diff --git a/ai21/models/maestro/run.py b/ai21/models/maestro/run.py
index e9abfbf5..6cfdab76 100644
--- a/ai21/models/maestro/run.py
+++ b/ai21/models/maestro/run.py
@@ -125,9 +125,19 @@ class WebSearchResult(TypedDict, total=False):
     score: float
 
 
+class ToolCallResult(TypedDict, total=False):
+    tool_name: str
+    tool_type: Literal["mcp", "http"]
+    server_label: Optional[str]
+    parameters: Dict[str, Any]
+    response: Dict[str, Any]
+    status: Literal["success", "failure"]
+
+
 class DataSources(TypedDict, total=False):
     file_search: Optional[List[FileSearchResult]]
     web_search: Optional[List[WebSearchResult]]
+    tool_calls: Optional[List[ToolCallResult]]
 
 
 class RunError(TypedDict):