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/worker.py

@@ -4,41 +4,55 @@
 
 class FastMCPServer(DurableObject):
     def __init__(self, state, env):
+        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
-        self.state = state
-        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}"
-
-        # mcp depends on uvicorn and imports it at the top scope that we have to patch that to move the 
-        # import into the function that uses it.
-        # TODO(now): Change uvicorn to optional in mcp
-        app = mcp.sse_app()
-        # Starlette default http exception handler is sync which starlette tries to run in threadpool
-        # in https://github.com/encode/starlette/blob/master/starlette/_exception_handler.py#L61.
-        # Since we don't support threads we need to override it with the same function but async.
-        # TODO(now): change starlette's http_exception to be async, it is strictly slower to spawn a new
-        #            thread
+
+        app = FastAPI()
         app.add_exception_handler(HTTPException, http_exception)
+
+        mcp = FastApiMCP(
+            app,
+            # Optional parameters
+            name="My API MCP",
+            description="My API description",
+        )
+
+        # 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()
         self.mcp = mcp
         self.app = app
 

vendor.txt

@@ -1 +1,3 @@
-mcp
+fastapi-mcp
+fastapi
+pydantic