mcp-server-mas-sequential-thinking 0.1.3__py3-none-any.whl → 0.2.1__py3-none-any.whl

main.py

@@ -4,11 +4,14 @@ import sys
 from contextlib import asynccontextmanager
 from dataclasses import dataclass, field
 from datetime import datetime
-from typing import Any, AsyncIterator, Dict, List, Optional
+from typing import Any, AsyncIterator, Dict, List, Optional, Type
 
 from mcp.server.fastmcp import FastMCP
 from agno.agent import Agent
+from agno.models.base import Model
 from agno.models.deepseek import DeepSeek
+from agno.models.groq import Groq
+from agno.models.openrouter import OpenRouter
 from agno.team.team import Team
 from agno.tools.exa import ExaTools
 from agno.tools.thinking import ThinkingTools
@@ -16,12 +19,12 @@ from dotenv import load_dotenv
 from pydantic import (BaseModel, ConfigDict, Field, ValidationError,
                       field_validator, model_validator)
 
-# Add logging imports and setup
 import logging
 import logging.handlers
 from pathlib import Path
 
-# Configure logging system
+load_dotenv()
+
 def setup_logging() -> logging.Logger:
     """
     Set up application logging with both file and console handlers.
@@ -66,14 +69,8 @@ def setup_logging() -> logging.Logger:
 
     return logger
 
-# Initialize logger
 logger = setup_logging()
 
-# Load environment variables from .env file
-load_dotenv()
-
-# --- Pydantic Model for Tool Input Schema ---
-
 class ThoughtData(BaseModel):
     """
     Represents the data structure for a single thought in the sequential
@@ -294,6 +291,44 @@ def format_thought_for_log(thought_data: ThoughtData) -> str:
 
 # --- Agno Multi-Agent Team Setup ---
 
+def get_model_config() -> tuple[Type[Model], str, str]:
+    """
+    Determines the LLM provider, team model ID, and agent model ID based on environment variables.
+
+    Returns:
+        A tuple containing:
+        - ModelClass: The Agno model class (e.g., DeepSeek, Groq, OpenRouter).
+        - team_model_id: The model ID for the team coordinator.
+        - agent_model_id: The model ID for the specialist agents.
+    """
+    provider = os.environ.get("LLM_PROVIDER", "deepseek").lower()
+    logger.info(f"Selected LLM Provider: {provider}")
+
+    if provider == "deepseek":
+        ModelClass = DeepSeek
+        # Use environment variables for DeepSeek model IDs if set, otherwise use defaults
+        team_model_id = os.environ.get("DEEPSEEK_TEAM_MODEL_ID", "deepseek-chat")
+        agent_model_id = os.environ.get("DEEPSEEK_AGENT_MODEL_ID", "deepseek-reasoner")
+        logger.info(f"Using DeepSeek: Team Model='{team_model_id}', Agent Model='{agent_model_id}'")
+    elif provider == "groq":
+        ModelClass = Groq
+        team_model_id = os.environ.get("GROQ_TEAM_MODEL_ID", "deepseek-r1-distill-llama-70b")
+        agent_model_id = os.environ.get("GROQ_AGENT_MODEL_ID", "qwen-2.5-32b")
+        logger.info(f"Using Groq: Team Model='{team_model_id}', Agent Model='{agent_model_id}'")
+    elif provider == "openrouter":
+        ModelClass = OpenRouter
+        team_model_id = os.environ.get("OPENROUTER_TEAM_MODEL_ID", "deepseek/deepseek-chat-v3-0324")
+        agent_model_id = os.environ.get("OPENROUTER_AGENT_MODEL_ID", "deepseek/deepseek-r1")
+        logger.info(f"Using OpenRouter: Team Model='{team_model_id}', Agent Model='{agent_model_id}'")
+    else:
+        logger.error(f"Unsupported LLM_PROVIDER: {provider}. Defaulting to DeepSeek.")
+        ModelClass = DeepSeek
+        team_model_id = "deepseek-chat"
+        agent_model_id = "deepseek-chat"
+
+    return ModelClass, team_model_id, agent_model_id
+
+
 def create_sequential_thinking_team() -> Team:
     """
     Creates and configures the Agno multi-agent team for sequential thinking,
@@ -303,12 +338,13 @@ def create_sequential_thinking_team() -> Team:
         An initialized Team instance.
     """
     try:
-        # Use a capable model for the team coordinator logic and specialists
-        # The Team itself needs a model for its coordination instructions.
-        team_model = DeepSeek(id="deepseek-chat")
+        ModelClass, team_model_id, agent_model_id = get_model_config()
+        team_model_instance = ModelClass(id=team_model_id)
+        agent_model_instance = ModelClass(id=agent_model_id)
+
     except Exception as e:
-        logger.error(f"Error initializing base model: {e}")
-        logger.error("Please ensure the necessary API keys and configurations are set.")
+        logger.error(f"Error initializing models: {e}")
+        logger.error("Please ensure the necessary API keys and configurations are set for the selected provider ({os.environ.get('LLM_PROVIDER', 'deepseek')}).")
         sys.exit(1)
 
     # REMOVED the separate Coordinator Agent definition.
@@ -334,7 +370,7 @@ def create_sequential_thinking_team() -> Team:
             " 7. Return your response to the Team Coordinator.",
             "Focus on fulfilling the delegated planning sub-task accurately and efficiently.",
         ],
-        model=team_model, # Specialists can use the same model or different ones
+        model=agent_model_instance, # Use the designated agent model
         add_datetime_to_instructions=True,
         markdown=True
     )
@@ -358,7 +394,7 @@ def create_sequential_thinking_team() -> Team:
             " 7. Return your response to the Team Coordinator.",
             "Focus on accuracy and relevance for the delegated research request.",
         ],
-        model=team_model,
+        model=agent_model_instance, # Use the designated agent model
         add_datetime_to_instructions=True,
         markdown=True
     )
@@ -382,7 +418,7 @@ def create_sequential_thinking_team() -> Team:
             " 7. Return your response to the Team Coordinator.",
             "Focus on depth and clarity for the delegated analytical task.",
         ],
-        model=team_model,
+        model=agent_model_instance, # Use the designated agent model
         add_datetime_to_instructions=True,
         markdown=True
     )
@@ -407,7 +443,7 @@ def create_sequential_thinking_team() -> Team:
             " 8. Return your response to the Team Coordinator.",
             "Focus on rigorous and constructive critique for the delegated evaluation task.",
         ],
-        model=team_model,
+        model=agent_model_instance, # Use the designated agent model
         add_datetime_to_instructions=True,
         markdown=True
     )
@@ -430,7 +466,7 @@ def create_sequential_thinking_team() -> Team:
             " 6. Return your response to the Team Coordinator.",
             "Focus on creating clarity and coherence for the delegated synthesis task.",
         ],
-        model=team_model,
+        model=agent_model_instance, # Use the designated agent model
         add_datetime_to_instructions=True,
         markdown=True
     )
@@ -441,7 +477,7 @@ def create_sequential_thinking_team() -> Team:
         name="SequentialThinkingTeam",
         mode="coordinate",
         members=[planner, researcher, analyzer, critic, synthesizer], # ONLY specialist agents
-        model=team_model, # Model for the Team's coordination logic
+        model=team_model_instance, # Model for the Team's coordination logic
         description="You are the Coordinator of a specialist team processing sequential thoughts. Your role is to manage the flow, delegate tasks, and synthesize results.",
         instructions=[
             f"Current date and time: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}",
@@ -515,9 +551,17 @@ async def app_lifespan() -> AsyncIterator[None]:
     """Manages the application lifecycle."""
     global app_context
     logger.info("Initializing application resources (Coordinate Mode)...")
-    team = create_sequential_thinking_team()
-    app_context = AppContext(team=team)
-    logger.info("Agno team initialized in coordinate mode.")
+    try:
+        team = create_sequential_thinking_team()
+        app_context = AppContext(team=team)
+        provider = os.environ.get("LLM_PROVIDER", "deepseek").lower()
+        logger.info(f"Agno team initialized in coordinate mode using provider: {provider}.")
+    except Exception as e:
+        logger.critical(f"Failed to initialize Agno team during lifespan setup: {e}", exc_info=True)
+        # Decide how to handle this - re-raise, exit, or continue without a team?
+        # For now, re-raise to prevent server starting in a broken state.
+        raise e
+
     try:
         yield
     finally:
@@ -598,7 +642,20 @@ async def sequentialthinking(thought: str, thoughtNumber: int, totalThoughts: in
     global app_context
     if not app_context or not app_context.team:
         logger.error("Application context or Agno team not initialized during tool call.")
-        raise Exception("Critical Error: Application context not available.")
+        # Attempt re-initialization cautiously, or fail hard.
+        # Let's try re-initialization if app_lifespan wasn't used or failed silently.
+        logger.warning("Attempting to re-initialize team due to missing context...")
+        try:
+             team = create_sequential_thinking_team()
+             app_context = AppContext(team=team) # Re-create context
+             logger.info("Successfully re-initialized team and context.")
+        except Exception as init_err:
+             logger.critical(f"Failed to re-initialize Agno team during tool call: {init_err}", exc_info=True)
+             return json.dumps({
+                 "error": "Critical Error: Application context not available and re-initialization failed.",
+                 "status": "critical_failure"
+             }, indent=2, ensure_ascii=False)
+             # Or raise Exception("Critical Error: Application context not available.")
 
     MIN_TOTAL_THOUGHTS = 5 # Keep a minimum suggestion
 
@@ -711,7 +768,7 @@ async def sequentialthinking(thought: str, thoughtNumber: int, totalThoughts: in
             "status": "success"
         }
 
-        return json.dumps(result_data, indent=2)
+        return json.dumps(result_data, indent=2, ensure_ascii=False)
 
     except ValidationError as e:
         logger.error(f"Validation Error processing tool call: {e}")
@@ -719,19 +776,21 @@ async def sequentialthinking(thought: str, thoughtNumber: int, totalThoughts: in
         return json.dumps({
             "error": f"Input validation failed: {e}",
             "status": "validation_error"
-        }, indent=2)
+        }, indent=2, ensure_ascii=False)
     except Exception as e:
         logger.exception(f"Error processing tool call") # Log full traceback
         return json.dumps({
             "error": f"An unexpected error occurred: {str(e)}",
             "status": "failed"
-        }, indent=2)
+        }, indent=2, ensure_ascii=False)
 
 # --- Main Execution ---
 
 def run():
     """Initializes and runs the MCP server in coordinate mode."""
-    logger.info("Initializing Sequential Thinking Server (Coordinate Mode)...")
+    selected_provider = os.environ.get("LLM_PROVIDER", "deepseek").lower()
+    logger.info(f"Using provider: {selected_provider}")
+    logger.info(f"Initializing Sequential Thinking Server (Coordinate Mode) with Provider: {selected_provider}...")
 
     global app_context
     # Initialize application resources using the lifespan manager implicitly if running via framework
@@ -742,7 +801,7 @@ def run():
         try:
              team = create_sequential_thinking_team()
              app_context = AppContext(team=team)
-             logger.info("Agno team initialized directly in coordinate mode.")
+             logger.info(f"Agno team initialized directly in coordinate mode using provider: {selected_provider}.")
         except Exception as e:
              logger.critical(f"Failed to initialize Agno team: {e}", exc_info=True)
              sys.exit(1)
@@ -759,27 +818,37 @@ def run():
         logger.info("Shutting down application resources...")
         app_context = None # Clean up context if initialized directly
 
-if __name__ == "__main__":
-    # Check necessary environment variables
-    if "DEEPSEEK_API_KEY" not in os.environ:
-        logger.warning("DEEPSEEK_API_KEY environment variable not found. Model initialization might fail.")
-
-    # Check EXA_API_KEY only if ExaTools are actually used by any member agent
+def check_environment_variables():
+    """Checks for necessary environment variables based on the selected provider."""
+    provider = os.environ.get("LLM_PROVIDER", "deepseek").lower()
+    api_key_var = ""
+    base_url_var = "" # Some providers might not strictly need a base URL override
+
+    if provider == "deepseek":
+        api_key_var = "DEEPSEEK_API_KEY"
+    elif provider == "groq":
+        api_key_var = "GROQ_API_KEY"
+    elif provider == "openrouter":
+        api_key_var = "OPENROUTER_API_KEY"
+    if api_key_var and api_key_var not in os.environ:
+        logger.warning(f"{api_key_var} environment variable not found. Model initialization for '{provider}' might fail.")
     try:
-        team_for_check = create_sequential_thinking_team() # Create temporarily to check tools
-        uses_exa = False
-        # Iterate through members to check for ExaTools
-        for member in team_for_check.members:
-            if hasattr(member, 'tools') and member.tools:
-                 if any(isinstance(t, ExaTools) for t in member.tools):
-                     uses_exa = True
-                     break # Found it, no need to check further
+        ModelClass, _, _ = get_model_config() # Just need the class for dummy init
+        dummy_model = ModelClass(id="dummy-check") # Use a placeholder ID
+        researcher_for_check = Agent(name="CheckAgent", tools=[ExaTools()], model=dummy_model)
+        uses_exa = any(isinstance(t, ExaTools) for t in researcher_for_check.tools)
 
         if uses_exa and "EXA_API_KEY" not in os.environ:
              logger.warning("EXA_API_KEY environment variable not found, but ExaTools are configured in a team member. Researcher agent might fail.")
+    except Exception as e:
+        logger.error(f"Could not perform ExaTools check due to an error: {e}")
 
-        # Run the application
+
+if __name__ == "__main__":
+    check_environment_variables()
+    try:
         run()
     except Exception as e:
-        logger.critical(f"Failed during initial setup or checks: {e}", exc_info=True)
+        logger.critical(f"Failed during server run: {e}", exc_info=True)
         sys.exit(1)
+        

{mcp_server_mas_sequential_thinking-0.1.3.dist-info → mcp_server_mas_sequential_thinking-0.2.1.dist-info}/METADATA

@@ -1,12 +1,13 @@
 Metadata-Version: 2.4
 Name: mcp-server-mas-sequential-thinking
-Version: 0.1.3
+Version: 0.2.1
 Summary: MCP Agent Implementation for Sequential Thinking
 Author-email: Frad LEE <fradser@gmail.com>
 Requires-Python: >=3.10
 Requires-Dist: agno
 Requires-Dist: asyncio
 Requires-Dist: exa-py
+Requires-Dist: groq
 Requires-Dist: mcp
 Requires-Dist: python-dotenv
 Provides-Extra: dev
@@ -80,8 +81,11 @@ This parallel processing leads to substantially higher token usage (potentially
 ## Prerequisites
 
 * Python 3.10+
-* Access to a compatible LLM API (configured for `agno`, e.g., DeepSeek)
-    * `DEEPSEEK_API_KEY` environment variable.
+* Access to a compatible LLM API (configured for `agno`). The system now supports:
+    * **Groq:** Requires `GROQ_API_KEY`.
+    * **DeepSeek:** Requires `DEEPSEEK_API_KEY`.
+    * **OpenRouter:** Requires `OPENROUTER_API_KEY`.
+    * Configure the desired provider using the `LLM_PROVIDER` environment variable (defaults to `deepseek`).
 * Exa API Key (if using the Researcher agent's capabilities)
     * `EXA_API_KEY` environment variable.
 * `uv` package manager (recommended) or `pip`.
@@ -90,7 +94,9 @@ This parallel processing leads to substantially higher token usage (potentially
 
 This server runs as a standard executable script that communicates via stdio, as expected by MCP. The exact configuration method depends on your specific MCP client implementation. Consult your client's documentation for details.
 
-```
+The `env` section should include the API key for your chosen `LLM_PROVIDER`.
+
+```json
 {
   "mcpServers": {
       "mas-sequential-thinking": {
@@ -98,14 +104,17 @@ This server runs as a standard executable script that communicates via stdio, as
          "args": [
             "mcp-server-mas-sequential-thinking"
          ],
-         env": {
-            "DEEPSEEK_API_KEY": "your_deepseek_api_key",
-            "DEEPSEEK_BASE_URL": "your_base_url_if_needed", # Optional: If using a custom endpoint
-            "EXA_API_KEY": "your_exa_api_key"
+         "env": {
+            "LLM_PROVIDER": "deepseek", // Or "groq", "openrouter"
+            // "GROQ_API_KEY": "your_groq_api_key", // Only if LLM_PROVIDER="groq"
+            "DEEPSEEK_API_KEY": "your_deepseek_api_key", // Default provider
+            // "OPENROUTER_API_KEY": "your_openrouter_api_key", // Only if LLM_PROVIDER="openrouter"
+            "DEEPSEEK_BASE_URL": "your_base_url_if_needed", // Optional: If using a custom endpoint for DeepSeek
+            "EXA_API_KEY": "your_exa_api_key" // Only if using Exa
          }
       }
    }
-}  
+}
 ```
 
 ## Installation & Setup
@@ -119,14 +128,41 @@ This server runs as a standard executable script that communicates via stdio, as
 2.  **Set Environment Variables:**
     Create a `.env` file in the root directory or export the variables:
     ```dotenv
-    # Required for the LLM used by Agno agents/team
-    DEEPSEEK_API_KEY="your_deepseek_api_key"
-    # DEEPSEEK_BASE_URL="your_base_url_if_needed" # Optional: If using a custom endpoint
+    # --- LLM Configuration ---
+    # Select the LLM provider: "deepseek" (default), "groq", or "openrouter"
+    LLM_PROVIDER="deepseek"
 
+    # Provide the API key for the chosen provider:
+    # GROQ_API_KEY="your_groq_api_key"
+    DEEPSEEK_API_KEY="your_deepseek_api_key"
+    # OPENROUTER_API_KEY="your_openrouter_api_key"
+
+    # Optional: Base URL override (e.g., for custom DeepSeek endpoints)
+    DEEPSEEK_BASE_URL="your_base_url_if_needed"
+
+    # Optional: Specify different models for Team Coordinator and Specialist Agents
+    # Defaults are set within the code based on the provider if these are not set.
+    # Example for Groq:
+    # GROQ_TEAM_MODEL_ID="llama3-70b-8192"
+    # GROQ_AGENT_MODEL_ID="llama3-8b-8192"
+    # Example for DeepSeek:
+    # DEEPSEEK_TEAM_MODEL_ID="deepseek-reasoner" # Recommended for coordination
+    # DEEPSEEK_AGENT_MODEL_ID="deepseek-chat"     # Recommended for specialists
+    # Example for OpenRouter:
+    # OPENROUTER_TEAM_MODEL_ID="anthropic/claude-3-haiku-20240307"
+    # OPENROUTER_AGENT_MODEL_ID="google/gemini-flash-1.5"
+
+    # --- External Tools ---
     # Required ONLY if the Researcher agent is used and needs Exa
     EXA_API_KEY="your_exa_api_key"
     ```
 
+    **Note on Model Selection:**
+
+    *   The `TEAM_MODEL_ID` is used by the Coordinator (the `Team` object itself). This role requires strong reasoning, synthesis, and delegation capabilities. Using a more powerful model (like `deepseek-reasoner`, `claude-3-opus`, or `gpt-4-turbo`) is often beneficial here, even if it's slower or more expensive.
+    *   The `AGENT_MODEL_ID` is used by the specialist agents (Planner, Researcher, etc.). These agents handle more focused sub-tasks. You might choose a faster or more cost-effective model (like `deepseek-chat`, `claude-3-sonnet`, `llama3-70b`) for specialists, depending on the complexity of the tasks they typically handle and your budget/performance requirements.
+    *   The defaults provided in `main.py` (e.g., `deepseek-reasoner` for agents when using DeepSeek) are starting points. Experimentation is encouraged to find the optimal balance for your specific use case.
+
 3.  **Install Dependencies:**
 
     * **Using `uv` (Recommended):**

mcp_server_mas_sequential_thinking-0.2.1.dist-info/RECORD

@@ -0,0 +1,5 @@
+main.py,sha256=Vm6SBMDmvFy9CwEmqI-ZqZ0YDLUgPA_E689La_Qc4Yo,44071
+mcp_server_mas_sequential_thinking-0.2.1.dist-info/METADATA,sha256=bIJGHFTRT2hldImbnA0xKyp4i8BRcc78WihpJAcMf5A,15807
+mcp_server_mas_sequential_thinking-0.2.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mcp_server_mas_sequential_thinking-0.2.1.dist-info/entry_points.txt,sha256=wY2jq_6PmuqyKQzNnL6famc7DXnQiEhVnq3umzNVNiE,64
+mcp_server_mas_sequential_thinking-0.2.1.dist-info/RECORD,,

mcp_server_mas_sequential_thinking-0.1.3.dist-info/RECORD

@@ -1,5 +0,0 @@
-main.py,sha256=mDC2ayNFA5ON0v4PMhpWijIHxlk_5h7ak8x6shcqORQ,40049
-mcp_server_mas_sequential_thinking-0.1.3.dist-info/METADATA,sha256=XHTnbZ9kt63Lc5g_tgy6wCbgIup0Owy0kwwHffT7yNs,13349
-mcp_server_mas_sequential_thinking-0.1.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mcp_server_mas_sequential_thinking-0.1.3.dist-info/entry_points.txt,sha256=wY2jq_6PmuqyKQzNnL6famc7DXnQiEhVnq3umzNVNiE,64
-mcp_server_mas_sequential_thinking-0.1.3.dist-info/RECORD,,