[AUR-362] Add In-memory vector store (#22)

* [AUR-362] Add In-memory vector store

* [AUR-362] fix delete fun input format

* [AUR-362] revise persist and from persist path to save and load

* [AUR-362] revise simple.py to in_memory.py
This commit is contained in:
cin-jacky
2023-09-20 17:51:50 +09:00
committed by GitHub
parent b794051653
commit c329c4c03f
4 changed files with 104 additions and 2 deletions

View File

@@ -1,6 +1,7 @@
from abc import ABC, abstractmethod
from typing import Any, List, Optional, Tuple, Type, Union
from llama_index.schema import NodeRelationship, RelatedNodeInfo
from llama_index.vector_stores.types import BasePydanticVectorStore
from llama_index.vector_stores.types import VectorStore as LIVectorStore
from llama_index.vector_stores.types import VectorStoreQuery
@@ -118,6 +119,9 @@ class LlamaIndexVectorStore(BaseVectorStore):
if ids is not None:
for node, id in zip(nodes, ids):
node.id_ = id
node.relationships = {
NodeRelationship.SOURCE: RelatedNodeInfo(node_id=id)
}
return self._client.add(nodes=nodes) # type: ignore