Switch to streamable http transport

Author: danlapid

src/worker.py

@@ -1,17 +1,15 @@
 import sys
 
-from workers import DurableObject
-
 sys.path.insert(0, "/session/metadata/vendor")
 sys.path.insert(0, "/session/metadata")
 
+from mcp.server.fastmcp import FastMCP
 
-def setup_server():
-    from mcp.server.fastmcp import FastMCP
+from exceptions import HTTPException, http_exception
 
-    from exceptions import HTTPException, http_exception
 
-    mcp = FastMCP("Demo")
+def setup_server():
+    mcp = FastMCP("Demo", stateless_http=True)
 
     @mcp.tool()
     def add(a: int, b: int) -> int:
@@ -33,24 +31,13 @@ def echo_prompt(message: str) -> str:
         """Create an echo prompt"""
         return f"Please process this message: {message}"
 
-    app = mcp.sse_app()
+    app = mcp.streamable_http_app()
     app.add_exception_handler(HTTPException, http_exception)
     return mcp, app
 
 
-class FastMCPServer(DurableObject):
-    def __init__(self, ctx, env):
-        self.ctx = ctx
-        self.env = env
-        self.mcp, self.app = setup_server()
-
-    async def call(self, request):
-        import asgi
-
-        return await asgi.fetch(self.app, request, self.env, self.ctx)
-
+async def on_fetch(request, env, ctx):
+    mcp, app = setup_server()
+    import asgi
 
-async def on_fetch(request, env):
-    id = env.ns.idFromName("A")
-    obj = env.ns.get(id)
-    return await obj.call(request)
+    return await asgi.fetch(app, request, env, ctx)

wrangler.jsonc

@@ -17,22 +17,6 @@
             "fallthrough": true
         }
     ],
-    "durable_objects": {
-        "bindings": [
-            {
-                "name": "ns",
-                "class_name": "FastMCPServer"
-            }
-        ]
-    },
-    "migrations": [
-        {
-            "tag": "v1",
-            "new_sqlite_classes": [
-                "FastMCPServer"
-            ]
-        }
-    ],
     "observability": {
         "enabled": true
     }