Upload 8 files

arithmetic_server.py

@@ -1,4 +1,8 @@
 from mcp.server.fastmcp import FastMCP
+from starlette.applications import Starlette
+from starlette.routing import Mount
+import contextlib
+import uvicorn
 
 # Creates a server named "Arithmetic"
 mcp = FastMCP("Arithmetic")
@@ -25,6 +29,16 @@ def divide(a: int, b: int) -> float:
         raise ValueError("Division by zero is not allowed.")
     return a / b
 
+@contextlib.asynccontextmanager
+async def lifespan(app: Starlette):
+    async with contextlib.AsyncExitStack() as stack:
+        await stack.enter_async_context(mcp.session_manager.run())
+        yield
+
+app = Starlette(
+    routes=[Mount("/mcp", mcp.streamable_http_app())],
+    lifespan=lifespan,
+)
 
 if __name__ == "__main__":
-    mcp.run(transport="streamable_http", port=8001, address="0.0.0.0")
+    uvicorn.run(app, host="0.0.0.0", port=8001)

backend.py

@@ -175,11 +175,11 @@ class MCPAgent:
         self.client = MultiServerMCPClient({
             "arithmetic": {
                 "url": "http://localhost:8001/mcp",
-                "transport": "streamable_http",
+                "transport": "streamable-http",
             },
             "stocks": {
                 "url": "http://localhost:8002/mcp",
-                "transport": "streamable_http",
+                "transport": "streamable-http",
             },
         }
     )

stock_server.py

@@ -2,6 +2,10 @@ from mcp.server.fastmcp import FastMCP
 import aiohttp
 import json
 from typing import Optional
+from starlette.applications import Starlette
+from starlette.routing import Mount
+import contextlib
+import uvicorn
 
 mcp = FastMCP("StockMarket")
 
@@ -59,5 +63,16 @@ async def get_company_news(symbol: str, limit: int = 3) -> str:
     2. Analysts upgrade {symbol} to 'Buy' rating
     3. {symbol} unveils new product line for 2025"""
 
+@contextlib.asynccontextmanager
+async def lifespan(app: Starlette):
+    async with contextlib.AsyncExitStack() as stack:
+        await stack.enter_async_context(mcp.session_manager.run())
+        yield
+
+app = Starlette(
+    routes=[Mount("/mcp", mcp.streamable_http_app())],
+    lifespan=lifespan,
+)
+
 if __name__ == "__main__":
-    mcp.run(transport="streamable_http", port=8002, address="0.0.0.0")
+    uvicorn.run(app, host="0.0.0.0", port=8002)