Public/kotaemon
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update Base interface of Index/Retrieval pipeline (#36)

Browse Source 
* add base Tool

* minor update test_tool

* update test dependency

* update test dependency

* Fix namespace conflict

* update test

* add base Agent Interface, add ReWoo Agent

* minor update

* update test

* fix typo

* remove unneeded print

* update rewoo agent

* add LLMTool

* update BaseAgent type

* add ReAct agent

* add ReAct agent

* minor update

* minor update

* minor update

* minor update

* update base reader with BaseComponent

* add splitter

* update agent and tool

* update vectorstores

* update load/save for indexing and retrieving pipeline

* update test_agent for more use-cases

* add missing dependency for test

* update test case for in memory vectorstore

* add TextSplitter to BaseComponent

* update type hint basetool

---------

Co-authored-by: trducng <trungduc1992@gmail.com>
This commit is contained in:
Tuan Anh Nguyen Dang (Tadashi_Cin)
2023-10-04 14:27:44 +07:00
committed by GitHub
parent 49ed3f6994
commit 56bc41b673
13 changed files with 302 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
knowledgehub/vectorstores/in_memory.py
View File

@@ -44,9 +44,7 @@ class InMemoryVectorStore(LlamaIndexVectorStore):
"""
self._client.persist(persist_path=save_path, fs=fs)
def load(
self, load_path: str, fs: Optional[fsspec.AbstractFileSystem] = None
) -> "InMemoryVectorStore":
def load(self, load_path: str, fs: Optional[fsspec.AbstractFileSystem] = None):
"""Create a SimpleKVStore from a load directory.
@@ -54,4 +52,4 @@ class InMemoryVectorStore(LlamaIndexVectorStore):
load_path: Path of loading vector.
fs: An abstract super-class for pythonic file-systems
"""
return self._client.from_persist_path(persist_path=load_path, fs=fs)
self._client = self._client.from_persist_path(persist_path=load_path, fs=fs)