code-puppy 0.0.163__py3-none-any.whl → 0.0.164__py3-none-any.whl

code_puppy/agent.py

@@ -6,7 +6,6 @@ from pydantic_ai import Agent
 from pydantic_ai.settings import ModelSettings
 from pydantic_ai.usage import UsageLimits
 
-from code_puppy.agents import get_current_agent_config
 from code_puppy.message_history_processor import (
     get_model_context_length,
     message_history_accumulator,
@@ -136,11 +135,12 @@ def reload_code_generation_agent(message_group: str | None):
 
     # 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'):
+    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(
@@ -203,17 +203,18 @@ def get_code_generation_agent(force_reload=False, message_group: str | None = No
 
     # 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'):
+    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

code_puppy/agents/agent_creator_agent.py

@@ -28,15 +28,17 @@ class AgentCreatorAgent(BaseAgent):
     def get_system_prompt(self) -> str:
         available_tools = get_available_tool_names()
         agents_dir = get_user_agents_directory()
-        
+
         # Load available models dynamically
         models_config = ModelFactory.load_config()
         model_descriptions = []
         for model_name, model_info in models_config.items():
-            model_type = model_info.get('type', 'Unknown')
-            context_length = model_info.get('context_length', 'Unknown')
-            model_descriptions.append(f"- **{model_name}**: {model_type} model with {context_length} context")
-        
+            model_type = model_info.get("type", "Unknown")
+            context_length = model_info.get("context_length", "Unknown")
+            model_descriptions.append(
+                f"- **{model_name}**: {model_type} model with {context_length} context"
+            )
+
         available_models_str = "\n".join(model_descriptions)
 
         return f"""You are the Agent Creator! 🏗️ Your mission is to help users create awesome JSON agent files through an interactive process.

code_puppy/command_line/command_handler.py

@@ -57,7 +57,9 @@ def get_commands_help():
     )
     help_lines.append(
         Text("/compact", style="cyan")
-        + Text("              Summarize and compact current chat history (uses compaction_strategy config)")
+        + Text(
+            "              Summarize and compact current chat history (uses compaction_strategy config)"
+        )
     )
     help_lines.append(
         Text("/dump_context", style="cyan")
@@ -371,7 +373,7 @@ def handle_command(command: str):
 
         # If no model matched, show available models
         from code_puppy.command_line.model_picker_completion import load_model_names
-        
+
         new_input = update_model_in_input(model_command)
         if new_input is not None:
             from code_puppy.agents.runtime_manager import get_runtime_agent_manager
@@ -406,76 +408,76 @@ def handle_command(command: str):
         from code_puppy.agents.json_agent import discover_json_agents
         from code_puppy.command_line.model_picker_completion import load_model_names
         import json
-        
+
         tokens = command.split()
-        
+
         if len(tokens) != 3:
             emit_warning("Usage: /pin_model <agent-name> <model-name>")
-            
+
             # Show available models and JSON agents
             available_models = load_model_names()
             json_agents = discover_json_agents()
-            
+
             emit_info("Available models:")
             for model in available_models:
                 emit_info(f"  [cyan]{model}[/cyan]")
-                
+
             if json_agents:
                 emit_info("\nAvailable JSON agents:")
                 for agent_name, agent_path in json_agents.items():
                     emit_info(f"  [cyan]{agent_name}[/cyan] ({agent_path})")
             return True
-            
+
         agent_name = tokens[1].lower()
         model_name = tokens[2]
-        
+
         # Check if model exists
         available_models = load_model_names()
         if model_name not in available_models:
             emit_error(f"Model '{model_name}' not found")
             emit_warning(f"Available models: {', '.join(available_models)}")
             return True
-            
+
         # Check that we're modifying a JSON agent (not a built-in Python agent)
         json_agents = discover_json_agents()
         if agent_name not in json_agents:
             emit_error(f"JSON agent '{agent_name}' not found")
-            
+
             # Show available JSON agents
             if json_agents:
                 emit_info("Available JSON agents:")
                 for name, path in json_agents.items():
                     emit_info(f"  [cyan]{name}[/cyan] ({path})")
             return True
-            
+
         agent_file_path = json_agents[agent_name]
-        
+
         # Load, modify, and save the agent configuration
         try:
             with open(agent_file_path, "r", encoding="utf-8") as f:
                 agent_config = json.load(f)
-                
+
             # Set the model
             agent_config["model"] = model_name
-            
+
             # Save the updated configuration
             with open(agent_file_path, "w", encoding="utf-8") as f:
                 json.dump(agent_config, f, indent=2, ensure_ascii=False)
-                
+
             emit_success(f"Model '{model_name}' pinned to agent '{agent_name}'")
-            
+
             # If this is the current agent, reload it to use the new model
             from code_puppy.agents import get_current_agent_config
             from code_puppy.agents.runtime_manager import get_runtime_agent_manager
-            
+
             current_agent = get_current_agent_config()
             if current_agent.name == agent_name:
                 manager = get_runtime_agent_manager()
                 manager.reload_agent()
                 emit_info(f"Active agent reloaded with pinned model '{model_name}'")
-            
+
             return True
-            
+
         except Exception as e:
             emit_error(f"Failed to pin model to agent '{agent_name}': {e}")
             return True

code_puppy/command_line/prompt_toolkit_completion.py

@@ -147,9 +147,9 @@ def get_prompt_with_active_model(base: str = ">>> "):
 
     # Check if current agent has a pinned model
     agent_model = None
-    if current_agent and hasattr(current_agent, 'get_model_name'):
+    if current_agent and hasattr(current_agent, "get_model_name"):
         agent_model = current_agent.get_model_name()
-    
+
     # Determine which model to display
     if agent_model and agent_model != global_model:
         # Show both models when they differ

code_puppy/config.py

@@ -492,4 +492,4 @@ def save_command_to_history(command: str):
         error_msg = (
             f"❌ An unexpected error occurred while saving command history: {str(e)}"
         )
-        direct_console.print(f"[bold red]{error_msg}[/bold red]")
+        direct_console.print(f"[bold red]{error_msg}[/bold red]")

code_puppy/model_factory.py

@@ -95,9 +95,20 @@ class ModelFactory:
                 config = json.load(f)
 
         if pathlib.Path(EXTRA_MODELS_FILE).exists():
-            with open(EXTRA_MODELS_FILE, "r") as f:
-                extra_config = json.load(f)
-                config.update(extra_config)
+            try:
+                with open(EXTRA_MODELS_FILE, "r") as f:
+                    extra_config = json.load(f)
+                    config.update(extra_config)
+            except json.JSONDecodeError as e:
+                logging.getLogger(__name__).warning(
+                    f"Failed to load extra models config from {EXTRA_MODELS_FILE}: Invalid JSON - {e}\n"
+                    f"Please check your extra_models.json file for syntax errors."
+                )
+            except Exception as e:
+                logging.getLogger(__name__).warning(
+                    f"Failed to load extra models config from {EXTRA_MODELS_FILE}: {e}\n"
+                    f"The extra models configuration will be ignored."
+                )
         return config
 
     @staticmethod

code_puppy/tools/command_runner.py

@@ -12,7 +12,6 @@ from pydantic_ai import RunContext
 from rich.markdown import Markdown
 from rich.text import Text
 
-from code_puppy.callbacks import on_run_shell_command
 from code_puppy.messaging import (
     emit_divider,
     emit_error,

code_puppy/tools/file_modifications.py

@@ -20,7 +20,6 @@ import json_repair
 from pydantic import BaseModel
 from pydantic_ai import RunContext
 
-from code_puppy.callbacks import on_delete_file, on_edit_file
 from code_puppy.messaging import emit_error, emit_info, emit_warning
 from code_puppy.tools.common import _find_best_window, generate_group_id
 
@@ -567,9 +566,9 @@ def register_edit_file(agent):
             except Exception as e:
                 return {
                     "success": False,
-                    "path": 'Not retrievable in Payload',
+                    "path": "Not retrievable in Payload",
                     "message": f"edit_file call failed: {str(e)} - this means the tool failed to parse your inputs. Refer to the following examples: {parse_error_message}",
-                    "changed": False
+                    "changed": False,
                 }
 
         # Call _edit_file which will extract file_path from payload and handle group_id generation

code_puppy/tui/components/chat_view.py

@@ -262,29 +262,35 @@ class ChatView(VerticalScroll):
             separator = "\n"
 
         # Handle content concatenation carefully to preserve Rich objects
-        if hasattr(last_message.content, "__rich_console__") or hasattr(message.content, "__rich_console__"):
+        if hasattr(last_message.content, "__rich_console__") or hasattr(
+            message.content, "__rich_console__"
+        ):
             # If either content is a Rich object, convert both to text and concatenate
             from io import StringIO
             from rich.console import Console
-            
+
             # Convert existing content to string
             if hasattr(last_message.content, "__rich_console__"):
                 string_io = StringIO()
-                temp_console = Console(file=string_io, width=80, legacy_windows=False, markup=False)
+                temp_console = Console(
+                    file=string_io, width=80, legacy_windows=False, markup=False
+                )
                 temp_console.print(last_message.content)
                 existing_content = string_io.getvalue().rstrip("\n")
             else:
                 existing_content = str(last_message.content)
-            
+
             # Convert new content to string
             if hasattr(message.content, "__rich_console__"):
                 string_io = StringIO()
-                temp_console = Console(file=string_io, width=80, legacy_windows=False, markup=False)
+                temp_console = Console(
+                    file=string_io, width=80, legacy_windows=False, markup=False
+                )
                 temp_console.print(message.content)
                 new_content = string_io.getvalue().rstrip("\n")
             else:
                 new_content = str(message.content)
-            
+
             # Combine as plain text
             last_message.content = existing_content + separator + new_content
         else:

{code_puppy-0.0.163.dist-info → code_puppy-0.0.164.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: code-puppy
-Version: 0.0.163
+Version: 0.0.164
 Summary: Code generation agent
 Project-URL: repository, https://github.com/mpfaffenberger/code_puppy
 Project-URL: HomePage, https://github.com/mpfaffenberger/code_puppy

{code_puppy-0.0.163.dist-info → code_puppy-0.0.164.dist-info}/RECORD

@@ -1,12 +1,12 @@
 code_puppy/__init__.py,sha256=ehbM1-wMjNmOXk_DBhhJECFyBv2dRHwwo7ucjHeM68E,107
 code_puppy/__main__.py,sha256=pDVssJOWP8A83iFkxMLY9YteHYat0EyWDQqMkKHpWp4,203
-code_puppy/agent.py,sha256=hZOrJJJDwtLtb82b6Zyt5CZb8RbzDtvHMIBUPVYsgOM,8133
+code_puppy/agent.py,sha256=wCARhq7PcD38c4l83fuXm4Us_6c1TXVa0U7TnBnNPEY,8064
 code_puppy/callbacks.py,sha256=6wYB6K_fGSCkKKEFaYOYkJT45WaV5W_NhUIzcvVH_nU,5060
-code_puppy/config.py,sha256=Kp21bilRjqtgC8mYrLTIgR7CZ6_wbvRbsS_Rw7Cx6gs,16385
+code_puppy/config.py,sha256=J8XU0iOPtfzrkDD49b4TdrdHoQmW2kaP-25PPbGGdKU,16386
 code_puppy/http_utils.py,sha256=BAvt4hed7fVMXglA7eS9gOb08h2YTuOyai6VmQq09fg,3432
 code_puppy/main.py,sha256=tYLfhUjPTJ-4S1r-pr-jSbn6kIU1iYvt2Z8lxI7zDFY,22220
 code_puppy/message_history_processor.py,sha256=aV-vcRcOQJPZPlrokB4CaLMxEU3Y4nDiabb9Ov_sJeU,15933
-code_puppy/model_factory.py,sha256=_drvOUKx2ivLY_TfE-D6gdZDJzIMyPqf2IWU83KEIE4,13158
+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
@@ -16,20 +16,20 @@ code_puppy/summarization_agent.py,sha256=kos4_YK-l_YjYRq4Fs4X5YoTUbmAcDhhPqefL-r
 code_puppy/version_checker.py,sha256=bjLDmgGPrl7XnYwX1u13O8uFlsfikV90PK6nbA9Z9QU,1150
 code_puppy/agents/__init__.py,sha256=SwtHGNG1GIgDBv7y3EGIXOXEWrG_Ou7fEknNgFbrHv8,594
 code_puppy/agents/agent_code_puppy.py,sha256=sbuQxLzlkMbPOyLbILbXOAHecRsxbFdQt13HJ_GEqQo,7972
-code_puppy/agents/agent_creator_agent.py,sha256=jyOfqCa4d3Tqlv0ZODjK1J6cRwJjBxKyNa9RQSdsB44,23256
+code_puppy/agents/agent_creator_agent.py,sha256=chxZlvvl_12ol1d9V2WF7cjE1Px9d-S6NEH3CaKg_lc,23270
 code_puppy/agents/agent_manager.py,sha256=nXvro6fpX8KA-NedRoVJuhJW966trrePOrH4eAnqq40,17034
 code_puppy/agents/agent_orchestrator.json,sha256=Iqnc0p6ICoAlUTMkEsi1XXMXJi4pdxVnWZUMaih6s5o,1267
 code_puppy/agents/base_agent.py,sha256=XYSff6IQX9Z6C7hPVbN_YXC2xfjwd268e2jtG3ZGnVk,3450
 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=w0sE_oGRe6oV05a8-9vJmz0gZWRq_whE8a2mk8Tx7M8,24612
+code_puppy/command_line/command_handler.py,sha256=gdsOR04CIkUQjSMJtvrKDdAQNe6cF8JcsTObN-XfTLE,24414
 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/motd.py,sha256=PEdkp3ZnydVfvd7mNJylm8YyFNUKg9jmY6uwkA1em8c,2152
-code_puppy/command_line/prompt_toolkit_completion.py,sha256=saq1ZRjcxK8s2M-SKy-7I62xGC4Cpxdw02bvxbMwTxA,10010
+code_puppy/command_line/prompt_toolkit_completion.py,sha256=PLb1Yt_65iQI2mzPEPxPsSaRww5Ur951Qn-K-TLMCGQ,10006
 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=RghGmOCjwwYJfrbLiV5UZojMA4-FNPlM04LKci0rpAw,6415
@@ -78,16 +78,16 @@ code_puppy/messaging/spinner/textual_spinner.py,sha256=Omx9A-FSPkxYDMYgBXgYMBQnK
 code_puppy/plugins/__init__.py,sha256=fksDqMUiXPJ5WNuMsYsVR8ulueQRCXPlvECEyicHPtQ,1312
 code_puppy/tools/__init__.py,sha256=YiiXRqxU1BEJ5t0Oe163lSqOneI9sKtwDW0swCPgBt4,2119
 code_puppy/tools/agent_tools.py,sha256=bHMrFIbYRhuubR41G_XdLsk3cUKWfIPl2O4bVzo2pE0,5591
-code_puppy/tools/command_runner.py,sha256=kK_lKLTMRvJmNLfunTBTeHFT8Z9EiRC3c_mhyzDhvxM,20911
+code_puppy/tools/command_runner.py,sha256=6mGO90xks2OwJCEvwrsrNf3lRfcKXrMeLbkJEpX5bMw,20857
 code_puppy/tools/common.py,sha256=pL-9xcRs3rxU7Fl9X9EUgbDp2-csh2LLJ5DHH_KAHKY,10596
-code_puppy/tools/file_modifications.py,sha256=_8IkNa3zmboN1usTHhQEBar5ctSOtvH7j74mFmV4WWA,23279
+code_puppy/tools/file_modifications.py,sha256=EaDWcv6gi8wAvpgyeJdKSKPWg9fTpZoEkxQiLCE6rn4,23218
 code_puppy/tools/file_operations.py,sha256=Cxa5bqbVNwgViSdBjwGOrqWmtxBPhQRjDBepgt2Xuxk,30852
 code_puppy/tools/tools_content.py,sha256=bsBqW-ppd1XNAS_g50B3UHDQBWEALC1UneH6-afz1zo,2365
 code_puppy/tui/__init__.py,sha256=XesAxIn32zLPOmvpR2wIDxDAnnJr81a5pBJB4cZp1Xs,321
 code_puppy/tui/app.py,sha256=nPOzwlusjdWzBfu__EbC3Q0etkPrqRq-2g-mk4IcfG4,39378
 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=UsFdypI6RGcj0ieCWhbr0yba5Y84JVDQ-KKgbrktPLM,19851
+code_puppy/tui/components/chat_view.py,sha256=NfyNXuN2idPht1rKJB4YhHVXb1AIRNO5q_nLdt8Ocug,19913
 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
@@ -104,9 +104,9 @@ code_puppy/tui/screens/help.py,sha256=eJuPaOOCp7ZSUlecearqsuX6caxWv7NQszUh0tZJjB
 code_puppy/tui/screens/mcp_install_wizard.py,sha256=xqwN5omltMkfxWZwXj3D2PbXbtrxUi1dT0XT77oxOKk,27685
 code_puppy/tui/screens/settings.py,sha256=GMpv-qa08rorAE9mj3AjmqjZFPhmeJ_GWd-DBHG6iAA,10671
 code_puppy/tui/screens/tools.py,sha256=3pr2Xkpa9Js6Yhf1A3_wQVRzFOui-KDB82LwrsdBtyk,1715
-code_puppy-0.0.163.data/data/code_puppy/models.json,sha256=iXmLZGflnQcu2DRh4WUlgAhoXdvoxUc7KBhB8YxawXM,3088
-code_puppy-0.0.163.dist-info/METADATA,sha256=VEvu8qBW4fj-rcg7OVfn75GBzxcUECvRl7fz1PWCjto,19567
-code_puppy-0.0.163.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-code_puppy-0.0.163.dist-info/entry_points.txt,sha256=d8YkBvIUxF-dHNJAj-x4fPEqizbY5d_TwvYpc01U5kw,58
-code_puppy-0.0.163.dist-info/licenses/LICENSE,sha256=31u8x0SPgdOq3izJX41kgFazWsM43zPEF9eskzqbJMY,1075
-code_puppy-0.0.163.dist-info/RECORD,,
+code_puppy-0.0.164.data/data/code_puppy/models.json,sha256=iXmLZGflnQcu2DRh4WUlgAhoXdvoxUc7KBhB8YxawXM,3088
+code_puppy-0.0.164.dist-info/METADATA,sha256=3mQaqHuxj9zoChbe-Q5477PK05cpaOCpxUEdjh0lZwA,19567
+code_puppy-0.0.164.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+code_puppy-0.0.164.dist-info/entry_points.txt,sha256=d8YkBvIUxF-dHNJAj-x4fPEqizbY5d_TwvYpc01U5kw,58
+code_puppy-0.0.164.dist-info/licenses/LICENSE,sha256=31u8x0SPgdOq3izJX41kgFazWsM43zPEF9eskzqbJMY,1075
+code_puppy-0.0.164.dist-info/RECORD,,