Upload app.py

app.py

@@ -161,8 +161,19 @@ async def mcp_sse_endpoint(request: Request):
                 tool_name = body.get("params", {}).get("name")
                 arguments = body.get("params", {}).get("arguments", {})
                 
-                if tool_name in mcp._tool_manager._tools:
-                    tool_func = mcp._tool_manager._tools[tool_name]
+                # Directly call the tool functions we defined
+                tool_map = {
+                    "search_company": search_company,
+                    "get_company_info": get_company_info,
+                    "get_company_filings": get_company_filings,
+                    "get_financial_data": get_financial_data,
+                    "extract_financial_metrics": extract_financial_metrics,
+                    "get_latest_financial_data": get_latest_financial_data,
+                    "advanced_search_company": advanced_search_company
+                }
+                
+                if tool_name in tool_map:
+                    tool_func = tool_map[tool_name]
                     result = tool_func(**arguments)
                     
                     return JSONResponse({