Upload app.py

app.py

@@ -9,6 +9,7 @@ from huggingface_hub import InferenceClient
 
 MCP_SPACE = "JC321/EasyReportDateMCP"
 MCP_URL = "https://jc321-easyreportdatemcp.hf.space"
+MCP_ENDPOINT = "/mcp"  # MCP 工具调用端点
 
 # 设置请求头
 HEADERS = {
@@ -115,6 +116,48 @@ def format_value(value, value_type="money"):
     else:  # number
         return f"{value:.2f}"
 
+def call_mcp_tool(tool_name, arguments):
+    """调用 MCP 工具并返回结果"""
+    try:
+        # 构建完整的 URL
+        full_url = f"{MCP_URL}{MCP_ENDPOINT}"
+        
+        # FastMCP HTTP Server 使用 /mcp 端点
+        response = session.post(
+            full_url,
+            json={
+                "jsonrpc": "2.0",
+                "method": "tools/call",
+                "params": {
+                    "name": tool_name,
+                    "arguments": arguments
+                },
+                "id": 1
+            },
+            headers=HEADERS,
+            timeout=60
+        )
+        
+        # 调试信息
+        print(f"DEBUG: Calling {full_url}")
+        print(f"DEBUG: Tool: {tool_name}, Args: {arguments}")
+        print(f"DEBUG: Status Code: {response.status_code}")
+        print(f"DEBUG: Response: {response.text[:500]}")
+        
+        if response.status_code != 200:
+            return {
+                "error": f"HTTP {response.status_code}",
+                "detail": response.text,
+                "url": full_url
+            }
+        
+        return response.json()
+    except Exception as e:
+        return {
+            "error": str(e),
+            "url": full_url if 'full_url' in locals() else MCP_URL
+        }
+
 def normalize_cik(cik):
     """
     格式化 CIK 为标准的 10 位格式
@@ -229,24 +272,13 @@ def query_financial_data(company_name, query_type):
         # 使用 MCP 协议调用工具
         # 先搜索公司（使用 advanced_search_company）
         try:
-            search_resp = session.post(
-                f"{MCP_URL}/mcp",
-                json={
-                    "jsonrpc": "2.0",
-                    "method": "tools/call",
-                    "params": {
-                        "name": "advanced_search_company",
-                        "arguments": {"company_input": company_name}
-                    },
-                    "id": 1
-                },
-                headers=HEADERS,
-                timeout=60  # 增加到60秒
-            )
+            search_result = call_mcp_tool("advanced_search_company", {"company_input": company_name})
         except requests.exceptions.Timeout:
             return f"❌ MCP Server Timeout: The server took too long to respond (>60s).\n\n**Possible reasons**:\n1. MCP Server is cold starting (first request after idle)\n2. Server is overloaded\n3. Network issues\n\n**Suggestion**: Please try again in a few moments. If the problem persists, the MCP Server at {MCP_URL} may be down."
         
-        if search_resp.status_code != 200:
+        # 检查是否有错误
+        if "error" in search_result:
+            return f"❌ Server Error: {search_result.get('error')}\n\nResponse: {search_result.get('detail', 'N/A')}\n\nURL: {search_result.get('url', MCP_URL)}"
             return f"❌ Server Error: HTTP {search_resp.status_code}\n\nResponse: {search_resp.text[:500]}"
         
         try: