fix: vector cache generation

codeqai/vector_store.py

@@ -56,7 +56,7 @@ def index_documents(self, documents: list[Document]):
         index_to_docstore_id = self.db.index_to_docstore_id
         for i in range(len(documents)):
             document = self.db.docstore.search(index_to_docstore_id[i])
-            if document and document is type(Document):
+            if document and isinstance(document, Document):
                 # Check if the document is already present in the vector cache
                 # if yes, then add the vector id to the vector cache entry
                 if self.vector_cache.get(document.metadata["filename"]):

tests/vector_store_test.py

@@ -2,7 +2,7 @@
 
 import pytest
 from langchain.schema import Document
-from langchain_community.embeddings import FakeEmbeddings
+from langchain_core.embeddings import FakeEmbeddings
 
 from codeqai.cache import get_cache_path
 from codeqai.vector_store import VectorStore