airbyte-agent-mcp 0.1.30__py3-none-any.whl

airbyte_agent_mcp/_vendored/connector_sdk/constants.py

@@ -0,0 +1,78 @@
+"""
+Constants used throughout the Airbyte SDK.
+
+This module centralizes configuration defaults and commonly used values to improve
+maintainability and consistency across the codebase.
+"""
+
+from __future__ import annotations
+
+from importlib.metadata import PackageNotFoundError, version
+
+# ============================================================================
+# HTTP Connection and Timeout Defaults
+# ============================================================================
+
+# Connection pooling limits
+DEFAULT_MAX_CONNECTIONS = 100
+"""Maximum number of concurrent HTTP connections to maintain in the pool."""
+
+DEFAULT_MAX_KEEPALIVE_CONNECTIONS = 20
+"""Maximum number of keepalive connections to maintain in the pool."""
+
+# Timeout values (in seconds)
+DEFAULT_CONNECT_TIMEOUT = 5.0
+"""Default timeout for establishing a new connection (seconds)."""
+
+DEFAULT_READ_TIMEOUT = 30.0
+"""Default timeout for reading response data (seconds)."""
+
+DEFAULT_WRITE_TIMEOUT = 30.0
+"""Default timeout for writing request data (seconds)."""
+
+DEFAULT_POOL_TIMEOUT = 5.0
+"""Default timeout for acquiring a connection from the pool (seconds)."""
+
+DEFAULT_REQUEST_TIMEOUT = 30.0
+"""Default overall request timeout (seconds)."""
+
+# ============================================================================
+# OpenAPI Specification
+# ============================================================================
+
+OPENAPI_VERSION_PREFIX = "3.1."
+"""Required OpenAPI version prefix. Only 3.1.x specifications are supported."""
+
+OPENAPI_DEFAULT_VERSION = "1.0.0"
+"""Default version string for connectors that don't specify a version."""
+
+# ============================================================================
+# Performance and Metrics
+# ============================================================================
+
+MILLISECONDS_PER_SECOND = 1000
+"""Conversion factor from seconds to milliseconds."""
+
+# ============================================================================
+# Retry and Backoff Defaults
+# ============================================================================
+
+DEFAULT_INITIAL_DELAY_SECONDS = 1.0
+"""Default initial delay for retry backoff (seconds)."""
+
+DEFAULT_MAX_DELAY_SECONDS = 60.0
+"""Default maximum delay for retry backoff (seconds)."""
+
+# ============================================================================
+# SDK Version
+# ============================================================================
+
+try:
+    SDK_VERSION = version("connector-sdk")
+except PackageNotFoundError:
+    # Fallback for development when package isn't installed
+    SDK_VERSION = "0.0.0-dev"
+"""Current version of the Airbyte SDK."""
+
+MINIMUM_PYTHON_VERSION = "3.9"
+"""Minimum Python version required to run the SDK."""

airbyte_agent_mcp/_vendored/connector_sdk/exceptions.py

@@ -0,0 +1,23 @@
+"""Exceptions for the Airbyte SDK.
+
+DEPRECATED: HTTP exceptions have been moved to connector_sdk.http.exceptions.
+This module re-exports them for backward compatibility.
+"""
+
+from .http.exceptions import (
+    AuthenticationError,
+    HTTPClientError,
+    HTTPStatusError,
+    NetworkError,
+    RateLimitError,
+    TimeoutError,
+)
+
+__all__ = [
+    "HTTPClientError",
+    "HTTPStatusError",
+    "AuthenticationError",
+    "RateLimitError",
+    "NetworkError",
+    "TimeoutError",
+]

airbyte_agent_mcp/_vendored/connector_sdk/executor/__init__.py

@@ -0,0 +1,31 @@
+"""Executor implementations for connector operations."""
+
+from .hosted_executor import HostedExecutor
+from .local_executor import LocalExecutor
+from .models import (
+    ActionNotSupportedError,
+    EntityNotFoundError,
+    ExecutionConfig,
+    ExecutionResult,
+    ExecutorError,
+    ExecutorProtocol,
+    InvalidParameterError,
+    MissingParameterError,
+)
+
+__all__ = [
+    # Config and Result types
+    "ExecutionConfig",
+    "ExecutionResult",
+    # Protocol
+    "ExecutorProtocol",
+    # Executors
+    "LocalExecutor",
+    "HostedExecutor",
+    # Exceptions
+    "ExecutorError",
+    "EntityNotFoundError",
+    "ActionNotSupportedError",
+    "MissingParameterError",
+    "InvalidParameterError",
+]

airbyte_agent_mcp/_vendored/connector_sdk/executor/hosted_executor.py

@@ -0,0 +1,188 @@
+"""Hosted executor for proxying operations through the backend API."""
+
+from __future__ import annotations
+
+import os
+
+import httpx
+from opentelemetry import trace
+
+from .models import (
+    ExecutionConfig,
+    ExecutionResult,
+)
+
+
+class HostedExecutor:
+    """Executor that proxies execution through the Sonar backend API.
+
+    This is the "hosted mode" executor that makes HTTP calls to the backend API
+    instead of directly calling external services. The backend handles all
+    connector logic, secrets management, and execution.
+
+    The API URL is configured at initialization via the api_url parameter,
+    which defaults to the AIRBYTE_CONNECTOR_API_URL environment variable.
+
+    Implements ExecutorProtocol.
+
+    Example:
+        executor = HostedExecutor(
+            connector_id="stripe-prod-123",
+            airbyte_client_id="client_abc123",
+            airbyte_client_secret="secret_xyz789"
+        )
+
+        config = ExecutionConfig(
+            entity="customers",
+            action="list"
+        )
+
+        result = await executor.execute(config)
+        if result.success:
+            print(f"Data: {result.data}")
+        else:
+            print(f"Error: {result.error}")
+    """
+
+    def __init__(
+        self,
+        connector_id: str,
+        airbyte_client_id: str,
+        airbyte_client_secret: str,
+        api_url: str | None = None,
+    ):
+        """Initialize hosted executor.
+
+        Args:
+            connector_id: ID of the connector to execute (e.g., "stripe-prod-123")
+            airbyte_client_id: Airbyte client ID for authentication
+            airbyte_client_secret: Airbyte client secret for authentication
+            api_url: API URL for the hosted executor backend. Defaults to
+                AIRBYTE_CONNECTOR_API_URL environment variable or "http://localhost:8001"
+
+        Example:
+            executor = HostedExecutor(
+                connector_id="my-connector-id",
+                airbyte_client_id="client_abc123",
+                airbyte_client_secret="secret_xyz789"
+            )
+
+            # Or with custom API URL:
+            executor = HostedExecutor(
+                connector_id="my-connector-id",
+                airbyte_client_id="client_abc123",
+                airbyte_client_secret="secret_xyz789",
+                api_url="https://api.production.com"
+            )
+        """
+        self.connector_id = connector_id
+        self.airbyte_client_id = airbyte_client_id
+        self.airbyte_client_secret = airbyte_client_secret
+        self.api_url = api_url or os.getenv("AIRBYTE_CONNECTOR_API_URL", "http://localhost:8001")
+
+        # Create synchronous HTTP client
+        # We use sync client even though execute() is async for simplicity
+        # The async wrapper allows it to work with the protocol
+        self.client = httpx.Client(
+            timeout=httpx.Timeout(300.0),  # 5 minute timeout
+            follow_redirects=True,
+        )
+
+    async def execute(self, config: ExecutionConfig) -> ExecutionResult:
+        """Execute connector via backend API (ExecutorProtocol implementation).
+
+        Makes an HTTP POST request to /connectors/{connector_id}/execute with
+        OAuth authentication and the configuration in the request body.
+
+        Args:
+            config: Execution configuration (entity, action, params)
+
+        Returns:
+            ExecutionResult with success/failure status
+
+        Raises:
+            httpx.HTTPStatusError: If API returns 4xx/5xx status code
+            httpx.RequestError: If network request fails
+
+        Example:
+            config = ExecutionConfig(
+                entity="customers",
+                action="list"
+            )
+            result = await executor.execute(config)
+        """
+        tracer = trace.get_tracer("airbyte.connector-sdk.executor.hosted")
+
+        with tracer.start_as_current_span("airbyte.hosted_executor.execute") as span:
+            # Add span attributes
+            span.set_attribute("connector.id", self.connector_id)
+            span.set_attribute("connector.entity", config.entity)
+            span.set_attribute("connector.action", config.action)
+            span.set_attribute("connector.api_url", self.api_url)
+            if config.params:
+                # Only add non-sensitive param keys
+                span.set_attribute("connector.param_keys", list(config.params.keys()))
+
+            # Build API URL from instance api_url
+            url = f"{self.api_url}/connectors/{self.connector_id}/execute"
+            span.set_attribute("http.url", url)
+
+            # Build request body matching ExecutionRequest model
+            # Extract entity, action, and params from config attributes
+            request_body = {
+                "entity": config.entity,
+                "action": config.action,
+                "params": config.params,
+            }
+
+            try:
+                # Make synchronous HTTP request
+                # (wrapped in async method for protocol compatibility)
+                response = self.client.post(url, json=request_body)
+
+                # Add response status code to span
+                span.set_attribute("http.status_code", response.status_code)
+
+                # Raise exception for 4xx/5xx status codes
+                response.raise_for_status()
+
+                # Parse JSON response
+                result_data = response.json()
+
+                # Mark span as successful
+                span.set_attribute("connector.success", True)
+
+                # Return success result
+                return ExecutionResult(success=True, data=result_data, error=None)
+
+            except httpx.HTTPStatusError as e:
+                # HTTP error (4xx, 5xx) - record and re-raise
+                span.set_attribute("connector.success", False)
+                span.set_attribute("connector.error_type", "HTTPStatusError")
+                span.set_attribute("http.status_code", e.response.status_code)
+                span.record_exception(e)
+                raise
+
+            except Exception as e:
+                # Catch-all for any other unexpected exceptions
+                span.set_attribute("connector.success", False)
+                span.set_attribute("connector.error_type", type(e).__name__)
+                span.record_exception(e)
+                raise
+
+    def close(self):
+        """Close the HTTP client.
+
+        Call this when you're done using the executor to clean up resources.
+
+        Example:
+            executor = HostedExecutor(
+                workspace_id="workspace-123",
+                connector_id="my-connector"
+            )
+            try:
+                result = await executor.execute(config)
+            finally:
+                executor.close()
+        """
+        self.client.close()