feat: move source_id to path variable (#1171)

This commit is contained in:
Charles Packer
2024-03-20 21:37:52 -07:00
committed by GitHub
parent d0dfa7c34e
commit 6ee95c31e8
5 changed files with 40 additions and 29 deletions

View File

@@ -157,7 +157,7 @@ def test_attach_source_to_agent(server, user_id, agent_id):
assert len(passages_before) == 0
# attach source
server.attach_source_to_agent(user_id, agent_id, "test_source")
server.attach_source_to_agent(user_id=user_id, agent_id=agent_id, source_name="test_source")
# check archival memory size
passages_after = server.get_agent_archival(user_id=user_id, agent_id=agent_id, start=0, count=10000)