phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-16 15:42:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

chore: review update with score_threshold

Browse source
This commit is contained in:
Anush008 2024-10-23 00:54:47 +05:30
parent d1bef44e2b
commit bc27046f36
No known key found for this signature in database
6 changed files with 24 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -142,14 +142,13 @@ async def test_query_documents(memory_settings, sample_documents):
assert_valid_response(response4)
assert len(response4.chunks) <= 2
# Score threshold is not implemented in vector memory banks
# Test case 5: Query with threshold on similarity score
# query5 = "quantum computing" # Not directly related to any document
# params5 = {"score_threshold": 0.5}
# response5 = await memory_impl.query_documents("test_bank", query5, params5)
# assert_valid_response(response5)
# print("The scores are:", response5.scores)
# assert all(score >= 0.5 for score in response5.scores)
query5 = "quantum computing" # Not directly related to any document
params5 = {"score_threshold": 0.2}
response5 = await memory_impl.query_documents("test_bank", query5, params5)
assert_valid_response(response5)
print("The scores are:", response5.scores)
assert all(score >= 0.2 for score in response5.scores)
def assert_valid_response(response: QueryDocumentsResponse):