fix: preserve wrapped input display in colored prompts

s01_agent_loop/code.py

@@ -125,7 +125,7 @@ def agent_loop(messages: list):
     history = []
     while True:
         try:
-            query = input("\033[36ms01 >> \033[0m")
+            query = input("\001\033[36m\002s01 >> \001\033[0m\002")
         except (EOFError, KeyboardInterrupt):
             break
         if query.strip().lower() in ("q", "exit", ""):

s02_tool_use/code.py

@@ -178,7 +178,7 @@ def agent_loop(messages: list):
     history = []
     while True:
         try:
-            query = input("\033[36ms02 >> \033[0m")
+            query = input("\001\033[36m\002s02 >> \001\033[0m\002")
         except (EOFError, KeyboardInterrupt):
             break
         if query.strip().lower() in ("q", "exit", ""):

s03_permission/code.py

@@ -228,7 +228,7 @@ def agent_loop(messages: list):
     history = []
     while True:
         try:
-            query = input("\033[36ms03 >> \033[0m")
+            query = input("\001\033[36m\002s03 >> \001\033[0m\002")
         except (EOFError, KeyboardInterrupt):
             break
         if query.strip().lower() in ("q", "exit", ""):

s04_hooks/code.py

@@ -241,7 +241,7 @@ def agent_loop(messages: list):
     history = []
     while True:
         try:
-            query = input("\033[36ms04 >> \033[0m")
+            query = input("\001\033[36m\002s04 >> \001\033[0m\002")
         except (EOFError, KeyboardInterrupt):
             break
         if query.strip().lower() in ("q", "exit", ""):

s05_todo_write/code.py

@@ -333,7 +333,7 @@ def agent_loop(messages: list):
     history = []
     while True:
         try:
-            query = input("\033[36ms05 >> \033[0m")
+            query = input("\001\033[36m\002s05 >> \001\033[0m\002")
         except (EOFError, KeyboardInterrupt):
             break
         if query.strip().lower() in ("q", "exit", ""):

s06_subagent/code.py

@@ -349,7 +349,7 @@ def agent_loop(messages: list):
     history = []
     while True:
         try:
-            query = input("\033[36ms06 >> \033[0m")
+            query = input("\001\033[36m\002s06 >> \001\033[0m\002")
         except (EOFError, KeyboardInterrupt):
             break
         if query.strip().lower() in ("q", "exit", ""):

s07_skill_loading/code.py

@@ -364,7 +364,7 @@ def agent_loop(messages: list):
     history = []
     while True:
         try:
-            query = input("\033[36ms07 >> \033[0m")
+            query = input("\001\033[36m\002s07 >> \001\033[0m\002")
         except (EOFError, KeyboardInterrupt):
             break
         if query.strip().lower() in ("q", "exit", ""):

s08_context_compact/code.py

@@ -485,7 +485,7 @@ def agent_loop(messages: list, active_request: str):
     history = []
     while True:
         try:
-            query = input("\033[36ms08 >> \033[0m")
+            query = input("\001\033[36m\002s08 >> \001\033[0m\002")
         except (EOFError, KeyboardInterrupt):
             break
         if query.strip().lower() in ("q", "exit", ""):

s09_memory/code.py

@@ -743,7 +743,7 @@ def agent_loop(messages: list):
     history = []
     while True:
         try:
-            query = input("\033[36ms09 >> \033[0m")
+            query = input("\001\033[36m\002s09 >> \001\033[0m\002")
         except (EOFError, KeyboardInterrupt):
             break
         if query.strip().lower() in ("q", "exit", ""):

s10_task_system/code.py

@@ -567,7 +567,7 @@ def agent_loop(messages: list):
     history = []
     while True:
         try:
-            query = input("\033[36ms10 >> \033[0m")
+            query = input("\001\033[36m\002s10 >> \001\033[0m\002")
         except (EOFError, KeyboardInterrupt):
             break
         if query.strip().lower() in ("q", "exit", ""):

s11_background_tasks/code.py

@@ -485,7 +485,7 @@ def agent_loop(messages: list):
     history = []
     while True:
         try:
-            query = input("\033[36ms11 >> \033[0m")
+            query = input("\001\033[36m\002s11 >> \001\033[0m\002")
         except (EOFError, KeyboardInterrupt):
             break
         if query.strip().lower() in ("q", "exit", ""):

s12_cron_scheduler/code.py

@@ -758,7 +758,7 @@ def stop_runtime_threads():
     try:
         while True:
             try:
-                query = input("\033[36ms12 >> \033[0m")
+                query = input("\001\033[36m\002s12 >> \001\033[0m\002")
             except (EOFError, KeyboardInterrupt):
                 break
             if query.strip().lower() in ("q", "exit", ""):