airbyte-agent-zendesk-support 0.18.36__py3-none-any.whl → 0.18.42__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.
Files changed (30) hide show
  1. airbyte_agent_zendesk_support/__init__.py +111 -18
  2. airbyte_agent_zendesk_support/_vendored/connector_sdk/auth_strategies.py +2 -5
  3. airbyte_agent_zendesk_support/_vendored/connector_sdk/auth_template.py +1 -1
  4. airbyte_agent_zendesk_support/_vendored/connector_sdk/cloud_utils/client.py +26 -26
  5. airbyte_agent_zendesk_support/_vendored/connector_sdk/connector_model_loader.py +1 -2
  6. airbyte_agent_zendesk_support/_vendored/connector_sdk/executor/hosted_executor.py +10 -11
  7. airbyte_agent_zendesk_support/_vendored/connector_sdk/executor/local_executor.py +72 -13
  8. airbyte_agent_zendesk_support/_vendored/connector_sdk/extensions.py +1 -2
  9. airbyte_agent_zendesk_support/_vendored/connector_sdk/http/response.py +2 -0
  10. airbyte_agent_zendesk_support/_vendored/connector_sdk/logging/logger.py +9 -9
  11. airbyte_agent_zendesk_support/_vendored/connector_sdk/logging/types.py +10 -10
  12. airbyte_agent_zendesk_support/_vendored/connector_sdk/observability/config.py +2 -2
  13. airbyte_agent_zendesk_support/_vendored/connector_sdk/observability/models.py +6 -6
  14. airbyte_agent_zendesk_support/_vendored/connector_sdk/observability/session.py +7 -5
  15. airbyte_agent_zendesk_support/_vendored/connector_sdk/performance/metrics.py +3 -3
  16. airbyte_agent_zendesk_support/_vendored/connector_sdk/schema/base.py +17 -17
  17. airbyte_agent_zendesk_support/_vendored/connector_sdk/schema/components.py +58 -58
  18. airbyte_agent_zendesk_support/_vendored/connector_sdk/schema/connector.py +22 -33
  19. airbyte_agent_zendesk_support/_vendored/connector_sdk/schema/extensions.py +9 -9
  20. airbyte_agent_zendesk_support/_vendored/connector_sdk/schema/operations.py +31 -31
  21. airbyte_agent_zendesk_support/_vendored/connector_sdk/schema/security.py +36 -36
  22. airbyte_agent_zendesk_support/_vendored/connector_sdk/secrets.py +2 -2
  23. airbyte_agent_zendesk_support/_vendored/connector_sdk/telemetry/events.py +7 -7
  24. airbyte_agent_zendesk_support/_vendored/connector_sdk/telemetry/tracker.py +6 -5
  25. airbyte_agent_zendesk_support/_vendored/connector_sdk/types.py +2 -2
  26. airbyte_agent_zendesk_support/connector.py +153 -157
  27. airbyte_agent_zendesk_support/models.py +21 -69
  28. {airbyte_agent_zendesk_support-0.18.36.dist-info → airbyte_agent_zendesk_support-0.18.42.dist-info}/METADATA +11 -8
  29. {airbyte_agent_zendesk_support-0.18.36.dist-info → airbyte_agent_zendesk_support-0.18.42.dist-info}/RECORD +30 -30
  30. {airbyte_agent_zendesk_support-0.18.36.dist-info → airbyte_agent_zendesk_support-0.18.42.dist-info}/WHEEL +0 -0
@@ -2,7 +2,7 @@
2
2
 
3
3
  import base64
4
4
  from datetime import UTC, datetime
5
- from typing import Any, Dict, List, Optional
5
+ from typing import Any, Dict, List
6
6
 
7
7
  from pydantic import BaseModel, ConfigDict, Field, field_serializer, field_validator
8
8
 
@@ -27,12 +27,12 @@ class RequestLog(BaseModel):
27
27
  url: str
28
28
  path: str
29
29
  headers: Dict[str, str] = Field(default_factory=dict)
30
- params: Optional[Dict[str, Any]] = None
31
- body: Optional[Any] = None
32
- response_status: Optional[int] = None
33
- response_body: Optional[Any] = None
34
- timing_ms: Optional[float] = None
35
- error: Optional[str] = None
30
+ params: Dict[str, Any] | None = None
31
+ body: Any | None = None
32
+ response_status: int | None = None
33
+ response_body: Any | None = None
34
+ timing_ms: float | None = None
35
+ error: str | None = None
36
36
 
37
37
  @field_serializer("timestamp")
38
38
  def serialize_datetime(self, value: datetime) -> str:
@@ -50,9 +50,9 @@ class LogSession(BaseModel):
50
50
 
51
51
  session_id: str
52
52
  started_at: datetime = Field(default_factory=_utc_now)
53
- connector_name: Optional[str] = None
53
+ connector_name: str | None = None
54
54
  logs: List[RequestLog] = Field(default_factory=list)
55
- max_logs: Optional[int] = Field(
55
+ max_logs: int | None = Field(
56
56
  default=10000,
57
57
  description="Maximum number of logs to keep in memory. "
58
58
  "When limit is reached, oldest logs should be flushed before removal. "
@@ -60,7 +60,7 @@ class LogSession(BaseModel):
60
60
  )
61
61
  chunk_logs: List[bytes] = Field(
62
62
  default_factory=list,
63
- description="Captured chunks from streaming responses. " "Each chunk is logged when log_chunk_fetch() is called.",
63
+ description="Captured chunks from streaming responses. Each chunk is logged when log_chunk_fetch() is called.",
64
64
  )
65
65
 
66
66
  @field_validator("chunk_logs", mode="before")
@@ -6,7 +6,7 @@ import tempfile
6
6
  import uuid
7
7
  from dataclasses import dataclass, field
8
8
  from pathlib import Path
9
- from typing import Any, Optional
9
+ from typing import Any
10
10
 
11
11
  import yaml
12
12
 
@@ -53,7 +53,7 @@ def _delete_legacy_files() -> None:
53
53
  logger.debug(f"Could not delete legacy file {legacy_path}: {e}")
54
54
 
55
55
 
56
- def _migrate_legacy_config() -> Optional[SDKConfig]:
56
+ def _migrate_legacy_config() -> SDKConfig | None:
57
57
  """
58
58
  Migrate from legacy file-based config to new YAML format.
59
59
 
@@ -2,7 +2,7 @@
2
2
 
3
3
  from dataclasses import dataclass
4
4
  from datetime import datetime
5
- from typing import Any, Dict, Optional
5
+ from typing import Any, Dict
6
6
 
7
7
 
8
8
  @dataclass
@@ -12,8 +12,8 @@ class OperationMetadata:
12
12
  entity: str
13
13
  action: str
14
14
  timestamp: datetime
15
- timing_ms: Optional[float] = None
16
- status_code: Optional[int] = None
17
- error_type: Optional[str] = None
18
- error_message: Optional[str] = None
19
- params: Optional[Dict[str, Any]] = None
15
+ timing_ms: float | None = None
16
+ status_code: int | None = None
17
+ error_type: str | None = None
18
+ error_message: str | None = None
19
+ params: Dict[str, Any] | None = None
@@ -3,14 +3,14 @@
3
3
  import logging
4
4
  import uuid
5
5
  from datetime import UTC, datetime
6
- from typing import Any, Dict, Optional
6
+ from typing import Any, Dict
7
7
 
8
8
  from .config import SDKConfig, load_config
9
9
 
10
10
  logger = logging.getLogger(__name__)
11
11
 
12
12
  # Cache the config at module level to avoid repeated reads
13
- _cached_config: Optional[SDKConfig] = None
13
+ _cached_config: SDKConfig | None = None
14
14
 
15
15
 
16
16
  def _get_config() -> SDKConfig:
@@ -39,7 +39,7 @@ def get_persistent_user_id() -> str:
39
39
  return _get_config().user_id
40
40
 
41
41
 
42
- def get_public_ip() -> Optional[str]:
42
+ def get_public_ip() -> str | None:
43
43
  """
44
44
  Fetch the public IP address of the user.
45
45
 
@@ -47,6 +47,8 @@ def get_public_ip() -> Optional[str]:
47
47
  Uses httpx for a robust HTTP request to a public IP service.
48
48
  """
49
49
  try:
50
+ # NOTE: Import here intentionally - this is a non-critical network call
51
+ # that may fail. Importing at module level would make httpx a hard dependency.
50
52
  import httpx
51
53
 
52
54
  # Use a short timeout to avoid blocking
@@ -77,9 +79,9 @@ class ObservabilitySession:
77
79
  def __init__(
78
80
  self,
79
81
  connector_name: str,
80
- connector_version: Optional[str] = None,
82
+ connector_version: str | None = None,
81
83
  execution_context: str = "direct",
82
- session_id: Optional[str] = None,
84
+ session_id: str | None = None,
83
85
  ):
84
86
  self.session_id = session_id or str(uuid.uuid4())
85
87
  self.user_id = get_persistent_user_id()
@@ -2,7 +2,7 @@
2
2
 
3
3
  import time
4
4
  from contextlib import asynccontextmanager
5
- from typing import Dict, Optional
5
+ from typing import Dict
6
6
 
7
7
 
8
8
  class PerformanceMonitor:
@@ -33,7 +33,7 @@ class PerformanceMonitor:
33
33
  metrics["min"] = min(metrics["min"], duration)
34
34
  metrics["max"] = max(metrics["max"], duration)
35
35
 
36
- def get_stats(self, metric_name: str) -> Optional[Dict[str, float]]:
36
+ def get_stats(self, metric_name: str) -> Dict[str, float] | None:
37
37
  """Get statistics for a metric.
38
38
 
39
39
  Args:
@@ -62,7 +62,7 @@ class PerformanceMonitor:
62
62
  """
63
63
  return {name: self.get_stats(name) for name in self._metrics.keys()}
64
64
 
65
- def reset(self, metric_name: Optional[str] = None):
65
+ def reset(self, metric_name: str | None = None):
66
66
  """Reset metrics.
67
67
 
68
68
  Args:
@@ -7,7 +7,7 @@ References:
7
7
  """
8
8
 
9
9
  from enum import StrEnum
10
- from typing import Dict, Optional
10
+ from typing import Dict
11
11
  from uuid import UUID
12
12
 
13
13
  from pydantic import BaseModel, ConfigDict, Field, field_validator
@@ -45,9 +45,9 @@ class Contact(BaseModel):
45
45
 
46
46
  model_config = ConfigDict(populate_by_name=True, extra="forbid")
47
47
 
48
- name: Optional[str] = None
49
- url: Optional[str] = None
50
- email: Optional[str] = None
48
+ name: str | None = None
49
+ url: str | None = None
50
+ email: str | None = None
51
51
 
52
52
 
53
53
  class License(BaseModel):
@@ -60,7 +60,7 @@ class License(BaseModel):
60
60
  model_config = ConfigDict(populate_by_name=True, extra="forbid")
61
61
 
62
62
  name: str
63
- url: Optional[str] = None
63
+ url: str | None = None
64
64
 
65
65
 
66
66
  class DocUrlType(StrEnum):
@@ -85,7 +85,7 @@ class DocUrl(BaseModel):
85
85
 
86
86
  url: str
87
87
  type: DocUrlType
88
- title: Optional[str] = None
88
+ title: str | None = None
89
89
 
90
90
  @field_validator("url")
91
91
  def validate_url(cls, v):
@@ -111,17 +111,17 @@ class Info(BaseModel):
111
111
 
112
112
  title: str
113
113
  version: str
114
- description: Optional[str] = None
115
- terms_of_service: Optional[str] = Field(None, alias="termsOfService")
116
- contact: Optional[Contact] = None
117
- license: Optional[License] = None
114
+ description: str | None = None
115
+ terms_of_service: str | None = Field(None, alias="termsOfService")
116
+ contact: Contact | None = None
117
+ license: License | None = None
118
118
 
119
119
  # Airbyte extension
120
- x_airbyte_connector_name: Optional[str] = Field(None, alias="x-airbyte-connector-name")
121
- x_airbyte_connector_id: Optional[UUID] = Field(None, alias="x-airbyte-connector-id")
120
+ x_airbyte_connector_name: str | None = Field(None, alias="x-airbyte-connector-name")
121
+ x_airbyte_connector_id: UUID | None = Field(None, alias="x-airbyte-connector-id")
122
122
  x_airbyte_external_documentation_urls: list[DocUrl] = Field(..., alias="x-airbyte-external-documentation-urls")
123
- x_airbyte_retry_config: Optional[RetryConfig] = Field(None, alias="x-airbyte-retry-config")
124
- x_airbyte_example_questions: Optional[ExampleQuestions] = Field(None, alias="x-airbyte-example-questions")
123
+ x_airbyte_retry_config: RetryConfig | None = Field(None, alias="x-airbyte-retry-config")
124
+ x_airbyte_example_questions: ExampleQuestions | None = Field(None, alias="x-airbyte-example-questions")
125
125
 
126
126
 
127
127
  class ServerVariable(BaseModel):
@@ -133,9 +133,9 @@ class ServerVariable(BaseModel):
133
133
 
134
134
  model_config = ConfigDict(populate_by_name=True, extra="forbid")
135
135
 
136
- enum: Optional[list[str]] = None
136
+ enum: list[str] | None = None
137
137
  default: str
138
- description: Optional[str] = None
138
+ description: str | None = None
139
139
 
140
140
 
141
141
  class Server(BaseModel):
@@ -148,7 +148,7 @@ class Server(BaseModel):
148
148
  model_config = ConfigDict(populate_by_name=True, extra="forbid")
149
149
 
150
150
  url: str
151
- description: Optional[str] = None
151
+ description: str | None = None
152
152
  variables: Dict[str, ServerVariable] = Field(default_factory=dict)
153
153
 
154
154
  @field_validator("url")
@@ -7,7 +7,7 @@ References:
7
7
  - https://spec.openapis.org/oas/v3.1.0#parameter-object
8
8
  """
9
9
 
10
- from typing import Any, Dict, List, Literal, Optional, Union
10
+ from typing import Any, Dict, List, Literal, Union
11
11
 
12
12
  from pydantic import BaseModel, ConfigDict, Field
13
13
 
@@ -30,44 +30,44 @@ class Schema(BaseModel):
30
30
  model_config = ConfigDict(populate_by_name=True, extra="forbid")
31
31
 
32
32
  # Core JSON Schema fields
33
- type: Optional[str] = None
34
- format: Optional[str] = None
35
- title: Optional[str] = None
36
- description: Optional[str] = None
37
- default: Optional[Any] = None
38
- example: Optional[Any] = None
33
+ type: str | None = None
34
+ format: str | None = None
35
+ title: str | None = None
36
+ description: str | None = None
37
+ default: Any | None = None
38
+ example: Any | None = None
39
39
 
40
40
  # Object properties
41
41
  properties: Dict[str, Any] = Field(default_factory=dict) # May contain $ref
42
42
  required: List[str] = Field(default_factory=list)
43
- additional_properties: Optional[Any] = Field(None, alias="additionalProperties")
43
+ additional_properties: Any | None = Field(None, alias="additionalProperties")
44
44
 
45
45
  # Array properties
46
- items: Optional[Any] = None # May be Schema or $ref
46
+ items: Any | None = None # May be Schema or $ref
47
47
 
48
48
  # Validation
49
- enum: Optional[List[Any]] = None
50
- min_length: Optional[int] = Field(None, alias="minLength")
51
- max_length: Optional[int] = Field(None, alias="maxLength")
52
- minimum: Optional[float] = None
53
- maximum: Optional[float] = None
54
- pattern: Optional[str] = None
49
+ enum: List[Any] | None = None
50
+ min_length: int | None = Field(None, alias="minLength")
51
+ max_length: int | None = Field(None, alias="maxLength")
52
+ minimum: float | None = None
53
+ maximum: float | None = None
54
+ pattern: str | None = None
55
55
 
56
56
  # Composition
57
- all_of: Optional[List[Any]] = Field(None, alias="allOf")
58
- any_of: Optional[List[Any]] = Field(None, alias="anyOf")
59
- one_of: Optional[List[Any]] = Field(None, alias="oneOf")
60
- not_: Optional[Any] = Field(None, alias="not")
57
+ all_of: List[Any] | None = Field(None, alias="allOf")
58
+ any_of: List[Any] | None = Field(None, alias="anyOf")
59
+ one_of: List[Any] | None = Field(None, alias="oneOf")
60
+ not_: Any | None = Field(None, alias="not")
61
61
 
62
62
  # Metadata
63
- nullable: Optional[bool] = Field(None, deprecated="Use type union with null instead (OpenAPI 3.1)")
64
- read_only: Optional[bool] = Field(None, alias="readOnly")
65
- write_only: Optional[bool] = Field(None, alias="writeOnly")
66
- deprecated: Optional[bool] = None
63
+ nullable: bool | None = Field(None, deprecated="Use type union with null instead (OpenAPI 3.1)")
64
+ read_only: bool | None = Field(None, alias="readOnly")
65
+ write_only: bool | None = Field(None, alias="writeOnly")
66
+ deprecated: bool | None = None
67
67
 
68
68
  # Airbyte extensions
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")
69
+ x_airbyte_entity_name: str | None = Field(None, alias="x-airbyte-entity-name")
70
+ x_airbyte_stream_name: str | None = Field(None, alias="x-airbyte-stream-name")
71
71
 
72
72
 
73
73
  class Parameter(BaseModel):
@@ -81,19 +81,19 @@ class Parameter(BaseModel):
81
81
 
82
82
  name: str
83
83
  in_: Literal["query", "header", "path", "cookie"] = Field(alias="in")
84
- description: Optional[str] = None
85
- required: Optional[bool] = None
86
- deprecated: Optional[bool] = None
87
- allow_empty_value: Optional[bool] = Field(None, alias="allowEmptyValue")
84
+ description: str | None = None
85
+ required: bool | None = None
86
+ deprecated: bool | None = None
87
+ allow_empty_value: bool | None = Field(None, alias="allowEmptyValue")
88
88
 
89
89
  # Schema can be inline or reference
90
- schema_: Optional[Dict[str, Any]] = Field(None, alias="schema")
90
+ schema_: Dict[str, Any] | None = Field(None, alias="schema")
91
91
 
92
92
  # Style and examples
93
- style: Optional[str] = None
94
- explode: Optional[bool] = None
95
- example: Optional[Any] = None
96
- examples: Optional[Dict[str, Any]] = None
93
+ style: str | None = None
94
+ explode: bool | None = None
95
+ example: Any | None = None
96
+ examples: Dict[str, Any] | None = None
97
97
 
98
98
 
99
99
  class MediaType(BaseModel):
@@ -105,10 +105,10 @@ class MediaType(BaseModel):
105
105
 
106
106
  model_config = ConfigDict(populate_by_name=True, extra="forbid")
107
107
 
108
- schema_: Optional[Dict[str, Any]] = Field(None, alias="schema")
109
- example: Optional[Any] = None
110
- examples: Optional[Dict[str, Any]] = None
111
- encoding: Optional[Dict[str, Any]] = None
108
+ schema_: Dict[str, Any] | None = Field(None, alias="schema")
109
+ example: Any | None = None
110
+ examples: Dict[str, Any] | None = None
111
+ encoding: Dict[str, Any] | None = None
112
112
 
113
113
 
114
114
  class GraphQLBodyConfig(BaseModel):
@@ -125,12 +125,12 @@ class GraphQLBodyConfig(BaseModel):
125
125
  ...,
126
126
  description="GraphQL query or mutation string with optional template placeholders (e.g., {{ variable }})",
127
127
  )
128
- variables: Optional[Dict[str, Any]] = Field(
128
+ variables: Dict[str, Any] | None = Field(
129
129
  None,
130
130
  description="Variables to substitute in the GraphQL query using template syntax (e.g., {{ param_name }})",
131
131
  )
132
- operationName: Optional[str] = Field(None, description="Operation name for queries with multiple operations")
133
- default_fields: Optional[Union[str, List[str]]] = Field(
132
+ operationName: str | None = Field(None, description="Operation name for queries with multiple operations")
133
+ default_fields: Union[str, List[str]] | None = Field(
134
134
  None,
135
135
  description="Default fields to select if not provided in request parameters. Can be a string or array of field names.",
136
136
  )
@@ -156,7 +156,7 @@ class PathOverrideConfig(BaseModel):
156
156
 
157
157
  path: str = Field(
158
158
  ...,
159
- description=("Actual HTTP path to use for requests (e.g., '/graphql'). " "Must start with '/'"),
159
+ description=("Actual HTTP path to use for requests (e.g., '/graphql'). Must start with '/'"),
160
160
  )
161
161
 
162
162
 
@@ -173,17 +173,17 @@ class RequestBody(BaseModel):
173
173
 
174
174
  model_config = ConfigDict(populate_by_name=True, extra="forbid")
175
175
 
176
- description: Optional[str] = None
176
+ description: str | None = None
177
177
  content: Dict[str, MediaType] = Field(default_factory=dict)
178
- required: Optional[bool] = None
178
+ required: bool | None = None
179
179
 
180
180
  # Airbyte extensions for GraphQL support
181
181
  # See connector_sdk.extensions for AIRBYTE_BODY_TYPE constant
182
- x_airbyte_body_type: Optional[BodyTypeConfig] = Field(
182
+ x_airbyte_body_type: BodyTypeConfig | None = Field(
183
183
  None,
184
184
  alias="x-airbyte-body-type", # AIRBYTE_BODY_TYPE
185
185
  description=(
186
- "Body type and configuration. Contains 'type' field (e.g., 'graphql') " "and type-specific configuration (query, variables, etc.)."
186
+ "Body type and configuration. Contains 'type' field (e.g., 'graphql') and type-specific configuration (query, variables, etc.)."
187
187
  ),
188
188
  )
189
189
 
@@ -197,11 +197,11 @@ class Header(BaseModel):
197
197
 
198
198
  model_config = ConfigDict(populate_by_name=True, extra="forbid")
199
199
 
200
- description: Optional[str] = None
201
- required: Optional[bool] = None
202
- deprecated: Optional[bool] = None
203
- schema_: Optional[Dict[str, Any]] = Field(None, alias="schema")
204
- example: Optional[Any] = None
200
+ description: str | None = None
201
+ required: bool | None = None
202
+ deprecated: bool | None = None
203
+ schema_: Dict[str, Any] | None = Field(None, alias="schema")
204
+ example: Any | None = None
205
205
 
206
206
 
207
207
  class Response(BaseModel):
@@ -214,9 +214,9 @@ class Response(BaseModel):
214
214
  model_config = ConfigDict(populate_by_name=True, extra="forbid")
215
215
 
216
216
  description: str
217
- headers: Optional[Dict[str, Header]] = None
218
- content: Optional[Dict[str, MediaType]] = None
219
- links: Optional[Dict[str, Any]] = None
217
+ headers: Dict[str, Header] | None = None
218
+ content: Dict[str, MediaType] | None = None
219
+ links: Dict[str, Any] | None = None
220
220
 
221
221
 
222
222
  class Components(BaseModel):
@@ -231,9 +231,9 @@ class Components(BaseModel):
231
231
  schemas: Dict[str, Schema] = Field(default_factory=dict)
232
232
  responses: Dict[str, Response] = Field(default_factory=dict)
233
233
  parameters: Dict[str, Parameter] = Field(default_factory=dict)
234
- examples: Optional[Dict[str, Any]] = None
234
+ examples: Dict[str, Any] | None = None
235
235
  request_bodies: Dict[str, RequestBody] = Field(default_factory=dict, alias="requestBodies")
236
- headers: Optional[Dict[str, Header]] = None
236
+ headers: Dict[str, Header] | None = None
237
237
  security_schemes: Dict[str, SecurityScheme] = Field(default_factory=dict, alias="securitySchemes")
238
- links: Optional[Dict[str, Any]] = None
239
- callbacks: Optional[Dict[str, Any]] = None
238
+ links: Dict[str, Any] | None = None
239
+ callbacks: Dict[str, Any] | None = None
@@ -7,6 +7,7 @@ References:
7
7
 
8
8
  from __future__ import annotations
9
9
 
10
+ from collections.abc import Iterator
10
11
  from typing import Any
11
12
 
12
13
  from pydantic import BaseModel, ConfigDict, Field, field_validator
@@ -15,7 +16,7 @@ from ..constants import OPENAPI_VERSION_PREFIX
15
16
 
16
17
  from .base import Info, Server
17
18
  from .components import Components
18
- from .operations import PathItem
19
+ from .operations import Operation, PathItem
19
20
  from .security import SecurityRequirement
20
21
 
21
22
 
@@ -33,6 +34,10 @@ class Tag(BaseModel):
33
34
  external_docs: dict[str, Any] | None = Field(None, alias="externalDocs")
34
35
 
35
36
 
37
+ # HTTP methods supported by OpenAPI operations
38
+ HTTP_METHODS = frozenset({"get", "post", "put", "patch", "delete", "options", "head", "trace"})
39
+
40
+
36
41
  class ExternalDocs(BaseModel):
37
42
  """
38
43
  External documentation reference.
@@ -79,7 +84,7 @@ class OpenAPIConnector(BaseModel):
79
84
  raise ValueError(f"OpenAPI version must be {OPENAPI_VERSION_PREFIX}x, got: {v}")
80
85
  return v
81
86
 
82
- def get_entity_operations(self, entity_name: str) -> list[tuple[str, str, Any]]:
87
+ def get_entity_operations(self, entity_name: str) -> list[tuple[str, str, Operation]]:
83
88
  """
84
89
  Get all operations for a specific entity.
85
90
 
@@ -89,22 +94,7 @@ class OpenAPIConnector(BaseModel):
89
94
  Returns:
90
95
  List of tuples: (path, method, operation)
91
96
  """
92
- results = []
93
- for path, path_item in self.paths.items():
94
- for method in [
95
- "get",
96
- "post",
97
- "put",
98
- "patch",
99
- "delete",
100
- "options",
101
- "head",
102
- "trace",
103
- ]:
104
- operation = getattr(path_item, method, None)
105
- if operation and operation.x_airbyte_entity == entity_name:
106
- results.append((path, method, operation))
107
- return results
97
+ return [(path, method, op) for path, method, op in self._iter_operations() if op.x_airbyte_entity == entity_name]
108
98
 
109
99
  def list_entities(self) -> list[str]:
110
100
  """
@@ -113,19 +103,18 @@ class OpenAPIConnector(BaseModel):
113
103
  Returns:
114
104
  Sorted list of unique entity names
115
105
  """
116
- entities = set()
117
- for path_item in self.paths.values():
118
- for method in [
119
- "get",
120
- "post",
121
- "put",
122
- "patch",
123
- "delete",
124
- "options",
125
- "head",
126
- "trace",
127
- ]:
128
- operation = getattr(path_item, method, None)
129
- if operation and operation.x_airbyte_entity:
130
- entities.add(operation.x_airbyte_entity)
106
+ entities = {op.x_airbyte_entity for _, _, op in self._iter_operations() if op.x_airbyte_entity}
131
107
  return sorted(entities)
108
+
109
+ def _iter_operations(self) -> Iterator[tuple[str, str, Operation]]:
110
+ """
111
+ Iterate over all operations in the spec.
112
+
113
+ Yields:
114
+ Tuples of (path, method, operation) for each defined operation
115
+ """
116
+ for path, path_item in self.paths.items():
117
+ for method in HTTP_METHODS:
118
+ operation = getattr(path_item, method, None)
119
+ if operation:
120
+ yield path, method, operation
@@ -12,7 +12,7 @@ to Operation, Schema, or other models when their respective features
12
12
  are implemented.
13
13
  """
14
14
 
15
- from typing import Literal, Optional
15
+ from typing import Literal
16
16
 
17
17
  from pydantic import BaseModel, ConfigDict
18
18
 
@@ -33,22 +33,22 @@ class PaginationConfig(BaseModel):
33
33
  limit_param: str = "limit"
34
34
 
35
35
  # Cursor-based pagination
36
- cursor_param: Optional[str] = None
37
- cursor_source: Optional[Literal["body", "headers"]] = "body"
38
- cursor_path: Optional[str] = None
36
+ cursor_param: str | None = None
37
+ cursor_source: Literal["body", "headers"] | None = "body"
38
+ cursor_path: str | None = None
39
39
 
40
40
  # Offset-based pagination
41
- offset_param: Optional[str] = None
41
+ offset_param: str | None = None
42
42
 
43
43
  # Page-based pagination
44
- page_param: Optional[str] = None
44
+ page_param: str | None = None
45
45
 
46
46
  # Response parsing
47
47
  data_path: str = "data"
48
- has_more_path: Optional[str] = None
48
+ has_more_path: str | None = None
49
49
 
50
50
  # Limits
51
- max_page_size: Optional[int] = None
51
+ max_page_size: int | None = None
52
52
  default_page_size: int = 100
53
53
 
54
54
 
@@ -66,7 +66,7 @@ class RateLimitConfig(BaseModel):
66
66
 
67
67
  max_requests: int
68
68
  time_window_seconds: int
69
- retry_after_header: Optional[str] = "Retry-After"
69
+ retry_after_header: str | None = "Retry-After"
70
70
  respect_retry_after: bool = True
71
71
 
72
72