Skip to content

Commit

Permalink
fix: do a pre-validation validation on the inputs assigned to `agent_…
Browse files Browse the repository at this point in the history
…state.message_ids` (#2098)
  • Loading branch information
cpacker authored Nov 25, 2024
1 parent 251619e commit 84031a9
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions letta/agent.py
Original file line number Diff line number Diff line change
Expand Up @@ -1325,6 +1325,12 @@ def remove_function(self, function_name: str) -> str:

def update_state(self) -> AgentState:
message_ids = [msg.id for msg in self._messages]

# Assert that these are all strings
if any(not isinstance(m_id, str) for m_id in message_ids):
warnings.warn(f"Non-string message IDs found in agent state: {message_ids}")
message_ids = [m_id for m_id in message_ids if isinstance(m_id, str)]

assert isinstance(self.memory, Memory), f"Memory is not a Memory object: {type(self.memory)}"

# override any fields that may have been updated
Expand Down

0 comments on commit 84031a9

Please sign in to comment.