Update mcp_schemas.py

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,