andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-05-01 23:32:39 +00:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

fix: add l2 distance metric support for backward compatibility (#3540)

Browse Source
This commit is contained in:
Greyson LaLonde
2025-09-18 18:36:33 -04:00
committed by GitHub
parent 2f682e1564
commit 39e8792ae5
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/crewai/rag/chromadb/utils.py
View File

@@ -133,6 +133,9 @@ def _convert_distance_to_score(
if distance_metric == "cosine": if distance_metric == "cosine":
score = 1.0 - 0.5 * distance score = 1.0 - 0.5 * distance
return max(0.0, min(1.0, score)) return max(0.0, min(1.0, score))
if distance_metric == "l2":
score = 1.0 / (1.0 + distance)
return max(0.0, min(1.0, score))
raise ValueError(f"Unsupported distance metric: {distance_metric}") raise ValueError(f"Unsupported distance metric: {distance_metric}")