airbyte-agent-orb 0.1.6__py3-none-any.whl → 0.1.7__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.
- airbyte_agent_orb/__init__.py +2 -2
- airbyte_agent_orb/_vendored/connector_sdk/schema/base.py +32 -1
- airbyte_agent_orb/_vendored/connector_sdk/schema/extensions.py +19 -1
- airbyte_agent_orb/_vendored/connector_sdk/schema/security.py +37 -0
- airbyte_agent_orb/models.py +16 -16
- {airbyte_agent_orb-0.1.6.dist-info → airbyte_agent_orb-0.1.7.dist-info}/METADATA +3 -3
- {airbyte_agent_orb-0.1.6.dist-info → airbyte_agent_orb-0.1.7.dist-info}/RECORD +8 -8
- {airbyte_agent_orb-0.1.6.dist-info → airbyte_agent_orb-0.1.7.dist-info}/WHEEL +0 -0
airbyte_agent_orb/__init__.py
CHANGED
|
@@ -20,9 +20,9 @@ from .models import (
|
|
|
20
20
|
PlanPricesItem,
|
|
21
21
|
Plan,
|
|
22
22
|
PlansList,
|
|
23
|
-
InvoiceLineItemsItem,
|
|
24
23
|
InvoiceSubscription,
|
|
25
24
|
InvoiceCustomer,
|
|
25
|
+
InvoiceLineItemsItem,
|
|
26
26
|
Invoice,
|
|
27
27
|
InvoicesList,
|
|
28
28
|
CustomersListResultMeta,
|
|
@@ -88,9 +88,9 @@ __all__ = [
|
|
|
88
88
|
"PlanPricesItem",
|
|
89
89
|
"Plan",
|
|
90
90
|
"PlansList",
|
|
91
|
-
"InvoiceLineItemsItem",
|
|
92
91
|
"InvoiceSubscription",
|
|
93
92
|
"InvoiceCustomer",
|
|
93
|
+
"InvoiceLineItemsItem",
|
|
94
94
|
"Invoice",
|
|
95
95
|
"InvoicesList",
|
|
96
96
|
"CustomersListResultMeta",
|
|
@@ -10,7 +10,7 @@ from enum import StrEnum
|
|
|
10
10
|
from typing import Any, Dict
|
|
11
11
|
from uuid import UUID
|
|
12
12
|
|
|
13
|
-
from pydantic import BaseModel, ConfigDict, Field, field_validator
|
|
13
|
+
from pydantic import BaseModel, ConfigDict, Field, field_validator, model_validator
|
|
14
14
|
from pydantic_core import Url
|
|
15
15
|
|
|
16
16
|
from .extensions import CacheConfig, ReplicationConfig, RetryConfig
|
|
@@ -210,3 +210,34 @@ class Server(BaseModel):
|
|
|
210
210
|
raise ValueError("Server URL cannot be empty")
|
|
211
211
|
# Allow both absolute URLs and relative paths
|
|
212
212
|
return v
|
|
213
|
+
|
|
214
|
+
@model_validator(mode="after")
|
|
215
|
+
def validate_replication_environment_mapping(self) -> "Server":
|
|
216
|
+
"""Validate that x-airbyte-replication-environment-mapping sources exist in variables.
|
|
217
|
+
|
|
218
|
+
For simple mappings like {"subdomain": "subdomain"}, the key is the source variable.
|
|
219
|
+
For transform mappings like {"domain": {"source": "subdomain", "format": "..."}},
|
|
220
|
+
the "source" field is the source variable.
|
|
221
|
+
"""
|
|
222
|
+
env_mapping = self.x_airbyte_replication_environment_mapping
|
|
223
|
+
if not env_mapping or not self.variables:
|
|
224
|
+
return self
|
|
225
|
+
|
|
226
|
+
variable_names = set(self.variables.keys())
|
|
227
|
+
|
|
228
|
+
for env_key, mapping_value in env_mapping.items():
|
|
229
|
+
if isinstance(mapping_value, str):
|
|
230
|
+
source_var = env_key
|
|
231
|
+
elif isinstance(mapping_value, EnvironmentMappingTransform):
|
|
232
|
+
source_var = mapping_value.source
|
|
233
|
+
else:
|
|
234
|
+
continue
|
|
235
|
+
|
|
236
|
+
if source_var not in variable_names:
|
|
237
|
+
available = ", ".join(sorted(variable_names)) if variable_names else "(none)"
|
|
238
|
+
raise ValueError(
|
|
239
|
+
f"x-airbyte-replication-environment-mapping: source variable '{source_var}' "
|
|
240
|
+
f"not found in server variables. Available: {available}"
|
|
241
|
+
)
|
|
242
|
+
|
|
243
|
+
return self
|
|
@@ -14,7 +14,7 @@ are implemented.
|
|
|
14
14
|
|
|
15
15
|
from typing import Literal
|
|
16
16
|
|
|
17
|
-
from pydantic import BaseModel, ConfigDict, Field
|
|
17
|
+
from pydantic import BaseModel, ConfigDict, Field, model_validator
|
|
18
18
|
|
|
19
19
|
|
|
20
20
|
class PaginationConfig(BaseModel):
|
|
@@ -252,6 +252,24 @@ class ReplicationConfig(BaseModel):
|
|
|
252
252
|
description="Mapping from replication_config field names to source_config field names",
|
|
253
253
|
)
|
|
254
254
|
|
|
255
|
+
@model_validator(mode="after")
|
|
256
|
+
def validate_replication_config_key_mapping(self) -> "ReplicationConfig":
|
|
257
|
+
"""Validate that replication_config_key_mapping keys exist in properties.
|
|
258
|
+
|
|
259
|
+
The mapping is: {local_key: airbyte_path}
|
|
260
|
+
We validate that local_key exists in our properties.
|
|
261
|
+
"""
|
|
262
|
+
if self.replication_config_key_mapping and self.properties:
|
|
263
|
+
property_names = set(self.properties.keys())
|
|
264
|
+
for local_key, airbyte_path in self.replication_config_key_mapping.items():
|
|
265
|
+
if local_key not in property_names:
|
|
266
|
+
available = ", ".join(sorted(property_names)) if property_names else "(none)"
|
|
267
|
+
raise ValueError(
|
|
268
|
+
f"replication_config_key_mapping: local key '{local_key}' "
|
|
269
|
+
f"(mapped to '{airbyte_path}') not found in properties. Available: {available}"
|
|
270
|
+
)
|
|
271
|
+
return self
|
|
272
|
+
|
|
255
273
|
|
|
256
274
|
class CacheConfig(BaseModel):
|
|
257
275
|
"""
|
|
@@ -81,6 +81,21 @@ class AuthConfigOption(BaseModel):
|
|
|
81
81
|
description="Mapping from source config paths (e.g., 'credentials.api_key') to auth config keys for direct connectors",
|
|
82
82
|
)
|
|
83
83
|
|
|
84
|
+
@model_validator(mode="after")
|
|
85
|
+
def validate_replication_auth_key_mapping(self) -> "AuthConfigOption":
|
|
86
|
+
"""Validate that replication_auth_key_mapping target keys exist in properties."""
|
|
87
|
+
if self.replication_auth_key_mapping and self.properties:
|
|
88
|
+
property_names = set(self.properties.keys())
|
|
89
|
+
for airbyte_path, our_key in self.replication_auth_key_mapping.items():
|
|
90
|
+
if our_key not in property_names:
|
|
91
|
+
option_context = f"oneOf option '{self.title}'" if self.title else "oneOf option"
|
|
92
|
+
available = ", ".join(sorted(property_names)) if property_names else "(none)"
|
|
93
|
+
raise ValueError(
|
|
94
|
+
f"replication_auth_key_mapping in {option_context}: target key '{our_key}' "
|
|
95
|
+
f"(mapped from '{airbyte_path}') not found in properties. Available: {available}"
|
|
96
|
+
)
|
|
97
|
+
return self
|
|
98
|
+
|
|
84
99
|
|
|
85
100
|
class AirbyteAuthConfig(BaseModel):
|
|
86
101
|
"""
|
|
@@ -146,8 +161,30 @@ class AirbyteAuthConfig(BaseModel):
|
|
|
146
161
|
if not self.auth_mapping:
|
|
147
162
|
raise ValueError("Single auth option must have auth_mapping")
|
|
148
163
|
|
|
164
|
+
# Validate replication_auth_key_mapping targets exist in properties
|
|
165
|
+
if self.replication_auth_key_mapping and self.properties:
|
|
166
|
+
self._validate_replication_auth_key_mapping(self.replication_auth_key_mapping, self.properties, context="x-airbyte-auth-config")
|
|
167
|
+
|
|
149
168
|
return self
|
|
150
169
|
|
|
170
|
+
@staticmethod
|
|
171
|
+
def _validate_replication_auth_key_mapping(mapping: Dict[str, str], properties: Dict[str, AuthConfigFieldSpec], context: str) -> None:
|
|
172
|
+
"""Validate that replication_auth_key_mapping target keys exist in properties.
|
|
173
|
+
|
|
174
|
+
Args:
|
|
175
|
+
mapping: The replication_auth_key_mapping dict (airbyte_path -> our_key)
|
|
176
|
+
properties: The properties dict from x-airbyte-auth-config
|
|
177
|
+
context: Context string for error messages
|
|
178
|
+
"""
|
|
179
|
+
property_names = set(properties.keys())
|
|
180
|
+
for airbyte_path, our_key in mapping.items():
|
|
181
|
+
if our_key not in property_names:
|
|
182
|
+
available = ", ".join(sorted(property_names)) if property_names else "(none)"
|
|
183
|
+
raise ValueError(
|
|
184
|
+
f"replication_auth_key_mapping in {context}: target key '{our_key}' "
|
|
185
|
+
f"(mapped from '{airbyte_path}') not found in properties. Available: {available}"
|
|
186
|
+
)
|
|
187
|
+
|
|
151
188
|
|
|
152
189
|
class SecurityScheme(BaseModel):
|
|
153
190
|
"""
|
airbyte_agent_orb/models.py
CHANGED
|
@@ -180,6 +180,22 @@ class PlansList(BaseModel):
|
|
|
180
180
|
data: Union[list[Plan], Any] = Field(default=None)
|
|
181
181
|
pagination_metadata: Union[PaginationMetadata, Any] = Field(default=None)
|
|
182
182
|
|
|
183
|
+
class InvoiceSubscription(BaseModel):
|
|
184
|
+
"""The subscription associated with the invoice"""
|
|
185
|
+
model_config = ConfigDict(extra="allow", populate_by_name=True)
|
|
186
|
+
|
|
187
|
+
id: Union[str | None, Any] = Field(default=None, description="The subscription ID")
|
|
188
|
+
"""The subscription ID"""
|
|
189
|
+
|
|
190
|
+
class InvoiceCustomer(BaseModel):
|
|
191
|
+
"""The customer associated with the invoice"""
|
|
192
|
+
model_config = ConfigDict(extra="allow", populate_by_name=True)
|
|
193
|
+
|
|
194
|
+
id: Union[str | None, Any] = Field(default=None, description="The customer ID")
|
|
195
|
+
"""The customer ID"""
|
|
196
|
+
external_customer_id: Union[str | None, Any] = Field(default=None, description="The external customer ID")
|
|
197
|
+
"""The external customer ID"""
|
|
198
|
+
|
|
183
199
|
class InvoiceLineItemsItem(BaseModel):
|
|
184
200
|
"""Nested schema for Invoice.line_items_item"""
|
|
185
201
|
model_config = ConfigDict(extra="allow", populate_by_name=True)
|
|
@@ -197,22 +213,6 @@ class InvoiceLineItemsItem(BaseModel):
|
|
|
197
213
|
end_date: Union[str | None, Any] = Field(default=None, description="The end date of the line item")
|
|
198
214
|
"""The end date of the line item"""
|
|
199
215
|
|
|
200
|
-
class InvoiceSubscription(BaseModel):
|
|
201
|
-
"""The subscription associated with the invoice"""
|
|
202
|
-
model_config = ConfigDict(extra="allow", populate_by_name=True)
|
|
203
|
-
|
|
204
|
-
id: Union[str | None, Any] = Field(default=None, description="The subscription ID")
|
|
205
|
-
"""The subscription ID"""
|
|
206
|
-
|
|
207
|
-
class InvoiceCustomer(BaseModel):
|
|
208
|
-
"""The customer associated with the invoice"""
|
|
209
|
-
model_config = ConfigDict(extra="allow", populate_by_name=True)
|
|
210
|
-
|
|
211
|
-
id: Union[str | None, Any] = Field(default=None, description="The customer ID")
|
|
212
|
-
"""The customer ID"""
|
|
213
|
-
external_customer_id: Union[str | None, Any] = Field(default=None, description="The external customer ID")
|
|
214
|
-
"""The external customer ID"""
|
|
215
|
-
|
|
216
216
|
class Invoice(BaseModel):
|
|
217
217
|
"""Invoice object"""
|
|
218
218
|
model_config = ConfigDict(extra="allow", populate_by_name=True)
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
Metadata-Version: 2.4
|
|
2
2
|
Name: airbyte-agent-orb
|
|
3
|
-
Version: 0.1.
|
|
3
|
+
Version: 0.1.7
|
|
4
4
|
Summary: Airbyte Orb 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/
|
|
@@ -147,7 +147,7 @@ See the official [Orb API reference](https://docs.withorb.com/api-reference).
|
|
|
147
147
|
|
|
148
148
|
## Version information
|
|
149
149
|
|
|
150
|
-
- **Package version:** 0.1.
|
|
150
|
+
- **Package version:** 0.1.7
|
|
151
151
|
- **Connector version:** 0.1.1
|
|
152
|
-
- **Generated with Connector SDK commit SHA:**
|
|
152
|
+
- **Generated with Connector SDK commit SHA:** 940246757c7476ed4edd7d16b873ebe54ea2b456
|
|
153
153
|
- **Changelog:** [View changelog](https://github.com/airbytehq/airbyte-agent-connectors/blob/main/connectors/orb/CHANGELOG.md)
|
|
@@ -1,7 +1,7 @@
|
|
|
1
|
-
airbyte_agent_orb/__init__.py,sha256=
|
|
1
|
+
airbyte_agent_orb/__init__.py,sha256=jHWqgCDpIiDZB1WLwcKo4aNXkp5ypSdV0WQWkCNJv_Y,3225
|
|
2
2
|
airbyte_agent_orb/connector.py,sha256=dCsSG2QFX4hdIyMJzNiP0sCj4xWoAKAKKrGxH83AVmc,41497
|
|
3
3
|
airbyte_agent_orb/connector_model.py,sha256=mhxwbi31hLA-oU2I8oQq7uMK4S0KV8XGjCELQRGth50,120056
|
|
4
|
-
airbyte_agent_orb/models.py,sha256=
|
|
4
|
+
airbyte_agent_orb/models.py,sha256=dFg7AeDxlgoGO8eU9ypRTLqULqSAHDA3cuaXKEKbsyk,22410
|
|
5
5
|
airbyte_agent_orb/types.py,sha256=O6Q2AOp8BYjYhpEFvOZXnzQDf-7GAKnpCorER2Puh2E,37077
|
|
6
6
|
airbyte_agent_orb/_vendored/__init__.py,sha256=ILl7AHXMui__swyrjxrh9yRa4dLiwBvV6axPWFWty80,38
|
|
7
7
|
airbyte_agent_orb/_vendored/connector_sdk/__init__.py,sha256=T5o7roU6NSpH-lCAGZ338sE5dlh4ZU6i6IkeG1zpems,1949
|
|
@@ -43,16 +43,16 @@ airbyte_agent_orb/_vendored/connector_sdk/performance/__init__.py,sha256=Sp5fSd1
|
|
|
43
43
|
airbyte_agent_orb/_vendored/connector_sdk/performance/instrumentation.py,sha256=_dXvNiqdndIBwDjeDKNViWzn_M5FkSUsMmJtFldrmsM,1504
|
|
44
44
|
airbyte_agent_orb/_vendored/connector_sdk/performance/metrics.py,sha256=FRff7dKt4iwt_A7pxV5n9kAGBR756PC7q8-weWygPSM,2817
|
|
45
45
|
airbyte_agent_orb/_vendored/connector_sdk/schema/__init__.py,sha256=Uymu-QuzGJuMxexBagIvUxpVAigIuIhz3KeBl_Vu4Ko,1638
|
|
46
|
-
airbyte_agent_orb/_vendored/connector_sdk/schema/base.py,sha256=
|
|
46
|
+
airbyte_agent_orb/_vendored/connector_sdk/schema/base.py,sha256=0R0aR4HjcJpu3vN0jZ4nPBEwVZ_7J9_XLQgWaQGT3Fs,8476
|
|
47
47
|
airbyte_agent_orb/_vendored/connector_sdk/schema/components.py,sha256=nJIPieavwX3o3ODvdtLHPk84d_V229xmg6LDfwEHjzc,8119
|
|
48
48
|
airbyte_agent_orb/_vendored/connector_sdk/schema/connector.py,sha256=mSZk1wr2YSdRj9tTRsPAuIlCzd_xZLw-Bzl1sMwE0rE,3731
|
|
49
|
-
airbyte_agent_orb/_vendored/connector_sdk/schema/extensions.py,sha256=
|
|
49
|
+
airbyte_agent_orb/_vendored/connector_sdk/schema/extensions.py,sha256=DqWCvyWKLnPxAbJirq6uITf-MBeJRVCcsEjySF4fuhQ,10558
|
|
50
50
|
airbyte_agent_orb/_vendored/connector_sdk/schema/operations.py,sha256=St-A75m6sZUZlsoM6WcoPaShYu_X1K19pdyPvJbabOE,6214
|
|
51
|
-
airbyte_agent_orb/_vendored/connector_sdk/schema/security.py,sha256=
|
|
51
|
+
airbyte_agent_orb/_vendored/connector_sdk/schema/security.py,sha256=5UeaN63InuInS_XHNZSm5yDXrMLm5lhz298l-3GSc_4,11435
|
|
52
52
|
airbyte_agent_orb/_vendored/connector_sdk/telemetry/__init__.py,sha256=RaLgkBU4dfxn1LC5Y0Q9rr2PJbrwjxvPgBLmq8_WafE,211
|
|
53
53
|
airbyte_agent_orb/_vendored/connector_sdk/telemetry/config.py,sha256=tLmQwAFD0kP1WyBGWBS3ysaudN9H3e-3EopKZi6cGKg,885
|
|
54
54
|
airbyte_agent_orb/_vendored/connector_sdk/telemetry/events.py,sha256=8Y1NbXiwISX-V_wRofY7PqcwEXD0dLMnntKkY6XFU2s,1328
|
|
55
55
|
airbyte_agent_orb/_vendored/connector_sdk/telemetry/tracker.py,sha256=SginFQbHqVUVYG82NnNzG34O-tAQ_wZYjGDcuo0q4Kk,5584
|
|
56
|
-
airbyte_agent_orb-0.1.
|
|
57
|
-
airbyte_agent_orb-0.1.
|
|
58
|
-
airbyte_agent_orb-0.1.
|
|
56
|
+
airbyte_agent_orb-0.1.7.dist-info/METADATA,sha256=I_Biiotj9EZ2BMBPaxWxlCDt8h57_CjyGn3d95gEt_0,5670
|
|
57
|
+
airbyte_agent_orb-0.1.7.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
|
|
58
|
+
airbyte_agent_orb-0.1.7.dist-info/RECORD,,
|
|
File without changes
|