code-puppy 0.0.120__tar.gz → 0.0.122__tar.gz

{code_puppy-0.0.120 → code_puppy-0.0.122}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: code-puppy
-Version: 0.0.120
+Version: 0.0.122
 Summary: Code generation agent
 Author: Michael Pfaffenberger
 License: MIT

{code_puppy-0.0.120 → code_puppy-0.0.122}/code_puppy/main.py

@@ -28,7 +28,7 @@ from code_puppy.message_history_processor import (
     message_history_accumulator,
     prune_interrupted_tool_calls,
 )
-from code_puppy.state_management import is_tui_mode, set_tui_mode
+from code_puppy.state_management import is_tui_mode, set_tui_mode, set_message_history
 from code_puppy.tools.common import console
 from code_puppy.version_checker import default_version_mismatch_behavior
 
@@ -320,6 +320,8 @@ async def interactive_mode(message_renderer, initial_command: str = None) -> Non
                         response = await agent.run(
                             initial_command, usage_limits=get_custom_usage_limits()
                         )
+                    finally:
+                        set_message_history(prune_interrupted_tool_calls(get_message_history()))
 
             agent_response = response.output
 
@@ -328,7 +330,7 @@ async def interactive_mode(message_renderer, initial_command: str = None) -> Non
             )
             new_msgs = response.all_messages()
             message_history_accumulator(new_msgs)
-
+            set_message_history(prune_interrupted_tool_calls(get_message_history()))
             emit_system_message("\n" + "=" * 50)
             emit_info("[bold green]🐶 Continuing in Interactive Mode[/bold green]")
             emit_system_message(
@@ -463,6 +465,8 @@ async def interactive_mode(message_renderer, initial_command: str = None) -> Non
                                 message_history=get_message_history(),
                                 usage_limits=get_custom_usage_limits(),
                             )
+                        finally:
+                            set_message_history(prune_interrupted_tool_calls(get_message_history()))
 
                     # Create the task
                     agent_task = asyncio.create_task(run_agent_task())

{code_puppy-0.0.120 → code_puppy-0.0.122}/code_puppy/tui/app.py

@@ -27,7 +27,7 @@ from code_puppy.message_history_processor import (
 
 # Import our message queue system
 from code_puppy.messaging import TUIRenderer, get_global_queue
-from code_puppy.state_management import clear_message_history, get_message_history
+from code_puppy.state_management import clear_message_history, get_message_history, set_message_history
 from code_puppy.tui.components import (
     ChatView,
     CustomTextArea,
@@ -498,6 +498,8 @@ class CodePuppyTUI(App):
                         else:
                             # Handle regular exceptions
                             self.add_error_message(f"MCP/Agent error: {str(eg)}")
+                    finally:
+                        set_message_history(prune_interrupted_tool_calls(get_message_history()))
                 except Exception as agent_error:
                     # Handle any other errors in agent processing
                     self.add_error_message(

{code_puppy-0.0.120 → code_puppy-0.0.122}/code_puppy/tui/components/chat_view.py

@@ -30,7 +30,7 @@ class ChatView(VerticalScroll):
     }
 
     .user-message {
-        background: #1e3a8a;
+        background: transparent;
         color: #ffffff;
         margin: 0 0 1 0;
         margin-top: 0;
@@ -41,7 +41,7 @@ class ChatView(VerticalScroll):
     }
 
     .agent-message {
-        background: #374151;
+        background: transparent;
         color: #f3f4f6;
         margin: 0 0 1 0;
         margin-top: 0;
@@ -52,7 +52,7 @@ class ChatView(VerticalScroll):
     }
 
     .system-message {
-        background: #1f2937;
+        background: transparent;
         color: #d1d5db;
         margin: 0 0 1 0;
         margin-top: 0;
@@ -64,7 +64,7 @@ class ChatView(VerticalScroll):
     }
 
     .error-message {
-        background: #7f1d1d;
+        background: transparent;
         color: #fef2f2;
         margin: 0 0 1 0;
         margin-top: 0;
@@ -75,7 +75,7 @@ class ChatView(VerticalScroll):
     }
 
     .agent_reasoning-message {
-        background: #1f2937;
+        background: transparent;
         color: #f3e8ff;
         margin: 0 0 1 0;
         margin-top: 0;
@@ -87,7 +87,7 @@ class ChatView(VerticalScroll):
     }
 
     .planned_next_steps-message {
-        background: #1f2937;
+        background: transparent;
         color: #f3e8ff;
         margin: 0 0 1 0;
         margin-top: 0;
@@ -99,7 +99,7 @@ class ChatView(VerticalScroll):
     }
 
     .agent_response-message {
-        background: #1f2937;
+        background: transparent;
         color: #f3e8ff;
         margin: 0 0 1 0;
         margin-top: 0;
@@ -110,7 +110,7 @@ class ChatView(VerticalScroll):
     }
 
     .info-message {
-        background: #065f46;
+        background: transparent;
         color: #d1fae5;
         margin: 0 0 1 0;
         margin-top: 0;
@@ -121,7 +121,7 @@ class ChatView(VerticalScroll):
     }
 
     .success-message {
-        background: #064e3b;
+        background: #0d9488;
         color: #d1fae5;
         margin: 0 0 1 0;
         margin-top: 0;
@@ -132,7 +132,7 @@ class ChatView(VerticalScroll):
     }
 
     .warning-message {
-        background: #92400e;
+        background: #d97706;
         color: #fef3c7;
         margin: 0 0 1 0;
         margin-top: 0;
@@ -143,7 +143,7 @@ class ChatView(VerticalScroll):
     }
 
     .tool_output-message {
-        background: #1e40af;
+        background: #5b21b6;
         color: #dbeafe;
         margin: 0 0 1 0;
         margin-top: 0;
@@ -154,7 +154,7 @@ class ChatView(VerticalScroll):
     }
 
     .command_output-message {
-        background: #7c2d12;
+        background: #9a3412;
         color: #fed7aa;
         margin: 0 0 1 0;
         margin-top: 0;

{code_puppy-0.0.120 → code_puppy-0.0.122}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "code-puppy"
-version = "0.0.120"
+version = "0.0.122"
 description = "Code generation agent"
 readme = "README.md"
 requires-python = ">=3.10"