feat:add status emojis to system messages

ErikBjare · Jan 31, 2025 · ffa7f04 · ffa7f04
1 parent e02e119
commit ffa7f04
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/gptme/message.py b/gptme/message.py
@@ -213,8 +213,8 @@ def format_msgs(
             userprefix = f"[bold {color}]{userprefix}[/bold {color}]"
         if msg.role == "system":
             first_line = msg.content.split("\n", 1)[0].lower()
-            isSuccess = first_line.startswith(("saved", "executed", "ran", "appended", "patch successfully"))
-            isError = first_line.startswith(("error", "failed","missing arguments", "unknown command", "no patch"))
+            isSuccess = first_line.startswith(("saved", "appended", "patch successfully"))
+            isError = first_line.startswith(("error", "failed"))
             if isSuccess:
                 userprefix = f"{userprefix} ✅"
             elif isError:

diff --git a/gptme/tools/patch.py b/gptme/tools/patch.py
@@ -257,7 +257,7 @@ def execute_patch(
         code = kwargs.get("patch", code)
 
     if not code:
-        yield Message("system", "No patch provided by the assistant")
+        yield Message("system", "Error: No patch provided by the assistant")
         return
 
     yield from execute_with_confirmation(

diff --git a/gptme/tools/tmux.py b/gptme/tools/tmux.py
@@ -223,7 +223,7 @@ def execute_tmux(
             continue
 
         if len(parts) < 2:
-            yield Message("system", f"Missing arguments for command: {command}")
+            yield Message("system", f"Error: Missing arguments for command: {command}")
             continue
 
         _args = parts[1]
@@ -237,7 +237,7 @@ def execute_tmux(
         elif command == "kill_session":
             yield kill_session(_args)
         else:
-            yield Message("system", f"Unknown command: {command}")
+            yield Message("system", f"Error: Unknown command: {command}")
 
 
 instructions = """