airbyte-agent-slack 0.1.37__py3-none-any.whl → 0.1.39__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -7,6 +7,7 @@ Auto-generated from OpenAPI specification.
7
7
  from .connector import SlackConnector
8
8
  from .models import (
9
9
  SlackAuthConfig,
10
+ SlackReplicationConfig,
10
11
  User,
11
12
  UserProfile,
12
13
  ResponseMetadata,
@@ -17,9 +18,9 @@ from .models import (
17
18
  ChannelPurpose,
18
19
  ChannelsListResponse,
19
20
  ChannelResponse,
20
- Reaction,
21
21
  Attachment,
22
22
  File,
23
+ Reaction,
23
24
  Message,
24
25
  Thread,
25
26
  EditedInfo,
@@ -86,6 +87,7 @@ from .types import (
86
87
  __all__ = [
87
88
  "SlackConnector",
88
89
  "SlackAuthConfig",
90
+ "SlackReplicationConfig",
89
91
  "User",
90
92
  "UserProfile",
91
93
  "ResponseMetadata",
@@ -96,9 +98,9 @@ __all__ = [
96
98
  "ChannelPurpose",
97
99
  "ChannelsListResponse",
98
100
  "ChannelResponse",
99
- "Reaction",
100
101
  "Attachment",
101
102
  "File",
103
+ "Reaction",
102
104
  "Message",
103
105
  "Thread",
104
106
  "EditedInfo",
@@ -161,6 +161,68 @@ class AirbyteCloudClient:
161
161
  connector_id = connectors[0]["id"]
162
162
  return connector_id
163
163
 
164
+ async def create_source(
165
+ self,
166
+ name: str,
167
+ connector_definition_id: str,
168
+ external_user_id: str,
169
+ credentials: dict[str, Any],
170
+ replication_config: dict[str, Any] | None = None,
171
+ ) -> str:
172
+ """Create a new source on Airbyte Agent Platform.
173
+
174
+ Args:
175
+ name: Source name
176
+ connector_definition_id: UUID of the connector definition
177
+ external_user_id: User identifier
178
+ credentials: Connector auth config dict
179
+ replication_config: Optional replication settings (e.g., start_date for
180
+ connectors with x-airbyte-replication-config). Required for REPLICATION
181
+ mode sources like Intercom.
182
+
183
+ Returns:
184
+ The created source ID (UUID string)
185
+
186
+ Raises:
187
+ httpx.HTTPStatusError: If creation fails
188
+
189
+ Example:
190
+ source_id = await client.create_source(
191
+ name="My Gong Source",
192
+ connector_definition_id="32382e40-3b49-4b99-9c5c-4076501914e7",
193
+ external_user_id="my-workspace",
194
+ credentials={"access_key": "...", "access_key_secret": "..."}
195
+ )
196
+
197
+ # For REPLICATION mode sources (e.g., Intercom):
198
+ source_id = await client.create_source(
199
+ name="My Intercom Source",
200
+ connector_definition_id="d8313939-3782-41b0-be29-b3ca20d8dd3a",
201
+ external_user_id="my-workspace",
202
+ credentials={"access_token": "..."},
203
+ replication_config={"start_date": "2024-01-01T00:00:00Z"}
204
+ )
205
+ """
206
+ token = await self.get_bearer_token()
207
+ url = f"{self.API_BASE_URL}/v1/integrations/connectors"
208
+ headers = {"Authorization": f"Bearer {token}"}
209
+
210
+ request_body: dict[str, Any] = {
211
+ "name": name,
212
+ "definition_id": connector_definition_id,
213
+ "external_user_id": external_user_id,
214
+ "credentials": credentials,
215
+ }
216
+
217
+ if replication_config is not None:
218
+ request_body["replication_config"] = replication_config
219
+
220
+ response = await self._http_client.post(url, json=request_body, headers=headers)
221
+ response.raise_for_status()
222
+
223
+ data = response.json()
224
+ return data["id"]
225
+
164
226
  async def execute_connector(
165
227
  self,
166
228
  connector_id: str,
@@ -1014,6 +1014,7 @@ def _parse_security_scheme_to_option(scheme_name: str, scheme: Any) -> AuthOptio
1014
1014
  type=single_auth.type,
1015
1015
  config=single_auth.config,
1016
1016
  user_config_spec=single_auth.user_config_spec,
1017
+ untested=getattr(scheme, "x_airbyte_untested", False),
1017
1018
  )
1018
1019
 
1019
1020
 
@@ -19,19 +19,26 @@ class HostedExecutor:
19
19
  instead of directly calling external services. The cloud API handles all
20
20
  connector logic, secrets management, and execution.
21
21
 
22
- The executor takes an external_user_id and uses the AirbyteCloudClient to:
22
+ The executor uses the AirbyteCloudClient to:
23
23
  1. Authenticate with the Airbyte Platform (bearer token with caching)
24
- 2. Look up the user's connector
24
+ 2. Look up the user's connector (if connector_id not provided)
25
25
  3. Execute the connector operation via the cloud API
26
26
 
27
27
  Implements ExecutorProtocol.
28
28
 
29
29
  Example:
30
- # Create executor with user ID, credentials, and connector definition ID
30
+ # Create executor with explicit connector_id (no lookup needed)
31
+ executor = HostedExecutor(
32
+ airbyte_client_id="client_abc123",
33
+ airbyte_client_secret="secret_xyz789",
34
+ connector_id="existing-source-uuid",
35
+ )
36
+
37
+ # Or create executor with user ID for lookup
31
38
  executor = HostedExecutor(
32
- external_user_id="user-123",
33
39
  airbyte_client_id="client_abc123",
34
40
  airbyte_client_secret="secret_xyz789",
41
+ external_user_id="user-123",
35
42
  connector_definition_id="abc123-def456-ghi789",
36
43
  )
37
44
 
@@ -51,28 +58,48 @@ class HostedExecutor:
51
58
 
52
59
  def __init__(
53
60
  self,
54
- external_user_id: str,
55
61
  airbyte_client_id: str,
56
62
  airbyte_client_secret: str,
57
- connector_definition_id: str,
63
+ connector_id: str | None = None,
64
+ external_user_id: str | None = None,
65
+ connector_definition_id: str | None = None,
58
66
  ):
59
67
  """Initialize hosted executor.
60
68
 
69
+ Either provide connector_id directly OR (external_user_id + connector_definition_id)
70
+ for lookup.
71
+
61
72
  Args:
62
- external_user_id: User identifier in the Airbyte system
63
73
  airbyte_client_id: Airbyte client ID for authentication
64
74
  airbyte_client_secret: Airbyte client secret for authentication
65
- connector_definition_id: Connector definition ID used to look up
66
- the user's connector.
75
+ connector_id: Direct connector/source ID (skips lookup if provided)
76
+ external_user_id: User identifier in the Airbyte system (for lookup)
77
+ connector_definition_id: Connector definition ID (for lookup)
78
+
79
+ Raises:
80
+ ValueError: If neither connector_id nor (external_user_id + connector_definition_id) provided
67
81
 
68
82
  Example:
83
+ # With explicit connector_id (no lookup)
84
+ executor = HostedExecutor(
85
+ airbyte_client_id="client_abc123",
86
+ airbyte_client_secret="secret_xyz789",
87
+ connector_id="existing-source-uuid",
88
+ )
89
+
90
+ # With lookup by user + definition
69
91
  executor = HostedExecutor(
70
- external_user_id="user-123",
71
92
  airbyte_client_id="client_abc123",
72
93
  airbyte_client_secret="secret_xyz789",
94
+ external_user_id="user-123",
73
95
  connector_definition_id="abc123-def456-ghi789",
74
96
  )
75
97
  """
98
+ # Validate: either connector_id OR (external_user_id + connector_definition_id) required
99
+ if not connector_id and not (external_user_id and connector_definition_id):
100
+ raise ValueError("Either connector_id OR (external_user_id + connector_definition_id) must be provided")
101
+
102
+ self._connector_id = connector_id
76
103
  self._external_user_id = external_user_id
77
104
  self._connector_definition_id = connector_definition_id
78
105
 
@@ -86,10 +113,9 @@ class HostedExecutor:
86
113
  """Execute connector via cloud API (ExecutorProtocol implementation).
87
114
 
88
115
  Flow:
89
- 1. Get connector definition id from executor config
90
- 2. Look up the user's connector ID
91
- 3. Execute the connector operation via the cloud API
92
- 4. Parse the response into ExecutionResult
116
+ 1. Use provided connector_id or look up from external_user_id + definition_id
117
+ 2. Execute the connector operation via the cloud API
118
+ 3. Parse the response into ExecutionResult
93
119
 
94
120
  Args:
95
121
  config: Execution configuration (entity, action, params)
@@ -98,7 +124,7 @@ class HostedExecutor:
98
124
  ExecutionResult with success/failure status
99
125
 
100
126
  Raises:
101
- ValueError: If no connector or multiple connectors found for user
127
+ ValueError: If no connector or multiple connectors found for user (when doing lookup)
102
128
  httpx.HTTPStatusError: If API returns 4xx/5xx status code
103
129
  httpx.RequestError: If network request fails
104
130
 
@@ -114,23 +140,26 @@ class HostedExecutor:
114
140
 
115
141
  with tracer.start_as_current_span("airbyte.hosted_executor.execute") as span:
116
142
  # Add span attributes for observability
117
- span.set_attribute("connector.definition_id", self._connector_definition_id)
143
+ if self._connector_definition_id:
144
+ span.set_attribute("connector.definition_id", self._connector_definition_id)
118
145
  span.set_attribute("connector.entity", config.entity)
119
146
  span.set_attribute("connector.action", config.action)
120
- span.set_attribute("user.external_id", self._external_user_id)
147
+ if self._external_user_id:
148
+ span.set_attribute("user.external_id", self._external_user_id)
121
149
  if config.params:
122
150
  # Only add non-sensitive param keys
123
151
  span.set_attribute("connector.param_keys", list(config.params.keys()))
124
152
 
125
153
  try:
126
- # Step 1: Get connector definition id
127
- connector_definition_id = self._connector_definition_id
128
-
129
- # Step 2: Get the connector ID for this user
130
- connector_id = await self._cloud_client.get_connector_id(
131
- external_user_id=self._external_user_id,
132
- connector_definition_id=connector_definition_id,
133
- )
154
+ # Use provided connector_id or look it up
155
+ if self._connector_id:
156
+ connector_id = self._connector_id
157
+ else:
158
+ # Look up connector by external_user_id + definition_id
159
+ connector_id = await self._cloud_client.get_connector_id(
160
+ external_user_id=self._external_user_id, # type: ignore[arg-type]
161
+ connector_definition_id=self._connector_definition_id, # type: ignore[arg-type]
162
+ )
134
163
 
135
164
  span.set_attribute("connector.connector_id", connector_id)
136
165
 
@@ -36,6 +36,7 @@ from ..types import (
36
36
  EndpointDefinition,
37
37
  EntityDefinition,
38
38
  )
39
+ from ..utils import find_matching_auth_options
39
40
 
40
41
  from .models import (
41
42
  ActionNotSupportedError,
@@ -356,8 +357,8 @@ class LocalExecutor:
356
357
  ) -> tuple[AuthOption, dict[str, SecretStr]]:
357
358
  """Infer authentication scheme from provided credentials.
358
359
 
359
- Matches user credentials against each auth option's required fields
360
- to determine which scheme to use.
360
+ Uses shared utility find_matching_auth_options to match credentials
361
+ against each auth option's required fields.
361
362
 
362
363
  Args:
363
364
  user_credentials: User-provided credentials
@@ -375,16 +376,8 @@ class LocalExecutor:
375
376
  # Get the credential keys provided by the user
376
377
  provided_keys = set(user_credentials.keys())
377
378
 
378
- # Find all options where all required fields are present
379
- matching_options: list[AuthOption] = []
380
- for option in options:
381
- if option.user_config_spec and option.user_config_spec.required:
382
- required_fields = set(option.user_config_spec.required)
383
- if required_fields.issubset(provided_keys):
384
- matching_options.append(option)
385
- elif not option.user_config_spec or not option.user_config_spec.required:
386
- # Option has no required fields - it matches any credentials
387
- matching_options.append(option)
379
+ # Use shared utility to find matching options
380
+ matching_options = find_matching_auth_options(provided_keys, options)
388
381
 
389
382
  # Handle matching results
390
383
  if len(matching_options) == 0:
@@ -199,6 +199,11 @@ class SecurityScheme(BaseModel):
199
199
  alias="x-airbyte-token-extract",
200
200
  description="List of fields to extract from OAuth2 token responses and use as server variables",
201
201
  )
202
+ x_airbyte_untested: bool = Field(
203
+ False,
204
+ alias="x-airbyte-untested",
205
+ description="Mark this auth scheme as untested to skip cassette coverage validation",
206
+ )
202
207
 
203
208
  @field_validator("x_airbyte_token_extract", mode="after")
204
209
  @classmethod
@@ -90,6 +90,10 @@ class AuthOption(BaseModel):
90
90
  None,
91
91
  description="User-facing credential specification from x-airbyte-auth-config",
92
92
  )
93
+ untested: bool = Field(
94
+ False,
95
+ description="Mark this auth scheme as untested to skip cassette coverage validation",
96
+ )
93
97
 
94
98
 
95
99
  class AuthConfig(BaseModel):
@@ -1,7 +1,13 @@
1
1
  """Utility functions for working with connectors."""
2
2
 
3
+ from __future__ import annotations
4
+
3
5
  from collections.abc import AsyncIterator
4
6
  from pathlib import Path
7
+ from typing import TYPE_CHECKING
8
+
9
+ if TYPE_CHECKING:
10
+ from .types import AuthOption
5
11
 
6
12
 
7
13
  async def save_download(
@@ -58,3 +64,64 @@ async def save_download(
58
64
  raise OSError(f"Failed to write file {file_path}: {e}") from e
59
65
 
60
66
  return file_path
67
+
68
+
69
+ def find_matching_auth_options(
70
+ provided_keys: set[str],
71
+ auth_options: list[AuthOption],
72
+ ) -> list[AuthOption]:
73
+ """Find auth options that match the provided credential keys.
74
+
75
+ This is the single source of truth for auth scheme inference logic,
76
+ used by both the executor (at runtime) and validation (for cassettes).
77
+
78
+ Matching logic:
79
+ - An option matches if all its required fields are present in provided_keys
80
+ - Options with no required fields match any credentials
81
+
82
+ Args:
83
+ provided_keys: Set of credential/auth_config keys
84
+ auth_options: List of AuthOption from the connector model
85
+
86
+ Returns:
87
+ List of AuthOption that match the provided keys
88
+ """
89
+ matching_options: list[AuthOption] = []
90
+
91
+ for option in auth_options:
92
+ if option.user_config_spec and option.user_config_spec.required:
93
+ required_fields = set(option.user_config_spec.required)
94
+ if required_fields.issubset(provided_keys):
95
+ matching_options.append(option)
96
+ elif not option.user_config_spec or not option.user_config_spec.required:
97
+ # Option has no required fields - it matches any credentials
98
+ matching_options.append(option)
99
+
100
+ return matching_options
101
+
102
+
103
+ def infer_auth_scheme_name(
104
+ provided_keys: set[str],
105
+ auth_options: list[AuthOption],
106
+ ) -> str | None:
107
+ """Infer the auth scheme name from provided credential keys.
108
+
109
+ Uses find_matching_auth_options to find matches, then returns
110
+ the scheme name only if exactly one option matches.
111
+
112
+ Args:
113
+ provided_keys: Set of credential/auth_config keys
114
+ auth_options: List of AuthOption from the connector model
115
+
116
+ Returns:
117
+ The scheme_name if exactly one match, None otherwise
118
+ """
119
+ if not provided_keys or not auth_options:
120
+ return None
121
+
122
+ matching = find_matching_auth_options(provided_keys, auth_options)
123
+
124
+ if len(matching) == 1:
125
+ return matching[0].scheme_name
126
+
127
+ return None
@@ -21,7 +21,8 @@ from .connector_model_loader import (
21
21
  load_connector_model,
22
22
  )
23
23
  from .testing.spec_loader import load_test_spec
24
- from .types import Action, EndpointDefinition
24
+ from .types import Action, ConnectorModel, EndpointDefinition
25
+ from .utils import infer_auth_scheme_name
25
26
  from .validation_replication import validate_replication_compatibility
26
27
 
27
28
 
@@ -53,6 +54,112 @@ def build_cassette_map(cassettes_dir: Path) -> Dict[Tuple[str, str], List[Path]]
53
54
  return dict(cassette_map)
54
55
 
55
56
 
57
+ def build_auth_scheme_coverage(
58
+ cassettes_dir: Path,
59
+ auth_options: list | None = None,
60
+ ) -> Tuple[Dict[str | None, List[Path]], List[Tuple[Path, set[str]]]]:
61
+ """Build a map of auth_scheme -> list of cassette paths.
62
+
63
+ For multi-auth connectors, infers the auth scheme from the cassette's auth_config
64
+ keys using the same matching logic as the executor.
65
+
66
+ Args:
67
+ cassettes_dir: Directory containing cassette YAML files
68
+ auth_options: List of AuthOption from the connector model (for inference)
69
+
70
+ Returns:
71
+ Tuple of:
72
+ - Dictionary mapping auth_scheme names (or None for single-auth) to cassette paths
73
+ - List of (cassette_path, auth_config_keys) for cassettes that couldn't be matched
74
+ """
75
+ auth_scheme_map: Dict[str | None, List[Path]] = defaultdict(list)
76
+ unmatched_cassettes: List[Tuple[Path, set[str]]] = []
77
+
78
+ if not cassettes_dir.exists() or not cassettes_dir.is_dir():
79
+ return {}, []
80
+
81
+ for cassette_file in cassettes_dir.glob("*.yaml"):
82
+ try:
83
+ spec = load_test_spec(cassette_file, auth_config={})
84
+
85
+ # First, check if auth_scheme is explicitly set in the cassette
86
+ if spec.auth_scheme:
87
+ auth_scheme_map[spec.auth_scheme].append(cassette_file)
88
+ # Otherwise, try to infer from auth_config keys
89
+ elif spec.auth_config and auth_options:
90
+ auth_config_keys = set(spec.auth_config.keys())
91
+ inferred_scheme = infer_auth_scheme_name(auth_config_keys, auth_options)
92
+ if inferred_scheme is not None:
93
+ auth_scheme_map[inferred_scheme].append(cassette_file)
94
+ else:
95
+ # Couldn't infer - track as unmatched
96
+ unmatched_cassettes.append((cassette_file, auth_config_keys))
97
+ else:
98
+ # No auth_scheme and no auth_config - treat as None
99
+ auth_scheme_map[None].append(cassette_file)
100
+ except Exception:
101
+ continue
102
+
103
+ return dict(auth_scheme_map), unmatched_cassettes
104
+
105
+
106
+ def validate_auth_scheme_coverage(
107
+ config: ConnectorModel,
108
+ cassettes_dir: Path,
109
+ ) -> Tuple[bool, List[str], List[str], List[str], List[Tuple[Path, set[str]]]]:
110
+ """Validate that each auth scheme has at least one cassette.
111
+
112
+ For multi-auth connectors, every defined auth scheme must have coverage
113
+ unless marked with x-airbyte-untested: true.
114
+ For single-auth connectors, this check is skipped (existing cassette checks suffice).
115
+
116
+ Args:
117
+ config: Loaded connector model
118
+ cassettes_dir: Directory containing cassette files
119
+
120
+ Returns:
121
+ Tuple of (is_valid, errors, warnings, covered_schemes, unmatched_cassettes)
122
+ """
123
+ errors: List[str] = []
124
+ warnings: List[str] = []
125
+
126
+ # Skip check for single-auth connectors
127
+ if not config.auth.is_multi_auth():
128
+ return True, errors, warnings, [], []
129
+
130
+ # Get all defined auth schemes, separating tested from untested
131
+ options = config.auth.options or []
132
+
133
+ # Build auth scheme coverage from cassettes (pass options for inference)
134
+ auth_scheme_coverage, unmatched_cassettes = build_auth_scheme_coverage(cassettes_dir, options)
135
+ tested_schemes = {opt.scheme_name for opt in options if not opt.untested}
136
+ untested_schemes = {opt.scheme_name for opt in options if opt.untested}
137
+ covered_schemes = {scheme for scheme in auth_scheme_coverage.keys() if scheme is not None}
138
+
139
+ # Find missing tested schemes (errors)
140
+ missing_tested = tested_schemes - covered_schemes
141
+ for scheme in sorted(missing_tested):
142
+ errors.append(
143
+ f"Auth scheme '{scheme}' has no cassette coverage. "
144
+ f"Record at least one cassette using this authentication method, "
145
+ f"or add 'x-airbyte-untested: true' to skip this check."
146
+ )
147
+
148
+ # Warn about untested schemes without coverage
149
+ missing_untested = untested_schemes - covered_schemes
150
+ for scheme in sorted(missing_untested):
151
+ warnings.append(
152
+ f"Auth scheme '{scheme}' is marked as untested (x-airbyte-untested: true) " f"and has no cassette coverage. Validation skipped."
153
+ )
154
+
155
+ # Warn about cassettes that couldn't be matched to any auth scheme
156
+ for cassette_path, auth_config_keys in unmatched_cassettes:
157
+ warnings.append(f"Cassette '{cassette_path.name}' could not be matched to any auth scheme. " f"auth_config keys: {sorted(auth_config_keys)}")
158
+
159
+ is_valid = len(missing_tested) == 0
160
+ return is_valid, errors, warnings, sorted(covered_schemes), unmatched_cassettes
161
+
162
+
56
163
  def validate_response_against_schema(response_body: Any, schema: Dict[str, Any]) -> Tuple[bool, List[str]]:
57
164
  """Validate a response body against a JSON schema.
58
165
 
@@ -588,6 +695,9 @@ def validate_connector_readiness(connector_dir: str | Path) -> Dict[str, Any]:
588
695
  cassettes_dir = connector_path / "tests" / "cassettes"
589
696
  cassette_map = build_cassette_map(cassettes_dir)
590
697
 
698
+ # Validate auth scheme coverage for multi-auth connectors
699
+ auth_valid, auth_errors, auth_warnings, auth_covered_schemes, auth_unmatched_cassettes = validate_auth_scheme_coverage(config, cassettes_dir)
700
+
591
701
  validation_results = []
592
702
  total_operations = 0
593
703
  operations_with_cassettes = 0
@@ -827,8 +937,12 @@ def validate_connector_readiness(connector_dir: str | Path) -> Dict[str, Any]:
827
937
  if replication_result.get("registry_found", False):
828
938
  total_warnings += len(replication_warnings)
829
939
 
830
- # Update success criteria to include replication validation
831
- success = operations_missing_cassettes == 0 and cassettes_invalid == 0 and total_operations > 0 and len(replication_errors) == 0
940
+ # Merge auth scheme validation errors/warnings into totals
941
+ total_errors += len(auth_errors)
942
+ total_warnings += len(auth_warnings)
943
+
944
+ # Update success criteria to include replication and auth scheme validation
945
+ success = operations_missing_cassettes == 0 and cassettes_invalid == 0 and total_operations > 0 and len(replication_errors) == 0 and auth_valid
832
946
 
833
947
  # Check for preferred_for_check on at least one list operation
834
948
  has_preferred_check = False
@@ -849,12 +963,26 @@ def validate_connector_readiness(connector_dir: str | Path) -> Dict[str, Any]:
849
963
  "to enable reliable health checks."
850
964
  )
851
965
 
966
+ # Build auth scheme validation result
967
+ options = config.auth.options or []
968
+ tested_schemes = [opt.scheme_name for opt in options if not opt.untested]
969
+ untested_schemes_list = [opt.scheme_name for opt in options if opt.untested]
970
+ missing_tested = [s for s in tested_schemes if s not in auth_covered_schemes]
971
+
852
972
  return {
853
973
  "success": success,
854
974
  "connector_name": config.name,
855
975
  "connector_path": str(connector_path),
856
976
  "validation_results": validation_results,
857
977
  "replication_validation": replication_result,
978
+ "auth_scheme_validation": {
979
+ "valid": auth_valid,
980
+ "errors": auth_errors,
981
+ "warnings": auth_warnings,
982
+ "covered_schemes": auth_covered_schemes,
983
+ "missing_schemes": missing_tested,
984
+ "untested_schemes": untested_schemes_list,
985
+ },
858
986
  "readiness_warnings": readiness_warnings,
859
987
  "summary": {
860
988
  "total_operations": total_operations,
@@ -38,6 +38,8 @@ from .types import (
38
38
  )
39
39
  if TYPE_CHECKING:
40
40
  from .models import SlackAuthConfig
41
+ from .models import SlackReplicationConfig
42
+
41
43
  # Import specific auth config classes for multi-auth isinstance checks
42
44
  from .models import SlackTokenAuthenticationAuthConfig, SlackOauth20AuthenticationAuthConfig
43
45
  # Import response models and envelope models at runtime
@@ -108,7 +110,7 @@ class SlackConnector:
108
110
  """
109
111
 
110
112
  connector_name = "slack"
111
- connector_version = "0.1.11"
113
+ connector_version = "0.1.12"
112
114
  vendored_sdk_version = "0.1.0" # Version of vendored connector-sdk
113
115
 
114
116
  # Map of (entity, action) -> needs_envelope for envelope wrapping decision
@@ -152,26 +154,35 @@ class SlackConnector:
152
154
  external_user_id: str | None = None,
153
155
  airbyte_client_id: str | None = None,
154
156
  airbyte_client_secret: str | None = None,
157
+ connector_id: str | None = None,
155
158
  on_token_refresh: Any | None = None ):
156
159
  """
157
160
  Initialize a new slack connector instance.
158
161
 
159
162
  Supports both local and hosted execution modes:
160
163
  - Local mode: Provide `auth_config` for direct API calls
161
- - Hosted mode: Provide `external_user_id`, `airbyte_client_id`, and `airbyte_client_secret` for hosted execution
164
+ - Hosted mode: Provide Airbyte credentials with either `connector_id` or `external_user_id`
162
165
 
163
166
  Args:
164
167
  auth_config: Typed authentication configuration (required for local mode)
165
- external_user_id: External user ID (required for hosted mode)
168
+ external_user_id: External user ID (for hosted mode lookup)
166
169
  airbyte_client_id: Airbyte OAuth client ID (required for hosted mode)
167
170
  airbyte_client_secret: Airbyte OAuth client secret (required for hosted mode)
171
+ connector_id: Specific connector/source ID (for hosted mode, skips lookup)
168
172
  on_token_refresh: Optional callback for OAuth2 token refresh persistence.
169
173
  Called with new_tokens dict when tokens are refreshed. Can be sync or async.
170
174
  Example: lambda tokens: save_to_database(tokens)
171
175
  Examples:
172
176
  # Local mode (direct API calls)
173
177
  connector = SlackConnector(auth_config=SlackAuthConfig(api_token="..."))
174
- # Hosted mode (executed on Airbyte cloud)
178
+ # Hosted mode with explicit connector_id (no lookup needed)
179
+ connector = SlackConnector(
180
+ airbyte_client_id="client_abc123",
181
+ airbyte_client_secret="secret_xyz789",
182
+ connector_id="existing-source-uuid"
183
+ )
184
+
185
+ # Hosted mode with lookup by external_user_id
175
186
  connector = SlackConnector(
176
187
  external_user_id="user-123",
177
188
  airbyte_client_id="client_abc123",
@@ -189,21 +200,24 @@ class SlackConnector:
189
200
  on_token_refresh=save_tokens
190
201
  )
191
202
  """
192
- # Hosted mode: external_user_id, airbyte_client_id, and airbyte_client_secret provided
193
- if external_user_id and airbyte_client_id and airbyte_client_secret:
203
+ # Hosted mode: Airbyte credentials + either connector_id OR external_user_id
204
+ is_hosted = airbyte_client_id and airbyte_client_secret and (connector_id or external_user_id)
205
+
206
+ if is_hosted:
194
207
  from ._vendored.connector_sdk.executor import HostedExecutor
195
208
  self._executor = HostedExecutor(
196
- external_user_id=external_user_id,
197
209
  airbyte_client_id=airbyte_client_id,
198
210
  airbyte_client_secret=airbyte_client_secret,
199
- connector_definition_id=str(SlackConnectorModel.id),
211
+ connector_id=connector_id,
212
+ external_user_id=external_user_id,
213
+ connector_definition_id=str(SlackConnectorModel.id) if not connector_id else None,
200
214
  )
201
215
  else:
202
216
  # Local mode: auth_config required
203
217
  if not auth_config:
204
218
  raise ValueError(
205
- "Either provide (external_user_id, airbyte_client_id, airbyte_client_secret) for hosted mode "
206
- "or auth_config for local mode"
219
+ "Either provide Airbyte credentials (airbyte_client_id, airbyte_client_secret) with "
220
+ "connector_id or external_user_id for hosted mode, or auth_config for local mode"
207
221
  )
208
222
 
209
223
  from ._vendored.connector_sdk.executor import LocalExecutor
@@ -567,6 +581,105 @@ class SlackConnector:
567
581
  )
568
582
  return entity_def.entity_schema if entity_def else None
569
583
 
584
+ @property
585
+ def connector_id(self) -> str | None:
586
+ """Get the connector/source ID (only available in hosted mode).
587
+
588
+ Returns:
589
+ The connector ID if in hosted mode, None if in local mode.
590
+
591
+ Example:
592
+ connector = await SlackConnector.create_hosted(...)
593
+ print(f"Created connector: {connector.connector_id}")
594
+ """
595
+ if hasattr(self, '_executor') and hasattr(self._executor, '_connector_id'):
596
+ return self._executor._connector_id
597
+ return None
598
+
599
+ # ===== HOSTED MODE FACTORY =====
600
+
601
+ @classmethod
602
+ async def create_hosted(
603
+ cls,
604
+ *,
605
+ external_user_id: str,
606
+ airbyte_client_id: str,
607
+ airbyte_client_secret: str,
608
+ auth_config: "SlackAuthConfig",
609
+ name: str | None = None,
610
+ replication_config: "SlackReplicationConfig" | None = None,
611
+ ) -> "SlackConnector":
612
+ """
613
+ Create a new hosted connector on Airbyte Cloud.
614
+
615
+ This factory method:
616
+ 1. Creates a source on Airbyte Cloud with the provided credentials
617
+ 2. Returns a connector configured with the new connector_id
618
+
619
+ Args:
620
+ external_user_id: Workspace identifier in Airbyte Cloud
621
+ airbyte_client_id: Airbyte OAuth client ID
622
+ airbyte_client_secret: Airbyte OAuth client secret
623
+ auth_config: Typed auth config (same as local mode)
624
+ name: Optional source name (defaults to connector name + external_user_id)
625
+ replication_config: Typed replication settings.
626
+ Required for connectors with x-airbyte-replication-config (REPLICATION mode sources).
627
+
628
+ Returns:
629
+ A SlackConnector instance configured in hosted mode
630
+
631
+ Example:
632
+ # Create a new hosted connector with API key auth
633
+ connector = await SlackConnector.create_hosted(
634
+ external_user_id="my-workspace",
635
+ airbyte_client_id="client_abc",
636
+ airbyte_client_secret="secret_xyz",
637
+ auth_config=SlackAuthConfig(api_token="..."),
638
+ )
639
+
640
+ # With replication config (required for this connector):
641
+ connector = await SlackConnector.create_hosted(
642
+ external_user_id="my-workspace",
643
+ airbyte_client_id="client_abc",
644
+ airbyte_client_secret="secret_xyz",
645
+ auth_config=SlackAuthConfig(api_token="..."),
646
+ replication_config=SlackReplicationConfig(start_date="...", lookback_window="...", join_channels="..."),
647
+ )
648
+
649
+ # Use the connector
650
+ result = await connector.execute("entity", "list", {})
651
+ """
652
+ from ._vendored.connector_sdk.cloud_utils import AirbyteCloudClient
653
+
654
+ client = AirbyteCloudClient(
655
+ client_id=airbyte_client_id,
656
+ client_secret=airbyte_client_secret,
657
+ )
658
+
659
+ try:
660
+ # Build credentials from auth_config
661
+ credentials = auth_config.model_dump(exclude_none=True)
662
+ replication_config_dict = replication_config.model_dump(exclude_none=True) if replication_config else None
663
+
664
+ # Create source on Airbyte Cloud
665
+ source_name = name or f"{cls.connector_name} - {external_user_id}"
666
+ source_id = await client.create_source(
667
+ name=source_name,
668
+ connector_definition_id=str(SlackConnectorModel.id),
669
+ external_user_id=external_user_id,
670
+ credentials=credentials,
671
+ replication_config=replication_config_dict,
672
+ )
673
+ finally:
674
+ await client.close()
675
+
676
+ # Return connector configured with the new connector_id
677
+ return cls(
678
+ airbyte_client_id=airbyte_client_id,
679
+ airbyte_client_secret=airbyte_client_secret,
680
+ connector_id=source_id,
681
+ )
682
+
570
683
 
571
684
 
572
685
  class UsersQuery:
@@ -27,7 +27,7 @@ from uuid import (
27
27
  SlackConnectorModel: ConnectorModel = ConnectorModel(
28
28
  id=UUID('c2281cee-86f9-4a86-bb48-d23286b4c7bd'),
29
29
  name='slack',
30
- version='0.1.11',
30
+ version='0.1.12',
31
31
  base_url='https://slack.com/api',
32
32
  auth=AuthConfig(
33
33
  options=[
@@ -88,6 +88,7 @@ SlackConnectorModel: ConnectorModel = ConnectorModel(
88
88
  },
89
89
  replication_auth_key_constants={'credentials.option_title': 'Default OAuth2.0 authorization'},
90
90
  ),
91
+ untested=True,
91
92
  ),
92
93
  ],
93
94
  ),
@@ -34,6 +34,20 @@ class SlackOauth20AuthenticationAuthConfig(BaseModel):
34
34
 
35
35
  SlackAuthConfig = SlackTokenAuthenticationAuthConfig | SlackOauth20AuthenticationAuthConfig
36
36
 
37
+ # Replication configuration
38
+
39
+ class SlackReplicationConfig(BaseModel):
40
+ """Replication Configuration - Settings for data replication from Slack."""
41
+
42
+ model_config = ConfigDict(extra="forbid")
43
+
44
+ start_date: str
45
+ """UTC date and time in the format YYYY-MM-DDTHH:mm:ssZ from which to start replicating data."""
46
+ lookback_window: int
47
+ """Number of days to look back when syncing data (0-365)."""
48
+ join_channels: bool
49
+ """Whether to automatically join public channels to sync messages."""
50
+
37
51
  # ===== RESPONSE TYPE DEFINITIONS (PYDANTIC) =====
38
52
 
39
53
  class User(BaseModel):
@@ -176,14 +190,6 @@ class ChannelResponse(BaseModel):
176
190
  ok: Union[bool, Any] = Field(default=None)
177
191
  channel: Union[Channel, Any] = Field(default=None)
178
192
 
179
- class Reaction(BaseModel):
180
- """Message reaction"""
181
- model_config = ConfigDict(extra="allow", populate_by_name=True)
182
-
183
- name: Union[str | None, Any] = Field(default=None)
184
- users: Union[list[str] | None, Any] = Field(default=None)
185
- count: Union[int | None, Any] = Field(default=None)
186
-
187
193
  class Attachment(BaseModel):
188
194
  """Message attachment"""
189
195
  model_config = ConfigDict(extra="allow", populate_by_name=True)
@@ -229,6 +235,14 @@ class File(BaseModel):
229
235
  created: Union[int | None, Any] = Field(default=None)
230
236
  timestamp: Union[int | None, Any] = Field(default=None)
231
237
 
238
+ class Reaction(BaseModel):
239
+ """Message reaction"""
240
+ model_config = ConfigDict(extra="allow", populate_by_name=True)
241
+
242
+ name: Union[str | None, Any] = Field(default=None)
243
+ users: Union[list[str] | None, Any] = Field(default=None)
244
+ count: Union[int | None, Any] = Field(default=None)
245
+
232
246
  class Message(BaseModel):
233
247
  """Slack message object"""
234
248
  model_config = ConfigDict(extra="allow", populate_by_name=True)
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: airbyte-agent-slack
3
- Version: 0.1.37
3
+ Version: 0.1.39
4
4
  Summary: Airbyte Slack Connector for AI platforms
5
5
  Project-URL: Homepage, https://github.com/airbytehq/airbyte-agent-connectors
6
6
  Project-URL: Documentation, https://docs.airbyte.com/ai-agents/
@@ -158,7 +158,7 @@ See the official [Slack API reference](https://api.slack.com/methods).
158
158
 
159
159
  ## Version information
160
160
 
161
- - **Package version:** 0.1.37
162
- - **Connector version:** 0.1.11
163
- - **Generated with Connector SDK commit SHA:** 580ea1221eff062f41b1065a155124c85861cb18
161
+ - **Package version:** 0.1.39
162
+ - **Connector version:** 0.1.12
163
+ - **Generated with Connector SDK commit SHA:** b184da3e22ef8521d2eeebf3c96a0fe8da2424f5
164
164
  - **Changelog:** [View changelog](https://github.com/airbytehq/airbyte-agent-connectors/blob/main/connectors/slack/CHANGELOG.md)
@@ -1,28 +1,28 @@
1
- airbyte_agent_slack/__init__.py,sha256=BeUNOHVHPcONYMOwcQjQWhamKVjdDju5m72ZXmASWj0,3749
2
- airbyte_agent_slack/connector.py,sha256=qNu3u9mWv4prN3xLejXYazFyII6e5pfXHfHo0pUxxd8,41130
3
- airbyte_agent_slack/connector_model.py,sha256=YMQvFPtRdvjCvsiB8Hmk95I-vnBfSCWmWb3aLCU42Yo,194546
4
- airbyte_agent_slack/models.py,sha256=Tw-n-wzmdiWmaGGBtYI_6Rd9JD-rQ3Ylp5kZRnm8BUw,29511
1
+ airbyte_agent_slack/__init__.py,sha256=Qkfb6sqZC1mlHqGIp87Le__5wQP-ZU0SZf6asFljrt4,3807
2
+ airbyte_agent_slack/connector.py,sha256=2am44Z6ezc1oI81VbvJaUIOvkgcOh4FRcZZFvLong3Q,45619
3
+ airbyte_agent_slack/connector_model.py,sha256=W8DFMX20VxPr2IxTzEIjYcYNSt-E0JNj1b5UizuaWV0,194577
4
+ airbyte_agent_slack/models.py,sha256=AqOTLa-0w6csj8113pM5YHLXVwpEOrtVq1KDHWZm-EE,30019
5
5
  airbyte_agent_slack/types.py,sha256=QDLOYuH23w4H_5fxVjTcBBz6BMYm2PWbuM1K_EOcD4o,32154
6
6
  airbyte_agent_slack/_vendored/__init__.py,sha256=ILl7AHXMui__swyrjxrh9yRa4dLiwBvV6axPWFWty80,38
7
7
  airbyte_agent_slack/_vendored/connector_sdk/__init__.py,sha256=T5o7roU6NSpH-lCAGZ338sE5dlh4ZU6i6IkeG1zpems,1949
8
8
  airbyte_agent_slack/_vendored/connector_sdk/auth_strategies.py,sha256=5Sb9moUp623o67Q2wMa8iZldJH08y4gQdoutoO_75Iw,42088
9
9
  airbyte_agent_slack/_vendored/connector_sdk/auth_template.py,sha256=nju4jqlFC_KI82ILNumNIyiUtRJcy7J94INIZ0QraI4,4454
10
- airbyte_agent_slack/_vendored/connector_sdk/connector_model_loader.py,sha256=ecm0Cdj1SyhDOpEi2wUzzrJNkgt0wViB0Q-YTrDPsjU,41698
10
+ airbyte_agent_slack/_vendored/connector_sdk/connector_model_loader.py,sha256=1AAvSvjxM9Nuto6w7D6skN5VXGb4e6na0lMFcFmmVkI,41761
11
11
  airbyte_agent_slack/_vendored/connector_sdk/constants.py,sha256=AtzOvhDMWbRJgpsQNWl5tkogHD6mWgEY668PgRmgtOY,2737
12
12
  airbyte_agent_slack/_vendored/connector_sdk/exceptions.py,sha256=ss5MGv9eVPmsbLcLWetuu3sDmvturwfo6Pw3M37Oq5k,481
13
13
  airbyte_agent_slack/_vendored/connector_sdk/extensions.py,sha256=XWRRoJOOrwUHSKbuQt5DU7CCu8ePzhd_HuP7c_uD77w,21376
14
14
  airbyte_agent_slack/_vendored/connector_sdk/http_client.py,sha256=09Fclbq4wrg38EM2Yh2kHiykQVXqdAGby024elcEz8E,28027
15
15
  airbyte_agent_slack/_vendored/connector_sdk/introspection.py,sha256=e9uWn2ofpeehoBbzNgts_bjlKLn8ayA1Y3OpDC3b7ZA,19517
16
16
  airbyte_agent_slack/_vendored/connector_sdk/secrets.py,sha256=J9ezMu4xNnLW11xY5RCre6DHP7YMKZCqwGJfk7ufHAM,6855
17
- airbyte_agent_slack/_vendored/connector_sdk/types.py,sha256=sRZ4rRkJXGfqdjfF3qb0kzrw_tu1cn-CmFSZKMMgF7o,9269
18
- airbyte_agent_slack/_vendored/connector_sdk/utils.py,sha256=G4LUXOC2HzPoND2v4tQW68R9uuPX9NQyCjaGxb7Kpl0,1958
19
- airbyte_agent_slack/_vendored/connector_sdk/validation.py,sha256=iz1wo5CrHT_yD4elBGM9TSCFAegVmkEZiRKCpjSu7IA,33997
17
+ airbyte_agent_slack/_vendored/connector_sdk/types.py,sha256=MsWJsQy779r7Mqiqf_gh_4Vs6VDqieoMjLPyWt7qhu8,9412
18
+ airbyte_agent_slack/_vendored/connector_sdk/utils.py,sha256=UYwYuSLhsDD-4C0dBs7Qy0E0gIcFZXb6VWadJORhQQU,4080
19
+ airbyte_agent_slack/_vendored/connector_sdk/validation.py,sha256=w5WGnmILkdBslpXhAXhKhE-c8ANBc_OZQxr_fUeAgtc,39666
20
20
  airbyte_agent_slack/_vendored/connector_sdk/validation_replication.py,sha256=v7F5YWd5m4diIF7_4m4nOkC9crg97vqRUUkt9ka9HZ4,36043
21
21
  airbyte_agent_slack/_vendored/connector_sdk/cloud_utils/__init__.py,sha256=4799Hv9f2zxDVj1aLyQ8JpTEuFTp_oOZMRz-NZCdBJg,134
22
- airbyte_agent_slack/_vendored/connector_sdk/cloud_utils/client.py,sha256=YxdRpQr9XjDzih6csSseBVGn9kfMtaqbOCXP0TPuzFY,7189
22
+ airbyte_agent_slack/_vendored/connector_sdk/cloud_utils/client.py,sha256=-_ibaHVa7KBBw8SnMuxpWz6XkrSgNTFdMgDfTChtywg,9505
23
23
  airbyte_agent_slack/_vendored/connector_sdk/executor/__init__.py,sha256=EmG9YQNAjSuYCVB4D5VoLm4qpD1KfeiiOf7bpALj8p8,702
24
- airbyte_agent_slack/_vendored/connector_sdk/executor/hosted_executor.py,sha256=AC5aJtdcMPQfRuam0ZGE4QdhUBu2oGEmNZ6oVQLHTE8,7212
25
- airbyte_agent_slack/_vendored/connector_sdk/executor/local_executor.py,sha256=l2b0VILgtOlUNyG19J0UkbCa6rJXkU8g1uzLot8BJjQ,77008
24
+ airbyte_agent_slack/_vendored/connector_sdk/executor/hosted_executor.py,sha256=tv0njAdy-gdHBg4izgcxhEWYbrNiBifEYEca9AWzaL0,8693
25
+ airbyte_agent_slack/_vendored/connector_sdk/executor/local_executor.py,sha256=RtdTXFzfoJz5Coz9nwQi81Df1402BRgO1Mgd3ZzTkfw,76581
26
26
  airbyte_agent_slack/_vendored/connector_sdk/executor/models.py,sha256=mUUBnuShKXxVIfsTOhMiI2rn2a-50jJG7SFGKT_P6Jk,6281
27
27
  airbyte_agent_slack/_vendored/connector_sdk/http/__init__.py,sha256=y8fbzZn-3yV9OxtYz8Dy6FFGI5v6TOqADd1G3xHH3Hw,911
28
28
  airbyte_agent_slack/_vendored/connector_sdk/http/config.py,sha256=6J7YIIwHC6sRu9i-yKa5XvArwK2KU60rlnmxzDZq3lw,3283
@@ -48,11 +48,11 @@ airbyte_agent_slack/_vendored/connector_sdk/schema/components.py,sha256=nJIPieav
48
48
  airbyte_agent_slack/_vendored/connector_sdk/schema/connector.py,sha256=mSZk1wr2YSdRj9tTRsPAuIlCzd_xZLw-Bzl1sMwE0rE,3731
49
49
  airbyte_agent_slack/_vendored/connector_sdk/schema/extensions.py,sha256=5hgpFHK7fzpzegCkJk882DeIP79bCx_qairKJhvPMZ8,9590
50
50
  airbyte_agent_slack/_vendored/connector_sdk/schema/operations.py,sha256=St-A75m6sZUZlsoM6WcoPaShYu_X1K19pdyPvJbabOE,6214
51
- airbyte_agent_slack/_vendored/connector_sdk/schema/security.py,sha256=1CVCavrPdHHyk7B6JtUD75yRS_hWLCemZF1zwGbdqxg,9036
51
+ airbyte_agent_slack/_vendored/connector_sdk/schema/security.py,sha256=R-21DLnp-ANIRO1Dzqo53TYFJL6lCp0aO8GSuxa_bDI,9225
52
52
  airbyte_agent_slack/_vendored/connector_sdk/telemetry/__init__.py,sha256=RaLgkBU4dfxn1LC5Y0Q9rr2PJbrwjxvPgBLmq8_WafE,211
53
53
  airbyte_agent_slack/_vendored/connector_sdk/telemetry/config.py,sha256=tLmQwAFD0kP1WyBGWBS3ysaudN9H3e-3EopKZi6cGKg,885
54
54
  airbyte_agent_slack/_vendored/connector_sdk/telemetry/events.py,sha256=8Y1NbXiwISX-V_wRofY7PqcwEXD0dLMnntKkY6XFU2s,1328
55
55
  airbyte_agent_slack/_vendored/connector_sdk/telemetry/tracker.py,sha256=SginFQbHqVUVYG82NnNzG34O-tAQ_wZYjGDcuo0q4Kk,5584
56
- airbyte_agent_slack-0.1.37.dist-info/METADATA,sha256=Ej9k0_poIJ3le8QrprurVhdTlMTFamzDq_ierH9awTk,6597
57
- airbyte_agent_slack-0.1.37.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
58
- airbyte_agent_slack-0.1.37.dist-info/RECORD,,
56
+ airbyte_agent_slack-0.1.39.dist-info/METADATA,sha256=8C5smFY0ODal6gobxu5y0xr4GWtlqTlmArcv518wcvk,6597
57
+ airbyte_agent_slack-0.1.39.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
58
+ airbyte_agent_slack-0.1.39.dist-info/RECORD,,