From 81ba53848f6d2b286a034cefa9564fb4a2d1dc3e Mon Sep 17 00:00:00 2001
From: nazaranton <97673072+nazaranton@users.noreply.github.com>
Date: Tue, 4 Nov 2025 08:47:01 +0200
Subject: [PATCH 1/2] feat: add tool_calls list to Maestro run data sources

---
 ai21/models/maestro/run.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/ai21/models/maestro/run.py b/ai21/models/maestro/run.py
index e9abfbf5..21e75549 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):

From bbeacae40fcd36e48722afe731b43dc2ce4608e6 Mon Sep 17 00:00:00 2001
From: nazaranton <97673072+nazaranton@users.noreply.github.com>
Date: Tue, 4 Nov 2025 13:51:35 +0200
Subject: [PATCH 2/2] fix: unittests

---
 ai21/models/maestro/run.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/ai21/models/maestro/run.py b/ai21/models/maestro/run.py
index 21e75549..6cfdab76 100644
--- a/ai21/models/maestro/run.py
+++ b/ai21/models/maestro/run.py
@@ -129,8 +129,8 @@ 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]
+    parameters: Dict[str, Any]
+    response: Dict[str, Any]
     status: Literal["success", "failure"]