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

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

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

{code_puppy-0.0.121 → 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.121 → 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.121 → code_puppy-0.0.122}/pyproject.toml

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