code-puppy 0.0.50__py3-none-any.whl → 0.0.51__py3-none-any.whl

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

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

@@ -1,7 +1,7 @@
 code_puppy/__init__.py,sha256=orgffM-uGp8g1XCqXGKWaFB4tCCz8TVgsLMPKCOGNx0,81
 code_puppy/agent.py,sha256=avoOEorAYUyQYnYVVCezLz1QKDm0Qgx3i_C-BqgDiQQ,3198
 code_puppy/agent_prompts.py,sha256=A6ADydqbHozIAtwOy_UY-fYZ2XDE9_5oV3oxOMtVCFA,6782
-code_puppy/config.py,sha256=eVUWtcRHuD7cwHuBOgypTV7qINjT7Z5sBmXprKrOjEs,2869
+code_puppy/config.py,sha256=jsPnwDjz_H69Oww5JyEg3eud3rYMKp3ah2WrnablRak,3894
 code_puppy/main.py,sha256=bc27bk6rFge95H2BumTTzRLtOx43z5FnsmjIjQx_RpU,10052
 code_puppy/model_factory.py,sha256=vDlEoEKkOoikExCqi5oxb_PuTF8STUMABXAbPHV_p34,11639
 code_puppy/models.json,sha256=7H-y97YK9BXhag5wJU19rtg24JtZWYx60RsBLBW3WiI,2162
@@ -9,9 +9,9 @@ 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=C14QL_MMNnKQh1rKQ7KdQkHCJRl5oRTKd9NYSe8OCqk,4220
+code_puppy/command_line/meta_command_handler.py,sha256=lavg91Qak68v0S7nxZfcIObHfd-hQiJJaeOGAR1BGzo,5290
 code_puppy/command_line/model_picker_completion.py,sha256=5VEa6OE9shsF0EafJXzBLpFXhFIkqevJi9RRD-eUvZA,3718
-code_puppy/command_line/prompt_toolkit_completion.py,sha256=QtXqqEzc6C8ODHTfvIDD1sY0fLaX40S7Sfbol43A9i8,5338
+code_puppy/command_line/prompt_toolkit_completion.py,sha256=R-ykRmhVTieE4zIQnn7H1GDKk32yvalot0rZpFizLAQ,7358
 code_puppy/command_line/utils.py,sha256=L1PnV9tNupEW1zeziyb5aGAq8DYP8sMiuQbFYLO5Nus,1236
 code_puppy/tools/__init__.py,sha256=48BVpMt0HAMtz8G_z9SQhX6LnRqR83_AVfMQMf7bY0g,557
 code_puppy/tools/code_map.py,sha256=BghDHaebhGDfDGvA34gwO_5r92Py4O0Q3J4RV-WtnWs,3155
@@ -20,9 +20,9 @@ code_puppy/tools/common.py,sha256=dbmyZTrTBQh_0WWpaYN6jEync62W2mMrzNS8UFK0co4,14
 code_puppy/tools/file_modifications.py,sha256=nT87uAoY14RTAapnFCgkLTmW9g9P9bymxts2MpSpoo0,19297
 code_puppy/tools/file_operations.py,sha256=LJU_1b3WCXTAHa2B5VAbckrn1VVWb-HhcI3TF3BxYWs,11625
 code_puppy/tools/web_search.py,sha256=HhcwX0MMvMDPFO8gr8gzgesD5wPXOypjkxyLZeNwL5g,589
-code_puppy-0.0.50.data/data/code_puppy/models.json,sha256=7H-y97YK9BXhag5wJU19rtg24JtZWYx60RsBLBW3WiI,2162
-code_puppy-0.0.50.dist-info/METADATA,sha256=LRlwtx444fmT-6hL6p99vIYRKb8zIL30CaHG5k3lSRw,4716
-code_puppy-0.0.50.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-code_puppy-0.0.50.dist-info/entry_points.txt,sha256=d8YkBvIUxF-dHNJAj-x4fPEqizbY5d_TwvYpc01U5kw,58
-code_puppy-0.0.50.dist-info/licenses/LICENSE,sha256=31u8x0SPgdOq3izJX41kgFazWsM43zPEF9eskzqbJMY,1075
-code_puppy-0.0.50.dist-info/RECORD,,
+code_puppy-0.0.51.data/data/code_puppy/models.json,sha256=7H-y97YK9BXhag5wJU19rtg24JtZWYx60RsBLBW3WiI,2162
+code_puppy-0.0.51.dist-info/METADATA,sha256=sobI_laF57EIXLW_rLfU-E3ibYWAKLsRukigD5I6yMM,4716
+code_puppy-0.0.51.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+code_puppy-0.0.51.dist-info/entry_points.txt,sha256=d8YkBvIUxF-dHNJAj-x4fPEqizbY5d_TwvYpc01U5kw,58
+code_puppy-0.0.51.dist-info/licenses/LICENSE,sha256=31u8x0SPgdOq3izJX41kgFazWsM43zPEF9eskzqbJMY,1075
+code_puppy-0.0.51.dist-info/RECORD,,