Expose fastapi-mcp

Author: danlapid

README.md

@@ -1,6 +1,6 @@
-# Python Workers: FastMCP Example
+# Python Workers: FastAPI-MCP Example
 
-This is an example of a Python Worker that uses the FastMCP package.
+This is an example of a Python Worker that uses the FastAPI-MCP package.
 
 ## Adding Packages
 

src/httpx_patch.py

@@ -0,0 +1,11 @@
+from httpx._transports.jsfetch import AsyncJavascriptFetchTransport
+
+orig_handle_async_request = AsyncJavascriptFetchTransport.handle_async_request
+
+async def handle_async_request(self, request):
+    response = await orig_handle_async_request(self, request)
+    # fix content-encoding headers because the javascript fetch handles that
+    response.headers.update({"content-encoding": "identity"})
+    return response
+
+AsyncJavascriptFetchTransport.handle_async_request = handle_async_request

src/worker.py

@@ -2,48 +2,81 @@
 from workers import DurableObject
 from logger import logger
 import sys
+import urllib.parse
+import httpx_patch  # noqa: F401
 sys.path.insert(0, "/session/metadata/vendor")
 sys.path.insert(0, "/session/metadata")
 
 
-def setup_server():
+def setup_server(base_url: str):
+    from fastapi import FastAPI, Request
+    from pydantic import BaseModel
+    from fastapi_mcp import FastApiMCP
     from exceptions import HTTPException, http_exception
-    from mcp.server.fastmcp import FastMCP
-    mcp = FastMCP("Demo")
-
-    @mcp.tool()
-    def add(a: int, b: int) -> int:
-        """Add two numbers"""
-        return a + b
-
-    @mcp.resource("greeting://{name}")
-    def get_greeting(name: str) -> str:
-        """Get a personalized greeting"""
-        return f"Hello, {name}!"
-
-    @mcp.tool()
-    def calculate_bmi(weight_kg: float, height_m: float) -> float:
-        """Calculate BMI given weight in kg and height in meters"""
-        return weight_kg / (height_m**2)
-
-    @mcp.prompt()
-    def echo_prompt(message: str) -> str:
-        """Create an echo prompt"""
-        return f"Please process this message: {message}"
-
-    app = mcp.sse_app()
+
+    app = FastAPI()
     app.add_exception_handler(HTTPException, http_exception)
+
+    mcp = FastApiMCP(
+        app,
+        # Optional parameters
+        name="My API MCP",
+        description="My API description",
+        base_url=base_url
+    )
+
+    # Mount the MCP server directly to your FastAPI app
+    mcp.mount()
+    # Auto-generated operation_id (something like "read_user_users__user_id__get")
+    @app.get("/")
+    async def root():
+        return {"message": "Hello, World!"}
+
+    @app.get("/env")
+    async def root(req: Request):
+        env = req.scope["env"]
+        return {"message": "Here is an example of getting an environment variable: " + env.MESSAGE}
+
+    class Item(BaseModel):
+        name: str
+        description: str | None = None
+        price: float
+        tax: float | None = None
+
+    @app.post("/items/")
+    async def create_item(item: Item):
+        return item
+
+    @app.put("/items/{item_id}")
+    async def create_item(item_id: int, item: Item, q: str | None = None):
+        result = {"item_id": item_id, **item.dict()}
+        if q:
+            result.update({"q": q})
+        return result
+
+    @app.get("/items/{item_id}")
+    async def read_item(item_id: int):
+        return {"item_id": item_id}
+    
+    mcp.setup_server()
     return mcp, app
 
 
 class FastMCPServer(DurableObject):
     def __init__(self, ctx, env):
         self.ctx = ctx
         self.env = env
-        self.mcp, self.app = setup_server()
+        self.mcp = None
+        self.app = None
+        self.base_url = None
 
     async def call(self, request):
         import asgi
+        request_url = urllib.parse.urlparse(request.url)
+        base_url = f"{request_url.scheme}://{request_url.netloc}/"
+        if self.mcp is None or self.app is None or self.base_url is None or self.base_url != base_url:
+            self.mcp, self.app = setup_server(base_url)
+            self.base_url = base_url
         return await asgi.fetch(self.app, request, self.env, self.ctx)
 
 async def on_fetch(request, env):

vendor.txt

@@ -1,2 +1,4 @@
-mcp
-structlog
+fastapi-mcp
+fastapi
+pydantic
+structlog