Fix fomratting

homanp · Oct 28, 2023 · de99252 · de99252
1 parent d5eb355
commit de99252
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git 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
@@ -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},