Clean up existing tools to prepare for GentPool

gentopia/tools/__init__.py

@@ -1,20 +1,15 @@
 from .basetool import BaseTool
 from .calculator import Calculator
-from .doc_store_look_up import DocStoreLookUp
-# from .dummy_tool import DummyTool
 from .google_search import GoogleSearch
 from .llm import LLM
 from .search_doc import SearchDoc
 from .wikipedia import Wikipedia
 from .wolfram_alpha import WolframAlpha
-from .zip_code import ZipCodeRetriever
 
 
 def load_tools(name: str) -> BaseTool:
     if name == "calculator":
         return Calculator
-    elif name == "doc_store_look_up":
-        return DocStoreLookUp
     elif name == "google_search":
         return GoogleSearch
     elif name == "llm":
@@ -25,7 +20,5 @@ def load_tools(name: str) -> BaseTool:
         return Wikipedia
     elif name == "wolfram_alpha":
         return WolframAlpha
-    elif name == "zip_code":
-        return ZipCodeRetriever
     else:
         raise NotImplementedError

gentopia/tools/search_doc.py

@@ -10,10 +10,9 @@ class SearchDoc(BaseTool):
     name = "SearchDoc"
     args_schema: Optional[Type[BaseModel]] = create_model("SearchDocArgs", query=(str, ...))
     doc_path: Optional[str] = None
-    doc_name: Optional[str] = None
-    description: str = f"A vector store that searches for similar and related content in document: {doc_name}. " \
-                       f"The result is a huge chunk of text related to your search but can also " \
-                       f"contain irrelevant info. Input should be a search query."
+    doc_name: Optional[str] = ""
+    description: str = f"A search engine looking for relevant text chunk in a document: {doc_name}. " \
+                       f"Input should be a search query."
 
     def _run(self, query: AnyStr) -> AnyStr:
         loader = TextLoader(self.doc_path)