From e7052ea547361c71ec0481baefc33fd80e74c5e4 Mon Sep 17 00:00:00 2001 From: zhanggangyong Date: Thu, 5 Feb 2026 05:50:09 +0000 Subject: [PATCH] Initial commit: ccjinaaiagent with 1 tools - src/server/mcp_server.py --- src/server/mcp_server.py | 144 +++++++++++++++++++++++++++++++++++++++ 1 file changed, 144 insertions(+) create mode 100644 src/server/mcp_server.py diff --git a/src/server/mcp_server.py b/src/server/mcp_server.py new file mode 100644 index 0000000..b2fcb8a --- /dev/null +++ b/src/server/mcp_server.py @@ -0,0 +1,144 @@ +""" +MCP 服务器 - ccjinaaiagent-195d28 + +Agent with 1 external tools +使用 Pydantic AI 和 FastMCP 框架。 +自动生成时间: 2026-02-05T05:49:57.409392 +""" +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('ccjinaaiagent-195d28') + +# 系统提示词 +SYSTEM_PROMPT = """你是 ccjinaaiagent-195d28,一个专业的 AI 智能助手。 + +## 角色定位 +Agent with 1 external tools + +## 可用工具 +- CCJinaAItools: 通过post传递url爬取网站 + +## 工作原则 +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 ccjinaaitools(url: str) -> str: + """ + 通过post传递url爬取网站 + + Args: + url: url 参数 + + Returns: + str: JSON格式的响应结果 + """ + api_url = "https://r.jina.ai/" + + headers = { + "Authorization": f"Bearer {os.getenv('TOOL_API_KEY')}", + "Content-Type": "application/json" + } + + body = { + "url": url + } + + try: + async with httpx.AsyncClient(timeout=30.0) as client: + response = await client.post( + api_url, + headers=headers, + json=body + ) + response.raise_for_status() + + return json.dumps({ + "success": True, + "data": response.text, + "status_code": response.status_code + }, ensure_ascii=False) + + except httpx.HTTPStatusError as e: + return json.dumps({ + "success": False, + "error": f"HTTP错误: {e.response.status_code}", + "detail": str(e) + }, ensure_ascii=False) + + except httpx.RequestError as e: + return json.dumps({ + "success": False, + "error": "请求失败", + "detail": str(e) + }, ensure_ascii=False) + + except Exception as e: + return json.dumps({ + "success": False, + "error": "未知错误", + "detail": str(e) + }, ensure_ascii=False) + + +# ==================== 工具映射(供 API 使用)==================== + +TOOL_MAP = { + 'ccjinaaitools': ccjinaaitools, +} + +TOOL_LIST = [ + { + "name": "ccjinaaitools", + "description": "通过post传递url爬取网站", + "inputSchema": { + "type": "object", + "properties": {}, + "required": [] + } + }, +] + + +if __name__ == '__main__': + server.run()