code-puppy 0.0.121__py3-none-any.whl → 0.0.122__py3-none-any.whl

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/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.dist-info → code_puppy-0.0.122.dist-info}/METADATA

@@ -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.dist-info → code_puppy-0.0.122.dist-info}/RECORD

@@ -5,7 +5,7 @@ code_puppy/agent_prompts.py,sha256=pBDC5zsWozEiFHmoOS6CnxortMvN8VZqnUEHM8sx8Ug,6
 code_puppy/callbacks.py,sha256=AOm39_w7vvMlPXzALoRIYRlhvjZjIiAAouPbbWDdj78,4420
 code_puppy/config.py,sha256=FPyhds_nJ_olG7knP5EhHH3qG9qg14WYIOveosCcVEc,13345
 code_puppy/http_utils.py,sha256=zN0_F8cq4agamL3GRRLLhgEjBgjKtWrgGe8thORC7YE,3434
-code_puppy/main.py,sha256=T1nBn7Y2gTFEonPQ6akpA3HtDesgkMcNtIVxnWAxdHc,26809
+code_puppy/main.py,sha256=awLuvGCeX8JuMFJjLOYhl95JXQMZ3TC2HIEk_vpfJIA,27174
 code_puppy/message_history_processor.py,sha256=wEKZ2AsuMPt2biqTd-bMfG9tIoFoIYzvEt_65qTl3gU,14461
 code_puppy/model_factory.py,sha256=cqYpDAtUcFU4iB0PFSA_nLbuXv4cWFB59BE5-1nZqXI,10202
 code_puppy/models.json,sha256=jr0-LW87aJS79GosVwoZdHeeq5eflPzgdPoMbcqpVA8,2728
@@ -41,7 +41,7 @@ code_puppy/tools/file_operations.py,sha256=7nSueKqlJcnlnBg7zXkX7EWiOJX8i3NNWGE7Q
 code_puppy/tools/token_check.py,sha256=cNrGOOKahXsnWsvh5xnMkL1NS9FjYur9QIRZGQFW-pE,1189
 code_puppy/tools/tools_content.py,sha256=pi9ig2qahZFkUj7gBBN2TX2QldvwnqmTHrRKP8my_2k,2209
 code_puppy/tui/__init__.py,sha256=XesAxIn32zLPOmvpR2wIDxDAnnJr81a5pBJB4cZp1Xs,321
-code_puppy/tui/app.py,sha256=lBjnNLAcpJwRzYs8w8jqj5y1THyctCLtBHYGRhzKuRM,43057
+code_puppy/tui/app.py,sha256=zKToIrL3SyB9-EeYpijxAPaLsB9GgJisj8DRjT5v1Zo,43204
 code_puppy/tui/messages.py,sha256=zQoToWI0eWdT36NEsY6RdCFzcDfAmfvoPlHv8jiCbgo,720
 code_puppy/tui/components/__init__.py,sha256=uj5pnk3s6SEN3SbFI0ZnzaA2KK1NNg8TfUj6U-Z732U,455
 code_puppy/tui/components/chat_view.py,sha256=u6yFqz7sz5RYxVpobnAHXDfDeIDRwDnXXTBwBl5Wn7M,17995
@@ -78,9 +78,9 @@ code_puppy/tui/tests/test_sidebar_history_navigation.py,sha256=JGiyua8A2B8dLfwiE
 code_puppy/tui/tests/test_status_bar.py,sha256=nYT_FZGdmqnnbn6o0ZuOkLtNUtJzLSmtX8P72liQ5Vo,1797
 code_puppy/tui/tests/test_timestamped_history.py,sha256=nVXt9hExZZ_8MFP-AZj4L4bB_1Eo_mc-ZhVICzTuw3I,1799
 code_puppy/tui/tests/test_tools.py,sha256=kgzzAkK4r0DPzQwHHD4cePpVNgrHor6cFr05Pg6DBWg,2687
-code_puppy-0.0.121.data/data/code_puppy/models.json,sha256=jr0-LW87aJS79GosVwoZdHeeq5eflPzgdPoMbcqpVA8,2728
-code_puppy-0.0.121.dist-info/METADATA,sha256=nx60hBpbjqQo6faKi_ypVf08JtE2MzaQqMhltoO5kZ0,6537
-code_puppy-0.0.121.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-code_puppy-0.0.121.dist-info/entry_points.txt,sha256=d8YkBvIUxF-dHNJAj-x4fPEqizbY5d_TwvYpc01U5kw,58
-code_puppy-0.0.121.dist-info/licenses/LICENSE,sha256=31u8x0SPgdOq3izJX41kgFazWsM43zPEF9eskzqbJMY,1075
-code_puppy-0.0.121.dist-info/RECORD,,
+code_puppy-0.0.122.data/data/code_puppy/models.json,sha256=jr0-LW87aJS79GosVwoZdHeeq5eflPzgdPoMbcqpVA8,2728
+code_puppy-0.0.122.dist-info/METADATA,sha256=lDnpy3zaF0ePq78sE2rD0k7lVo5k2ufFtfCJiC3KqH4,6537
+code_puppy-0.0.122.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+code_puppy-0.0.122.dist-info/entry_points.txt,sha256=d8YkBvIUxF-dHNJAj-x4fPEqizbY5d_TwvYpc01U5kw,58
+code_puppy-0.0.122.dist-info/licenses/LICENSE,sha256=31u8x0SPgdOq3izJX41kgFazWsM43zPEF9eskzqbJMY,1075
+code_puppy-0.0.122.dist-info/RECORD,,