code-puppy 0.0.43__py3-none-any.whl → 0.0.45__py3-none-any.whl

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/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/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/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.dist-info → code_puppy-0.0.45.dist-info}/METADATA

@@ -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.dist-info → code_puppy-0.0.45.dist-info}/RECORD

@@ -1,7 +1,7 @@
 code_puppy/__init__.py,sha256=orgffM-uGp8g1XCqXGKWaFB4tCCz8TVgsLMPKCOGNx0,81
-code_puppy/agent.py,sha256=OtDhM9j5IrxuLStxM2X0h8tWXcaDAMS4mMZlbmAH7UU,3146
+code_puppy/agent.py,sha256=avoOEorAYUyQYnYVVCezLz1QKDm0Qgx3i_C-BqgDiQQ,3198
 code_puppy/agent_prompts.py,sha256=O8vzlEPct6qRNhKSRctJRwq3aDOhT_-vyVozS5DtmVA,6554
-code_puppy/config.py,sha256=DvmctmrHsOmjiXaiwT-4vAUiTZ4Cib-imyjKewI70eU,1661
+code_puppy/config.py,sha256=vBC4JFKNNXUSJ8l4yHF-Vp4lPwMlHNRafb1yxjIpxAE,2202
 code_puppy/main.py,sha256=bc27bk6rFge95H2BumTTzRLtOx43z5FnsmjIjQx_RpU,10052
 code_puppy/model_factory.py,sha256=Pi46jRr7JaBtJfNpSFk-fK2DlBy0Dv6tG2RcXLf4ZVI,11560
 code_puppy/models.json,sha256=7H-y97YK9BXhag5wJU19rtg24JtZWYx60RsBLBW3WiI,2162
@@ -9,8 +9,8 @@ code_puppy/session_memory.py,sha256=vuLSw1Pfa-MXD4lD8hj2qt65OR_aL2WdoMuF6Jwnc1k,
 code_puppy/version_checker.py,sha256=cK-eU7Y_yYjn7feIONqzzZUdiedozL0iiKXlGKjfq68,395
 code_puppy/command_line/__init__.py,sha256=y7WeRemfYppk8KVbCGeAIiTuiOszIURCDjOMZv_YRmU,45
 code_puppy/command_line/file_path_completion.py,sha256=HAlOu9XVYgJ7FbjdrhKBL0rFmCVFxSGGewdcfKTUsPw,2865
-code_puppy/command_line/meta_command_handler.py,sha256=Q-aDOZfefF0m2Bo-N6_DNVvT0gLrg8nBalrK1uJwEUA,3321
-code_puppy/command_line/model_picker_completion.py,sha256=5D2HkrMCtee7Eo6augls4KU6bh7tu7xR2zH5triRuZg,3658
+code_puppy/command_line/meta_command_handler.py,sha256=w9aZoMZVhPQj2Vix3nqW_K9Psou6nU_Jb2dFwr37yKs,3453
+code_puppy/command_line/model_picker_completion.py,sha256=5VEa6OE9shsF0EafJXzBLpFXhFIkqevJi9RRD-eUvZA,3718
 code_puppy/command_line/prompt_toolkit_completion.py,sha256=pOX3gd2-nn42ReM3_pp3g-NxvslZ9tBSd0S_IeyNelk,5156
 code_puppy/command_line/utils.py,sha256=L1PnV9tNupEW1zeziyb5aGAq8DYP8sMiuQbFYLO5Nus,1236
 code_puppy/tools/__init__.py,sha256=48BVpMt0HAMtz8G_z9SQhX6LnRqR83_AVfMQMf7bY0g,557
@@ -20,9 +20,9 @@ code_puppy/tools/common.py,sha256=dbmyZTrTBQh_0WWpaYN6jEync62W2mMrzNS8UFK0co4,14
 code_puppy/tools/file_modifications.py,sha256=V8as4Pp_qBJSUFuFl5Y-jpiJKH0tUEQ8dNdkGHzpQM0,11645
 code_puppy/tools/file_operations.py,sha256=9M-JntSSTElsLZ1EMLo9Xu2ifje-N0hjXbwARoBm2Tw,9722
 code_puppy/tools/web_search.py,sha256=HhcwX0MMvMDPFO8gr8gzgesD5wPXOypjkxyLZeNwL5g,589
-code_puppy-0.0.43.data/data/code_puppy/models.json,sha256=7H-y97YK9BXhag5wJU19rtg24JtZWYx60RsBLBW3WiI,2162
-code_puppy-0.0.43.dist-info/METADATA,sha256=AiR6F_KVYn0DmXNjwkSKL3KEl-dPHZCmYt2QJVs57D8,4716
-code_puppy-0.0.43.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-code_puppy-0.0.43.dist-info/entry_points.txt,sha256=d8YkBvIUxF-dHNJAj-x4fPEqizbY5d_TwvYpc01U5kw,58
-code_puppy-0.0.43.dist-info/licenses/LICENSE,sha256=31u8x0SPgdOq3izJX41kgFazWsM43zPEF9eskzqbJMY,1075
-code_puppy-0.0.43.dist-info/RECORD,,
+code_puppy-0.0.45.data/data/code_puppy/models.json,sha256=7H-y97YK9BXhag5wJU19rtg24JtZWYx60RsBLBW3WiI,2162
+code_puppy-0.0.45.dist-info/METADATA,sha256=4Z0WVe0KJCQULd1GaPPLptRYEmk7MNGkJP5J_SByFKw,4716
+code_puppy-0.0.45.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+code_puppy-0.0.45.dist-info/entry_points.txt,sha256=d8YkBvIUxF-dHNJAj-x4fPEqizbY5d_TwvYpc01U5kw,58
+code_puppy-0.0.45.dist-info/licenses/LICENSE,sha256=31u8x0SPgdOq3izJX41kgFazWsM43zPEF9eskzqbJMY,1075
+code_puppy-0.0.45.dist-info/RECORD,,