airbyte-agent-zendesk-support 0.18.21__py3-none-any.whl → 0.18.32__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.
@@ -393,16 +393,24 @@ def convert_openapi_to_connector_model(spec: OpenAPIConnector) -> ConnectorModel
393
393
  for entity_name, endpoints_dict in entities_map.items():
394
394
  actions = list(endpoints_dict.keys())
395
395
 
396
- # Get schema from components if available
396
+ # Get schema and stream_name from components if available
397
397
  schema = None
398
+ entity_stream_name = None
398
399
  if spec.components:
399
400
  # Look for a schema matching the entity name
400
401
  for schema_name, schema_def in spec.components.schemas.items():
401
402
  if schema_def.x_airbyte_entity_name == entity_name or schema_name.lower() == entity_name.lower():
402
403
  schema = schema_def.model_dump(by_alias=True)
404
+ entity_stream_name = schema_def.x_airbyte_stream_name
403
405
  break
404
406
 
405
- entity = EntityDefinition(name=entity_name, actions=actions, endpoints=endpoints_dict, schema=schema)
407
+ entity = EntityDefinition(
408
+ name=entity_name,
409
+ stream_name=entity_stream_name,
410
+ actions=actions,
411
+ endpoints=endpoints_dict,
412
+ schema=schema,
413
+ )
406
414
  entities.append(entity)
407
415
 
408
416
  # Extract retry config from x-airbyte-retry-config extension
@@ -674,16 +674,16 @@ class LocalExecutor:
674
674
  return {key: value for key, value in params.items() if key in allowed_params}
675
675
 
676
676
  def _extract_body(self, allowed_fields: list[str], params: dict[str, Any]) -> dict[str, Any]:
677
- """Extract body fields from params.
677
+ """Extract body fields from params, filtering out None values.
678
678
 
679
679
  Args:
680
680
  allowed_fields: List of allowed body field names
681
681
  params: All parameters
682
682
 
683
683
  Returns:
684
- Dictionary of body fields
684
+ Dictionary of body fields with None values filtered out
685
685
  """
686
- return {key: value for key, value in params.items() if key in allowed_fields}
686
+ return {key: value for key, value in params.items() if key in allowed_fields and value is not None}
687
687
 
688
688
  def _serialize_deep_object_params(self, params: dict[str, Any], deep_object_param_names: list[str]) -> dict[str, Any]:
689
689
  """Serialize deepObject parameters to bracket notation format.
@@ -837,15 +837,65 @@ class LocalExecutor:
837
837
  Request body dict or None if no body needed
838
838
  """
839
839
  if endpoint.graphql_body:
840
- return self._build_graphql_body(endpoint.graphql_body, params)
840
+ # Extract defaults from query_params_schema for GraphQL variable interpolation
841
+ param_defaults = {name: schema.get("default") for name, schema in endpoint.query_params_schema.items() if "default" in schema}
842
+ return self._build_graphql_body(endpoint.graphql_body, params, param_defaults)
841
843
  elif endpoint.body_fields:
842
844
  return self._extract_body(endpoint.body_fields, params)
843
845
  return None
844
846
 
847
+ def _flatten_form_data(self, data: dict[str, Any], parent_key: str = "") -> dict[str, Any]:
848
+ """Flatten nested dict/list structures into bracket notation for form encoding.
849
+
850
+ Stripe and similar APIs require nested arrays/objects to be encoded using bracket
851
+ notation when using application/x-www-form-urlencoded content type.
852
+
853
+ Args:
854
+ data: Nested dict with arrays/objects to flatten
855
+ parent_key: Parent key for nested structures (used in recursion)
856
+
857
+ Returns:
858
+ Flattened dict with bracket notation keys
859
+
860
+ Examples:
861
+ >>> _flatten_form_data({"items": [{"price": "p1", "qty": 1}]})
862
+ {"items[0][price]": "p1", "items[0][qty]": 1}
863
+
864
+ >>> _flatten_form_data({"customer": "cus_123", "metadata": {"key": "value"}})
865
+ {"customer": "cus_123", "metadata[key]": "value"}
866
+ """
867
+ flattened = {}
868
+
869
+ for key, value in data.items():
870
+ new_key = f"{parent_key}[{key}]" if parent_key else key
871
+
872
+ if isinstance(value, dict):
873
+ # Recursively flatten nested dicts
874
+ flattened.update(self._flatten_form_data(value, new_key))
875
+ elif isinstance(value, list):
876
+ # Flatten arrays with indexed bracket notation
877
+ for i, item in enumerate(value):
878
+ indexed_key = f"{new_key}[{i}]"
879
+ if isinstance(item, dict):
880
+ # Nested dict in array - recurse
881
+ flattened.update(self._flatten_form_data(item, indexed_key))
882
+ elif isinstance(item, list):
883
+ # Nested list in array - recurse
884
+ flattened.update(self._flatten_form_data({str(i): item}, new_key))
885
+ else:
886
+ # Primitive value in array
887
+ flattened[indexed_key] = item
888
+ else:
889
+ # Primitive value - add directly
890
+ flattened[new_key] = value
891
+
892
+ return flattened
893
+
845
894
  def _determine_request_format(self, endpoint: EndpointDefinition, body: dict[str, Any] | None) -> dict[str, Any]:
846
895
  """Determine json/data parameters for HTTP request.
847
896
 
848
897
  GraphQL always uses JSON, regardless of content_type setting.
898
+ For form-encoded requests, nested structures are flattened into bracket notation.
849
899
 
850
900
  Args:
851
901
  endpoint: Endpoint definition
@@ -862,7 +912,9 @@ class LocalExecutor:
862
912
  if is_graphql or endpoint.content_type.value == "application/json":
863
913
  return {"json": body}
864
914
  elif endpoint.content_type.value == "application/x-www-form-urlencoded":
865
- return {"data": body}
915
+ # Flatten nested structures for form encoding
916
+ flattened_body = self._flatten_form_data(body)
917
+ return {"data": flattened_body}
866
918
 
867
919
  return {}
868
920
 
@@ -903,12 +955,18 @@ class LocalExecutor:
903
955
 
904
956
  return query
905
957
 
906
- def _build_graphql_body(self, graphql_config: dict[str, Any], params: dict[str, Any]) -> dict[str, Any]:
958
+ def _build_graphql_body(
959
+ self,
960
+ graphql_config: dict[str, Any],
961
+ params: dict[str, Any],
962
+ param_defaults: dict[str, Any] | None = None,
963
+ ) -> dict[str, Any]:
907
964
  """Build GraphQL request body with variable substitution and field selection.
908
965
 
909
966
  Args:
910
967
  graphql_config: GraphQL configuration from x-airbyte-body-type extension
911
968
  params: Parameters from execute() call
969
+ param_defaults: Default values for params from query_params_schema
912
970
 
913
971
  Returns:
914
972
  GraphQL request body: {"query": "...", "variables": {...}}
@@ -922,7 +980,7 @@ class LocalExecutor:
922
980
 
923
981
  # Substitute variables from params
924
982
  if "variables" in graphql_config and graphql_config["variables"]:
925
- body["variables"] = self._interpolate_variables(graphql_config["variables"], params)
983
+ body["variables"] = self._interpolate_variables(graphql_config["variables"], params, param_defaults)
926
984
 
927
985
  # Add operation name if specified
928
986
  if "operationName" in graphql_config:
@@ -981,7 +1039,12 @@ class LocalExecutor:
981
1039
  fields_str = " ".join(graphql_fields)
982
1040
  return query.replace("{{ fields }}", fields_str)
983
1041
 
984
- def _interpolate_variables(self, variables: dict[str, Any], params: dict[str, Any]) -> dict[str, Any]:
1042
+ def _interpolate_variables(
1043
+ self,
1044
+ variables: dict[str, Any],
1045
+ params: dict[str, Any],
1046
+ param_defaults: dict[str, Any] | None = None,
1047
+ ) -> dict[str, Any]:
985
1048
  """Recursively interpolate variables using params.
986
1049
 
987
1050
  Preserves types (doesn't stringify everything).
@@ -990,15 +1053,18 @@ class LocalExecutor:
990
1053
  - Direct replacement: "{{ owner }}" → params["owner"] (preserves type)
991
1054
  - Nested objects: {"input": {"name": "{{ name }}"}}
992
1055
  - Arrays: [{"id": "{{ id }}"}]
993
- - Unsubstituted placeholders: "{{ states }}" → None (for optional params)
1056
+ - Default values: "{{ per_page }}" → param_defaults["per_page"] if not in params
1057
+ - Unsubstituted placeholders: "{{ states }}" → None (for optional params without defaults)
994
1058
 
995
1059
  Args:
996
1060
  variables: Variables dict with template placeholders
997
1061
  params: Parameters to substitute
1062
+ param_defaults: Default values for params from query_params_schema
998
1063
 
999
1064
  Returns:
1000
1065
  Interpolated variables dict with types preserved
1001
1066
  """
1067
+ defaults = param_defaults or {}
1002
1068
 
1003
1069
  def interpolate_value(value: Any) -> Any:
1004
1070
  if isinstance(value, str):
@@ -1012,8 +1078,15 @@ class LocalExecutor:
1012
1078
  value = value.replace(placeholder, str(param_value))
1013
1079
 
1014
1080
  # Check if any unsubstituted placeholders remain
1015
- # If so, return None (treats as "not provided" for optional params)
1016
1081
  if re.search(r"\{\{\s*\w+\s*\}\}", value):
1082
+ # Extract placeholder name and check for default value
1083
+ match = re.search(r"\{\{\s*(\w+)\s*\}\}", value)
1084
+ if match:
1085
+ param_name = match.group(1)
1086
+ if param_name in defaults:
1087
+ # Use default value (preserves type)
1088
+ return defaults[param_name]
1089
+ # No default found - return None (for optional params)
1017
1090
  return None
1018
1091
 
1019
1092
  return value
@@ -1151,21 +1224,18 @@ class LocalExecutor:
1151
1224
  if action not in (Action.CREATE, Action.UPDATE):
1152
1225
  return
1153
1226
 
1154
- # Check if endpoint has body fields defined
1155
- if not endpoint.body_fields:
1227
+ # Get the request schema to find truly required fields
1228
+ request_schema = endpoint.request_schema
1229
+ if not request_schema:
1156
1230
  return
1157
1231
 
1158
- # For now, we treat all body_fields as potentially required for CREATE/UPDATE
1159
- # In a more advanced implementation, we could parse the request schema
1160
- # to identify truly required fields
1161
- missing_fields = []
1162
- for field in endpoint.body_fields:
1163
- if field not in params:
1164
- missing_fields.append(field)
1232
+ # Only validate fields explicitly marked as required in the schema
1233
+ required_fields = request_schema.get("required", [])
1234
+ missing_fields = [field for field in required_fields if field not in params]
1165
1235
 
1166
1236
  if missing_fields:
1167
1237
  raise MissingParameterError(
1168
- f"Missing required body fields for {entity}.{action.value}: {missing_fields}. Provided parameters: {list(params.keys())}"
1238
+ f"Missing required body fields for {entity}.{action.value}: {missing_fields}. " f"Provided parameters: {list(params.keys())}"
1169
1239
  )
1170
1240
 
1171
1241
  async def close(self):
@@ -159,6 +159,38 @@ Example:
159
159
  ```
160
160
  """
161
161
 
162
+ AIRBYTE_STREAM_NAME = "x-airbyte-stream-name"
163
+ """
164
+ Extension: x-airbyte-stream-name
165
+ Location: Schema object (in components.schemas)
166
+ Type: string
167
+ Required: No
168
+
169
+ Description:
170
+ Specifies the Airbyte stream name for cache lookup purposes. This maps the entity
171
+ to the corresponding Airbyte stream, enabling cache-based data retrieval. When
172
+ specified, the EntityDefinition.stream_name field will be populated with this value.
173
+
174
+ This extension is placed on Schema objects alongside x-airbyte-entity-name, following
175
+ the same pattern. The stream name is an entity-level property (not operation-level)
176
+ since an entity maps to exactly one Airbyte stream.
177
+
178
+ Example:
179
+ ```yaml
180
+ components:
181
+ schemas:
182
+ Customer:
183
+ type: object
184
+ x-airbyte-entity-name: customers
185
+ x-airbyte-stream-name: customers
186
+ properties:
187
+ id:
188
+ type: string
189
+ name:
190
+ type: string
191
+ ```
192
+ """
193
+
162
194
  AIRBYTE_TOKEN_PATH = "x-airbyte-token-path"
163
195
  """
164
196
  Extension: x-airbyte-token-path
@@ -548,6 +580,7 @@ def get_all_extension_names() -> list[str]:
548
580
  AIRBYTE_ENTITY,
549
581
  AIRBYTE_ACTION,
550
582
  AIRBYTE_ENTITY_NAME,
583
+ AIRBYTE_STREAM_NAME,
551
584
  AIRBYTE_TOKEN_PATH,
552
585
  AIRBYTE_BODY_TYPE,
553
586
  AIRBYTE_PATH_OVERRIDE,
@@ -594,6 +627,12 @@ EXTENSION_REGISTRY = {
594
627
  "required": False,
595
628
  "description": "Links schema to an entity/stream",
596
629
  },
630
+ AIRBYTE_STREAM_NAME: {
631
+ "location": "schema",
632
+ "type": "string",
633
+ "required": False,
634
+ "description": "Maps entity to Airbyte stream for cache lookup",
635
+ },
597
636
  AIRBYTE_TOKEN_PATH: {
598
637
  "location": "securityScheme",
599
638
  "type": "string",
@@ -0,0 +1,262 @@
1
+ """
2
+ Shared introspection utilities for connector metadata.
3
+
4
+ This module provides utilities for introspecting connector metadata,
5
+ generating descriptions, and formatting parameter signatures. These
6
+ functions are used by both the runtime decorators and the generated
7
+ connector code.
8
+
9
+ The module is designed to work with any object conforming to the
10
+ ConnectorModel and EndpointDefinition interfaces from connector_sdk.types.
11
+ """
12
+
13
+ from __future__ import annotations
14
+
15
+ from typing import Any, Protocol
16
+
17
+ # Constants
18
+ MAX_EXAMPLE_QUESTIONS = 5 # Maximum number of example questions to include in description
19
+
20
+
21
+ class EndpointProtocol(Protocol):
22
+ """Protocol defining the expected interface for endpoint parameters.
23
+
24
+ This allows functions to work with any endpoint-like object
25
+ that has these attributes, including EndpointDefinition and mock objects.
26
+ """
27
+
28
+ path_params: list[str]
29
+ path_params_schema: dict[str, dict[str, Any]]
30
+ query_params: list[str]
31
+ query_params_schema: dict[str, dict[str, Any]]
32
+ body_fields: list[str]
33
+ request_schema: dict[str, Any] | None
34
+
35
+
36
+ class EntityProtocol(Protocol):
37
+ """Protocol defining the expected interface for entity definitions."""
38
+
39
+ name: str
40
+ actions: list[Any]
41
+ endpoints: dict[Any, EndpointProtocol]
42
+
43
+
44
+ class ConnectorModelProtocol(Protocol):
45
+ """Protocol defining the expected interface for connector model parameters.
46
+
47
+ This allows functions to work with any connector-like object
48
+ that has these attributes, including ConnectorModel and mock objects.
49
+ """
50
+
51
+ @property
52
+ def entities(self) -> list[EntityProtocol]: ...
53
+
54
+ @property
55
+ def openapi_spec(self) -> Any: ...
56
+
57
+
58
+ def format_param_signature(endpoint: EndpointProtocol) -> str:
59
+ """Format parameter signature for an endpoint action.
60
+
61
+ Returns a string like: (id*) or (limit?, starting_after?, email?)
62
+ where * = required, ? = optional
63
+
64
+ Args:
65
+ endpoint: Object conforming to EndpointProtocol (e.g., EndpointDefinition)
66
+
67
+ Returns:
68
+ Formatted parameter signature string
69
+ """
70
+ params = []
71
+
72
+ # Defensive: safely access attributes with defaults for malformed endpoints
73
+ path_params = getattr(endpoint, "path_params", []) or []
74
+ query_params = getattr(endpoint, "query_params", []) or []
75
+ query_params_schema = getattr(endpoint, "query_params_schema", {}) or {}
76
+ body_fields = getattr(endpoint, "body_fields", []) or []
77
+ request_schema = getattr(endpoint, "request_schema", None)
78
+
79
+ # Path params (always required)
80
+ for name in path_params:
81
+ params.append(f"{name}*")
82
+
83
+ # Query params
84
+ for name in query_params:
85
+ schema = query_params_schema.get(name, {})
86
+ required = schema.get("required", False)
87
+ params.append(f"{name}{'*' if required else '?'}")
88
+
89
+ # Body fields
90
+ if request_schema:
91
+ required_fields = set(request_schema.get("required", []))
92
+ for name in body_fields:
93
+ params.append(f"{name}{'*' if name in required_fields else '?'}")
94
+
95
+ return f"({', '.join(params)})" if params else "()"
96
+
97
+
98
+ def describe_entities(model: ConnectorModelProtocol) -> list[dict[str, Any]]:
99
+ """Generate entity descriptions from ConnectorModel.
100
+
101
+ Returns a list of entity descriptions with detailed parameter information
102
+ for each action. This is used by generated connectors' describe() method.
103
+
104
+ Args:
105
+ model: Object conforming to ConnectorModelProtocol (e.g., ConnectorModel)
106
+
107
+ Returns:
108
+ List of entity description dicts with keys:
109
+ - entity_name: Name of the entity (e.g., "contacts", "deals")
110
+ - description: Entity description from the first endpoint
111
+ - available_actions: List of actions (e.g., ["list", "get", "create"])
112
+ - parameters: Dict mapping action -> list of parameter dicts
113
+ """
114
+ entities = []
115
+ for entity_def in model.entities:
116
+ description = ""
117
+ parameters: dict[str, list[dict[str, Any]]] = {}
118
+
119
+ endpoints = getattr(entity_def, "endpoints", {}) or {}
120
+ if endpoints:
121
+ for action, endpoint in endpoints.items():
122
+ # Get description from first endpoint that has one
123
+ if not description:
124
+ endpoint_desc = getattr(endpoint, "description", None)
125
+ if endpoint_desc:
126
+ description = endpoint_desc
127
+
128
+ action_params: list[dict[str, Any]] = []
129
+
130
+ # Defensive: safely access endpoint attributes
131
+ path_params = getattr(endpoint, "path_params", []) or []
132
+ path_params_schema = getattr(endpoint, "path_params_schema", {}) or {}
133
+ query_params = getattr(endpoint, "query_params", []) or []
134
+ query_params_schema = getattr(endpoint, "query_params_schema", {}) or {}
135
+ body_fields = getattr(endpoint, "body_fields", []) or []
136
+ request_schema = getattr(endpoint, "request_schema", None)
137
+
138
+ # Path params (always required)
139
+ for param_name in path_params:
140
+ schema = path_params_schema.get(param_name, {})
141
+ action_params.append(
142
+ {
143
+ "name": param_name,
144
+ "in": "path",
145
+ "required": True,
146
+ "type": schema.get("type", "string"),
147
+ "description": schema.get("description", ""),
148
+ }
149
+ )
150
+
151
+ # Query params
152
+ for param_name in query_params:
153
+ schema = query_params_schema.get(param_name, {})
154
+ action_params.append(
155
+ {
156
+ "name": param_name,
157
+ "in": "query",
158
+ "required": schema.get("required", False),
159
+ "type": schema.get("type", "string"),
160
+ "description": schema.get("description", ""),
161
+ }
162
+ )
163
+
164
+ # Body fields
165
+ if request_schema:
166
+ required_fields = request_schema.get("required", [])
167
+ properties = request_schema.get("properties", {})
168
+ for param_name in body_fields:
169
+ prop = properties.get(param_name, {})
170
+ action_params.append(
171
+ {
172
+ "name": param_name,
173
+ "in": "body",
174
+ "required": param_name in required_fields,
175
+ "type": prop.get("type", "string"),
176
+ "description": prop.get("description", ""),
177
+ }
178
+ )
179
+
180
+ if action_params:
181
+ # Action is an enum, use .value to get string
182
+ action_key = action.value if hasattr(action, "value") else str(action)
183
+ parameters[action_key] = action_params
184
+
185
+ actions = getattr(entity_def, "actions", []) or []
186
+ entities.append(
187
+ {
188
+ "entity_name": entity_def.name,
189
+ "description": description,
190
+ "available_actions": [a.value if hasattr(a, "value") else str(a) for a in actions],
191
+ "parameters": parameters,
192
+ }
193
+ )
194
+
195
+ return entities
196
+
197
+
198
+ def generate_tool_description(model: ConnectorModelProtocol) -> str:
199
+ """Generate AI tool description from connector metadata.
200
+
201
+ Produces a detailed description that includes:
202
+ - Per-entity/action parameter signatures with required (*) and optional (?) markers
203
+ - Response structure documentation with pagination hints
204
+ - Example questions if available in the OpenAPI spec
205
+
206
+ This is used by the Connector.describe class method decorator to populate
207
+ function docstrings for AI framework integration.
208
+
209
+ Args:
210
+ model: Object conforming to ConnectorModelProtocol (e.g., ConnectorModel)
211
+
212
+ Returns:
213
+ Formatted description string suitable for AI tool documentation
214
+ """
215
+ lines = []
216
+
217
+ # Entity/action parameter details (including pagination params like limit, starting_after)
218
+ lines.append("ENTITIES AND PARAMETERS:")
219
+ for entity in model.entities:
220
+ lines.append(f" {entity.name}:")
221
+ actions = getattr(entity, "actions", []) or []
222
+ endpoints = getattr(entity, "endpoints", {}) or {}
223
+ for action in actions:
224
+ action_str = action.value if hasattr(action, "value") else str(action)
225
+ endpoint = endpoints.get(action)
226
+ if endpoint:
227
+ param_sig = format_param_signature(endpoint)
228
+ lines.append(f" - {action_str}{param_sig}")
229
+ else:
230
+ lines.append(f" - {action_str}()")
231
+
232
+ # Response structure (brief, includes pagination hint)
233
+ lines.append("")
234
+ lines.append("RESPONSE STRUCTURE:")
235
+ lines.append(" - list/search: {data: [...], meta: {has_more: bool}}")
236
+ lines.append(" - get: Returns entity directly (no envelope)")
237
+ lines.append(" To paginate: pass starting_after=<last_id> while has_more is true")
238
+
239
+ # Add example questions if available in openapi_spec
240
+ openapi_spec = getattr(model, "openapi_spec", None)
241
+ if openapi_spec:
242
+ info = getattr(openapi_spec, "info", None)
243
+ if info:
244
+ example_questions = getattr(info, "x_airbyte_example_questions", None)
245
+ if example_questions:
246
+ supported = getattr(example_questions, "supported", None)
247
+ if supported:
248
+ lines.append("")
249
+ lines.append("EXAMPLE QUESTIONS:")
250
+ for q in supported[:MAX_EXAMPLE_QUESTIONS]:
251
+ lines.append(f" - {q}")
252
+
253
+ # Generic parameter description for function signature
254
+ lines.append("")
255
+ lines.append("FUNCTION PARAMETERS:")
256
+ lines.append(" - entity: Entity name (string)")
257
+ lines.append(" - action: Operation to perform (string)")
258
+ lines.append(" - params: Operation parameters (dict) - see entity details above")
259
+ lines.append("")
260
+ lines.append("Parameter markers: * = required, ? = optional")
261
+
262
+ return "\n".join(lines)
@@ -65,8 +65,9 @@ class Schema(BaseModel):
65
65
  write_only: Optional[bool] = Field(None, alias="writeOnly")
66
66
  deprecated: Optional[bool] = None
67
67
 
68
- # Airbyte extension
68
+ # Airbyte extensions
69
69
  x_airbyte_entity_name: Optional[str] = Field(None, alias="x-airbyte-entity-name")
70
+ x_airbyte_stream_name: Optional[str] = Field(None, alias="x-airbyte-stream-name")
70
71
 
71
72
 
72
73
  class Parameter(BaseModel):
@@ -77,6 +77,10 @@ class AuthConfigOption(BaseModel):
77
77
  default_factory=dict,
78
78
  description="Mapping from auth parameters (e.g., 'username', 'password', 'token') to template strings using ${field} syntax",
79
79
  )
80
+ replication_auth_key_mapping: Optional[Dict[str, str]] = Field(
81
+ None,
82
+ description="Mapping from source config paths (e.g., 'credentials.api_key') to auth config keys for direct connectors",
83
+ )
80
84
 
81
85
 
82
86
  class AirbyteAuthConfig(BaseModel):
@@ -99,6 +103,12 @@ class AirbyteAuthConfig(BaseModel):
99
103
  properties: Optional[Dict[str, AuthConfigFieldSpec]] = None
100
104
  auth_mapping: Optional[Dict[str, str]] = None
101
105
 
106
+ # Replication connector auth mapping
107
+ replication_auth_key_mapping: Optional[Dict[str, str]] = Field(
108
+ None,
109
+ description="Mapping from source config paths (e.g., 'credentials.api_key') to auth config keys for direct connectors",
110
+ )
111
+
102
112
  # Multiple options (oneOf)
103
113
  one_of: Optional[List[AuthConfigOption]] = Field(None, alias="oneOf")
104
114
 
@@ -221,6 +221,10 @@ class EntityDefinition(BaseModel):
221
221
  model_config = {"populate_by_name": True}
222
222
 
223
223
  name: str
224
+ stream_name: str | None = Field(
225
+ default=None,
226
+ description="Airbyte stream name for cache lookup (from x-airbyte-stream-name schema extension)",
227
+ )
224
228
  actions: list[Action]
225
229
  endpoints: dict[Action, EndpointDefinition]
226
230
  entity_schema: dict[str, Any] | None = Field(default=None, alias="schema")
@@ -4,14 +4,15 @@ zendesk-support connector.
4
4
 
5
5
  from __future__ import annotations
6
6
 
7
- from typing import TYPE_CHECKING, Any, AsyncIterator, overload
7
+ import logging
8
+ from typing import TYPE_CHECKING, Any, Callable, TypeVar, AsyncIterator, overload
8
9
  try:
9
10
  from typing import Literal
10
11
  except ImportError:
11
12
  from typing_extensions import Literal
12
13
 
13
14
  from .connector_model import ZendeskSupportConnectorModel
14
-
15
+ from ._vendored.connector_sdk.introspection import describe_entities, generate_tool_description
15
16
  from .types import (
16
17
  ArticleAttachmentsDownloadParams,
17
18
  ArticleAttachmentsGetParams,
@@ -53,7 +54,6 @@ from .types import (
53
54
  ViewsGetParams,
54
55
  ViewsListParams,
55
56
  )
56
-
57
57
  if TYPE_CHECKING:
58
58
  from .models import ZendeskSupportAuthConfig
59
59
  # Import specific auth config classes for multi-auth isinstance checks
@@ -102,6 +102,9 @@ from .models import (
102
102
  ArticleAttachmentsGetResult,
103
103
  )
104
104
 
105
+ # TypeVar for decorator type preservation
106
+ _F = TypeVar("_F", bound=Callable[..., Any])
107
+
105
108
 
106
109
  class ZendeskSupportConnector:
107
110
  """
@@ -111,7 +114,7 @@ class ZendeskSupportConnector:
111
114
  """
112
115
 
113
116
  connector_name = "zendesk-support"
114
- connector_version = "0.1.3"
117
+ connector_version = "0.1.4"
115
118
  vendored_sdk_version = "0.1.0" # Version of vendored connector-sdk
116
119
 
117
120
  # Map of (entity, action) -> has_extractors for envelope wrapping decision
@@ -712,6 +715,88 @@ class ZendeskSupportConnector:
712
715
  # No extractors - return raw response data
713
716
  return result.data
714
717
 
718
+ # ===== INTROSPECTION METHODS =====
719
+
720
+ @classmethod
721
+ def describe(cls, func: _F) -> _F:
722
+ """
723
+ Decorator that populates a function's docstring with connector capabilities.
724
+
725
+ This class method can be used as a decorator to automatically generate
726
+ comprehensive documentation for AI tool functions.
727
+
728
+ Usage:
729
+ @mcp.tool()
730
+ @ZendeskSupportConnector.describe
731
+ async def execute(entity: str, action: str, params: dict):
732
+ '''Execute operations.'''
733
+ ...
734
+
735
+ The decorated function's __doc__ will be updated with:
736
+ - Available entities and their actions
737
+ - Parameter signatures with required (*) and optional (?) markers
738
+ - Response structure documentation
739
+ - Example questions (if available in OpenAPI spec)
740
+
741
+ Args:
742
+ func: The function to decorate
743
+
744
+ Returns:
745
+ The same function with updated __doc__
746
+ """
747
+ description = generate_tool_description(ZendeskSupportConnectorModel)
748
+
749
+ original_doc = func.__doc__ or ""
750
+ if original_doc.strip():
751
+ func.__doc__ = f"{original_doc.strip()}\n\n{description}"
752
+ else:
753
+ func.__doc__ = description
754
+
755
+ return func
756
+
757
+ def list_entities(self) -> list[dict[str, Any]]:
758
+ """
759
+ Get structured data about available entities, actions, and parameters.
760
+
761
+ Returns a list of entity descriptions with:
762
+ - entity_name: Name of the entity (e.g., "contacts", "deals")
763
+ - description: Entity description from the first endpoint
764
+ - available_actions: List of actions (e.g., ["list", "get", "create"])
765
+ - parameters: Dict mapping action -> list of parameter dicts
766
+
767
+ Example:
768
+ entities = connector.list_entities()
769
+ for entity in entities:
770
+ print(f"{entity['entity_name']}: {entity['available_actions']}")
771
+ """
772
+ return describe_entities(ZendeskSupportConnectorModel)
773
+
774
+ def entity_schema(self, entity: str) -> dict[str, Any] | None:
775
+ """
776
+ Get the JSON schema for an entity.
777
+
778
+ Args:
779
+ entity: Entity name (e.g., "contacts", "companies")
780
+
781
+ Returns:
782
+ JSON schema dict describing the entity structure, or None if not found.
783
+
784
+ Example:
785
+ schema = connector.entity_schema("contacts")
786
+ if schema:
787
+ print(f"Contact properties: {list(schema.get('properties', {}).keys())}")
788
+ """
789
+ entity_def = next(
790
+ (e for e in ZendeskSupportConnectorModel.entities if e.name == entity),
791
+ None
792
+ )
793
+ if entity_def is None:
794
+ logging.getLogger(__name__).warning(
795
+ f"Entity '{entity}' not found. Available entities: "
796
+ f"{[e.name for e in ZendeskSupportConnectorModel.entities]}"
797
+ )
798
+ return entity_def.entity_schema if entity_def else None
799
+
715
800
 
716
801
 
717
802
  class TicketsQuery:
@@ -30,7 +30,7 @@ from uuid import (
30
30
  ZendeskSupportConnectorModel: ConnectorModel = ConnectorModel(
31
31
  id=UUID('79c1aa37-dae3-42ae-b333-d1c105477715'),
32
32
  name='zendesk-support',
33
- version='0.1.3',
33
+ version='0.1.4',
34
34
  base_url='https://{subdomain}.zendesk.com/api/v2',
35
35
  auth=AuthConfig(
36
36
  options=[
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: airbyte-agent-zendesk-support
3
- Version: 0.18.21
3
+ Version: 0.18.32
4
4
  Summary: Airbyte Zendesk-Support Connector for AI platforms
5
5
  Project-URL: Homepage, https://github.com/airbytehq/airbyte-embedded
6
6
  Project-URL: Documentation, https://github.com/airbytehq/airbyte-embedded/tree/main/integrations
@@ -43,22 +43,26 @@ for customer support analytics and service performance insights.
43
43
 
44
44
  ## Example questions
45
45
 
46
+ The Zendesk-Support connector is optimized to handle prompts like these.
47
+
46
48
  - Show me the tickets assigned to me last week
47
49
  - What are the top 5 support issues our organization has faced this month?
48
- - List all unresolved tickets for [customerX]
50
+ - List all unresolved tickets for \{customer\}
49
51
  - Analyze the satisfaction ratings for our support team in the last 30 days
50
52
  - Compare ticket resolution times across different support groups
51
- - Show me the details of recent tickets tagged with [specific tag]
53
+ - Show me the details of recent tickets tagged with \{tag\}
52
54
  - Identify the most common ticket fields used in our support workflow
53
55
  - Summarize the performance of our SLA policies this quarter
54
56
 
55
57
  ## Unsupported questions
56
58
 
57
- - Create a new support ticket for [customerX]
59
+ The Zendesk-Support connector isn't currently able to handle prompts like these.
60
+
61
+ - Create a new support ticket for \{customer\}
58
62
  - Update the priority of this ticket
59
- - Assign this ticket to [teamMember]
63
+ - Assign this ticket to \{team_member\}
60
64
  - Delete these old support tickets
61
- - Send an automatic response to [customerX]
65
+ - Send an automatic response to \{customer\}
62
66
 
63
67
  ## Installation
64
68
 
@@ -68,11 +72,16 @@ uv pip install airbyte-agent-zendesk-support
68
72
 
69
73
  ## Usage
70
74
 
75
+ This connector supports multiple authentication methods:
76
+
77
+ ### OAuth 2.0
78
+
71
79
  ```python
72
- from airbyte_agent_zendesk_support import ZendeskSupportConnector, ZendeskSupportAuthConfig
80
+ from airbyte_agent_zendesk_support import ZendeskSupportConnector
81
+ from airbyte_agent_zendesk_support.models import ZendeskSupportOauth20AuthConfig
73
82
 
74
83
  connector = ZendeskSupportConnector(
75
- auth_config=ZendeskSupportAuthConfig(
84
+ auth_config=ZendeskSupportOauth20AuthConfig(
76
85
  access_token="...",
77
86
  refresh_token="..."
78
87
  )
@@ -80,6 +89,22 @@ connector = ZendeskSupportConnector(
80
89
  result = await connector.tickets.list()
81
90
  ```
82
91
 
92
+ ### API Token
93
+
94
+ ```python
95
+ from airbyte_agent_zendesk_support import ZendeskSupportConnector
96
+ from airbyte_agent_zendesk_support.models import ZendeskSupportApiTokenAuthConfig
97
+
98
+ connector = ZendeskSupportConnector(
99
+ auth_config=ZendeskSupportApiTokenAuthConfig(
100
+ email="...",
101
+ api_token="..."
102
+ )
103
+ )
104
+ result = await connector.tickets.list()
105
+ ```
106
+
107
+
83
108
  ## Full documentation
84
109
 
85
110
  This connector supports the following entities and actions.
@@ -116,6 +141,6 @@ For the service's official API docs, see the [Zendesk-Support API reference](htt
116
141
 
117
142
  ## Version information
118
143
 
119
- - **Package version:** 0.18.21
120
- - **Connector version:** 0.1.3
121
- - **Generated with Connector SDK commit SHA:** f7c55d3e3cdc7568cab2da9d736285eec58f044b
144
+ - **Package version:** 0.18.32
145
+ - **Connector version:** 0.1.4
146
+ - **Generated with Connector SDK commit SHA:** 3c7bfdfd1100dd20420a61cec56549b65820ea0f
@@ -1,26 +1,27 @@
1
1
  airbyte_agent_zendesk_support/__init__.py,sha256=MPz4HU055DRA3-1qgbGXh2E0YHmhcexQCFl-Tz21gm4,6227
2
- airbyte_agent_zendesk_support/connector.py,sha256=7AYDMDbkygncAVYtxo0q880C6SrVZQhaCvK8p0bKiOQ,63878
3
- airbyte_agent_zendesk_support/connector_model.py,sha256=q-rn6aRvFt4H-neQ1HQRX7oq2MDEJY7MMuSaEAhGYCk,241131
2
+ airbyte_agent_zendesk_support/connector.py,sha256=VYOTY7bEtXsvjyZSggoeS4lbSbGUsnVmtq3B2awkW5o,67058
3
+ airbyte_agent_zendesk_support/connector_model.py,sha256=SAEWsLhW517Gc5eajkkYV1fQEGpsLlMALs2pTJ9BcPk,241131
4
4
  airbyte_agent_zendesk_support/models.py,sha256=31bsOmf4nBdf8EXN3JpYzXW8mx6gv1xaZjeuEBgSzws,36399
5
5
  airbyte_agent_zendesk_support/types.py,sha256=3CxJ8HosRMyzNEbVmRbybNCTVj9Ycxr7io25TP3YcCQ,6337
6
6
  airbyte_agent_zendesk_support/_vendored/__init__.py,sha256=ILl7AHXMui__swyrjxrh9yRa4dLiwBvV6axPWFWty80,38
7
7
  airbyte_agent_zendesk_support/_vendored/connector_sdk/__init__.py,sha256=T5o7roU6NSpH-lCAGZ338sE5dlh4ZU6i6IkeG1zpems,1949
8
8
  airbyte_agent_zendesk_support/_vendored/connector_sdk/auth_strategies.py,sha256=0BfIISVzuvZTAYZjQFOOhKTpw0QuKDlLQBQ1PQo-V2M,39967
9
9
  airbyte_agent_zendesk_support/_vendored/connector_sdk/auth_template.py,sha256=vKnyA21Jp33EuDjkIUAf1PGicwk4t9kZAPJuAgAZKzU,4458
10
- airbyte_agent_zendesk_support/_vendored/connector_sdk/connector_model_loader.py,sha256=BeX4QykMUQZk5qY--WuwxXClI7FBDxxbGguqcztAd8A,34663
10
+ airbyte_agent_zendesk_support/_vendored/connector_sdk/connector_model_loader.py,sha256=nKrXfe-FAyvNMkW7AqGzxrp5wXdaHiqC0yIFJoIVwlY,34890
11
11
  airbyte_agent_zendesk_support/_vendored/connector_sdk/constants.py,sha256=uH4rjBX6WsBP8M0jt7AUJI9w5Adn4wvJwib7Gdfkr1M,2736
12
12
  airbyte_agent_zendesk_support/_vendored/connector_sdk/exceptions.py,sha256=ss5MGv9eVPmsbLcLWetuu3sDmvturwfo6Pw3M37Oq5k,481
13
- airbyte_agent_zendesk_support/_vendored/connector_sdk/extensions.py,sha256=iWA2i0kiiGZY84H8P25A6QmfbuZwu7euMcj4-Vx2DOQ,20185
13
+ airbyte_agent_zendesk_support/_vendored/connector_sdk/extensions.py,sha256=fWy9uwGUCjPO1KDYuGZo9nkrNU35P-dLcqi4K6UF4uA,21371
14
14
  airbyte_agent_zendesk_support/_vendored/connector_sdk/http_client.py,sha256=NdccrrBHI5rW56XnXcP54arCwywIVKnMeSQPas6KlOM,27466
15
+ airbyte_agent_zendesk_support/_vendored/connector_sdk/introspection.py,sha256=6v3YNdca8qe8qIz3m97GZ_ll_Ih3oUKMrqrdipPcpRk,10331
15
16
  airbyte_agent_zendesk_support/_vendored/connector_sdk/secrets.py,sha256=UWcO9fP-vZwcfkAuvlZahlOCTOwdNN860BIwe8X4jxw,6868
16
- airbyte_agent_zendesk_support/_vendored/connector_sdk/types.py,sha256=sS9olOyT-kVemHmcFll2ePFRhTdGMbWcz7bSgV-MuSw,8114
17
+ airbyte_agent_zendesk_support/_vendored/connector_sdk/types.py,sha256=TI-O7EyWAoppGc9G7kXHwceYWekt_sxXmXKxD1xC_7U,8285
17
18
  airbyte_agent_zendesk_support/_vendored/connector_sdk/utils.py,sha256=G4LUXOC2HzPoND2v4tQW68R9uuPX9NQyCjaGxb7Kpl0,1958
18
19
  airbyte_agent_zendesk_support/_vendored/connector_sdk/validation.py,sha256=CDjCux1eg35a0Y4BegSivzIwZjiTqOxYWotWNLqTSVU,31792
19
20
  airbyte_agent_zendesk_support/_vendored/connector_sdk/cloud_utils/__init__.py,sha256=4799Hv9f2zxDVj1aLyQ8JpTEuFTp_oOZMRz-NZCdBJg,134
20
21
  airbyte_agent_zendesk_support/_vendored/connector_sdk/cloud_utils/client.py,sha256=HoDgZuEgGHj78P-BGwUf6HGPVWynbdKjGOmjb-JDk58,7188
21
22
  airbyte_agent_zendesk_support/_vendored/connector_sdk/executor/__init__.py,sha256=EmG9YQNAjSuYCVB4D5VoLm4qpD1KfeiiOf7bpALj8p8,702
22
23
  airbyte_agent_zendesk_support/_vendored/connector_sdk/executor/hosted_executor.py,sha256=YQ-qfT7PZh9izNFHHe7SAcETiZOKrWjTU-okVb0_VL8,7079
23
- airbyte_agent_zendesk_support/_vendored/connector_sdk/executor/local_executor.py,sha256=hHlBTtvykrUcfypzyW0e61fU4e3vlxc90mypCFzgSl0,61879
24
+ airbyte_agent_zendesk_support/_vendored/connector_sdk/executor/local_executor.py,sha256=IPe4q3FsN22zqCQfT836uQwhnnl3Y9g6o_ZBrZEjTuc,65271
24
25
  airbyte_agent_zendesk_support/_vendored/connector_sdk/executor/models.py,sha256=lYVT_bNcw-PoIks4WHNyl2VY-lJVf2FntzINSOBIheE,5845
25
26
  airbyte_agent_zendesk_support/_vendored/connector_sdk/http/__init__.py,sha256=y8fbzZn-3yV9OxtYz8Dy6FFGI5v6TOqADd1G3xHH3Hw,911
26
27
  airbyte_agent_zendesk_support/_vendored/connector_sdk/http/config.py,sha256=6J7YIIwHC6sRu9i-yKa5XvArwK2KU60rlnmxzDZq3lw,3283
@@ -41,15 +42,15 @@ airbyte_agent_zendesk_support/_vendored/connector_sdk/performance/instrumentatio
41
42
  airbyte_agent_zendesk_support/_vendored/connector_sdk/performance/metrics.py,sha256=3-wPwlJyfVLUIG3y7ESxk0avhkILk3z8K7zSrnlZf5U,2833
42
43
  airbyte_agent_zendesk_support/_vendored/connector_sdk/schema/__init__.py,sha256=Uymu-QuzGJuMxexBagIvUxpVAigIuIhz3KeBl_Vu4Ko,1638
43
44
  airbyte_agent_zendesk_support/_vendored/connector_sdk/schema/base.py,sha256=eN6YfHwsYNz_0CE-S715Me8x3gQWTtaRk1vhFjEZF8I,4799
44
- airbyte_agent_zendesk_support/_vendored/connector_sdk/schema/components.py,sha256=2ivMNhUAcovw88qvDkW221ILf1L63eXteztTDZL46us,7989
45
+ airbyte_agent_zendesk_support/_vendored/connector_sdk/schema/components.py,sha256=4clYbIJNqfmFj7Te31ZqC3Rr5E2v33BeXyQx1wp3oj0,8076
45
46
  airbyte_agent_zendesk_support/_vendored/connector_sdk/schema/connector.py,sha256=VFBOzIkLgYBR1XUTmyrPGqBkX8PP-zsG8avQcdJUqXs,3864
46
47
  airbyte_agent_zendesk_support/_vendored/connector_sdk/schema/extensions.py,sha256=LdoCuMLNdCj68O47qAL4v8xmqGz5tJgRiNZL5JnL9uw,3311
47
48
  airbyte_agent_zendesk_support/_vendored/connector_sdk/schema/operations.py,sha256=zInMjx9iOEVZo-CCWw06Uk2SFg7HtUAXXpO5kFGUwNk,5825
48
- airbyte_agent_zendesk_support/_vendored/connector_sdk/schema/security.py,sha256=Jtrhb58d1zssN5yR7LhLJi-QK0PsZKT7uIvPqanF0co,8114
49
+ airbyte_agent_zendesk_support/_vendored/connector_sdk/schema/security.py,sha256=Xo6Z0-NCryMY4EDZgvg8ABSPglskpXANvm_iI34meV8,8588
49
50
  airbyte_agent_zendesk_support/_vendored/connector_sdk/telemetry/__init__.py,sha256=RaLgkBU4dfxn1LC5Y0Q9rr2PJbrwjxvPgBLmq8_WafE,211
50
51
  airbyte_agent_zendesk_support/_vendored/connector_sdk/telemetry/config.py,sha256=tLmQwAFD0kP1WyBGWBS3ysaudN9H3e-3EopKZi6cGKg,885
51
52
  airbyte_agent_zendesk_support/_vendored/connector_sdk/telemetry/events.py,sha256=NvqjlUbkm6cbGh4ffKxYxtjdwwgzfPF4MKJ2GfgWeFg,1285
52
53
  airbyte_agent_zendesk_support/_vendored/connector_sdk/telemetry/tracker.py,sha256=KacNdbHatvPPhnNrycp5YUuD5xpkp56AFcHd-zguBgk,5247
53
- airbyte_agent_zendesk_support-0.18.21.dist-info/METADATA,sha256=h_eKn-nbGYB3MCSchx0bZNgcJcpCCSKAImsE8EQwnSA,5629
54
- airbyte_agent_zendesk_support-0.18.21.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
55
- airbyte_agent_zendesk_support-0.18.21.dist-info/RECORD,,
54
+ airbyte_agent_zendesk_support-0.18.32.dist-info/METADATA,sha256=cfq9fi6SsUk98-QqBlnLPfX6Bvtw9wcRpG5AteNq2Nw,6267
55
+ airbyte_agent_zendesk_support-0.18.32.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
56
+ airbyte_agent_zendesk_support-0.18.32.dist-info/RECORD,,