From 39e8792ae5c2531fa2b39943256e707220bec58e Mon Sep 17 00:00:00 2001 From: Greyson LaLonde Date: Thu, 18 Sep 2025 18:36:33 -0400 Subject: [PATCH] fix: add l2 distance metric support for backward compatibility (#3540) --- src/crewai/rag/chromadb/utils.py | 3 +++ 1 file changed, 3 insertions(+) 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}")