Timothy J. Baek 7 months ago
parent
commit
bc6f23f82f
1 changed files with 3 additions and 1 deletions
  1. 3 1
      backend/open_webui/apps/rag/main.py

+ 3 - 1
backend/open_webui/apps/rag/main.py

@@ -206,7 +206,9 @@ def update_reranking_model(
                     self.ckpt = Checkpoint(name, colbert_config=ColBERTConfig())
                     self.ckpt = Checkpoint(name, colbert_config=ColBERTConfig())
                     pass
                     pass
 
 
-                def calculate_similarity_scores(query_embeddings, document_embeddings):
+                def calculate_similarity_scores(
+                    self, query_embeddings, document_embeddings
+                ):
                     # Validate dimensions to ensure compatibility
                     # Validate dimensions to ensure compatibility
                     if query_embeddings.dim() != 3:
                     if query_embeddings.dim() != 3:
                         raise ValueError(
                         raise ValueError(