code-puppy 0.0.31__py3-none-any.whl → 0.0.33__py3-none-any.whl

code_puppy/command_line/prompt_toolkit_completion.py

@@ -1,5 +1,6 @@
 import os
 from code_puppy.command_line.utils import list_directory
+from code_puppy.config import get_puppy_name
 # 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.
@@ -59,14 +60,14 @@ from prompt_toolkit.formatted_text import FormattedText
 def get_prompt_with_active_model(base: str = '>>> '):
     model = get_active_model() or '(default)'
     cwd = os.getcwd()
-    # Abbreviate the home directory to ~ for brevity in the prompt
     home = os.path.expanduser('~')
     if cwd.startswith(home):
         cwd_display = '~' + cwd[len(home):]
     else:
         cwd_display = cwd
+    puppy_name = get_puppy_name()
     return FormattedText([
-        ('bold', '🐶'),
+        ('bold', f'🐶 {puppy_name} '),
         ('class:model', f'[' + str(model) + '] '),
         ('class:cwd', f'(' + str(cwd_display) + ') '),
         ('class:arrow', str(base)),

code_puppy/config.py

@@ -0,0 +1,53 @@
+import os
+import configparser
+
+CONFIG_DIR = os.path.join(os.path.expanduser("~"), ".code_puppy")
+CONFIG_FILE = os.path.join(CONFIG_DIR, "puppy.cfg")
+
+DEFAULT_SECTION = "puppy"
+REQUIRED_KEYS = ["puppy_name", "owner_name"]
+
+
+def ensure_config_exists():
+    """
+    Ensure that the .code_puppy dir and puppy.cfg exist, prompting if needed.
+    Returns configparser.ConfigParser for reading.
+    """
+    if not os.path.exists(CONFIG_DIR):
+        os.makedirs(CONFIG_DIR, exist_ok=True)
+    exists = os.path.isfile(CONFIG_FILE)
+    config = configparser.ConfigParser()
+    if exists:
+        config.read(CONFIG_FILE)
+    missing = []
+    if DEFAULT_SECTION not in config:
+        config[DEFAULT_SECTION] = {}
+    for key in REQUIRED_KEYS:
+        if not config[DEFAULT_SECTION].get(key):
+            missing.append(key)
+    if missing:
+        print("🐾 Let's get your Puppy ready!")
+        for key in missing:
+            if key == "puppy_name":
+                val = input("What should we name the puppy? ").strip()
+            elif key == "owner_name":
+                val = input("What's your name (so Code Puppy knows its master)? ").strip()
+            else:
+                val = input(f"Enter {key}: ").strip()
+            config[DEFAULT_SECTION][key] = val
+        with open(CONFIG_FILE, "w") as f:
+            config.write(f)
+    return config
+
+def get_value(key: str):
+    config = configparser.ConfigParser()
+    config.read(CONFIG_FILE)
+    val = config.get(DEFAULT_SECTION, key, fallback=None)
+    return val
+
+
+def get_puppy_name():
+    return get_value("puppy_name") or "Puppy"
+
+def get_owner_name():
+    return get_value("owner_name") or "Master"

code_puppy/main.py

@@ -5,6 +5,7 @@ from code_puppy.version_checker import fetch_latest_version
 from code_puppy import __version__
 import sys
 from dotenv import load_dotenv
+from code_puppy.config import ensure_config_exists
 from rich.console import Console
 from rich.markdown import Markdown
 from rich.console import ConsoleOptions, RenderResult
@@ -32,6 +33,8 @@ def get_secret_file_path():
 
 
 async def main():
+    # Ensure the config directory and puppy.cfg with name info exist (prompt user if needed)
+    ensure_config_exists()
     current_version = __version__
     latest_version = fetch_latest_version('code-puppy')
     console.print(f'Current version: {current_version}')
@@ -223,7 +226,7 @@ async def interactive_mode(history_file_path: str) -> None:
                 )
 
             except Exception:
-                console.print_exception(show_locals=True)
+                console.print_exception()
 
 
 def prettier_code_blocks():

{code_puppy-0.0.31.dist-info → code_puppy-0.0.33.dist-info}/METADATA

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

{code_puppy-0.0.31.dist-info → code_puppy-0.0.33.dist-info}/RECORD

@@ -1,7 +1,8 @@
 code_puppy/__init__.py,sha256=orgffM-uGp8g1XCqXGKWaFB4tCCz8TVgsLMPKCOGNx0,81
 code_puppy/agent.py,sha256=2rPSpYuj3ouLi3IlKxXM3WPZ06rhhcoVQLKQdtYqqLs,3144
 code_puppy/agent_prompts.py,sha256=6bM_3Dxv9QAxoDBTFd3jSIJKB6WwAaYcW5NH21N2cx0,6095
-code_puppy/main.py,sha256=whW4CITLOMUsI63Bx16zH1jm7Sy7VDWXwN6wxXOchkI,9897
+code_puppy/config.py,sha256=DvmctmrHsOmjiXaiwT-4vAUiTZ4Cib-imyjKewI70eU,1661
+code_puppy/main.py,sha256=bc27bk6rFge95H2BumTTzRLtOx43z5FnsmjIjQx_RpU,10052
 code_puppy/model_factory.py,sha256=IAiOBWxL2LC8PZ3xzesRvKsLAOuZBgWzn5Mh_gT8rRM,8466
 code_puppy/models.json,sha256=Bt4rkn2tHR4UA8kanovpxtW0_-T5TooKrtUpzJNipmo,2212
 code_puppy/session_memory.py,sha256=vuLSw1Pfa-MXD4lD8hj2qt65OR_aL2WdoMuF6Jwnc1k,2597
@@ -10,7 +11,7 @@ code_puppy/command_line/__init__.py,sha256=y7WeRemfYppk8KVbCGeAIiTuiOszIURCDjOMZ
 code_puppy/command_line/file_path_completion.py,sha256=HAlOu9XVYgJ7FbjdrhKBL0rFmCVFxSGGewdcfKTUsPw,2865
 code_puppy/command_line/meta_command_handler.py,sha256=9lg2-UFKin8M67mdleECglWq30sFe61hHhaVsxBpnas,3326
 code_puppy/command_line/model_picker_completion.py,sha256=5D2HkrMCtee7Eo6augls4KU6bh7tu7xR2zH5triRuZg,3658
-code_puppy/command_line/prompt_toolkit_completion.py,sha256=BUOpkOJ3GlYjuArReCiPmlZS4XAo95gH6e8-2foNQZQ,4640
+code_puppy/command_line/prompt_toolkit_completion.py,sha256=KpPRl7t5RaD5zf4xC2WgY_rkIhD3LS3Yt4lrXUvFL44,4667
 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
@@ -19,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.31.data/data/code_puppy/models.json,sha256=Bt4rkn2tHR4UA8kanovpxtW0_-T5TooKrtUpzJNipmo,2212
-code_puppy-0.0.31.dist-info/METADATA,sha256=Pc7YI0lRchu0XDU-F6tG1bG_Xn0HU7yZJC2ssHaYtTY,4716
-code_puppy-0.0.31.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-code_puppy-0.0.31.dist-info/entry_points.txt,sha256=d8YkBvIUxF-dHNJAj-x4fPEqizbY5d_TwvYpc01U5kw,58
-code_puppy-0.0.31.dist-info/licenses/LICENSE,sha256=31u8x0SPgdOq3izJX41kgFazWsM43zPEF9eskzqbJMY,1075
-code_puppy-0.0.31.dist-info/RECORD,,
+code_puppy-0.0.33.data/data/code_puppy/models.json,sha256=Bt4rkn2tHR4UA8kanovpxtW0_-T5TooKrtUpzJNipmo,2212
+code_puppy-0.0.33.dist-info/METADATA,sha256=pvyumUtOyXpgMNnlTjOar9GNzQjkPJ7ZFOin3INUq6w,4716
+code_puppy-0.0.33.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+code_puppy-0.0.33.dist-info/entry_points.txt,sha256=d8YkBvIUxF-dHNJAj-x4fPEqizbY5d_TwvYpc01U5kw,58
+code_puppy-0.0.33.dist-info/licenses/LICENSE,sha256=31u8x0SPgdOq3izJX41kgFazWsM43zPEF9eskzqbJMY,1075
+code_puppy-0.0.33.dist-info/RECORD,,