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

src/server/mcp_server.py

@@ -0,0 +1,143 @@
+"""
+MCP 服务器 - lasttest-0db7a0
+
+Agent with 1 external tools
+使用 Pydantic AI 和 FastMCP 框架。
+自动生成时间: 2026-02-13T15:50:34.879589
+"""
+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('lasttest-0db7a0')
+
+# 系统提示词
+SYSTEM_PROMPT = """你是 lasttest-0db7a0，一个专业的 AI 智能助手。
+
+## 角色定位
+Agent with 1 external tools
+
+## 可用工具
+- jinaaitool: abcd
+
+## 工作原则
+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 jinaaitool(url: str) -> str:
+    """
+    调用 jinaaitool API
+    
+    Args:
+        url: url 参数
+        
+    Returns:
+        JSON 字符串格式的响应结果
+    """
+    try:
+        # 构建完整的 API URL，将 url 参数拼接到路径中
+        api_url = f"https://r.jina.ai/{url}"
+        
+        # 从环境变量获取 token 并设置 Bearer 认证
+        token = os.getenv("TOOL_API_KEY")
+        if not token:
+            return json.dumps({"error": "TOOL_API_KEY environment variable not set"})
+        
+        headers = {
+            "Authorization": f"Bearer {token}",
+            "Content-Type": "application/json"
+        }
+        
+        # 发送 POST 请求
+        async with httpx.AsyncClient(timeout=30.0) as client:
+            response = await client.post(
+                api_url,
+                headers=headers
+            )
+            response.raise_for_status()
+            
+            # 尝试解析 JSON 响应
+            try:
+                result = response.json()
+            except json.JSONDecodeError:
+                result = {"content": response.text}
+            
+            return json.dumps(result, ensure_ascii=False)
+            
+    except httpx.HTTPStatusError as e:
+        return json.dumps({
+            "error": f"HTTP error occurred: {e.response.status_code}",
+            "detail": str(e)
+        })
+    except httpx.RequestError as e:
+        return json.dumps({
+            "error": "Request error occurred",
+            "detail": str(e)
+        })
+    except Exception as e:
+        return json.dumps({
+            "error": "Unexpected error occurred",
+            "detail": str(e)
+        })
+
+
+# ==================== 工具映射（供 API 使用）====================
+
+TOOL_MAP = {
+    'jinaaitool': jinaaitool,
+}
+
+TOOL_LIST = [
+    {
+        "name": "jinaaitool",
+        "description": "abcd",
+        "inputSchema": {
+            "type": "object",
+            "properties": {"url": {"type": "string", "description": "url 参数"}},
+            "required": []
+        }
+    },
+]
+
+
+if __name__ == '__main__':
+    server.run()