broken imports

config.py

@@ -15,7 +15,6 @@ from langchain.vectorstores import Pinecone
 from langchain.embeddings.openai import OpenAIEmbeddings
 
 from magic.prompts import PROMPT, EXAMPLE_PROMPT
-from magic.vectordb import VectorIndex
 from magic.self_query_retriever import SelfQueryRetriever
 
 from utils import get_courses
@@ -38,7 +37,7 @@ def set_openai_key(key):
 
 class State:
     settings: Settings
-    store: Pinecone | VectorIndex
+    store: Pinecone
     chain: RetrievalQAWithSourcesChain
     courses: list
 
@@ -49,9 +48,6 @@ class State:
 
         # OPENAI
         set_openai_key(self.settings.OPENAI_API_KEY)
-
-        # LOCAL VECTORSTORE
-        # self.store: VectorIndex = VectorIndex()
         
         # PINECONE VECTORSTORE
         embeddings = OpenAIEmbeddings()

magic/vectordb.py

@@ -1,13 +0,0 @@
-import logging
-logger = logging.getLogger()
-import os
-from langchain.embeddings.openai import OpenAIEmbeddings
-from langchain.vectorstores import Chroma
-from ingest import PERSIST_DIR
-
-embedding = OpenAIEmbeddings()
-class VectorIndex(Chroma):
-    def __init__(self):
-        if len(os.listdir(PERSIST_DIR)) < 2: # check if there are files in the directory
-            logger.error(f"VectorIndex: No files in {PERSIST_DIR}, have you run ingest.py?")
-        super().__init__(persist_directory=PERSIST_DIR, embedding_function=embedding)