diff --git a/backend/open_webui/apps/rag/main.py b/backend/open_webui/apps/rag/main.py index f801d7423..eab197273 100644 --- a/backend/open_webui/apps/rag/main.py +++ b/backend/open_webui/apps/rag/main.py @@ -206,7 +206,9 @@ def update_reranking_model( self.ckpt = Checkpoint(name, colbert_config=ColBERTConfig()) pass - def calculate_similarity_scores(query_embeddings, document_embeddings): + def calculate_similarity_scores( + self, query_embeddings, document_embeddings + ): # Validate dimensions to ensure compatibility if query_embeddings.dim() != 3: raise ValueError(