Skip to content
Snippets Groups Projects
Unverified Commit f1ecae7b authored by Jerry Liu's avatar Jerry Liu Committed by GitHub
Browse files

fix insert for simple vector index (#934)

parent d036d92f
Branches
Tags
No related merge requests found
......@@ -15,6 +15,7 @@ from gpt_index.data_structs.data_structs import (
SimpleIndexDict,
WeaviateIndexDict,
)
from gpt_index.schema import BaseDocument
from gpt_index.embeddings.base import BaseEmbedding
from gpt_index.indices.base import DOCUMENTS_INPUT, BaseGPTIndex
from gpt_index.indices.query.base import BaseGPTIndexQuery
......@@ -102,6 +103,14 @@ class GPTSimpleVectorIndex(GPTVectorStoreIndex):
# update docstore with current struct
self._docstore.add_documents([self.index_struct], allow_update=True)
def _insert(self, document: BaseDocument, **insert_kwargs: Any) -> None:
"""Insert a document."""
super()._insert(document, **insert_kwargs)
# TODO: Temporary hack to also store embeddings in index_struct
vector_store = cast(SimpleVectorStore, self._vector_store)
embedding_dict = vector_store._data.embedding_dict
self._index_struct.embeddings_dict = embedding_dict
@classmethod
def get_query_map(self) -> Dict[str, Type[BaseGPTIndexQuery]]:
"""Get query map."""
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment