antfly-sdk 0.0.2.dev2__py3-none-any.whl

antfly/client_generated/models/regexp_query.py

@@ -0,0 +1,93 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import Any, TypeVar, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="RegexpQuery")
+
+
+@_attrs_define
+class RegexpQuery:
+    """
+    Attributes:
+        regexp (str):
+        field (str | Unset):
+        boost (float | None | Unset): A floating-point number used to decrease or increase the relevance scores of a
+            query.
+    """
+
+    regexp: str
+    field: str | Unset = UNSET
+    boost: float | None | Unset = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        regexp = self.regexp
+
+        field = self.field
+
+        boost: float | None | Unset
+        if isinstance(self.boost, Unset):
+            boost = UNSET
+        else:
+            boost = self.boost
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "regexp": regexp,
+            }
+        )
+        if field is not UNSET:
+            field_dict["field"] = field
+        if boost is not UNSET:
+            field_dict["boost"] = boost
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        regexp = d.pop("regexp")
+
+        field = d.pop("field", UNSET)
+
+        def _parse_boost(data: object) -> float | None | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(float | None | Unset, data)
+
+        boost = _parse_boost(d.pop("boost", UNSET))
+
+        regexp_query = cls(
+            regexp=regexp,
+            field=field,
+            boost=boost,
+        )
+
+        regexp_query.additional_properties = d
+        return regexp_query
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

antfly/client_generated/models/replication_source_action_hint.py

@@ -0,0 +1,81 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="ReplicationSourceActionHint")
+
+
+@_attrs_define
+class ReplicationSourceActionHint:
+    """Action hint for this replication source when remediation is recommended.
+    Present only in GET table detail responses.
+
+        Attributes:
+            action (str | Unset): Recommended action (e.g., "reseed_exact_cutover").
+            reason (str | Unset): Human-readable reason for the recommendation.
+            reseed_exact_cutover_path (str | Unset): API path to trigger a reseed exact cutover, if applicable.
+    """
+
+    action: str | Unset = UNSET
+    reason: str | Unset = UNSET
+    reseed_exact_cutover_path: str | Unset = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        action = self.action
+
+        reason = self.reason
+
+        reseed_exact_cutover_path = self.reseed_exact_cutover_path
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if action is not UNSET:
+            field_dict["action"] = action
+        if reason is not UNSET:
+            field_dict["reason"] = reason
+        if reseed_exact_cutover_path is not UNSET:
+            field_dict["reseed_exact_cutover_path"] = reseed_exact_cutover_path
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        action = d.pop("action", UNSET)
+
+        reason = d.pop("reason", UNSET)
+
+        reseed_exact_cutover_path = d.pop("reseed_exact_cutover_path", UNSET)
+
+        replication_source_action_hint = cls(
+            action=action,
+            reason=reason,
+            reseed_exact_cutover_path=reseed_exact_cutover_path,
+        )
+
+        replication_source_action_hint.additional_properties = d
+        return replication_source_action_hint
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

antfly/client_generated/models/replication_source_status.py

@@ -0,0 +1,225 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="ReplicationSourceStatus")
+
+
+@_attrs_define
+class ReplicationSourceStatus:
+    """Runtime status of this replication source. Present only in GET table
+    detail responses, not in create/update requests.
+
+        Attributes:
+            source_kind (str | Unset):
+            external_table (str | Unset):
+            cutover_mode (str | Unset):
+            slot_name (str | Unset):
+            publication_name (str | Unset):
+            phase (str | Unset):
+            checkpoint (str | Unset):
+            snapshot_offset (int | Unset):
+            prepared_checkpoint (str | Unset):
+            stream_checkpoint (str | Unset):
+            last_error (str | Unset):
+            failure_class (str | Unset):
+            lag_records (int | Unset):
+            lag_millis (int | Unset):
+            consecutive_failures (int | Unset):
+            last_source_commit_at_ms (int | Unset):
+            last_success_at_ms (int | Unset):
+            last_change_applied_at_ms (int | Unset):
+            updated_at_ms (int | Unset):
+    """
+
+    source_kind: str | Unset = UNSET
+    external_table: str | Unset = UNSET
+    cutover_mode: str | Unset = UNSET
+    slot_name: str | Unset = UNSET
+    publication_name: str | Unset = UNSET
+    phase: str | Unset = UNSET
+    checkpoint: str | Unset = UNSET
+    snapshot_offset: int | Unset = UNSET
+    prepared_checkpoint: str | Unset = UNSET
+    stream_checkpoint: str | Unset = UNSET
+    last_error: str | Unset = UNSET
+    failure_class: str | Unset = UNSET
+    lag_records: int | Unset = UNSET
+    lag_millis: int | Unset = UNSET
+    consecutive_failures: int | Unset = UNSET
+    last_source_commit_at_ms: int | Unset = UNSET
+    last_success_at_ms: int | Unset = UNSET
+    last_change_applied_at_ms: int | Unset = UNSET
+    updated_at_ms: int | Unset = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        source_kind = self.source_kind
+
+        external_table = self.external_table
+
+        cutover_mode = self.cutover_mode
+
+        slot_name = self.slot_name
+
+        publication_name = self.publication_name
+
+        phase = self.phase
+
+        checkpoint = self.checkpoint
+
+        snapshot_offset = self.snapshot_offset
+
+        prepared_checkpoint = self.prepared_checkpoint
+
+        stream_checkpoint = self.stream_checkpoint
+
+        last_error = self.last_error
+
+        failure_class = self.failure_class
+
+        lag_records = self.lag_records
+
+        lag_millis = self.lag_millis
+
+        consecutive_failures = self.consecutive_failures
+
+        last_source_commit_at_ms = self.last_source_commit_at_ms
+
+        last_success_at_ms = self.last_success_at_ms
+
+        last_change_applied_at_ms = self.last_change_applied_at_ms
+
+        updated_at_ms = self.updated_at_ms
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if source_kind is not UNSET:
+            field_dict["source_kind"] = source_kind
+        if external_table is not UNSET:
+            field_dict["external_table"] = external_table
+        if cutover_mode is not UNSET:
+            field_dict["cutover_mode"] = cutover_mode
+        if slot_name is not UNSET:
+            field_dict["slot_name"] = slot_name
+        if publication_name is not UNSET:
+            field_dict["publication_name"] = publication_name
+        if phase is not UNSET:
+            field_dict["phase"] = phase
+        if checkpoint is not UNSET:
+            field_dict["checkpoint"] = checkpoint
+        if snapshot_offset is not UNSET:
+            field_dict["snapshot_offset"] = snapshot_offset
+        if prepared_checkpoint is not UNSET:
+            field_dict["prepared_checkpoint"] = prepared_checkpoint
+        if stream_checkpoint is not UNSET:
+            field_dict["stream_checkpoint"] = stream_checkpoint
+        if last_error is not UNSET:
+            field_dict["last_error"] = last_error
+        if failure_class is not UNSET:
+            field_dict["failure_class"] = failure_class
+        if lag_records is not UNSET:
+            field_dict["lag_records"] = lag_records
+        if lag_millis is not UNSET:
+            field_dict["lag_millis"] = lag_millis
+        if consecutive_failures is not UNSET:
+            field_dict["consecutive_failures"] = consecutive_failures
+        if last_source_commit_at_ms is not UNSET:
+            field_dict["last_source_commit_at_ms"] = last_source_commit_at_ms
+        if last_success_at_ms is not UNSET:
+            field_dict["last_success_at_ms"] = last_success_at_ms
+        if last_change_applied_at_ms is not UNSET:
+            field_dict["last_change_applied_at_ms"] = last_change_applied_at_ms
+        if updated_at_ms is not UNSET:
+            field_dict["updated_at_ms"] = updated_at_ms
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        source_kind = d.pop("source_kind", UNSET)
+
+        external_table = d.pop("external_table", UNSET)
+
+        cutover_mode = d.pop("cutover_mode", UNSET)
+
+        slot_name = d.pop("slot_name", UNSET)
+
+        publication_name = d.pop("publication_name", UNSET)
+
+        phase = d.pop("phase", UNSET)
+
+        checkpoint = d.pop("checkpoint", UNSET)
+
+        snapshot_offset = d.pop("snapshot_offset", UNSET)
+
+        prepared_checkpoint = d.pop("prepared_checkpoint", UNSET)
+
+        stream_checkpoint = d.pop("stream_checkpoint", UNSET)
+
+        last_error = d.pop("last_error", UNSET)
+
+        failure_class = d.pop("failure_class", UNSET)
+
+        lag_records = d.pop("lag_records", UNSET)
+
+        lag_millis = d.pop("lag_millis", UNSET)
+
+        consecutive_failures = d.pop("consecutive_failures", UNSET)
+
+        last_source_commit_at_ms = d.pop("last_source_commit_at_ms", UNSET)
+
+        last_success_at_ms = d.pop("last_success_at_ms", UNSET)
+
+        last_change_applied_at_ms = d.pop("last_change_applied_at_ms", UNSET)
+
+        updated_at_ms = d.pop("updated_at_ms", UNSET)
+
+        replication_source_status = cls(
+            source_kind=source_kind,
+            external_table=external_table,
+            cutover_mode=cutover_mode,
+            slot_name=slot_name,
+            publication_name=publication_name,
+            phase=phase,
+            checkpoint=checkpoint,
+            snapshot_offset=snapshot_offset,
+            prepared_checkpoint=prepared_checkpoint,
+            stream_checkpoint=stream_checkpoint,
+            last_error=last_error,
+            failure_class=failure_class,
+            lag_records=lag_records,
+            lag_millis=lag_millis,
+            consecutive_failures=consecutive_failures,
+            last_source_commit_at_ms=last_source_commit_at_ms,
+            last_success_at_ms=last_success_at_ms,
+            last_change_applied_at_ms=last_change_applied_at_ms,
+            updated_at_ms=updated_at_ms,
+        )
+
+        replication_source_status.additional_properties = d
+        return replication_source_status
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

antfly/client_generated/models/replication_transform_op.py

@@ -0,0 +1,87 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="ReplicationTransformOp")
+
+
+@_attrs_define
+class ReplicationTransformOp:
+    """
+    Attributes:
+        op (str): Transform operation. Standard ops: `$set`, `$unset`, `$inc`, `$push`, `$pull`,
+            `$addToSet`, `$pop`, `$mul`, `$min`, `$max`, `$currentDate`, `$rename`.
+            Replication-specific: `$merge` (flatten JSONB into top-level fields),
+            `$delete_document` (delete entire Antfly doc, `on_delete` only).
+             Example: $set.
+        path (str | Unset): Antfly document field path. Required for `$set`, `$unset`, etc. Example: email.
+        value (Any | Unset): Value for the operation. Can be a literal (string, number, boolean)
+            or a `{{column}}` reference to a PG column value. Use `{{col.key}}` to
+            navigate into decoded JSONB columns.
+    """
+
+    op: str
+    path: str | Unset = UNSET
+    value: Any | Unset = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        op = self.op
+
+        path = self.path
+
+        value = self.value
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "op": op,
+            }
+        )
+        if path is not UNSET:
+            field_dict["path"] = path
+        if value is not UNSET:
+            field_dict["value"] = value
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        op = d.pop("op")
+
+        path = d.pop("path", UNSET)
+
+        value = d.pop("value", UNSET)
+
+        replication_transform_op = cls(
+            op=op,
+            path=path,
+            value=value,
+        )
+
+        replication_transform_op.additional_properties = d
+        return replication_transform_op
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

antfly/client_generated/models/reranker_config.py

@@ -0,0 +1,86 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..models.reranker_provider import RerankerProvider
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="RerankerConfig")
+
+
+@_attrs_define
+class RerankerConfig:
+    """A unified configuration for a reranking provider.
+
+    Example:
+        {'provider': 'ollama', 'model': 'dengcao/Qwen3-Reranker-0.6B:F16', 'field': 'content'}
+
+    Attributes:
+        provider (RerankerProvider): The reranking provider to use.
+        field (str | Unset): Field name to extract from documents for reranking.
+        template (str | Unset): Handlebars template to render document text for reranking.
+    """
+
+    provider: RerankerProvider
+    field: str | Unset = UNSET
+    template: str | Unset = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        provider = self.provider.value
+
+        field = self.field
+
+        template = self.template
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "provider": provider,
+            }
+        )
+        if field is not UNSET:
+            field_dict["field"] = field
+        if template is not UNSET:
+            field_dict["template"] = template
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        provider = RerankerProvider(d.pop("provider"))
+
+        field = d.pop("field", UNSET)
+
+        template = d.pop("template", UNSET)
+
+        reranker_config = cls(
+            provider=provider,
+            field=field,
+            template=template,
+        )
+
+        reranker_config.additional_properties = d
+        return reranker_config
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

antfly/client_generated/models/reranker_profile.py

@@ -0,0 +1,80 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="RerankerProfile")
+
+
+@_attrs_define
+class RerankerProfile:
+    """Reranking execution statistics.
+
+    Attributes:
+        model (str | Unset): Reranker model that was used.
+        documents_reranked (int | Unset): Number of documents that were reranked.
+        duration_ms (int | Unset): Time spent reranking in milliseconds.
+    """
+
+    model: str | Unset = UNSET
+    documents_reranked: int | Unset = UNSET
+    duration_ms: int | Unset = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        model = self.model
+
+        documents_reranked = self.documents_reranked
+
+        duration_ms = self.duration_ms
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if model is not UNSET:
+            field_dict["model"] = model
+        if documents_reranked is not UNSET:
+            field_dict["documents_reranked"] = documents_reranked
+        if duration_ms is not UNSET:
+            field_dict["duration_ms"] = duration_ms
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        model = d.pop("model", UNSET)
+
+        documents_reranked = d.pop("documents_reranked", UNSET)
+
+        duration_ms = d.pop("duration_ms", UNSET)
+
+        reranker_profile = cls(
+            model=model,
+            documents_reranked=documents_reranked,
+            duration_ms=duration_ms,
+        )
+
+        reranker_profile.additional_properties = d
+        return reranker_profile
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

antfly/client_generated/models/reranker_provider.py

@@ -0,0 +1,11 @@
+from enum import Enum
+
+
+class RerankerProvider(str, Enum):
+    ANTFLY = "antfly"
+    COHERE = "cohere"
+    OLLAMA = "ollama"
+    VERTEX = "vertex"
+
+    def __str__(self) -> str:
+        return str(self.value)

antfly/client_generated/models/resource_type.py

@@ -0,0 +1,10 @@
+from enum import Enum
+
+
+class ResourceType(str, Enum):
+    TABLE = "table"
+    USER = "user"
+    VALUE_2 = "*"
+
+    def __str__(self) -> str:
+        return str(self.value)