code-puppy 0.0.72__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.72.dist-info → code_puppy-0.0.74.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: code-puppy
-Version: 0.0.72
+Version: 0.0.74
 Summary: Code generation agent
 Author: Michael Pfaffenberger
 License: MIT
@@ -146,7 +146,7 @@ Ensure that all components follow these color schemes to promote consistency in
 
 ## Using MCP Servers for External Tools
 
-Code Puppy supports **MCP (Model Context Protocol) servers** to give you access to external code tools and advanced features like code search, documentation lookups, and more—including Context7 integration for deep docs and search!
+Code Puppy supports **MCP (Model Context Protocol) servers** to give you access to external code tools and advanced features like code search, documentation lookups, and more—including Context7 (https://context7.com/) integration for deep docs and search!
 
 ### What is an MCP Server?
 An MCP server is a standalone process (can be local or remote) that offers specialized functionality (plugins, doc search, code analysis, etc.). Code Puppy can connect to one or more MCP servers at startup, unlocking these extra commands inside your coding agent.

{code_puppy-0.0.72.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.72.data/data/code_puppy/models.json,sha256=AY1dcFd3bMFaJBjU_u-PjsrnmcKbjjtGhjL3uYfqpW8,1692
-code_puppy-0.0.72.dist-info/METADATA,sha256=qgbVd3QO10NGEMWmSQJJb6VbQeqgQ4KMIGadboT5S28,6393
-code_puppy-0.0.72.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-code_puppy-0.0.72.dist-info/entry_points.txt,sha256=d8YkBvIUxF-dHNJAj-x4fPEqizbY5d_TwvYpc01U5kw,58
-code_puppy-0.0.72.dist-info/licenses/LICENSE,sha256=31u8x0SPgdOq3izJX41kgFazWsM43zPEF9eskzqbJMY,1075
-code_puppy-0.0.72.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,,