code-puppy 0.0.50__tar.gz → 0.0.51__tar.gz

{code_puppy-0.0.50 → code_puppy-0.0.51}/PKG-INFO

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

{code_puppy-0.0.50 → code_puppy-0.0.51}/code_puppy/command_line/meta_command_handler.py

@@ -47,7 +47,6 @@ def handle_meta_command(command: str, console: Console) -> bool:
     if command.strip().startswith("~show"):
         from code_puppy.config import get_puppy_name, get_owner_name
         from code_puppy.command_line.model_picker_completion import get_active_model
-        import os
         puppy_name = get_puppy_name()
         owner_name = get_owner_name()
         model = get_active_model()
@@ -61,6 +60,34 @@ def handle_meta_command(command: str, console: Console) -> bool:
 ''')
         return True
 
+    if command.startswith("~set"):
+        # Syntax: ~set KEY=VALUE or ~set KEY VALUE
+        from code_puppy.config import set_config_value, get_config_keys
+        tokens = command.split(None, 2)
+        argstr = command[len('~set'):].strip()
+        key = None
+        value = None
+        if '=' in argstr:
+            key, value = argstr.split('=', 1)
+            key = key.strip()
+            value = value.strip()
+        elif len(tokens) >= 3:
+            key = tokens[1]
+            value = tokens[2]
+        elif len(tokens) == 2:
+            key = tokens[1]
+            value = ''
+        else:
+            console.print('[yellow]Usage:[/yellow] ~set KEY=VALUE or ~set KEY VALUE')
+            console.print('Config keys: ' + ', '.join(get_config_keys()))
+            return True
+        if key:
+            set_config_value(key, value)
+            console.print(f'[green]🌶 Set[/green] [cyan]{key}[/cyan] = "{value}" in puppy.cfg!')
+        else:
+            console.print('[red]You must supply a key.[/red]')
+        return True
+
     if command.startswith("~m"):
         # Try setting model and show confirmation
         new_input = update_model_in_input(command)
@@ -76,7 +103,7 @@ def handle_meta_command(command: str, console: Console) -> bool:
         console.print(f"[yellow]Usage:[/yellow] ~m <model_name>")
         return True
     if command in ("~help", "~h"):
-        console.print("[bold magenta]Meta commands available:[/bold magenta]\n  ~m <model>: Pick a model from your list!\n  ~cd [dir]: Change directories\n  ~codemap [dir]: Visualize project code structure\n  ~help: Show this help\n  (More soon. Woof!)")
+        console.print("[bold magenta]Meta commands available:[/bold magenta]\n  ~m <model>: Pick a model from your list!\n  ~cd [dir]: Change directories\n  ~codemap [dir]: Visualize project code structure\n  ~set KEY=VALUE: Set a puppy.cfg setting!\n  ~help: Show this help\n  (More soon. Woof!)")
         return True
     if command.startswith("~"):
         name = command[1:].split()[0] if len(command)>1 else ""

{code_puppy-0.0.50 → code_puppy-0.0.51}/code_puppy/command_line/prompt_toolkit_completion.py

@@ -1,6 +1,6 @@
 import os
 from code_puppy.command_line.utils import list_directory
-from code_puppy.config import get_puppy_name, get_owner_name
+from code_puppy.config import get_puppy_name, get_owner_name, get_config_keys, get_value
 # ANSI color codes are no longer necessary because prompt_toolkit handles
 # styling via the `Style` class. We keep them here commented-out in case
 # someone needs raw ANSI later, but they are unused in the current code.
@@ -27,6 +27,43 @@ from code_puppy.command_line.file_path_completion import FilePathCompleter
 
 from prompt_toolkit.completion import Completer, Completion
 
+class SetCompleter(Completer):
+    def __init__(self, trigger: str = '~set'):
+        self.trigger = trigger
+    def get_completions(self, document, complete_event):
+        text = document.text_before_cursor
+        if not text.strip().startswith(self.trigger):
+            return
+        # If the only thing typed is exactly '~set', suggest space
+        if text.strip() == self.trigger:
+            yield Completion(self.trigger + ' ', start_position=-len(self.trigger), display=f'{self.trigger} ', display_meta='set config')
+        tokens = text.strip().split()
+        # completion for the first arg after ~set
+        if len(tokens) == 1:
+            # user just typed ~set <-- suggest config keys
+            base = ''
+        else:
+            base = tokens[1]
+        # --- SPECIAL HANDLING FOR 'model' KEY ---
+        if base == 'model':
+            # Don't return any completions -- let ModelNameCompleter handle it
+            return
+        for key in get_config_keys():
+            if key == 'model':
+                continue  # exclude 'model' from regular ~set completions
+            if key.startswith(base):
+                prev_value = get_value(key)
+                # Ensure there's a space after '~set' if it's the only thing typed
+                if text.strip() == self.trigger or text.strip() == self.trigger + '':
+                    prefix = self.trigger + ' '  # Always enforce a space
+                    insert_text = f'{prefix}{key} = {prev_value}' if prev_value is not None else f'{prefix}{key} = '
+                    sp = -len(text)
+                else:
+                    insert_text = f'{key} = {prev_value}' if prev_value is not None else f'{key} = '
+                    sp = -len(base)
+                # Make it obvious the value part is from before
+                yield Completion(insert_text, start_position=sp, display_meta=f'puppy.cfg key (was: {prev_value})' if prev_value is not None else 'puppy.cfg key')
+
 class CDCompleter(Completer):
     def __init__(self, trigger: str = '~cd'):
         self.trigger = trigger
@@ -84,6 +121,7 @@ async def get_input_with_combined_completion(prompt_str = '>>> ', history_file:
         FilePathCompleter(symbol='@'),
         ModelNameCompleter(trigger='~m'),
         CDCompleter(trigger='~cd'),
+        SetCompleter(trigger='~set'),
     ])
     # Add custom key bindings for Alt+M to insert a new line without submitting
     bindings = KeyBindings()

{code_puppy-0.0.50 → code_puppy-0.0.51}/code_puppy/config.py

@@ -50,6 +50,31 @@ def get_puppy_name():
 def get_owner_name():
     return get_value("owner_name") or "Master"
 
+# --- CONFIG SETTER STARTS HERE ---
+def get_config_keys():
+    '''
+    Returns the list of all config keys currently in puppy.cfg,
+    plus certain preset expected keys (e.g. "yolo_mode", "model").
+    '''
+    default_keys = ['yolo_mode', 'model']
+    config = configparser.ConfigParser()
+    config.read(CONFIG_FILE)
+    keys = set(config[DEFAULT_SECTION].keys()) if DEFAULT_SECTION in config else set()
+    keys.update(default_keys)
+    return sorted(keys)
+
+def set_config_value(key: str, value: str):
+    '''
+    Sets a config value in the persistent config file.
+    '''
+    config = configparser.ConfigParser()
+    config.read(CONFIG_FILE)
+    if DEFAULT_SECTION not in config:
+        config[DEFAULT_SECTION] = {}
+    config[DEFAULT_SECTION][key] = value
+    with open(CONFIG_FILE, 'w') as f:
+        config.write(f)
+
 # --- MODEL STICKY EXTENSION STARTS HERE ---
 def get_model_name():
     """Returns the last used model name stored in config, or None if unset."""
@@ -66,20 +91,25 @@ def set_model_name(model: str):
         config.write(f)
 
 def get_yolo_mode():
-    """Checks env var CODE_PUPPY_YOLO or puppy.cfg for 'yolo_mode'.
-    Returns True if either is explicitly truthy, else False by default.
-    Env var wins if both are set.
-    Allowed env/cfg values: 1, '1', 'true', 'yes', 'on' (case-insensitive).
     """
-    env_val = os.getenv('YOLO_MODE')
+    Checks puppy.cfg for 'yolo_mode' (case-insensitive in value only).
+    If not set, checks YOLO_MODE env var:
+    - If found in env, saves that value to puppy.cfg for future use.
+    - If neither present, defaults to False.
+    Allowed values for ON: 1, '1', 'true', 'yes', 'on' (all case-insensitive for value).
+    Always prioritizes the config once set!
+    """
     true_vals = {'1', 'true', 'yes', 'on'}
-    if env_val is not None:
-        if str(env_val).strip().lower() in true_vals:
-            return True
-        return False
     cfg_val = get_value('yolo_mode')
     if cfg_val is not None:
         if str(cfg_val).strip().lower() in true_vals:
             return True
         return False
+    env_val = os.getenv('YOLO_MODE')
+    if env_val is not None:
+        # Persist the env value now
+        set_config_value('yolo_mode', env_val)
+        if str(env_val).strip().lower() in true_vals:
+            return True
+        return False
     return False

{code_puppy-0.0.50 → code_puppy-0.0.51}/pyproject.toml

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