code-puppy 0.0.43__tar.gz → 0.0.45__tar.gz

{code_puppy-0.0.43 → code_puppy-0.0.45}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: code-puppy
-Version: 0.0.43
+Version: 0.0.45
 Summary: Code generation agent
 Author: Michael Pfaffenberger
 License: MIT

{code_puppy-0.0.43 → code_puppy-0.0.45}/code_puppy/agent.py

@@ -50,7 +50,8 @@ def session_memory():
 def reload_code_generation_agent():
     """Force-reload the agent, usually after a model change."""
     global _code_generation_agent, _LAST_MODEL_NAME
-    model_name = os.environ.get("MODEL_NAME", "gpt-4.1")
+    from code_puppy.config import get_model_name
+    model_name = get_model_name()
     console.print(f'[bold cyan]Loading Model: {model_name}[/bold cyan]')
     models_path = Path(MODELS_JSON_PATH) if MODELS_JSON_PATH else Path(__file__).parent / "models.json"
     model = ModelFactory.get_model(model_name, ModelFactory.load_config(models_path))
@@ -79,7 +80,8 @@ def get_code_generation_agent(force_reload=False):
     Forces a reload if the model has changed, or if force_reload is passed.
     """
     global _code_generation_agent, _LAST_MODEL_NAME
-    model_name = os.environ.get("MODEL_NAME", "gpt-4.1")
+    from code_puppy.config import get_model_name
+    model_name = get_model_name()
     if _code_generation_agent is None or _LAST_MODEL_NAME != model_name or force_reload:
         return reload_code_generation_agent()
     return _code_generation_agent

{code_puppy-0.0.43 → code_puppy-0.0.45}/code_puppy/command_line/meta_command_handler.py

@@ -48,8 +48,10 @@ def handle_meta_command(command: str, console: Console) -> bool:
         # Try setting model and show confirmation
         new_input = update_model_in_input(command)
         if new_input is not None:
+            from code_puppy.agent import get_code_generation_agent
             model = get_active_model()
-            console.print(f"[bold green]Active model set to:[/bold green] [cyan]{model}[/cyan]")
+            get_code_generation_agent(force_reload=True)
+            console.print(f"[bold green]Active model set and loaded:[/bold green] [cyan]{model}[/cyan]")
             return True
         # If no model matched, show available models
         model_names = load_model_names()

{code_puppy-0.0.43 → code_puppy-0.0.45}/code_puppy/command_line/model_picker_completion.py

@@ -5,12 +5,12 @@ from prompt_toolkit.completion import Completer, Completion
 from prompt_toolkit.history import FileHistory
 from prompt_toolkit.document import Document
 from prompt_toolkit import PromptSession
+from code_puppy.config import get_model_name, set_model_name
 
 MODELS_JSON_PATH = os.environ.get("MODELS_JSON_PATH")
 if not MODELS_JSON_PATH:
     MODELS_JSON_PATH = os.path.join(os.path.dirname(__file__), '..', 'models.json')
     MODELS_JSON_PATH = os.path.abspath(MODELS_JSON_PATH)
-MODEL_STATE_PATH = os.path.expanduser('~/.code_puppy_model')
 
 def load_model_names():
     with open(MODELS_JSON_PATH, 'r') as f:
@@ -18,18 +18,18 @@ def load_model_names():
     return list(models.keys())
 
 def get_active_model():
-    env_model = os.environ.get('MODEL_NAME')
-    if env_model:
-        return env_model
-    try:
-        with open(MODEL_STATE_PATH, 'r') as f:
-            return f.read().strip()
-    except Exception:
-        return None
+    """
+    Returns the active model from the config using get_model_name().
+    This ensures consistency across the codebase by always using the config value.
+    """
+    return get_model_name()
 
 def set_active_model(model_name: str):
-    with open(MODEL_STATE_PATH, 'w') as f:
-        f.write(model_name.strip())
+    """
+    Sets the active model name by updating both config (for persistence)
+    and env (for process lifetime override).
+    """
+    set_model_name(model_name)
     os.environ['MODEL_NAME'] = model_name.strip()
     # Reload agent globally
     try:

{code_puppy-0.0.43 → code_puppy-0.0.45}/code_puppy/config.py

@@ -51,3 +51,18 @@ def get_puppy_name():
 
 def get_owner_name():
     return get_value("owner_name") or "Master"
+
+# --- MODEL STICKY EXTENSION STARTS HERE ---
+def get_model_name():
+    """Returns the last used model name stored in config, or None if unset."""
+    return get_value("model") or "gpt-4o"
+
+def set_model_name(model: str):
+    """Sets the model name in the persistent config file."""
+    config = configparser.ConfigParser()
+    config.read(CONFIG_FILE)
+    if DEFAULT_SECTION not in config:
+        config[DEFAULT_SECTION] = {}
+    config[DEFAULT_SECTION]["model"] = model or ""
+    with open(CONFIG_FILE, "w") as f:
+        config.write(f)

{code_puppy-0.0.43 → code_puppy-0.0.45}/pyproject.toml

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