edda-framework 0.3.0__tar.gz → 0.3.1__tar.gz

{edda_framework-0.3.0 → edda_framework-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: edda-framework
-Version: 0.3.0
+Version: 0.3.1
 Summary: Lightweight Durable Execution Framework
 Project-URL: Homepage, https://github.com/i2y/edda
 Project-URL: Documentation, https://github.com/i2y/edda#readme

{edda_framework-0.3.0 → edda_framework-0.3.1}/edda/integrations/mcp/decorators.py

@@ -4,7 +4,7 @@ from __future__ import annotations
 
 import inspect
 from collections.abc import Callable
-from typing import TYPE_CHECKING, Any
+from typing import TYPE_CHECKING, Any, cast
 
 from edda.workflow import workflow
 
@@ -15,7 +15,7 @@ if TYPE_CHECKING:
 
 def create_durable_tool(
     server: EddaMCPServer,
-    func: Callable,
+    func: Callable[..., Any],
     *,
     description: str = "",
 ) -> Workflow:
@@ -38,7 +38,7 @@ def create_durable_tool(
         Workflow instance
     """
     # 1. Create Edda workflow
-    workflow_instance: Workflow = workflow(func, event_handler=False)
+    workflow_instance = cast(Workflow, workflow(func, event_handler=False))
     workflow_name = func.__name__
 
     # Register in server's workflow registry
@@ -56,7 +56,7 @@ def create_durable_tool(
     ]
 
     # Create the tool function
-    async def start_tool(**kwargs: Any) -> dict:
+    async def start_tool(**kwargs: Any) -> dict[str, Any]:
         """
         Start workflow and return instance_id.
 
@@ -94,11 +94,21 @@ def create_durable_tool(
     status_tool_name = f"{workflow_name}_status"
     status_tool_description = f"Check status of {workflow_name} workflow"
 
-    @server._mcp.tool(name=status_tool_name, description=status_tool_description)
-    async def status_tool(instance_id: str) -> dict:
+    @server._mcp.tool(name=status_tool_name, description=status_tool_description)  # type: ignore[misc]
+    async def status_tool(instance_id: str) -> dict[str, Any]:
         """Check workflow status."""
         try:
             instance = await server._edda_app.storage.get_instance(instance_id)
+            if instance is None:
+                return {
+                    "content": [
+                        {
+                            "type": "text",
+                            "text": f"Workflow instance not found: {instance_id}",
+                        }
+                    ],
+                    "isError": True,
+                }
 
             status = instance["status"]
             current_activity_id = instance.get("current_activity_id", "N/A")
@@ -128,11 +138,21 @@ def create_durable_tool(
     result_tool_name = f"{workflow_name}_result"
     result_tool_description = f"Get result of {workflow_name} workflow (if completed)"
 
-    @server._mcp.tool(name=result_tool_name, description=result_tool_description)
-    async def result_tool(instance_id: str) -> dict:
+    @server._mcp.tool(name=result_tool_name, description=result_tool_description)  # type: ignore[misc]
+    async def result_tool(instance_id: str) -> dict[str, Any]:
         """Get workflow result (if completed)."""
         try:
             instance = await server._edda_app.storage.get_instance(instance_id)
+            if instance is None:
+                return {
+                    "content": [
+                        {
+                            "type": "text",
+                            "text": f"Workflow instance not found: {instance_id}",
+                        }
+                    ],
+                    "isError": True,
+                }
 
             status = instance["status"]
 

{edda_framework-0.3.0 → edda_framework-0.3.1}/edda/integrations/mcp/server.py

@@ -3,13 +3,13 @@
 from __future__ import annotations
 
 from collections.abc import Callable
-from typing import Any
+from typing import Any, cast
 
 from edda.app import EddaApp
 from edda.workflow import Workflow
 
 try:
-    from mcp.server.fastmcp import FastMCP
+    from mcp.server.fastmcp import FastMCP  # type: ignore[import-not-found]
 except ImportError as e:
     raise ImportError(
         "MCP Python SDK is required for MCP integration. "
@@ -89,7 +89,7 @@ class EddaMCPServer:
             service_name=name,
             db_url=db_url,
             outbox_enabled=outbox_enabled,
-            broker_url=broker_url,
+            broker_url=broker_url or "",
         )
         self._mcp = FastMCP(name, json_response=True, stateless_http=True)
         self._token_verifier = token_verifier
@@ -99,10 +99,10 @@ class EddaMCPServer:
 
     def durable_tool(
         self,
-        func: Callable | None = None,
+        func: Callable[..., Any] | None = None,
         *,
         description: str = "",
-    ) -> Callable:
+    ) -> Callable[..., Any]:
         """
         Decorator to define a durable workflow tool.
 
@@ -128,14 +128,14 @@ class EddaMCPServer:
         """
         from edda.integrations.mcp.decorators import create_durable_tool
 
-        def decorator(f: Callable) -> Workflow:
+        def decorator(f: Callable[..., Any]) -> Workflow:
             return create_durable_tool(self, f, description=description)
 
         if func is None:
             return decorator
         return decorator(func)
 
-    def asgi_app(self) -> Callable:
+    def asgi_app(self) -> Callable[..., Any]:
         """
         Create ASGI application with MCP + CloudEvents support.
 
@@ -150,8 +150,8 @@ class EddaMCPServer:
         Returns:
             ASGI callable (Starlette app)
         """
-        from starlette.requests import Request
-        from starlette.responses import Response
+        from starlette.requests import Request  # type: ignore[import-not-found]
+        from starlette.responses import Response  # type: ignore[import-not-found]
 
         # Get MCP's Starlette app (Issue #1367 workaround: use directly)
         app = self._mcp.streamable_http_app()
@@ -169,9 +169,9 @@ class EddaMCPServer:
             scope["path"] = f"/cancel/{instance_id}"
 
             # Capture response
-            response_data = {"status": 200, "headers": [], "body": b""}
+            response_data: dict[str, Any] = {"status": 200, "headers": [], "body": b""}
 
-            async def send(message: dict) -> None:
+            async def send(message: dict[str, Any]) -> None:
                 if message["type"] == "http.response.start":
                     response_data["status"] = message["status"]
                     response_data["headers"] = message.get("headers", [])
@@ -185,7 +185,7 @@ class EddaMCPServer:
             return Response(
                 content=response_data["body"],
                 status_code=response_data["status"],
-                headers=dict(response_data["headers"]),
+                headers=cast(dict[str, str], dict(response_data["headers"])),
             )
 
         # Add cancel route
@@ -193,14 +193,18 @@ class EddaMCPServer:
 
         # Add authentication middleware if token_verifier provided (AFTER adding routes)
         if self._token_verifier is not None:
-            from starlette.middleware.base import BaseHTTPMiddleware
+            from starlette.middleware.base import (  # type: ignore[import-not-found]
+                BaseHTTPMiddleware,
+            )
 
-            class AuthMiddleware(BaseHTTPMiddleware):
-                def __init__(self, app: Any, token_verifier: Callable):
+            class AuthMiddleware(BaseHTTPMiddleware):  # type: ignore[misc]
+                def __init__(self, app: Any, token_verifier: Callable[[str], bool]):
                     super().__init__(app)
                     self.token_verifier = token_verifier
 
-                async def dispatch(self, request: Request, call_next: Callable):
+                async def dispatch(
+                    self, request: Request, call_next: Callable[..., Any]
+                ) -> Response:
                     auth_header = request.headers.get("authorization", "")
                     if auth_header.startswith("Bearer "):
                         token = auth_header[7:]
@@ -211,7 +215,7 @@ class EddaMCPServer:
             # Wrap app with auth middleware
             app = AuthMiddleware(app, self._token_verifier)
 
-        return app
+        return cast(Callable[..., Any], app)
 
     async def initialize(self) -> None:
         """

{edda_framework-0.3.0 → edda_framework-0.3.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "edda-framework"
-version = "0.3.0"
+version = "0.3.1"
 description = "Lightweight Durable Execution Framework"
 authors = [
     { name = "Yasushi Itoh", email = "6240399+i2y@users.noreply.github.com" }

{edda_framework-0.3.0 → edda_framework-0.3.1}/uv.lock

@@ -664,7 +664,7 @@ wheels = [
 
 [[package]]
 name = "edda-framework"
-version = "0.3.0"
+version = "0.3.1"
 source = { editable = "." }
 dependencies = [
     { name = "a2wsgi" },