code-puppy 0.0.173__py3-none-any.whl → 0.0.174__py3-none-any.whl

code_puppy/tui/app.py

@@ -12,27 +12,19 @@ from textual.events import Resize
 from textual.reactive import reactive
 from textual.widgets import Footer, ListView
 
-from code_puppy.agent import get_code_generation_agent, get_custom_usage_limits
-from code_puppy.agents.runtime_manager import get_runtime_agent_manager
 from code_puppy.command_line.command_handler import handle_command
 from code_puppy.config import (
-    get_model_name,
+    get_global_model_name,
     get_puppy_name,
     initialize_command_history_file,
     save_command_to_history,
 )
-from code_puppy.message_history_processor import (
-    message_history_accumulator,
-    prune_interrupted_tool_calls,
-)
+# message_history_accumulator and prune_interrupted_tool_calls have been moved to BaseAgent class
+from code_puppy.agents.agent_manager import get_current_agent
 
 # 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,
-    set_message_history,
-)
+
 from code_puppy.tui.components import (
     ChatView,
     CustomTextArea,
@@ -41,7 +33,6 @@ from code_puppy.tui.components import (
     StatusBar,
 )
 
-from .. import state_management
 
 # Import shared message classes
 from .messages import CommandSelected, HistoryEntrySelected
@@ -112,9 +103,9 @@ class CodePuppyTUI(App):
     def _on_agent_reload(self, agent_id: str, agent_name: str) -> None:
         """Callback for when agent is reloaded/changed."""
         # Get the updated agent configuration
-        from code_puppy.agents.agent_manager import get_current_agent_config
+        from code_puppy.agents.agent_manager import get_current_agent
 
-        current_agent_config = get_current_agent_config()
+        current_agent_config = get_current_agent()
         new_agent_display = (
             current_agent_config.display_name if current_agent_config else "code-puppy"
         )
@@ -127,7 +118,6 @@ class CodePuppyTUI(App):
 
     def __init__(self, initial_command: str = None, **kwargs):
         super().__init__(**kwargs)
-        self.agent_manager = None
         self._current_worker = None
         self.initial_command = initial_command
 
@@ -159,13 +149,13 @@ class CodePuppyTUI(App):
         register_callback("agent_reload", self._on_agent_reload)
 
         # Load configuration
-        self.current_model = get_model_name()
+        self.current_model = get_global_model_name()
         self.puppy_name = get_puppy_name()
 
         # Get current agent information
-        from code_puppy.agents.agent_manager import get_current_agent_config
+        from code_puppy.agents.agent_manager import get_current_agent
 
-        current_agent_config = get_current_agent_config()
+        current_agent_config = get_current_agent()
         self.current_agent = (
             current_agent_config.display_name if current_agent_config else "code-puppy"
         )
@@ -174,8 +164,6 @@ class CodePuppyTUI(App):
         self._update_title()
 
         # Use runtime manager to ensure we always have the current agent
-        self.agent_manager = get_runtime_agent_manager()
-
         # Update status bar
         status_bar = self.query_one(StatusBar)
         status_bar.current_model = self.current_model
@@ -188,7 +176,7 @@ class CodePuppyTUI(App):
         )
 
         # Get current agent and display info
-        get_code_generation_agent()
+        agent = get_current_agent()
         self.add_system_message(
             f"🐕 Loaded agent '{self.puppy_name}' with model '{self.current_model}'"
         )
@@ -430,11 +418,6 @@ class CodePuppyTUI(App):
                 else:
                     # Only cancel the agent task if NO processes were killed
                     self._current_worker.cancel()
-                    state_management.set_message_history(
-                        prune_interrupted_tool_calls(
-                            state_management.get_message_history()
-                        )
-                    )
                     self.add_system_message("⚠️  Processing cancelled by user")
                     # Stop spinner and clear state only when agent is actually cancelled
                     self._current_worker = None
@@ -491,55 +474,44 @@ class CodePuppyTUI(App):
                 return
 
             # Process with agent
-            if self.agent_manager:
-                try:
-                    self.update_agent_progress("Processing", 25)
-
-                    # Use agent_manager's run_with_mcp to handle MCP servers properly
-                    try:
-                        self.update_agent_progress("Processing", 50)
-                        result = await self.agent_manager.run_with_mcp(
-                            message,
-                            message_history=get_message_history(),
-                            usage_limits=get_custom_usage_limits(),
-                        )
-
-                        if not result or not hasattr(result, "output"):
-                            self.add_error_message("Invalid response format from agent")
-                            return
-
-                        self.update_agent_progress("Processing", 75)
-                        agent_response = result.output
-                        self.add_agent_message(agent_response)
-
-                        # Update message history
-                        new_msgs = result.new_messages()
-                        message_history_accumulator(new_msgs)
-
-                        # Refresh history display to show new interaction
-                        self.refresh_history_display()
+            try:
+                self.update_agent_progress("Processing", 25)
 
-                    except Exception as eg:
-                        # Handle TaskGroup and other exceptions
-                        # BaseExceptionGroup is only available in Python 3.11+
-                        if hasattr(eg, "exceptions"):
-                            # Handle TaskGroup exceptions specifically (Python 3.11+)
-                            for e in eg.exceptions:
-                                self.add_error_message(f"MCP/Agent error: {str(e)}")
-                        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(
-                        f"Agent processing failed: {str(agent_error)}"
+                # Use agent_manager's run_with_mcp to handle MCP servers properly
+                try:
+                    agent = get_current_agent()
+                    self.update_agent_progress("Processing", 50)
+                    result = await agent.run_with_mcp(
+                        message,
                     )
-            else:
-                self.add_error_message("Agent manager not initialized")
+
+                    if not result or not hasattr(result, "output"):
+                        self.add_error_message("Invalid response format from agent")
+                        return
+
+                    self.update_agent_progress("Processing", 75)
+                    agent_response = result.output
+                    self.add_agent_message(agent_response)
+                    # Refresh history display to show new interaction
+                    self.refresh_history_display()
+
+                except Exception as eg:
+                    # Handle TaskGroup and other exceptions
+                    # BaseExceptionGroup is only available in Python 3.11+
+                    if hasattr(eg, "exceptions"):
+                        # Handle TaskGroup exceptions specifically (Python 3.11+)
+                        for e in eg.exceptions:
+                            self.add_error_message(f"MCP/Agent error: {str(e)}")
+                    else:
+                        # Handle regular exceptions
+                        self.add_error_message(f"MCP/Agent error: {str(eg)}")
+                finally:
+                    pass
+            except Exception as agent_error:
+                # Handle any other errors in agent processing
+                self.add_error_message(
+                    f"Agent processing failed: {str(agent_error)}"
+                )
 
         except Exception as e:
             self.add_error_message(f"Error processing message: {str(e)}")
@@ -553,7 +525,6 @@ class CodePuppyTUI(App):
         """Clear the chat history."""
         chat_view = self.query_one("#chat-view", ChatView)
         chat_view.clear_messages()
-        clear_message_history()
         self.add_system_message("Chat history cleared")
 
     def action_show_help(self) -> None:
@@ -652,13 +623,13 @@ class CodePuppyTUI(App):
         def handle_settings_result(result):
             if result and result.get("success"):
                 # Update reactive variables
-                from code_puppy.config import get_model_name, get_puppy_name
+                from code_puppy.config import get_global_model_name, get_puppy_name
 
                 self.puppy_name = get_puppy_name()
 
                 # Handle model change if needed
                 if result.get("model_changed"):
-                    new_model = get_model_name()
+                    new_model = get_global_model_name()
                     self.current_model = new_model
                     # Reinitialize agent with new model
                     self.agent_manager.reload_agent()

code_puppy/tui/screens/settings.py

@@ -125,7 +125,7 @@ class SettingsScreen(ModalScreen):
     def on_mount(self) -> None:
         """Load current settings when the screen mounts."""
         from code_puppy.config import (
-            get_model_name,
+            get_global_model_name,
             get_owner_name,
             get_protected_token_count,
             get_puppy_name,
@@ -155,7 +155,7 @@ class SettingsScreen(ModalScreen):
         self.load_model_options(model_select)
 
         # Set current model selection
-        current_model = get_model_name()
+        current_model = get_global_model_name()
         model_select.value = current_model
 
         # YOLO mode is always enabled in TUI mode

{code_puppy-0.0.173.dist-info → code_puppy-0.0.174.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: code-puppy
-Version: 0.0.173
+Version: 0.0.174
 Summary: Code generation agent
 Project-URL: repository, https://github.com/mpfaffenberger/code_puppy
 Project-URL: HomePage, https://github.com/mpfaffenberger/code_puppy
@@ -26,7 +26,7 @@ Requires-Dist: openai>=1.99.1
 Requires-Dist: pathspec>=0.11.0
 Requires-Dist: playwright>=1.40.0
 Requires-Dist: prompt-toolkit>=3.0.38
-Requires-Dist: pydantic-ai>=1.0.10
+Requires-Dist: pydantic-ai<=0.8
 Requires-Dist: pydantic>=2.4.0
 Requires-Dist: pyjwt>=2.8.0
 Requires-Dist: pytest-cov>=6.1.1

{code_puppy-0.0.173.dist-info → code_puppy-0.0.174.dist-info}/RECORD

@@ -1,37 +1,32 @@
 code_puppy/__init__.py,sha256=ehbM1-wMjNmOXk_DBhhJECFyBv2dRHwwo7ucjHeM68E,107
 code_puppy/__main__.py,sha256=pDVssJOWP8A83iFkxMLY9YteHYat0EyWDQqMkKHpWp4,203
-code_puppy/agent.py,sha256=rBe7OS_ekZvMuteLuIp9lor9zIP7KHBsxZ22pWM9_fg,8526
+code_puppy/agent.py,sha256=CGdK-vIwRQh27c552Glc39eqhnXJh2ZeE9D2BaYkhms,8505
 code_puppy/callbacks.py,sha256=6wYB6K_fGSCkKKEFaYOYkJT45WaV5W_NhUIzcvVH_nU,5060
-code_puppy/config.py,sha256=A3l5gSJGz9kGXJb2pZVyoksDGclIhMdKmAv-EdDSleY,20625
+code_puppy/config.py,sha256=ruPGe9f3kFI9F563NQOKuNQY12kw5eXPzbm2utJ0DO8,20639
 code_puppy/http_utils.py,sha256=jnElpVNNIK-_gcqwsBYRImDGU-tYNfsLR7Gtme5CYzs,8423
-code_puppy/main.py,sha256=VlH8fXPapXblSagOjrezeSTtFij_-X3dzq1cfHPXwDI,22252
-code_puppy/message_history_processor.py,sha256=SnVpUCMkbtM4RX5b6ioeG-fh--ri0bh8_yHp7ZQIAPE,16005
+code_puppy/main.py,sha256=hVPFNN8QOuct2FFH6OMGQ3E04cmEnmO88ffTbbpfuWY,20516
 code_puppy/model_factory.py,sha256=z9vQbcGllgMwU0On8rPvzYxkygW2Uyd3NJmRzbKv-is,13759
 code_puppy/models.json,sha256=iXmLZGflnQcu2DRh4WUlgAhoXdvoxUc7KBhB8YxawXM,3088
 code_puppy/reopenable_async_client.py,sha256=4UJRaMp5np8cbef9F0zKQ7TPKOfyf5U-Kv-0zYUWDho,8274
 code_puppy/round_robin_model.py,sha256=SEN3VSwTgC5wHjx2sZsHQLPWOycf4jGwzB-EydgqkdY,5643
-code_puppy/state_management.py,sha256=OyBSfdicKN7ObZEGs6Kbyy1JC8R8n4FZBOLIBvQZTBc,1916
 code_puppy/status_display.py,sha256=F6eEAkGePDp4StM2BWj-uLLQTDGtJrf0IufzCeP1rRg,8336
-code_puppy/summarization_agent.py,sha256=kos4_YK-l_YjYRq4Fs4X5YoTUbmAcDhhPqefL-rdenI,3197
+code_puppy/summarization_agent.py,sha256=LnObgtLmM6N4z2553XXQlXAOf8R1BPSNmFSfXkjpivg,3211
 code_puppy/tui_state.py,sha256=TT76XBVapKj6fKjFzz6oxCONeN_BZwcMILxxZcxu6-Y,1171
 code_puppy/version_checker.py,sha256=bjLDmgGPrl7XnYwX1u13O8uFlsfikV90PK6nbA9Z9QU,1150
-code_puppy/agents/__init__.py,sha256=SwtHGNG1GIgDBv7y3EGIXOXEWrG_Ou7fEknNgFbrHv8,594
+code_puppy/agents/__init__.py,sha256=37co1Xux3E10ynvbCBkyeTZAZpjEb-ci2uUy6nzM1hA,518
 code_puppy/agents/agent_code_puppy.py,sha256=sbuQxLzlkMbPOyLbILbXOAHecRsxbFdQt13HJ_GEqQo,7972
 code_puppy/agents/agent_creator_agent.py,sha256=eNOlJssQdyoQm1F7d-TZWcMXpkYmZ-w9WN-NDFXCgtw,23042
-code_puppy/agents/agent_manager.py,sha256=nXvro6fpX8KA-NedRoVJuhJW966trrePOrH4eAnqq40,17034
-code_puppy/agents/agent_orchestrator.json,sha256=Iqnc0p6ICoAlUTMkEsi1XXMXJi4pdxVnWZUMaih6s5o,1267
+code_puppy/agents/agent_manager.py,sha256=BmIxguNN3qq1S7UebxqrcdB0Q4ZuwWRVRsnYUDMgfYg,11050
 code_puppy/agents/agent_qa_kitten.py,sha256=5PeFFSwCFlTUvP6h5bGntx0xv5NmRwBiw0HnMqY8nLI,9107
-code_puppy/agents/base_agent.py,sha256=XUsziIf0ZGCe6ZLjt-kJqbeYTy9gTQE3_BPq7THQ3X8,19538
+code_puppy/agents/base_agent.py,sha256=LRBj-ap7-pjiXGbAUfNMt45W1KnCBZjfYPKENlmg89U,36701
 code_puppy/agents/json_agent.py,sha256=y6AYE3Fx9LhmemcPzt46d7359MNnkGIjU83YBGNer2g,4533
-code_puppy/agents/runtime_manager.py,sha256=fUOBpmETo3wTyLc5wWBfGKSX1HFRQWSpuwxYAOyA-_8,10059
 code_puppy/command_line/__init__.py,sha256=y7WeRemfYppk8KVbCGeAIiTuiOszIURCDjOMZv_YRmU,45
-code_puppy/command_line/command_handler.py,sha256=45jUScWIikVi2OgCL7LrcpWoDcrmfWv0do-y_mS4i8U,27248
+code_puppy/command_line/command_handler.py,sha256=z4p62jwdJXUJVVzTHStTOFyzv-PwjjNhu8bziIo8gns,27191
 code_puppy/command_line/file_path_completion.py,sha256=gw8NpIxa6GOpczUJRyh7VNZwoXKKn-yvCqit7h2y6Gg,2931
 code_puppy/command_line/load_context_completion.py,sha256=6eZxV6Bs-EFwZjN93V8ZDZUC-6RaWxvtZk-04Wtikyw,2240
-code_puppy/command_line/meta_command_handler.py,sha256=80aK5JQOaqjt149qBmSsM02uy2Cikkee8zaQnu5u2KQ,5712
-code_puppy/command_line/model_picker_completion.py,sha256=adxp3NZaDV67YqaGv0SG7WVvOTXN0UwkkLqxTsknAvs,4126
+code_puppy/command_line/model_picker_completion.py,sha256=vYNCZS1QWu6fxF__hTwpc7jwH7h_48wUxrnITawc83E,4140
 code_puppy/command_line/motd.py,sha256=PEdkp3ZnydVfvd7mNJylm8YyFNUKg9jmY6uwkA1em8c,2152
-code_puppy/command_line/prompt_toolkit_completion.py,sha256=PLb1Yt_65iQI2mzPEPxPsSaRww5Ur951Qn-K-TLMCGQ,10006
+code_puppy/command_line/prompt_toolkit_completion.py,sha256=O-wTYIGb9J_ONEBa-bWNFeQLnqC6LIFKsKQ7hjMbssM,9992
 code_puppy/command_line/utils.py,sha256=7eyxDHjPjPB9wGDJQQcXV_zOsGdYsFgI0SGCetVmTqE,1251
 code_puppy/command_line/mcp/__init__.py,sha256=0-OQuwjq_pLiTVJ1_NrirVwdRerghyKs_MTZkwPC7YY,315
 code_puppy/command_line/mcp/add_command.py,sha256=lZ09RpFDIeghX1zhc2YIAqBASs5Ra52x5YAasUKvqJg,6409
@@ -44,11 +39,11 @@ code_puppy/command_line/mcp/logs_command.py,sha256=x_QsVGPpI5XY7RCZtiNFegc6R3csi
 code_puppy/command_line/mcp/remove_command.py,sha256=MrWmXQ9jZTq1wrohFDO3ls0a1faTHCqRZocN-ynTzh8,2753
 code_puppy/command_line/mcp/restart_command.py,sha256=-arZhayEVjTg3WErQKiBoFxbpbQcFlPXTzUI5o8AQLg,3254
 code_puppy/command_line/mcp/search_command.py,sha256=k4CGiNJ1MKJ1_dPHd3wlwXo899nzizSUr7PsR0V9y4Y,4113
-code_puppy/command_line/mcp/start_all_command.py,sha256=9f8BPTijs7BVBQKKNXjtH3laQVuqhNyfPSgISvBUcfM,4295
-code_puppy/command_line/mcp/start_command.py,sha256=r00onOpfSSL-gtJLDOvmD-8G4RZhhmd-nv3TyIWGMUI,3350
+code_puppy/command_line/mcp/start_all_command.py,sha256=cdFi-eEAp4SKTiUvSkCAHaeq3PiujW35a3ZT7UPNlUc,4197
+code_puppy/command_line/mcp/start_command.py,sha256=dFB64MJMHAh8v_Jq07Ux4K74P9CP239l_0BHk7wH_2A,3108
 code_puppy/command_line/mcp/status_command.py,sha256=tQJrEDlN4DJ3n9jnKW-4kD0vL3TyUQbovb4V3nz-RsA,6640
-code_puppy/command_line/mcp/stop_all_command.py,sha256=QioqCPFjIbbdT_P5tyVakB2idUD1rnlhwYXLpa-G4TI,3716
-code_puppy/command_line/mcp/stop_command.py,sha256=zuZO8pIP2Ix11LATf-cwwX3fPFIGAbRDaZ6qnfPTqLY,2594
+code_puppy/command_line/mcp/stop_all_command.py,sha256=BcsvkGcSGCexCnWfpirBNxJNHP0HSGrjQP0LslHbano,3618
+code_puppy/command_line/mcp/stop_command.py,sha256=6ruOlABIgGmuFmoakAsaqStaCUctBsA_KbvYNKmlt8U,2456
 code_puppy/command_line/mcp/test_command.py,sha256=Pjod77DYRcA9WDZcxaCe6AUgpEXKJr42QRkxMhSjXhM,3689
 code_puppy/command_line/mcp/utils.py,sha256=0Wt4ttYgSlVvtusYmBLKXSkjAjcsDiUxcZQAoFLUNnE,3625
 code_puppy/command_line/mcp/wizard_utils.py,sha256=mWuE5epVgCxgGoj-VGwPO_XWVkkXjegXoIJpC9LC2TM,11075
@@ -79,7 +74,7 @@ code_puppy/messaging/spinner/spinner_base.py,sha256=474qMrTYpNfWcprFzmhaOJEOC-2r
 code_puppy/messaging/spinner/textual_spinner.py,sha256=Omx9A-FSPkxYDMYgBXgYMBQnK-DMlyqLOgkFVG8cmo4,3465
 code_puppy/plugins/__init__.py,sha256=fksDqMUiXPJ5WNuMsYsVR8ulueQRCXPlvECEyicHPtQ,1312
 code_puppy/tools/__init__.py,sha256=BVTZ85jLHgDANwOnUSOz3UDlp8VQDq4DoGF23BRlyWw,6032
-code_puppy/tools/agent_tools.py,sha256=bHMrFIbYRhuubR41G_XdLsk3cUKWfIPl2O4bVzo2pE0,5591
+code_puppy/tools/agent_tools.py,sha256=h3i4c6KOxdjLr-fFjUDJdL0OR_XHo_AWx7uwfvavQJ8,5640
 code_puppy/tools/browser_control.py,sha256=6E_Kq63cErBk-iM9-03Cq8sTWWLh-Tk2kpdrB_rMmbg,8603
 code_puppy/tools/browser_interactions.py,sha256=m-bVsAUbdrS-P1GbS0ChO2tMMg4tMIsDX34azn_Olzs,17070
 code_puppy/tools/browser_locators.py,sha256=ZshjhYuV600vJqXUC23oNX89AZLMvY8rlvj-9oy7GMw,19574
@@ -102,9 +97,9 @@ code_puppy/tools/browser/browser_screenshot.py,sha256=zbOMmN9_9aVDWJkC54-3zv4OEG
 code_puppy/tools/browser/browser_scripts.py,sha256=BLSx1Q2F_mOOoGCoyXat3HvazTb1XaFYPXAF8CYVeX8,15071
 code_puppy/tools/browser/browser_workflows.py,sha256=HZ0lPmEyAobPIWR-SK1E0ngW1OfULLqw8XILVT4N8Fg,5979
 code_puppy/tools/browser/camoufox_manager.py,sha256=E1CJwQtzPFoDQiXSHHk28niWuJhk3soZ-ItxXCXWO0M,6938
-code_puppy/tools/browser/vqa_agent.py,sha256=u3EhZQHNBiWUn-XLDK9MnVyd2ChshPnVZbB9iEkEqH4,2118
+code_puppy/tools/browser/vqa_agent.py,sha256=jUvzsk9okOZ9jskuBuij2n8wRTpTkUU2JDaHmfY8PLc,2016
 code_puppy/tui/__init__.py,sha256=XesAxIn32zLPOmvpR2wIDxDAnnJr81a5pBJB4cZp1Xs,321
-code_puppy/tui/app.py,sha256=WW-O8myM_yl05NIdj_aSp7zN56AwimgdMA9TmEOE54Y,39427
+code_puppy/tui/app.py,sha256=gZvg7ZOUp3JurZyKZ7ehpHEhmyC0NnfmQYWF_yZm36o,38163
 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=NfyNXuN2idPht1rKJB4YhHVXb1AIRNO5q_nLdt8Ocug,19913
@@ -122,11 +117,11 @@ code_puppy/tui/models/enums.py,sha256=1ulsei95Gxy4r1sk-m-Sm5rdmejYCGRI-YtUwJmKFf
 code_puppy/tui/screens/__init__.py,sha256=tJ00d0aYQ9kzOGHRChqy6cCQ6JUKKXBzLUTEbk_eA2Y,286
 code_puppy/tui/screens/help.py,sha256=eJuPaOOCp7ZSUlecearqsuX6caxWv7NQszUh0tZJjBM,3232
 code_puppy/tui/screens/mcp_install_wizard.py,sha256=vObpQwLbXjQsxmSg-WCasoev1usEi0pollKnL0SHu9U,27693
-code_puppy/tui/screens/settings.py,sha256=GMpv-qa08rorAE9mj3AjmqjZFPhmeJ_GWd-DBHG6iAA,10671
+code_puppy/tui/screens/settings.py,sha256=W22sevojC1_HZBoeoJTH3HWkehini3bGi_ic0OKgeLI,10685
 code_puppy/tui/screens/tools.py,sha256=3pr2Xkpa9Js6Yhf1A3_wQVRzFOui-KDB82LwrsdBtyk,1715
-code_puppy-0.0.173.data/data/code_puppy/models.json,sha256=iXmLZGflnQcu2DRh4WUlgAhoXdvoxUc7KBhB8YxawXM,3088
-code_puppy-0.0.173.dist-info/METADATA,sha256=4qImoiksA3Y_7gc-YlHc7cAdJJa6pnP2xhOUQbQ9vCY,20106
-code_puppy-0.0.173.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-code_puppy-0.0.173.dist-info/entry_points.txt,sha256=Tp4eQC99WY3HOKd3sdvb22vZODRq0XkZVNpXOag_KdI,91
-code_puppy-0.0.173.dist-info/licenses/LICENSE,sha256=31u8x0SPgdOq3izJX41kgFazWsM43zPEF9eskzqbJMY,1075
-code_puppy-0.0.173.dist-info/RECORD,,
+code_puppy-0.0.174.data/data/code_puppy/models.json,sha256=iXmLZGflnQcu2DRh4WUlgAhoXdvoxUc7KBhB8YxawXM,3088
+code_puppy-0.0.174.dist-info/METADATA,sha256=cnh7U5TS6UOkwhUSw8Nek8vBxdinFzq-K7FO5FRZiIU,20103
+code_puppy-0.0.174.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+code_puppy-0.0.174.dist-info/entry_points.txt,sha256=Tp4eQC99WY3HOKd3sdvb22vZODRq0XkZVNpXOag_KdI,91
+code_puppy-0.0.174.dist-info/licenses/LICENSE,sha256=31u8x0SPgdOq3izJX41kgFazWsM43zPEF9eskzqbJMY,1075
+code_puppy-0.0.174.dist-info/RECORD,,

code_puppy/agents/agent_orchestrator.json

@@ -1,26 +0,0 @@
-{
-  "id": "agent-orchestrator-id",
-  "name": "agent-orchestrator",
-  "display_name": "Agent Orchestrator 🎭",
-  "description": "Coordinates and manages various specialized agents to accomplish tasks",
-  "system_prompt": [
-    "You are an agent orchestrator that coordinates various specialized agents.",
-    "When given a task, first list the available agents to understand what's at your disposal.",
-    "Then, invoke the most appropriate agent to handle the task. If needed, you can invoke multiple agents.",
-    "",
-    "#### `list_agents()`",
-    "Use this to list all available sub-agents that can be invoked",
-    "",
-    "#### `invoke_agent(agent_name: str, user_prompt: str)`",
-    "Use this to invoke another agent with a specific prompt. This allows agents to delegate tasks to specialized sub-agents.",
-    "Arguments:",
-    "- agent_name (required): Name of the agent to invoke",
-    "- user_prompt (required): The prompt to send to the invoked agent",
-    "Example usage:",
-    "```python",
-    "invoke_agent(agent_name=\"python-tutor\", user_prompt=\"Explain how to use list comprehensions\")",
-    "```"
-  ],
-  "tools": ["list_agents", "invoke_agent", "agent_share_your_reasoning"],
-  "user_prompt": "What would you like me to coordinate for you?"
-}

code_puppy/agents/runtime_manager.py

@@ -1,272 +0,0 @@
-"""
-Runtime agent manager that ensures proper agent instance updates.
-
-This module provides a wrapper around the agent singleton that ensures
-all references to the agent are properly updated when it's reloaded.
-"""
-
-import asyncio
-import signal
-import sys
-import uuid
-from typing import Any, Optional
-
-# ExceptionGroup is available in Python 3.11+
-if sys.version_info >= (3, 11):
-    from builtins import ExceptionGroup
-else:
-    # For Python 3.10 and below, we can define a simple fallback
-    class ExceptionGroup(Exception):
-        def __init__(self, message, exceptions):
-            super().__init__(message)
-            self.exceptions = exceptions
-
-
-import mcp
-from pydantic_ai import Agent
-from pydantic_ai.exceptions import UsageLimitExceeded
-from pydantic_ai.usage import UsageLimits
-
-from code_puppy.messaging.message_queue import emit_info
-
-
-class RuntimeAgentManager:
-    """
-    Manages the runtime agent instance and ensures proper updates.
-
-    This class acts as a proxy that always returns the current agent instance,
-    ensuring that when the agent is reloaded, all code using this manager
-    automatically gets the updated instance.
-    """
-
-    def __init__(self):
-        """Initialize the runtime agent manager."""
-        self._agent: Optional[Agent] = None
-        self._last_model_name: Optional[str] = None
-
-    def get_agent(self, force_reload: bool = False, message_group: str = "") -> Agent:
-        """
-        Get the current agent instance.
-
-        This method always returns the most recent agent instance,
-        automatically handling reloads when the model changes.
-
-        Args:
-            force_reload: If True, force a reload of the agent
-
-        Returns:
-            The current agent instance
-        """
-        from code_puppy.agent import get_code_generation_agent
-
-        # Always get the current singleton - this ensures we have the latest
-        current_agent = get_code_generation_agent(
-            force_reload=force_reload, message_group=message_group
-        )
-        self._agent = current_agent
-
-        return self._agent
-
-    def reload_agent(self) -> Agent:
-        """
-        Force reload the agent.
-
-        This is typically called after MCP servers are started/stopped.
-
-        Returns:
-            The newly loaded agent instance
-        """
-        message_group = uuid.uuid4()
-        emit_info(
-            "[bold cyan]Reloading agent with updated configuration...[/bold cyan]",
-            message_group=message_group,
-        )
-        return self.get_agent(force_reload=True, message_group=message_group)
-
-    async def run_with_mcp(
-        self, prompt: str, usage_limits: Optional[UsageLimits] = None, **kwargs
-    ) -> Any:
-        """
-        Run the agent with MCP servers and full cancellation support.
-
-        This method ensures we're always using the current agent instance
-        and handles Ctrl+C interruption properly by creating a cancellable task.
-
-        Args:
-            prompt: The user prompt to process
-            usage_limits: Optional usage limits for the agent
-            **kwargs: Additional arguments to pass to agent.run (e.g., message_history)
-
-        Returns:
-            The agent's response
-
-        Raises:
-            asyncio.CancelledError: When execution is cancelled by user
-        """
-        agent = self.get_agent()
-        group_id = str(uuid.uuid4())
-
-        # Function to run agent with MCP
-        async def run_agent_task():
-            try:
-                async with agent:
-                    return await agent.run(prompt, usage_limits=usage_limits, **kwargs)
-            except* UsageLimitExceeded as ule:
-                emit_info(f"Usage limit exceeded: {str(ule)}", group_id=group_id)
-                emit_info(
-                    "The agent has reached its usage limit. You can ask it to continue by saying 'please continue' or similar.",
-                    group_id=group_id,
-                )
-            except* mcp.shared.exceptions.McpError as mcp_error:
-                emit_info(f"MCP server error: {str(mcp_error)}", group_id=group_id)
-                emit_info(f"{str(mcp_error)}", group_id=group_id)
-                emit_info(
-                    "Try disabling any malfunctioning MCP servers", group_id=group_id
-                )
-            except* asyncio.exceptions.CancelledError:
-                emit_info("Cancelled")
-            except* InterruptedError as ie:
-                emit_info(f"Interrupted: {str(ie)}")
-            except* Exception as other_error:
-                # Filter out CancelledError and UsageLimitExceeded from the exception group - let it propagate
-                remaining_exceptions = []
-
-                def collect_non_cancelled_exceptions(exc):
-                    if isinstance(exc, ExceptionGroup):
-                        for sub_exc in exc.exceptions:
-                            collect_non_cancelled_exceptions(sub_exc)
-                    elif not isinstance(
-                        exc, (asyncio.CancelledError, UsageLimitExceeded)
-                    ):
-                        remaining_exceptions.append(exc)
-                        emit_info(f"Unexpected error: {str(exc)}", group_id=group_id)
-                        emit_info(f"{str(exc.args)}", group_id=group_id)
-
-                collect_non_cancelled_exceptions(other_error)
-
-                # If there are CancelledError exceptions in the group, re-raise them
-                cancelled_exceptions = []
-
-                def collect_cancelled_exceptions(exc):
-                    if isinstance(exc, ExceptionGroup):
-                        for sub_exc in exc.exceptions:
-                            collect_cancelled_exceptions(sub_exc)
-                    elif isinstance(exc, asyncio.CancelledError):
-                        cancelled_exceptions.append(exc)
-
-                collect_cancelled_exceptions(other_error)
-
-                if cancelled_exceptions:
-                    # Re-raise the first CancelledError to propagate cancellation
-                    raise cancelled_exceptions[0]
-
-        # Create the task FIRST
-        agent_task = asyncio.create_task(run_agent_task())
-
-        # Import shell process killer
-        from code_puppy.tools.command_runner import kill_all_running_shell_processes
-
-        # Ensure the interrupt handler only acts once per task
-        def keyboard_interrupt_handler(sig, frame):
-            """Signal handler for Ctrl+C - replicating exact original logic"""
-
-            # First, nuke any running shell processes triggered by tools
-            try:
-                killed = kill_all_running_shell_processes()
-                if killed:
-                    emit_info(f"Cancelled {killed} running shell process(es).")
-                else:
-                    # Only cancel the agent task if no shell processes were killed
-                    if not agent_task.done():
-                        agent_task.cancel()
-            except Exception as e:
-                emit_info(f"Shell kill error: {e}")
-                # If shell kill failed, still try to cancel the agent task
-                if not agent_task.done():
-                    agent_task.cancel()
-            # Don't call the original handler
-            # This prevents the application from exiting
-
-        try:
-            # Save original handler and set our custom one AFTER task is created
-            original_handler = signal.signal(signal.SIGINT, keyboard_interrupt_handler)
-
-            # Wait for the task to complete or be cancelled
-            result = await agent_task
-            return result
-        except asyncio.CancelledError:
-            # Task was cancelled by our handler
-            raise
-        except KeyboardInterrupt:
-            # Handle direct keyboard interrupt during await
-            if not agent_task.done():
-                agent_task.cancel()
-            try:
-                await agent_task
-            except asyncio.CancelledError:
-                pass
-            raise asyncio.CancelledError()
-        finally:
-            # Restore original signal handler
-            if original_handler:
-                signal.signal(signal.SIGINT, original_handler)
-
-    async def run(
-        self, prompt: str, usage_limits: Optional[UsageLimits] = None, **kwargs
-    ) -> Any:
-        """
-        Run the agent without explicitly managing MCP servers.
-
-        Args:
-            prompt: The user prompt to process
-            usage_limits: Optional usage limits for the agent
-            **kwargs: Additional arguments to pass to agent.run (e.g., message_history)
-
-        Returns:
-            The agent's response
-        """
-        agent = self.get_agent()
-        try:
-            return await agent.run(prompt, usage_limits=usage_limits, **kwargs)
-        except UsageLimitExceeded as ule:
-            group_id = str(uuid.uuid4())
-            emit_info(f"Usage limit exceeded: {str(ule)}", group_id=group_id)
-            emit_info(
-                "The agent has reached its usage limit. You can ask it to continue by saying 'please continue' or similar.",
-                group_id=group_id,
-            )
-            # Return None or some default value to indicate the limit was reached
-            return None
-
-    def __getattr__(self, name: str) -> Any:
-        """
-        Proxy all other attribute access to the current agent.
-
-        This allows the manager to be used as a drop-in replacement
-        for direct agent access.
-
-        Args:
-            name: The attribute name to access
-
-        Returns:
-            The attribute from the current agent
-        """
-        agent = self.get_agent()
-        return getattr(agent, name)
-
-
-# Global singleton instance
-_runtime_manager: Optional[RuntimeAgentManager] = None
-
-
-def get_runtime_agent_manager() -> RuntimeAgentManager:
-    """
-    Get the global runtime agent manager instance.
-
-    Returns:
-        The singleton RuntimeAgentManager instance
-    """
-    global _runtime_manager
-    if _runtime_manager is None:
-        _runtime_manager = RuntimeAgentManager()
-    return _runtime_manager