Browse Source

add nltk punkt resource (#7063)

Jyong 8 months ago
parent
commit
169cde6c3c
2 changed files with 4 additions and 2 deletions
  1. 3 0
      api/Dockerfile
  2. 1 2
      api/core/rag/rerank/weight_rerank.py

+ 3 - 0
api/Dockerfile

@@ -55,6 +55,9 @@ ENV VIRTUAL_ENV=/app/api/.venv
 COPY --from=packages ${VIRTUAL_ENV} ${VIRTUAL_ENV}
 ENV PATH="${VIRTUAL_ENV}/bin:${PATH}"
 
+# Download nltk data
+RUN python -c "import nltk; nltk.download('punkt')"
+
 # Copy source code
 COPY . /app/api/
 

+ 1 - 2
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)