code-puppy 0.0.329__py3-none-any.whl → 0.0.331__py3-none-any.whl

code_puppy/cli_runner.py

@@ -421,6 +421,10 @@ async def interactive_mode(message_renderer, initial_command: str = None) -> Non
     current_agent_task = None
 
     while True:
+        # Windows-specific: Aggressively reset terminal at the start of every loop
+        # This fixes terminal corruption after Ctrl+C, especially when running via uvx
+        reset_windows_terminal_full()
+
         from code_puppy.agents.agent_manager import get_current_agent
         from code_puppy.messaging import emit_info
 
@@ -707,6 +711,9 @@ async def run_prompt_with_attachments(
     attachments = [attachment.content for attachment in processed_prompt.attachments]
     link_attachments = [link.url_part for link in processed_prompt.link_attachments]
 
+    # Trigger invoke_agent callbacks (e.g., Claude Code OAuth token refresh)
+    await callbacks.on_invoke_agent()
+
     # IMPORTANT: Set the shared console on the agent so that streaming output
     # uses the same console as the spinner. This prevents Live display conflicts
     # that cause line duplication during markdown streaming.

code_puppy/plugins/claude_code_oauth/register_callbacks.py

@@ -25,6 +25,7 @@ from .utils import (
     fetch_claude_code_models,
     load_claude_models_filtered,
     load_stored_tokens,
+    maybe_refresh_token,
     prepare_oauth_context,
     remove_claude_code_models,
     save_tokens,
@@ -302,5 +303,19 @@ def _handle_custom_command(command: str, name: str) -> Optional[bool]:
     return None
 
 
+def _on_invoke_agent(*args, **kwargs) -> None:
+    """Called before each agent invocation.
+
+    Checks if 30 minutes have passed since last token refresh and
+    refreshes the Claude Code OAuth token if needed.
+    """
+    try:
+        if maybe_refresh_token():
+            logger.debug("Token refresh check completed")
+    except Exception as exc:  # pragma: no cover - defensive
+        logger.warning("Token refresh check failed: %s", exc)
+
+
 register_callback("custom_command_help", _custom_help)
 register_callback("custom_command", _handle_custom_command)
+register_callback("invoke_agent", _on_invoke_agent)

code_puppy/plugins/claude_code_oauth/utils.py

@@ -23,6 +23,10 @@ from .config import (
 
 logger = logging.getLogger(__name__)
 
+# Token refresh tracking
+_last_refresh_time: float = 0.0
+REFRESH_INTERVAL_SECONDS: float = 30 * 60  # 30 minutes
+
 
 @dataclass
 class OAuthContext:
@@ -397,3 +401,201 @@ def remove_claude_code_models() -> int:
     except Exception as exc:  # pragma: no cover - defensive logging
         logger.error("Error removing Claude Code models: %s", exc)
     return 0
+
+
+def _update_model_tokens(new_access_token: str) -> bool:
+    """Update all Claude Code models with the new access token.
+
+    Args:
+        new_access_token: The new access token to set
+
+    Returns:
+        True if successful, False otherwise
+    """
+    try:
+        claude_models = load_claude_models()
+        if not claude_models:
+            logger.debug("No models to update")
+            return True
+
+        updated = 0
+        for _model_name, config in claude_models.items():
+            if config.get("oauth_source") == "claude-code-plugin":
+                if (
+                    "custom_endpoint" in config
+                    and "api_key" in config["custom_endpoint"]
+                ):
+                    config["custom_endpoint"]["api_key"] = new_access_token
+                    updated += 1
+
+        if updated > 0:
+            if save_claude_models(claude_models):
+                logger.info("Updated %s model configurations with new token", updated)
+                return True
+            else:
+                logger.error("Failed to save updated model configurations")
+                return False
+
+        return True
+
+    except Exception as exc:  # pragma: no cover - defensive logging
+        logger.error("Error updating model tokens: %s", exc)
+        return False
+
+
+def refresh_access_token() -> Optional[str]:
+    """Refresh the access token using the refresh token.
+
+    Returns:
+        New access token if successful, None otherwise
+    """
+    try:
+        tokens = load_stored_tokens()
+        if not tokens:
+            logger.debug("No stored tokens found for refresh")
+            return None
+
+        if "refresh_token" not in tokens:
+            logger.debug("No refresh token available")
+            return None
+
+        refresh_token = tokens["refresh_token"]
+
+        # Prepare refresh request
+        payload = {
+            "grant_type": "refresh_token",
+            "client_id": CLAUDE_CODE_OAUTH_CONFIG["client_id"],
+            "refresh_token": refresh_token,
+        }
+
+        headers = {
+            "Content-Type": "application/json",
+            "Accept": "application/json",
+            "anthropic-beta": "oauth-2025-04-20",
+        }
+
+        logger.info("Refreshing Claude Code access token...")
+        response = requests.post(
+            CLAUDE_CODE_OAUTH_CONFIG["token_url"],
+            json=payload,
+            headers=headers,
+            timeout=30,
+        )
+
+        if response.status_code == 200:
+            token_data = response.json()
+
+            # Update tokens with new access token and expiry
+            new_access_token = token_data.get("access_token")
+            if not new_access_token:
+                logger.error("No access_token in refresh response")
+                return None
+
+            # Update stored tokens
+            tokens["access_token"] = new_access_token
+
+            # Update expiry if provided
+            if "expires_in" in token_data:
+                tokens["expires_at"] = time.time() + token_data["expires_in"]
+
+            # Update refresh token if a new one was provided
+            if "refresh_token" in token_data:
+                tokens["refresh_token"] = token_data["refresh_token"]
+
+            # Save updated tokens
+            if save_tokens(tokens):
+                logger.info("Claude Code access token refreshed successfully")
+
+                # Update model configurations with new token
+                _update_model_tokens(new_access_token)
+
+                return new_access_token
+            else:
+                logger.error("Failed to save refreshed tokens")
+                return None
+        else:
+            logger.warning(
+                "Token refresh failed: %s - %s",
+                response.status_code,
+                response.text,
+            )
+            return None
+
+    except Exception as exc:  # pragma: no cover - defensive logging
+        logger.error("Error refreshing access token: %s", exc)
+        return None
+
+
+def _is_using_claude_code_model() -> bool:
+    """Check if the current agent is using a Claude Code OAuth model.
+
+    Returns:
+        True if currently using a claude-code-* model, False otherwise
+    """
+    try:
+        from code_puppy.agents import get_current_agent
+        from code_puppy.model_utils import is_claude_code_model
+
+        agent = get_current_agent()
+        if agent is None:
+            return False
+
+        model_name = agent.get_model_name()
+        if not model_name:
+            return False
+
+        return is_claude_code_model(model_name)
+    except Exception as exc:
+        logger.debug("Could not determine current model: %s", exc)
+        return False
+
+
+def maybe_refresh_token() -> bool:
+    """Refresh the token if 30 minutes have passed since last refresh.
+
+    This function is designed to be called on every prompt, but will only
+    actually refresh the token if:
+    1. The current model is a Claude Code OAuth model (starts with 'claude-code-')
+    2. REFRESH_INTERVAL_SECONDS (30 min) has passed since last refresh
+
+    Returns:
+        True if refresh was attempted (regardless of success), False if skipped
+    """
+    global _last_refresh_time
+
+    # Only refresh if we're actually using a Claude Code model
+    if not _is_using_claude_code_model():
+        return False
+
+    # Check if we have tokens at all
+    tokens = load_stored_tokens()
+    if not tokens or "refresh_token" not in tokens:
+        return False
+
+    current_time = time.time()
+    time_since_last = current_time - _last_refresh_time
+
+    if time_since_last < REFRESH_INTERVAL_SECONDS:
+        logger.debug(
+            "Skipping token refresh, %.1f minutes until next refresh",
+            (REFRESH_INTERVAL_SECONDS - time_since_last) / 60,
+        )
+        return False
+
+    logger.info(
+        "Token refresh interval reached (%.1f min since last refresh)",
+        time_since_last / 60,
+    )
+
+    # Attempt refresh
+    result = refresh_access_token()
+    if result:
+        _last_refresh_time = current_time
+        logger.info("Token refresh successful, next refresh in 30 minutes")
+    else:
+        # Even on failure, update the timestamp to avoid hammering the API
+        # We'll retry on the next 30-min interval
+        _last_refresh_time = current_time
+        logger.warning("Token refresh failed, will retry in 30 minutes")
+
+    return True

code_puppy/terminal_utils.py

@@ -87,16 +87,20 @@ def reset_windows_console_mode() -> None:
 
 
 def reset_windows_terminal_full() -> None:
-    """Perform a full Windows terminal reset (ANSI + console mode).
+    """Perform a full Windows terminal reset (console mode + ANSI).
 
-    Combines both ANSI reset and console mode reset for complete
+    Combines both console mode reset and ANSI reset for complete
     terminal state restoration after interrupts.
+
+    IMPORTANT: Console mode must be reset FIRST to re-enable
+    ENABLE_VIRTUAL_TERMINAL_PROCESSING, otherwise the ANSI escape
+    sequences will be printed as literal text (e.g., '[0m').
     """
     if platform.system() != "Windows":
         return
 
-    reset_windows_terminal_ansi()
-    reset_windows_console_mode()
+    reset_windows_console_mode()  # Must be first! Enables ANSI processing
+    reset_windows_terminal_ansi()  # Now ANSI escapes will be interpreted
 
 
 def reset_unix_terminal() -> None:

{code_puppy-0.0.329.dist-info → code_puppy-0.0.331.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: code-puppy
-Version: 0.0.329
+Version: 0.0.331
 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.329.dist-info → code_puppy-0.0.331.dist-info}/RECORD

@@ -3,7 +3,7 @@ code_puppy/__main__.py,sha256=pDVssJOWP8A83iFkxMLY9YteHYat0EyWDQqMkKHpWp4,203
 code_puppy/callbacks.py,sha256=hqTV--dNxG5vwWWm3MrEjmb8MZuHFFdmHePl23NXPHk,8621
 code_puppy/chatgpt_codex_client.py,sha256=Om0ANB_kpHubhCwNzF9ENf8RvKBqs0IYzBLl_SNw0Vk,9833
 code_puppy/claude_cache_client.py,sha256=hZr_YtXZSQvBoJFtRbbecKucYqJgoMopqUmm0IxFYGY,6071
-code_puppy/cli_runner.py,sha256=qIwmg3Di2dkxoRxsyd6o1uqd_wqkbOAe1OLQpzw80BU,29666
+code_puppy/cli_runner.py,sha256=9ioQkdt-J0ifG_b7tGn_aulde9RIJBOIv8Df_QGROx0,29991
 code_puppy/config.py,sha256=qqeJrQP7gqADqeYqVzfksP7NYGROLrBQCuYic5PuQfY,52295
 code_puppy/error_logging.py,sha256=a80OILCUtJhexI6a9GM-r5LqIdjvSRzggfgPp2jv1X0,3297
 code_puppy/gemini_code_assist.py,sha256=KGS7sO5OLc83nDF3xxS-QiU6vxW9vcm6hmzilu79Ef8,13867
@@ -21,7 +21,7 @@ code_puppy/round_robin_model.py,sha256=kSawwPUiPgg0yg8r4AAVgvjzsWkptxpSORd75-HP7
 code_puppy/session_storage.py,sha256=T4hOsAl9z0yz2JZCptjJBOnN8fCmkLZx5eLy1hTdv6Q,9631
 code_puppy/status_display.py,sha256=qHzIQGAPEa2_-4gQSg7_rE1ihOosBq8WO73MWFNmmlo,8938
 code_puppy/summarization_agent.py,sha256=6Pu_Wp_rF-HAhoX9u2uXTabRVkOZUYwRoMP1lzNS4ew,4485
-code_puppy/terminal_utils.py,sha256=oa8SF7Pel4o15Erd7a7vA7Y9wi6SJx196qxXITebDqA,3650
+code_puppy/terminal_utils.py,sha256=P6ProVD_3nHW0zHNRIMWTcg8HvBk9e0ir5bhUgJpT5k,3921
 code_puppy/version_checker.py,sha256=aq2Mwxl1CR9sEFBgrPt3OQOowLOBUp9VaQYWJhuUv8Q,1780
 code_puppy/agents/__init__.py,sha256=PtPB7Z5MSwmUKipgt_qxvIuGggcuVaYwNbnp1UP4tPc,518
 code_puppy/agents/agent_c_reviewer.py,sha256=1kO_89hcrhlS4sJ6elDLSEx-h43jAaWGgvIL0SZUuKo,8214
@@ -128,9 +128,9 @@ code_puppy/plugins/claude_code_oauth/README.md,sha256=76nHhMlhk61DZa5g0Q2fc0Atpp
 code_puppy/plugins/claude_code_oauth/SETUP.md,sha256=lnGzofPLogBy3oPPFLv5_cZ7vjg_GYrIyYnF-EoTJKg,3278
 code_puppy/plugins/claude_code_oauth/__init__.py,sha256=mCcOU-wM7LNCDjr-w-WLPzom8nTF1UNt4nqxGE6Rt0k,187
 code_puppy/plugins/claude_code_oauth/config.py,sha256=DjGySCkvjSGZds6DYErLMAi3TItt8iSLGvyJN98nSEM,2013
-code_puppy/plugins/claude_code_oauth/register_callbacks.py,sha256=0NeX1hhkYIlVfPmjZ1xmcf1yueDAJh_FMUmvJlxSO-E,10057
+code_puppy/plugins/claude_code_oauth/register_callbacks.py,sha256=szexPtlFvCiR098wJXXvbl0PUyr330Y5TUrq7ey6Ntc,10579
 code_puppy/plugins/claude_code_oauth/test_plugin.py,sha256=yQy4EeZl4bjrcog1d8BjknoDTRK75mRXXvkSQJYSSEM,9286
-code_puppy/plugins/claude_code_oauth/utils.py,sha256=wDaOU21zB3y6PWkuMXwE4mFjQuffyDae-vXysPTS-w8,13438
+code_puppy/plugins/claude_code_oauth/utils.py,sha256=vkkwsrcyBps25C_zvuK2Mquh5CMReQPReDRZ_gVOIZE,19901
 code_puppy/plugins/customizable_commands/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_puppy/plugins/customizable_commands/register_callbacks.py,sha256=zVMfIzr--hVn0IOXxIicbmgj2s-HZUgtrOc0NCDOnDw,5183
 code_puppy/plugins/example_custom_command/README.md,sha256=5c5Zkm7CW6BDSfe3WoLU7GW6t5mjjYAbu9-_pu-b3p4,8244
@@ -159,10 +159,10 @@ code_puppy/tools/browser/browser_scripts.py,sha256=sNb8eLEyzhasy5hV4B9OjM8yIVMLV
 code_puppy/tools/browser/browser_workflows.py,sha256=nitW42vCf0ieTX1gLabozTugNQ8phtoFzZbiAhw1V90,6491
 code_puppy/tools/browser/camoufox_manager.py,sha256=RZjGOEftE5sI_tsercUyXFSZI2wpStXf-q0PdYh2G3I,8680
 code_puppy/tools/browser/vqa_agent.py,sha256=DBn9HKloILqJSTSdNZzH_PYWT0B2h9VwmY6akFQI_uU,2913
-code_puppy-0.0.329.data/data/code_puppy/models.json,sha256=IPABdOrDw2OZJxa0XGBWSWmBRerV6_pIEmKVLRtUbAk,3105
-code_puppy-0.0.329.data/data/code_puppy/models_dev_api.json,sha256=wHjkj-IM_fx1oHki6-GqtOoCrRMR0ScK0f-Iz0UEcy8,548187
-code_puppy-0.0.329.dist-info/METADATA,sha256=PacPOzPdJBJx5yDJC2H5IJHS_Otpm7J0dGeyoDburBA,28854
-code_puppy-0.0.329.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-code_puppy-0.0.329.dist-info/entry_points.txt,sha256=Tp4eQC99WY3HOKd3sdvb22vZODRq0XkZVNpXOag_KdI,91
-code_puppy-0.0.329.dist-info/licenses/LICENSE,sha256=31u8x0SPgdOq3izJX41kgFazWsM43zPEF9eskzqbJMY,1075
-code_puppy-0.0.329.dist-info/RECORD,,
+code_puppy-0.0.331.data/data/code_puppy/models.json,sha256=IPABdOrDw2OZJxa0XGBWSWmBRerV6_pIEmKVLRtUbAk,3105
+code_puppy-0.0.331.data/data/code_puppy/models_dev_api.json,sha256=wHjkj-IM_fx1oHki6-GqtOoCrRMR0ScK0f-Iz0UEcy8,548187
+code_puppy-0.0.331.dist-info/METADATA,sha256=a_tpJr_TVCZGLAjZiyqiF5mG51Kni05wqqQuzVQmy-w,28854
+code_puppy-0.0.331.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+code_puppy-0.0.331.dist-info/entry_points.txt,sha256=Tp4eQC99WY3HOKd3sdvb22vZODRq0XkZVNpXOag_KdI,91
+code_puppy-0.0.331.dist-info/licenses/LICENSE,sha256=31u8x0SPgdOq3izJX41kgFazWsM43zPEF9eskzqbJMY,1075
+code_puppy-0.0.331.dist-info/RECORD,,