airbyte-agent-amazon-ads 0.1.24__py3-none-any.whl → 0.1.25__py3-none-any.whl

airbyte_agent_amazon_ads/_vendored/connector_sdk/cloud_utils/client.py

@@ -161,24 +161,80 @@ class AirbyteCloudClient:
         connector_id = connectors[0]["id"]
         return connector_id
 
+    async def initiate_oauth(
+        self,
+        definition_id: str,
+        external_user_id: str,
+        redirect_url: str,
+    ) -> str:
+        """Initiate a server-side OAuth flow.
+
+        Starts the OAuth flow for a connector. Returns a consent URL where the
+        end user should be redirected to grant access. After completing consent,
+        they'll be redirected to your redirect_url with a `server_side_oauth_secret_id`
+        query parameter that can be used with `create_source()`.
+
+        Args:
+            definition_id: Connector definition UUID
+            external_user_id: Workspace identifier
+            redirect_url: URL where users will be redirected after OAuth consent
+
+        Returns:
+            The OAuth consent URL
+
+        Raises:
+            httpx.HTTPStatusError: If the request fails
+
+        Example:
+            consent_url = await client.initiate_oauth(
+                definition_id="d8313939-3782-41b0-be29-b3ca20d8dd3a",
+                external_user_id="my-workspace",
+                redirect_url="https://myapp.com/oauth/callback",
+            )
+            # Redirect user to: consent_url
+            # After consent: https://myapp.com/oauth/callback?server_side_oauth_secret_id=...
+        """
+        token = await self.get_bearer_token()
+        url = f"{self.API_BASE_URL}/api/v1/integrations/connectors/oauth/initiate"
+        headers = {"Authorization": f"Bearer {token}"}
+        request_body = {
+            "external_user_id": external_user_id,
+            "definition_id": definition_id,
+            "redirect_url": redirect_url,
+        }
+
+        response = await self._http_client.post(url, json=request_body, headers=headers)
+        response.raise_for_status()
+        return response.json()["consent_url"]
+
     async def create_source(
         self,
         name: str,
         connector_definition_id: str,
         external_user_id: str,
-        credentials: dict[str, Any],
+        credentials: dict[str, Any] | None = None,
         replication_config: dict[str, Any] | None = None,
+        server_side_oauth_secret_id: str | None = None,
+        source_template_id: str | None = None,
     ) -> str:
-        """Create a new source on Airbyte Agent Platform.
+        """Create a new source on Airbyte Cloud.
+
+        Supports two authentication modes:
+        1. Direct credentials: Provide `credentials` dict
+        2. Server-side OAuth: Provide `server_side_oauth_secret_id` from OAuth flow
 
         Args:
             name: Source name
             connector_definition_id: UUID of the connector definition
             external_user_id: User identifier
-            credentials: Connector auth config dict
+            credentials: Connector auth config dict. Required unless using OAuth.
             replication_config: Optional replication settings (e.g., start_date for
                 connectors with x-airbyte-replication-config). Required for REPLICATION
                 mode sources like Intercom.
+            server_side_oauth_secret_id: OAuth secret ID from initiate_oauth redirect.
+                When provided, credentials are not required.
+            source_template_id: Source template ID. Required when organization has
+                multiple source templates for this connector type.
 
         Returns:
             The created source ID (UUID string)
@@ -187,19 +243,21 @@ class AirbyteCloudClient:
             httpx.HTTPStatusError: If creation fails
 
         Example:
+            # With direct credentials:
             source_id = await client.create_source(
-                name="My Gong Source",
-                connector_definition_id="32382e40-3b49-4b99-9c5c-4076501914e7",
+                name="My Intercom Source",
+                connector_definition_id="d8313939-3782-41b0-be29-b3ca20d8dd3a",
                 external_user_id="my-workspace",
-                credentials={"access_key": "...", "access_key_secret": "..."}
+                credentials={"access_token": "..."},
+                replication_config={"start_date": "2024-01-01T00:00:00Z"}
             )
 
-            # For REPLICATION mode sources (e.g., Intercom):
+            # With server-side OAuth:
             source_id = await client.create_source(
                 name="My Intercom Source",
                 connector_definition_id="d8313939-3782-41b0-be29-b3ca20d8dd3a",
                 external_user_id="my-workspace",
-                credentials={"access_token": "..."},
+                server_side_oauth_secret_id="airbyte_oauth_..._secret_...",
                 replication_config={"start_date": "2024-01-01T00:00:00Z"}
             )
         """
@@ -211,11 +269,16 @@ class AirbyteCloudClient:
             "name": name,
             "definition_id": connector_definition_id,
             "external_user_id": external_user_id,
-            "credentials": credentials,
         }
 
+        if credentials is not None:
+            request_body["credentials"] = credentials
         if replication_config is not None:
             request_body["replication_config"] = replication_config
+        if server_side_oauth_secret_id is not None:
+            request_body["server_side_oauth_secret_id"] = server_side_oauth_secret_id
+        if source_template_id is not None:
+            request_body["source_template_id"] = source_template_id
 
         response = await self._http_client.post(url, json=request_body, headers=headers)
         response.raise_for_status()

airbyte_agent_amazon_ads/connector.py

@@ -508,6 +508,59 @@ class AmazonAdsConnector:
 
     # ===== HOSTED MODE FACTORY =====
 
+    @classmethod
+    async def initiate_oauth(
+        cls,
+        *,
+        external_user_id: str,
+        redirect_url: str,
+        airbyte_client_id: str,
+        airbyte_client_secret: str,
+    ) -> str:
+        """
+        Initiate server-side OAuth flow for this connector.
+
+        Returns a consent URL where the end user should be redirected to grant access.
+        After completing consent, they'll be redirected to your redirect_url with a
+        `server_side_oauth_secret_id` query parameter that can be used with `create_hosted()`.
+
+        Args:
+            external_user_id: Workspace identifier in Airbyte Cloud
+            redirect_url: URL where users will be redirected after OAuth consent
+            airbyte_client_id: Airbyte OAuth client ID
+            airbyte_client_secret: Airbyte OAuth client secret
+
+        Returns:
+            The OAuth consent URL
+
+        Example:
+            consent_url = await AmazonAdsConnector.initiate_oauth(
+                external_user_id="my-workspace",
+                redirect_url="https://myapp.com/oauth/callback",
+                airbyte_client_id="client_abc",
+                airbyte_client_secret="secret_xyz",
+            )
+            # Redirect user to: consent_url
+            # After consent, user arrives at: https://myapp.com/oauth/callback?server_side_oauth_secret_id=...
+        """
+        from ._vendored.connector_sdk.cloud_utils import AirbyteCloudClient
+
+        client = AirbyteCloudClient(
+            client_id=airbyte_client_id,
+            client_secret=airbyte_client_secret,
+        )
+
+        try:
+            consent_url = await client.initiate_oauth(
+                definition_id=str(AmazonAdsConnectorModel.id),
+                external_user_id=external_user_id,
+                redirect_url=redirect_url,
+            )
+        finally:
+            await client.close()
+
+        return consent_url
+
     @classmethod
     async def create_hosted(
         cls,
@@ -515,9 +568,11 @@ class AmazonAdsConnector:
         external_user_id: str,
         airbyte_client_id: str,
         airbyte_client_secret: str,
-        auth_config: "AmazonAdsAuthConfig",
+        auth_config: "AmazonAdsAuthConfig" | None = None,
+        server_side_oauth_secret_id: str | None = None,
         name: str | None = None,
         replication_config: dict[str, Any] | None = None,
+        source_template_id: str | None = None,
     ) -> "AmazonAdsConnector":
         """
         Create a new hosted connector on Airbyte Cloud.
@@ -526,18 +581,29 @@ class AmazonAdsConnector:
         1. Creates a source on Airbyte Cloud with the provided credentials
         2. Returns a connector configured with the new connector_id
 
+        Supports two authentication modes:
+        1. Direct credentials: Provide `auth_config` with typed credentials
+        2. Server-side OAuth: Provide `server_side_oauth_secret_id` from OAuth flow
+
         Args:
             external_user_id: Workspace identifier in Airbyte Cloud
             airbyte_client_id: Airbyte OAuth client ID
             airbyte_client_secret: Airbyte OAuth client secret
-            auth_config: Typed auth config (same as local mode)
+            auth_config: Typed auth config. Required unless using server_side_oauth_secret_id.
+            server_side_oauth_secret_id: OAuth secret ID from initiate_oauth redirect.
+                When provided, auth_config is not required.
             name: Optional source name (defaults to connector name + external_user_id)
             replication_config: Optional replication settings dict.
                 Required for connectors with x-airbyte-replication-config (REPLICATION mode sources).
+            source_template_id: Source template ID. Required when organization has
+                multiple source templates for this connector type.
 
         Returns:
             A AmazonAdsConnector instance configured in hosted mode
 
+        Raises:
+            ValueError: If neither or both auth_config and server_side_oauth_secret_id provided
+
         Example:
             # Create a new hosted connector with API key auth
             connector = await AmazonAdsConnector.create_hosted(
@@ -547,9 +613,27 @@ class AmazonAdsConnector:
                 auth_config=AmazonAdsAuthConfig(client_id="...", client_secret="...", refresh_token="..."),
             )
 
+            # With server-side OAuth:
+            connector = await AmazonAdsConnector.create_hosted(
+                external_user_id="my-workspace",
+                airbyte_client_id="client_abc",
+                airbyte_client_secret="secret_xyz",
+                server_side_oauth_secret_id="airbyte_oauth_..._secret_...",
+            )
+
             # Use the connector
             result = await connector.execute("entity", "list", {})
         """
+        # Validate: exactly one of auth_config or server_side_oauth_secret_id required
+        if auth_config is None and server_side_oauth_secret_id is None:
+            raise ValueError(
+                "Either auth_config or server_side_oauth_secret_id must be provided"
+            )
+        if auth_config is not None and server_side_oauth_secret_id is not None:
+            raise ValueError(
+                "Cannot provide both auth_config and server_side_oauth_secret_id"
+            )
+
         from ._vendored.connector_sdk.cloud_utils import AirbyteCloudClient
 
         client = AirbyteCloudClient(
@@ -558,8 +642,8 @@ class AmazonAdsConnector:
         )
 
         try:
-            # Build credentials from auth_config
-            credentials = auth_config.model_dump(exclude_none=True)
+            # Build credentials from auth_config (if provided)
+            credentials = auth_config.model_dump(exclude_none=True) if auth_config else None
             replication_config_dict = replication_config.model_dump(exclude_none=True) if replication_config else None
 
             # Create source on Airbyte Cloud
@@ -570,6 +654,8 @@ class AmazonAdsConnector:
                 external_user_id=external_user_id,
                 credentials=credentials,
                 replication_config=replication_config_dict,
+                server_side_oauth_secret_id=server_side_oauth_secret_id,
+                source_template_id=source_template_id,
             )
         finally:
             await client.close()
@@ -583,6 +669,7 @@ class AmazonAdsConnector:
 
 
 
+
 class ProfilesQuery:
     """
     Query class for Profiles entity operations.

{airbyte_agent_amazon_ads-0.1.24.dist-info → airbyte_agent_amazon_ads-0.1.25.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: airbyte-agent-amazon-ads
-Version: 0.1.24
+Version: 0.1.25
 Summary: Airbyte Amazon-Ads Connector for AI platforms
 Project-URL: Homepage, https://github.com/airbytehq/airbyte-agent-connectors
 Project-URL: Documentation, https://docs.airbyte.com/ai-agents/
@@ -135,7 +135,7 @@ See the official [Amazon-Ads API reference](https://advertising.amazon.com/API/d
 
 ## Version information
 
-- **Package version:** 0.1.24
+- **Package version:** 0.1.25
 - **Connector version:** 1.0.5
-- **Generated with Connector SDK commit SHA:** b184da3e22ef8521d2eeebf3c96a0fe8da2424f5
+- **Generated with Connector SDK commit SHA:** 9d9866b0aae8c3494d04d34e193b9bd860bfc1c6
 - **Changelog:** [View changelog](https://github.com/airbytehq/airbyte-agent-connectors/blob/main/connectors/amazon-ads/CHANGELOG.md)

{airbyte_agent_amazon_ads-0.1.24.dist-info → airbyte_agent_amazon_ads-0.1.25.dist-info}/RECORD

@@ -1,5 +1,5 @@
 airbyte_agent_amazon_ads/__init__.py,sha256=MICR9abWiXLcSQ0jd4W7S6xyo2KgwRjOPIZTeHpz67I,1950
-airbyte_agent_amazon_ads/connector.py,sha256=emUP1S6OIO0WowecjbiUtCnUB2vopT7S68AIppHb8ls,29494
+airbyte_agent_amazon_ads/connector.py,sha256=9r5mxeos61jzP-tTNJf4UB2h6XCXoMs5T5E7r2xhqFg,33158
 airbyte_agent_amazon_ads/connector_model.py,sha256=QAxt_l2-RjUBxg2izmclOPTqVrjF9CL70fOkkYcwpTk,98041
 airbyte_agent_amazon_ads/models.py,sha256=KmYWz46e-8HuQdnuz6FQT99uDKwpcToEc9Xda2M7FL4,9781
 airbyte_agent_amazon_ads/types.py,sha256=EW-jvOmXyJQqe2ESiDCs7nCIIsFreDCIZBlA6aplxpA,6728
@@ -19,7 +19,7 @@ airbyte_agent_amazon_ads/_vendored/connector_sdk/utils.py,sha256=UYwYuSLhsDD-4C0
 airbyte_agent_amazon_ads/_vendored/connector_sdk/validation.py,sha256=w5WGnmILkdBslpXhAXhKhE-c8ANBc_OZQxr_fUeAgtc,39666
 airbyte_agent_amazon_ads/_vendored/connector_sdk/validation_replication.py,sha256=v7F5YWd5m4diIF7_4m4nOkC9crg97vqRUUkt9ka9HZ4,36043
 airbyte_agent_amazon_ads/_vendored/connector_sdk/cloud_utils/__init__.py,sha256=4799Hv9f2zxDVj1aLyQ8JpTEuFTp_oOZMRz-NZCdBJg,134
-airbyte_agent_amazon_ads/_vendored/connector_sdk/cloud_utils/client.py,sha256=-_ibaHVa7KBBw8SnMuxpWz6XkrSgNTFdMgDfTChtywg,9505
+airbyte_agent_amazon_ads/_vendored/connector_sdk/cloud_utils/client.py,sha256=e0VLNCmesGGfo2uD0GiICgXsXTeTkh0GYiVgx_e4VEc,12296
 airbyte_agent_amazon_ads/_vendored/connector_sdk/executor/__init__.py,sha256=EmG9YQNAjSuYCVB4D5VoLm4qpD1KfeiiOf7bpALj8p8,702
 airbyte_agent_amazon_ads/_vendored/connector_sdk/executor/hosted_executor.py,sha256=tv0njAdy-gdHBg4izgcxhEWYbrNiBifEYEca9AWzaL0,8693
 airbyte_agent_amazon_ads/_vendored/connector_sdk/executor/local_executor.py,sha256=RtdTXFzfoJz5Coz9nwQi81Df1402BRgO1Mgd3ZzTkfw,76581
@@ -53,6 +53,6 @@ airbyte_agent_amazon_ads/_vendored/connector_sdk/telemetry/__init__.py,sha256=Ra
 airbyte_agent_amazon_ads/_vendored/connector_sdk/telemetry/config.py,sha256=tLmQwAFD0kP1WyBGWBS3ysaudN9H3e-3EopKZi6cGKg,885
 airbyte_agent_amazon_ads/_vendored/connector_sdk/telemetry/events.py,sha256=8Y1NbXiwISX-V_wRofY7PqcwEXD0dLMnntKkY6XFU2s,1328
 airbyte_agent_amazon_ads/_vendored/connector_sdk/telemetry/tracker.py,sha256=SginFQbHqVUVYG82NnNzG34O-tAQ_wZYjGDcuo0q4Kk,5584
-airbyte_agent_amazon_ads-0.1.24.dist-info/METADATA,sha256=izTt3eLQjsnHCbUprS1Dgo2oUgynI7l425iKamEW6Sc,5326
-airbyte_agent_amazon_ads-0.1.24.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-airbyte_agent_amazon_ads-0.1.24.dist-info/RECORD,,
+airbyte_agent_amazon_ads-0.1.25.dist-info/METADATA,sha256=mgsYZFR2QZ1zOj3x28WcdeuxnD6HibWamIMoq6vwF3o,5326
+airbyte_agent_amazon_ads-0.1.25.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+airbyte_agent_amazon_ads-0.1.25.dist-info/RECORD,,