Ask user command

loopgpt/tools/__init__.py

@@ -4,6 +4,7 @@
     DeleteAgent,
     ListAgents,
 )
+from loopgpt.tools.user_manager import AskUser
 from loopgpt.tools.base_tool import BaseTool
 from loopgpt.tools.browser import Browser
 from loopgpt.tools.code import ExecutePythonFile, ReviewCode, ImproveCode, WriteTests
@@ -55,4 +56,6 @@ def builtin_tools():
         WriteTests,
         ExecutePythonFile,
         EvaluateMath,
+        AskUser,
+        Shell,
     ]

loopgpt/tools/agent_manager.py

@@ -3,11 +3,11 @@
 from uuid import uuid4
 
 
-class _AgentMangerTool(BaseTool):
+class _AgentManagerTool(BaseTool):
     pass
 
 
-class CreateAgent(_AgentMangerTool):
+class CreateAgent(_AgentManagerTool):
     @property
     def args(self):
         return {
@@ -36,7 +36,7 @@ def run(self, name="", task="", prompt=""):
         return {"uuid": id, "resp": resp}
 
 
-class MessageAgent(_AgentMangerTool):
+class MessageAgent(_AgentManagerTool):
     @property
     def args(self):
         return {
@@ -55,7 +55,7 @@ def run(self, id, message):
         return {"resp": resp}
 
 
-class DeleteAgent(_AgentMangerTool):
+class DeleteAgent(_AgentManagerTool):
     @property
     def args(self):
         return {"id": "Agent id"}
@@ -72,7 +72,7 @@ def run(self, id):
             return {f"resp": "Specified agent (id={id} not found.)"}
 
 
-class ListAgents(_AgentMangerTool):
+class ListAgents(_AgentManagerTool):
     @property
     def args(self):
         return {}

loopgpt/tools/shell.py

@@ -5,7 +5,7 @@
 class Shell(BaseTool):
     @property
     def desc(self):
-        return "Command line shell. (Non interactive commands only)"
+        return "Use this command to execute any terminal commands. (Non interactive commands only)"
 
     @property
     def args(self):

loopgpt/tools/user_manager.py

@@ -0,0 +1,24 @@
+from loopgpt.tools.base_tool import BaseTool
+from loopgpt.utils.spinner import hide_spinner
+
+
+class _UserManagerTool(BaseTool):
+    pass
+
+
+class AskUser(_UserManagerTool):
+    @property
+    def args(self):
+        return {"message": "The question for the user."}
+
+    @property
+    def resp(self):
+        return {"response": "The response from the user."}
+
+    @property
+    def desc(self):
+        return "Use this command to get input from the user."
+
+    @hide_spinner
+    def run(self, message):
+        return {"response": input(f"{message}: ")}

loopgpt/utils/spinner.py

@@ -137,3 +137,17 @@ def inner(*args, **kwargs):
             return func(*args, **kwargs)
 
     return inner
+
+
+def hide_spinner(func):
+    @wraps(func)
+    def inner(*args, **kwargs):
+        if ACTIVE_SPINNER:
+            ACTIVE_SPINNER.hide()
+        try:
+            return func(*args, **kwargs)
+        finally:
+            if ACTIVE_SPINNER:
+                ACTIVE_SPINNER.show()
+
+    return inner