0.1.3

Author: arabot777

.gitignore

@@ -102,3 +102,4 @@ wheels/
 *.bak
 *.swp
 *~.nib
+example/

README.md

@@ -80,7 +80,7 @@ WavespeedMCP can be configured through:
    - `WAVESPEED_MCP_BASE_PATH`: Base path for output files (default: ~/Desktop)
    - `WAVESPEED_API_RESOURCE_MODE`: Resource output mode (options: url, base64, local; default: url)
    - `WAVESPEED_LOG_LEVEL`: Logging level (options: DEBUG, INFO, WARNING, ERROR; default: INFO)
-   - `WAVESPEED_API_TEXT_TO_IMAGE_ENDPOINT`: Custom endpoint for text-to-image generation (default: /wavespeed-ai/flux-dev-lora)
+   - `WAVESPEED_API_TEXT_TO_IMAGE_ENDPOINT`: Custom endpoint for text-to-image generation (default: /wavespeed-ai/flux-dev)
    - `WAVESPEED_API_IMAGE_TO_IMAGE_ENDPOINT`: Custom endpoint for image-to-image generation (default: /wavespeed-ai/flux-kontext-pro)
    - `WAVESPEED_API_VIDEO_ENDPOINT`: Custom endpoint for video generation (default: /wavespeed-ai/wan-2.1/i2v-480p-lora)
 

README.zh.md

@@ -78,7 +78,7 @@ WavespeedMCP 可以通过以下方式进行配置：
    - `WAVESPEED_MCP_BASE_PATH`：输出文件的基本路径（默认：~/Desktop）
    - `WAVESPEED_API_RESOURCE_MODE`：资源输出模式（选项：url、base64、local；默认：url）
    - `WAVESPEED_LOG_LEVEL`：日志级别（选项：DEBUG、INFO、WARNING、ERROR；默认：INFO）
-   - `WAVESPEED_API_TEXT_TO_IMAGE_ENDPOINT`：文本生成图像的自定义端点（默认：/wavespeed-ai/flux-dev-lora）
+   - `WAVESPEED_API_TEXT_TO_IMAGE_ENDPOINT`：文本生成图像的自定义端点（默认：/wavespeed-ai/flux-dev）
    - `WAVESPEED_API_IMAGE_TO_IMAGE_ENDPOINT`：图像编辑的自定义端点（默认：/wavespeed-ai/flux-kontext-pro）
    - `WAVESPEED_API_VIDEO_ENDPOINT`：视频生成的自定义端点（默认：/wavespeed-ai/wan-2.1/i2v-480p-lora）
 

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "wavespeed-mcp"
-version = "0.1.2"
+version = "0.1.3"
 description = "WaveSpeed MCP Server"
 authors = [
     { name = "Wavespeed AI Team", email = "support@wavespeed.ai" },

tests/test_integration.py

@@ -56,7 +56,7 @@ def test_generate_image(self):
         }
 
         print("Calling API with params:", params)
-        result = asyncio.run(self._run_client("generate_image", params))
+        result = asyncio.run(self._run_client("text_to_image", params))
         print("Received API response: ", result)
 
         # Verify result

wavespeed_mcp/client.py

@@ -47,7 +47,9 @@ def _make_request(self, method: str, endpoint: str, **kwargs) -> Dict[str, Any]:
             WavespeedAuthError: If authentication fails
             WavespeedRequestError: If the request fails
         """
-        url = f"{self.api_host}{endpoint}"
+        host = self.api_host.rstrip('/')
+        path = endpoint if endpoint.startswith('/') else f'/{endpoint}'
+        url = f"{host}{path}"
 
         logger.debug(f"Making {method} request to {url}")
 

wavespeed_mcp/const.py

@@ -34,7 +34,7 @@
 # API endpoints
 API_VERSION = "v3"
 API_BASE_PATH = "/api"
-API_IMAGE_ENDPOINT = "/wavespeed-ai/flux-dev-lora"
+API_IMAGE_ENDPOINT = "/wavespeed-ai/flux-dev"
 API_IMAGE_TO_IMAGE_ENDPOINT = "/wavespeed-ai/flux-kontext-pro"
 API_VIDEO_ENDPOINT = "/wavespeed-ai/wan-2.1/i2v-480p-lora"
 ENV_API_TEXT_TO_IMAGE_ENDPOINT = "WAVESPEED_API_TEXT_TO_IMAGE_ENDPOINT"

wavespeed_mcp/server.py

@@ -13,7 +13,6 @@
 import json
 import logging
 from typing import Dict, List, Optional, Union
-from pathlib import Path
 from pydantic import BaseModel
 from dotenv import load_dotenv
 from mcp.server.fastmcp import FastMCP
@@ -71,7 +70,7 @@
 # Get configuration from environment variables
 api_key = os.getenv(ENV_WAVESPEED_API_KEY)
 api_host = os.getenv(ENV_WAVESPEED_API_HOST, "https://api.wavespeed.ai")
-base_path = os.getenv(ENV_WAVESPEED_MCP_BASE_PATH) or str(Path.home() / "Desktop")
+base_path = os.getenv(ENV_WAVESPEED_MCP_BASE_PATH) or "~/Desktop"
 resource_mode = os.getenv(ENV_RESOURCE_MODE, RESOURCE_MODE_URL)
 
 # Validate required environment variables