From 609f594733c56f6785f4a59b68bd91dd6569fb69 Mon Sep 17 00:00:00 2001
From: CI-DEV <154627941+IlumCI@users.noreply.github.com>
Date: Fri, 15 Aug 2025 00:01:06 +0300
Subject: [PATCH] Update mcp_schemas.py

---
 swarms/schemas/mcp_schemas.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

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,