diff --git a/swarms/schemas/mcp_schemas.py b/swarms/schemas/mcp_schemas.py index 7a61f086..c1bf72f2 100644 --- a/swarms/schemas/mcp_schemas.py +++ b/swarms/schemas/mcp_schemas.py @@ -1,5 +1,5 @@ from pydantic import BaseModel, Field -from typing import Dict, List, Any, Optional, Literal +from typing import Dict, List, Any, Optional, Literal, Callable class MCPConnection(BaseModel): @@ -70,6 +70,11 @@ class MCPConnection(BaseModel): description="Timeout for streaming operations in seconds" ) + streaming_callback: Optional[Callable[[str], None]] = Field( + default=None, + description="Callback function for streaming chunks" + ) + # Tool configurations tool_configurations: Optional[Dict[Any, Any]] = Field( default=None,