phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-20 09:18:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

weaviate fixes

Browse source
This commit is contained in:
Dinesh Yeduguru 2024-12-10 16:45:33 -08:00
parent 0e451525e5
commit b509d59dcd
2 changed files with 11 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/providers/tests/memory/test_memory.py
View file

@ -169,13 +169,13 @@ class TestMemory:
# Test case 5: Query with threshold on similarity score
query5 = "quantum computing" # Not directly related to any document
params5 = {"score_threshold": 0.2}
params5 = {"score_threshold": 0.01}
response5 = await memory_impl.query_documents(
registered_bank.memory_bank_id, query5, params5
)
assert_valid_response(response5)
print("The scores are:", response5.scores)
assert all(score >= 0.2 for score in response5.scores)
assert all(score >= 0.01 for score in response5.scores)
def assert_valid_response(response: QueryDocumentsResponse):