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

@@ -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.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,10 +41,10 @@ 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=L79D_N90UH-SITFxaySnlRK7ciXQ7w1LOBLMVUDQbNA,17963
+code_puppy/tui/components/chat_view.py,sha256=u6yFqz7sz5RYxVpobnAHXDfDeIDRwDnXXTBwBl5Wn7M,17995
 code_puppy/tui/components/command_history_modal.py,sha256=pUPEQvoCWa2iUnuMgNwO22y8eUbyw0HpcPH3wAosHvU,7097
 code_puppy/tui/components/copy_button.py,sha256=E4-OJYk5YNzDf-E81NyiVGKsTRPrUX-RnQ8qFuVnabw,4375
 code_puppy/tui/components/custom_widgets.py,sha256=pnjkB3ZNa5lwSrAXUFlhN9AHNh4uMTpSap8AdbpecKw,1986
@@ -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.120.data/data/code_puppy/models.json,sha256=jr0-LW87aJS79GosVwoZdHeeq5eflPzgdPoMbcqpVA8,2728
-code_puppy-0.0.120.dist-info/METADATA,sha256=g1_tvSCvW26oEstJC0p_fc9dGjlHy6QA4MFCWIn3B-g,6537
-code_puppy-0.0.120.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-code_puppy-0.0.120.dist-info/entry_points.txt,sha256=d8YkBvIUxF-dHNJAj-x4fPEqizbY5d_TwvYpc01U5kw,58
-code_puppy-0.0.120.dist-info/licenses/LICENSE,sha256=31u8x0SPgdOq3izJX41kgFazWsM43zPEF9eskzqbJMY,1075
-code_puppy-0.0.120.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,,