code-puppy 0.0.175__py3-none-any.whl → 0.0.177__py3-none-any.whl

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,12 +3,13 @@
 from typing import Optional
 
 import camoufox
-from playwright.async_api import Browser, BrowserContext, Page, Playwright, async_playwright
+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
-from camoufox.locale import ALLOW_GEOIP, download_mmdb
-from camoufox.addons import maybe_download_addons, DefaultAddons
 
 
 class CamoufoxManager:
@@ -17,7 +18,6 @@ class CamoufoxManager:
     _instance: Optional["CamoufoxManager"] = None
     _browser: Optional[Browser] = None
     _context: Optional[BrowserContext] = None
-    _playwright: Optional[Playwright] = None
     _initialized: bool = False
 
     def __new__(cls):
@@ -52,27 +52,17 @@ class CamoufoxManager:
 
         try:
             emit_info("[yellow]Initializing Camoufox (privacy Firefox)...[/yellow]")
-            
+
             # Ensure Camoufox binary and dependencies are fetched before launching
             await self._prefetch_camoufox()
 
-            try:
-                await self._initialize_camoufox()
-                emit_info(
-                    "[green]✅ Camoufox initialized successfully (privacy-focused Firefox)[/green]"
-                )
-            except Exception as camoufox_error:
-                error_reason = str(camoufox_error).splitlines()[0]
-                emit_info(
-                    "[yellow]⚠️ Camoufox failed to initialize, falling back to Playwright Firefox[/yellow]"
-                )
-                await self._cleanup()
-                await self._initialize_playwright_firefox(error_reason)
-
+            await self._initialize_camoufox()
+            emit_info(
+                "[green]✅ Camoufox initialized successfully (privacy-focused Firefox)[/green]"
+            )
             self._initialized = True
 
-        except Exception as e:
-            emit_info(f"[red]❌ Failed to initialize browser: {e}[/red]")
+        except Exception:
             await self._cleanup()
             raise
 
@@ -82,6 +72,8 @@ class CamoufoxManager:
             headless=self.headless,
             block_webrtc=self.block_webrtc,
             humanize=self.humanize,
+            exclude_addons=list(DefaultAddons),
+            addons=[],
         )
         self._browser = await camoufox_instance.start()
         self._context = await self._browser.new_context(
@@ -91,20 +83,6 @@ class CamoufoxManager:
         page = await self._context.new_page()
         await page.goto(self.homepage)
 
-    async def _initialize_playwright_firefox(self, error_reason: str) -> None:
-        """Fallback to vanilla Playwright Firefox when Camoufox fails."""
-        self._playwright = await async_playwright().start()
-        self._browser = await self._playwright.firefox.launch(headless=self.headless)
-        self._context = await self._browser.new_context(
-            viewport={"width": 1920, "height": 1080},
-            ignore_https_errors=True,
-        )
-        page = await self._context.new_page()
-        await page.goto(self.homepage)
-        emit_info(
-            f"[green]✅ Playwright Firefox fallback ready (Camoufox error: {error_reason})[/green]"
-        )
-
     async def get_current_page(self) -> Optional[Page]:
         """Get the currently active page."""
         if not self._initialized or not self._context:
@@ -127,18 +105,28 @@ 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]")
-        
-        # Fetch Camoufox binary if needed
-        CamoufoxFetcher().install()
-        
+        emit_info(
+            "[cyan]🔍 Ensuring Camoufox binary and dependencies are up-to-date...[/cyan]"
+        )
+
+        needs_install = False
+        try:
+            camoufox_path(download_if_missing=False)
+            emit_info("[cyan]🗃️ Using cached Camoufox installation[/cyan]")
+        except (CamoufoxNotInstalled, FileNotFoundError):
+            emit_info("[cyan]📥 Camoufox not found, installing fresh copy[/cyan]")
+            needs_install = True
+        except UnsupportedVersion:
+            emit_info("[cyan]♻️ Camoufox update required, reinstalling[/cyan]")
+            needs_install = True
+
+        if needs_install:
+            CamoufoxFetcher().install()
+
         # Fetch GeoIP database if enabled
         if ALLOW_GEOIP:
             download_mmdb()
-        
-        # Download default addons
-        maybe_download_addons(list(DefaultAddons))
-        
+
         emit_info("[cyan]📦 Camoufox dependencies ready[/cyan]")
 
     async def close_page(self, page: Page) -> None:
@@ -160,9 +148,6 @@ class CamoufoxManager:
             if self._browser:
                 await self._browser.close()
                 self._browser = None
-            if self._playwright:
-                await self._playwright.stop()
-                self._playwright = None
             self._initialized = False
         except Exception as e:
             emit_info(f"[yellow]Warning during cleanup: {e}[/yellow]")
@@ -184,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/vqa_agent.py

@@ -3,7 +3,6 @@
 from __future__ import annotations
 
 from functools import lru_cache
-from typing import Optional
 
 from pydantic import BaseModel, Field
 from pydantic_ai import Agent, BinaryContent
@@ -37,7 +36,6 @@ def _load_vqa_agent(model_name: str) -> Agent[None, VisualAnalysisResult]:
         instructions=instructions,
         output_type=VisualAnalysisResult,
         retries=2,
-        instrument=instrumentation,
     )
 
 

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.175.dist-info → code_puppy-0.0.177.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: code-puppy
-Version: 0.0.175
+Version: 0.0.177
 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",