Fimeg/letta-server
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

feat: Add new tool_calls field to ToolCallMessage [LET-5333] (#5267)

Browse Source 
* Add new tool_calls field to ToolCallMessage

* fern autogen
This commit is contained in:
Matthew Zhou
2025-10-08 16:07:14 -07:00
committed by Caren Thomas
parent caa3f8b28d
commit 45101fd97a
2 changed files with 28 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
fern/openapi.json
View File

@@ -33842,7 +33842,25 @@
"$ref": "#/components/schemas/ToolCallDelta"
}
],
"title": "Tool Call"
"title": "Tool Call",
"deprecated": true
},
"tool_calls": {
"anyOf": [
{
"items": {
"$ref": "#/components/schemas/ToolCall"
},
"type": "array"
},
{
"$ref": "#/components/schemas/ToolCallDelta"
},
{
"type": "null"
}
],
"title": "Tool Calls"
}
},
"type": "object",

11
letta/schemas/letta_message.py
View File

@@ -190,7 +190,8 @@ class ToolCallMessage(LettaMessage):
message_type: Literal[MessageType.tool_call_message] = Field(
default=MessageType.tool_call_message, description="The type of the message."
)
tool_call: Union[ToolCall, ToolCallDelta]
tool_call: Union[ToolCall, ToolCallDelta] = Field(..., deprecated=True)
tool_calls: Optional[Union[List[ToolCall], ToolCallDelta]] = None
def model_dump(self, *args, **kwargs):
"""
@@ -198,8 +199,14 @@ class ToolCallMessage(LettaMessage):
"""
kwargs["exclude_none"] = True
data = super().model_dump(*args, **kwargs)
if isinstance(data["tool_call"], dict):
if isinstance(data.get("tool_call"), dict):
data["tool_call"] = {k: v for k, v in data["tool_call"].items() if v is not None}
if isinstance(data.get("tool_calls"), dict):
data["tool_calls"] = {k: v for k, v in data["tool_calls"].items() if v is not None}
elif isinstance(data.get("tool_calls"), list):
data["tool_calls"] = [
{k: v for k, v in item.items() if v is not None} if isinstance(item, dict) else item for item in data["tool_calls"]
]
return data
class Config: