airbyte-agent-hubspot 0.15.28__py3-none-any.whl → 0.15.43__py3-none-any.whl

airbyte_agent_hubspot/_vendored/connector_sdk/schema/components.py

@@ -7,7 +7,7 @@ References:
 - https://spec.openapis.org/oas/v3.1.0#parameter-object
 """
 
-from typing import Any, Dict, List, Literal, Optional, Union
+from typing import Any, Dict, List, Literal, Union
 
 from pydantic import BaseModel, ConfigDict, Field
 
@@ -30,43 +30,44 @@ class Schema(BaseModel):
     model_config = ConfigDict(populate_by_name=True, extra="forbid")
 
     # Core JSON Schema fields
-    type: Optional[str] = None
-    format: Optional[str] = None
-    title: Optional[str] = None
-    description: Optional[str] = None
-    default: Optional[Any] = None
-    example: Optional[Any] = None
+    type: str | None = None
+    format: str | None = None
+    title: str | None = None
+    description: str | None = None
+    default: Any | None = None
+    example: Any | None = None
 
     # Object properties
     properties: Dict[str, Any] = Field(default_factory=dict)  # May contain $ref
     required: List[str] = Field(default_factory=list)
-    additional_properties: Optional[Any] = Field(None, alias="additionalProperties")
+    additional_properties: Any | None = Field(None, alias="additionalProperties")
 
     # Array properties
-    items: Optional[Any] = None  # May be Schema or $ref
+    items: Any | None = None  # May be Schema or $ref
 
     # Validation
-    enum: Optional[List[Any]] = None
-    min_length: Optional[int] = Field(None, alias="minLength")
-    max_length: Optional[int] = Field(None, alias="maxLength")
-    minimum: Optional[float] = None
-    maximum: Optional[float] = None
-    pattern: Optional[str] = None
+    enum: List[Any] | None = None
+    min_length: int | None = Field(None, alias="minLength")
+    max_length: int | None = Field(None, alias="maxLength")
+    minimum: float | None = None
+    maximum: float | None = None
+    pattern: str | None = None
 
     # Composition
-    all_of: Optional[List[Any]] = Field(None, alias="allOf")
-    any_of: Optional[List[Any]] = Field(None, alias="anyOf")
-    one_of: Optional[List[Any]] = Field(None, alias="oneOf")
-    not_: Optional[Any] = Field(None, alias="not")
+    all_of: List[Any] | None = Field(None, alias="allOf")
+    any_of: List[Any] | None = Field(None, alias="anyOf")
+    one_of: List[Any] | None = Field(None, alias="oneOf")
+    not_: Any | None = Field(None, alias="not")
 
     # Metadata
-    nullable: Optional[bool] = Field(None, deprecated="Use type union with null instead (OpenAPI 3.1)")
-    read_only: Optional[bool] = Field(None, alias="readOnly")
-    write_only: Optional[bool] = Field(None, alias="writeOnly")
-    deprecated: Optional[bool] = None
+    nullable: bool | None = Field(None, deprecated="Use type union with null instead (OpenAPI 3.1)")
+    read_only: bool | None = Field(None, alias="readOnly")
+    write_only: bool | None = Field(None, alias="writeOnly")
+    deprecated: bool | None = None
 
-    # Airbyte extension
-    x_airbyte_entity_name: Optional[str] = Field(None, alias="x-airbyte-entity-name")
+    # Airbyte extensions
+    x_airbyte_entity_name: str | None = Field(None, alias="x-airbyte-entity-name")
+    x_airbyte_stream_name: str | None = Field(None, alias="x-airbyte-stream-name")
 
 
 class Parameter(BaseModel):
@@ -80,19 +81,19 @@ class Parameter(BaseModel):
 
     name: str
     in_: Literal["query", "header", "path", "cookie"] = Field(alias="in")
-    description: Optional[str] = None
-    required: Optional[bool] = None
-    deprecated: Optional[bool] = None
-    allow_empty_value: Optional[bool] = Field(None, alias="allowEmptyValue")
+    description: str | None = None
+    required: bool | None = None
+    deprecated: bool | None = None
+    allow_empty_value: bool | None = Field(None, alias="allowEmptyValue")
 
     # Schema can be inline or reference
-    schema_: Optional[Dict[str, Any]] = Field(None, alias="schema")
+    schema_: Dict[str, Any] | None = Field(None, alias="schema")
 
     # Style and examples
-    style: Optional[str] = None
-    explode: Optional[bool] = None
-    example: Optional[Any] = None
-    examples: Optional[Dict[str, Any]] = None
+    style: str | None = None
+    explode: bool | None = None
+    example: Any | None = None
+    examples: Dict[str, Any] | None = None
 
 
 class MediaType(BaseModel):
@@ -104,10 +105,10 @@ class MediaType(BaseModel):
 
     model_config = ConfigDict(populate_by_name=True, extra="forbid")
 
-    schema_: Optional[Dict[str, Any]] = Field(None, alias="schema")
-    example: Optional[Any] = None
-    examples: Optional[Dict[str, Any]] = None
-    encoding: Optional[Dict[str, Any]] = None
+    schema_: Dict[str, Any] | None = Field(None, alias="schema")
+    example: Any | None = None
+    examples: Dict[str, Any] | None = None
+    encoding: Dict[str, Any] | None = None
 
 
 class GraphQLBodyConfig(BaseModel):
@@ -124,12 +125,12 @@ class GraphQLBodyConfig(BaseModel):
         ...,
         description="GraphQL query or mutation string with optional template placeholders (e.g., {{ variable }})",
     )
-    variables: Optional[Dict[str, Any]] = Field(
+    variables: Dict[str, Any] | None = Field(
         None,
         description="Variables to substitute in the GraphQL query using template syntax (e.g., {{ param_name }})",
     )
-    operationName: Optional[str] = Field(None, description="Operation name for queries with multiple operations")
-    default_fields: Optional[Union[str, List[str]]] = Field(
+    operationName: str | None = Field(None, description="Operation name for queries with multiple operations")
+    default_fields: Union[str, List[str]] | None = Field(
         None,
         description="Default fields to select if not provided in request parameters. Can be a string or array of field names.",
     )
@@ -155,7 +156,7 @@ class PathOverrideConfig(BaseModel):
 
     path: str = Field(
         ...,
-        description=("Actual HTTP path to use for requests (e.g., '/graphql'). " "Must start with '/'"),
+        description=("Actual HTTP path to use for requests (e.g., '/graphql'). Must start with '/'"),
     )
 
 
@@ -172,17 +173,17 @@ class RequestBody(BaseModel):
 
     model_config = ConfigDict(populate_by_name=True, extra="forbid")
 
-    description: Optional[str] = None
+    description: str | None = None
     content: Dict[str, MediaType] = Field(default_factory=dict)
-    required: Optional[bool] = None
+    required: bool | None = None
 
     # Airbyte extensions for GraphQL support
     # See connector_sdk.extensions for AIRBYTE_BODY_TYPE constant
-    x_airbyte_body_type: Optional[BodyTypeConfig] = Field(
+    x_airbyte_body_type: BodyTypeConfig | None = Field(
         None,
         alias="x-airbyte-body-type",  # AIRBYTE_BODY_TYPE
         description=(
-            "Body type and configuration. Contains 'type' field (e.g., 'graphql') " "and type-specific configuration (query, variables, etc.)."
+            "Body type and configuration. Contains 'type' field (e.g., 'graphql') and type-specific configuration (query, variables, etc.)."
         ),
     )
 
@@ -196,11 +197,11 @@ class Header(BaseModel):
 
     model_config = ConfigDict(populate_by_name=True, extra="forbid")
 
-    description: Optional[str] = None
-    required: Optional[bool] = None
-    deprecated: Optional[bool] = None
-    schema_: Optional[Dict[str, Any]] = Field(None, alias="schema")
-    example: Optional[Any] = None
+    description: str | None = None
+    required: bool | None = None
+    deprecated: bool | None = None
+    schema_: Dict[str, Any] | None = Field(None, alias="schema")
+    example: Any | None = None
 
 
 class Response(BaseModel):
@@ -213,9 +214,9 @@ class Response(BaseModel):
     model_config = ConfigDict(populate_by_name=True, extra="forbid")
 
     description: str
-    headers: Optional[Dict[str, Header]] = None
-    content: Optional[Dict[str, MediaType]] = None
-    links: Optional[Dict[str, Any]] = None
+    headers: Dict[str, Header] | None = None
+    content: Dict[str, MediaType] | None = None
+    links: Dict[str, Any] | None = None
 
 
 class Components(BaseModel):
@@ -230,9 +231,9 @@ class Components(BaseModel):
     schemas: Dict[str, Schema] = Field(default_factory=dict)
     responses: Dict[str, Response] = Field(default_factory=dict)
     parameters: Dict[str, Parameter] = Field(default_factory=dict)
-    examples: Optional[Dict[str, Any]] = None
+    examples: Dict[str, Any] | None = None
     request_bodies: Dict[str, RequestBody] = Field(default_factory=dict, alias="requestBodies")
-    headers: Optional[Dict[str, Header]] = None
+    headers: Dict[str, Header] | None = None
     security_schemes: Dict[str, SecurityScheme] = Field(default_factory=dict, alias="securitySchemes")
-    links: Optional[Dict[str, Any]] = None
-    callbacks: Optional[Dict[str, Any]] = None
+    links: Dict[str, Any] | None = None
+    callbacks: Dict[str, Any] | None = None

airbyte_agent_hubspot/_vendored/connector_sdk/schema/connector.py

@@ -7,6 +7,7 @@ References:
 
 from __future__ import annotations
 
+from collections.abc import Iterator
 from typing import Any
 
 from pydantic import BaseModel, ConfigDict, Field, field_validator
@@ -15,7 +16,7 @@ from ..constants import OPENAPI_VERSION_PREFIX
 
 from .base import Info, Server
 from .components import Components
-from .operations import PathItem
+from .operations import Operation, PathItem
 from .security import SecurityRequirement
 
 
@@ -33,6 +34,10 @@ class Tag(BaseModel):
     external_docs: dict[str, Any] | None = Field(None, alias="externalDocs")
 
 
+# HTTP methods supported by OpenAPI operations
+HTTP_METHODS = frozenset({"get", "post", "put", "patch", "delete", "options", "head", "trace"})
+
+
 class ExternalDocs(BaseModel):
     """
     External documentation reference.
@@ -79,7 +84,7 @@ class OpenAPIConnector(BaseModel):
             raise ValueError(f"OpenAPI version must be {OPENAPI_VERSION_PREFIX}x, got: {v}")
         return v
 
-    def get_entity_operations(self, entity_name: str) -> list[tuple[str, str, Any]]:
+    def get_entity_operations(self, entity_name: str) -> list[tuple[str, str, Operation]]:
         """
         Get all operations for a specific entity.
 
@@ -89,22 +94,7 @@ class OpenAPIConnector(BaseModel):
         Returns:
             List of tuples: (path, method, operation)
         """
-        results = []
-        for path, path_item in self.paths.items():
-            for method in [
-                "get",
-                "post",
-                "put",
-                "patch",
-                "delete",
-                "options",
-                "head",
-                "trace",
-            ]:
-                operation = getattr(path_item, method, None)
-                if operation and operation.x_airbyte_entity == entity_name:
-                    results.append((path, method, operation))
-        return results
+        return [(path, method, op) for path, method, op in self._iter_operations() if op.x_airbyte_entity == entity_name]
 
     def list_entities(self) -> list[str]:
         """
@@ -113,19 +103,18 @@ class OpenAPIConnector(BaseModel):
         Returns:
             Sorted list of unique entity names
         """
-        entities = set()
-        for path_item in self.paths.values():
-            for method in [
-                "get",
-                "post",
-                "put",
-                "patch",
-                "delete",
-                "options",
-                "head",
-                "trace",
-            ]:
-                operation = getattr(path_item, method, None)
-                if operation and operation.x_airbyte_entity:
-                    entities.add(operation.x_airbyte_entity)
+        entities = {op.x_airbyte_entity for _, _, op in self._iter_operations() if op.x_airbyte_entity}
         return sorted(entities)
+
+    def _iter_operations(self) -> Iterator[tuple[str, str, Operation]]:
+        """
+        Iterate over all operations in the spec.
+
+        Yields:
+            Tuples of (path, method, operation) for each defined operation
+        """
+        for path, path_item in self.paths.items():
+            for method in HTTP_METHODS:
+                operation = getattr(path_item, method, None)
+                if operation:
+                    yield path, method, operation

airbyte_agent_hubspot/_vendored/connector_sdk/schema/extensions.py

@@ -12,9 +12,9 @@ to Operation, Schema, or other models when their respective features
 are implemented.
 """
 
-from typing import Literal, Optional
+from typing import Literal
 
-from pydantic import BaseModel, ConfigDict
+from pydantic import BaseModel, ConfigDict, Field
 
 
 class PaginationConfig(BaseModel):
@@ -33,22 +33,22 @@ class PaginationConfig(BaseModel):
     limit_param: str = "limit"
 
     # Cursor-based pagination
-    cursor_param: Optional[str] = None
-    cursor_source: Optional[Literal["body", "headers"]] = "body"
-    cursor_path: Optional[str] = None
+    cursor_param: str | None = None
+    cursor_source: Literal["body", "headers"] | None = "body"
+    cursor_path: str | None = None
 
     # Offset-based pagination
-    offset_param: Optional[str] = None
+    offset_param: str | None = None
 
     # Page-based pagination
-    page_param: Optional[str] = None
+    page_param: str | None = None
 
     # Response parsing
     data_path: str = "data"
-    has_more_path: Optional[str] = None
+    has_more_path: str | None = None
 
     # Limits
-    max_page_size: Optional[int] = None
+    max_page_size: int | None = None
     default_page_size: int = 100
 
 
@@ -66,7 +66,7 @@ class RateLimitConfig(BaseModel):
 
     max_requests: int
     time_window_seconds: int
-    retry_after_header: Optional[str] = "Retry-After"
+    retry_after_header: str | None = "Retry-After"
     respect_retry_after: bool = True
 
 
@@ -107,3 +107,96 @@ class RetryConfig(BaseModel):
     # Header-based delay extraction
     retry_after_header: str = "Retry-After"
     retry_after_format: Literal["seconds", "milliseconds", "unix_timestamp"] = "seconds"
+
+
+class CacheFieldConfig(BaseModel):
+    """
+    Field configuration for cache mapping.
+
+    Defines a single field in a cache entity, with optional name aliasing
+    to map between user-facing field names and cache storage names.
+
+    Used in x-airbyte-cache extension for api_search operations.
+    """
+
+    model_config = ConfigDict(populate_by_name=True, extra="forbid")
+
+    name: str
+    x_airbyte_name: str | None = Field(default=None, alias="x-airbyte-name")
+    type: str | list[str]
+    description: str
+
+    @property
+    def cache_name(self) -> str:
+        """Return cache name, falling back to name if alias not specified."""
+        return self.x_airbyte_name or self.name
+
+
+class CacheEntityConfig(BaseModel):
+    """
+    Entity configuration for cache mapping.
+
+    Defines a cache-enabled entity with its fields and optional name aliasing
+    to map between user-facing entity names and cache storage names.
+
+    Used in x-airbyte-cache extension for api_search operations.
+    """
+
+    model_config = ConfigDict(populate_by_name=True, extra="forbid")
+
+    entity: str
+    x_airbyte_name: str | None = Field(default=None, alias="x-airbyte-name")
+    fields: list[CacheFieldConfig]
+
+    @property
+    def cache_name(self) -> str:
+        """Return cache entity name, falling back to entity if alias not specified."""
+        return self.x_airbyte_name or self.entity
+
+
+class CacheConfig(BaseModel):
+    """
+    Cache configuration extension (x-airbyte-cache).
+
+    Defines cache-enabled entities and their field mappings for api_search operations.
+    Supports optional name aliasing via x-airbyte-name for both entities and fields,
+    enabling bidirectional mapping between user-facing names and cache storage names.
+
+    This extension is added to the Info model and provides field-level mapping for
+    search operations that use cached data.
+
+    Example YAML usage:
+        info:
+          title: Stripe API
+          x-airbyte-cache:
+            entities:
+              - entity: customers
+                stream: customers
+                fields:
+                  - name: email
+                    type: ["null", "string"]
+                    description: "Customer email address"
+                  - name: customer_name
+                    x-airbyte-name: name
+                    type: ["null", "string"]
+                    description: "Customer full name"
+    """
+
+    model_config = ConfigDict(populate_by_name=True, extra="forbid")
+
+    entities: list[CacheEntityConfig]
+
+    def get_entity_mapping(self, user_entity: str) -> CacheEntityConfig | None:
+        """
+        Get entity config by user-facing name.
+
+        Args:
+            user_entity: User-facing entity name to look up
+
+        Returns:
+            CacheEntityConfig if found, None otherwise
+        """
+        for entity in self.entities:
+            if entity.entity == user_entity:
+                return entity
+        return None

airbyte_agent_hubspot/_vendored/connector_sdk/schema/operations.py

@@ -6,7 +6,7 @@ References:
 - https://spec.openapis.org/oas/v3.1.0#path-item-object
 """
 
-from typing import Any, Dict, List, Optional
+from typing import Any, Dict, List
 
 from pydantic import BaseModel, ConfigDict, Field, model_validator
 
@@ -34,38 +34,38 @@ class Operation(BaseModel):
     model_config = ConfigDict(populate_by_name=True, extra="forbid")
 
     # Standard OpenAPI fields
-    tags: Optional[List[str]] = None
-    summary: Optional[str] = None
-    description: Optional[str] = None
-    external_docs: Optional[Dict[str, Any]] = Field(None, alias="externalDocs")
-    operation_id: Optional[str] = Field(None, alias="operationId")
-    parameters: Optional[List[Parameter]] = None
-    request_body: Optional[RequestBody] = Field(None, alias="requestBody")
+    tags: List[str] | None = None
+    summary: str | None = None
+    description: str | None = None
+    external_docs: Dict[str, Any] | None = Field(None, alias="externalDocs")
+    operation_id: str | None = Field(None, alias="operationId")
+    parameters: List[Parameter] | None = None
+    request_body: RequestBody | None = Field(None, alias="requestBody")
     responses: Dict[str, Response] = Field(default_factory=dict)
-    callbacks: Optional[Dict[str, Any]] = None
-    deprecated: Optional[bool] = None
-    security: Optional[List[SecurityRequirement]] = None
-    servers: Optional[List[Any]] = None  # Can override root servers
+    callbacks: Dict[str, Any] | None = None
+    deprecated: bool | None = None
+    security: List[SecurityRequirement] | None = None
+    servers: List[Any] | None = None  # Can override root servers
 
     # Airbyte extensions
     x_airbyte_entity: str = Field(..., alias="x-airbyte-entity")
     x_airbyte_action: ActionTypeLiteral = Field(..., alias="x-airbyte-action")
-    x_airbyte_path_override: Optional[PathOverrideConfig] = Field(
+    x_airbyte_path_override: PathOverrideConfig | None = Field(
         None,
         alias="x-airbyte-path-override",
-        description=("Override path for HTTP requests when OpenAPI path " "differs from actual endpoint"),
+        description=("Override path for HTTP requests when OpenAPI path differs from actual endpoint"),
     )
-    x_airbyte_record_extractor: Optional[str] = Field(
+    x_airbyte_record_extractor: str | None = Field(
         None,
         alias="x-airbyte-record-extractor",
         description=(
             "JSONPath expression to extract records from API response envelopes. "
             "When specified, executor extracts data at this path instead of returning "
-            "full response. Returns array for list/search actions, single record for "
+            "full response. Returns array for list/api_search actions, single record for "
             "get/create/update/delete actions."
         ),
     )
-    x_airbyte_meta_extractor: Optional[Dict[str, str]] = Field(
+    x_airbyte_meta_extractor: Dict[str, str] | None = Field(
         None,
         alias="x-airbyte-meta-extractor",
         description=(
@@ -76,8 +76,8 @@ class Operation(BaseModel):
             "Example: {'pagination': '$.pagination', 'request_id': '$.requestId'}"
         ),
     )
-    x_airbyte_file_url: Optional[str] = Field(None, alias="x-airbyte-file-url")
-    x_airbyte_untested: Optional[bool] = Field(
+    x_airbyte_file_url: str | None = Field(None, alias="x-airbyte-file-url")
+    x_airbyte_untested: bool | None = Field(
         None,
         alias="x-airbyte-untested",
         description=(
@@ -127,20 +127,20 @@ class PathItem(BaseModel):
     model_config = ConfigDict(populate_by_name=True, extra="forbid")
 
     # Common fields for all operations
-    summary: Optional[str] = None
-    description: Optional[str] = None
-    servers: Optional[List[Any]] = None
-    parameters: Optional[List[Parameter]] = None
+    summary: str | None = None
+    description: str | None = None
+    servers: List[Any] | None = None
+    parameters: List[Parameter] | None = None
 
     # HTTP methods (all optional)
-    get: Optional[Operation] = None
-    put: Optional[Operation] = None
-    post: Optional[Operation] = None
-    delete: Optional[Operation] = None
-    options: Optional[Operation] = None
-    head: Optional[Operation] = None
-    patch: Optional[Operation] = None
-    trace: Optional[Operation] = None
+    get: Operation | None = None
+    put: Operation | None = None
+    post: Operation | None = None
+    delete: Operation | None = None
+    options: Operation | None = None
+    head: Operation | None = None
+    patch: Operation | None = None
+    trace: Operation | None = None
 
     # Reference support
-    ref: Optional[str] = Field(None, alias="$ref")
+    ref: str | None = Field(None, alias="$ref")