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

* 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>

knowledgehub/vectorstores/chroma.py

@@ -67,6 +67,9 @@ class ChromaVectorStore(LlamaIndexVectorStore):
             collection_name = self._client.client.name
         self._client.client._client.delete_collection(collection_name)
 
+    def count(self) -> int:
+        return self._collection.count()
+
     def save(self, *args, **kwargs):
         pass
 

knowledgehub/vectorstores/in_memory.py

@@ -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)