| @@ -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/ | |||
| @@ -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) | |||