From 1208f19efc3e38fec9d9e9cf82b5ae9375b4df26 Mon Sep 17 00:00:00 2001
From: Ismail Pelaseyed <homanp@gmail.com>
Date: Wed, 4 Oct 2023 23:58:58 +0200
Subject: [PATCH] Fix formatting

---
 lib/api/ingest.py        |  2 +-
 lib/service/embedding.py | 16 +++++++---------
 lib/service/vectordb.py  | 11 ++++++-----
 3 files changed, 14 insertions(+), 15 deletions(-)

diff --git a/lib/api/ingest.py b/lib/api/ingest.py
index 35f2a1b..af4b626 100644
--- a/lib/api/ingest.py
+++ b/lib/api/ingest.py
@@ -1,7 +1,7 @@
 from fastapi import APIRouter
-from lib.utils.prisma import prisma
 
 from lib.service.embedding import EmbeddingService
+from lib.utils.prisma import prisma
 
 router = APIRouter()
 
diff --git a/lib/service/embedding.py b/lib/service/embedding.py
index e914959..9aecd9d 100644
--- a/lib/service/embedding.py
+++ b/lib/service/embedding.py
@@ -1,17 +1,15 @@
-import requests
-
-from numpy import ndarray
-from typing import Any, Dict, List, Union
-from prisma.models import Datasource
 from tempfile import NamedTemporaryFile
-from decouple import config
-from sentence_transformers import SentenceTransformer
+from typing import List, Union
 
-from llama_index import SimpleDirectoryReader
+import requests
+from decouple import config
+from llama_index import Document, SimpleDirectoryReader
 from llama_index.node_parser import SimpleNodeParser
-from llama_index import Document
+from numpy import ndarray
+from sentence_transformers import SentenceTransformer
 
 from lib.service.vectordb import get_vector_service
+from prisma.models import Datasource
 
 
 class EmbeddingService:
diff --git a/lib/service/vectordb.py b/lib/service/vectordb.py
index 0fd3520..de59374 100644
--- a/lib/service/vectordb.py
+++ b/lib/service/vectordb.py
@@ -1,9 +1,9 @@
-import pinecone
-
-from numpy import ndarray
-from typing import List
 from abc import ABC, abstractmethod
+from typing import List
+
+import pinecone
 from decouple import config
+from numpy import ndarray
 
 
 class VectorDBService(ABC):
@@ -27,7 +27,8 @@ def __init__(self, index_name: str, dimension: int, namespace: str = None):
             index_name=index_name, dimension=dimension, namespace=namespace
         )
         pinecone.init(api_key=config("PINECONE_API_KEY"))
-        # Create a new vector index if it doesn't exist dimensions should be passed in the arguments
+        # Create a new vector index if it doesn't
+        # exist dimensions should be passed in the arguments
         if index_name not in pinecone.list_indexes():
             pinecone.create_index(
                 name=index_name, metric="cosine", shards=1, dimension=dimension