From 156359ab84414e2bba7d513cf57cc2c3ddf20e49 Mon Sep 17 00:00:00 2001 From: Cameron Morin Date: Wed, 27 Nov 2024 21:25:35 +0000 Subject: [PATCH] Run pre-commit hook --- .../retrievers/opensearch/langchain/requirements.txt | 4 ++-- .../opensearch/langchain/retriever_opensearch.py | 11 ++++------- 2 files changed, 6 insertions(+), 9 deletions(-) diff --git a/comps/retrievers/opensearch/langchain/requirements.txt b/comps/retrievers/opensearch/langchain/requirements.txt index 7fb987ccb8..5690118bbb 100644 --- a/comps/retrievers/opensearch/langchain/requirements.txt +++ b/comps/retrievers/opensearch/langchain/requirements.txt @@ -3,14 +3,14 @@ easyocr fastapi langchain_community langchain_huggingface +numpy opensearch-py opentelemetry-api opentelemetry-exporter-otlp opentelemetry-sdk prometheus-fastapi-instrumentator +pydantic pymupdf sentence_transformers shortuuid uvicorn -pydantic -numpy diff --git a/comps/retrievers/opensearch/langchain/retriever_opensearch.py b/comps/retrievers/opensearch/langchain/retriever_opensearch.py index f5b2602307..c570cb6db5 100644 --- a/comps/retrievers/opensearch/langchain/retriever_opensearch.py +++ b/comps/retrievers/opensearch/langchain/retriever_opensearch.py @@ -3,9 +3,9 @@ import os import time -import numpy as np -from typing import Union, List, Callable +from typing import Callable, List, Union +import numpy as np from langchain_community.embeddings import HuggingFaceBgeEmbeddings from langchain_community.vectorstores import OpenSearchVectorSearch from langchain_huggingface import HuggingFaceEndpointEmbeddings @@ -37,11 +37,8 @@ async def search_all_embeddings_vectors( - embeddings: Union[conlist(float, min_length=0), List[conlist(float, min_length=0)]], - func: Callable, - *args, - **kwargs - ): + embeddings: Union[conlist(float, min_length=0), List[conlist(float, min_length=0)]], func: Callable, *args, **kwargs +): try: if not isinstance(embeddings, np.ndarray): embeddings = np.array(embeddings)