everyrow 0.1.0__py3-none-any.whl

everyrow/generated/api/default/whoami_whoami_get.py

@@ -0,0 +1,127 @@
+from http import HTTPStatus
+from typing import Any
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+from ...models.whoami_whoami_get_response_whoami_whoami_get import WhoamiWhoamiGetResponseWhoamiWhoamiGet
+from ...types import Response
+
+
+def _get_kwargs() -> dict[str, Any]:
+    _kwargs: dict[str, Any] = {
+        "method": "get",
+        "url": "/whoami",
+    }
+
+    return _kwargs
+
+
+def _parse_response(
+    *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> WhoamiWhoamiGetResponseWhoamiWhoamiGet | None:
+    if response.status_code == 200:
+        response_200 = WhoamiWhoamiGetResponseWhoamiWhoamiGet.from_dict(response.json())
+
+        return response_200
+
+    if client.raise_on_unexpected_status:
+        raise errors.UnexpectedStatus(response.status_code, response.content)
+    else:
+        return None
+
+
+def _build_response(
+    *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Response[WhoamiWhoamiGetResponseWhoamiWhoamiGet]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(client=client, response=response),
+    )
+
+
+def sync_detailed(
+    *,
+    client: AuthenticatedClient,
+) -> Response[WhoamiWhoamiGetResponseWhoamiWhoamiGet]:
+    """Whoami
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[WhoamiWhoamiGetResponseWhoamiWhoamiGet]
+    """
+
+    kwargs = _get_kwargs()
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+def sync(
+    *,
+    client: AuthenticatedClient,
+) -> WhoamiWhoamiGetResponseWhoamiWhoamiGet | None:
+    """Whoami
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        WhoamiWhoamiGetResponseWhoamiWhoamiGet
+    """
+
+    return sync_detailed(
+        client=client,
+    ).parsed
+
+
+async def asyncio_detailed(
+    *,
+    client: AuthenticatedClient,
+) -> Response[WhoamiWhoamiGetResponseWhoamiWhoamiGet]:
+    """Whoami
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[WhoamiWhoamiGetResponseWhoamiWhoamiGet]
+    """
+
+    kwargs = _get_kwargs()
+
+    response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio(
+    *,
+    client: AuthenticatedClient,
+) -> WhoamiWhoamiGetResponseWhoamiWhoamiGet | None:
+    """Whoami
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        WhoamiWhoamiGetResponseWhoamiWhoamiGet
+    """
+
+    return (
+        await asyncio_detailed(
+            client=client,
+        )
+    ).parsed

everyrow/generated/client.py

@@ -0,0 +1,268 @@
+import ssl
+from typing import Any
+
+import httpx
+from attrs import define, evolve, field
+
+
+@define
+class Client:
+    """A class for keeping track of data related to the API
+
+    The following are accepted as keyword arguments and will be used to construct httpx Clients internally:
+
+        ``base_url``: The base URL for the API, all requests are made to a relative path to this URL
+
+        ``cookies``: A dictionary of cookies to be sent with every request
+
+        ``headers``: A dictionary of headers to be sent with every request
+
+        ``timeout``: The maximum amount of a time a request can take. API functions will raise
+        httpx.TimeoutException if this is exceeded.
+
+        ``verify_ssl``: Whether or not to verify the SSL certificate of the API server. This should be True in production,
+        but can be set to False for testing purposes.
+
+        ``follow_redirects``: Whether or not to follow redirects. Default value is False.
+
+        ``httpx_args``: A dictionary of additional arguments to be passed to the ``httpx.Client`` and ``httpx.AsyncClient`` constructor.
+
+
+    Attributes:
+        raise_on_unexpected_status: Whether or not to raise an errors.UnexpectedStatus if the API returns a
+            status code that was not documented in the source OpenAPI document. Can also be provided as a keyword
+            argument to the constructor.
+    """
+
+    raise_on_unexpected_status: bool = field(default=False, kw_only=True)
+    _base_url: str = field(alias="base_url")
+    _cookies: dict[str, str] = field(factory=dict, kw_only=True, alias="cookies")
+    _headers: dict[str, str] = field(factory=dict, kw_only=True, alias="headers")
+    _timeout: httpx.Timeout | None = field(default=None, kw_only=True, alias="timeout")
+    _verify_ssl: str | bool | ssl.SSLContext = field(default=True, kw_only=True, alias="verify_ssl")
+    _follow_redirects: bool = field(default=False, kw_only=True, alias="follow_redirects")
+    _httpx_args: dict[str, Any] = field(factory=dict, kw_only=True, alias="httpx_args")
+    _client: httpx.Client | None = field(default=None, init=False)
+    _async_client: httpx.AsyncClient | None = field(default=None, init=False)
+
+    def with_headers(self, headers: dict[str, str]) -> "Client":
+        """Get a new client matching this one with additional headers"""
+        if self._client is not None:
+            self._client.headers.update(headers)
+        if self._async_client is not None:
+            self._async_client.headers.update(headers)
+        return evolve(self, headers={**self._headers, **headers})
+
+    def with_cookies(self, cookies: dict[str, str]) -> "Client":
+        """Get a new client matching this one with additional cookies"""
+        if self._client is not None:
+            self._client.cookies.update(cookies)
+        if self._async_client is not None:
+            self._async_client.cookies.update(cookies)
+        return evolve(self, cookies={**self._cookies, **cookies})
+
+    def with_timeout(self, timeout: httpx.Timeout) -> "Client":
+        """Get a new client matching this one with a new timeout configuration"""
+        if self._client is not None:
+            self._client.timeout = timeout
+        if self._async_client is not None:
+            self._async_client.timeout = timeout
+        return evolve(self, timeout=timeout)
+
+    def set_httpx_client(self, client: httpx.Client) -> "Client":
+        """Manually set the underlying httpx.Client
+
+        **NOTE**: This will override any other settings on the client, including cookies, headers, and timeout.
+        """
+        self._client = client
+        return self
+
+    def get_httpx_client(self) -> httpx.Client:
+        """Get the underlying httpx.Client, constructing a new one if not previously set"""
+        if self._client is None:
+            self._client = httpx.Client(
+                base_url=self._base_url,
+                cookies=self._cookies,
+                headers=self._headers,
+                timeout=self._timeout,
+                verify=self._verify_ssl,
+                follow_redirects=self._follow_redirects,
+                **self._httpx_args,
+            )
+        return self._client
+
+    def __enter__(self) -> "Client":
+        """Enter a context manager for self.client—you cannot enter twice (see httpx docs)"""
+        self.get_httpx_client().__enter__()
+        return self
+
+    def __exit__(self, *args: Any, **kwargs: Any) -> None:
+        """Exit a context manager for internal httpx.Client (see httpx docs)"""
+        self.get_httpx_client().__exit__(*args, **kwargs)
+
+    def set_async_httpx_client(self, async_client: httpx.AsyncClient) -> "Client":
+        """Manually set the underlying httpx.AsyncClient
+
+        **NOTE**: This will override any other settings on the client, including cookies, headers, and timeout.
+        """
+        self._async_client = async_client
+        return self
+
+    def get_async_httpx_client(self) -> httpx.AsyncClient:
+        """Get the underlying httpx.AsyncClient, constructing a new one if not previously set"""
+        if self._async_client is None:
+            self._async_client = httpx.AsyncClient(
+                base_url=self._base_url,
+                cookies=self._cookies,
+                headers=self._headers,
+                timeout=self._timeout,
+                verify=self._verify_ssl,
+                follow_redirects=self._follow_redirects,
+                **self._httpx_args,
+            )
+        return self._async_client
+
+    async def __aenter__(self) -> "Client":
+        """Enter a context manager for underlying httpx.AsyncClient—you cannot enter twice (see httpx docs)"""
+        await self.get_async_httpx_client().__aenter__()
+        return self
+
+    async def __aexit__(self, *args: Any, **kwargs: Any) -> None:
+        """Exit a context manager for underlying httpx.AsyncClient (see httpx docs)"""
+        await self.get_async_httpx_client().__aexit__(*args, **kwargs)
+
+
+@define
+class AuthenticatedClient:
+    """A Client which has been authenticated for use on secured endpoints
+
+    The following are accepted as keyword arguments and will be used to construct httpx Clients internally:
+
+        ``base_url``: The base URL for the API, all requests are made to a relative path to this URL
+
+        ``cookies``: A dictionary of cookies to be sent with every request
+
+        ``headers``: A dictionary of headers to be sent with every request
+
+        ``timeout``: The maximum amount of a time a request can take. API functions will raise
+        httpx.TimeoutException if this is exceeded.
+
+        ``verify_ssl``: Whether or not to verify the SSL certificate of the API server. This should be True in production,
+        but can be set to False for testing purposes.
+
+        ``follow_redirects``: Whether or not to follow redirects. Default value is False.
+
+        ``httpx_args``: A dictionary of additional arguments to be passed to the ``httpx.Client`` and ``httpx.AsyncClient`` constructor.
+
+
+    Attributes:
+        raise_on_unexpected_status: Whether or not to raise an errors.UnexpectedStatus if the API returns a
+            status code that was not documented in the source OpenAPI document. Can also be provided as a keyword
+            argument to the constructor.
+        token: The token to use for authentication
+        prefix: The prefix to use for the Authorization header
+        auth_header_name: The name of the Authorization header
+    """
+
+    raise_on_unexpected_status: bool = field(default=False, kw_only=True)
+    _base_url: str = field(alias="base_url")
+    _cookies: dict[str, str] = field(factory=dict, kw_only=True, alias="cookies")
+    _headers: dict[str, str] = field(factory=dict, kw_only=True, alias="headers")
+    _timeout: httpx.Timeout | None = field(default=None, kw_only=True, alias="timeout")
+    _verify_ssl: str | bool | ssl.SSLContext = field(default=True, kw_only=True, alias="verify_ssl")
+    _follow_redirects: bool = field(default=False, kw_only=True, alias="follow_redirects")
+    _httpx_args: dict[str, Any] = field(factory=dict, kw_only=True, alias="httpx_args")
+    _client: httpx.Client | None = field(default=None, init=False)
+    _async_client: httpx.AsyncClient | None = field(default=None, init=False)
+
+    token: str
+    prefix: str = "Bearer"
+    auth_header_name: str = "Authorization"
+
+    def with_headers(self, headers: dict[str, str]) -> "AuthenticatedClient":
+        """Get a new client matching this one with additional headers"""
+        if self._client is not None:
+            self._client.headers.update(headers)
+        if self._async_client is not None:
+            self._async_client.headers.update(headers)
+        return evolve(self, headers={**self._headers, **headers})
+
+    def with_cookies(self, cookies: dict[str, str]) -> "AuthenticatedClient":
+        """Get a new client matching this one with additional cookies"""
+        if self._client is not None:
+            self._client.cookies.update(cookies)
+        if self._async_client is not None:
+            self._async_client.cookies.update(cookies)
+        return evolve(self, cookies={**self._cookies, **cookies})
+
+    def with_timeout(self, timeout: httpx.Timeout) -> "AuthenticatedClient":
+        """Get a new client matching this one with a new timeout configuration"""
+        if self._client is not None:
+            self._client.timeout = timeout
+        if self._async_client is not None:
+            self._async_client.timeout = timeout
+        return evolve(self, timeout=timeout)
+
+    def set_httpx_client(self, client: httpx.Client) -> "AuthenticatedClient":
+        """Manually set the underlying httpx.Client
+
+        **NOTE**: This will override any other settings on the client, including cookies, headers, and timeout.
+        """
+        self._client = client
+        return self
+
+    def get_httpx_client(self) -> httpx.Client:
+        """Get the underlying httpx.Client, constructing a new one if not previously set"""
+        if self._client is None:
+            self._headers[self.auth_header_name] = f"{self.prefix} {self.token}" if self.prefix else self.token
+            self._client = httpx.Client(
+                base_url=self._base_url,
+                cookies=self._cookies,
+                headers=self._headers,
+                timeout=self._timeout,
+                verify=self._verify_ssl,
+                follow_redirects=self._follow_redirects,
+                **self._httpx_args,
+            )
+        return self._client
+
+    def __enter__(self) -> "AuthenticatedClient":
+        """Enter a context manager for self.client—you cannot enter twice (see httpx docs)"""
+        self.get_httpx_client().__enter__()
+        return self
+
+    def __exit__(self, *args: Any, **kwargs: Any) -> None:
+        """Exit a context manager for internal httpx.Client (see httpx docs)"""
+        self.get_httpx_client().__exit__(*args, **kwargs)
+
+    def set_async_httpx_client(self, async_client: httpx.AsyncClient) -> "AuthenticatedClient":
+        """Manually set the underlying httpx.AsyncClient
+
+        **NOTE**: This will override any other settings on the client, including cookies, headers, and timeout.
+        """
+        self._async_client = async_client
+        return self
+
+    def get_async_httpx_client(self) -> httpx.AsyncClient:
+        """Get the underlying httpx.AsyncClient, constructing a new one if not previously set"""
+        if self._async_client is None:
+            self._headers[self.auth_header_name] = f"{self.prefix} {self.token}" if self.prefix else self.token
+            self._async_client = httpx.AsyncClient(
+                base_url=self._base_url,
+                cookies=self._cookies,
+                headers=self._headers,
+                timeout=self._timeout,
+                verify=self._verify_ssl,
+                follow_redirects=self._follow_redirects,
+                **self._httpx_args,
+            )
+        return self._async_client
+
+    async def __aenter__(self) -> "AuthenticatedClient":
+        """Enter a context manager for underlying httpx.AsyncClient—you cannot enter twice (see httpx docs)"""
+        await self.get_async_httpx_client().__aenter__()
+        return self
+
+    async def __aexit__(self, *args: Any, **kwargs: Any) -> None:
+        """Exit a context manager for underlying httpx.AsyncClient (see httpx docs)"""
+        await self.get_async_httpx_client().__aexit__(*args, **kwargs)

everyrow/generated/errors.py

@@ -0,0 +1,16 @@
+"""Contains shared errors types that can be raised from API functions"""
+
+
+class UnexpectedStatus(Exception):
+    """Raised by api functions when the response status an undocumented status and Client.raise_on_unexpected_status is True"""
+
+    def __init__(self, status_code: int, content: bytes):
+        self.status_code = status_code
+        self.content = content
+
+        super().__init__(
+            f"Unexpected status code: {status_code}\n\nResponse content:\n{content.decode(errors='ignore')}"
+        )
+
+
+__all__ = ["UnexpectedStatus"]

everyrow/generated/models/__init__.py

@@ -0,0 +1,281 @@
+"""Contains all the data models used in inputs/outputs"""
+
+from .agent_improvement_instruction import AgentImprovementInstruction
+from .agent_query_params import AgentQueryParams
+from .agent_query_params_system_prompt_kind_type_0 import AgentQueryParamsSystemPromptKindType0
+from .agent_task_args import AgentTaskArgs
+from .agent_task_args_processing_mode import AgentTaskArgsProcessingMode
+from .allowed_suggestions import AllowedSuggestions
+from .api_key_info import APIKeyInfo
+from .artifact_changed_payload import ArtifactChangedPayload
+from .artifact_group_record import ArtifactGroupRecord
+from .artifact_group_record_metadata_type_0 import ArtifactGroupRecordMetadataType0
+from .artifact_group_record_trace_mapping_type_0 import ArtifactGroupRecordTraceMappingType0
+from .artifact_status import ArtifactStatus
+from .auto_cohort_conversation_message import AutoCohortConversationMessage
+from .aux_data import AuxData
+from .aux_data_source_bank import AuxDataSourceBank
+from .chat_completion_message_tool_call import ChatCompletionMessageToolCall
+from .chat_message_metadata import ChatMessageMetadata
+from .concatenate_query_params import ConcatenateQueryParams
+from .concatenate_request import ConcatenateRequest
+from .continue_reason import ContinueReason
+from .continue_task_request import ContinueTaskRequest
+from .controller_improvement_round import ControllerImprovementRound
+from .conversation_changed_payload import ConversationChangedPayload
+from .copy_artifacts_request import CopyArtifactsRequest
+from .copy_artifacts_response import CopyArtifactsResponse
+from .copy_workflow_request import CopyWorkflowRequest
+from .copy_workflow_response import CopyWorkflowResponse
+from .create_api_key_request import CreateAPIKeyRequest
+from .create_api_key_response import CreateAPIKeyResponse
+from .create_group_query_params import CreateGroupQueryParams
+from .create_group_request import CreateGroupRequest
+from .create_query_params import CreateQueryParams
+from .create_request import CreateRequest
+from .create_session_request import CreateSessionRequest
+from .create_session_response import CreateSessionResponse
+from .create_workflow_from_artifact_request import CreateWorkflowFromArtifactRequest
+from .create_workflow_from_artifact_response import CreateWorkflowFromArtifactResponse
+from .data_frame_method import DataFrameMethod
+from .date_cutoffs import DateCutoffs
+from .dedupe_mode import DedupeMode
+from .dedupe_query_params import DedupeQueryParams
+from .dedupe_request_params import DedupeRequestParams
+from .deep_merge_public_params import DeepMergePublicParams
+from .deep_merge_request import DeepMergeRequest
+from .deep_rank_public_params import DeepRankPublicParams
+from .deep_rank_request import DeepRankRequest
+from .deep_screen_public_params import DeepScreenPublicParams
+from .deep_screen_request import DeepScreenRequest
+from .derive_expression import DeriveExpression
+from .derive_query_params import DeriveQueryParams
+from .derive_request import DeriveRequest
+from .document_query_tool import DocumentQueryTool
+from .drop_columns_query_params import DropColumnsQueryParams
+from .drop_columns_request import DropColumnsRequest
+from .embedding_models import EmbeddingModels
+from .event_type import EventType
+from .execution_metadata import ExecutionMetadata
+from .export_request import ExportRequest
+from .export_request_token_data import ExportRequestTokenData
+from .export_to_google_sheets_export_post_response_export_to_google_sheets_export_post import (
+    ExportToGoogleSheetsExportPostResponseExportToGoogleSheetsExportPost,
+)
+from .filter_query_params import FilterQueryParams
+from .filter_request import FilterRequest
+from .flatten_query_params import FlattenQueryParams
+from .flatten_request import FlattenRequest
+from .generate_feedback_request import GenerateFeedbackRequest
+from .group_by_query_params import GroupByQueryParams
+from .group_by_request import GroupByRequest
+from .healthz_healthz_get_response_healthz_healthz_get import HealthzHealthzGetResponseHealthzHealthzGet
+from .http_validation_error import HTTPValidationError
+from .image_chat_content_part import ImageChatContentPart
+from .image_chat_content_part_image_url import ImageChatContentPartImageUrl
+from .import_from_google_sheets_import_post_response_import_from_google_sheets_import_post import (
+    ImportFromGoogleSheetsImportPostResponseImportFromGoogleSheetsImportPost,
+)
+from .import_request import ImportRequest
+from .import_request_token_data import ImportRequestTokenData
+from .insufficient_balance_error import InsufficientBalanceError
+from .join_query_params import JoinQueryParams
+from .join_request import JoinRequest
+from .llm_enum import LLMEnum
+from .map_agent_request_params import MapAgentRequestParams
+from .map_multi_agent_request_params import MapMultiAgentRequestParams
+from .message_created_payload import MessageCreatedPayload
+from .multi_agent_effort_level import MultiAgentEffortLevel
+from .multi_agent_query_params import MultiAgentQueryParams
+from .multi_modal_chat_message import MultiModalChatMessage
+from .multi_modal_chat_message_role import MultiModalChatMessageRole
+from .preview_metadata import PreviewMetadata
+from .processing_mode import ProcessingMode
+from .progress_status import ProgressStatus
+from .queue_stats import QueueStats
+from .reduce_agent_request_params import ReduceAgentRequestParams
+from .reduce_multi_agent_request_params import ReduceMultiAgentRequestParams
+from .resource_estimation_response import ResourceEstimationResponse
+from .response_schema_type import ResponseSchemaType
+from .revoke_api_key_response import RevokeAPIKeyResponse
+from .rollback_to_message_request import RollbackToMessageRequest
+from .rollback_to_message_response import RollbackToMessageResponse
+from .session_changed_payload import SessionChangedPayload
+from .simple_chat_message import SimpleChatMessage
+from .simple_chat_message_role import SimpleChatMessageRole
+from .simple_chat_message_with_tool_calls import SimpleChatMessageWithToolCalls
+from .source_database_entry import SourceDatabaseEntry
+from .standalone_artifact_record import StandaloneArtifactRecord
+from .standalone_artifact_record_metadata_type_0 import StandaloneArtifactRecordMetadataType0
+from .standalone_artifact_record_trace_mapping_type_0 import StandaloneArtifactRecordTraceMappingType0
+from .status_count import StatusCount
+from .status_count_status import StatusCountStatus
+from .submit_chat_task_body import SubmitChatTaskBody
+from .submit_chat_task_body_selected_task_type_type_0 import SubmitChatTaskBodySelectedTaskTypeType0
+from .submit_task_body import SubmitTaskBody
+from .task_changed_payload import TaskChangedPayload
+from .task_effort import TaskEffort
+from .task_id_request import TaskIdRequest
+from .task_insert import TaskInsert
+from .task_insert_query_params import TaskInsertQueryParams
+from .task_metadata import TaskMetadata
+from .task_metadata_cols_to_rename_type_0 import TaskMetadataColsToRenameType0
+from .task_response import TaskResponse
+from .task_status import TaskStatus
+from .task_status_response import TaskStatusResponse
+from .task_type import TaskType
+from .text_chat_content_part import TextChatContentPart
+from .tool_response_message import ToolResponseMessage
+from .toolkit_constants import ToolkitConstants
+from .trace_changed_payload import TraceChangedPayload
+from .trace_info import TraceInfo
+from .trigger_workflow_execution_request import TriggerWorkflowExecutionRequest
+from .trigger_workflow_execution_request_task_params import TriggerWorkflowExecutionRequestTaskParams
+from .trigger_workflow_execution_request_task_params_additional_property import (
+    TriggerWorkflowExecutionRequestTaskParamsAdditionalProperty,
+)
+from .trigger_workflow_execution_response import TriggerWorkflowExecutionResponse
+from .upload_csv_payload import UploadCsvPayload
+from .upload_csv_query_params import UploadCsvQueryParams
+from .usage_response import UsageResponse
+from .validation_error import ValidationError
+from .whoami_whoami_get_response_whoami_whoami_get import WhoamiWhoamiGetResponseWhoamiWhoamiGet
+from .workflow_leaf_node_input import WorkflowLeafNodeInput
+
+__all__ = (
+    "AgentImprovementInstruction",
+    "AgentQueryParams",
+    "AgentQueryParamsSystemPromptKindType0",
+    "AgentTaskArgs",
+    "AgentTaskArgsProcessingMode",
+    "AllowedSuggestions",
+    "APIKeyInfo",
+    "ArtifactChangedPayload",
+    "ArtifactGroupRecord",
+    "ArtifactGroupRecordMetadataType0",
+    "ArtifactGroupRecordTraceMappingType0",
+    "ArtifactStatus",
+    "AutoCohortConversationMessage",
+    "AuxData",
+    "AuxDataSourceBank",
+    "ChatCompletionMessageToolCall",
+    "ChatMessageMetadata",
+    "ConcatenateQueryParams",
+    "ConcatenateRequest",
+    "ContinueReason",
+    "ContinueTaskRequest",
+    "ControllerImprovementRound",
+    "ConversationChangedPayload",
+    "CopyArtifactsRequest",
+    "CopyArtifactsResponse",
+    "CopyWorkflowRequest",
+    "CopyWorkflowResponse",
+    "CreateAPIKeyRequest",
+    "CreateAPIKeyResponse",
+    "CreateGroupQueryParams",
+    "CreateGroupRequest",
+    "CreateQueryParams",
+    "CreateRequest",
+    "CreateSessionRequest",
+    "CreateSessionResponse",
+    "CreateWorkflowFromArtifactRequest",
+    "CreateWorkflowFromArtifactResponse",
+    "DataFrameMethod",
+    "DateCutoffs",
+    "DedupeMode",
+    "DedupeQueryParams",
+    "DedupeRequestParams",
+    "DeepMergePublicParams",
+    "DeepMergeRequest",
+    "DeepRankPublicParams",
+    "DeepRankRequest",
+    "DeepScreenPublicParams",
+    "DeepScreenRequest",
+    "DeriveExpression",
+    "DeriveQueryParams",
+    "DeriveRequest",
+    "DocumentQueryTool",
+    "DropColumnsQueryParams",
+    "DropColumnsRequest",
+    "EmbeddingModels",
+    "EventType",
+    "ExecutionMetadata",
+    "ExportRequest",
+    "ExportRequestTokenData",
+    "ExportToGoogleSheetsExportPostResponseExportToGoogleSheetsExportPost",
+    "FilterQueryParams",
+    "FilterRequest",
+    "FlattenQueryParams",
+    "FlattenRequest",
+    "GenerateFeedbackRequest",
+    "GroupByQueryParams",
+    "GroupByRequest",
+    "HealthzHealthzGetResponseHealthzHealthzGet",
+    "HTTPValidationError",
+    "ImageChatContentPart",
+    "ImageChatContentPartImageUrl",
+    "ImportFromGoogleSheetsImportPostResponseImportFromGoogleSheetsImportPost",
+    "ImportRequest",
+    "ImportRequestTokenData",
+    "InsufficientBalanceError",
+    "JoinQueryParams",
+    "JoinRequest",
+    "LLMEnum",
+    "MapAgentRequestParams",
+    "MapMultiAgentRequestParams",
+    "MessageCreatedPayload",
+    "MultiAgentEffortLevel",
+    "MultiAgentQueryParams",
+    "MultiModalChatMessage",
+    "MultiModalChatMessageRole",
+    "PreviewMetadata",
+    "ProcessingMode",
+    "ProgressStatus",
+    "QueueStats",
+    "ReduceAgentRequestParams",
+    "ReduceMultiAgentRequestParams",
+    "ResourceEstimationResponse",
+    "ResponseSchemaType",
+    "RevokeAPIKeyResponse",
+    "RollbackToMessageRequest",
+    "RollbackToMessageResponse",
+    "SessionChangedPayload",
+    "SimpleChatMessage",
+    "SimpleChatMessageRole",
+    "SimpleChatMessageWithToolCalls",
+    "SourceDatabaseEntry",
+    "StandaloneArtifactRecord",
+    "StandaloneArtifactRecordMetadataType0",
+    "StandaloneArtifactRecordTraceMappingType0",
+    "StatusCount",
+    "StatusCountStatus",
+    "SubmitChatTaskBody",
+    "SubmitChatTaskBodySelectedTaskTypeType0",
+    "SubmitTaskBody",
+    "TaskChangedPayload",
+    "TaskEffort",
+    "TaskIdRequest",
+    "TaskInsert",
+    "TaskInsertQueryParams",
+    "TaskMetadata",
+    "TaskMetadataColsToRenameType0",
+    "TaskResponse",
+    "TaskStatus",
+    "TaskStatusResponse",
+    "TaskType",
+    "TextChatContentPart",
+    "ToolkitConstants",
+    "ToolResponseMessage",
+    "TraceChangedPayload",
+    "TraceInfo",
+    "TriggerWorkflowExecutionRequest",
+    "TriggerWorkflowExecutionRequestTaskParams",
+    "TriggerWorkflowExecutionRequestTaskParamsAdditionalProperty",
+    "TriggerWorkflowExecutionResponse",
+    "UploadCsvPayload",
+    "UploadCsvQueryParams",
+    "UsageResponse",
+    "ValidationError",
+    "WhoamiWhoamiGetResponseWhoamiWhoamiGet",
+    "WorkflowLeafNodeInput",
+)