code-puppy 0.0.176__py3-none-any.whl → 0.0.178__py3-none-any.whl

code_puppy/round_robin_model.py

@@ -5,14 +5,12 @@ from typing import Any, AsyncIterator, List
 from pydantic_ai.models import (
     Model,
     ModelMessage,
-    ModelSettings,
     ModelRequestParameters,
     ModelResponse,
+    ModelSettings,
     StreamedResponse,
 )
-from pydantic_ai.models.fallback import (
-    merge_model_settings,
-)
+from pydantic_ai.models.fallback import merge_model_settings
 from pydantic_ai.result import RunContext
 
 try:

code_puppy/tools/agent_tools.py

@@ -1,21 +1,21 @@
 # agent_tools.py
 
 from typing import List
+
 from pydantic import BaseModel
-from pydantic_ai import RunContext
 
+# Import Agent from pydantic_ai to create temporary agents for invocation
+from pydantic_ai import Agent, RunContext
+
+from code_puppy.config import get_global_model_name
 from code_puppy.messaging import (
-    emit_info,
     emit_divider,
-    emit_system_message,
     emit_error,
+    emit_info,
+    emit_system_message,
 )
-from code_puppy.tools.common import generate_group_id
-
-# Import Agent from pydantic_ai to create temporary agents for invocation
-from pydantic_ai import Agent
 from code_puppy.model_factory import ModelFactory
-from code_puppy.config import get_global_model_name
+from code_puppy.tools.common import generate_group_id
 
 
 class AgentInfo(BaseModel):
@@ -65,6 +65,7 @@ def register_list_agents(agent):
 
         try:
             from code_puppy.agents import get_available_agents
+
             # Get available agents from the agent manager
             agents_dict = get_available_agents()
 
@@ -114,6 +115,7 @@ def register_invoke_agent(agent):
             AgentInvokeOutput: The agent's response to the prompt
         """
         from code_puppy.agents.agent_manager import load_agent
+
         # Generate a group ID for this tool execution
         group_id = generate_group_id("invoke_agent", agent_name)
 

code_puppy/tools/browser/browser_screenshot.py

@@ -13,10 +13,11 @@ from code_puppy.messaging import emit_error, emit_info
 from code_puppy.tools.common import generate_group_id
 
 from .camoufox_manager import get_camoufox_manager
-from .vqa_agent import VisualAnalysisResult, run_vqa_analysis
+from .vqa_agent import run_vqa_analysis
 
-
-_TEMP_SCREENSHOT_ROOT = Path(mkdtemp(prefix="code_puppy_screenshots_", dir=gettempdir()))
+_TEMP_SCREENSHOT_ROOT = Path(
+    mkdtemp(prefix="code_puppy_screenshots_", dir=gettempdir())
+)
 
 
 def _build_screenshot_path(timestamp: str) -> Path:

code_puppy/tools/browser/browser_scripts.py

@@ -236,9 +236,6 @@ async def wait_for_element(
         return {"success": False, "error": str(e), "selector": selector, "state": state}
 
 
-
-
-
 async def highlight_element(
     selector: str,
     color: str = "red",
@@ -437,9 +434,6 @@ def register_wait_for_element(agent):
         return await wait_for_element(selector, state, timeout)
 
 
-
-
-
 def register_browser_highlight_element(agent):
     """Register the element highlighting tool."""
 

code_puppy/tools/browser/browser_workflows.py

@@ -29,18 +29,18 @@ async def save_workflow(name: str, content: str) -> Dict[str, Any]:
         workflows_dir = get_workflows_directory()
 
         # Clean up the filename - remove spaces, special chars, etc.
-        safe_name = "".join(c for c in name if c.isalnum() or c in ('-', '_')).lower()
+        safe_name = "".join(c for c in name if c.isalnum() or c in ("-", "_")).lower()
         if not safe_name:
             safe_name = "workflow"
 
         # Ensure .md extension
-        if not safe_name.endswith('.md'):
-            safe_name += '.md'
+        if not safe_name.endswith(".md"):
+            safe_name += ".md"
 
         workflow_path = workflows_dir / safe_name
 
         # Write the workflow content
-        with open(workflow_path, 'w', encoding='utf-8') as f:
+        with open(workflow_path, "w", encoding="utf-8") as f:
             f.write(content)
 
         emit_info(
@@ -52,7 +52,7 @@ async def save_workflow(name: str, content: str) -> Dict[str, Any]:
             "success": True,
             "path": str(workflow_path),
             "name": safe_name,
-            "size": len(content)
+            "size": len(content),
         }
 
     except Exception as e:
@@ -75,23 +75,27 @@ async def list_workflows() -> Dict[str, Any]:
         workflows_dir = get_workflows_directory()
 
         # Find all .md files in the workflows directory
-        workflow_files = list(workflows_dir.glob('*.md'))
+        workflow_files = list(workflows_dir.glob("*.md"))
 
         workflows = []
         for workflow_file in workflow_files:
             try:
                 stat = workflow_file.stat()
-                workflows.append({
-                    "name": workflow_file.name,
-                    "path": str(workflow_file),
-                    "size": stat.st_size,
-                    "modified": stat.st_mtime
-                })
+                workflows.append(
+                    {
+                        "name": workflow_file.name,
+                        "path": str(workflow_file),
+                        "size": stat.st_size,
+                        "modified": stat.st_mtime,
+                    }
+                )
             except Exception as e:
-                emit_info(f"[yellow]Warning: Could not read {workflow_file}: {e}[/yellow]")
+                emit_info(
+                    f"[yellow]Warning: Could not read {workflow_file}: {e}[/yellow]"
+                )
 
         # Sort by modification time (newest first)
-        workflows.sort(key=lambda x: x['modified'], reverse=True)
+        workflows.sort(key=lambda x: x["modified"], reverse=True)
 
         emit_info(
             f"[green]✅ Found {len(workflows)} workflow(s)[/green]",
@@ -102,7 +106,7 @@ async def list_workflows() -> Dict[str, Any]:
             "success": True,
             "workflows": workflows,
             "count": len(workflows),
-            "directory": str(workflows_dir)
+            "directory": str(workflows_dir),
         }
 
     except Exception as e:
@@ -125,8 +129,8 @@ async def read_workflow(name: str) -> Dict[str, Any]:
         workflows_dir = get_workflows_directory()
 
         # Handle both with and without .md extension
-        if not name.endswith('.md'):
-            name += '.md'
+        if not name.endswith(".md"):
+            name += ".md"
 
         workflow_path = workflows_dir / name
 
@@ -135,10 +139,14 @@ async def read_workflow(name: str) -> Dict[str, Any]:
                 f"[red]❌ Workflow not found: {name}[/red]",
                 message_group=group_id,
             )
-            return {"success": False, "error": f"Workflow '{name}' not found", "name": name}
+            return {
+                "success": False,
+                "error": f"Workflow '{name}' not found",
+                "name": name,
+            }
 
         # Read the workflow content
-        with open(workflow_path, 'r', encoding='utf-8') as f:
+        with open(workflow_path, "r", encoding="utf-8") as f:
             content = f.read()
 
         emit_info(
@@ -151,7 +159,7 @@ async def read_workflow(name: str) -> Dict[str, Any]:
             "name": name,
             "content": content,
             "path": str(workflow_path),
-            "size": len(content)
+            "size": len(content),
         }
 
     except Exception as e:

code_puppy/tools/browser/camoufox_manager.py

@@ -3,13 +3,13 @@
 from typing import Optional
 
 import camoufox
+from camoufox.addons import DefaultAddons
+from camoufox.exceptions import CamoufoxNotInstalled, UnsupportedVersion
+from camoufox.locale import ALLOW_GEOIP, download_mmdb
+from camoufox.pkgman import CamoufoxFetcher, camoufox_path
 from playwright.async_api import Browser, BrowserContext, Page
 
 from code_puppy.messaging import emit_info
-from camoufox.pkgman import CamoufoxFetcher, camoufox_path
-from camoufox.locale import ALLOW_GEOIP, download_mmdb
-from camoufox.addons import DefaultAddons
-from camoufox.exceptions import CamoufoxNotInstalled, UnsupportedVersion
 
 
 class CamoufoxManager:
@@ -52,7 +52,7 @@ class CamoufoxManager:
 
         try:
             emit_info("[yellow]Initializing Camoufox (privacy Firefox)...[/yellow]")
-            
+
             # Ensure Camoufox binary and dependencies are fetched before launching
             await self._prefetch_camoufox()
 
@@ -62,7 +62,7 @@ class CamoufoxManager:
             )
             self._initialized = True
 
-        except Exception as e:
+        except Exception:
             await self._cleanup()
             raise
 
@@ -83,7 +83,6 @@ class CamoufoxManager:
         page = await self._context.new_page()
         await page.goto(self.homepage)
 
-
     async def get_current_page(self) -> Optional[Page]:
         """Get the currently active page."""
         if not self._initialized or not self._context:
@@ -106,7 +105,9 @@ class CamoufoxManager:
 
     async def _prefetch_camoufox(self) -> None:
         """Prefetch Camoufox binary and dependencies."""
-        emit_info("[cyan]🔍 Ensuring Camoufox binary and dependencies are up-to-date...[/cyan]")
+        emit_info(
+            "[cyan]🔍 Ensuring Camoufox binary and dependencies are up-to-date...[/cyan]"
+        )
 
         needs_install = False
         try:
@@ -168,7 +169,7 @@ class CamoufoxManager:
                     loop.create_task(self._cleanup())
                 else:
                     loop.run_until_complete(self._cleanup())
-            except:
+            except Exception:
                 pass  # Best effort cleanup
 
 

code_puppy/tools/browser_scripts.py

@@ -236,9 +236,6 @@ async def wait_for_element(
         return {"success": False, "error": str(e), "selector": selector, "state": state}
 
 
-
-
-
 async def highlight_element(
     selector: str,
     color: str = "red",
@@ -437,9 +434,6 @@ def register_wait_for_element(agent):
         return await wait_for_element(selector, state, timeout)
 
 
-
-
-
 def register_browser_highlight_element(agent):
     """Register the element highlighting tool."""
 

code_puppy/tools/browser_workflows.py

@@ -29,18 +29,18 @@ async def save_workflow(name: str, content: str) -> Dict[str, Any]:
         workflows_dir = get_workflows_directory()
 
         # Clean up the filename - remove spaces, special chars, etc.
-        safe_name = "".join(c for c in name if c.isalnum() or c in ('-', '_')).lower()
+        safe_name = "".join(c for c in name if c.isalnum() or c in ("-", "_")).lower()
         if not safe_name:
             safe_name = "workflow"
 
         # Ensure .md extension
-        if not safe_name.endswith('.md'):
-            safe_name += '.md'
+        if not safe_name.endswith(".md"):
+            safe_name += ".md"
 
         workflow_path = workflows_dir / safe_name
 
         # Write the workflow content
-        with open(workflow_path, 'w', encoding='utf-8') as f:
+        with open(workflow_path, "w", encoding="utf-8") as f:
             f.write(content)
 
         emit_info(
@@ -52,7 +52,7 @@ async def save_workflow(name: str, content: str) -> Dict[str, Any]:
             "success": True,
             "path": str(workflow_path),
             "name": safe_name,
-            "size": len(content)
+            "size": len(content),
         }
 
     except Exception as e:
@@ -75,23 +75,27 @@ async def list_workflows() -> Dict[str, Any]:
         workflows_dir = get_workflows_directory()
 
         # Find all .md files in the workflows directory
-        workflow_files = list(workflows_dir.glob('*.md'))
+        workflow_files = list(workflows_dir.glob("*.md"))
 
         workflows = []
         for workflow_file in workflow_files:
             try:
                 stat = workflow_file.stat()
-                workflows.append({
-                    "name": workflow_file.name,
-                    "path": str(workflow_file),
-                    "size": stat.st_size,
-                    "modified": stat.st_mtime
-                })
+                workflows.append(
+                    {
+                        "name": workflow_file.name,
+                        "path": str(workflow_file),
+                        "size": stat.st_size,
+                        "modified": stat.st_mtime,
+                    }
+                )
             except Exception as e:
-                emit_info(f"[yellow]Warning: Could not read {workflow_file}: {e}[/yellow]")
+                emit_info(
+                    f"[yellow]Warning: Could not read {workflow_file}: {e}[/yellow]"
+                )
 
         # Sort by modification time (newest first)
-        workflows.sort(key=lambda x: x['modified'], reverse=True)
+        workflows.sort(key=lambda x: x["modified"], reverse=True)
 
         emit_info(
             f"[green]✅ Found {len(workflows)} workflow(s)[/green]",
@@ -102,7 +106,7 @@ async def list_workflows() -> Dict[str, Any]:
             "success": True,
             "workflows": workflows,
             "count": len(workflows),
-            "directory": str(workflows_dir)
+            "directory": str(workflows_dir),
         }
 
     except Exception as e:
@@ -125,8 +129,8 @@ async def read_workflow(name: str) -> Dict[str, Any]:
         workflows_dir = get_workflows_directory()
 
         # Handle both with and without .md extension
-        if not name.endswith('.md'):
-            name += '.md'
+        if not name.endswith(".md"):
+            name += ".md"
 
         workflow_path = workflows_dir / name
 
@@ -135,10 +139,14 @@ async def read_workflow(name: str) -> Dict[str, Any]:
                 f"[red]❌ Workflow not found: {name}[/red]",
                 message_group=group_id,
             )
-            return {"success": False, "error": f"Workflow '{name}' not found", "name": name}
+            return {
+                "success": False,
+                "error": f"Workflow '{name}' not found",
+                "name": name,
+            }
 
         # Read the workflow content
-        with open(workflow_path, 'r', encoding='utf-8') as f:
+        with open(workflow_path, "r", encoding="utf-8") as f:
             content = f.read()
 
         emit_info(
@@ -151,7 +159,7 @@ async def read_workflow(name: str) -> Dict[str, Any]:
             "name": name,
             "content": content,
             "path": str(workflow_path),
-            "size": len(content)
+            "size": len(content),
         }
 
     except Exception as e:

code_puppy/tools/command_runner.py

@@ -19,8 +19,8 @@ from code_puppy.messaging import (
     emit_system_message,
     emit_warning,
 )
-from code_puppy.tui_state import is_tui_mode
 from code_puppy.tools.common import generate_group_id
+from code_puppy.tui_state import is_tui_mode
 
 # Maximum line length for shell command output to prevent massive token usage
 # This helps avoid exceeding model context limits when commands produce very long lines

code_puppy/tui/app.py

@@ -12,6 +12,8 @@ from textual.events import Resize
 from textual.reactive import reactive
 from textual.widgets import Footer, ListView
 
+# message_history_accumulator and prune_interrupted_tool_calls have been moved to BaseAgent class
+from code_puppy.agents.agent_manager import get_current_agent
 from code_puppy.command_line.command_handler import handle_command
 from code_puppy.config import (
     get_global_model_name,
@@ -19,12 +21,9 @@ from code_puppy.config import (
     initialize_command_history_file,
     save_command_to_history,
 )
-# message_history_accumulator and prune_interrupted_tool_calls have been moved to BaseAgent class
-from code_puppy.agents.agent_manager import get_current_agent
 
 # Import our message queue system
 from code_puppy.messaging import TUIRenderer, get_global_queue
-
 from code_puppy.tui.components import (
     ChatView,
     CustomTextArea,
@@ -33,7 +32,6 @@ from code_puppy.tui.components import (
     StatusBar,
 )
 
-
 # Import shared message classes
 from .messages import CommandSelected, HistoryEntrySelected
 from .models import ChatMessage, MessageType
@@ -175,12 +173,6 @@ class CodePuppyTUI(App):
             "Welcome to Code Puppy 🐶!\n💨 YOLO mode is enabled in TUI: commands will execute without confirmation."
         )
 
-        # Get current agent and display info
-        agent = get_current_agent()
-        self.add_system_message(
-            f"🐕 Loaded agent '{self.puppy_name}' with model '{self.current_model}'"
-        )
-
         # Start the message renderer EARLY to catch startup messages
         # Using call_after_refresh to start it as soon as possible after mount
         self.call_after_refresh(self.start_message_renderer_sync)
@@ -509,9 +501,7 @@ class CodePuppyTUI(App):
                     pass
             except Exception as agent_error:
                 # Handle any other errors in agent processing
-                self.add_error_message(
-                    f"Agent processing failed: {str(agent_error)}"
-                )
+                self.add_error_message(f"Agent processing failed: {str(agent_error)}")
 
         except Exception as e:
             self.add_error_message(f"Error processing message: {str(e)}")

code_puppy/tui/components/chat_view.py

@@ -267,6 +267,7 @@ class ChatView(VerticalScroll):
         ):
             # If either content is a Rich object, convert both to text and concatenate
             from io import StringIO
+
             from rich.console import Console
 
             # Convert existing content to string

code_puppy/tui/screens/settings.py

@@ -125,12 +125,12 @@ class SettingsScreen(ModalScreen):
     def on_mount(self) -> None:
         """Load current settings when the screen mounts."""
         from code_puppy.config import (
+            get_compaction_strategy,
+            get_compaction_threshold,
             get_global_model_name,
             get_owner_name,
             get_protected_token_count,
             get_puppy_name,
-            get_compaction_strategy,
-            get_compaction_threshold,
         )
 
         # Load current values
@@ -188,9 +188,9 @@ class SettingsScreen(ModalScreen):
     def save_settings(self) -> None:
         """Save the modified settings."""
         from code_puppy.config import (
+            get_model_context_length,
             set_config_value,
             set_model_name,
-            get_model_context_length,
         )
 
         try:

{code_puppy-0.0.176.dist-info → code_puppy-0.0.178.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: code-puppy
-Version: 0.0.176
+Version: 0.0.178
 Summary: Code generation agent
 Project-URL: repository, https://github.com/mpfaffenberger/code_puppy
 Project-URL: HomePage, https://github.com/mpfaffenberger/code_puppy
@@ -54,15 +54,15 @@ Description-Content-Type: text/markdown
 
 ## Overview
 
-*This project was coded angrily in reaction to Windsurf and Cursor removing access to models and raising prices.* 
+*This project was coded angrily in reaction to Windsurf and Cursor removing access to models and raising prices.*
 
 *You could also run 50 code puppies at once if you were insane enough.*
 
-*Would you rather plow a field with one ox or 1024 puppies?* 
+*Would you rather plow a field with one ox or 1024 puppies?*
     - If you pick the ox, better slam that back button in your browser.
-    
 
-Code Puppy is an AI-powered code generation agent, designed to understand programming tasks, generate high-quality code, and explain its reasoning similar to tools like Windsurf and Cursor. 
+
+Code Puppy is an AI-powered code generation agent, designed to understand programming tasks, generate high-quality code, and explain its reasoning similar to tools like Windsurf and Cursor.
 
 ## Quick start
 
@@ -497,22 +497,22 @@ class MyCustomAgent(BaseAgent):
     @property
     def name(self) -> str:
         return "my-agent"
-    
+
     @property
     def display_name(self) -> str:
         return "My Custom Agent ✨"
-    
+
     @property
     def description(self) -> str:
         return "A custom agent for specialized tasks"
-    
+
     def get_system_prompt(self) -> str:
         return "Your custom system prompt here..."
-    
+
     def get_available_tools(self) -> list[str]:
         return [
             "list_files",
-            "read_file", 
+            "read_file",
             "grep",
             "edit_file",
             "delete_file",