Upload app.py

app.py

@@ -88,13 +88,16 @@ def query_financial_data(company_name, query_type):
     internal_query_type = query_type_mapping.get(query_type, query_type)
     
     try:
-        # 调用 MCP Server 工具（使用 MCP 协议格式）
-        # 先搜索公司 - 尝试使用 tools/call 端点
+        # 使用 MCP 协议调用工具
+        # 先搜索公司
         search_resp = requests.post(
-            f"{MCP_URL}/tools/call",
+            f"{MCP_URL}/message",
             json={
-                "name": "advanced_search",
-                "arguments": {"company_input": company_name}
+                "method": "tools/call",
+                "params": {
+                    "name": "advanced_search",
+                    "arguments": {"company_input": company_name}
+                }
             },
             headers=HEADERS,
             timeout=30
@@ -104,22 +107,17 @@ def query_financial_data(company_name, query_type):
             return f"❌ Server Error: HTTP {search_resp.status_code}\n\nResponse: {search_resp.text[:500]}"
         
         try:
-            company_result = search_resp.json()
-            # MCP 协议返回格式可能是 {"content": [{"type": "text", "text": "..."}]} 或直接返回数据
-            if isinstance(company_result, dict):
-                if "content" in company_result:
-                    # MCP 标准返回格式
-                    content = company_result.get("content", [])
-                    if content and len(content) > 0:
-                        text_content = content[0].get("text", "{}")
-                        company = json.loads(text_content) if isinstance(text_content, str) else text_content
-                    else:
-                        company = {}
+            result = search_resp.json()
+            # MCP 协议返回格式: {"content": [{"type": "text", "text": "..."}]}
+            if isinstance(result, dict) and "content" in result:
+                content = result.get("content", [])
+                if content and len(content) > 0:
+                    text_content = content[0].get("text", "{}")
+                    company = json.loads(text_content) if isinstance(text_content, str) else text_content
                 else:
-                    # 直接返回数据
-                    company = company_result
+                    company = {}
             else:
-                company = company_result
+                company = result
         except (ValueError, KeyError) as e:
             return f"❌ JSON Parse Error: {str(e)}\n\nResponse: {search_resp.text[:500]}"
         
@@ -135,10 +133,13 @@ def query_financial_data(company_name, query_type):
         # 根据查询类型获取数据
         if internal_query_type == "最新财务数据":
             data_resp = requests.post(
-                f"{MCP_URL}/tools/call",
+                f"{MCP_URL}/message",
                 json={
-                    "name": "get_latest_financial_data",
-                    "arguments": {"cik": cik}
+                    "method": "tools/call",
+                    "params": {
+                        "name": "get_latest_financial_data",
+                        "arguments": {"cik": cik}
+                    }
                 },
                 headers=HEADERS,
                 timeout=30
@@ -149,7 +150,7 @@ def query_financial_data(company_name, query_type):
             
             try:
                 data_result = data_resp.json()
-                # 处理 MCP 协议返回格式
+                # MCP 协议返回格式
                 if isinstance(data_result, dict) and "content" in data_result:
                     content = data_result.get("content", [])
                     if content and len(content) > 0:
@@ -186,10 +187,13 @@ def query_financial_data(company_name, query_type):
         
         elif internal_query_type == "3年趋势":
             metrics_resp = requests.post(
-                f"{MCP_URL}/tools/call",
+                f"{MCP_URL}/message",
                 json={
-                    "name": "extract_financial_metrics",
-                    "arguments": {"cik": cik, "years": 3}
+                    "method": "tools/call",
+                    "params": {
+                        "name": "extract_financial_metrics",
+                        "arguments": {"cik": cik, "years": 3}
+                    }
                 },
                 headers=HEADERS,
                 timeout=60
@@ -200,7 +204,7 @@ def query_financial_data(company_name, query_type):
             
             try:
                 metrics_result = metrics_resp.json()
-                # 处理 MCP 协议返回格式
+                # MCP 协议返回格式
                 if isinstance(metrics_result, dict) and "content" in metrics_result:
                     content = metrics_result.get("content", [])
                     if content and len(content) > 0:
@@ -273,10 +277,13 @@ def query_financial_data(company_name, query_type):
         
         elif internal_query_type == "5年趋势":
             metrics_resp = requests.post(
-                f"{MCP_URL}/tools/call",
+                f"{MCP_URL}/message",
                 json={
-                    "name": "extract_financial_metrics",
-                    "arguments": {"cik": cik, "years": 5}
+                    "method": "tools/call",
+                    "params": {
+                        "name": "extract_financial_metrics",
+                        "arguments": {"cik": cik, "years": 5}
+                    }
                 },
                 headers=HEADERS,
                 timeout=60
@@ -287,7 +294,7 @@ def query_financial_data(company_name, query_type):
             
             try:
                 metrics_result = metrics_resp.json()
-                # 处理 MCP 协议返回格式
+                # MCP 协议返回格式
                 if isinstance(metrics_result, dict) and "content" in metrics_result:
                     content = metrics_result.get("content", [])
                     if content and len(content) > 0:
@@ -360,10 +367,13 @@ def query_financial_data(company_name, query_type):
         elif internal_query_type == "公司报表列表":
             # 查询公司所有报表
             filings_resp = requests.post(
-                f"{MCP_URL}/tools/call",
+                f"{MCP_URL}/message",
                 json={
-                    "name": "get_company_filings",
-                    "arguments": {"cik": cik, "limit": 50}
+                    "method": "tools/call",
+                    "params": {
+                        "name": "get_company_filings",
+                        "arguments": {"cik": cik, "limit": 50}
+                    }
                 },
                 headers=HEADERS,
                 timeout=60
@@ -374,7 +384,7 @@ def query_financial_data(company_name, query_type):
             
             try:
                 filings_result = filings_resp.json()
-                # 处理 MCP 协议返回格式
+                # MCP 协议返回格式
                 if isinstance(filings_result, dict) and "content" in filings_result:
                     content = filings_result.get("content", [])
                     if content and len(content) > 0: