Update retrieving + agent pipeline (#71)

2023-11-14 16:40:13 +07:00
parent 693ed39de4
commit 640962e916
8 changed files with 65 additions and 21 deletions
--- a/knowledgehub/pipelines/tools/wikipedia.py
+++ b/knowledgehub/pipelines/tools/wikipedia.py
@@ -52,7 +52,8 @@ class WikipediaTool(BaseTool):
    description = (
        "Search engine from Wikipedia, retrieving relevant wiki page. "
        "Useful when you need to get holistic knowledge about people, "
-        "places, companies, historical events, or other subjects."
+        "places, companies, historical events, or other subjects. "
+        "Input should be a search query."
    )
    args_schema: Optional[Type[BaseModel]] = WikipediaArgs
    doc_store: Any = None