code-puppy 0.0.201__tar.gz → 0.0.203__tar.gz

{code_puppy-0.0.201 → code_puppy-0.0.203}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: code-puppy
-Version: 0.0.201
+Version: 0.0.203
 Summary: Code generation agent
 Project-URL: repository, https://github.com/mpfaffenberger/code_puppy
 Project-URL: HomePage, https://github.com/mpfaffenberger/code_puppy
@@ -26,7 +26,7 @@ Requires-Dist: openai>=1.99.1
 Requires-Dist: pathspec>=0.11.0
 Requires-Dist: playwright>=1.40.0
 Requires-Dist: prompt-toolkit>=3.0.52
-Requires-Dist: pydantic-ai==1.0.6
+Requires-Dist: pydantic-ai==1.0.5
 Requires-Dist: pydantic>=2.4.0
 Requires-Dist: pyjwt>=2.8.0
 Requires-Dist: pytest-cov>=6.1.1

{code_puppy-0.0.201 → code_puppy-0.0.203}/code_puppy/agents/agent_manager.py

@@ -252,7 +252,8 @@ def set_current_agent(agent_name: str) -> bool:
     global _CURRENT_AGENT
     curr_agent = get_current_agent()
     if curr_agent is not None:
-        _AGENT_HISTORIES[curr_agent.name] = curr_agent.get_message_history()
+        # Store a shallow copy so future mutations don't affect saved history
+        _AGENT_HISTORIES[curr_agent.name] = list(curr_agent.get_message_history())
     # Generate a message group ID for agent switching
     message_group_id = str(uuid.uuid4())
     _discover_agents(message_group_id=message_group_id)
@@ -269,7 +270,8 @@ def set_current_agent(agent_name: str) -> bool:
     _SESSION_AGENTS_CACHE[session_id] = agent_name
     _save_session_data(_SESSION_AGENTS_CACHE)
     if agent_obj.name in _AGENT_HISTORIES:
-        agent_obj.set_message_history(_AGENT_HISTORIES[agent_obj.name])
+        # Restore a copy to avoid sharing the same list instance
+        agent_obj.set_message_history(list(_AGENT_HISTORIES[agent_obj.name]))
     on_agent_reload(agent_obj.id, agent_name)
     return True
 

{code_puppy-0.0.201 → code_puppy-0.0.203}/code_puppy/agents/base_agent.py

@@ -13,7 +13,7 @@ import pydantic
 import pydantic_ai.models
 from pydantic_ai import Agent as PydanticAgent
 from pydantic_ai import BinaryContent, DocumentUrl, ImageUrl
-from pydantic_ai import RunContext, UsageLimitExceeded
+from pydantic_ai import RunContext, UsageLimitExceeded, UsageLimits
 from pydantic_ai.messages import (
     ModelMessage,
     ModelRequest,
@@ -950,7 +950,7 @@ class BaseAgent(ABC):
                 self.set_message_history(
                     self.prune_interrupted_tool_calls(self.get_message_history())
                 )
-                usage_limits = pydantic_ai.agent._usage.UsageLimits(request_limit=get_message_limit())
+                usage_limits = UsageLimits(request_limit=get_message_limit())
                 result_ = await pydantic_agent.run(
                     prompt_payload,
                     message_history=self.get_message_history(),

{code_puppy-0.0.201 → code_puppy-0.0.203}/code_puppy/command_line/attachments.py

@@ -151,20 +151,8 @@ def _is_supported_extension(path: Path) -> bool:
 
 
 def _parse_link(token: str) -> PromptLinkAttachment | None:
-    if "://" not in token:
-        return None
-    scheme = token.split(":", 1)[0].lower()
-    if scheme not in SUPPORTED_INLINE_SCHEMES:
-        return None
-    if token.lower().endswith(".pdf"):
-        return PromptLinkAttachment(
-            placeholder=token,
-            url_part=DocumentUrl(url=token),
-        )
-    return PromptLinkAttachment(
-        placeholder=token,
-        url_part=ImageUrl(url=token),
-    )
+    """URL parsing disabled: no URLs are treated as attachments."""
+    return None
 
 
 @dataclass

{code_puppy-0.0.201 → code_puppy-0.0.203}/code_puppy/tools/browser/camoufox_manager.py

@@ -1,6 +1,7 @@
 """Camoufox browser manager - privacy-focused Firefox automation."""
 
-from typing import Optional
+from pathlib import Path
+from typing import Optional, TypeAlias
 
 import camoufox
 from camoufox.addons import DefaultAddons
@@ -9,6 +10,8 @@ from camoufox.locale import ALLOW_GEOIP, download_mmdb
 from camoufox.pkgman import CamoufoxFetcher, camoufox_path
 from playwright.async_api import Browser, BrowserContext, Page
 
+_MIN_VIEWPORT_DIMENSION = 640
+
 from code_puppy.messaging import emit_info
 
 
@@ -38,6 +41,9 @@ class CamoufoxManager:
         self.block_webrtc = True  # Block WebRTC for privacy
         self.humanize = True  # Add human-like behavior
 
+        # Persistent profile directory for consistent browser state across runs
+        self.profile_dir = self._get_profile_directory()
+
     @classmethod
     def get_instance(cls) -> "CamoufoxManager":
         """Get the singleton instance."""
@@ -45,6 +51,16 @@ class CamoufoxManager:
             cls._instance = cls()
         return cls._instance
 
+    def _get_profile_directory(self) -> Path:
+        """Get or create the persistent profile directory.
+
+        Returns a Path object pointing to ~/.code_puppy/camoufox_profile
+        where browser data (cookies, history, bookmarks, etc.) will be stored.
+        """
+        profile_path = Path.home() / ".code_puppy" / "camoufox_profile"
+        profile_path.mkdir(parents=True, exist_ok=True)
+        return profile_path
+
     async def async_initialize(self) -> None:
         """Initialize Camoufox browser."""
         if self._initialized:
@@ -68,30 +84,40 @@ class CamoufoxManager:
 
     async def _initialize_camoufox(self) -> None:
         """Try to start Camoufox with the configured privacy settings."""
+        emit_info(f"[cyan]📁 Using persistent profile: {self.profile_dir}[/cyan]")
+
         camoufox_instance = camoufox.AsyncCamoufox(
             headless=self.headless,
             block_webrtc=self.block_webrtc,
             humanize=self.humanize,
             exclude_addons=list(DefaultAddons),
+            persistent_context=True,
+            user_data_dir=str(self.profile_dir),
             addons=[],
         )
-        self._browser = await camoufox_instance.start()
-        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)
+
+        self._browser = camoufox_instance.browser
+        # Use persistent storage directory for browser context
+        # This ensures cookies, localStorage, history, etc. persist across runs
+        if not self._initialized:
+            self._context = await camoufox_instance.start()
+            self._initialized = True
+            # Do not auto-open a page here to avoid duplicate windows/tabs.
 
     async def get_current_page(self) -> Optional[Page]:
-        """Get the currently active page."""
+        """Get the currently active page. Lazily creates one if none exist."""
         if not self._initialized or not self._context:
             await self.async_initialize()
 
-        if self._context:
-            pages = self._context.pages
-            return pages[0] if pages else None
-        return None
+        if not self._context:
+            return None
+
+        pages = self._context.pages
+        if pages:
+            return pages[0]
+
+        # Lazily create a new blank page without navigation
+        return await self._context.new_page()
 
     async def new_page(self, url: Optional[str] = None) -> Page:
         """Create a new page and optionally navigate to URL."""
@@ -140,9 +166,21 @@ class CamoufoxManager:
         return self._context.pages
 
     async def _cleanup(self) -> None:
-        """Clean up browser resources."""
+        """Clean up browser resources and save persistent state."""
         try:
+            # Save browser state before closing (cookies, localStorage, etc.)
             if self._context:
+                try:
+                    storage_state_path = self.profile_dir / "storage_state.json"
+                    await self._context.storage_state(path=str(storage_state_path))
+                    emit_info(
+                        f"[green]💾 Browser state saved to {storage_state_path}[/green]"
+                    )
+                except Exception as e:
+                    emit_info(
+                        f"[yellow]Warning: Could not save storage state: {e}[/yellow]"
+                    )
+
                 await self._context.close()
                 self._context = None
             if self._browser:

{code_puppy-0.0.201 → code_puppy-0.0.203}/pyproject.toml

@@ -4,12 +4,12 @@ build-backend = "hatchling.build"
 
 [project]
 name = "code-puppy"
-version = "0.0.201"
+version = "0.0.203"
 description = "Code generation agent"
 readme = "README.md"
 requires-python = ">=3.11"
 dependencies = [
-    "pydantic-ai==1.0.6",
+    "pydantic-ai==1.0.5",
     "httpx>=0.24.1",
     "rich>=13.4.2",
     "logfire>=0.7.1",