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

feat: add step id on letta message (#1974)

Browse Source
This commit is contained in:
cthomas
2025-05-01 11:21:59 -07:00
committed by GitHub
parent a6040c9e92
commit 9722596a74
2 changed files with 16 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
letta/schemas/letta_message.py
View File

@@ -48,6 +48,7 @@ class LettaMessage(BaseModel):
message_type: MessageType = Field(..., description="The type of the message.")
otid: Optional[str] = None
sender_id: Optional[str] = None
step_id: Optional[str] = None
@field_serializer("date")
def serialize_datetime(self, dt: datetime, _info):

15
letta/schemas/message.py
View File

@@ -138,6 +138,11 @@ class Message(BaseMessage):
created_at (datetime): The time the message was created.
tool_calls (List[OpenAIToolCall,]): The list of tool calls requested.
tool_call_id (str): The id of the tool call.
step_id (str): The id of the step that this message was created in.
otid (str): The offline threading id associated with this message.
tool_returns (List[ToolReturn]): The list of tool returns requested.
group_id (str): The multi-agent group that the message was sent in.
sender_id (str): The id of the sender of the message, can be an identity id or agent id.
"""
@@ -253,6 +258,7 @@ class Message(BaseMessage):
name=self.name,
otid=otid,
sender_id=self.sender_id,
step_id=self.step_id,
)
)
# Otherwise, we may have a list of multiple types
@@ -270,6 +276,7 @@ class Message(BaseMessage):
name=self.name,
otid=otid,
sender_id=self.sender_id,
step_id=self.step_id,
)
)
elif isinstance(content_part, ReasoningContent):
@@ -283,6 +290,7 @@ class Message(BaseMessage):
signature=content_part.signature,
name=self.name,
otid=otid,
step_id=self.step_id,
)
)
elif isinstance(content_part, RedactedReasoningContent):
@@ -296,6 +304,7 @@ class Message(BaseMessage):
name=self.name,
otid=otid,
sender_id=self.sender_id,
step_id=self.step_id,
)
)
elif isinstance(content_part, OmittedReasoningContent):
@@ -308,6 +317,7 @@ class Message(BaseMessage):
state="omitted",
name=self.name,
otid=otid,
step_id=self.step_id,
)
)
else:
@@ -334,6 +344,7 @@ class Message(BaseMessage):
name=self.name,
otid=otid,
sender_id=self.sender_id,
step_id=self.step_id,
)
)
else:
@@ -349,6 +360,7 @@ class Message(BaseMessage):
name=self.name,
otid=otid,
sender_id=self.sender_id,
step_id=self.step_id,
)
)
elif self.role == MessageRole.tool:
@@ -392,6 +404,7 @@ class Message(BaseMessage):
name=self.name,
otid=self.id.replace("message-", ""),
sender_id=self.sender_id,
step_id=self.step_id,
)
)
elif self.role == MessageRole.user:
@@ -410,6 +423,7 @@ class Message(BaseMessage):
name=self.name,
otid=self.otid,
sender_id=self.sender_id,
step_id=self.step_id,
)
)
elif self.role == MessageRole.system:
@@ -427,6 +441,7 @@ class Message(BaseMessage):
name=self.name,
otid=self.otid,
sender_id=self.sender_id,
step_id=self.step_id,
)
)
else: