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

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,26 @@ def load_model_names():
     return list(models.keys())
 
 def get_active_model():
+    """
+    Returns the active model in order of priority:
+    1. Model stored in config (sticky, highest priority)
+    2. MODEL_NAME environment variable (for temporary override/testing)
+    3. None if unset
+    """
+    model = get_model_name()
+    if model:
+        return 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
+    return None
 
 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")
+
+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.44.dist-info}/METADATA

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

{code_puppy-0.0.43.dist-info → code_puppy-0.0.44.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_prompts.py,sha256=O8vzlEPct6qRNhKSRctJRwq3aDOhT_-vyVozS5DtmVA,6554
-code_puppy/config.py,sha256=DvmctmrHsOmjiXaiwT-4vAUiTZ4Cib-imyjKewI70eU,1661
+code_puppy/config.py,sha256=LZMyrRH2YqUCvqZ6R9V1MlXG6MV9Zi800QnElja-jmk,2190
 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
@@ -10,7 +10,7 @@ code_puppy/version_checker.py,sha256=cK-eU7Y_yYjn7feIONqzzZUdiedozL0iiKXlGKjfq68
 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/model_picker_completion.py,sha256=bs-tor8vR5OShu6GB1yKt4RKdMaAI0W8tuFDeum0V1c,3907
 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.44.data/data/code_puppy/models.json,sha256=7H-y97YK9BXhag5wJU19rtg24JtZWYx60RsBLBW3WiI,2162
+code_puppy-0.0.44.dist-info/METADATA,sha256=WoVZ0GiAyd05MLZhQzxnTe-wegvFvxQD5-oB_QCulnc,4716
+code_puppy-0.0.44.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+code_puppy-0.0.44.dist-info/entry_points.txt,sha256=d8YkBvIUxF-dHNJAj-x4fPEqizbY5d_TwvYpc01U5kw,58
+code_puppy-0.0.44.dist-info/licenses/LICENSE,sha256=31u8x0SPgdOq3izJX41kgFazWsM43zPEF9eskzqbJMY,1075
+code_puppy-0.0.44.dist-info/RECORD,,