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

code_puppy/agents/base_agent.py

@@ -829,15 +829,15 @@ class BaseAgent(ABC):
 
         async def run_agent_task():
             try:
+                self.set_message_history(
+                    self.prune_interrupted_tool_calls(self.get_message_history())
+                )
                 result_ = await pydantic_agent.run(
                     prompt,
                     message_history=self.get_message_history(),
                     usage_limits=usage_limits,
                     **kwargs,
                 )
-                self.set_message_history(
-                    self.prune_interrupted_tool_calls(self.get_message_history())
-                )
                 return result_
             except* UsageLimitExceeded as ule:
                 emit_info(f"Usage limit exceeded: {str(ule)}", group_id=group_id)
@@ -883,10 +883,10 @@ class BaseAgent(ABC):
                         cancelled_exceptions.append(exc)
 
                 collect_cancelled_exceptions(other_error)
-
-                if cancelled_exceptions:
-                    # Re-raise the first CancelledError to propagate cancellation
-                    raise cancelled_exceptions[0]
+            finally:
+                self.set_message_history(
+                    self.prune_interrupted_tool_calls(self.get_message_history())
+                )
 
         # Create the task FIRST
         agent_task = asyncio.create_task(run_agent_task())
@@ -909,7 +909,6 @@ class BaseAgent(ABC):
                         agent_task.cancel()
             except Exception as e:
                 emit_info(f"Shell kill error: {e}")
-                # If shell kill failed, still try to cancel the agent task
                 if not agent_task.done():
                     agent_task.cancel()
             # Don't call the original handler
@@ -923,8 +922,7 @@ class BaseAgent(ABC):
             result = await agent_task
             return result
         except asyncio.CancelledError:
-            # Task was cancelled by our handler
-            raise
+            agent_task.cancel()
         except KeyboardInterrupt:
             # Handle direct keyboard interrupt during await
             if not agent_task.done():
@@ -933,7 +931,6 @@ class BaseAgent(ABC):
                 await agent_task
             except asyncio.CancelledError:
                 pass
-            raise asyncio.CancelledError()
         finally:
             # Restore original signal handler
             if original_handler:

code_puppy/main.py

@@ -383,6 +383,8 @@ async def interactive_mode(message_renderer, initial_command: str = None) -> Non
         if task.strip().lower() in ("clear", "/clear"):
             from code_puppy.messaging import emit_system_message, emit_warning
 
+            agent = get_current_agent()
+            agent.clear_message_history()
             emit_warning("Conversation history cleared!")
             emit_system_message("The agent will not remember previous interactions.\n")
             continue

code_puppy/tui/app.py

@@ -515,6 +515,8 @@ class CodePuppyTUI(App):
         """Clear the chat history."""
         chat_view = self.query_one("#chat-view", ChatView)
         chat_view.clear_messages()
+        agent = get_current_agent()
+        agent.clear_message_history()
         self.add_system_message("Chat history cleared")
 
     def action_show_help(self) -> None:

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: code-puppy
-Version: 0.0.178
+Version: 0.0.180
 Summary: Code generation agent
 Project-URL: repository, https://github.com/mpfaffenberger/code_puppy
 Project-URL: HomePage, https://github.com/mpfaffenberger/code_puppy

{code_puppy-0.0.178.dist-info → code_puppy-0.0.180.dist-info}/RECORD

@@ -3,7 +3,7 @@ code_puppy/__main__.py,sha256=pDVssJOWP8A83iFkxMLY9YteHYat0EyWDQqMkKHpWp4,203
 code_puppy/callbacks.py,sha256=6wYB6K_fGSCkKKEFaYOYkJT45WaV5W_NhUIzcvVH_nU,5060
 code_puppy/config.py,sha256=smjGUs18VJQxPeVLvV8U2T_YuGh7E_Sg5ZgaXhdBWWg,20677
 code_puppy/http_utils.py,sha256=YLd8Y16idbI32JGeBXG8n5rT4o4X_zxk9FgUvK9XFo8,8248
-code_puppy/main.py,sha256=c5nxJ6eRve9KFFTxVtioWJ393e6sxy7quld9TQpsH94,20519
+code_puppy/main.py,sha256=_TvwZFFepyq6UCC29ApFUeXeODsRHV2OyFuzM3xkiwo,20601
 code_puppy/model_factory.py,sha256=KLyxUrZJual6uo_TncjdPhdDbvIvHiztWvK9bfdQmmc,13759
 code_puppy/models.json,sha256=iXmLZGflnQcu2DRh4WUlgAhoXdvoxUc7KBhB8YxawXM,3088
 code_puppy/reopenable_async_client.py,sha256=4UJRaMp5np8cbef9F0zKQ7TPKOfyf5U-Kv-0zYUWDho,8274
@@ -13,12 +13,20 @@ code_puppy/summarization_agent.py,sha256=LnObgtLmM6N4z2553XXQlXAOf8R1BPSNmFSfXkj
 code_puppy/tui_state.py,sha256=TT76XBVapKj6fKjFzz6oxCONeN_BZwcMILxxZcxu6-Y,1171
 code_puppy/version_checker.py,sha256=bjLDmgGPrl7XnYwX1u13O8uFlsfikV90PK6nbA9Z9QU,1150
 code_puppy/agents/__init__.py,sha256=PtPB7Z5MSwmUKipgt_qxvIuGggcuVaYwNbnp1UP4tPc,518
+code_puppy/agents/agent_c_reviewer.py,sha256=XVOONfnbCS6nEdcXoiowuSB5pnqrXyoWIAjG2cKm7BU,5044
 code_puppy/agents/agent_code_puppy.py,sha256=vk25J5YSjt30ItfML8KyQc5QR8QumBv5_U8RhqaXUz4,7973
+code_puppy/agents/agent_code_reviewer.py,sha256=8XHktiFrOVeFZ81WIefuTPQw_IzcNddm4bk46mW46ok,4003
+code_puppy/agents/agent_cpp_reviewer.py,sha256=H4INgJo2OJ84QT7bfTkw4s1Ml7luwokhAgTsdig2TQQ,3653
 code_puppy/agents/agent_creator_agent.py,sha256=IiwVirB6uoIeGOmtetut9eDv6o055ykND3V-fvyA8Lw,23042
-code_puppy/agents/agent_golang_reviewer.py,sha256=cdJEK0J5oZEIW9Sw5Hj_FiVP0iN-_NoAnjP4b6xB_Rw,3308
+code_puppy/agents/agent_golang_reviewer.py,sha256=-OMuT8hkapVf2Oox46Ck9SRHlsfd8ab8uefbVfdW72M,3348
+code_puppy/agents/agent_javascript_reviewer.py,sha256=5YC4kRSvorcNgObjHjD2Rrgnvf8jlKhPqWdjOMjU9A0,3636
 code_puppy/agents/agent_manager.py,sha256=USzTbsGGZ9CH17LFeb6P6Im8RPY8qZUGKoH9EZZyU3k,11413
+code_puppy/agents/agent_python_reviewer.py,sha256=D0M3VA12QKdsyg2zIBI2FECxz0IP2fSIfg24xGzDhw0,3837
+code_puppy/agents/agent_qa_expert.py,sha256=wCGXzuAVElT5c-QigQVb8JX9Gw0JmViCUQQnADMSbVc,3796
 code_puppy/agents/agent_qa_kitten.py,sha256=5PeFFSwCFlTUvP6h5bGntx0xv5NmRwBiw0HnMqY8nLI,9107
-code_puppy/agents/base_agent.py,sha256=Ld9WTYPMr2uewivHmjBr99L6vlkR1F7jdlz6XG_ZBUQ,37155
+code_puppy/agents/agent_security_auditor.py,sha256=ADafi2x4gqXw6m-Nch5vjiKjO0Urcbj0x4zxHti3gDw,3712
+code_puppy/agents/agent_typescript_reviewer.py,sha256=EDY1mFkVpuJ1BPXsJFu2wQ2pfAV-90ipc_8w9ymrKPg,4054
+code_puppy/agents/base_agent.py,sha256=MFyC4YSxlQ1xzQRBnSXgtyO7XPpHZuSJVzRATU8j7J0,36992
 code_puppy/agents/json_agent.py,sha256=y6AYE3Fx9LhmemcPzt46d7359MNnkGIjU83YBGNer2g,4533
 code_puppy/command_line/__init__.py,sha256=y7WeRemfYppk8KVbCGeAIiTuiOszIURCDjOMZv_YRmU,45
 code_puppy/command_line/command_handler.py,sha256=JeJo6kc9-4B7S0_pbMp1LO7jtAT6XMXtSSnOShfoG-Y,27237
@@ -75,13 +83,6 @@ code_puppy/messaging/spinner/textual_spinner.py,sha256=Omx9A-FSPkxYDMYgBXgYMBQnK
 code_puppy/plugins/__init__.py,sha256=fksDqMUiXPJ5WNuMsYsVR8ulueQRCXPlvECEyicHPtQ,1312
 code_puppy/tools/__init__.py,sha256=BVTZ85jLHgDANwOnUSOz3UDlp8VQDq4DoGF23BRlyWw,6032
 code_puppy/tools/agent_tools.py,sha256=VMTKPfNUiK90ifyJUtEvUpOHmmifpWCxDKv7j6oTaxI,5620
-code_puppy/tools/browser_control.py,sha256=6E_Kq63cErBk-iM9-03Cq8sTWWLh-Tk2kpdrB_rMmbg,8603
-code_puppy/tools/browser_interactions.py,sha256=m-bVsAUbdrS-P1GbS0ChO2tMMg4tMIsDX34azn_Olzs,17070
-code_puppy/tools/browser_locators.py,sha256=ZshjhYuV600vJqXUC23oNX89AZLMvY8rlvj-9oy7GMw,19574
-code_puppy/tools/browser_navigation.py,sha256=Tj_fNcM3KGpkM2UTKcGQX8BpI373Sv7xZAJf-U7pO5M,7621
-code_puppy/tools/browser_screenshot.py,sha256=QSwxS37G4LSo-Q9SBiuIofxWKnyInM90TY-_fiWQLrs,9222
-code_puppy/tools/browser_scripts.py,sha256=MMO5KRjdrhuLOoJGoKGG1jm6UAqhFhUznz02aWqhMAE,15065
-code_puppy/tools/browser_workflows.py,sha256=SJfiM2SgZr-qeeQJEivGzUNc51AlLIoqj9h1-mTvVsA,6593
 code_puppy/tools/command_runner.py,sha256=SRhTACgKklHUm_ztbFHnefGbLw4CJOm601or6L1cPGM,22417
 code_puppy/tools/common.py,sha256=pL-9xcRs3rxU7Fl9X9EUgbDp2-csh2LLJ5DHH_KAHKY,10596
 code_puppy/tools/file_modifications.py,sha256=EaDWcv6gi8wAvpgyeJdKSKPWg9fTpZoEkxQiLCE6rn4,23218
@@ -98,7 +99,7 @@ code_puppy/tools/browser/browser_workflows.py,sha256=jplJ1T60W3G4-dhVJX-CXkm9ssk
 code_puppy/tools/browser/camoufox_manager.py,sha256=RYvLcs0iAoVNtpLjrrA1uu6a5k9tAdBbmhWFGSWjX_A,6106
 code_puppy/tools/browser/vqa_agent.py,sha256=0GMDgJAK728rIuSQxAVytFSNagjo0LCjCUxBTm3w9Po,1952
 code_puppy/tui/__init__.py,sha256=XesAxIn32zLPOmvpR2wIDxDAnnJr81a5pBJB4cZp1Xs,321
-code_puppy/tui/app.py,sha256=f_8c9t9WF1twn4GCNNnmEgklSyUDC-9-d8_Lkosszjk,37911
+code_puppy/tui/app.py,sha256=fiOKrhtYBpruUPM-aqnsDmEiOM_dCbD5DnBRKO219NI,37985
 code_puppy/tui/messages.py,sha256=zQoToWI0eWdT36NEsY6RdCFzcDfAmfvoPlHv8jiCbgo,720
 code_puppy/tui/components/__init__.py,sha256=uj5pnk3s6SEN3SbFI0ZnzaA2KK1NNg8TfUj6U-Z732U,455
 code_puppy/tui/components/chat_view.py,sha256=Ff6uM6J0yENISNAOYroX7F-JL73_ajUUcP5IZSf2mng,19914
@@ -118,9 +119,9 @@ code_puppy/tui/screens/help.py,sha256=eJuPaOOCp7ZSUlecearqsuX6caxWv7NQszUh0tZJjB
 code_puppy/tui/screens/mcp_install_wizard.py,sha256=vObpQwLbXjQsxmSg-WCasoev1usEi0pollKnL0SHu9U,27693
 code_puppy/tui/screens/settings.py,sha256=-WLldnKyWVKUYVPJcfOn1UU6eP9t8lLPUAVI317SOOM,10685
 code_puppy/tui/screens/tools.py,sha256=3pr2Xkpa9Js6Yhf1A3_wQVRzFOui-KDB82LwrsdBtyk,1715
-code_puppy-0.0.178.data/data/code_puppy/models.json,sha256=iXmLZGflnQcu2DRh4WUlgAhoXdvoxUc7KBhB8YxawXM,3088
-code_puppy-0.0.178.dist-info/METADATA,sha256=sn-XrM1AZoJQWMYG6YSkGknl0cmfCjfLOdb6NLEMzYE,20079
-code_puppy-0.0.178.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-code_puppy-0.0.178.dist-info/entry_points.txt,sha256=Tp4eQC99WY3HOKd3sdvb22vZODRq0XkZVNpXOag_KdI,91
-code_puppy-0.0.178.dist-info/licenses/LICENSE,sha256=31u8x0SPgdOq3izJX41kgFazWsM43zPEF9eskzqbJMY,1075
-code_puppy-0.0.178.dist-info/RECORD,,
+code_puppy-0.0.180.data/data/code_puppy/models.json,sha256=iXmLZGflnQcu2DRh4WUlgAhoXdvoxUc7KBhB8YxawXM,3088
+code_puppy-0.0.180.dist-info/METADATA,sha256=vYolNp6ezYwpXSXhhzSdt_6GZdqFOOMVutES6BiN5jM,20079
+code_puppy-0.0.180.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+code_puppy-0.0.180.dist-info/entry_points.txt,sha256=Tp4eQC99WY3HOKd3sdvb22vZODRq0XkZVNpXOag_KdI,91
+code_puppy-0.0.180.dist-info/licenses/LICENSE,sha256=31u8x0SPgdOq3izJX41kgFazWsM43zPEF9eskzqbJMY,1075
+code_puppy-0.0.180.dist-info/RECORD,,

code_puppy/tools/browser_control.py

@@ -1,293 +0,0 @@
-"""Browser initialization and control tools."""
-
-from typing import Any, Dict, Optional
-
-from pydantic_ai import RunContext
-
-from code_puppy.messaging import emit_info
-from code_puppy.tools.common import generate_group_id
-
-from .camoufox_manager import get_camoufox_manager
-
-
-async def initialize_browser(
-    headless: bool = False,
-    browser_type: str = "chromium",
-    homepage: str = "https://www.google.com",
-) -> Dict[str, Any]:
-    """Initialize the browser with specified settings."""
-    group_id = generate_group_id("browser_initialize", f"{browser_type}_{homepage}")
-    emit_info(
-        f"[bold white on blue] BROWSER INITIALIZE [/bold white on blue] 🌐 {browser_type} → {homepage}",
-        message_group=group_id,
-    )
-    try:
-        browser_manager = get_camoufox_manager()
-
-        # Configure browser settings
-        browser_manager.headless = headless
-        browser_manager.browser_type = browser_type
-        browser_manager.homepage = homepage
-
-        # Initialize browser
-        await browser_manager.async_initialize()
-
-        # Get page info
-        page = await browser_manager.get_current_page()
-        if page:
-            url = page.url
-            title = await page.title()
-        else:
-            url = "Unknown"
-            title = "Unknown"
-
-        emit_info(
-            "[green]Browser initialized successfully[/green]", message_group=group_id
-        )
-
-        return {
-            "success": True,
-            "browser_type": browser_type,
-            "headless": headless,
-            "homepage": homepage,
-            "current_url": url,
-            "current_title": title,
-        }
-
-    except Exception as e:
-        emit_info(
-            f"[red]Browser initialization failed: {str(e)}[/red]",
-            message_group=group_id,
-        )
-        return {
-            "success": False,
-            "error": str(e),
-            "browser_type": browser_type,
-            "headless": headless,
-        }
-
-
-async def close_browser() -> Dict[str, Any]:
-    """Close the browser and clean up resources."""
-    group_id = generate_group_id("browser_close")
-    emit_info(
-        "[bold white on blue] BROWSER CLOSE [/bold white on blue] 🔒",
-        message_group=group_id,
-    )
-    try:
-        browser_manager = get_camoufox_manager()
-        await browser_manager.close()
-
-        emit_info(
-            "[yellow]Browser closed successfully[/yellow]", message_group=group_id
-        )
-
-        return {"success": True, "message": "Browser closed"}
-
-    except Exception as e:
-        return {"success": False, "error": str(e)}
-
-
-async def get_browser_status() -> Dict[str, Any]:
-    """Get current browser status and information."""
-    group_id = generate_group_id("browser_status")
-    emit_info(
-        "[bold white on blue] BROWSER STATUS [/bold white on blue] 📊",
-        message_group=group_id,
-    )
-    try:
-        browser_manager = get_camoufox_manager()
-
-        if not browser_manager._initialized:
-            return {
-                "success": True,
-                "status": "not_initialized",
-                "browser_type": browser_manager.browser_type,
-                "headless": browser_manager.headless,
-            }
-
-        page = await browser_manager.get_current_page()
-        if page:
-            url = page.url
-            title = await page.title()
-
-            # Get all pages
-            all_pages = await browser_manager.get_all_pages()
-            page_count = len(all_pages)
-        else:
-            url = None
-            title = None
-            page_count = 0
-
-        return {
-            "success": True,
-            "status": "initialized",
-            "browser_type": browser_manager.browser_type,
-            "headless": browser_manager.headless,
-            "current_url": url,
-            "current_title": title,
-            "page_count": page_count,
-        }
-
-    except Exception as e:
-        return {"success": False, "error": str(e)}
-
-
-async def create_new_page(url: Optional[str] = None) -> Dict[str, Any]:
-    """Create a new browser page/tab."""
-    group_id = generate_group_id("browser_new_page", url or "blank")
-    emit_info(
-        f"[bold white on blue] BROWSER NEW PAGE [/bold white on blue] 📄 {url or 'blank page'}",
-        message_group=group_id,
-    )
-    try:
-        browser_manager = get_camoufox_manager()
-
-        if not browser_manager._initialized:
-            return {
-                "success": False,
-                "error": "Browser not initialized. Use browser_initialize first.",
-            }
-
-        page = await browser_manager.new_page(url)
-
-        final_url = page.url
-        title = await page.title()
-
-        emit_info(
-            f"[green]Created new page: {final_url}[/green]", message_group=group_id
-        )
-
-        return {"success": True, "url": final_url, "title": title, "requested_url": url}
-
-    except Exception as e:
-        return {"success": False, "error": str(e), "url": url}
-
-
-async def list_pages() -> Dict[str, Any]:
-    """List all open browser pages/tabs."""
-    group_id = generate_group_id("browser_list_pages")
-    emit_info(
-        "[bold white on blue] BROWSER LIST PAGES [/bold white on blue] 📋",
-        message_group=group_id,
-    )
-    try:
-        browser_manager = get_camoufox_manager()
-
-        if not browser_manager._initialized:
-            return {"success": False, "error": "Browser not initialized"}
-
-        all_pages = await browser_manager.get_all_pages()
-
-        pages_info = []
-        for i, page in enumerate(all_pages):
-            try:
-                url = page.url
-                title = await page.title()
-                is_closed = page.is_closed()
-
-                pages_info.append(
-                    {"index": i, "url": url, "title": title, "closed": is_closed}
-                )
-            except Exception as e:
-                pages_info.append(
-                    {
-                        "index": i,
-                        "url": "Error",
-                        "title": "Error",
-                        "error": str(e),
-                        "closed": True,
-                    }
-                )
-
-        return {"success": True, "page_count": len(all_pages), "pages": pages_info}
-
-    except Exception as e:
-        return {"success": False, "error": str(e)}
-
-
-# Tool registration functions
-def register_initialize_browser(agent):
-    """Register the browser initialization tool."""
-
-    @agent.tool
-    async def browser_initialize(
-        context: RunContext,
-        headless: bool = False,
-        browser_type: str = "chromium",
-        homepage: str = "https://www.google.com",
-    ) -> Dict[str, Any]:
-        """
-        Initialize the browser with specified settings. Must be called before using other browser tools.
-
-        Args:
-            headless: Run browser in headless mode (no GUI)
-            browser_type: Browser engine (chromium, firefox, webkit)
-            homepage: Initial page to load
-
-        Returns:
-            Dict with initialization results
-        """
-        return await initialize_browser(headless, browser_type, homepage)
-
-
-def register_close_browser(agent):
-    """Register the browser close tool."""
-
-    @agent.tool
-    async def browser_close(context: RunContext) -> Dict[str, Any]:
-        """
-        Close the browser and clean up all resources.
-
-        Returns:
-            Dict with close results
-        """
-        return await close_browser()
-
-
-def register_get_browser_status(agent):
-    """Register the browser status tool."""
-
-    @agent.tool
-    async def browser_status(context: RunContext) -> Dict[str, Any]:
-        """
-        Get current browser status and information.
-
-        Returns:
-            Dict with browser status and metadata
-        """
-        return await get_browser_status()
-
-
-def register_create_new_page(agent):
-    """Register the new page creation tool."""
-
-    @agent.tool
-    async def browser_new_page(
-        context: RunContext,
-        url: Optional[str] = None,
-    ) -> Dict[str, Any]:
-        """
-        Create a new browser page/tab.
-
-        Args:
-            url: Optional URL to navigate to in the new page
-
-        Returns:
-            Dict with new page results
-        """
-        return await create_new_page(url)
-
-
-def register_list_pages(agent):
-    """Register the list pages tool."""
-
-    @agent.tool
-    async def browser_list_pages(context: RunContext) -> Dict[str, Any]:
-        """
-        List all open browser pages/tabs.
-
-        Returns:
-            Dict with information about all open pages
-        """
-        return await list_pages()