dexa-ai-sdk 0.3.2__py3-none-any.whl

dexa_ai_sdk/__init__.py

@@ -0,0 +1,197 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+
+import typing
+from importlib import import_module
+
+if typing.TYPE_CHECKING:
+    from .types import (
+        AudioInput,
+        AudioInputType,
+        Choice,
+        ChoiceFinishReason,
+        ContentPart,
+        ContentPartLanguage,
+        ContentPartType,
+        DecideResponse,
+        DecideResult,
+        DecideTask,
+        DecideTaskTargetLanguage,
+        ErrorResponse,
+        ErrorResponseError,
+        ExtractResponse,
+        ExtractTask,
+        ExtractTaskTargetLanguage,
+        ImageInput,
+        ImageInputType,
+        ImageUrl,
+        ImageUrlDetail,
+        Message,
+        MessageContent,
+        MessageLanguage,
+        MessageRole,
+        ResponseMessage,
+        ResponseMessageLanguage,
+        ResponseMessageRole,
+        RunTasksRequest,
+        RunTasksRequestOutputLanguage,
+        RunTasksRequestResponseFormat,
+        RunTasksRequestResponseFormatType,
+        RunTasksRequestTargetLanguage,
+        RunTasksResponse,
+        Task,
+        Task_Decide,
+        Task_Extract,
+        Usage,
+    )
+    from .errors import BadRequestError, InternalServerError, UnprocessableEntityError
+    from . import tasks
+    from ._default_clients import DefaultAioHttpClient, DefaultAsyncHttpxClient
+    from .client import AsyncDexaAiSdk, DexaAiSdk
+    from .environment import DexaAiSdkEnvironment
+    from .tasks import (
+        DecideRequestOutputLanguage,
+        DecideRequestResponseFormat,
+        DecideRequestResponseFormatType,
+        DecideRequestTargetLanguage,
+        ExtractRequestOutputLanguage,
+        ExtractRequestResponseFormat,
+        ExtractRequestResponseFormatType,
+        ExtractRequestTargetLanguage,
+    )
+_dynamic_imports: typing.Dict[str, str] = {
+    "AsyncDexaAiSdk": ".client",
+    "AudioInput": ".types",
+    "AudioInputType": ".types",
+    "BadRequestError": ".errors",
+    "Choice": ".types",
+    "ChoiceFinishReason": ".types",
+    "ContentPart": ".types",
+    "ContentPartLanguage": ".types",
+    "ContentPartType": ".types",
+    "DecideRequestOutputLanguage": ".tasks",
+    "DecideRequestResponseFormat": ".tasks",
+    "DecideRequestResponseFormatType": ".tasks",
+    "DecideRequestTargetLanguage": ".tasks",
+    "DecideResponse": ".types",
+    "DecideResult": ".types",
+    "DecideTask": ".types",
+    "DecideTaskTargetLanguage": ".types",
+    "DefaultAioHttpClient": "._default_clients",
+    "DefaultAsyncHttpxClient": "._default_clients",
+    "DexaAiSdk": ".client",
+    "DexaAiSdkEnvironment": ".environment",
+    "ErrorResponse": ".types",
+    "ErrorResponseError": ".types",
+    "ExtractRequestOutputLanguage": ".tasks",
+    "ExtractRequestResponseFormat": ".tasks",
+    "ExtractRequestResponseFormatType": ".tasks",
+    "ExtractRequestTargetLanguage": ".tasks",
+    "ExtractResponse": ".types",
+    "ExtractTask": ".types",
+    "ExtractTaskTargetLanguage": ".types",
+    "ImageInput": ".types",
+    "ImageInputType": ".types",
+    "ImageUrl": ".types",
+    "ImageUrlDetail": ".types",
+    "InternalServerError": ".errors",
+    "Message": ".types",
+    "MessageContent": ".types",
+    "MessageLanguage": ".types",
+    "MessageRole": ".types",
+    "ResponseMessage": ".types",
+    "ResponseMessageLanguage": ".types",
+    "ResponseMessageRole": ".types",
+    "RunTasksRequest": ".types",
+    "RunTasksRequestOutputLanguage": ".types",
+    "RunTasksRequestResponseFormat": ".types",
+    "RunTasksRequestResponseFormatType": ".types",
+    "RunTasksRequestTargetLanguage": ".types",
+    "RunTasksResponse": ".types",
+    "Task": ".types",
+    "Task_Decide": ".types",
+    "Task_Extract": ".types",
+    "UnprocessableEntityError": ".errors",
+    "Usage": ".types",
+    "tasks": ".tasks",
+}
+
+
+def __getattr__(attr_name: str) -> typing.Any:
+    module_name = _dynamic_imports.get(attr_name)
+    if module_name is None:
+        raise AttributeError(f"No {attr_name} found in _dynamic_imports for module name -> {__name__}")
+    try:
+        module = import_module(module_name, __package__)
+        if module_name == f".{attr_name}":
+            return module
+        else:
+            return getattr(module, attr_name)
+    except ImportError as e:
+        raise ImportError(f"Failed to import {attr_name} from {module_name}: {e}") from e
+    except AttributeError as e:
+        raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
+
+
+def __dir__():
+    lazy_attrs = list(_dynamic_imports.keys())
+    return sorted(lazy_attrs)
+
+
+__all__ = [
+    "AsyncDexaAiSdk",
+    "AudioInput",
+    "AudioInputType",
+    "BadRequestError",
+    "Choice",
+    "ChoiceFinishReason",
+    "ContentPart",
+    "ContentPartLanguage",
+    "ContentPartType",
+    "DecideRequestOutputLanguage",
+    "DecideRequestResponseFormat",
+    "DecideRequestResponseFormatType",
+    "DecideRequestTargetLanguage",
+    "DecideResponse",
+    "DecideResult",
+    "DecideTask",
+    "DecideTaskTargetLanguage",
+    "DefaultAioHttpClient",
+    "DefaultAsyncHttpxClient",
+    "DexaAiSdk",
+    "DexaAiSdkEnvironment",
+    "ErrorResponse",
+    "ErrorResponseError",
+    "ExtractRequestOutputLanguage",
+    "ExtractRequestResponseFormat",
+    "ExtractRequestResponseFormatType",
+    "ExtractRequestTargetLanguage",
+    "ExtractResponse",
+    "ExtractTask",
+    "ExtractTaskTargetLanguage",
+    "ImageInput",
+    "ImageInputType",
+    "ImageUrl",
+    "ImageUrlDetail",
+    "InternalServerError",
+    "Message",
+    "MessageContent",
+    "MessageLanguage",
+    "MessageRole",
+    "ResponseMessage",
+    "ResponseMessageLanguage",
+    "ResponseMessageRole",
+    "RunTasksRequest",
+    "RunTasksRequestOutputLanguage",
+    "RunTasksRequestResponseFormat",
+    "RunTasksRequestResponseFormatType",
+    "RunTasksRequestTargetLanguage",
+    "RunTasksResponse",
+    "Task",
+    "Task_Decide",
+    "Task_Extract",
+    "UnprocessableEntityError",
+    "Usage",
+    "tasks",
+]

dexa_ai_sdk/_default_clients.py

@@ -0,0 +1,33 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import httpx
+
+SDK_DEFAULT_TIMEOUT = 60
+
+try:
+    import httpx_aiohttp  # type: ignore[import-not-found]
+except ImportError:
+
+    class DefaultAioHttpClient(httpx.AsyncClient):  # type: ignore
+        def __init__(self, **kwargs: typing.Any) -> None:
+            raise RuntimeError(
+                "To use the aiohttp client, install the aiohttp extra: "
+                "pip install dsainvg_decide_extractor_sdk[aiohttp]"
+            )
+
+else:
+
+    class DefaultAioHttpClient(httpx_aiohttp.HttpxAiohttpClient):  # type: ignore
+        def __init__(self, **kwargs: typing.Any) -> None:
+            kwargs.setdefault("timeout", SDK_DEFAULT_TIMEOUT)
+            kwargs.setdefault("follow_redirects", True)
+            super().__init__(**kwargs)
+
+
+class DefaultAsyncHttpxClient(httpx.AsyncClient):
+    def __init__(self, **kwargs: typing.Any) -> None:
+        kwargs.setdefault("timeout", SDK_DEFAULT_TIMEOUT)
+        kwargs.setdefault("follow_redirects", True)
+        super().__init__(**kwargs)

dexa_ai_sdk/client.py

@@ -0,0 +1,218 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+
+import typing
+
+import httpx
+from .core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from .core.logging import LogConfig, Logger
+from .environment import DexaAiSdkEnvironment
+
+if typing.TYPE_CHECKING:
+    from .tasks.client import AsyncTasksClient, TasksClient
+
+
+class DexaAiSdk:
+    """
+    Use this class to access the different functions within the SDK. You can instantiate any number of clients with different configuration that will propagate to these functions.
+
+    Parameters
+    ----------
+    base_url : typing.Optional[str]
+        The base url to use for requests from the client.
+
+    environment : DexaAiSdkEnvironment
+        The environment to use for requests from the client. from .environment import DexaAiSdkEnvironment
+
+
+
+        Defaults to DexaAiSdkEnvironment.DEFAULT
+
+
+
+    headers : typing.Optional[typing.Dict[str, str]]
+        Additional headers to send with every request.
+
+    timeout : typing.Optional[float]
+        The timeout to be used, in seconds, for requests. By default the timeout is 60 seconds, unless a custom httpx client is used, in which case this default is not enforced.
+
+    max_retries : typing.Optional[int]
+        The default maximum number of retries for failed requests. Defaults to 2. Per-request `max_retries` in `request_options` takes precedence over this value.
+
+    stream_reconnection_enabled : typing.Optional[bool]
+        Whether to automatically reconnect on stream disconnection for resumable streaming endpoints. Defaults to True. Per-request `stream_reconnection_enabled` in `request_options` takes precedence over this value.
+
+    max_stream_reconnection_attempts : typing.Optional[int]
+        The maximum number of reconnection attempts for resumable streaming endpoints. Defaults to no limit. Per-request `max_stream_reconnection_attempts` in `request_options` takes precedence over this value.
+
+    follow_redirects : typing.Optional[bool]
+        Whether the default httpx client follows redirects or not, this is irrelevant if a custom httpx client is passed in.
+
+    httpx_client : typing.Optional[httpx.Client]
+        The httpx client to use for making requests, a preconfigured client is used by default, however this is useful should you want to pass in any custom httpx configuration.
+
+    logging : typing.Optional[typing.Union[LogConfig, Logger]]
+        Configure logging for the SDK. Accepts a LogConfig dict with 'level' (debug/info/warn/error), 'logger' (custom logger implementation), and 'silent' (boolean, defaults to True) fields. You can also pass a pre-configured Logger instance.
+
+    Examples
+    --------
+    from dsainvg_decide_extractor_sdk import DexaAiSdk
+
+    client = DexaAiSdk()
+    """
+
+    def __init__(
+        self,
+        *,
+        base_url: typing.Optional[str] = None,
+        environment: DexaAiSdkEnvironment = DexaAiSdkEnvironment.DEFAULT,
+        headers: typing.Optional[typing.Dict[str, str]] = None,
+        timeout: typing.Optional[float] = None,
+        max_retries: typing.Optional[int] = None,
+        stream_reconnection_enabled: typing.Optional[bool] = None,
+        max_stream_reconnection_attempts: typing.Optional[int] = None,
+        follow_redirects: typing.Optional[bool] = True,
+        httpx_client: typing.Optional[httpx.Client] = None,
+        logging: typing.Optional[typing.Union[LogConfig, Logger]] = None,
+    ):
+        _defaulted_timeout = timeout if timeout is not None else 60 if httpx_client is None else None
+        _defaulted_max_retries = max_retries if max_retries is not None else 2
+        self._client_wrapper = SyncClientWrapper(
+            base_url=_get_base_url(base_url=base_url, environment=environment),
+            headers=headers,
+            httpx_client=httpx_client
+            if httpx_client is not None
+            else httpx.Client(timeout=_defaulted_timeout, follow_redirects=follow_redirects)
+            if follow_redirects is not None
+            else httpx.Client(timeout=_defaulted_timeout),
+            timeout=_defaulted_timeout,
+            max_retries=_defaulted_max_retries,
+            stream_reconnection_enabled=stream_reconnection_enabled,
+            max_stream_reconnection_attempts=max_stream_reconnection_attempts,
+            logging=logging,
+        )
+        self._tasks: typing.Optional[TasksClient] = None
+
+    @property
+    def tasks(self):
+        if self._tasks is None:
+            from .tasks.client import TasksClient  # noqa: E402
+
+            self._tasks = TasksClient(client_wrapper=self._client_wrapper)
+        return self._tasks
+
+
+def _make_default_async_client(
+    timeout: typing.Optional[float],
+    follow_redirects: typing.Optional[bool],
+) -> httpx.AsyncClient:
+    try:
+        import httpx_aiohttp  # type: ignore[import-not-found]
+    except ImportError:
+        pass
+    else:
+        if follow_redirects is not None:
+            return httpx_aiohttp.HttpxAiohttpClient(timeout=timeout, follow_redirects=follow_redirects)
+        return httpx_aiohttp.HttpxAiohttpClient(timeout=timeout)
+
+    if follow_redirects is not None:
+        return httpx.AsyncClient(timeout=timeout, follow_redirects=follow_redirects)
+    return httpx.AsyncClient(timeout=timeout)
+
+
+class AsyncDexaAiSdk:
+    """
+    Use this class to access the different functions within the SDK. You can instantiate any number of clients with different configuration that will propagate to these functions.
+
+    Parameters
+    ----------
+    base_url : typing.Optional[str]
+        The base url to use for requests from the client.
+
+    environment : DexaAiSdkEnvironment
+        The environment to use for requests from the client. from .environment import DexaAiSdkEnvironment
+
+
+
+        Defaults to DexaAiSdkEnvironment.DEFAULT
+
+
+
+    headers : typing.Optional[typing.Dict[str, str]]
+        Additional headers to send with every request.
+
+    timeout : typing.Optional[float]
+        The timeout to be used, in seconds, for requests. By default the timeout is 60 seconds, unless a custom httpx client is used, in which case this default is not enforced.
+
+    max_retries : typing.Optional[int]
+        The default maximum number of retries for failed requests. Defaults to 2. Per-request `max_retries` in `request_options` takes precedence over this value.
+
+    stream_reconnection_enabled : typing.Optional[bool]
+        Whether to automatically reconnect on stream disconnection for resumable streaming endpoints. Defaults to True. Per-request `stream_reconnection_enabled` in `request_options` takes precedence over this value.
+
+    max_stream_reconnection_attempts : typing.Optional[int]
+        The maximum number of reconnection attempts for resumable streaming endpoints. Defaults to no limit. Per-request `max_stream_reconnection_attempts` in `request_options` takes precedence over this value.
+
+    follow_redirects : typing.Optional[bool]
+        Whether the default httpx client follows redirects or not, this is irrelevant if a custom httpx client is passed in.
+
+    httpx_client : typing.Optional[httpx.AsyncClient]
+        The httpx client to use for making requests, a preconfigured client is used by default, however this is useful should you want to pass in any custom httpx configuration.
+
+    logging : typing.Optional[typing.Union[LogConfig, Logger]]
+        Configure logging for the SDK. Accepts a LogConfig dict with 'level' (debug/info/warn/error), 'logger' (custom logger implementation), and 'silent' (boolean, defaults to True) fields. You can also pass a pre-configured Logger instance.
+
+    Examples
+    --------
+    from dsainvg_decide_extractor_sdk import AsyncDexaAiSdk
+
+    client = AsyncDexaAiSdk()
+    """
+
+    def __init__(
+        self,
+        *,
+        base_url: typing.Optional[str] = None,
+        environment: DexaAiSdkEnvironment = DexaAiSdkEnvironment.DEFAULT,
+        headers: typing.Optional[typing.Dict[str, str]] = None,
+        timeout: typing.Optional[float] = None,
+        max_retries: typing.Optional[int] = None,
+        stream_reconnection_enabled: typing.Optional[bool] = None,
+        max_stream_reconnection_attempts: typing.Optional[int] = None,
+        follow_redirects: typing.Optional[bool] = True,
+        httpx_client: typing.Optional[httpx.AsyncClient] = None,
+        logging: typing.Optional[typing.Union[LogConfig, Logger]] = None,
+    ):
+        _defaulted_timeout = timeout if timeout is not None else 60 if httpx_client is None else None
+        _defaulted_max_retries = max_retries if max_retries is not None else 2
+        self._client_wrapper = AsyncClientWrapper(
+            base_url=_get_base_url(base_url=base_url, environment=environment),
+            headers=headers,
+            httpx_client=httpx_client
+            if httpx_client is not None
+            else _make_default_async_client(timeout=_defaulted_timeout, follow_redirects=follow_redirects),
+            timeout=_defaulted_timeout,
+            max_retries=_defaulted_max_retries,
+            stream_reconnection_enabled=stream_reconnection_enabled,
+            max_stream_reconnection_attempts=max_stream_reconnection_attempts,
+            logging=logging,
+        )
+        self._tasks: typing.Optional[AsyncTasksClient] = None
+
+    @property
+    def tasks(self):
+        if self._tasks is None:
+            from .tasks.client import AsyncTasksClient  # noqa: E402
+
+            self._tasks = AsyncTasksClient(client_wrapper=self._client_wrapper)
+        return self._tasks
+
+
+def _get_base_url(*, base_url: typing.Optional[str] = None, environment: DexaAiSdkEnvironment) -> str:
+    if base_url is not None:
+        return base_url
+    elif environment is not None:
+        return environment.value
+    else:
+        raise Exception("Please pass in either base_url or environment to construct the client")

dexa_ai_sdk/core/__init__.py

@@ -0,0 +1,127 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+
+import typing
+from importlib import import_module
+
+if typing.TYPE_CHECKING:
+    from .api_error import ApiError
+    from .client_wrapper import AsyncClientWrapper, BaseClientWrapper, SyncClientWrapper
+    from .datetime_utils import Rfc2822DateTime, parse_rfc2822_datetime, serialize_datetime
+    from .file import File, convert_file_dict_to_httpx_tuples, with_content_type
+    from .http_client import AsyncHttpClient, HttpClient
+    from .http_response import AsyncHttpResponse, HttpResponse
+    from .jsonable_encoder import encode_path_param, jsonable_encoder
+    from .logging import ConsoleLogger, ILogger, LogConfig, LogLevel, Logger, create_logger
+    from .parse_error import ParsingError
+    from .pydantic_utilities import (
+        IS_PYDANTIC_V2,
+        UniversalBaseModel,
+        UniversalRootModel,
+        parse_obj_as,
+        universal_field_validator,
+        universal_root_validator,
+        update_forward_refs,
+    )
+    from .query_encoder import encode_query
+    from .remove_none_from_dict import remove_none_from_dict
+    from .request_options import RequestOptions
+    from .serialization import FieldMetadata, convert_and_respect_annotation_metadata
+_dynamic_imports: typing.Dict[str, str] = {
+    "ApiError": ".api_error",
+    "AsyncClientWrapper": ".client_wrapper",
+    "AsyncHttpClient": ".http_client",
+    "AsyncHttpResponse": ".http_response",
+    "BaseClientWrapper": ".client_wrapper",
+    "ConsoleLogger": ".logging",
+    "FieldMetadata": ".serialization",
+    "File": ".file",
+    "HttpClient": ".http_client",
+    "HttpResponse": ".http_response",
+    "ILogger": ".logging",
+    "IS_PYDANTIC_V2": ".pydantic_utilities",
+    "LogConfig": ".logging",
+    "LogLevel": ".logging",
+    "Logger": ".logging",
+    "ParsingError": ".parse_error",
+    "RequestOptions": ".request_options",
+    "Rfc2822DateTime": ".datetime_utils",
+    "SyncClientWrapper": ".client_wrapper",
+    "UniversalBaseModel": ".pydantic_utilities",
+    "UniversalRootModel": ".pydantic_utilities",
+    "convert_and_respect_annotation_metadata": ".serialization",
+    "convert_file_dict_to_httpx_tuples": ".file",
+    "create_logger": ".logging",
+    "encode_path_param": ".jsonable_encoder",
+    "encode_query": ".query_encoder",
+    "jsonable_encoder": ".jsonable_encoder",
+    "parse_obj_as": ".pydantic_utilities",
+    "parse_rfc2822_datetime": ".datetime_utils",
+    "remove_none_from_dict": ".remove_none_from_dict",
+    "serialize_datetime": ".datetime_utils",
+    "universal_field_validator": ".pydantic_utilities",
+    "universal_root_validator": ".pydantic_utilities",
+    "update_forward_refs": ".pydantic_utilities",
+    "with_content_type": ".file",
+}
+
+
+def __getattr__(attr_name: str) -> typing.Any:
+    module_name = _dynamic_imports.get(attr_name)
+    if module_name is None:
+        raise AttributeError(f"No {attr_name} found in _dynamic_imports for module name -> {__name__}")
+    try:
+        module = import_module(module_name, __package__)
+        if module_name == f".{attr_name}":
+            return module
+        else:
+            return getattr(module, attr_name)
+    except ImportError as e:
+        raise ImportError(f"Failed to import {attr_name} from {module_name}: {e}") from e
+    except AttributeError as e:
+        raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
+
+
+def __dir__():
+    lazy_attrs = list(_dynamic_imports.keys())
+    return sorted(lazy_attrs)
+
+
+__all__ = [
+    "ApiError",
+    "AsyncClientWrapper",
+    "AsyncHttpClient",
+    "AsyncHttpResponse",
+    "BaseClientWrapper",
+    "ConsoleLogger",
+    "FieldMetadata",
+    "File",
+    "HttpClient",
+    "HttpResponse",
+    "ILogger",
+    "IS_PYDANTIC_V2",
+    "LogConfig",
+    "LogLevel",
+    "Logger",
+    "ParsingError",
+    "RequestOptions",
+    "Rfc2822DateTime",
+    "SyncClientWrapper",
+    "UniversalBaseModel",
+    "UniversalRootModel",
+    "convert_and_respect_annotation_metadata",
+    "convert_file_dict_to_httpx_tuples",
+    "create_logger",
+    "encode_path_param",
+    "encode_query",
+    "jsonable_encoder",
+    "parse_obj_as",
+    "parse_rfc2822_datetime",
+    "remove_none_from_dict",
+    "serialize_datetime",
+    "universal_field_validator",
+    "universal_root_validator",
+    "update_forward_refs",
+    "with_content_type",
+]

dexa_ai_sdk/core/api_error.py

@@ -0,0 +1,23 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from typing import Any, Dict, Optional
+
+
+class ApiError(Exception):
+    headers: Optional[Dict[str, str]]
+    status_code: Optional[int]
+    body: Any
+
+    def __init__(
+        self,
+        *,
+        headers: Optional[Dict[str, str]] = None,
+        status_code: Optional[int] = None,
+        body: Any = None,
+    ) -> None:
+        self.headers = headers
+        self.status_code = status_code
+        self.body = body
+
+    def __str__(self) -> str:
+        return f"headers: {self.headers}, status_code: {self.status_code}, body: {self.body}"