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

mistralai/models/uploadfileout.py

@@ -1,20 +1,16 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
+from .filepurpose import FilePurpose
 from .sampletype import SampleType
 from .source import Source
-from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL, UnrecognizedStr
+from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from mistralai.utils import validate_open_enum
-import pydantic
 from pydantic import model_serializer
 from pydantic.functional_validators import PlainValidator
-from typing import Final, Literal, TypedDict, Union
-from typing_extensions import Annotated, NotRequired
+from typing_extensions import Annotated, NotRequired, TypedDict
 
 
-Purpose = Union[Literal["fine-tune"], UnrecognizedStr]
-r"""The intended purpose of the uploaded file. Only accepts fine-tuning (`fine-tune`) for now."""
-
 class UploadFileOutTypedDict(TypedDict):
     id: str
     r"""The unique identifier of the file."""
@@ -26,28 +22,36 @@ class UploadFileOutTypedDict(TypedDict):
     r"""The UNIX timestamp (in seconds) of the event."""
     filename: str
     r"""The name of the uploaded file."""
+    purpose: FilePurpose
     sample_type: SampleType
     source: Source
     num_lines: NotRequired[Nullable[int]]
-    
+
 
 class UploadFileOut(BaseModel):
     id: str
     r"""The unique identifier of the file."""
+
     object: str
     r"""The object type, which is always \"file\"."""
+
     bytes: int
     r"""The size of the file, in bytes."""
+
     created_at: int
     r"""The UNIX timestamp (in seconds) of the event."""
+
     filename: str
     r"""The name of the uploaded file."""
-    sample_type: SampleType
-    source: Source
-    PURPOSE: Annotated[Final[Annotated[Purpose, PlainValidator(validate_open_enum(False))]], pydantic.Field(alias="purpose")] = "fine-tune" # type: ignore
-    r"""The intended purpose of the uploaded file. Only accepts fine-tuning (`fine-tune`) for now."""
+
+    purpose: Annotated[FilePurpose, PlainValidator(validate_open_enum(False))]
+
+    sample_type: Annotated[SampleType, PlainValidator(validate_open_enum(False))]
+
+    source: Annotated[Source, PlainValidator(validate_open_enum(False))]
+
     num_lines: OptionalNullable[int] = UNSET
-    
+
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
         optional_fields = ["num_lines"]
@@ -61,9 +65,13 @@ class UploadFileOut(BaseModel):
         for n, f in self.model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
+            serialized.pop(k, None)
 
             optional_nullable = k in optional_fields and k in nullable_fields
-            is_set = (self.__pydantic_fields_set__.intersection({n}) or k in null_default_fields) # pylint: disable=no-member
+            is_set = (
+                self.__pydantic_fields_set__.intersection({n})
+                or k in null_default_fields
+            )  # pylint: disable=no-member
 
             if val is not None and val != UNSET_SENTINEL:
                 m[k] = val
@@ -73,4 +81,3 @@ class UploadFileOut(BaseModel):
                 m[k] = val
 
         return m
-        

mistralai/models/usageinfo.py

@@ -2,17 +2,18 @@
 
 from __future__ import annotations
 from mistralai.types import BaseModel
-from typing import TypedDict
+from typing_extensions import TypedDict
 
 
 class UsageInfoTypedDict(TypedDict):
     prompt_tokens: int
     completion_tokens: int
     total_tokens: int
-    
+
 
 class UsageInfo(BaseModel):
     prompt_tokens: int
+
     completion_tokens: int
+
     total_tokens: int
-    

mistralai/models/usermessage.py

@@ -1,26 +1,58 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from .textchunk import TextChunk, TextChunkTypedDict
-from mistralai.types import BaseModel
-from typing import List, Literal, Optional, TypedDict, Union
-from typing_extensions import NotRequired
+from .contentchunk import ContentChunk, ContentChunkTypedDict
+from mistralai.types import BaseModel, Nullable, UNSET_SENTINEL
+from pydantic import model_serializer
+from typing import List, Literal, Optional, Union
+from typing_extensions import NotRequired, TypedDict
 
 
-UserMessageContentTypedDict = Union[str, List[TextChunkTypedDict]]
+UserMessageContentTypedDict = Union[str, List[ContentChunkTypedDict]]
 
 
-UserMessageContent = Union[str, List[TextChunk]]
+UserMessageContent = Union[str, List[ContentChunk]]
 
 
 UserMessageRole = Literal["user"]
 
+
 class UserMessageTypedDict(TypedDict):
-    content: UserMessageContentTypedDict
+    content: Nullable[UserMessageContentTypedDict]
     role: NotRequired[UserMessageRole]
-    
+
 
 class UserMessage(BaseModel):
-    content: UserMessageContent
+    content: Nullable[UserMessageContent]
+
     role: Optional[UserMessageRole] = "user"
-    
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["role"]
+        nullable_fields = ["content"]
+        null_default_fields = []
+
+        serialized = handler(self)
+
+        m = {}
+
+        for n, f in self.model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+            serialized.pop(k, None)
+
+            optional_nullable = k in optional_fields and k in nullable_fields
+            is_set = (
+                self.__pydantic_fields_set__.intersection({n})
+                or k in null_default_fields
+            )  # pylint: disable=no-member
+
+            if val is not None and val != UNSET_SENTINEL:
+                m[k] = val
+            elif val != UNSET_SENTINEL and (
+                not k in optional_fields or (optional_nullable and is_set)
+            ):
+                m[k] = val
+
+        return m

mistralai/models/validationerror.py

@@ -2,7 +2,8 @@
 
 from __future__ import annotations
 from mistralai.types import BaseModel
-from typing import List, TypedDict, Union
+from typing import List, Union
+from typing_extensions import TypedDict
 
 
 LocTypedDict = Union[str, int]
@@ -15,10 +16,11 @@ class ValidationErrorTypedDict(TypedDict):
     loc: List[LocTypedDict]
     msg: str
     type: str
-    
+
 
 class ValidationError(BaseModel):
     loc: List[Loc]
+
     msg: str
+
     type: str
-    

mistralai/models/wandbintegration.py

@@ -2,34 +2,47 @@
 
 from __future__ import annotations
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
+from mistralai.utils import validate_const
 import pydantic
 from pydantic import model_serializer
-from typing import Final, Literal, Optional, TypedDict
-from typing_extensions import Annotated, NotRequired
+from pydantic.functional_validators import AfterValidator
+from typing import Literal, Optional
+from typing_extensions import Annotated, NotRequired, TypedDict
 
 
 WandbIntegrationType = Literal["wandb"]
 
+
 class WandbIntegrationTypedDict(TypedDict):
     project: str
     r"""The name of the project that the new run will be created under."""
     api_key: str
     r"""The WandB API key to use for authentication."""
+    type: WandbIntegrationType
     name: NotRequired[Nullable[str]]
     r"""A display name to set for the run. If not set, will use the job ID as the name."""
     run_name: NotRequired[Nullable[str]]
-    
+
 
 class WandbIntegration(BaseModel):
     project: str
     r"""The name of the project that the new run will be created under."""
+
     api_key: str
     r"""The WandB API key to use for authentication."""
-    TYPE: Annotated[Final[Optional[WandbIntegrationType]], pydantic.Field(alias="type")] = "wandb" # type: ignore
+
+    TYPE: Annotated[
+        Annotated[
+            Optional[WandbIntegrationType], AfterValidator(validate_const("wandb"))
+        ],
+        pydantic.Field(alias="type"),
+    ] = "wandb"
+
     name: OptionalNullable[str] = UNSET
     r"""A display name to set for the run. If not set, will use the job ID as the name."""
+
     run_name: OptionalNullable[str] = UNSET
-    
+
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
         optional_fields = ["type", "name", "run_name"]
@@ -43,9 +56,13 @@ class WandbIntegration(BaseModel):
         for n, f in self.model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
+            serialized.pop(k, None)
 
             optional_nullable = k in optional_fields and k in nullable_fields
-            is_set = (self.__pydantic_fields_set__.intersection({n}) or k in null_default_fields) # pylint: disable=no-member
+            is_set = (
+                self.__pydantic_fields_set__.intersection({n})
+                or k in null_default_fields
+            )  # pylint: disable=no-member
 
             if val is not None and val != UNSET_SENTINEL:
                 m[k] = val
@@ -55,4 +72,3 @@ class WandbIntegration(BaseModel):
                 m[k] = val
 
         return m
-        

mistralai/models/wandbintegrationout.py

@@ -2,30 +2,42 @@
 
 from __future__ import annotations
 from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
+from mistralai.utils import validate_const
 import pydantic
 from pydantic import model_serializer
-from typing import Final, Literal, Optional, TypedDict
-from typing_extensions import Annotated, NotRequired
+from pydantic.functional_validators import AfterValidator
+from typing import Literal, Optional
+from typing_extensions import Annotated, NotRequired, TypedDict
 
 
-Type = Literal["wandb"]
+WandbIntegrationOutType = Literal["wandb"]
+
 
 class WandbIntegrationOutTypedDict(TypedDict):
     project: str
     r"""The name of the project that the new run will be created under."""
+    type: WandbIntegrationOutType
     name: NotRequired[Nullable[str]]
     r"""A display name to set for the run. If not set, will use the job ID as the name."""
     run_name: NotRequired[Nullable[str]]
-    
+
 
 class WandbIntegrationOut(BaseModel):
     project: str
     r"""The name of the project that the new run will be created under."""
-    TYPE: Annotated[Final[Optional[Type]], pydantic.Field(alias="type")] = "wandb" # type: ignore
+
+    TYPE: Annotated[
+        Annotated[
+            Optional[WandbIntegrationOutType], AfterValidator(validate_const("wandb"))
+        ],
+        pydantic.Field(alias="type"),
+    ] = "wandb"
+
     name: OptionalNullable[str] = UNSET
     r"""A display name to set for the run. If not set, will use the job ID as the name."""
+
     run_name: OptionalNullable[str] = UNSET
-    
+
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
         optional_fields = ["type", "name", "run_name"]
@@ -39,9 +51,13 @@ class WandbIntegrationOut(BaseModel):
         for n, f in self.model_fields.items():
             k = f.alias or n
             val = serialized.get(k)
+            serialized.pop(k, None)
 
             optional_nullable = k in optional_fields and k in nullable_fields
-            is_set = (self.__pydantic_fields_set__.intersection({n}) or k in null_default_fields) # pylint: disable=no-member
+            is_set = (
+                self.__pydantic_fields_set__.intersection({n})
+                or k in null_default_fields
+            )  # pylint: disable=no-member
 
             if val is not None and val != UNSET_SENTINEL:
                 m[k] = val
@@ -51,4 +67,3 @@ class WandbIntegrationOut(BaseModel):
                 m[k] = val
 
         return m
-