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

fix: add message union type ref for LettaResponse (#507)

Browse Source
This commit is contained in:
cthomas
2025-01-06 10:05:03 -08:00
committed by GitHub
parent 9cd326ddb8
commit e2d94614e6
1 changed files with 8 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
letta/schemas/letta_response.py
View File

@@ -29,19 +29,20 @@ class LettaResponse(BaseModel):
json_schema_extra={
"items": {
"oneOf": [
{"x-ref-name": "SystemMessage"},
{"x-ref-name": "UserMessage"},
{"x-ref-name": "ReasoningMessage"},
{"x-ref-name": "ToolCallMessage"},
{"x-ref-name": "ToolReturnMessage"},
{"x-ref-name": "AssistantMessage"},
{"$ref": "#/components/schemas/SystemMessage-Output"},
{"$ref": "#/components/schemas/UserMessage-Output"},
{"$ref": "#/components/schemas/ReasoningMessage"},
{"$ref": "#/components/schemas/ToolCallMessage"},
{"$ref": "#/components/schemas/ToolReturnMessage"},
{"$ref": "#/components/schemas/AssistantMessage-Output"},
],
"discriminator": {"propertyName": "message_type"},
}
},
)
usage: LettaUsageStatistics = Field(
..., description="The usage statistics of the agent.", json_schema_extra={"x-ref-name": "LettaUsageStatistics"}
...,
description="The usage statistics of the agent.",
)
def __str__(self):