Upload mcp_server_sse.py

mcp_server_sse.py

@@ -2,6 +2,7 @@
 MCP Server for Hugging Face Space Deployment
 Uses Server-Sent Events (SSE) transport for remote MCP access
 Fully compatible with MCP clients like Claude Desktop
+Returns clean JSON without emoji formatting
 """
 
 from fastapi import FastAPI, Request, HTTPException
@@ -176,30 +177,22 @@ MCP_TOOLS = [
 ]
 
 
-def format_tool_result(data: Any) -> str:
-    """Format tool execution result as readable text"""
-    if isinstance(data, dict):
-        return json.dumps(data, indent=2, ensure_ascii=False)
-    elif isinstance(data, list):
-        return json.dumps(data, indent=2, ensure_ascii=False)
-    else:
-        return str(data)
-
-
 def execute_tool(tool_name: str, arguments: Dict[str, Any]) -> Dict[str, Any]:
-    """Execute MCP tool and return formatted result"""
+    """Execute MCP tool and return clean JSON result"""
     try:
         if tool_name == "search_company":
             result = edgar_client.search_company_by_name(arguments["company_name"])
             if result:
                 return {
                     "type": "text",
-                    "text": f"✅ Company Found:\n\n{format_tool_result(result)}"
+                    "text": json.dumps(result, ensure_ascii=False)
                 }
             else:
                 return {
                     "type": "text",
-                    "text": f"❌ No company found with name: {arguments['company_name']}"
+                    "text": json.dumps({
+                        "error": f"No company found with name: {arguments['company_name']}"
+                    }, ensure_ascii=False)
                 }
         
         elif tool_name == "get_company_info":
@@ -207,26 +200,36 @@ def execute_tool(tool_name: str, arguments: Dict[str, Any]) -> Dict[str, Any]:
             if result:
                 return {
                     "type": "text",
-                    "text": f"📊 Company Information:\n\n{format_tool_result(result)}"
+                    "text": json.dumps(result, ensure_ascii=False)
                 }
             else:
                 return {
                     "type": "text",
-                    "text": f"❌ No company found with CIK: {arguments['cik']}"
+                    "text": json.dumps({
+                        "error": f"No company found with CIK: {arguments['cik']}"
+                    }, ensure_ascii=False)
                 }
         
         elif tool_name == "get_company_filings":
             form_types = arguments.get("form_types")
             result = edgar_client.get_company_filings(arguments["cik"], form_types)
             if result:
+                # Limit to 20 results
+                limited_result = result[:20]
                 return {
                     "type": "text",
-                    "text": f"📁 Company Filings ({len(result)} found):\n\n{format_tool_result(result[:20])}"  # Limit to 20
+                    "text": json.dumps({
+                        "total": len(result),
+                        "returned": len(limited_result),
+                        "filings": limited_result
+                    }, ensure_ascii=False)
                 }
             else:
                 return {
                     "type": "text",
-                    "text": f"❌ No filings found for CIK: {arguments['cik']}"
+                    "text": json.dumps({
+                        "error": f"No filings found for CIK: {arguments['cik']}"
+                    }, ensure_ascii=False)
                 }
         
         elif tool_name == "get_financial_data":
@@ -234,12 +237,14 @@ def execute_tool(tool_name: str, arguments: Dict[str, Any]) -> Dict[str, Any]:
             if result and "period" in result:
                 return {
                     "type": "text",
-                    "text": f"💰 Financial Data for {result['period']}:\n\n{format_tool_result(result)}"
+                    "text": json.dumps(result, ensure_ascii=False)
                 }
             else:
                 return {
                     "type": "text",
-                    "text": f"❌ No financial data found for CIK: {arguments['cik']}, Period: {arguments['period']}"
+                    "text": json.dumps({
+                        "error": f"No financial data found for CIK: {arguments['cik']}, Period: {arguments['period']}"
+                    }, ensure_ascii=False)
                 }
         
         elif tool_name == "extract_financial_metrics":
@@ -247,7 +252,9 @@ def execute_tool(tool_name: str, arguments: Dict[str, Any]) -> Dict[str, Any]:
             if years < 1 or years > 10:
                 return {
                     "type": "text",
-                    "text": "❌ Years parameter must be between 1 and 10"
+                    "text": json.dumps({
+                        "error": "Years parameter must be between 1 and 10"
+                    }, ensure_ascii=False)
                 }
             
             metrics = financial_analyzer.extract_financial_metrics(arguments["cik"], years)
@@ -255,12 +262,17 @@ def execute_tool(tool_name: str, arguments: Dict[str, Any]) -> Dict[str, Any]:
                 formatted = financial_analyzer.format_financial_data(metrics)
                 return {
                     "type": "text",
-                    "text": f"📈 Financial Metrics ({len(formatted)} periods):\n\n{format_tool_result(formatted)}"
+                    "text": json.dumps({
+                        "periods": len(formatted),
+                        "data": formatted
+                    }, ensure_ascii=False)
                 }
             else:
                 return {
                     "type": "text",
-                    "text": f"❌ No financial metrics found for CIK: {arguments['cik']}"
+                    "text": json.dumps({
+                        "error": f"No financial metrics found for CIK: {arguments['cik']}"
+                    }, ensure_ascii=False)
                 }
         
         elif tool_name == "get_latest_financial_data":
@@ -268,36 +280,45 @@ def execute_tool(tool_name: str, arguments: Dict[str, Any]) -> Dict[str, Any]:
             if result and "period" in result:
                 return {
                     "type": "text",
-                    "text": f"🆕 Latest Financial Data:\n\n{format_tool_result(result)}"
+                    "text": json.dumps(result, ensure_ascii=False)
                 }
             else:
                 return {
                     "type": "text",
-                    "text": f"❌ No latest financial data found for CIK: {arguments['cik']}"
+                    "text": json.dumps({
+                        "error": f"No latest financial data found for CIK: {arguments['cik']}"
+                    }, ensure_ascii=False)
                 }
         
-        elif tool_name == "advanced_search_company":
+        elif tool_name == "advanced_search_company" or tool_name == "advanced_search":
+            # Support both names for backward compatibility
             result = financial_analyzer.search_company(arguments["company_input"])
             if result.get("error"):
                 return {
                     "type": "text",
-                    "text": f"❌ {result['error']}"
+                    "text": json.dumps({
+                        "error": result["error"]
+                    }, ensure_ascii=False)
                 }
             return {
                 "type": "text",
-                "text": f"🔍 Search Result:\n\n{format_tool_result(result)}"
+                "text": json.dumps(result, ensure_ascii=False)
             }
         
         else:
             return {
                 "type": "text",
-                "text": f"❌ Unknown tool: {tool_name}"
+                "text": json.dumps({
+                    "error": f"Unknown tool: {tool_name}"
+                }, ensure_ascii=False)
             }
     
     except Exception as e:
         return {
             "type": "text",
-            "text": f"❌ Error executing {tool_name}: {str(e)}"
+            "text": json.dumps({
+                "error": f"Error executing {tool_name}: {str(e)}"
+            }, ensure_ascii=False)
         }
 
 
@@ -482,10 +503,10 @@ async def root():
     </head>
     <body>
         <div class="container">
-            <h1>🚀 SEC Financial Report MCP Server</h1>
+            <h1>SEC Financial Report MCP Server</h1>
             <p><span class="badge">MCP READY</span> Model Context Protocol Server for SEC EDGAR Financial Data</p>
             
-            <h2>📡 Connect to this MCP Server</h2>
+            <h2>Connect to this MCP Server</h2>
             <p>Add the following configuration to your MCP client (e.g., Claude Desktop):</p>
             
             <div class="config-box">
@@ -500,7 +521,7 @@ async def root():
             
             <p><strong>Use the URL above to connect to this MCP Server.</strong></p>
             
-            <h2>🛠️ Available Tools (7)</h2>
+            <h2>Available Tools (7)</h2>
             <div class="tools">
                 <div class="tool">
                     <h3>search_company</h3>
@@ -519,7 +540,7 @@ async def root():
                     <p>Get financial data for a specific period (annual or quarterly)</p>
                 </div>
                 <div class="tool">
-                    <h3>extract_financial_metrics ⭐</h3>
+                    <h3>extract_financial_metrics</h3>
                     <p>Extract comprehensive financial metrics for multiple years</p>
                 </div>
                 <div class="tool">
@@ -532,7 +553,7 @@ async def root():
                 </div>
             </div>
             
-            <h2>💬 Example Usage</h2>
+            <h2>Example Usage</h2>
             <p>Once connected, you can ask your AI assistant:</p>
             <ul>
                 <li>"Search for Microsoft and show me their latest financial data"</li>
@@ -540,7 +561,7 @@ async def root():
                 <li>"Compare NVIDIA and AMD's revenue trends"</li>
             </ul>
             
-            <h2>📚 Documentation</h2>
+            <h2>Documentation</h2>
             <p>Visit <code>/tools</code> to see all available tools in JSON format.</p>
             
             <hr style="margin: 30px 0; border: none; border-top: 1px solid #e0e0e0;">