mistralai 1.9.7__py3-none-any.whl → 1.9.10__py3-none-any.whl

mistralai_azure/utils/__init__.py

@@ -1,50 +1,55 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
-from .annotations import get_discriminator
-from .enums import OpenEnumMeta
-from .headers import get_headers, get_response_headers
-from .metadata import (
-    FieldMetadata,
-    find_metadata,
-    FormMetadata,
-    HeaderMetadata,
-    MultipartFormMetadata,
-    PathParamMetadata,
-    QueryParamMetadata,
-    RequestMetadata,
-    SecurityMetadata,
-)
-from .queryparams import get_query_params
-from .retries import BackoffStrategy, Retries, retry, retry_async, RetryConfig
-from .requestbodies import serialize_request_body, SerializedRequestBody
-from .security import get_security
-from .serializers import (
-    get_pydantic_model,
-    marshal_json,
-    unmarshal,
-    unmarshal_json,
-    serialize_decimal,
-    serialize_float,
-    serialize_int,
-    stream_to_text,
-    stream_to_text_async,
-    stream_to_bytes,
-    stream_to_bytes_async,
-    validate_const,
-    validate_decimal,
-    validate_float,
-    validate_int,
-    validate_open_enum,
-)
-from .url import generate_url, template_url, remove_suffix
-from .values import (
-    get_global_from_env,
-    match_content_type,
-    match_status_codes,
-    match_response,
-    cast_partial,
-)
-from .logger import Logger, get_body_content, get_default_logger
+from typing import TYPE_CHECKING
+from importlib import import_module
+
+if TYPE_CHECKING:
+    from .annotations import get_discriminator
+    from .datetimes import parse_datetime
+    from .enums import OpenEnumMeta
+    from .headers import get_headers, get_response_headers
+    from .metadata import (
+        FieldMetadata,
+        find_metadata,
+        FormMetadata,
+        HeaderMetadata,
+        MultipartFormMetadata,
+        PathParamMetadata,
+        QueryParamMetadata,
+        RequestMetadata,
+        SecurityMetadata,
+    )
+    from .queryparams import get_query_params
+    from .retries import BackoffStrategy, Retries, retry, retry_async, RetryConfig
+    from .requestbodies import serialize_request_body, SerializedRequestBody
+    from .security import get_security
+    from .serializers import (
+        get_pydantic_model,
+        marshal_json,
+        unmarshal,
+        unmarshal_json,
+        serialize_decimal,
+        serialize_float,
+        serialize_int,
+        stream_to_text,
+        stream_to_text_async,
+        stream_to_bytes,
+        stream_to_bytes_async,
+        validate_const,
+        validate_decimal,
+        validate_float,
+        validate_int,
+        validate_open_enum,
+    )
+    from .url import generate_url, template_url, remove_suffix
+    from .values import (
+        get_global_from_env,
+        match_content_type,
+        match_status_codes,
+        match_response,
+        cast_partial,
+    )
+    from .logger import Logger, get_body_content, get_default_logger
 
 __all__ = [
     "BackoffStrategy",
@@ -55,6 +60,7 @@ __all__ = [
     "get_body_content",
     "get_default_logger",
     "get_discriminator",
+    "parse_datetime",
     "get_global_from_env",
     "get_headers",
     "get_pydantic_model",
@@ -97,3 +103,82 @@ __all__ = [
     "validate_open_enum",
     "cast_partial",
 ]
+
+_dynamic_imports: dict[str, str] = {
+    "BackoffStrategy": ".retries",
+    "FieldMetadata": ".metadata",
+    "find_metadata": ".metadata",
+    "FormMetadata": ".metadata",
+    "generate_url": ".url",
+    "get_body_content": ".logger",
+    "get_default_logger": ".logger",
+    "get_discriminator": ".annotations",
+    "parse_datetime": ".datetimes",
+    "get_global_from_env": ".values",
+    "get_headers": ".headers",
+    "get_pydantic_model": ".serializers",
+    "get_query_params": ".queryparams",
+    "get_response_headers": ".headers",
+    "get_security": ".security",
+    "HeaderMetadata": ".metadata",
+    "Logger": ".logger",
+    "marshal_json": ".serializers",
+    "match_content_type": ".values",
+    "match_status_codes": ".values",
+    "match_response": ".values",
+    "MultipartFormMetadata": ".metadata",
+    "OpenEnumMeta": ".enums",
+    "PathParamMetadata": ".metadata",
+    "QueryParamMetadata": ".metadata",
+    "remove_suffix": ".url",
+    "Retries": ".retries",
+    "retry": ".retries",
+    "retry_async": ".retries",
+    "RetryConfig": ".retries",
+    "RequestMetadata": ".metadata",
+    "SecurityMetadata": ".metadata",
+    "serialize_decimal": ".serializers",
+    "serialize_float": ".serializers",
+    "serialize_int": ".serializers",
+    "serialize_request_body": ".requestbodies",
+    "SerializedRequestBody": ".requestbodies",
+    "stream_to_text": ".serializers",
+    "stream_to_text_async": ".serializers",
+    "stream_to_bytes": ".serializers",
+    "stream_to_bytes_async": ".serializers",
+    "template_url": ".url",
+    "unmarshal": ".serializers",
+    "unmarshal_json": ".serializers",
+    "validate_decimal": ".serializers",
+    "validate_const": ".serializers",
+    "validate_float": ".serializers",
+    "validate_int": ".serializers",
+    "validate_open_enum": ".serializers",
+    "cast_partial": ".values",
+}
+
+
+def __getattr__(attr_name: str) -> object:
+    module_name = _dynamic_imports.get(attr_name)
+    if module_name is None:
+        raise AttributeError(
+            f"no {attr_name} found in _dynamic_imports, module name -> {__name__} "
+        )
+
+    try:
+        module = import_module(module_name, __package__)
+        result = getattr(module, attr_name)
+        return result
+    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)

mistralai_azure/utils/datetimes.py

@@ -0,0 +1,23 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from datetime import datetime
+import sys
+
+
+def parse_datetime(datetime_string: str) -> datetime:
+    """
+    Convert a RFC 3339 / ISO 8601 formatted string into a datetime object.
+    Python versions 3.11 and later support parsing RFC 3339 directly with
+    datetime.fromisoformat(), but for earlier versions, this function
+    encapsulates the necessary extra logic.
+    """
+    # Python 3.11 and later can parse RFC 3339 directly
+    if sys.version_info >= (3, 11):
+        return datetime.fromisoformat(datetime_string)
+    
+    # For Python 3.10 and earlier, a common ValueError is trailing 'Z' suffix,
+    # so fix that upfront.
+    if datetime_string.endswith("Z"):
+        datetime_string = datetime_string[:-1] + "+00:00"
+
+    return datetime.fromisoformat(datetime_string)

mistralai_azure/utils/enums.py

@@ -1,34 +1,74 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 import enum
-
+import sys
 
 class OpenEnumMeta(enum.EnumMeta):
-    def __call__(
-        cls, value, names=None, *, module=None, qualname=None, type=None, start=1
-    ):
-        # The `type` kwarg also happens to be a built-in that pylint flags as
-        # redeclared. Safe to ignore this lint rule with this scope.
-        # pylint: disable=redefined-builtin
+    # The __call__ method `boundary` kwarg was added in 3.11 and must be present
+    # for pyright. Refer also: https://github.com/pylint-dev/pylint/issues/9622
+    # pylint: disable=unexpected-keyword-arg
+    # The __call__ method `values` varg must be named for pyright.
+    # pylint: disable=keyword-arg-before-vararg
+
+    if sys.version_info >= (3, 11):
+        def __call__(
+            cls, value, names=None, *values, module=None, qualname=None, type=None, start=1, boundary=None
+        ):
+            # The `type` kwarg also happens to be a built-in that pylint flags as
+            # redeclared. Safe to ignore this lint rule with this scope.
+            # pylint: disable=redefined-builtin
+
+            if names is not None:
+                return super().__call__(
+                    value,
+                    names=names,
+                    *values,
+                    module=module,
+                    qualname=qualname,
+                    type=type,
+                    start=start,
+                    boundary=boundary,
+                )
+
+            try:
+                return super().__call__(
+                    value,
+                    names=names,  # pyright: ignore[reportArgumentType]
+                    *values,
+                    module=module,
+                    qualname=qualname,
+                    type=type,
+                    start=start,
+                    boundary=boundary,
+                )
+            except ValueError:
+                return value
+    else:
+        def __call__(
+            cls, value, names=None, *, module=None, qualname=None, type=None, start=1
+        ):
+            # The `type` kwarg also happens to be a built-in that pylint flags as
+            # redeclared. Safe to ignore this lint rule with this scope.
+            # pylint: disable=redefined-builtin
 
-        if names is not None:
-            return super().__call__(
-                value,
-                names=names,
-                module=module,
-                qualname=qualname,
-                type=type,
-                start=start,
-            )
+            if names is not None:
+                return super().__call__(
+                    value,
+                    names=names,
+                    module=module,
+                    qualname=qualname,
+                    type=type,
+                    start=start,
+                )
 
-        try:
-            return super().__call__(
-                value,
-                names=names,  # pyright: ignore[reportArgumentType]
-                module=module,
-                qualname=qualname,
-                type=type,
-                start=start,
-            )
-        except ValueError:
-            return value
+            try:
+                return super().__call__(
+                    value,
+                    names=names,  # pyright: ignore[reportArgumentType]
+                    module=module,
+                    qualname=qualname,
+                    type=type,
+                    start=start,
+                )
+            except ValueError:
+                return value

mistralai_azure/utils/forms.py

@@ -86,11 +86,39 @@ def _populate_form(
     return form
 
 
+def _extract_file_properties(file_obj: Any) -> Tuple[str, Any, Any]:
+    """Extract file name, content, and content type from a file object."""
+    file_fields: Dict[str, FieldInfo] = file_obj.__class__.model_fields
+
+    file_name = ""
+    content = None
+    content_type = None
+
+    for file_field_name in file_fields:
+        file_field = file_fields[file_field_name]
+
+        file_metadata = find_field_metadata(file_field, MultipartFormMetadata)
+        if file_metadata is None:
+            continue
+
+        if file_metadata.content:
+            content = getattr(file_obj, file_field_name, None)
+        elif file_field_name == "content_type":
+            content_type = getattr(file_obj, file_field_name, None)
+        else:
+            file_name = getattr(file_obj, file_field_name)
+
+    if file_name == "" or content is None:
+        raise ValueError("invalid multipart/form-data file")
+
+    return file_name, content, content_type
+
+
 def serialize_multipart_form(
     media_type: str, request: Any
-) -> Tuple[str, Dict[str, Any], Dict[str, Any]]:
+) -> Tuple[str, Dict[str, Any], List[Tuple[str, Any]]]:
     form: Dict[str, Any] = {}
-    files: Dict[str, Any] = {}
+    files: List[Tuple[str, Any]] = []
 
     if not isinstance(request, BaseModel):
         raise TypeError("invalid request body type")
@@ -112,39 +140,32 @@ def serialize_multipart_form(
         f_name = field.alias if field.alias else name
 
         if field_metadata.file:
-            file_fields: Dict[str, FieldInfo] = val.__class__.model_fields
-
-            file_name = ""
-            content = None
-            content_type = None
-
-            for file_field_name in file_fields:
-                file_field = file_fields[file_field_name]
+            if isinstance(val, List):
+                # Handle array of files
+                for file_obj in val:
+                    if not _is_set(file_obj):
+                        continue
+                        
+                    file_name, content, content_type = _extract_file_properties(file_obj)
 
-                file_metadata = find_field_metadata(file_field, MultipartFormMetadata)
-                if file_metadata is None:
-                    continue
+                    if content_type is not None:
+                        files.append((f_name + "[]", (file_name, content, content_type)))
+                    else:
+                        files.append((f_name + "[]", (file_name, content)))
+            else:
+                # Handle single file
+                file_name, content, content_type = _extract_file_properties(val)
 
-                if file_metadata.content:
-                    content = getattr(val, file_field_name, None)
-                elif file_field_name == "content_type":
-                    content_type = getattr(val, file_field_name, None)
+                if content_type is not None:
+                    files.append((f_name, (file_name, content, content_type)))
                 else:
-                    file_name = getattr(val, file_field_name)
-
-            if file_name == "" or content is None:
-                raise ValueError("invalid multipart/form-data file")
-
-            if content_type is not None:
-                files[f_name] = (file_name, content, content_type)
-            else:
-                files[f_name] = (file_name, content)
+                    files.append((f_name, (file_name, content)))
         elif field_metadata.json:
-            files[f_name] = (
+            files.append((f_name, (
                 None,
                 marshal_json(val, request_field_types[name]),
                 "application/json",
-            )
+            )))
         else:
             if isinstance(val, List):
                 values = []

mistralai_azure/utils/serializers.py

@@ -1,13 +1,16 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from decimal import Decimal
+import functools
 import json
-from typing import Any, Dict, List, Union, get_args
-import httpx
+import typing
+from typing import Any, Dict, List, Tuple, Union, get_args
+import typing_extensions
 from typing_extensions import get_origin
+
+import httpx
 from pydantic import ConfigDict, create_model
 from pydantic_core import from_json
-from typing_inspection.typing_objects import is_union
 
 from ..types.basemodel import BaseModel, Nullable, OptionalNullable, Unset
 
@@ -185,6 +188,13 @@ def is_nullable(field):
     return False
 
 
+def is_union(obj: object) -> bool:
+    """
+    Returns True if the given object is a typing.Union or typing_extensions.Union.
+    """
+    return any(obj is typing_obj for typing_obj in _get_typing_objects_by_name_of("Union"))
+
+
 def stream_to_text(stream: httpx.Response) -> str:
     return "".join(stream.iter_text())
 
@@ -217,3 +227,22 @@ def _contains_pydantic_model(data: Any) -> bool:
         return any(_contains_pydantic_model(value) for value in data.values())
 
     return False
+
+
+@functools.cache
+def _get_typing_objects_by_name_of(name: str) -> Tuple[Any, ...]:
+    """
+    Get typing objects by name from typing and typing_extensions.
+    Reference: https://typing-extensions.readthedocs.io/en/latest/#runtime-use-of-types
+    """
+    result = tuple(
+        getattr(module, name)
+        for module in (typing, typing_extensions)
+        if hasattr(module, name)
+    )
+    if not result:
+        raise ValueError(
+            f"Neither typing nor typing_extensions has an object called {name!r}"
+        )
+    return result
+