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

antfly/client_generated/models/tree_search_config.py

@@ -0,0 +1,97 @@
+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="TreeSearchConfig")
+
+
+@_attrs_define
+class TreeSearchConfig:
+    """Configuration for tree search strategy. Tree search navigates hierarchical
+    document structures by evaluating summaries at each level.
+
+        Attributes:
+            index (str): Name of the graph index to use for tree navigation Example: doc_hierarchy.
+            start_nodes (str | Unset): Starting nodes for tree search:
+                - "$roots" - Query for root nodes (nodes with no parents)
+                - Comma-separated explicit node IDs
+                When omitted and combined with a QueryRequest in a RetrievalQueryRequest,
+                the query results are used as start nodes.
+                 Example: $roots.
+            max_depth (int | Unset): Maximum depth to traverse in the tree Default: 5.
+            beam_width (int | Unset): Number of branches to explore at each level Default: 3.
+    """
+
+    index: str
+    start_nodes: str | Unset = UNSET
+    max_depth: int | Unset = 5
+    beam_width: int | Unset = 3
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        index = self.index
+
+        start_nodes = self.start_nodes
+
+        max_depth = self.max_depth
+
+        beam_width = self.beam_width
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "index": index,
+            }
+        )
+        if start_nodes is not UNSET:
+            field_dict["start_nodes"] = start_nodes
+        if max_depth is not UNSET:
+            field_dict["max_depth"] = max_depth
+        if beam_width is not UNSET:
+            field_dict["beam_width"] = beam_width
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        index = d.pop("index")
+
+        start_nodes = d.pop("start_nodes", UNSET)
+
+        max_depth = d.pop("max_depth", UNSET)
+
+        beam_width = d.pop("beam_width", UNSET)
+
+        tree_search_config = cls(
+            index=index,
+            start_nodes=start_nodes,
+            max_depth=max_depth,
+            beam_width=beam_width,
+        )
+
+        tree_search_config.additional_properties = d
+        return tree_search_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/update_password_request.py

@@ -0,0 +1,61 @@
+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
+
+T = TypeVar("T", bound="UpdatePasswordRequest")
+
+
+@_attrs_define
+class UpdatePasswordRequest:
+    """
+    Attributes:
+        new_password (str):  Example: newS3cr3tP@sswOrd.
+    """
+
+    new_password: str
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        new_password = self.new_password
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "new_password": new_password,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        new_password = d.pop("new_password")
+
+        update_password_request = cls(
+            new_password=new_password,
+        )
+
+        update_password_request.additional_properties = d
+        return update_password_request
+
+    @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/user.py

@@ -0,0 +1,110 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, TypeVar, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from ..models.user_metadata_type_0 import UserMetadataType0
+
+
+T = TypeVar("T", bound="User")
+
+
+@_attrs_define
+class User:
+    """
+    Attributes:
+        username (str):  Example: johndoe.
+        password_hash (str): Base64 encoded password hash. Exposing this is a security risk. Example: JGFyZ29uMm....
+        metadata (None | Unset | UserMetadataType0): Server-side auth metadata available to stored row-filter policies
+            through $auth metadata paths. Example: {'tenant_id': 'acme', 'department': 'eng'}.
+    """
+
+    username: str
+    password_hash: str
+    metadata: None | Unset | UserMetadataType0 = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        from ..models.user_metadata_type_0 import UserMetadataType0
+
+        username = self.username
+
+        password_hash = self.password_hash
+
+        metadata: dict[str, Any] | None | Unset
+        if isinstance(self.metadata, Unset):
+            metadata = UNSET
+        elif isinstance(self.metadata, UserMetadataType0):
+            metadata = self.metadata.to_dict()
+        else:
+            metadata = self.metadata
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "username": username,
+                "password_hash": password_hash,
+            }
+        )
+        if metadata is not UNSET:
+            field_dict["metadata"] = metadata
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.user_metadata_type_0 import UserMetadataType0
+
+        d = dict(src_dict)
+        username = d.pop("username")
+
+        password_hash = d.pop("password_hash")
+
+        def _parse_metadata(data: object) -> None | Unset | UserMetadataType0:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                metadata_type_0 = UserMetadataType0.from_dict(data)
+
+                return metadata_type_0
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            return cast(None | Unset | UserMetadataType0, data)
+
+        metadata = _parse_metadata(d.pop("metadata", UNSET))
+
+        user = cls(
+            username=username,
+            password_hash=password_hash,
+            metadata=metadata,
+        )
+
+        user.additional_properties = d
+        return user
+
+    @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/user_metadata_type_0.py

@@ -0,0 +1,52 @@
+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
+
+T = TypeVar("T", bound="UserMetadataType0")
+
+
+@_attrs_define
+class UserMetadataType0:
+    """Server-side auth metadata available to stored row-filter policies through $auth metadata paths.
+
+    Example:
+        {'tenant_id': 'acme', 'department': 'eng'}
+
+    """
+
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        user_metadata_type_0 = cls()
+
+        user_metadata_type_0.additional_properties = d
+        return user_metadata_type_0
+
+    @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/vertex_embedder_config.py

@@ -0,0 +1,115 @@
+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="VertexEmbedderConfig")
+
+
+@_attrs_define
+class VertexEmbedderConfig:
+    """Configuration for Google Cloud Vertex AI embedding models (enterprise-grade).
+
+    Uses Application Default Credentials (ADC) for authentication. Requires IAM role `roles/aiplatform.user`.
+
+    **Example Models:** gemini-embedding-001 (default, 3072 dims), multimodalembedding (images/audio/video)
+
+    **Docs:** https://cloud.google.com/vertex-ai/generative-ai/docs/embeddings/get-text-embeddings
+
+        Example:
+            {'provider': 'vertex', 'model': 'gemini-embedding-001', 'project_id': 'my-gcp-project', 'location': 'us-
+                central1', 'dimension': 3072}
+
+        Attributes:
+            model (str): The name of the Vertex AI embedding model to use. Default: 'gemini-embedding-001'. Example: gemini-
+                embedding-001.
+            project_id (str | Unset): Google Cloud project ID. Can also be set via GOOGLE_CLOUD_PROJECT environment
+                variable.
+            location (str | Unset): Google Cloud region for Vertex AI API (e.g., 'us-central1', 'europe-west1'). Can also be
+                set via GOOGLE_CLOUD_LOCATION. Defaults to 'us-central1'. Default: 'us-central1'.
+            credentials_path (str | Unset): Path to service account JSON key file. Alternative to ADC for non-GCP
+                environments.
+            dimension (int | Unset): The dimension of the embedding vector (768, 1536, or 3072 for gemini-embedding-001;
+                128-1408 for multimodalembedding). Default: 3072.
+    """
+
+    model: str = "gemini-embedding-001"
+    project_id: str | Unset = UNSET
+    location: str | Unset = "us-central1"
+    credentials_path: str | Unset = UNSET
+    dimension: int | Unset = 3072
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        model = self.model
+
+        project_id = self.project_id
+
+        location = self.location
+
+        credentials_path = self.credentials_path
+
+        dimension = self.dimension
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "model": model,
+            }
+        )
+        if project_id is not UNSET:
+            field_dict["project_id"] = project_id
+        if location is not UNSET:
+            field_dict["location"] = location
+        if credentials_path is not UNSET:
+            field_dict["credentials_path"] = credentials_path
+        if dimension is not UNSET:
+            field_dict["dimension"] = dimension
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        model = d.pop("model")
+
+        project_id = d.pop("project_id", UNSET)
+
+        location = d.pop("location", UNSET)
+
+        credentials_path = d.pop("credentials_path", UNSET)
+
+        dimension = d.pop("dimension", UNSET)
+
+        vertex_embedder_config = cls(
+            model=model,
+            project_id=project_id,
+            location=location,
+            credentials_path=credentials_path,
+            dimension=dimension,
+        )
+
+        vertex_embedder_config.additional_properties = d
+        return vertex_embedder_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/vertex_generator_config.py

@@ -0,0 +1,127 @@
+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="VertexGeneratorConfig")
+
+
+@_attrs_define
+class VertexGeneratorConfig:
+    """Configuration for Google Cloud Vertex AI generative models.
+
+    Attributes:
+        model (str): The name of the Vertex AI model to use. Default: 'gemini-2.5-flash'. Example: gemini-2.5-flash.
+        project_id (str | Unset): Google Cloud project ID.
+        location (str | Unset): Google Cloud region for Vertex AI API. Default: 'us-central1'.
+        credentials_path (str | Unset): Path to service account JSON key file.
+        temperature (float | Unset): Controls randomness in generation (0.0-2.0).
+        max_tokens (int | Unset): Maximum number of tokens to generate in the response.
+        top_p (float | Unset): Nucleus sampling parameter (0.0-1.0).
+        top_k (int | Unset): Top-k sampling parameter.
+    """
+
+    model: str = "gemini-2.5-flash"
+    project_id: str | Unset = UNSET
+    location: str | Unset = "us-central1"
+    credentials_path: str | Unset = UNSET
+    temperature: float | Unset = UNSET
+    max_tokens: int | Unset = UNSET
+    top_p: float | Unset = UNSET
+    top_k: int | Unset = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        model = self.model
+
+        project_id = self.project_id
+
+        location = self.location
+
+        credentials_path = self.credentials_path
+
+        temperature = self.temperature
+
+        max_tokens = self.max_tokens
+
+        top_p = self.top_p
+
+        top_k = self.top_k
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "model": model,
+            }
+        )
+        if project_id is not UNSET:
+            field_dict["project_id"] = project_id
+        if location is not UNSET:
+            field_dict["location"] = location
+        if credentials_path is not UNSET:
+            field_dict["credentials_path"] = credentials_path
+        if temperature is not UNSET:
+            field_dict["temperature"] = temperature
+        if max_tokens is not UNSET:
+            field_dict["max_tokens"] = max_tokens
+        if top_p is not UNSET:
+            field_dict["top_p"] = top_p
+        if top_k is not UNSET:
+            field_dict["top_k"] = top_k
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        model = d.pop("model")
+
+        project_id = d.pop("project_id", UNSET)
+
+        location = d.pop("location", UNSET)
+
+        credentials_path = d.pop("credentials_path", UNSET)
+
+        temperature = d.pop("temperature", UNSET)
+
+        max_tokens = d.pop("max_tokens", UNSET)
+
+        top_p = d.pop("top_p", UNSET)
+
+        top_k = d.pop("top_k", UNSET)
+
+        vertex_generator_config = cls(
+            model=model,
+            project_id=project_id,
+            location=location,
+            credentials_path=credentials_path,
+            temperature=temperature,
+            max_tokens=max_tokens,
+            top_p=top_p,
+            top_k=top_k,
+        )
+
+        vertex_generator_config.additional_properties = d
+        return vertex_generator_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/vertex_reranker_config.py

@@ -0,0 +1,108 @@
+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="VertexRerankerConfig")
+
+
+@_attrs_define
+class VertexRerankerConfig:
+    """Configuration for the Google Vertex AI Ranking API.
+
+    Uses Application Default Credentials (ADC) or explicit credentials path.
+
+    **Prerequisites:**
+    - Enable Discovery Engine API: `gcloud services enable discoveryengine.googleapis.com`
+    - Grant IAM role: `roles/discoveryengine.admin` (includes `discoveryengine.rankingConfigs.rank` permission)
+
+    **Models:** semantic-ranker-default@latest (default), semantic-ranker-fast-004
+
+    **Docs:** https://cloud.google.com/generative-ai-app-builder/docs/ranking
+
+    **IAM:** https://cloud.google.com/generative-ai-app-builder/docs/access-control
+
+        Example:
+            {'provider': 'vertex', 'model': 'semantic-ranker-default@latest', 'project_id': 'my-gcp-project'}
+
+        Attributes:
+            model (str): The ranking model to use. Default: 'semantic-ranker-default@latest'. Example: semantic-ranker-
+                default@latest.
+            project_id (str | Unset): Google Cloud project ID. Falls back to GOOGLE_CLOUD_PROJECT environment variable.
+            credentials_path (str | Unset): Path to service account JSON file. Falls back to GOOGLE_APPLICATION_CREDENTIALS
+                environment variable.
+            top_n (int | Unset): Maximum number of records to return. If not specified, returns all documents with scores.
+    """
+
+    model: str = "semantic-ranker-default@latest"
+    project_id: str | Unset = UNSET
+    credentials_path: str | Unset = UNSET
+    top_n: int | Unset = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        model = self.model
+
+        project_id = self.project_id
+
+        credentials_path = self.credentials_path
+
+        top_n = self.top_n
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "model": model,
+            }
+        )
+        if project_id is not UNSET:
+            field_dict["project_id"] = project_id
+        if credentials_path is not UNSET:
+            field_dict["credentials_path"] = credentials_path
+        if top_n is not UNSET:
+            field_dict["top_n"] = top_n
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        model = d.pop("model")
+
+        project_id = d.pop("project_id", UNSET)
+
+        credentials_path = d.pop("credentials_path", UNSET)
+
+        top_n = d.pop("top_n", UNSET)
+
+        vertex_reranker_config = cls(
+            model=model,
+            project_id=project_id,
+            credentials_path=credentials_path,
+            top_n=top_n,
+        )
+
+        vertex_reranker_config.additional_properties = d
+        return vertex_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