diff --git a/src/crewai/rag/chromadb/utils.py b/src/crewai/rag/chromadb/utils.py index 93865b203..3a6a6369c 100644 --- a/src/crewai/rag/chromadb/utils.py +++ b/src/crewai/rag/chromadb/utils.py @@ -133,6 +133,9 @@ def _convert_distance_to_score( if distance_metric == "cosine": score = 1.0 - 0.5 * distance 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}")