fix(vefaas): mcp server in app.py

Author: floritange

veadk/a2a/ve_a2a_server.py

@@ -21,7 +21,6 @@
 from veadk.a2a.agent_card import get_agent_card
 from veadk.a2a.ve_agent_executor import VeAgentExecutor
 from veadk.memory.short_term_memory import ShortTermMemory
-from veadk.runner import Runner
 
 
 class VeA2AServer:
@@ -49,39 +48,6 @@ def build(self) -> FastAPI:
         )
         app = app_application.build()  # build routes
 
-        runner = Runner(
-            agent=self.agent_executor.agent,
-            short_term_memory=self.agent_executor.short_term_memory,
-            app_name=self.agent_executor.app_name,
-            user_id="",
-        )
-
-        @app.post("/run_agent", operation_id="run_agent", tags=["mcp"])
-        async def run_agent(
-            user_input: str,
-            user_id: str = "unknown_user",
-            session_id: str = "unknown_session",
-        ) -> str:
-            """
-            Execute agent with user input and return final output
-            Args:
-                user_input: User's input message
-                user_id: User identifier
-                session_id: Session identifier
-            Returns:
-                Final agent response
-            """
-            # Set user_id for runner
-            runner.user_id = user_id
-
-            # Running agent and get final output
-            final_output = await runner.run(
-                messages=user_input,
-                session_id=session_id,
-            )
-
-            return final_output
-
         return app
 
 

veadk/cli/services/vefaas/template/deploy.py

@@ -63,7 +63,7 @@ async def main():
         async with client:
             # List available tools
             tools = await client.list_tools()
-            print(f"tool_0: {tools[0].__dict__}\n")
+            print(f"tools: {tools}\n")
 
             # Call run_agent tool, pass user input and session information
             res = await client.call_tool(

veadk/cli/services/vefaas/template/src/app.py

@@ -17,6 +17,7 @@
 from veadk.a2a.ve_a2a_server import init_app
 from veadk.tracing.base_tracer import BaseTracer
 from veadk.tracing.telemetry.opentelemetry_tracer import OpentelemetryTracer
+from veadk.runner import Runner
 from contextlib import asynccontextmanager
 from fastmcp import FastMCP
 from fastapi import FastAPI
@@ -72,6 +73,42 @@
     short_term_memory=short_term_memory,
 )
 
+# Add a2a app to fastmcp
+runner = Runner(
+    agent=agent,
+    short_term_memory=short_term_memory,
+    app_name=app_name,
+    user_id="",
+)
+
+
+# mcp server
+@a2a_app.post("/run_agent", operation_id="run_agent", tags=["mcp"])
+async def run_agent(
+    user_input: str,
+    user_id: str = "unknown_user",
+    session_id: str = "unknown_session",
+) -> str:
+    """
+    Execute agent with user input and return final output
+    Args:
+        user_input: User's input message
+        user_id: User identifier
+        session_id: Session identifier
+    Returns:
+        Final agent response
+    """
+    # Set user_id for runner
+    runner.user_id = user_id
+
+    # Running agent and get final output
+    final_output = await runner.run(
+        messages=user_input,
+        session_id=session_id,
+    )
+    return final_output
+
+
 mcp = FastMCP.from_fastapi(app=a2a_app, name=app_name, include_tags={"mcp"})
 
 # Create MCP ASGI app