code-puppy 0.0.73__py3-none-any.whl → 0.0.74__py3-none-any.whl

code_puppy/agent.py

@@ -67,6 +67,7 @@ def session_memory():
 
 def _load_mcp_servers():
     from code_puppy.config import load_mcp_server_configs
+
     configs = load_mcp_server_configs()
     servers = []
     for name, conf in configs.items():
@@ -76,6 +77,7 @@ def _load_mcp_servers():
             servers.append(MCPServerSSE(url))
     return servers
 
+
 def reload_code_generation_agent():
     """Force-reload the agent, usually after a model change."""
     global _code_generation_agent, _LAST_MODEL_NAME
@@ -104,7 +106,7 @@ def reload_code_generation_agent():
         instructions=instructions,
         output_type=AgentResponse,
         retries=3,
-        mcp_servers=mcp_servers
+        mcp_servers=mcp_servers,
     )
     register_all_tools(agent)
     _code_generation_agent = agent

code_puppy/command_line/meta_command_handler.py

@@ -8,6 +8,7 @@ from code_puppy.command_line.model_picker_completion import (
 )
 from code_puppy.config import get_config_keys
 from code_puppy.command_line.utils import make_directory_table
+from code_puppy.command_line.motd import print_motd
 
 META_COMMANDS_HELP = """
 [bold magenta]Meta Commands Help[/bold magenta]
@@ -22,8 +23,6 @@ META_COMMANDS_HELP = """
 """
 
 
-from code_puppy.command_line.motd import print_motd
-
 def handle_meta_command(command: str, console: Console) -> bool:
     """
     Handle meta/config commands prefixed with '~'.
@@ -75,20 +74,26 @@ def handle_meta_command(command: str, console: Console) -> bool:
 
     if command.strip().startswith("~show"):
         from code_puppy.command_line.model_picker_completion import get_active_model
-        from code_puppy.config import get_owner_name, get_puppy_name, get_yolo_mode, get_message_history_limit
+        from code_puppy.config import (
+            get_owner_name,
+            get_puppy_name,
+            get_yolo_mode,
+            get_message_history_limit,
+        )
+
         puppy_name = get_puppy_name()
         owner_name = get_owner_name()
         model = get_active_model()
         yolo_mode = get_yolo_mode()
         msg_limit = get_message_history_limit()
-        console.print(f'''[bold magenta]🐶 Puppy Status[/bold magenta]
+        console.print(f"""[bold magenta]🐶 Puppy Status[/bold magenta]
 
 [bold]puppy_name:[/bold]     [cyan]{puppy_name}[/cyan]
 [bold]owner_name:[/bold]     [cyan]{owner_name}[/cyan]
 [bold]model:[/bold]          [green]{model}[/green]
-[bold]YOLO_MODE:[/bold]      {'[red]ON[/red]' if yolo_mode else '[yellow]off[/yellow]'}
+[bold]YOLO_MODE:[/bold]      {"[red]ON[/red]" if yolo_mode else "[yellow]off[/yellow]"}
 [bold]message_history_limit:[/bold]   Keeping last [cyan]{msg_limit}[/cyan] messages in context
-''')
+""")
         return True
 
     if command.startswith("~set"):

code_puppy/command_line/motd.py

@@ -2,8 +2,8 @@
 MOTD (Message of the Day) feature for code-puppy.
 Stores seen versions in ~/.puppy_cfg/motd.txt.
 """
+
 import os
-from typing import Optional
 
 MOTD_VERSION = "20240621"
 MOTD_MESSAGE = """

code_puppy/config.py

@@ -72,6 +72,7 @@ def get_message_history_limit():
     except (ValueError, TypeError):
         return 40
 
+
 # --- CONFIG SETTER STARTS HERE ---
 def get_config_keys():
     """
@@ -117,6 +118,7 @@ def load_mcp_server_configs():
         print(f"Failed to load MCP servers - {str(e)}")
         return {}
 
+
 def get_model_name():
     """Returns the last used model name stored in config, or None if unset."""
     return get_value("model") or "gpt-4.1"

code_puppy/main.py

@@ -4,7 +4,6 @@ import os
 import sys
 
 from dotenv import load_dotenv
-from pydantic_ai.messages import ToolCallPart, ToolReturnPart
 from rich.console import Console, ConsoleOptions, RenderResult
 from rich.markdown import CodeBlock, Markdown
 from rich.syntax import Syntax
@@ -116,9 +115,10 @@ async def interactive_mode(history_file_path: str) -> None:
     # Show MOTD if user hasn't seen it after an update
     try:
         from code_puppy.command_line.motd import print_motd
+
         print_motd(console, force=False)
     except Exception as e:
-        console.print(f'[yellow]MOTD error: {e}[/yellow]')
+        console.print(f"[yellow]MOTD error: {e}[/yellow]")
 
     # Check if prompt_toolkit is installed
     try:
@@ -235,19 +235,21 @@ async def interactive_mode(history_file_path: str) -> None:
                 ]
                 # 2. Append to existing history and keep only the most recent set by config
                 from code_puppy.config import get_message_history_limit
+
                 message_history.extend(filtered)
 
                 # --- BEGIN GROUP-AWARE TRUNCATION LOGIC ---
                 limit = get_message_history_limit()
                 if len(message_history) > limit:
+
                     def group_by_tool_call_id(msgs):
                         grouped = {}
                         no_group = []
                         for m in msgs:
                             # Find all tool_call_id in message parts
                             tool_call_ids = set()
-                            for part in getattr(m, 'parts', []):
-                                if hasattr(part, 'tool_call_id') and part.tool_call_id:
+                            for part in getattr(m, "parts", []):
+                                if hasattr(part, "tool_call_id") and part.tool_call_id:
                                     tool_call_ids.add(part.tool_call_id)
                             if tool_call_ids:
                                 for tcid in tool_call_ids:
@@ -271,7 +273,6 @@ async def interactive_mode(history_file_path: str) -> None:
                     message_history = truncated
                 # --- END GROUP-AWARE TRUNCATION LOGIC ---
 
-
                 if agent_response and agent_response.awaiting_user_input:
                     console.print(
                         "\n[bold yellow]\u26a0 Agent needs your input to continue.[/bold yellow]"

code_puppy/tools/file_modifications.py

@@ -110,7 +110,7 @@ def _replace_in_file(
         orig_lines = modified.splitlines()
         loc, score = _find_best_window(orig_lines, old_snippet)
 
-        if score < 0.95 or loc == None:
+        if score < 0.95 or loc is None:
             return {
                 "error": "No suitable match in file (JW < 0.95)",
                 "jw_score": score,

code_puppy/tools/ts_code_map.py

@@ -392,7 +392,11 @@ def _walk_fix(ts_node, rich_parent, info):
         n_type = child.type
         if n_type in nodes_cfg:
             style = nodes_cfg[n_type].keywords["style"]
-            ident = child.child_by_field_name(name_field) if name_field else _first_identifier(child)
+            ident = (
+                child.child_by_field_name(name_field)
+                if name_field
+                else _first_identifier(child)
+            )
             label_text = ident.text.decode() if ident else "<anon>"
             label = nodes_cfg[n_type].func(label_text)
             emoji = _emoji_for_node_type(n_type)
@@ -402,6 +406,8 @@ def _walk_fix(ts_node, rich_parent, info):
             _walk_fix(child, branch, info)
         else:
             _walk_fix(child, rich_parent, info)
+
+
 # ----------------------------------------------------------------------
 
 
@@ -465,7 +471,9 @@ def make_code_map(directory: str, ignore_tests: bool = True) -> str:
     base_tree = RichTree(Text(Path(directory).name, style="bold magenta"))
 
     # Cache to ensure we reuse RichTree nodes per directory path
-    dir_nodes: dict[str, RichTree] = {Path(directory).resolve(): base_tree}  # key=abs path
+    dir_nodes: dict[str, RichTree] = {
+        Path(directory).resolve(): base_tree
+    }  # key=abs path
 
     for root, dirs, files in os.walk(directory):
         # ignore dot-folders early

{code_puppy-0.0.73.dist-info → code_puppy-0.0.74.dist-info}/METADATA

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

{code_puppy-0.0.73.dist-info → code_puppy-0.0.74.dist-info}/RECORD

@@ -1,29 +1,29 @@
 code_puppy/__init__.py,sha256=-ANvE6Xe5NlWDIRCIfL1x-rgtCZ6zM2Ye9NphFoULSY,82
-code_puppy/agent.py,sha256=hi064QRtDwMfDSsfW3p31SX4rdsMCRgJMftOK2Nv7_M,4081
+code_puppy/agent.py,sha256=hhVrBhT3MAhzs61cSnZ7Au7lrjoFAewfh6fGkawEQQo,4084
 code_puppy/agent_prompts.py,sha256=DnPNubiFtD6Ot76cIsxedJEaEVYS5AaxR3JWJM5nXNg,6925
-code_puppy/config.py,sha256=3xqEIysX-fV_M3oPDtnYuiH8tMGTZzhHpwzCzx8ghAQ,5011
-code_puppy/main.py,sha256=FYqZhne46DCOwC7GdFbfx0reWqRQRI4_JOsq0rHNygo,12538
+code_puppy/config.py,sha256=r5nw5ChOP8xd_K5yo8U5OtO2gy2bFhARiyNtDp1JrwQ,5013
+code_puppy/main.py,sha256=-wC_Bs2vbu9r9kdajKcSjddQ1nHFnpifLUX6nhbATZw,12478
 code_puppy/model_factory.py,sha256=jZH96zE5GMTK3RWjUpZB4HjbSO3fSaAzjRoQh8Y8lfg,7711
 code_puppy/models.json,sha256=AY1dcFd3bMFaJBjU_u-PjsrnmcKbjjtGhjL3uYfqpW8,1692
 code_puppy/session_memory.py,sha256=4sgAAjbXdLSi8hETpd56tgtrG6hqMUuZWDlJOu6BQjA,2735
 code_puppy/version_checker.py,sha256=aRGulzuY4C4CdFvU1rITduyL-1xTFsn4GiD1uSfOl_Y,396
 code_puppy/command_line/__init__.py,sha256=y7WeRemfYppk8KVbCGeAIiTuiOszIURCDjOMZv_YRmU,45
 code_puppy/command_line/file_path_completion.py,sha256=gw8NpIxa6GOpczUJRyh7VNZwoXKKn-yvCqit7h2y6Gg,2931
-code_puppy/command_line/meta_command_handler.py,sha256=kP57BU7gH23ten0CHX1XmrCbe89WFxoHzz6ddeRGrCE,6121
+code_puppy/command_line/meta_command_handler.py,sha256=L7qP2g0Faz0V7bMH4YK3s03OWWuQFtK7Sh-Kt2zmmEQ,6182
 code_puppy/command_line/model_picker_completion.py,sha256=NkyZZG7IhcVWSJ3ADytwCA5f8DpNeVs759Qtqs4fQtY,3733
-code_puppy/command_line/motd.py,sha256=TDk5xTe6Atp6tyTFEH3SgK8HWBqsdiffVkhxaoGs-lg,1873
+code_puppy/command_line/motd.py,sha256=3HuwCMuu0_hZa7UCGnOY5EcG103y2jMFWo28NwY-GDQ,1846
 code_puppy/command_line/prompt_toolkit_completion.py,sha256=_gP0FIOgHDNHTTWLNL0XNzr6sO0ISe7Mec1uQNo9kcM,8337
 code_puppy/command_line/utils.py,sha256=7eyxDHjPjPB9wGDJQQcXV_zOsGdYsFgI0SGCetVmTqE,1251
 code_puppy/tools/__init__.py,sha256=J_HCbkivdr1rP5vfucxttXhGmTBx0S2LNoDMrbaE-Fc,558
 code_puppy/tools/command_runner.py,sha256=cEphrDyr12CoqpKizzaxhMHhG1HiVcqXArTkwc6lvxU,6760
 code_puppy/tools/common.py,sha256=M53zhiXZAmPdvi1Y_bzCxgvEmifOvRRJvYPARYRZqHw,2253
-code_puppy/tools/file_modifications.py,sha256=d0fC2CCnkmv0kEJ8kvKTuUMTww3S_mgF06h9g93212A,13072
+code_puppy/tools/file_modifications.py,sha256=uEUXoB0F4h3Blzgeq4ggejv1GKAc8OdeTvntf58gsrQ,13072
 code_puppy/tools/file_operations.py,sha256=xqj7MiK6sGTaC8NMrMRTybWlXG32S-DjybqJtnufRcI,11715
-code_puppy/tools/ts_code_map.py,sha256=-CXLzFxFMefoYYdWxfPwFugZhkfVxdiftGb3LZnEttk,17567
+code_puppy/tools/ts_code_map.py,sha256=o-u8p5vsYwitfDtVEoPS-7MwWn2xHzwtIQLo1_WMhQs,17647
 code_puppy/tools/web_search.py,sha256=sA2ierjuuYA517-uhb5s53SgeVsyOe1nExoZsrU1Fps,1284
-code_puppy-0.0.73.data/data/code_puppy/models.json,sha256=AY1dcFd3bMFaJBjU_u-PjsrnmcKbjjtGhjL3uYfqpW8,1692
-code_puppy-0.0.73.dist-info/METADATA,sha256=zn6uo0kWwpb4-2hi4hhSHJ2QU5mSjym6RyZc9dgnNrI,6417
-code_puppy-0.0.73.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-code_puppy-0.0.73.dist-info/entry_points.txt,sha256=d8YkBvIUxF-dHNJAj-x4fPEqizbY5d_TwvYpc01U5kw,58
-code_puppy-0.0.73.dist-info/licenses/LICENSE,sha256=31u8x0SPgdOq3izJX41kgFazWsM43zPEF9eskzqbJMY,1075
-code_puppy-0.0.73.dist-info/RECORD,,
+code_puppy-0.0.74.data/data/code_puppy/models.json,sha256=AY1dcFd3bMFaJBjU_u-PjsrnmcKbjjtGhjL3uYfqpW8,1692
+code_puppy-0.0.74.dist-info/METADATA,sha256=biL4TR0F1r-BMFHSjpTLvKxNdtYJ3zXhh5_3HQXVzYQ,6417
+code_puppy-0.0.74.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+code_puppy-0.0.74.dist-info/entry_points.txt,sha256=d8YkBvIUxF-dHNJAj-x4fPEqizbY5d_TwvYpc01U5kw,58
+code_puppy-0.0.74.dist-info/licenses/LICENSE,sha256=31u8x0SPgdOq3izJX41kgFazWsM43zPEF9eskzqbJMY,1075
+code_puppy-0.0.74.dist-info/RECORD,,