fast-agent-mcp 0.2.29__tar.gz → 0.2.30__tar.gz

{fast_agent_mcp-0.2.29 → fast_agent_mcp-0.2.30}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fast-agent-mcp
-Version: 0.2.29
+Version: 0.2.30
 Summary: Define, Prompt and Test MCP enabled Agents and Workflows
 Author-email: Shaun Smith <fastagent@llmindset.co.uk>
 License:                                  Apache License

{fast_agent_mcp-0.2.29 → fast_agent_mcp-0.2.30}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "fast-agent-mcp"
-version = "0.2.29"
+version = "0.2.30"
 description = "Define, Prompt and Test MCP enabled Agents and Workflows"
 readme = "README.md"
 license = { file = "LICENSE" }

{fast_agent_mcp-0.2.29 → fast_agent_mcp-0.2.30}/src/mcp_agent/cli/commands/check_config.py

@@ -226,8 +226,17 @@ def get_config_summary(config_path: Optional[Path]) -> dict:
 
                 # Determine transport type
                 if "url" in server_config:
-                    server_info["transport"] = "SSE"
-                    server_info["url"] = server_config.get("url", "")
+                    url = server_config.get("url", "")
+                    server_info["url"] = url
+                    
+                    # Use URL path to determine transport type
+                    try:
+                        from .url_parser import parse_server_url
+                        _, transport_type, _ = parse_server_url(url)
+                        server_info["transport"] = transport_type.upper()
+                    except Exception:
+                        # Fallback to HTTP if URL parsing fails
+                        server_info["transport"] = "HTTP"
 
                 # Get command and args
                 command = server_config.get("command", "")

{fast_agent_mcp-0.2.29 → fast_agent_mcp-0.2.30}/src/mcp_agent/config.py

@@ -222,6 +222,15 @@ class TensorZeroSettings(BaseModel):
     model_config = ConfigDict(extra="allow", arbitrary_types_allowed=True)
 
 
+class HuggingFaceSettings(BaseModel):
+    """
+    Settings for HuggingFace authentication (used for MCP connections).
+    """
+
+    api_key: Optional[str] = None
+    model_config = ConfigDict(extra="allow", arbitrary_types_allowed=True)
+
+
 class LoggerSettings(BaseModel):
     """
     Logger settings for the fast-agent application.
@@ -291,7 +300,7 @@ class Settings(BaseSettings):
     Default model for agents. Format is provider.model_name.<reasoning_effort>, for example openai.o3-mini.low
     Aliases are provided for common models e.g. sonnet, haiku, gpt-4.1, o3-mini etc.
     """
-    
+
     auto_sampling: bool = True
     """Enable automatic sampling model selection if not explicitly configured"""
 
@@ -322,6 +331,12 @@ class Settings(BaseSettings):
     azure: AzureSettings | None = None
     """Settings for using Azure OpenAI Service in the fast-agent application"""
 
+    aliyun: OpenAISettings | None = None
+    """Settings for using Aliyun OpenAI Service in the fast-agent application"""
+
+    huggingface: HuggingFaceSettings | None = None
+    """Settings for HuggingFace authentication (used for MCP connections)"""
+
     logger: LoggerSettings | None = LoggerSettings()
     """Logger settings for the fast-agent application"""
 

{fast_agent_mcp-0.2.29 → fast_agent_mcp-0.2.30}/src/mcp_agent/core/enhanced_prompt.py

@@ -2,6 +2,11 @@
 Enhanced prompt functionality with advanced prompt_toolkit features.
 """
 
+import asyncio
+import os
+import shlex
+import subprocess
+import tempfile
 from importlib.metadata import version
 from typing import List, Optional
 
@@ -96,6 +101,85 @@ class AgentCompleter(Completer):
                     )
 
 
+# Helper function to open text in an external editor
+def get_text_from_editor(initial_text: str = "") -> str:
+    """
+    Opens the user\'s configured editor ($VISUAL or $EDITOR) to edit the initial_text.
+    Falls back to \'nano\' (Unix) or \'notepad\' (Windows) if neither is set.
+    Returns the edited text, or the original text if an error occurs.
+    """
+    editor_cmd_str = os.environ.get("VISUAL") or os.environ.get("EDITOR")
+
+    if not editor_cmd_str:
+        if os.name == "nt":  # Windows
+            editor_cmd_str = "notepad"
+        else:  # Unix-like (Linux, macOS)
+            editor_cmd_str = "nano"  # A common, usually available, simple editor
+
+    # Use shlex.split to handle editors with arguments (e.g., "code --wait")
+    try:
+        editor_cmd_list = shlex.split(editor_cmd_str)
+        if not editor_cmd_list:  # Handle empty string from shlex.split
+            raise ValueError("Editor command string is empty or invalid.")
+    except ValueError as e:
+        rich_print(f"[red]Error: Invalid editor command string ('{editor_cmd_str}'): {e}[/red]")
+        return initial_text
+
+    # Create a temporary file for the editor to use.
+    # Using a suffix can help some editors with syntax highlighting or mode.
+    try:
+        with tempfile.NamedTemporaryFile(
+            mode="w+", delete=False, suffix=".txt", encoding="utf-8"
+        ) as tmp_file:
+            if initial_text:
+                tmp_file.write(initial_text)
+                tmp_file.flush()  # Ensure content is written to disk before editor opens it
+            temp_file_path = tmp_file.name
+    except Exception as e:
+        rich_print(f"[red]Error: Could not create temporary file for editor: {e}[/red]")
+        return initial_text
+
+    try:
+        # Construct the full command: editor_parts + [temp_file_path]
+        # e.g., [\'vim\', \'/tmp/somefile.txt\'] or [\'code\', \'--wait\', \'/tmp/somefile.txt\']
+        full_cmd = editor_cmd_list + [temp_file_path]
+
+        # Run the editor. This is a blocking call.
+        subprocess.run(full_cmd, check=True)
+
+        # Read the content back from the temporary file.
+        with open(temp_file_path, "r", encoding="utf-8") as f:
+            edited_text = f.read()
+
+    except FileNotFoundError:
+        rich_print(
+            f"[red]Error: Editor command '{editor_cmd_list[0]}' not found. "
+            f"Please set $VISUAL or $EDITOR correctly, or install '{editor_cmd_list[0]}'.[/red]"
+        )
+        return initial_text
+    except subprocess.CalledProcessError as e:
+        rich_print(
+            f"[red]Error: Editor '{editor_cmd_list[0]}' closed with an error (code {e.returncode}).[/red]"
+        )
+        return initial_text
+    except Exception as e:
+        rich_print(
+            f"[red]An unexpected error occurred while launching or using the editor: {e}[/red]"
+        )
+        return initial_text
+    finally:
+        # Always attempt to clean up the temporary file.
+        if "temp_file_path" in locals() and os.path.exists(temp_file_path):
+            try:
+                os.remove(temp_file_path)
+            except Exception as e:
+                rich_print(
+                    f"[yellow]Warning: Could not remove temporary file {temp_file_path}: {e}[/yellow]"
+                )
+
+    return edited_text.strip()  # Added strip() to remove trailing newlines often added by editors
+
+
 def create_keybindings(on_toggle_multiline=None, app=None):
     """Create custom key bindings."""
     kb = KeyBindings()
@@ -140,6 +224,27 @@ def create_keybindings(on_toggle_multiline=None, app=None):
         """Ctrl+L: Clear the input buffer."""
         event.current_buffer.text = ""
 
+    @kb.add("c-e")
+    async def _(event) -> None:
+        """Ctrl+E: Edit current buffer in $EDITOR."""
+        current_text = event.app.current_buffer.text
+        try:
+            # Run the synchronous editor function in a thread
+            edited_text = await event.app.loop.run_in_executor(
+                None, get_text_from_editor, current_text
+            )
+            event.app.current_buffer.text = edited_text
+            # Optionally, move cursor to the end of the edited text
+            event.app.current_buffer.cursor_position = len(edited_text)
+        except asyncio.CancelledError:
+            rich_print("[yellow]Editor interaction cancelled.[/yellow]")
+        except Exception as e:
+            rich_print(f"[red]Error during editor interaction: {e}[/red]")
+        finally:
+            # Ensure the UI is updated
+            if event.app:
+                event.app.invalidate()
+
     return kb
 
 

{fast_agent_mcp-0.2.29 → fast_agent_mcp-0.2.30}/src/mcp_agent/llm/model_factory.py

@@ -10,6 +10,7 @@ from mcp_agent.llm.augmented_llm_passthrough import PassthroughLLM
 from mcp_agent.llm.augmented_llm_playback import PlaybackLLM
 from mcp_agent.llm.augmented_llm_slow import SlowLLM
 from mcp_agent.llm.provider_types import Provider
+from mcp_agent.llm.providers.augmented_llm_aliyun import AliyunAugmentedLLM
 from mcp_agent.llm.providers.augmented_llm_anthropic import AnthropicAugmentedLLM
 from mcp_agent.llm.providers.augmented_llm_azure import AzureOpenAIAugmentedLLM
 from mcp_agent.llm.providers.augmented_llm_deepseek import DeepSeekAugmentedLLM
@@ -103,6 +104,10 @@ class ModelFactory:
         "gemini-2.0-flash": Provider.GOOGLE,
         "gemini-2.5-flash-preview-05-20": Provider.GOOGLE,
         "gemini-2.5-pro-preview-05-06": Provider.GOOGLE,
+        "qwen-turbo": Provider.ALIYUN,
+        "qwen-plus": Provider.ALIYUN,
+        "qwen-max": Provider.ALIYUN,
+        "qwen-long": Provider.ALIYUN,
     }
 
     MODEL_ALIASES = {
@@ -136,6 +141,7 @@ class ModelFactory:
         Provider.OPENROUTER: OpenRouterAugmentedLLM,
         Provider.TENSORZERO: TensorZeroAugmentedLLM,
         Provider.AZURE: AzureOpenAIAugmentedLLM,
+        Provider.ALIYUN: AliyunAugmentedLLM,
     }
 
     # Mapping of special model names to their specific LLM classes

{fast_agent_mcp-0.2.29 → fast_agent_mcp-0.2.30}/src/mcp_agent/llm/provider_key_manager.py

@@ -17,6 +17,7 @@ PROVIDER_ENVIRONMENT_MAP: Dict[str, str] = {
     "google": "GOOGLE_API_KEY",
     "openrouter": "OPENROUTER_API_KEY",
     "generic": "GENERIC_API_KEY",
+    "huggingface": "HF_TOKEN",
 }
 API_KEY_HINT_TEXT = "<your-api-key-here>"
 

{fast_agent_mcp-0.2.29 → fast_agent_mcp-0.2.30}/src/mcp_agent/llm/provider_types.py

@@ -18,3 +18,5 @@ class Provider(Enum):
     OPENROUTER = "openrouter"
     TENSORZERO = "tensorzero"  # For TensorZero Gateway
     AZURE = "azure"  # Azure OpenAI Service
+    ALIYUN = "aliyun"  # Aliyun Bailian OpenAI Service
+    HUGGINGFACE = "huggingface"  # For HuggingFace MCP connections

fast_agent_mcp-0.2.30/src/mcp_agent/llm/providers/augmented_llm_aliyun.py

@@ -0,0 +1,30 @@
+from mcp_agent.core.request_params import RequestParams
+from mcp_agent.llm.provider_types import Provider
+from mcp_agent.llm.providers.augmented_llm_openai import OpenAIAugmentedLLM
+
+ALIYUN_BASE_URL = "https://dashscope.aliyuncs.com/compatible-mode/v1"
+DEFAULT_QWEN_MODEL = "qwen-turbo"
+
+
+class AliyunAugmentedLLM(OpenAIAugmentedLLM):
+    def __init__(self, *args, **kwargs) -> None:
+        super().__init__(*args, provider=Provider.ALIYUN, **kwargs)
+
+    def _initialize_default_params(self, kwargs: dict) -> RequestParams:
+        """Initialize Aliyun-specific default parameters"""
+        chosen_model = kwargs.get("model", DEFAULT_QWEN_MODEL)
+
+        return RequestParams(
+            model=chosen_model,
+            systemPrompt=self.instruction,
+            parallel_tool_calls=True,
+            max_iterations=10,
+            use_history=True,
+        )
+
+    def _base_url(self) -> str:
+        base_url = None
+        if self.context.config and self.context.config.aliyun:
+            base_url = self.context.config.aliyun.base_url
+
+        return base_url if base_url else ALIYUN_BASE_URL

{fast_agent_mcp-0.2.29 → fast_agent_mcp-0.2.30}/src/mcp_agent/llm/providers/augmented_llm_deepseek.py

@@ -1,4 +1,10 @@
-from typing import List, Tuple, Type
+from copy import copy
+from typing import List, Tuple, Type, cast
+
+from openai.types.chat import (
+    ChatCompletionAssistantMessageParam,
+    ChatCompletionMessage,
+)
 
 from mcp_agent.core.request_params import RequestParams
 from mcp_agent.llm.provider_types import Provider
@@ -77,3 +83,11 @@ class DeepSeekAugmentedLLM(OpenAIAugmentedLLM):
             multipart_messages, request_params
         )
         return self._structured_from_multipart(result, model)
+
+    @classmethod
+    def convert_message_to_message_param(cls, message: ChatCompletionMessage, **kwargs) -> ChatCompletionAssistantMessageParam:
+        """Convert a response object to an input parameter object to allow LLM calls to be chained."""
+        if hasattr(message, "reasoning_content"):
+            message = copy(message)
+            del message.reasoning_content
+        return cast("ChatCompletionAssistantMessageParam", message)

{fast_agent_mcp-0.2.29 → fast_agent_mcp-0.2.30}/src/mcp_agent/mcp/hf_auth.py

@@ -22,7 +22,26 @@ def is_huggingface_url(url: str) -> bool:
             return False
         
         # Check for HuggingFace domains
-        return hostname in {"hf.co", "huggingface.co"}
+        if hostname in {"hf.co", "huggingface.co"}:
+            return True
+            
+        # Check for HuggingFace Spaces (*.hf.space)
+        # Use endswith to match subdomains like space-name.hf.space
+        # but ensure exact match to prevent spoofing like evil.hf.space.com
+        if hostname.endswith(".hf.space") and hostname.count(".") >= 2:
+            # Additional validation: ensure it's a valid HF Space domain
+            # Format should be: {space-name}.hf.space
+            parts = hostname.split(".")
+            if len(parts) == 3 and parts[-2:] == ["hf", "space"]:
+                space_name = parts[0]
+                # Validate space name: not empty, not just hyphens/dots, no spaces
+                return (len(space_name) > 0 and 
+                        space_name != "-" and 
+                        not space_name.startswith(".") and
+                        not space_name.endswith(".") and
+                        " " not in space_name)
+            
+        return False
     except Exception:
         return False
 

{fast_agent_mcp-0.2.29 → fast_agent_mcp-0.2.30}/src/mcp_agent/mcp/mcp_aggregator.py

@@ -82,13 +82,14 @@ class MCPAggregator(ContextDependent):
                 await self.context._connection_manager.__aenter__()
             self._persistent_connection_manager = self.context._connection_manager
 
-        await self.load_servers()
         # Import the display component here to avoid circular imports
         from mcp_agent.ui.console_display import ConsoleDisplay
 
         # Initialize the display component
         self.display = ConsoleDisplay(config=self.context.config)
 
+        await self.load_servers()
+
         return self
 
     async def __aexit__(self, exc_type, exc_val, exc_tb):