mcp-server-mas-sequential-thinking 0.1.3__tar.gz → 0.2.0__tar.gz

mcp_server_mas_sequential_thinking-0.2.0/.env.example

@@ -0,0 +1,27 @@
+# --- 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-chat"
+# DEEPSEEK_AGENT_MODEL_ID="deepseek-coder"
+# 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"

{mcp_server_mas_sequential_thinking-0.1.3 → mcp_server_mas_sequential_thinking-0.2.0}/PKG-INFO

@@ -1,12 +1,13 @@
 Metadata-Version: 2.4
 Name: mcp-server-mas-sequential-thinking
-Version: 0.1.3
+Version: 0.2.0
 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,10 +128,31 @@ 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-chat"
+    # DEEPSEEK_AGENT_MODEL_ID="deepseek-coder"
+    # 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"
     ```

{mcp_server_mas_sequential_thinking-0.1.3 → mcp_server_mas_sequential_thinking-0.2.0}/README.md

@@ -62,8 +62,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`.
@@ -72,7 +75,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": {
@@ -80,14 +85,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
@@ -101,10 +109,31 @@ 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-chat"
+    # DEEPSEEK_AGENT_MODEL_ID="deepseek-coder"
+    # 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"
     ```

{mcp_server_mas_sequential_thinking-0.1.3 → mcp_server_mas_sequential_thinking-0.2.0}/README.zh-CN.md

@@ -63,8 +63,11 @@
 ## 先决条件
 
   * Python 3.10+
-  * 访问兼容的 LLM API（为 `agno` 配置，例如 DeepSeek）
-      * `DEEPSEEK_API_KEY` 环境变量。
+  * 访问兼容的 LLM API（为 `agno` 配置）。系统现在支持：
+      * **Groq:** 需要 `GROQ_API_KEY`。
+      * **DeepSeek:** 需要 `DEEPSEEK_API_KEY`。
+      * **OpenRouter:** 需要 `OPENROUTER_API_KEY`。
+      * 使用 `LLM_PROVIDER` 环境变量配置所需的提供商（默认为 `deepseek`）。
   * Exa API 密钥（如果使用研究员智能体的功能）
       * `EXA_API_KEY` 环境变量。
   * `uv` 包管理器（推荐）或 `pip`。
@@ -73,6 +76,8 @@
 
 此服务器作为标准可执行脚本运行，通过 stdio 进行通信，符合 MCP 的预期。确切的配置方法取决于您具体的 MCP 客户端实现。请查阅您客户端的文档以获取详细信息。
 
+`env` 部分应包含您选择的 `LLM_PROVIDER` 对应的 API 密钥。
+
 ```json
 {
   "mcpServers": {
@@ -82,9 +87,12 @@
         "mcp-server-mas-sequential-thinking"
       ],
       "env": {
-        "DEEPSEEK_API_KEY": "你的_deepseek_api_密钥",
-        "DEEPSEEK_BASE_URL": "你的_base_url_如果需要", // 可选：如果使用自定义端点
-        "EXA_API_KEY": "你的_exa_api_密钥"
+        "LLM_PROVIDER": "deepseek", // 或 "groq", "openrouter"
+        // "GROQ_API_KEY": "你的_groq_api_密钥", // 仅当 LLM_PROVIDER="groq" 时需要
+        "DEEPSEEK_API_KEY": "你的_deepseek_api_密钥", // 默认提供商
+        // "OPENROUTER_API_KEY": "你的_openrouter_api_密钥", // 仅当 LLM_PROVIDER="openrouter" 时需要
+        "DEEPSEEK_BASE_URL": "你的_base_url_如果需要", // 可选：如果为 DeepSeek 使用自定义端点
+        "EXA_API_KEY": "你的_exa_api_密钥" // 仅当使用 Exa 时需要
       }
     }
   }
@@ -104,10 +112,31 @@
     在根目录创建一个 `.env` 文件或导出变量：
 
     ```dotenv
-    # Agno 智能体/团队使用的 LLM 所必需
-    DEEPSEEK_API_KEY="你的_deepseek_api_密钥"
-    # DEEPSEEK_BASE_URL="你的_base_url_如果需要" # 可选：如果使用自定义端点
+    # --- LLM 配置 ---
+    # 选择 LLM 提供商: "deepseek" (默认), "groq", 或 "openrouter"
+    LLM_PROVIDER="deepseek"
 
+    # 提供所选提供商的 API 密钥:
+    # GROQ_API_KEY="你的_groq_api_密钥"
+    DEEPSEEK_API_KEY="你的_deepseek_api_密钥"
+    # OPENROUTER_API_KEY="你的_openrouter_api_密钥"
+
+    # 可选: 基础 URL 覆盖 (例如, 用于自定义 DeepSeek 端点)
+    DEEPSEEK_BASE_URL="你的_base_url_如果需要"
+
+    # 可选: 为团队协调器和专家智能体指定不同的模型
+    # 如果未设置这些变量，则代码会根据提供商设置默认值。
+    # Groq 示例:
+    # GROQ_TEAM_MODEL_ID="llama3-70b-8192"
+    # GROQ_AGENT_MODEL_ID="llama3-8b-8192"
+    # DeepSeek 示例:
+    # DEEPSEEK_TEAM_MODEL_ID="deepseek-chat"
+    # DEEPSEEK_AGENT_MODEL_ID="deepseek-coder"
+    # OpenRouter 示例:
+    # OPENROUTER_TEAM_MODEL_ID="anthropic/claude-3-haiku-20240307"
+    # OPENROUTER_AGENT_MODEL_ID="google/gemini-flash-1.5"
+
+    # --- 外部工具 ---
     # 仅当研究员智能体被使用且需要 Exa 时才必需
     EXA_API_KEY="你的_exa_api_密钥"
     ```
@@ -165,64 +194,8 @@ python 你的主脚本名称.py
 LLM 会迭代地与此工具交互：
 
 1.  **LLM:** 使用 `sequential-thinking-starter` 提示和问题。
-2.  **LLM:** 使用 `thoughtNumber: 1`、初始 `thought`（例如，“规划分析...”）、`totalThoughts` 预估、`nextThoughtNeeded: True` 调用 `sequentialthinking` 工具。
-3.  **服务器:** MAS 处理思考 -\> 协调器综合响应并提供指导（例如，“分析计划完成。建议下一步研究 X。暂不推荐修订。”）。
+2.  **LLM:** 使用 `thoughtNumber: 1`、初始 `thought`（例如，"规划分析..."）、`totalThoughts` 预估、`nextThoughtNeeded: True` 调用 `sequentialthinking` 工具。
+3.  **服务器:** MAS 处理思考 -\> 协调器综合响应并提供指导（例如，"分析计划完成。建议下一步研究 X。暂不推荐修订。"）。
 4.  **LLM:** 接收包含 `coordinatorResponse` 的 JSON 响应。
-5.  **LLM:** 根据 `coordinatorResponse` 构思下一个思考（例如，“使用 Exa 研究 X...”）。
-6.  **LLM:** 使用 `thoughtNumber: 2`、新的 `thought`、更新的 `totalThoughts`（如果需要）、`nextThoughtNeeded: True` 调用 `sequentialthinking` 工具。
-7.  **服务器:** MAS 处理 -\> 协调器综合（例如，“研究完成。发现表明思考 \#1 的假设存在缺陷。建议：修订思考 \#1...”）。
-8.  **LLM:** 接收响应，看到建议。
-9.  **LLM:** 构思一个修订思考。
-10. **LLM:** 使用 `thoughtNumber: 3`、修订后的 `thought`、`isRevision: True`、`revisesThought: 1`、`nextThoughtNeeded: True` 调用 `sequentialthinking` 工具。
-11. **... 以此类推，可能根据需要进行分支或扩展。**
-
-### 工具响应格式
-
-该工具返回一个 JSON 字符串，包含：
-
-```json
-{
-  "processedThoughtNumber": int,          // 处理的思考编号
-  "estimatedTotalThoughts": int,          // 预估总思考数
-  "nextThoughtNeeded": bool,              // 是否需要下一个思考
-  "coordinatorResponse": "来自智能体团队的综合输出，包括分析、发现和下一步指导...", // 协调器的综合响应
-  "branches": ["分支ID列表"],              // 所有分支 ID 的列表
-  "thoughtHistoryLength": int,           // 思考历史长度
-  "branchDetails": {                      // 分支详情
-    "currentBranchId": "main | branchId", // 当前分支 ID
-    "branchOriginThought": null | int,     // 分支起源的思考编号
-    "allBranches": {"main": 数量, "branchId": 数量, ...} // 所有分支及其包含的思考数
-  },
-  "isRevision": bool,                     // 是否为修订
-  "revisesThought": null | int,           // 修订的思考编号
-  "isBranch": bool,                       // 是否为分支操作产生的思考
-  "status": "success | validation_error | failed", // 状态
-  "error": "如果状态不是 success 时的错误信息" // 可选
-}
-```
-
-## 日志记录
-
-  * 日志写入 `~/.sequential_thinking/logs/sequential_thinking.log`。
-  * 使用 Python 标准的 `logging` 模块。
-  * 包含轮转文件处理器（10MB 限制，5 个备份）和控制台处理器（INFO 级别）。
-  * 日志包含时间戳、级别、记录器名称和消息，包括格式化的思考表示。
-
-## 开发
-
-（如果适用，在此处添加开发指南，例如设置开发环境、运行测试、代码检查等。）
-
-1.  克隆仓库。
-2.  设置虚拟环境。
-3.  安装依赖项，可能包括开发附加项：
-    ```bash
-    # 使用 uv
-    uv pip install -e ".[dev]"
-    # 使用 pip
-    pip install -e ".[dev]"
-    ```
-4.  运行代码检查器/格式化器/测试。
-
-## 许可证
-
-MIT
+5.  **LLM:** 根据 `coordinatorResponse` 构思下一个思考（例如，"使用 Exa 研究 X..."）。
+6.  **LLM:** 使用 `thoughtNumber: 2`、新的 `thought`、更新的 `totalThoughts`（如果需要）、`

{mcp_server_mas_sequential_thinking-0.1.3 → mcp_server_mas_sequential_thinking-0.2.0}/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-chat")
+        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", "deepseek-r1-distill-llama-70b")
+        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-chat-v3-0324")
+        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)
+             # Or raise Exception("Critical Error: Application context not available.")
 
     MIN_TOTAL_THOUGHTS = 5 # Keep a minimum suggestion
 
@@ -731,7 +788,9 @@ async def sequentialthinking(thought: str, thoughtNumber: int, totalThoughts: in
 
 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 → mcp_server_mas_sequential_thinking-0.2.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "mcp-server-mas-sequential-thinking"
-version = "0.1.3"
+version = "0.2.0"
 description = "MCP Agent Implementation for Sequential Thinking"
 readme = "README.md"
 requires-python = ">=3.10"
@@ -13,6 +13,7 @@ dependencies = [
     "exa-py",
     "python-dotenv",
     "mcp",
+    "groq",
 ]
 
 [project.optional-dependencies]
@@ -32,11 +33,3 @@ build-backend = "hatchling.build"
 
 [tool.hatch.build.targets.wheel]
 include = ["main.py"]
-
-[tool.uv]
-dev-dependencies = [
-    "pytest",
-    "black",
-    "isort",
-    "mypy",
-]