code-puppy 0.0.42__tar.gz → 0.0.44__tar.gz

{code_puppy-0.0.42 → code_puppy-0.0.44}/PKG-INFO

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

{code_puppy-0.0.42 → code_puppy-0.0.44}/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-0.0.42 → code_puppy-0.0.44}/code_puppy/command_line/prompt_toolkit_completion.py

@@ -15,6 +15,8 @@ from prompt_toolkit import PromptSession
 from prompt_toolkit.completion import merge_completers
 from prompt_toolkit.history import FileHistory
 from prompt_toolkit.styles import Style
+from prompt_toolkit.key_binding import KeyBindings
+from prompt_toolkit.keys import Keys
 
 from code_puppy.command_line.model_picker_completion import (
     ModelNameCompleter,
@@ -83,10 +85,17 @@ async def get_input_with_combined_completion(prompt_str = '>>> ', history_file:
         ModelNameCompleter(trigger='~m'),
         CDCompleter(trigger='~cd'),
     ])
+    # Add custom key bindings for Alt+M to insert a new line without submitting
+    bindings = KeyBindings()
+    @bindings.add(Keys.Escape, 'm')  # Alt+M
+    def _(event):
+        event.app.current_buffer.insert_text('\n')
+
     session = PromptSession(
         completer=completer,
         history=history,
-        complete_while_typing=True
+        complete_while_typing=True,
+        key_bindings=bindings
     )
     # If they pass a string, backward-compat: convert it to formatted_text
     if isinstance(prompt_str, str):

{code_puppy-0.0.42 → code_puppy-0.0.44}/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.42 → code_puppy-0.0.44}/pyproject.toml

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