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

fix: v1 sdk send message tests for message union rename (#6123)

Browse Source 
* add

* base:

---------

Co-authored-by: Letta Bot <noreply@letta.com>
This commit is contained in:
jnjpng
2025-11-11 18:50:05 -08:00
committed by Caren Thomas
parent d37ed2e056
commit 6139ab23d2
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/sdk_v1/integration/integration_test_send_message.py
View File

@@ -18,7 +18,7 @@ from letta_client.types import AgentState, MessageCreateParam, ToolReturnMessage
from letta_client.types.agents import (
AssistantMessage,
HiddenReasoningMessage,
LettaMessageUnion,
Message,
ReasoningMessage,
Run,
ToolCallMessage,
@@ -800,8 +800,8 @@ def accumulate_chunks(chunks: List[Any], verify_token_streaming: bool = False) -
return [m for m in messages if m is not None]
def cast_message_dict_to_messages(messages: List[Dict[str, Any]]) -> List[LettaMessageUnion]:
def cast_message(message: Dict[str, Any]) -> LettaMessageUnion:
def cast_message_dict_to_messages(messages: List[Dict[str, Any]]) -> List[Message]:
def cast_message(message: Dict[str, Any]) -> Message:
if message["message_type"] == "reasoning_message":
return ReasoningMessage(**message)
elif message["message_type"] == "assistant_message":