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

fix: remove letta ping schema override (#8790)

Browse Source
This commit is contained in:
cthomas
2026-01-15 15:37:33 -08:00
committed by Sarah Wooders
parent 4578a9203f
commit 6b6ca91183
2 changed files with 1 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
letta/schemas/letta_message.py
View File

@@ -458,24 +458,6 @@ def create_letta_message_union_schema():
}
def create_letta_ping_schema():
return {
"properties": {
"message_type": {
"type": "string",
"const": "ping",
"title": "Message Type",
"description": "The type of the message.",
"default": "ping",
}
},
"type": "object",
"required": ["message_type"],
"title": "LettaPing",
"description": "Ping messages are a keep-alive to prevent SSE streams from timing out during long running requests.",
}
def create_letta_error_message_schema():
return {
"properties": {

3
letta/server/rest_api/app.py
View File

@@ -60,7 +60,7 @@ from letta.jobs.scheduler import start_scheduler_with_leader_election
from letta.log import get_logger
from letta.orm.errors import DatabaseTimeoutError, ForeignKeyConstraintViolationError, NoResultFound, UniqueConstraintViolationError
from letta.otel.tracing import get_trace_id
from letta.schemas.letta_message import create_letta_error_message_schema, create_letta_message_union_schema, create_letta_ping_schema
from letta.schemas.letta_message import create_letta_error_message_schema, create_letta_message_union_schema
from letta.schemas.letta_message_content import (
create_letta_assistant_message_content_union_schema,
create_letta_message_content_union_schema,
@@ -106,7 +106,6 @@ def generate_openapi_schema(app: FastAPI):
letta_docs["components"]["schemas"]["LettaMessageContentUnion"] = create_letta_message_content_union_schema()
letta_docs["components"]["schemas"]["LettaAssistantMessageContentUnion"] = create_letta_assistant_message_content_union_schema()
letta_docs["components"]["schemas"]["LettaUserMessageContentUnion"] = create_letta_user_message_content_union_schema()
letta_docs["components"]["schemas"]["LettaPing"] = create_letta_ping_schema()
letta_docs["components"]["schemas"]["LettaErrorMessage"] = create_letta_error_message_schema()
# Update the app's schema with our modified version