code-puppy 0.0.31__tar.gz → 0.0.33__tar.gz

{code_puppy-0.0.31 → code_puppy-0.0.33}/PKG-INFO

@@ -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 → code_puppy-0.0.33}/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-0.0.33/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-0.0.31 → code_puppy-0.0.33}/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 → code_puppy-0.0.33}/pyproject.toml

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