mistralai 1.10.0__py3-none-any.whl → 1.10.1__py3-none-any.whl

mistralai_gcp/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_gcp/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,15 @@ 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 +229,21 @@ 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

mistralai_gcp/utils/unmarshal_json_response.py

@@ -0,0 +1,24 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from typing import Any, Optional
+
+import httpx
+
+from .serializers import unmarshal_json
+from mistralai_gcp import models
+
+
+def unmarshal_json_response(
+    typ: Any, http_res: httpx.Response, body: Optional[str] = None
+) -> Any:
+    if body is None:
+        body = http_res.text
+    try:
+        return unmarshal_json(body, typ)
+    except Exception as e:
+        raise models.ResponseValidationError(
+            "Response validation failed",
+            http_res,
+            e,
+            body,
+        ) from e