code-puppy 0.0.783__tar.gz → 0.0.784__tar.gz

{code_puppy-0.0.783 → code_puppy-0.0.784}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: code-puppy
-Version: 0.0.783
+Version: 0.0.784
 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.783 → code_puppy-0.0.784}/code_puppy/callbacks.py

@@ -49,6 +49,7 @@ PhaseType = Literal[
     "agent_run_end",
     "agent_run_result",
     "model_select",
+    "provider_credential_flow",
     "register_mcp_catalog_servers",
     "register_browser_types",
     "register_model_providers",
@@ -138,6 +139,7 @@ _callbacks: Dict[PhaseType, List[CallbackFunc]] = {
     "agent_run_end": [],
     "agent_run_result": [],
     "model_select": [],
+    "provider_credential_flow": [],
     "register_mcp_catalog_servers": [],
     "register_browser_types": [],
     "register_model_providers": [],
@@ -1225,6 +1227,44 @@ def on_model_select(
     return None
 
 
+def on_provider_credential_flow(*, provider_id: str, env_var: str) -> bool:
+    """Ask plugins to acquire a missing provider credential.
+
+    Fired by the ``/add_model`` flow for each required env var that is not
+    set, before falling back to manual key entry. A plugin that can mint the
+    credential itself (e.g. an OAuth flow that exchanges an authorization
+    code for an API key) should persist it so it takes effect immediately
+    (config + ``os.environ``) and return ``True``. Return ``None``/``False``
+    to defer to manual entry.
+
+    Unlike most hooks this one deliberately short-circuits: callbacks run
+    interactive, side-effectful flows, so the first one to return ``True``
+    wins and the rest are never invoked. Callbacks must gate on
+    ``provider_id``/``env_var`` and return ``None`` immediately for
+    providers they do not own.
+
+    Args:
+        provider_id: models.dev provider id (e.g. ``"openrouter"``).
+        env_var: The credential env var being requested.
+
+    Returns:
+        True when a callback reports the credential as acquired.
+    """
+    for callback in get_callbacks("provider_credential_flow"):
+        try:
+            result = callback(provider_id=provider_id, env_var=env_var)
+        except Exception as e:
+            logger.error(
+                f"Callback {getattr(callback, '__name__', callback)!r} failed "
+                f"in phase 'provider_credential_flow': {e}\n"
+                f"{traceback.format_exc()}"
+            )
+            continue
+        if result is True:
+            return True
+    return False
+
+
 async def on_agent_run_end(
     agent_name: str,
     model_name: str,

{code_puppy-0.0.783 → code_puppy-0.0.784}/code_puppy/command_line/add_model_menu.py

@@ -346,16 +346,34 @@ def _text_input(title: str, **kwargs):
     return builder
 
 
+def _plugin_credential_flow(provider: ProviderInfo, env_var: str) -> bool:
+    """Give plugins first crack at acquiring a missing credential.
+
+    A plugin (e.g. an OAuth flow that exchanges a code for an API key) saves
+    the credential itself and returns True via the ``provider_credential_flow``
+    hook. Trust but verify: the env var must actually be set afterwards,
+    otherwise we fall back to manual entry anyway.
+    """
+    from code_puppy.callbacks import on_provider_credential_flow
+
+    handled = on_provider_credential_flow(provider_id=provider.id, env_var=env_var)
+    return bool(handled and os.environ.get(env_var))
+
+
 def prompt_for_credentials(provider: ProviderInfo, **overrides) -> bool:
     """Prompt for each missing credential. Empty skips; Esc cancels all.
 
-    Saves via config + os.environ so the key is immediately usable.
+    Plugins get first crack via the ``provider_credential_flow`` hook (e.g.
+    browser OAuth); anything unhandled falls back to manual entry, which
+    saves via config + os.environ so the key is immediately usable.
     """
     missing = missing_env_vars(provider)
     if not missing:
         emit_info(t("model_menu.credentials.all_set", provider=provider.name))
         return True
     for env_var in missing:
+        if _plugin_credential_flow(provider, env_var):
+            continue
         result = (
             _text_input(
                 f"{provider.name}: {env_var}",

{code_puppy-0.0.783 → code_puppy-0.0.784}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "code-puppy"
-version = "0.0.783"
+version = "0.0.784"
 description = "Code generation agent"
 readme = "README.md"
 requires-python = ">=3.11,<3.15"