feat: Allow managed auth logins to disable learned skills

Stainless-Generated-From: 9c2ffa1553e5fe09f6a36fa423af3e7308eb7d14

Author: kernel-internal[bot]

src/kernel/resources/auth/connections.py

@@ -477,6 +477,7 @@ def login(
         browser_telemetry: Optional[connection_login_params.BrowserTelemetry] | Omit = omit,
         proxy: connection_login_params.Proxy | Omit = omit,
         record_session: bool | Omit = omit,
+        skill_mode: Literal["enabled", "disabled"] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -502,6 +503,10 @@ def login(
           record_session: Override the connection's default for recording this login's browser session.
               When omitted, the connection's record_session default is used.
 
+          skill_mode: Controls whether this login reads and writes learned domain skills. Automatic
+              reauths inherit the selected mode until a later accepted login sets enabled or
+              omits this field. Defaults to enabled when omitted.
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -520,6 +525,7 @@ def login(
                     "browser_telemetry": browser_telemetry,
                     "proxy": proxy,
                     "record_session": record_session,
+                    "skill_mode": skill_mode,
                 },
                 connection_login_params.ConnectionLoginParams,
             ),
@@ -1100,6 +1106,7 @@ async def login(
         browser_telemetry: Optional[connection_login_params.BrowserTelemetry] | Omit = omit,
         proxy: connection_login_params.Proxy | Omit = omit,
         record_session: bool | Omit = omit,
+        skill_mode: Literal["enabled", "disabled"] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -1125,6 +1132,10 @@ async def login(
           record_session: Override the connection's default for recording this login's browser session.
               When omitted, the connection's record_session default is used.
 
+          skill_mode: Controls whether this login reads and writes learned domain skills. Automatic
+              reauths inherit the selected mode until a later accepted login sets enabled or
+              omits this field. Defaults to enabled when omitted.
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -1143,6 +1154,7 @@ async def login(
                     "browser_telemetry": browser_telemetry,
                     "proxy": proxy,
                     "record_session": record_session,
+                    "skill_mode": skill_mode,
                 },
                 connection_login_params.ConnectionLoginParams,
             ),

src/kernel/types/auth/connection_login_params.py

@@ -3,7 +3,7 @@
 from __future__ import annotations
 
 from typing import Optional
-from typing_extensions import TypedDict
+from typing_extensions import Literal, TypedDict
 
 from .managed_auth_browser_config_param import ManagedAuthBrowserConfigParam
 from ..browsers.browser_telemetry_categories_config_param import BrowserTelemetryCategoriesConfigParam
@@ -40,6 +40,13 @@ class ConnectionLoginParams(TypedDict, total=False):
     When omitted, the connection's record_session default is used.
     """
 
+    skill_mode: Literal["enabled", "disabled"]
+    """Controls whether this login reads and writes learned domain skills.
+
+    Automatic reauths inherit the selected mode until a later accepted login sets
+    enabled or omits this field. Defaults to enabled when omitted.
+    """
+
 
 class BrowserTelemetryExportOtlpDestination(TypedDict, total=False):
     """OTLP destination to export this session's captured telemetry to.

tests/api_resources/auth/test_connections.py

@@ -525,6 +525,7 @@ def test_method_login_with_all_params(self, client: Kernel) -> None:
                 "name": "name",
             },
             record_session=True,
+            skill_mode="enabled",
         )
         assert_matches_type(LoginResponse, connection, path=["response"])
 
@@ -1187,6 +1188,7 @@ async def test_method_login_with_all_params(self, async_client: AsyncKernel) ->
                 "name": "name",
             },
             record_session=True,
+            skill_mode="enabled",
         )
         assert_matches_type(LoginResponse, connection, path=["response"])