code-puppy 0.0.173__py3-none-any.whl → 0.0.175__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.175.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: code-puppy
-Version: 0.0.173
+Version: 0.0.175
 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.175.dist-info}/RECORD

@@ -1,37 +1,31 @@
 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/callbacks.py,sha256=6wYB6K_fGSCkKKEFaYOYkJT45WaV5W_NhUIzcvVH_nU,5060
-code_puppy/config.py,sha256=A3l5gSJGz9kGXJb2pZVyoksDGclIhMdKmAv-EdDSleY,20625
-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/config.py,sha256=ruPGe9f3kFI9F563NQOKuNQY12kw5eXPzbm2utJ0DO8,20639
+code_puppy/http_utils.py,sha256=YLd8Y16idbI32JGeBXG8n5rT4o4X_zxk9FgUvK9XFo8,8248
+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=ZT6B7x4hRKdmEzP085_xR8OL8j0YpboWazrs0U7mP1I,11409
 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=Rv1jXxeFsXfwdyGoVWwoZRgovbi_laex6OivLzaGHpQ,36723
 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 +38,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 +73,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 +96,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 +116,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.175.data/data/code_puppy/models.json,sha256=iXmLZGflnQcu2DRh4WUlgAhoXdvoxUc7KBhB8YxawXM,3088
+code_puppy-0.0.175.dist-info/METADATA,sha256=eyUu2t6tGGAB4Urt_lhXVDtSkT0yw8aZ_IJEa2B-BAc,20103
+code_puppy-0.0.175.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+code_puppy-0.0.175.dist-info/entry_points.txt,sha256=Tp4eQC99WY3HOKd3sdvb22vZODRq0XkZVNpXOag_KdI,91
+code_puppy-0.0.175.dist-info/licenses/LICENSE,sha256=31u8x0SPgdOq3izJX41kgFazWsM43zPEF9eskzqbJMY,1075
+code_puppy-0.0.175.dist-info/RECORD,,

code_puppy/agent.py

@@ -1,239 +0,0 @@
-import uuid
-from pathlib import Path
-from pydantic_ai.models.openai import OpenAIModelSettings, OpenAIResponsesModelSettings
-from typing import Dict, Optional
-
-from pydantic_ai import Agent
-from pydantic_ai.settings import ModelSettings
-from pydantic_ai.usage import UsageLimits
-
-from code_puppy.message_history_processor import message_history_accumulator
-from code_puppy.messaging.message_queue import (
-    emit_error,
-    emit_info,
-    emit_system_message,
-)
-from code_puppy.model_factory import ModelFactory
-
-# Tool registration is imported on demand
-from code_puppy.tools.common import console
-
-
-def load_puppy_rules():
-    global PUPPY_RULES
-
-    # Check for all 4 combinations of the rules file
-    possible_paths = ["AGENTS.md", "AGENT.md", "agents.md", "agent.md"]
-
-    for path_str in possible_paths:
-        puppy_rules_path = Path(path_str)
-        if puppy_rules_path.exists():
-            with open(puppy_rules_path, "r") as f:
-                puppy_rules = f.read()
-                return puppy_rules
-
-    # If none of the files exist, return None
-    return None
-
-
-# Load at import
-PUPPY_RULES = load_puppy_rules()
-_LAST_MODEL_NAME = None
-_code_generation_agent = None
-
-
-def _load_mcp_servers(extra_headers: Optional[Dict[str, str]] = None):
-    """Load MCP servers using the new manager while maintaining backward compatibility."""
-    from code_puppy.config import get_value, load_mcp_server_configs
-    from code_puppy.mcp_ import ServerConfig, get_mcp_manager
-
-    # Check if MCP servers are disabled
-    mcp_disabled = get_value("disable_mcp_servers")
-    if mcp_disabled and str(mcp_disabled).lower() in ("1", "true", "yes", "on"):
-        emit_system_message("[dim]MCP servers disabled via config[/dim]")
-        return []
-
-    # Get the MCP manager singleton
-    manager = get_mcp_manager()
-
-    # Load configurations from legacy file for backward compatibility
-    configs = load_mcp_server_configs()
-    if not configs:
-        # Check if manager already has servers (could be from new system)
-        existing_servers = manager.list_servers()
-        if not existing_servers:
-            emit_system_message("[dim]No MCP servers configured[/dim]")
-            return []
-    else:
-        # Register servers from legacy config with manager
-        for name, conf in configs.items():
-            try:
-                # Convert legacy format to new ServerConfig
-                server_config = ServerConfig(
-                    id=conf.get("id", f"{name}_{hash(name)}"),
-                    name=name,
-                    type=conf.get("type", "sse"),
-                    enabled=conf.get("enabled", True),
-                    config=conf,
-                )
-
-                # Check if server already registered
-                existing = manager.get_server_by_name(name)
-                if not existing:
-                    # Register new server
-                    manager.register_server(server_config)
-                    emit_system_message(f"[dim]Registered MCP server: {name}[/dim]")
-                else:
-                    # Update existing server config if needed
-                    if existing.config != server_config.config:
-                        manager.update_server(existing.id, server_config)
-                        emit_system_message(f"[dim]Updated MCP server: {name}[/dim]")
-
-            except Exception as e:
-                emit_error(f"Failed to register MCP server '{name}': {str(e)}")
-                continue
-
-    # Get pydantic-ai compatible servers from manager
-    servers = manager.get_servers_for_agent()
-
-    if servers:
-        emit_system_message(
-            f"[green]Successfully loaded {len(servers)} MCP server(s)[/green]"
-        )
-    else:
-        emit_system_message(
-            "[yellow]No MCP servers available (check if servers are enabled)[/yellow]"
-        )
-
-    return servers
-
-
-def reload_mcp_servers():
-    """Reload MCP servers without restarting the agent."""
-    from code_puppy.mcp_ import get_mcp_manager
-
-    manager = get_mcp_manager()
-    # Reload configurations
-    _load_mcp_servers()
-    # Return updated servers
-    return manager.get_servers_for_agent()
-
-
-def reload_code_generation_agent(message_group: str | None):
-    """Force-reload the agent, usually after a model change."""
-    if message_group is None:
-        message_group = str(uuid.uuid4())
-    global _code_generation_agent, _LAST_MODEL_NAME
-    from code_puppy.agents import clear_agent_cache
-    from code_puppy.config import clear_model_cache, get_model_name
-
-    # Clear both ModelFactory cache and config cache when force reloading
-    clear_model_cache()
-    clear_agent_cache()
-
-    # Check if current agent has a pinned model
-    from code_puppy.agents import get_current_agent_config
-
-    agent_config = get_current_agent_config()
-    agent_model_name = None
-    if hasattr(agent_config, "get_model_name"):
-        agent_model_name = agent_config.get_model_name()
-
-    # Use agent-specific model if pinned, otherwise use global model
-    model_name = agent_model_name if agent_model_name else get_model_name()
-    emit_info(
-        f"[bold cyan]Loading Model: {model_name}[/bold cyan]",
-        message_group=message_group,
-    )
-    models_config = ModelFactory.load_config()
-    model = ModelFactory.get_model(model_name, models_config)
-
-    # Get agent-specific system prompt
-    agent_config = get_current_agent_config()
-    emit_info(
-        f"[bold magenta]Loading Agent: {agent_config.display_name}[/bold magenta]",
-        message_group=message_group,
-    )
-
-    instructions = agent_config.get_system_prompt()
-
-    if PUPPY_RULES:
-        instructions += f"\n{PUPPY_RULES}"
-
-    mcp_servers = _load_mcp_servers()
-
-    # Configure model settings with max_tokens if set
-    model_settings_dict = {"seed": 42}
-    # Get current agent to use its method
-    from code_puppy.agents import get_current_agent_config
-    current_agent = get_current_agent_config()
-    output_tokens = max(2048, min(int(0.05 * current_agent.get_model_context_length()) - 1024, 16384))
-    console.print(f"Max output tokens per message: {output_tokens}")
-    model_settings_dict["max_tokens"] = output_tokens
-
-
-    model_settings = ModelSettings(**model_settings_dict)
-    if "gpt-5" in model_name:
-        model_settings_dict["openai_reasoning_effort"] = "off"
-        model_settings_dict["extra_body"] = {
-            "verbosity": "low"
-        }
-        model_settings = OpenAIModelSettings(**model_settings_dict)
-    agent = Agent(
-        model=model,
-        instructions=instructions,
-        output_type=str,
-        retries=3,
-        mcp_servers=mcp_servers,
-        history_processors=[message_history_accumulator],
-        model_settings=model_settings,
-    )
-
-    # Register tools specified by the agent
-    from code_puppy.tools import register_tools_for_agent
-
-    agent_tools = agent_config.get_available_tools()
-    register_tools_for_agent(agent, agent_tools)
-    _code_generation_agent = agent
-    _LAST_MODEL_NAME = model_name
-    return _code_generation_agent
-
-
-def get_code_generation_agent(force_reload=False, message_group: str | None = None):
-    """
-    Retrieve the agent with the currently configured model.
-    Forces a reload if the model has changed, or if force_reload is passed.
-    """
-    global _code_generation_agent, _LAST_MODEL_NAME
-    if message_group is None:
-        message_group = str(uuid.uuid4())
-    from code_puppy.config import get_model_name
-
-    # Get the global model name
-    global_model_name = get_model_name()
-
-    # Check if current agent has a pinned model
-    from code_puppy.agents import get_current_agent_config
-
-    agent_config = get_current_agent_config()
-    agent_model_name = None
-    if hasattr(agent_config, "get_model_name"):
-        agent_model_name = agent_config.get_model_name()
-
-    # Use agent-specific model if pinned, otherwise use global model
-    model_name = agent_model_name if agent_model_name else global_model_name
-
-    if _code_generation_agent is None or _LAST_MODEL_NAME != model_name or force_reload:
-        return reload_code_generation_agent(message_group)
-    return _code_generation_agent
-
-
-def get_custom_usage_limits():
-    """
-    Returns custom usage limits with configurable request limit.
-    This centralizes the configuration of rate limiting for the agent.
-    Default pydantic-ai limit is 50, this increases it to the configured value (default 100).
-    """
-    from code_puppy.config import get_message_limit
-
-    return UsageLimits(request_limit=get_message_limit())

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?"
-}