mistralai 0.5.5a50__py3-none-any.whl → 1.0.0rc2__py3-none-any.whl

mistralai_gcp/sdk.py

@@ -11,11 +11,13 @@ import httpx
 from mistralai_gcp import models
 from mistralai_gcp._hooks import BeforeRequestHook, SDKHooks
 from mistralai_gcp.chat import Chat
+from mistralai_gcp.fim import Fim
 from mistralai_gcp.types import Nullable
 
 from .basesdk import BaseSDK
 from .httpclient import AsyncHttpClient, HttpClient
 from .sdkconfiguration import SDKConfiguration
+from .utils.logger import Logger, NoOpLogger
 from .utils.retries import RetryConfig
 
 
@@ -23,6 +25,7 @@ class MistralGoogleCloud(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."""
 
     chat: Chat
+    fim: Fim
     r"""Chat Completion API"""
 
     def __init__(
@@ -32,6 +35,7 @@ class MistralGoogleCloud(BaseSDK):
         client: Optional[HttpClient] = None,
         async_client: Optional[AsyncHttpClient] = None,
         retry_config: Optional[Nullable[RetryConfig]] = None,
+        debug_logger: Optional[Logger] = None,
     ) -> None:
         r"""Instantiates the SDK configuring it with the provided parameters.
 
@@ -73,6 +77,9 @@ class MistralGoogleCloud(BaseSDK):
         if async_client is None:
             async_client = httpx.AsyncClient()
 
+        if debug_logger is None:
+            debug_logger = NoOpLogger()
+
         assert issubclass(
             type(async_client), AsyncHttpClient
         ), "The provided async_client must implement the AsyncHttpClient protocol."
@@ -94,6 +101,7 @@ class MistralGoogleCloud(BaseSDK):
                 server_url=f"https://{region}-aiplatform.googleapis.com",
                 server=None,
                 retry_config=retry_config,
+                debug_logger=debug_logger,
             ),
         )
 
@@ -116,6 +124,7 @@ class MistralGoogleCloud(BaseSDK):
 
     def _init_sdks(self):
         self.chat = Chat(self.sdk_configuration)
+        self.fim = Fim(self.sdk_configuration)
 
 
 class GoogleCloudBeforeRequestHook(BeforeRequestHook):

mistralai_gcp/sdkconfiguration.py

@@ -3,7 +3,7 @@
 
 from ._hooks import SDKHooks
 from .httpclient import AsyncHttpClient, HttpClient
-from .utils import RetryConfig, remove_suffix
+from .utils import Logger, RetryConfig, remove_suffix
 from dataclasses import dataclass
 from mistralai_gcp import models
 from mistralai_gcp.types import OptionalNullable, UNSET
@@ -23,14 +23,15 @@ SERVERS = {
 class SDKConfiguration:
     client: HttpClient
     async_client: AsyncHttpClient
+    debug_logger: Logger
     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 = "0.5.5-alpha.53"
-    gen_version: str = "2.382.1"
-    user_agent: str = "speakeasy-sdk/python 0.5.5-alpha.53 2.382.1 0.0.2 mistralai-gcp"
+    sdk_version: str = "1.0.0-alpha.12"
+    gen_version: str = "2.386.0"
+    user_agent: str = "speakeasy-sdk/python 1.0.0-alpha.12 2.386.0 0.0.2 mistralai-gcp"
     retry_config: OptionalNullable[RetryConfig] = Field(default_factory=lambda: UNSET)
     timeout_ms: Optional[int] = None
 

mistralai_gcp/types/basemodel.py

@@ -2,8 +2,8 @@
 
 from pydantic import ConfigDict, model_serializer
 from pydantic import BaseModel as PydanticBaseModel
-from typing import Literal, Optional, TypeVar, Union, NewType
-from typing_extensions import TypeAliasType
+from typing import TYPE_CHECKING, Literal, Optional, TypeVar, Union, NewType
+from typing_extensions import TypeAliasType, TypeAlias
 
 
 class BaseModel(PydanticBaseModel):
@@ -26,10 +26,14 @@ UNSET_SENTINEL = "~?~unset~?~sentinel~?~"
 
 
 T = TypeVar("T")
-Nullable = TypeAliasType("Nullable", Union[T, None], type_params=(T,))
-OptionalNullable = TypeAliasType(
-    "OptionalNullable", Union[Optional[Nullable[T]], Unset], type_params=(T,)
-)
+if TYPE_CHECKING:
+    Nullable: TypeAlias = Union[T, None]
+    OptionalNullable: TypeAlias = Union[Optional[Nullable[T]], Unset]
+else:
+    Nullable = TypeAliasType("Nullable", Union[T, None], type_params=(T,))
+    OptionalNullable = TypeAliasType(
+        "OptionalNullable", Union[Optional[Nullable[T]], Unset], type_params=(T,)
+    )
 
 UnrecognizedInt = NewType("UnrecognizedInt", int)
 UnrecognizedStr = NewType("UnrecognizedStr", str)

mistralai_gcp/utils/__init__.py

@@ -34,6 +34,7 @@ from .serializers import (
 )
 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 .logger import Logger, get_body_content, NoOpLogger
 
 __all__ = [
     "BackoffStrategy",
@@ -41,6 +42,7 @@ __all__ = [
     "find_metadata",
     "FormMetadata",
     "generate_url",
+    "get_body_content",
     "get_discriminator",
     "get_global_from_env",
     "get_headers",
@@ -49,11 +51,13 @@ __all__ = [
     "get_response_headers",
     "get_security",
     "HeaderMetadata",
+    "Logger",
     "marshal_json",
     "match_content_type",
     "match_status_codes",
     "match_response",
     "MultipartFormMetadata",
+    "NoOpLogger",
     "OpenEnumMeta",
     "PathParamMetadata",
     "QueryParamMetadata",

mistralai_gcp/utils/eventstreaming.py

@@ -147,15 +147,14 @@ def _parse_event(
         data = data[:-1]
         event.data = data
 
-        if (
-            data.isnumeric()
-            or data == "true"
-            or data == "false"
-            or data == "null"
-            or data.startswith("{")
-            or data.startswith("[")
-            or data.startswith('"')
-        ):
+        data_is_primitive = (
+            data.isnumeric() or data == "true" or data == "false" or data == "null"
+        )
+        data_is_json = (
+            data.startswith("{") or data.startswith("[") or data.startswith('"')
+        )
+
+        if data_is_primitive or data_is_json:
             try:
                 event.data = json.loads(data)
             except Exception:

mistralai_gcp/utils/logger.py

@@ -0,0 +1,16 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+import httpx
+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)
+

mistralai_gcp/utils/retries.py

@@ -76,7 +76,7 @@ def retry(func, retries: Retries):
 
                         status_major = res.status_code / 100
 
-                        if status_major >= code_range and status_major < code_range + 1:
+                        if code_range <= status_major < code_range + 1:
                             raise TemporaryError(res)
                     else:
                         parsed_code = int(code)
@@ -125,7 +125,7 @@ async def retry_async(func, retries: Retries):
 
                         status_major = res.status_code / 100
 
-                        if status_major >= code_range and status_major < code_range + 1:
+                        if code_range <= status_major < code_range + 1:
                             raise TemporaryError(res)
                     else:
                         parsed_code = int(code)