forked from phoenix-oss/llama-stack-mirror
fix client-sdk memory/safety test
This commit is contained in:
parent
b6aca4c8bb
commit
70db039ff4
2 changed files with 2 additions and 5 deletions
|
@ -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
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue