Merge pull request #160 from hakeemo/main

Ask user for more context to finish a task

src/sherpa_ai/task_agent.py

@@ -203,7 +203,7 @@ def run(self, task: str) -> str:
                 tool = tools[action.name]
 
                 if tool.name == "UserInput":
-                    return {"type": "user_input", "query": action.args["query"]}
+                    return action.args["query"]
 
                 try:
                     observation = tool.run(action.args)

src/sherpa_ai/tools.py

@@ -15,14 +15,10 @@
 
 
 def get_tools(memory):
-    prompt = (
-        "You are an assistant helping user solve the task. Perform the task as writen"
-        "in the instruction.\nTask: {input}\nResult: "
-    )
-    prompt = Prompt.from_template(prompt)
     tools = []
 
     tools.append(ContextTool(memory=memory))
+    tools.append(UserInputTool())
 
     if cfg.SERPER_API_KEY is not None:
         search_tool = SearchTool(api_wrapper=GoogleSerperAPIWrapper())
@@ -151,7 +147,7 @@ class UserInputTool(BaseTool):
     name = "UserInput"
     description = (
         "Access the user input for the task."
-        "You use this tool if you need further clarification of the task from the user."
+        "You use this tool if you need more context and would like to ask clarifying questions to solve the task"
     )
 
     def _run(self, query: str) -> str: