From de992527bd52795734555b729e617750911f4c93 Mon Sep 17 00:00:00 2001 From: Ismail Pelaseyed Date: Sat, 28 Oct 2023 17:28:08 +0200 Subject: [PATCH] Fix fomratting --- nagato/service/__init__.py | 2 +- nagato/service/embedding.py | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/nagato/service/__init__.py b/nagato/service/__init__.py index cf37db7..749f529 100644 --- a/nagato/service/__init__.py +++ b/nagato/service/__init__.py @@ -111,7 +111,7 @@ def query_embedding( ) -> dict: from sentence_transformers import SentenceTransformer - from nagato.service.embedding import get_vector_service, MODEL_TO_INDEX + from nagato.service.embedding import MODEL_TO_INDEX, get_vector_service embedding_model = SentenceTransformer(model, use_auth_token=config("HF_API_KEY")) vectordb = get_vector_service( diff --git a/nagato/service/embedding.py b/nagato/service/embedding.py index 647399d..44c494c 100644 --- a/nagato/service/embedding.py +++ b/nagato/service/embedding.py @@ -11,7 +11,6 @@ from nagato.service.vectordb import get_vector_service - MODEL_TO_INDEX = { "all-MiniLM-L6-v2": {"index_name": "all-minilm-l6-v2", "dimensions": 384}, "thenlper/gte-base": {"index_name": "gte-base", "dimensions": 768},