mistralai 1.0.3__py3-none-any.whl → 1.2.0__py3-none-any.whl

mistralai/sdk.py

@@ -9,7 +9,9 @@ import httpx
 from mistralai import models, utils
 from mistralai._hooks import SDKHooks
 from mistralai.agents import Agents
+from mistralai.batch import Batch
 from mistralai.chat import Chat
+from mistralai.classifiers import Classifiers
 from mistralai.embeddings import Embeddings
 from mistralai.files import Files
 from mistralai.fim import Fim
@@ -18,13 +20,16 @@ from mistralai.models_ import Models
 from mistralai.types import OptionalNullable, UNSET
 from typing import Any, Callable, Dict, Optional, Union
 
+
 class Mistral(BaseSDK):
     r"""Mistral AI API: Our Chat Completion and Embeddings APIs specification. Create your account on [La Plateforme](https://console.mistral.ai) to get access and read the [docs](https://docs.mistral.ai) to learn how to use it."""
+
     models: Models
     r"""Model Management API"""
     files: Files
     r"""Files API"""
     fine_tuning: FineTuning
+    batch: Batch
     chat: Chat
     r"""Chat Completion API."""
     fim: Fim
@@ -33,6 +38,9 @@ class Mistral(BaseSDK):
     r"""Agents API."""
     embeddings: Embeddings
     r"""Embeddings API."""
+    classifiers: Classifiers
+    r"""Classifiers API."""
+
     def __init__(
         self,
         api_key: Optional[Union[Optional[str], Callable[[], Optional[str]]]] = None,
@@ -43,7 +51,7 @@ class Mistral(BaseSDK):
         async_client: Optional[AsyncHttpClient] = None,
         retry_config: OptionalNullable[RetryConfig] = UNSET,
         timeout_ms: Optional[int] = None,
-        debug_logger: Optional[Logger] = None
+        debug_logger: Optional[Logger] = None,
     ) -> None:
         r"""Instantiates the SDK configuring it with the provided parameters.
 
@@ -72,33 +80,37 @@ class Mistral(BaseSDK):
         assert issubclass(
             type(async_client), AsyncHttpClient
         ), "The provided async_client must implement the AsyncHttpClient protocol."
-        
+
         security: Any = None
         if callable(api_key):
-            security = lambda: models.Security(api_key = api_key()) # pylint: disable=unnecessary-lambda-assignment
+            security = lambda: models.Security(api_key=api_key())  # pylint: disable=unnecessary-lambda-assignment
         else:
-            security = models.Security(api_key = api_key)
+            security = models.Security(api_key=api_key)
 
         if server_url is not None:
             if url_params is not None:
                 server_url = utils.template_url(server_url, url_params)
-    
-
-        BaseSDK.__init__(self, SDKConfiguration(
-            client=client,
-            async_client=async_client,
-            security=security,
-            server_url=server_url,
-            server=server,
-            retry_config=retry_config,
-            timeout_ms=timeout_ms,
-            debug_logger=debug_logger
-        ))
+
+        BaseSDK.__init__(
+            self,
+            SDKConfiguration(
+                client=client,
+                async_client=async_client,
+                security=security,
+                server_url=server_url,
+                server=server,
+                retry_config=retry_config,
+                timeout_ms=timeout_ms,
+                debug_logger=debug_logger,
+            ),
+        )
 
         hooks = SDKHooks()
 
         current_server_url, *_ = self.sdk_configuration.get_server_details()
-        server_url, self.sdk_configuration.client = hooks.sdk_init(current_server_url, self.sdk_configuration.client)
+        server_url, self.sdk_configuration.client = hooks.sdk_init(
+            current_server_url, self.sdk_configuration.client
+        )
         if current_server_url != server_url:
             self.sdk_configuration.server_url = server_url
 
@@ -107,13 +119,13 @@ class Mistral(BaseSDK):
 
         self._init_sdks()
 
-
     def _init_sdks(self):
         self.models = Models(self.sdk_configuration)
         self.files = Files(self.sdk_configuration)
         self.fine_tuning = FineTuning(self.sdk_configuration)
+        self.batch = Batch(self.sdk_configuration)
         self.chat = Chat(self.sdk_configuration)
         self.fim = Fim(self.sdk_configuration)
         self.agents = Agents(self.sdk_configuration)
         self.embeddings = Embeddings(self.sdk_configuration)
-    
+        self.classifiers = Classifiers(self.sdk_configuration)

mistralai/sdkconfiguration.py

@@ -1,6 +1,5 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
-
 from ._hooks import SDKHooks
 from .httpclient import AsyncHttpClient, HttpClient
 from .utils import Logger, RetryConfig, remove_suffix
@@ -11,10 +10,10 @@ from pydantic import Field
 from typing import Callable, Dict, Optional, Tuple, Union
 
 
-SERVER_PROD = "prod"
-r"""Production server"""
+SERVER_EU = "eu"
+r"""EU Production server"""
 SERVERS = {
-	SERVER_PROD: "https://api.mistral.ai",
+    SERVER_EU: "https://api.mistral.ai",
 }
 """Contains the list of servers available to the SDK"""
 
@@ -24,14 +23,14 @@ class SDKConfiguration:
     client: HttpClient
     async_client: AsyncHttpClient
     debug_logger: Logger
-    security: Optional[Union[models.Security,Callable[[], models.Security]]] = None
+    security: Optional[Union[models.Security, Callable[[], models.Security]]] = None
     server_url: Optional[str] = ""
     server: Optional[str] = ""
     language: str = "python"
     openapi_doc_version: str = "0.0.2"
-    sdk_version: str = "1.0.3"
-    gen_version: str = "2.404.11"
-    user_agent: str = "speakeasy-sdk/python 1.0.3 2.404.11 0.0.2 mistralai"
+    sdk_version: str = "1.2.0"
+    gen_version: str = "2.452.0"
+    user_agent: str = "speakeasy-sdk/python 1.2.0 2.452.0 0.0.2 mistralai"
     retry_config: OptionalNullable[RetryConfig] = Field(default_factory=lambda: UNSET)
     timeout_ms: Optional[int] = None
 
@@ -42,13 +41,12 @@ class SDKConfiguration:
         if self.server_url is not None and self.server_url:
             return remove_suffix(self.server_url, "/"), {}
         if not self.server:
-            self.server = SERVER_PROD
+            self.server = SERVER_EU
 
         if self.server not in SERVERS:
-            raise ValueError(f"Invalid server \"{self.server}\"")
+            raise ValueError(f'Invalid server "{self.server}"')
 
         return SERVERS[self.server], {}
 
-
     def get_hooks(self) -> SDKHooks:
         return self._hooks

mistralai/utils/__init__.py

@@ -28,13 +28,22 @@ from .serializers import (
     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
+from .values import (
+    get_global_from_env,
+    match_content_type,
+    match_status_codes,
+    match_response,
+)
 from .logger import Logger, get_body_content, get_default_logger
 
 __all__ = [
@@ -76,10 +85,14 @@ __all__ = [
     "serialize_request_body",
     "SerializedRequestBody",
     "stream_to_text",
+    "stream_to_text_async",
+    "stream_to_bytes",
+    "stream_to_bytes_async",
     "template_url",
     "unmarshal",
     "unmarshal_json",
     "validate_decimal",
+    "validate_const",
     "validate_float",
     "validate_int",
     "validate_open_enum",

mistralai/utils/annotations.py

@@ -1,5 +1,6 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
+from enum import Enum
 from typing import Any
 
 def get_discriminator(model: Any, fieldname: str, key: str) -> str:
@@ -10,10 +11,20 @@ def get_discriminator(model: Any, fieldname: str, key: str) -> str:
             raise ValueError(f'Could not find discriminator key {key} in {model}') from e
 
     if hasattr(model, fieldname):
-        return f'{getattr(model, fieldname)}'
+        attr = getattr(model, fieldname)
+
+        if isinstance(attr, Enum):
+            return f'{attr.value}'
+
+        return f'{attr}'
 
     fieldname = fieldname.upper()
     if hasattr(model, fieldname):
-        return f'{getattr(model, fieldname)}'
+        attr = getattr(model, fieldname)
+
+        if isinstance(attr, Enum):
+            return f'{attr.value}'
+
+        return f'{attr}'
 
     raise ValueError(f'Could not find discriminator field {fieldname} in {model}')

mistralai/utils/logger.py

@@ -5,20 +5,23 @@ import logging
 import os
 from typing import Any, Protocol
 
+
 class Logger(Protocol):
     def debug(self, msg: str, *args: Any, **kwargs: Any) -> None:
         pass
 
+
 class NoOpLogger:
     def debug(self, msg: str, *args: Any, **kwargs: Any) -> None:
         pass
 
+
 def get_body_content(req: httpx.Request) -> str:
     return "<streaming body>" if not hasattr(req, "_content") else str(req.content)
 
+
 def get_default_logger() -> Logger:
     if os.getenv("MISTRAL_DEBUG"):
         logging.basicConfig(level=logging.DEBUG)
         return logging.getLogger("mistralai")
     return NoOpLogger()
-

mistralai/utils/retries.py

@@ -1,5 +1,6 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
+import asyncio
 import random
 import time
 from typing import List
@@ -212,5 +213,5 @@ async def retry_with_backoff_async(
                 raise
             sleep = (initial_interval / 1000) * exponent**retries + random.uniform(0, 1)
             sleep = min(sleep, max_interval / 1000)
-            time.sleep(sleep)
+            await asyncio.sleep(sleep)
             retries += 1

mistralai/utils/security.py

@@ -44,8 +44,10 @@ def get_security(security: Any) -> Tuple[Dict[str, str], Dict[str, List[str]]]:
             _parse_security_option(headers, query_params, value)
             return headers, query_params
         if metadata.scheme:
-            # Special case for basic auth which could be a flattened model
-            if metadata.sub_type == "basic" and not isinstance(value, BaseModel):
+            # Special case for basic auth or custom auth which could be a flattened model
+            if metadata.sub_type in ["basic", "custom"] and not isinstance(
+                value, BaseModel
+            ):
                 _parse_security_scheme(headers, query_params, metadata, name, security)
             else:
                 _parse_security_scheme(headers, query_params, metadata, name, value)
@@ -64,7 +66,7 @@ def get_security_from_env(security: Any, security_class: Any) -> Optional[BaseMo
 
     if os.getenv("MISTRAL_API_KEY"):
         security_dict["api_key"] = os.getenv("MISTRAL_API_KEY")
-    
+
     return security_class(**security_dict) if security_dict else None
 
 
@@ -97,9 +99,12 @@ def _parse_security_scheme(
     sub_type = scheme_metadata.sub_type
 
     if isinstance(scheme, BaseModel):
-        if scheme_type == "http" and sub_type == "basic":
-            _parse_basic_auth_scheme(headers, scheme)
-            return
+        if scheme_type == "http":
+            if sub_type == "basic":
+                _parse_basic_auth_scheme(headers, scheme)
+                return
+            if sub_type == "custom":
+                return
 
         scheme_fields: Dict[str, FieldInfo] = scheme.__class__.model_fields
         for name in scheme_fields:
@@ -148,6 +153,8 @@ def _parse_security_scheme_value(
     elif scheme_type == "http":
         if sub_type == "bearer":
             headers[header_name] = _apply_bearer(value)
+        elif sub_type == "custom":
+            return
         else:
             raise ValueError("sub type {sub_type} not supported")
     else:

mistralai/utils/serializers.py

@@ -116,6 +116,19 @@ def validate_open_enum(is_int: bool):
     return validate
 
 
+def validate_const(v):
+    def validate(c):
+        if is_optional_type(type(c)) and c is None:
+            return None
+
+        if v != c:
+            raise ValueError(f"Expected {v}")
+
+        return c
+
+    return validate
+
+
 def unmarshal_json(raw, typ: Any) -> Any:
     return unmarshal(from_json(raw), typ)
 
@@ -172,6 +185,18 @@ def stream_to_text(stream: httpx.Response) -> str:
     return "".join(stream.iter_text())
 
 
+async def stream_to_text_async(stream: httpx.Response) -> str:
+    return "".join([chunk async for chunk in stream.aiter_text()])
+
+
+def stream_to_bytes(stream: httpx.Response) -> bytes:
+    return stream.content
+
+
+async def stream_to_bytes_async(stream: httpx.Response) -> bytes:
+    return await stream.aread()
+
+
 def get_pydantic_model(data: Any, typ: Any) -> Any:
     if not _contains_pydantic_model(data):
         return unmarshal(data, typ)