code-puppy 0.0.198__tar.gz → 0.0.199__tar.gz

{code_puppy-0.0.198 → code_puppy-0.0.199}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: code-puppy
-Version: 0.0.198
+Version: 0.0.199
 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.198 → code_puppy-0.0.199}/code_puppy/agents/base_agent.py

@@ -460,16 +460,21 @@ class BaseAgent(ABC):
 
     def get_model_context_length(self) -> int:
         """
-        Get the context length for the currently configured model from models.json
-        """
-        model_configs = ModelFactory.load_config()
-        model_name = get_global_model_name()
-
-        # Get context length from model config
-        model_config = model_configs.get(model_name, {})
-        context_length = model_config.get("context_length", 128000)  # Default value
+        Return the context length for this agent's effective model.
 
-        return int(context_length)
+        Honors per-agent pinned model via `self.get_model_name()`; falls back
+        to global model when no pin is set. Defaults conservatively on failure.
+        """
+        try:
+            model_configs = ModelFactory.load_config()
+            # Use the agent's effective model (respects /pin_model)
+            model_name = self.get_model_name()
+            model_config = model_configs.get(model_name, {})
+            context_length = model_config.get("context_length", 128000)
+            return int(context_length)
+        except Exception:
+            # Be safe; don't blow up status/compaction if model lookup fails
+            return 128000
 
     def prune_interrupted_tool_calls(
         self, messages: List[ModelMessage]

{code_puppy-0.0.198 → code_puppy-0.0.199}/code_puppy/command_line/model_picker_completion.py

@@ -29,13 +29,22 @@ def set_active_model(model_name: str):
     Sets the active model name by updating the config (for persistence).
     """
     set_model_name(model_name)
-    # Reload agent globally
+    # Reload the currently active agent so the new model takes effect immediately
     try:
-        from code_puppy.agent import reload_code_generation_agent
-
-        reload_code_generation_agent()  # This will reload dynamically everywhere
+        from code_puppy.agents import get_current_agent
+
+        current_agent = get_current_agent()
+        # JSON agents may need to refresh their config before reload
+        if hasattr(current_agent, "refresh_config"):
+            try:
+                current_agent.refresh_config()
+            except Exception:
+                # Non-fatal, continue to reload
+                ...
+        current_agent.reload_code_generation_agent()
     except Exception:
-        pass  # If reload fails, agent will still be switched next interpreter run
+        # Swallow errors to avoid breaking the prompt flow; model persists for next run
+        pass
 
 
 class ModelNameCompleter(Completer):

{code_puppy-0.0.198 → code_puppy-0.0.199}/code_puppy/tui/screens/settings.py

@@ -217,6 +217,20 @@ class SettingsScreen(ModalScreen):
             # Save model selection
             if selected_model:
                 set_model_name(selected_model)
+                # Reload the active agent so model switch takes effect immediately
+                try:
+                    from code_puppy.agents import get_current_agent
+
+                    current_agent = get_current_agent()
+                    if hasattr(current_agent, "refresh_config"):
+                        try:
+                            current_agent.refresh_config()
+                        except Exception:
+                            ...
+                    current_agent.reload_code_generation_agent()
+                except Exception:
+                    # Non-fatal: settings saved; reload will happen on next run if needed
+                    pass
 
             set_config_value("yolo_mode", yolo_mode)
 

{code_puppy-0.0.198 → code_puppy-0.0.199}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "code-puppy"
-version = "0.0.198"
+version = "0.0.199"
 description = "Code generation agent"
 readme = "README.md"
 requires-python = ">=3.11"