airbyte-agent-orb 0.1.5__py3-none-any.whl → 0.1.7__py3-none-any.whl

airbyte_agent_orb/_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_orb/_vendored/connector_sdk/schema/base.py

@@ -10,7 +10,7 @@ from enum import StrEnum
 from typing import Any, Dict
 from uuid import UUID
 
-from pydantic import BaseModel, ConfigDict, Field, field_validator
+from pydantic import BaseModel, ConfigDict, Field, field_validator, model_validator
 from pydantic_core import Url
 
 from .extensions import CacheConfig, ReplicationConfig, RetryConfig
@@ -210,3 +210,34 @@ class Server(BaseModel):
             raise ValueError("Server URL cannot be empty")
         # Allow both absolute URLs and relative paths
         return v
+
+    @model_validator(mode="after")
+    def validate_replication_environment_mapping(self) -> "Server":
+        """Validate that x-airbyte-replication-environment-mapping sources exist in variables.
+
+        For simple mappings like {"subdomain": "subdomain"}, the key is the source variable.
+        For transform mappings like {"domain": {"source": "subdomain", "format": "..."}},
+        the "source" field is the source variable.
+        """
+        env_mapping = self.x_airbyte_replication_environment_mapping
+        if not env_mapping or not self.variables:
+            return self
+
+        variable_names = set(self.variables.keys())
+
+        for env_key, mapping_value in env_mapping.items():
+            if isinstance(mapping_value, str):
+                source_var = env_key
+            elif isinstance(mapping_value, EnvironmentMappingTransform):
+                source_var = mapping_value.source
+            else:
+                continue
+
+            if source_var not in variable_names:
+                available = ", ".join(sorted(variable_names)) if variable_names else "(none)"
+                raise ValueError(
+                    f"x-airbyte-replication-environment-mapping: source variable '{source_var}' "
+                    f"not found in server variables. Available: {available}"
+                )
+
+        return self

airbyte_agent_orb/_vendored/connector_sdk/schema/extensions.py

@@ -14,7 +14,7 @@ are implemented.
 
 from typing import Literal
 
-from pydantic import BaseModel, ConfigDict, Field
+from pydantic import BaseModel, ConfigDict, Field, model_validator
 
 
 class PaginationConfig(BaseModel):
@@ -252,6 +252,24 @@ class ReplicationConfig(BaseModel):
         description="Mapping from replication_config field names to source_config field names",
     )
 
+    @model_validator(mode="after")
+    def validate_replication_config_key_mapping(self) -> "ReplicationConfig":
+        """Validate that replication_config_key_mapping keys exist in properties.
+
+        The mapping is: {local_key: airbyte_path}
+        We validate that local_key exists in our properties.
+        """
+        if self.replication_config_key_mapping and self.properties:
+            property_names = set(self.properties.keys())
+            for local_key, airbyte_path in self.replication_config_key_mapping.items():
+                if local_key not in property_names:
+                    available = ", ".join(sorted(property_names)) if property_names else "(none)"
+                    raise ValueError(
+                        f"replication_config_key_mapping: local key '{local_key}' "
+                        f"(mapped to '{airbyte_path}') not found in properties. Available: {available}"
+                    )
+        return self
+
 
 class CacheConfig(BaseModel):
     """

airbyte_agent_orb/_vendored/connector_sdk/schema/security.py

@@ -81,6 +81,21 @@ class AuthConfigOption(BaseModel):
         description="Mapping from source config paths (e.g., 'credentials.api_key') to auth config keys for direct connectors",
     )
 
+    @model_validator(mode="after")
+    def validate_replication_auth_key_mapping(self) -> "AuthConfigOption":
+        """Validate that replication_auth_key_mapping target keys exist in properties."""
+        if self.replication_auth_key_mapping and self.properties:
+            property_names = set(self.properties.keys())
+            for airbyte_path, our_key in self.replication_auth_key_mapping.items():
+                if our_key not in property_names:
+                    option_context = f"oneOf option '{self.title}'" if self.title else "oneOf option"
+                    available = ", ".join(sorted(property_names)) if property_names else "(none)"
+                    raise ValueError(
+                        f"replication_auth_key_mapping in {option_context}: target key '{our_key}' "
+                        f"(mapped from '{airbyte_path}') not found in properties. Available: {available}"
+                    )
+        return self
+
 
 class AirbyteAuthConfig(BaseModel):
     """
@@ -146,8 +161,30 @@ class AirbyteAuthConfig(BaseModel):
             if not self.auth_mapping:
                 raise ValueError("Single auth option must have auth_mapping")
 
+            # Validate replication_auth_key_mapping targets exist in properties
+            if self.replication_auth_key_mapping and self.properties:
+                self._validate_replication_auth_key_mapping(self.replication_auth_key_mapping, self.properties, context="x-airbyte-auth-config")
+
         return self
 
+    @staticmethod
+    def _validate_replication_auth_key_mapping(mapping: Dict[str, str], properties: Dict[str, AuthConfigFieldSpec], context: str) -> None:
+        """Validate that replication_auth_key_mapping target keys exist in properties.
+
+        Args:
+            mapping: The replication_auth_key_mapping dict (airbyte_path -> our_key)
+            properties: The properties dict from x-airbyte-auth-config
+            context: Context string for error messages
+        """
+        property_names = set(properties.keys())
+        for airbyte_path, our_key in mapping.items():
+            if our_key not in property_names:
+                available = ", ".join(sorted(property_names)) if property_names else "(none)"
+                raise ValueError(
+                    f"replication_auth_key_mapping in {context}: target key '{our_key}' "
+                    f"(mapped from '{airbyte_path}') not found in properties. Available: {available}"
+                )
+
 
 class SecurityScheme(BaseModel):
     """

airbyte_agent_orb/connector.py

@@ -543,6 +543,7 @@ class OrbConnector:
         auth_config: "OrbAuthConfig",
         name: str | None = None,
         replication_config: dict[str, Any] | None = None,
+        source_template_id: str | None = None,
     ) -> "OrbConnector":
         """
         Create a new hosted connector on Airbyte Cloud.
@@ -559,6 +560,8 @@ class OrbConnector:
             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 OrbConnector instance configured in hosted mode
@@ -575,6 +578,7 @@ class OrbConnector:
             # Use the connector
             result = await connector.execute("entity", "list", {})
         """
+
         from ._vendored.connector_sdk.cloud_utils import AirbyteCloudClient
 
         client = AirbyteCloudClient(
@@ -583,8 +587,8 @@ class OrbConnector:
         )
 
         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
@@ -595,6 +599,7 @@ class OrbConnector:
                 external_user_id=external_user_id,
                 credentials=credentials,
                 replication_config=replication_config_dict,
+                source_template_id=source_template_id,
             )
         finally:
             await client.close()
@@ -608,6 +613,7 @@ class OrbConnector:
 
 
 
+
 class CustomersQuery:
     """
     Query class for Customers entity operations.

{airbyte_agent_orb-0.1.5.dist-info → airbyte_agent_orb-0.1.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: airbyte-agent-orb
-Version: 0.1.5
+Version: 0.1.7
 Summary: Airbyte Orb Connector for AI platforms
 Project-URL: Homepage, https://github.com/airbytehq/airbyte-agent-connectors
 Project-URL: Documentation, https://docs.airbyte.com/ai-agents/
@@ -147,7 +147,7 @@ See the official [Orb API reference](https://docs.withorb.com/api-reference).
 
 ## Version information
 
-- **Package version:** 0.1.5
+- **Package version:** 0.1.7
 - **Connector version:** 0.1.1
-- **Generated with Connector SDK commit SHA:** b184da3e22ef8521d2eeebf3c96a0fe8da2424f5
+- **Generated with Connector SDK commit SHA:** 940246757c7476ed4edd7d16b873ebe54ea2b456
 - **Changelog:** [View changelog](https://github.com/airbytehq/airbyte-agent-connectors/blob/main/connectors/orb/CHANGELOG.md)

{airbyte_agent_orb-0.1.5.dist-info → airbyte_agent_orb-0.1.7.dist-info}/RECORD

@@ -1,5 +1,5 @@
 airbyte_agent_orb/__init__.py,sha256=jHWqgCDpIiDZB1WLwcKo4aNXkp5ypSdV0WQWkCNJv_Y,3225
-airbyte_agent_orb/connector.py,sha256=KYeYW2DnDEYKMzp_nTdD17ydcB5Oh1Dwz0NNLIXswng,41204
+airbyte_agent_orb/connector.py,sha256=dCsSG2QFX4hdIyMJzNiP0sCj4xWoAKAKKrGxH83AVmc,41497
 airbyte_agent_orb/connector_model.py,sha256=mhxwbi31hLA-oU2I8oQq7uMK4S0KV8XGjCELQRGth50,120056
 airbyte_agent_orb/models.py,sha256=dFg7AeDxlgoGO8eU9ypRTLqULqSAHDA3cuaXKEKbsyk,22410
 airbyte_agent_orb/types.py,sha256=O6Q2AOp8BYjYhpEFvOZXnzQDf-7GAKnpCorER2Puh2E,37077
@@ -19,7 +19,7 @@ airbyte_agent_orb/_vendored/connector_sdk/utils.py,sha256=UYwYuSLhsDD-4C0dBs7Qy0
 airbyte_agent_orb/_vendored/connector_sdk/validation.py,sha256=w5WGnmILkdBslpXhAXhKhE-c8ANBc_OZQxr_fUeAgtc,39666
 airbyte_agent_orb/_vendored/connector_sdk/validation_replication.py,sha256=v7F5YWd5m4diIF7_4m4nOkC9crg97vqRUUkt9ka9HZ4,36043
 airbyte_agent_orb/_vendored/connector_sdk/cloud_utils/__init__.py,sha256=4799Hv9f2zxDVj1aLyQ8JpTEuFTp_oOZMRz-NZCdBJg,134
-airbyte_agent_orb/_vendored/connector_sdk/cloud_utils/client.py,sha256=-_ibaHVa7KBBw8SnMuxpWz6XkrSgNTFdMgDfTChtywg,9505
+airbyte_agent_orb/_vendored/connector_sdk/cloud_utils/client.py,sha256=e0VLNCmesGGfo2uD0GiICgXsXTeTkh0GYiVgx_e4VEc,12296
 airbyte_agent_orb/_vendored/connector_sdk/executor/__init__.py,sha256=EmG9YQNAjSuYCVB4D5VoLm4qpD1KfeiiOf7bpALj8p8,702
 airbyte_agent_orb/_vendored/connector_sdk/executor/hosted_executor.py,sha256=tv0njAdy-gdHBg4izgcxhEWYbrNiBifEYEca9AWzaL0,8693
 airbyte_agent_orb/_vendored/connector_sdk/executor/local_executor.py,sha256=RtdTXFzfoJz5Coz9nwQi81Df1402BRgO1Mgd3ZzTkfw,76581
@@ -43,16 +43,16 @@ airbyte_agent_orb/_vendored/connector_sdk/performance/__init__.py,sha256=Sp5fSd1
 airbyte_agent_orb/_vendored/connector_sdk/performance/instrumentation.py,sha256=_dXvNiqdndIBwDjeDKNViWzn_M5FkSUsMmJtFldrmsM,1504
 airbyte_agent_orb/_vendored/connector_sdk/performance/metrics.py,sha256=FRff7dKt4iwt_A7pxV5n9kAGBR756PC7q8-weWygPSM,2817
 airbyte_agent_orb/_vendored/connector_sdk/schema/__init__.py,sha256=Uymu-QuzGJuMxexBagIvUxpVAigIuIhz3KeBl_Vu4Ko,1638
-airbyte_agent_orb/_vendored/connector_sdk/schema/base.py,sha256=mOO5eZSK-FB7S-ZXpt5HFG5YBg8x-oM6RZRLPOEGxZM,7115
+airbyte_agent_orb/_vendored/connector_sdk/schema/base.py,sha256=0R0aR4HjcJpu3vN0jZ4nPBEwVZ_7J9_XLQgWaQGT3Fs,8476
 airbyte_agent_orb/_vendored/connector_sdk/schema/components.py,sha256=nJIPieavwX3o3ODvdtLHPk84d_V229xmg6LDfwEHjzc,8119
 airbyte_agent_orb/_vendored/connector_sdk/schema/connector.py,sha256=mSZk1wr2YSdRj9tTRsPAuIlCzd_xZLw-Bzl1sMwE0rE,3731
-airbyte_agent_orb/_vendored/connector_sdk/schema/extensions.py,sha256=5hgpFHK7fzpzegCkJk882DeIP79bCx_qairKJhvPMZ8,9590
+airbyte_agent_orb/_vendored/connector_sdk/schema/extensions.py,sha256=DqWCvyWKLnPxAbJirq6uITf-MBeJRVCcsEjySF4fuhQ,10558
 airbyte_agent_orb/_vendored/connector_sdk/schema/operations.py,sha256=St-A75m6sZUZlsoM6WcoPaShYu_X1K19pdyPvJbabOE,6214
-airbyte_agent_orb/_vendored/connector_sdk/schema/security.py,sha256=R-21DLnp-ANIRO1Dzqo53TYFJL6lCp0aO8GSuxa_bDI,9225
+airbyte_agent_orb/_vendored/connector_sdk/schema/security.py,sha256=5UeaN63InuInS_XHNZSm5yDXrMLm5lhz298l-3GSc_4,11435
 airbyte_agent_orb/_vendored/connector_sdk/telemetry/__init__.py,sha256=RaLgkBU4dfxn1LC5Y0Q9rr2PJbrwjxvPgBLmq8_WafE,211
 airbyte_agent_orb/_vendored/connector_sdk/telemetry/config.py,sha256=tLmQwAFD0kP1WyBGWBS3ysaudN9H3e-3EopKZi6cGKg,885
 airbyte_agent_orb/_vendored/connector_sdk/telemetry/events.py,sha256=8Y1NbXiwISX-V_wRofY7PqcwEXD0dLMnntKkY6XFU2s,1328
 airbyte_agent_orb/_vendored/connector_sdk/telemetry/tracker.py,sha256=SginFQbHqVUVYG82NnNzG34O-tAQ_wZYjGDcuo0q4Kk,5584
-airbyte_agent_orb-0.1.5.dist-info/METADATA,sha256=z_-GPecIUiy42GQZ_Yyrim1v_-J9O91ZNMPoBLVSnnw,5670
-airbyte_agent_orb-0.1.5.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-airbyte_agent_orb-0.1.5.dist-info/RECORD,,
+airbyte_agent_orb-0.1.7.dist-info/METADATA,sha256=I_Biiotj9EZ2BMBPaxWxlCDt8h57_CjyGn3d95gEt_0,5670
+airbyte_agent_orb-0.1.7.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+airbyte_agent_orb-0.1.7.dist-info/RECORD,,