phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-04 12:07:34 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(weaviate): correct score calcuation for cosine distance

Browse source
This commit is contained in:
ChristianZaccaria 2025-08-29 18:07:44 +01:00
parent f9794f8475
commit 0cd86e696c
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
llama_stack/providers/remote/vector_io/weaviate/weaviate.py
View file

@ -112,7 +112,10 @@ class WeaviateIndex(EmbeddingIndex):
log.exception(f"Failed to parse document: {chunk_json}") log.exception(f"Failed to parse document: {chunk_json}")
continue continue
score = 1.0 / doc.metadata.distance if doc.metadata.distance != 0 else float("inf") if doc.metadata.distance is None:
continue
# Convert cosine distance ∈ [0,2] → cosine similarity ∈ [-1,1]
score = 1.0 - float(doc.metadata.distance)
if score < score_threshold: if score < score_threshold:
continue continue