Skip to content

Commit 81ba538

Browse files
committed
feat: add tool_calls list to Maestro run data sources
1 parent 7115da6 commit 81ba538

1 file changed

Lines changed: 10 additions & 0 deletions

File tree

ai21/models/maestro/run.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -125,9 +125,19 @@ class WebSearchResult(TypedDict, total=False):
125125
score: float
126126

127127

128+
class ToolCallResult(TypedDict, total=False):
129+
tool_name: str
130+
tool_type: Literal["mcp", "http"]
131+
server_label: Optional[str]
132+
parameters: dict[str, Any]
133+
response: dict[str, Any]
134+
status: Literal["success", "failure"]
135+
136+
128137
class DataSources(TypedDict, total=False):
129138
file_search: Optional[List[FileSearchResult]]
130139
web_search: Optional[List[WebSearchResult]]
140+
tool_calls: Optional[List[ToolCallResult]]
131141

132142

133143
class RunError(TypedDict):

0 commit comments

Comments
 (0)