Skip to content

Commit 06e4c7f

Browse files
authored
feat: add input validation for messages field (#4328)
1 parent a7ed7ed commit 06e4c7f

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

letta/helpers/message_helper.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,8 +40,7 @@ def _convert_message_create_to_message(
4040
assert isinstance(message_create, MessageCreate)
4141

4242
# Extract message content
43-
if isinstance(message_create.content, str):
44-
assert message_create.content != "", "Message content must not be empty"
43+
if isinstance(message_create.content, str) and message_create.content != "":
4544
message_content = [TextContent(text=message_create.content)]
4645
elif isinstance(message_create.content, list) and len(message_create.content) > 0:
4746
message_content = message_create.content

letta/server/rest_api/routers/v1/agents.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1072,6 +1072,8 @@ async def send_message(
10721072
Process a user message and return the agent's response.
10731073
This endpoint accepts a message from a user and processes it through the agent.
10741074
"""
1075+
if len(request.messages) == 0:
1076+
raise ValueError("Messages must not be empty")
10751077
request_start_timestamp_ns = get_utc_timestamp_ns()
10761078
MetricRegistry().user_message_counter.add(1, get_ctx_attributes())
10771079

0 commit comments

Comments
 (0)