Skip to content

Commit

Permalink
Fix formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
homanp committed Oct 4, 2023
1 parent b2f2d4d commit 1208f19
Show file tree
Hide file tree
Showing 3 changed files with 14 additions and 15 deletions.
2 changes: 1 addition & 1 deletion lib/api/ingest.py
Original file line number Diff line number Diff line change
@@ -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()

Expand Down
16 changes: 7 additions & 9 deletions lib/service/embedding.py
Original file line number Diff line number Diff line change
@@ -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:
Expand Down
11 changes: 6 additions & 5 deletions lib/service/vectordb.py
Original file line number Diff line number Diff line change
@@ -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):
Expand All @@ -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
Expand Down

0 comments on commit 1208f19

Please sign in to comment.