code-puppy 0.0.70__tar.gz → 0.0.72__tar.gz

{code_puppy-0.0.70 → code_puppy-0.0.72}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: code-puppy
-Version: 0.0.70
+Version: 0.0.72
 Summary: Code generation agent
 Author: Michael Pfaffenberger
 License: MIT
@@ -20,7 +20,7 @@ Requires-Dist: json-repair>=0.46.2
 Requires-Dist: logfire>=0.7.1
 Requires-Dist: pathspec>=0.11.0
 Requires-Dist: prompt-toolkit>=3.0.38
-Requires-Dist: pydantic-ai>=0.1.0
+Requires-Dist: pydantic-ai>=0.3.2
 Requires-Dist: pydantic>=2.4.0
 Requires-Dist: pytest-cov>=6.1.1
 Requires-Dist: python-dotenv>=1.0.0
@@ -144,5 +144,45 @@ For instance, if you want to ensure that your application follows a specific des
 Ensure that all components follow these color schemes to promote consistency in design.
 ```
 
+## 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!
+
+### 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.
+
+### Configuration
+Create a config file at `~/.code_puppy/mcp_servers.json`. Here’s an example that connects to a local Context7 MCP server:
+
+```json
+{
+  "mcp_servers": {
+     "context7": { 
+        "url": "https://mcp.context7.com/sse"
+     }
+  }
+}
+```
+
+You can list multiple objects (one per server).
+
+### How to Use
+- Drop the config file in `~/.code_puppy/mcp_servers.json`.
+- Start your MCP (like context7, or anything compatible).
+- Run Code Puppy as usual. It’ll discover and use all configured MCP servers.
+
+#### Example usage
+```bash
+code-puppy --interactive
+# Then ask: Use context7 to look up FastAPI docs!
+```
+
+That’s it!
+If you need to run more exotic setups or connect to remote MCPs, just update your `mcp_servers.json` accordingly.
+
+**NOTE:** Want to add your own server or tool? Just follow the config pattern above—no code changes needed!
+
+---
+
 ## Conclusion
 By using Code Puppy, you can maintain code quality and adhere to design guidelines with ease.

{code_puppy-0.0.70 → code_puppy-0.0.72}/README.md

@@ -111,5 +111,45 @@ For instance, if you want to ensure that your application follows a specific des
 Ensure that all components follow these color schemes to promote consistency in design.
 ```
 
+## 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!
+
+### 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.
+
+### Configuration
+Create a config file at `~/.code_puppy/mcp_servers.json`. Here’s an example that connects to a local Context7 MCP server:
+
+```json
+{
+  "mcp_servers": {
+     "context7": { 
+        "url": "https://mcp.context7.com/sse"
+     }
+  }
+}
+```
+
+You can list multiple objects (one per server).
+
+### How to Use
+- Drop the config file in `~/.code_puppy/mcp_servers.json`.
+- Start your MCP (like context7, or anything compatible).
+- Run Code Puppy as usual. It’ll discover and use all configured MCP servers.
+
+#### Example usage
+```bash
+code-puppy --interactive
+# Then ask: Use context7 to look up FastAPI docs!
+```
+
+That’s it!
+If you need to run more exotic setups or connect to remote MCPs, just update your `mcp_servers.json` accordingly.
+
+**NOTE:** Want to add your own server or tool? Just follow the config pattern above—no code changes needed!
+
+---
+
 ## Conclusion
 By using Code Puppy, you can maintain code quality and adhere to design guidelines with ease.

{code_puppy-0.0.70 → code_puppy-0.0.72}/code_puppy/agent.py

@@ -3,6 +3,7 @@ from pathlib import Path
 
 import pydantic
 from pydantic_ai import Agent
+from pydantic_ai.mcp import MCPServerSSE
 
 from code_puppy.agent_prompts import get_system_prompt
 from code_puppy.model_factory import ModelFactory
@@ -64,11 +65,27 @@ def session_memory():
     return _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():
+        url = conf.get("url")
+        if url:
+            console.print(f"Registering MCP Server - {url}")
+            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
     from code_puppy.config import get_model_name
 
+    model_name = get_model_name()
+    console.print(f"[bold cyan]Loading Model: {model_name}")
+    global _code_generation_agent, _LAST_MODEL_NAME
+    from code_puppy.config import get_model_name
+
     model_name = get_model_name()
     console.print(f"[bold cyan]Loading Model: {model_name}[/bold cyan]")
     models_path = (
@@ -80,11 +97,14 @@ def reload_code_generation_agent():
     instructions = get_system_prompt()
     if PUPPY_RULES:
         instructions += f"\n{PUPPY_RULES}"
+
+    mcp_servers = _load_mcp_servers()
     agent = Agent(
         model=model,
         instructions=instructions,
         output_type=AgentResponse,
         retries=3,
+        mcp_servers=mcp_servers
     )
     register_all_tools(agent)
     _code_generation_agent = agent

code_puppy-0.0.72/code_puppy/command_line/motd.py

@@ -0,0 +1,57 @@
+"""
+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 = """
+June 21th, 2025 - 🚀 Woof-tastic news! Code Puppy now supports **MCP (Model Context Protocol) servers** for EXTREME PUPPY POWER!!!!.
+
+You can now connect plugins like doc search, Context7 integration, and more by simply dropping their info in your `~/.code_puppy/mcp_servers.json`. I’ll bark at remote docs or wrangle code tools for you—no extra fetches needed.
+
+Setup is easy:
+1. Add your MCP config to `~/.code_puppy/mcp_servers.json`.
+2. Fire up something like Context7, or any MCP server you want.
+3. Ask me to search docs, analyze, and more.
+
+The following example will let code_puppy use Context7! 
+Example config (+ more details in the README): 
+
+{
+  "mcp_servers": {
+     "context7": { 
+        "url": "https://mcp.context7.com/sse"
+     }
+  }
+}
+
+I fetch docs and power-ups via those servers. If you break stuff, please file an issue—bonus treat for reproducible bugs! 🦴
+
+This message-of-the-day won’t bug you again unless you run ~motd. Stay fluffy!
+
+"""
+MOTD_TRACK_FILE = os.path.expanduser("~/.puppy_cfg/motd.txt")
+
+
+def has_seen_motd(version: str) -> bool:
+    if not os.path.exists(MOTD_TRACK_FILE):
+        return False
+    with open(MOTD_TRACK_FILE, "r") as f:
+        seen_versions = {line.strip() for line in f if line.strip()}
+    return version in seen_versions
+
+
+def mark_motd_seen(version: str):
+    os.makedirs(os.path.dirname(MOTD_TRACK_FILE), exist_ok=True)
+    with open(MOTD_TRACK_FILE, "a") as f:
+        f.write(f"{version}\n")
+
+
+def print_motd(console, force: bool = False) -> bool:
+    if force or not has_seen_motd(MOTD_VERSION):
+        console.print(MOTD_MESSAGE)
+        mark_motd_seen(MOTD_VERSION)
+        return True
+    return False

{code_puppy-0.0.70 → code_puppy-0.0.72}/code_puppy/config.py

@@ -1,8 +1,11 @@
 import configparser
 import os
+import json
+import pathlib
 
 CONFIG_DIR = os.path.join(os.path.expanduser("~"), ".code_puppy")
 CONFIG_FILE = os.path.join(CONFIG_DIR, "puppy.cfg")
+MCP_SERVERS_FILE = os.path.join(CONFIG_DIR, "mcp_servers.json")
 
 DEFAULT_SECTION = "puppy"
 REQUIRED_KEYS = ["puppy_name", "owner_name"]
@@ -97,6 +100,23 @@ def set_config_value(key: str, value: str):
 
 
 # --- MODEL STICKY EXTENSION STARTS HERE ---
+def load_mcp_server_configs():
+    """
+    Loads the MCP server configurations from ~/.code_puppy/mcp_servers.json.
+    Returns a dict mapping names to their URL or config dict.
+    If file does not exist, returns an empty dict.
+    """
+    try:
+        if not pathlib.Path(MCP_SERVERS_FILE).exists():
+            print("No MCP configuration was found")
+            return {}
+        with open(MCP_SERVERS_FILE, "r") as f:
+            conf = json.loads(f.read())
+            return conf["mcp_servers"]
+    except Exception as e:
+        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-0.0.70 → code_puppy-0.0.72}/code_puppy/main.py

@@ -67,7 +67,8 @@ async def main():
         try:
             while not shutdown_flag:
                 agent = get_code_generation_agent()
-                response = await agent.run(command)
+                async with agent.run_mcp_servers():
+                    response = await agent.run(command)
                 agent_response = response.output
                 console.print(agent_response.output_message)
                 # Log to session memory
@@ -210,7 +211,8 @@ async def interactive_mode(history_file_path: str) -> None:
                 agent_response = None
 
                 agent = get_code_generation_agent()
-                result = await agent.run(task, message_history=message_history)
+                async with agent.run_mcp_servers():
+                    result = await agent.run(task, message_history=message_history)
                 # Get the structured response
                 agent_response = result.output
                 console.print(agent_response.output_message)

{code_puppy-0.0.70 → code_puppy-0.0.72}/pyproject.toml

@@ -4,12 +4,12 @@ build-backend = "hatchling.build"
 
 [project]
 name = "code-puppy"
-version = "0.0.70"
+version = "0.0.72"
 description = "Code generation agent"
 readme = "README.md"
 requires-python = ">=3.10"
 dependencies = [
-    "pydantic-ai>=0.1.0",
+    "pydantic-ai>=0.3.2",
     "httpx>=0.24.1",
     "rich>=13.4.2",
     "logfire>=0.7.1",

code_puppy-0.0.70/code_puppy/command_line/motd.py

@@ -1,45 +0,0 @@
-"""
-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 = "20240601"
-MOTD_MESSAGE = """
-June 14th, 2025 - Wow... code_puppy has been downloaded 10s of thousands of times. 
-
-This new update has a bug fix where message truncation would sometimes cause tool-calls and tool-replies
-to become isolated, and an exception would be raied creating a situation only recoverable by restarting
-code-puppy or using the `clear` command to get rid of all message history. 
-
-Thankfully that is fixed. Message truncation max-length is configurable with the following command:
-`~set message_history_length 25` if you want to truncate to 25 messages. The default is 40. 
-
-This message-of-the-day will not appear again unless you run ~motd.
-
-Please open issues on GitHub if you find any bugs! Cheers!
-"""
-MOTD_TRACK_FILE = os.path.expanduser("~/.puppy_cfg/motd.txt")
-
-
-def has_seen_motd(version: str) -> bool:
-    if not os.path.exists(MOTD_TRACK_FILE):
-        return False
-    with open(MOTD_TRACK_FILE, "r") as f:
-        seen_versions = {line.strip() for line in f if line.strip()}
-    return version in seen_versions
-
-
-def mark_motd_seen(version: str):
-    os.makedirs(os.path.dirname(MOTD_TRACK_FILE), exist_ok=True)
-    with open(MOTD_TRACK_FILE, "a") as f:
-        f.write(f"{version}\n")
-
-
-def print_motd(console, force: bool = False) -> bool:
-    if force or not has_seen_motd(MOTD_VERSION):
-        console.print(MOTD_MESSAGE)
-        mark_motd_seen(MOTD_VERSION)
-        return True
-    return False