fix client-sdk memory/safety test

This commit is contained in:
Xi Yan 2024-12-26 15:48:28 -08:00
parent b6aca4c8bb
commit 70db039ff4
2 changed files with 2 additions and 5 deletions

View file

@ -61,9 +61,7 @@ def test_memory_bank(llama_stack_client):
# query documents
response = llama_stack_client.memory.query(
bank_id=memory_bank_id,
query=[
"How do I use lora",
],
query="How do I use lora",
)
assert len(response.chunks) > 0