Update mcp_agent.py

mcp_agent.py

@@ -1,11 +1,11 @@
 # mcp_agent.py
 import asyncio
-from typing import List, Dict, Optional
+from typing import List, Dict, Optional, Generator, AsyncGenerator
 from fastmcp import Client
 from fastmcp.client.transports import SSETransport
 from llama_index.core.workflow import Context
 from llama_index.tools.mcp import McpToolSpec
-from llama_index.core.agent.workflow import FunctionAgent, ToolCall
+from llama_index.core.agent.workflow import FunctionAgent, ToolCall, ToolCallResult
 from mcp.client.sse import sse_client as original_sse_client
 from mcp.client.session import ClientSession
 from datetime import timedelta
@@ -91,7 +91,7 @@ class MCPAgent:
         handler = self.agent.run(message, ctx=self.agent_context)
         return str(await handler)
     
-    async def stream_response(self, message: str):
+    async def stream_response(self, message: str) -> AsyncGenerator[Dict, None]:
         """Genera la respuesta del agente en formato de streaming"""
         if not self.agent:
             await self.initialize()