fix: do a pre-validation validation on the inputs assigned to agent_state.message_ids (#2098)

This commit is contained in:
Charles Packer
2024-11-25 10:44:14 -08:00
committed by GitHub
parent 60cabae25a
commit f237717ce4

View File

@@ -1325,6 +1325,12 @@ class Agent(BaseAgent):
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