feat: Add lock around loading agent (#2141)

This commit is contained in:
Matthew Zhou
2024-12-02 17:46:48 -08:00
committed by GitHub
parent 1e5d74b4a7
commit 3b1f579aba
7 changed files with 48 additions and 58 deletions

View File

@@ -211,11 +211,10 @@ def check_agent_recall_chat_memory(filename: str) -> LettaResponse:
cleanup(client=client, agent_uuid=agent_uuid)
human_name = "BananaBoy"
agent_state = setup_agent(client, filename, memory_human_str=f"My name is {human_name}")
print("MEMORY", agent_state.memory.get_block("human").value)
response = client.user_message(agent_id=agent_state.id, message="Repeat my name back to me.")
agent_state = setup_agent(client, filename, memory_human_str=f"My name is {human_name}.")
response = client.user_message(
agent_id=agent_state.id, message="Repeat my name back to me. You should search in your human memory block."
)
# Basic checks
assert_sanity_checks(response)