add nltk punkt resource (#7063)

2024-08-08 14:23:22 +08:00
parent 34a9dbe826
commit 169cde6c3c
2 changed files with 4 additions and 2 deletions
--- a/api/core/rag/rerank/weight_rerank.py
+++ b/api/core/rag/rerank/weight_rerank.py
@@ -159,10 +159,9 @@ class WeightRerankRunner:
            if 'score' in document.metadata:
                query_vector_scores.append(document.metadata['score'])
            else:
-                content_vector = document.metadata['vector']
                # transform to NumPy
                vec1 = np.array(query_vector)
-                vec2 = np.array(document.metadata['vector'])
+                vec2 = np.array(document.vector)

                # calculate dot product
                dot_product = np.dot(vec1, vec2)