airbyte-agent-mcp 0.1.16__tar.gz → 0.1.51__tar.gz

{airbyte_agent_mcp-0.1.16 → airbyte_agent_mcp-0.1.51}/.gitignore

@@ -24,3 +24,6 @@ venv/
 # OS
 .DS_Store
 Thumbs.db
+
+.env
+configured_connectors.yaml

airbyte_agent_mcp-0.1.51/CHANGELOG.md

@@ -0,0 +1,261 @@
+# Changelog
+
+## [0.1.51] - 2026-01-14
+- Updated airbyte-agent-mcp package
+- Source commit: 6504630b
+- SDK version: 0.1.0
+
+## [0.1.50] - 2026-01-13
+- Updated airbyte-agent-mcp package
+- Source commit: e80a226e
+- SDK version: 0.1.0
+
+## [0.1.49] - 2026-01-13
+- Updated airbyte-agent-mcp package
+- Source commit: 78b1be67
+- SDK version: 0.1.0
+
+## [0.1.48] - 2026-01-11
+- Updated airbyte-agent-mcp package
+- Source commit: e519b73d
+- SDK version: 0.1.0
+
+## [0.1.47] - 2026-01-09
+- Updated airbyte-agent-mcp package
+- Source commit: 3c7bfdfd
+- SDK version: 0.1.0
+
+## [0.1.46] - 2026-01-09
+- Updated airbyte-agent-mcp package
+- Source commit: 3bcb33e8
+- SDK version: 0.1.0
+
+## [0.1.45] - 2026-01-09
+- Updated airbyte-agent-mcp package
+- Source commit: 9fe9d138
+- SDK version: 0.1.0
+
+## [0.1.44] - 2026-01-09
+- Updated airbyte-agent-mcp package
+- Source commit: da9b741b
+- SDK version: 0.1.0
+
+## [0.1.43] - 2026-01-07
+- Updated airbyte-agent-mcp package
+- Source commit: d023e05f
+- SDK version: 0.1.0
+
+## [0.1.42] - 2026-01-06
+- Updated airbyte-agent-mcp package
+- Source commit: 0580c727
+- SDK version: 0.1.0
+
+## [0.1.41] - 2026-01-06
+- Updated airbyte-agent-mcp package
+- Source commit: 42f65575
+- SDK version: 0.1.0
+
+## [0.1.40] - 2026-01-06
+- Updated airbyte-agent-mcp package
+- Source commit: 0deedc22
+- SDK version: 0.1.0
+
+## [0.1.39] - 2026-01-06
+- Updated airbyte-agent-mcp package
+- Source commit: e0e2f989
+- SDK version: 0.1.0
+
+## [0.1.38] - 2026-01-05
+- Updated airbyte-agent-mcp package
+- Source commit: 3e274293
+- SDK version: 0.1.0
+
+## [0.1.37] - 2025-12-19
+- Updated airbyte-agent-mcp package
+- Source commit: 12f6b994
+- SDK version: 0.1.0
+
+## [0.1.36] - 2025-12-19
+- Updated airbyte-agent-mcp package
+- Source commit: 5d11bfdf
+- SDK version: 0.1.0
+
+## [0.1.35] - 2025-12-19
+- Updated airbyte-agent-mcp package
+- Source commit: e996e848
+- SDK version: 0.1.0
+
+## [0.1.34] - 2025-12-18
+- Updated airbyte-agent-mcp package
+- Source commit: f7c55d3e
+- SDK version: 0.1.0
+
+## [0.1.33] - 2025-12-17
+- Updated airbyte-agent-mcp package
+- Source commit: af456521
+- SDK version: 0.1.0
+
+## [0.1.32] - 2025-12-17
+- Updated airbyte-agent-mcp package
+- Source commit: 6a6c981e
+- SDK version: 0.1.0
+
+## [0.1.31] - 2025-12-16
+- Updated airbyte-agent-mcp package
+- Source commit: 9c0af4ae
+- SDK version: 0.1.0
+
+## [0.1.30] - 2025-12-16
+- Updated airbyte-agent-mcp package
+- Source commit: d8fc544e
+- SDK version: 0.1.0
+
+## [0.1.29] - 2025-12-15
+- Updated airbyte-agent-mcp package
+- Source commit: c4c39c27
+- SDK version: 0.1.0
+
+## [0.1.28] - 2025-12-15
+- Updated airbyte-agent-mcp package
+- Source commit: 85f4e6b0
+- SDK version: 0.1.0
+
+## [0.1.27] - 2025-12-15
+- Updated airbyte-agent-mcp package
+- Source commit: 77ce3325
+- SDK version: 0.1.0
+
+## [0.1.26] - 2025-12-15
+- Updated airbyte-agent-mcp package
+- Source commit: 0bfa6500
+- SDK version: 0.1.0
+
+## [0.1.25] - 2025-12-15
+- Updated airbyte-agent-mcp package
+- Source commit: ea5a02a3
+- SDK version: 0.1.0
+
+## [0.1.24] - 2025-12-15
+- Updated airbyte-agent-mcp package
+- Source commit: f13dee0a
+- SDK version: 0.1.0
+
+## [0.1.23] - 2025-12-15
+- Updated airbyte-agent-mcp package
+- Source commit: d79da1e7
+- SDK version: 0.1.0
+
+## [0.1.22] - 2025-12-15
+- Updated airbyte-agent-mcp package
+- Source commit: 0e48ca4e
+- SDK version: 0.1.0
+
+## [0.1.21] - 2025-12-15
+- Updated airbyte-agent-mcp package
+- Source commit: 06e7d5c6
+- SDK version: 0.1.0
+
+## [0.1.20] - 2025-12-13
+- Updated airbyte-agent-mcp package
+- Source commit: 1ab72bd8
+- SDK version: 0.1.0
+
+## [0.1.19] - 2025-12-12
+- Updated airbyte-agent-mcp package
+- Source commit: dc79dc8b
+- SDK version: 0.1.0
+
+## [0.1.18] - 2025-12-12
+- Updated airbyte-agent-mcp package
+- Source commit: 380e7c20
+- SDK version: 0.1.0
+
+## [0.1.17] - 2025-12-12
+- Updated airbyte-agent-mcp package
+- Source commit: 1f622906
+- SDK version: 0.1.0
+
+## [0.1.16] - 2025-12-12
+- Updated airbyte-agent-mcp package
+- Source commit: 8cd69753
+- SDK version: 0.1.0
+
+## [0.1.15] - 2025-12-12
+- Updated airbyte-agent-mcp package
+- Source commit: 9f7f8a98
+- SDK version: 0.1.0
+
+## [0.1.14] - 2025-12-12
+- Updated airbyte-agent-mcp package
+- Source commit: 5cc8eaa0
+- SDK version: 0.1.0
+
+## [0.1.13] - 2025-12-11
+- Updated airbyte-agent-mcp package
+- Source commit: dfe01f50
+- SDK version: 0.1.0
+
+## [0.1.12] - 2025-12-11
+- Updated airbyte-agent-mcp package
+- Source commit: 8c06aa10
+- SDK version: 0.1.0
+
+## [0.1.11] - 2025-12-11
+- Updated airbyte-agent-mcp package
+- Source commit: 11427ac3
+- SDK version: 0.1.0
+
+## [0.1.10] - 2025-12-11
+- Updated airbyte-agent-mcp package
+- Source commit: bdd5df6d
+- SDK version: 0.1.0
+
+## [0.1.9] - 2025-12-11
+- Updated airbyte-agent-mcp package
+- Source commit: ec3dcc78
+- SDK version: 0.1.0
+
+## [0.1.8] - 2025-12-10
+- Updated airbyte-agent-mcp package
+- Source commit: 334e113b
+- SDK version: 0.1.0
+
+## [0.1.7] - 2025-12-10
+- Updated airbyte-agent-mcp package
+- Source commit: 30c77ce6
+- SDK version: 0.1.0
+
+## [0.1.6] - 2025-12-10
+- Updated airbyte-agent-mcp package
+- Source commit: 92482e66
+- SDK version: 0.1.0
+
+## [0.1.5] - 2025-12-10
+- Updated airbyte-agent-mcp package
+- Source commit: b3c6a176
+- SDK version: 0.1.0
+
+## [0.1.4] - 2025-12-10
+- Updated airbyte-agent-mcp package
+- Source commit: b3c6a176
+- SDK version: 0.1.0
+
+## [0.1.3] - 2025-12-10
+- Updated airbyte-agent-mcp package
+- Source commit: b3c6a176
+- SDK version: 0.1.0
+
+## [0.1.2] - 2025-12-10
+- Updated airbyte-agent-mcp package
+- Source commit: b3c6a176
+- SDK version: 0.1.0
+
+## [0.1.1] - 2025-12-10
+- Updated airbyte-agent-mcp package
+- Source commit: b3c6a176
+- SDK version: 0.1.0
+
+## [0.1.0] - 2025-12-09
+- Updated airbyte-agent-mcp package
+- Source commit: b3c6a176
+- SDK version: 0.1.0

{airbyte_agent_mcp-0.1.16 → airbyte_agent_mcp-0.1.51}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: airbyte-agent-mcp
-Version: 0.1.16
+Version: 0.1.51
 Summary: MCP server that exposes Airbyte Connector SDK as MCP tools
 Author-email: Airbyte Support <support@airbyte.io>
 Requires-Python: >=3.13

{airbyte_agent_mcp-0.1.16 → airbyte_agent_mcp-0.1.51}/airbyte_agent_mcp/_vendored/connector_sdk/__init__.py

@@ -11,31 +11,31 @@ Provides:
 from __future__ import annotations
 
 from .auth_strategies import AuthStrategy
+from .connector_model_loader import load_connector_model
 from .constants import SDK_VERSION
+from .exceptions import (
+    AuthenticationError,
+    HTTPClientError,
+    NetworkError,
+    RateLimitError,
+    TimeoutError,
+)
 from .executor import (
-    LocalExecutor,
-    HostedExecutor,
-    ExecutorProtocol,
+    ActionNotSupportedError,
+    EntityNotFoundError,
     ExecutionConfig,
     ExecutionResult,
     ExecutorError,
-    EntityNotFoundError,
-    ActionNotSupportedError,
-    MissingParameterError,
+    ExecutorProtocol,
+    HostedExecutor,
     InvalidParameterError,
+    LocalExecutor,
+    MissingParameterError,
 )
 from .http_client import HTTPClient
-from .types import ConnectorConfig, Action, AuthType, EntityDefinition
-from .config_loader import load_connector_config
-from .logging import RequestLogger, NullLogger, RequestLog, LogSession
+from .logging import LogSession, NullLogger, RequestLog, RequestLogger
 from .performance import PerformanceMonitor, instrument
-from .exceptions import (
-    HTTPClientError,
-    AuthenticationError,
-    RateLimitError,
-    NetworkError,
-    TimeoutError,
-)
+from .types import Action, AuthType, ConnectorModel, EntityDefinition
 from .utils import save_download
 
 __version__ = SDK_VERSION
@@ -50,11 +50,11 @@ __all__ = [
     "ExecutionConfig",
     "ExecutionResult",
     # Types
-    "ConnectorConfig",
+    "ConnectorModel",
     "Action",
     "AuthType",
     "EntityDefinition",
-    "load_connector_config",
+    "load_connector_model",
     # Authentication
     "AuthStrategy",
     # Executor Exceptions

{airbyte_agent_mcp-0.1.16 → airbyte_agent_mcp-0.1.51}/airbyte_agent_mcp/_vendored/connector_sdk/auth_strategies.py

@@ -611,8 +611,7 @@ class OAuth2AuthStrategy(AuthStrategy):
 
         if not has_access_token and not has_refresh_token:
             raise AuthenticationError(
-                "Missing OAuth2 credentials. Provide either 'access_token' "
-                "or 'refresh_token' (for refresh-token-only mode)."
+                "Missing OAuth2 credentials. Provide either 'access_token' " "or 'refresh_token' (for refresh-token-only mode)."
             )
 
     def can_refresh(self, secrets: OAuth2RefreshSecrets) -> bool:
@@ -742,9 +741,7 @@ class OAuth2AuthStrategy(AuthStrategy):
             )
 
         try:
-            logger.info(
-                "Proactively refreshing OAuth2 token (no access_token provided)"
-            )
+            logger.info("Proactively refreshing OAuth2 token (no access_token provided)")
 
             # Create token refresher and attempt refresh
             # Pass None for http_client - let refresher create its own
@@ -826,9 +823,7 @@ class OAuth2TokenRefresher:
         url = self._render_refresh_url(config, secrets)
         headers, body_params = self._build_refresh_request(config, secrets)
 
-        response = await self._execute_refresh_request(
-            url, headers, body_params, config
-        )
+        response = await self._execute_refresh_request(url, headers, body_params, config)
 
         # Get token_extract config if present
         token_extract: list[str] | None = config.get("token_extract")  # type: ignore[assignment]
@@ -893,14 +888,10 @@ class OAuth2TokenRefresher:
 
         # Build template context with priority: config_values > config > secrets
         template_context = dict(config)  # Auth config values
-        template_context.update(
-            self._config_values
-        )  # Non-secret config (higher priority)
+        template_context.update(self._config_values)  # Non-secret config (higher priority)
 
         # Add commonly needed secret values (but not sensitive tokens)
-        template_context["client_id"] = extract_secret_value(
-            secrets.get("client_id", "")
-        )
+        template_context["client_id"] = extract_secret_value(secrets.get("client_id", ""))
 
         return Template(refresh_url).render(template_context)
 
@@ -934,9 +925,7 @@ class OAuth2TokenRefresher:
         }
 
         # Build headers based on auth style
-        headers = self._build_auth_headers(
-            auth_style, client_id_value, client_secret_value
-        )
+        headers = self._build_auth_headers(auth_style, client_id_value, client_secret_value)
 
         # Add client credentials to body if using body auth style
         if auth_style == "body":

{airbyte_agent_mcp-0.1.16 → airbyte_agent_mcp-0.1.51}/airbyte_agent_mcp/_vendored/connector_sdk/auth_template.py

@@ -17,10 +17,7 @@ class MissingVariableError(ValueError):
     def __init__(self, var_name: str, available_fields: list):
         self.var_name = var_name
         self.available_fields = available_fields
-        super().__init__(
-            f"Template variable '${{{var_name}}}' not found in config. "
-            f"Available fields: {available_fields}"
-        )
+        super().__init__(f"Template variable '${{{var_name}}}' not found in config. " f"Available fields: {available_fields}")
 
 
 def apply_template(template: str, values: Dict[str, str]) -> str:
@@ -111,7 +108,10 @@ def apply_auth_mapping(
         user_config: Dict of user-provided field values
         required_fields: List of required field names. If a template references
             a variable not in user_config and that variable is not required,
-            the mapping is skipped. If None, all fields are treated as required.
+            the mapping is skipped. Behavior:
+            - None: all fields are treated as required (backward compatible)
+            - []: no fields are required (all optional)
+            - ["foo"]: only "foo" is required
 
     Returns:
         Dict of resolved auth parameters
@@ -120,7 +120,7 @@ def apply_auth_mapping(
         MissingVariableError: If a required template variable is not found
     """
     resolved = {}
-    required_set = set(required_fields) if required_fields else None
+    required_set = set(required_fields) if required_fields is not None else None
 
     for param, template in auth_mapping.items():
         try:

airbyte_agent_mcp-0.1.51/airbyte_agent_mcp/_vendored/connector_sdk/cloud_utils/__init__.py

@@ -0,0 +1,5 @@
+"""Cloud API utilities for Airbyte Platform integration."""
+
+from .client import AirbyteCloudClient
+
+__all__ = ["AirbyteCloudClient"]

airbyte_agent_mcp-0.1.51/airbyte_agent_mcp/_vendored/connector_sdk/cloud_utils/client.py

@@ -0,0 +1,213 @@
+"""AirbyteCloudClient for Airbyte Platform API integration."""
+
+from __future__ import annotations
+
+from datetime import datetime, timedelta
+from typing import Any
+
+import httpx
+
+
+class AirbyteCloudClient:
+    """Client for interacting with Airbyte Platform APIs.
+
+    Handles authentication, token caching, and API calls to:
+    - Get bearer tokens for authentication
+    - Look up connector instances for users
+    - Execute connectors via the cloud API
+
+    Example:
+        client = AirbyteCloudClient(
+            client_id="your-client-id",
+            client_secret="your-client-secret"
+        )
+
+        # Get a connector instance
+        instance_id = await client.get_connector_instance_id(
+            external_user_id="user-123",
+            connector_definition_id="stripe-def-456"
+        )
+
+        # Execute the connector
+        result = await client.execute_connector(
+            instance_id=instance_id,
+            entity="customers",
+            action="list",
+            params={"limit": 10}
+        )
+    """
+
+    AUTH_BASE_URL = "https://cloud.airbyte.com"  # For token endpoint
+    API_BASE_URL = "https://api.airbyte.ai"  # For instance lookup & execution
+
+    def __init__(self, client_id: str, client_secret: str):
+        """Initialize AirbyteCloudClient.
+
+        Args:
+            client_id: Airbyte client ID for authentication
+            client_secret: Airbyte client secret for authentication
+        """
+        self._client_id = client_id
+        self._client_secret = client_secret
+
+        # Token cache (instance-level)
+        self._cached_token: str | None = None
+        self._token_expires_at: datetime | None = None
+        self._http_client = httpx.AsyncClient(
+            timeout=httpx.Timeout(300.0),  # 5 minute timeout
+            follow_redirects=True,
+        )
+
+    async def get_bearer_token(self) -> str:
+        """Get bearer token for API authentication.
+
+        Caches the token and only requests a new one when the cached token
+        is expired or missing. Adds a 60-second buffer before expiration
+        to avoid edge cases.
+
+        Returns:
+            Bearer token string
+
+        Raises:
+            httpx.HTTPStatusError: If the token request fails with 4xx/5xx
+            httpx.RequestError: If the network request fails
+
+        Example:
+            token = await client.get_bearer_token()
+            # Use token in Authorization header: f"Bearer {token}"
+        """
+        # Check if we have a cached token that hasn't expired
+        if self._cached_token and self._token_expires_at:
+            # Add 60 second buffer before expiration to avoid edge cases
+            now = datetime.now()
+            if now < self._token_expires_at:
+                # Token is still valid, return cached version
+                return self._cached_token
+
+        # Token is missing or expired, fetch a new one
+        url = f"{self.AUTH_BASE_URL}/api/v1/applications/token"
+        request_body = {
+            "client_id": self._client_id,
+            "client_secret": self._client_secret,
+        }
+
+        response = await self._http_client.post(url, json=request_body)
+        response.raise_for_status()
+
+        data = response.json()
+        access_token = data["access_token"]
+        expires_in = 15 * 60  # default 15 min expiry time * 60 seconds
+
+        # Calculate expiration time with 60 second buffer
+        expires_at = datetime.now() + timedelta(seconds=expires_in - 60)
+        self._cached_token = access_token
+        self._token_expires_at = expires_at
+
+        return access_token
+
+    async def get_connector_instance_id(
+        self,
+        external_user_id: str,
+        connector_definition_id: str,
+    ) -> str:
+        """Get connector instance ID for a user.
+
+        Looks up the connector instance that belongs to the specified user
+        and connector definition. Validates that exactly one instance exists.
+
+        Args:
+            external_user_id: User identifier in the Airbyte system
+            connector_definition_id: UUID of the connector definition
+
+        Returns:
+            Connector instance ID (UUID string)
+
+        Raises:
+            ValueError: If 0 or more than 1 instance is found
+            httpx.HTTPStatusError: If API returns 4xx/5xx status code
+            httpx.RequestError: If network request fails
+
+        Example:
+            instance_id = await client.get_connector_instance_id(
+                external_user_id="user-123",
+                connector_definition_id="550e8400-e29b-41d4-a716-446655440000"
+            )
+        """
+
+        token = await self.get_bearer_token()
+        url = f"{self.API_BASE_URL}/api/v1/connectors/instances_for_user"
+        params = {
+            "external_user_id": external_user_id,
+            "definition_id": connector_definition_id,
+        }
+
+        headers = {"Authorization": f"Bearer {token}"}
+        response = await self._http_client.get(url, params=params, headers=headers)
+        response.raise_for_status()
+
+        data = response.json()
+        instances = data["instances"]
+
+        if len(instances) == 0:
+            raise ValueError(f"No connector instance found for user '{external_user_id}' " f"and connector '{connector_definition_id}'")
+
+        if len(instances) > 1:
+            raise ValueError(
+                f"Multiple connector instances found for user '{external_user_id}' "
+                f"and connector '{connector_definition_id}'. Expected exactly 1, "
+                f"found {len(instances)}"
+            )
+
+        instance_id = instances[0]["id"]
+        return instance_id
+
+    async def execute_connector(
+        self,
+        instance_id: str,
+        entity: str,
+        action: str,
+        params: dict[str, Any] | None,
+    ) -> dict[str, Any]:
+        """Execute a connector operation.
+
+        Args:
+            instance_id: Connector instance UUID
+            entity: Entity name (e.g., "customers", "invoices")
+            action: Operation action (e.g., "list", "get", "create")
+            params: Optional parameters for the operation
+
+        Returns:
+            Raw JSON response dict from the API
+
+        Raises:
+            httpx.HTTPStatusError: If API returns 4xx/5xx status code
+            httpx.RequestError: If network request fails
+
+        Example:
+            result = await client.execute_connector(
+                instance_id="inst-123",
+                entity="customers",
+                action="list",
+                params={"limit": 10}
+            )
+        """
+        token = await self.get_bearer_token()
+        url = f"{self.API_BASE_URL}/api/v1/connectors/instances/{instance_id}/execute"
+        headers = {"Authorization": f"Bearer {token}"}
+        request_body = {
+            "entity": entity,
+            "action": action,
+            "params": params,
+        }
+
+        response = await self._http_client.post(url, json=request_body, headers=headers)
+        response.raise_for_status()
+
+        return response.json()
+
+    async def close(self):
+        """Close the HTTP client.
+
+        Call this when you're done using the client to clean up resources.
+        """
+        await self._http_client.aclose()