Initial commit: mysearchagent with 1 tools - src/server/mcp_server.py

2026-02-10 11:58:09 +00:00
parent f747eaa339
commit c36ae1d9f0
1 changed files with 137 additions and 0 deletions
@@ -0,0 +1,137 @@
+"""
+MCP 服务器 - mysearchagent-9a8ab2
+
+Agent with 1 external tools
+使用 Pydantic AI 和 FastMCP 框架。
+自动生成时间: 2026-02-10T11:57:52.059593
+"""
+import json
+import os
+from typing import Optional
+
+import httpx
+from mcp.server.fastmcp import FastMCP
+from pydantic_ai import Agent
+
+# ==================== 配置 ====================
+
+# LiteLLM Gateway 配置
+_BASE_URL = os.getenv('OPENAI_BASE_URL', 
+    os.getenv('LLM_BASE_URL', 'https://litellm.graystone-fb459c5d.southeastasia.azurecontainerapps.io/v1'))
+_API_KEY = os.getenv('OPENAI_API_KEY', 'sk')
+
+os.environ.setdefault('OPENAI_API_KEY', _API_KEY)
+os.environ.setdefault('OPENAI_BASE_URL', _BASE_URL)
+
+# 模型名称（pydantic_ai 需要 openai: 前缀）
+def _get_model_name() -> str:
+    model = os.getenv('MODEL_NAME', os.getenv('LITELLM_MODEL', 'taiji/gpt-4o-mini'))
+    return model if ':' in model else f'openai:{model}'
+
+MODEL_NAME = _get_model_name()
+
+# ==================== MCP 服务器 ====================
+
+server = FastMCP('mysearchagent-9a8ab2')
+
+# 系统提示词
+SYSTEM_PROMPT = """你是 mysearchagent-9a8ab2，一个专业的 AI 智能助手。
+
+## 角色定位
+Agent with 1 external tools
+
+## 可用工具
+- mygoogletool: 在线google搜索
+
+## 工作原则
+1. 理解用户意图：仔细分析用户的请求，确保准确理解需求
+2. 选择合适工具：根据需求选择最合适的工具来完成任务
+3. 清晰反馈：以用户友好的方式呈现结果
+4. 错误处理：遇到问题时提供有用的错误信息和建议
+
+## 响应格式
+- 对于数据查询：返回结构化的 JSON 数据
+- 对于操作请求：返回操作状态和结果
+- 始终使用中文与用户交流（除非用户使用其他语言）"""
+
+
+def get_agent() -> Agent:
+    """创建 Agent 实例（每次调用使用最新的 API Key）"""
+    return Agent(MODEL_NAME, system_prompt=SYSTEM_PROMPT)
+
+
+# ==================== MCP 工具定义 ====================
+
+@server.tool()
+async def mygoogletool(q: str) -> str:
+    """
+    在线google搜索
+    
+    Args:
+        q: q 参数
+        
+    Returns:
+        str: JSON格式的响应结果
+    """
+    url = "https://google.serper.dev/search"
+    
+    headers = {
+        "X-API-Key": "8253b4f240b520194065312f90e85f9be0fa205f",
+        "Content-Type": "application/json"
+    }
+    
+    params = {
+        "q": q
+    }
+    
+    try:
+        async with httpx.AsyncClient(timeout=30.0) as client:
+            response = await client.post(
+                url,
+                headers=headers,
+                params=params
+            )
+            response.raise_for_status()
+            return json.dumps(response.json(), ensure_ascii=False)
+    except httpx.HTTPStatusError as e:
+        error_detail = {
+            "error": "HTTP错误",
+            "status_code": e.response.status_code,
+            "detail": str(e)
+        }
+        return json.dumps(error_detail, ensure_ascii=False)
+    except httpx.RequestError as e:
+        error_detail = {
+            "error": "请求错误",
+            "detail": str(e)
+        }
+        return json.dumps(error_detail, ensure_ascii=False)
+    except Exception as e:
+        error_detail = {
+            "error": "未知错误",
+            "detail": str(e)
+        }
+        return json.dumps(error_detail, ensure_ascii=False)
+
+
+# ==================== 工具映射（供 API 使用）====================
+
+TOOL_MAP = {
+    'mygoogletool': mygoogletool,
+}
+
+TOOL_LIST = [
+    {
+        "name": "mygoogletool",
+        "description": "在线google搜索",
+        "inputSchema": {
+            "type": "object",
+            "properties": {"q": {"type": "string", "description": "q 参数"}},
+            "required": []
+        }
+    },
+]
+
+
+if __name__ == '__main__':
+    server.run()