mistralai 1.0.0rc2__py3-none-any.whl → 1.0.1__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (48) hide show
  1. mistralai/agents.py +2 -2
  2. mistralai/client.py +1 -1
  3. mistralai/jobs.py +20 -30
  4. mistralai/models/__init__.py +3 -3
  5. mistralai/models/agentscompletionrequest.py +14 -14
  6. mistralai/models/agentscompletionstreamrequest.py +8 -8
  7. mistralai/models/chatcompletionrequest.py +14 -14
  8. mistralai/models/chatcompletionstreamrequest.py +14 -14
  9. mistralai/models/delete_model_v1_models_model_id_deleteop.py +2 -0
  10. mistralai/models/deltamessage.py +3 -3
  11. mistralai/models/embeddingrequest.py +8 -8
  12. mistralai/models/fimcompletionrequest.py +8 -8
  13. mistralai/models/fimcompletionstreamrequest.py +8 -8
  14. mistralai/models/functioncall.py +9 -3
  15. mistralai/models/jobs_api_routes_fine_tuning_archive_fine_tuned_modelop.py +2 -0
  16. mistralai/models/jobs_api_routes_fine_tuning_create_fine_tuning_jobop.py +1 -54
  17. mistralai/models/jobs_api_routes_fine_tuning_unarchive_fine_tuned_modelop.py +2 -0
  18. mistralai/models/jobs_api_routes_fine_tuning_update_fine_tuned_modelop.py +2 -0
  19. mistralai/models/retrieve_model_v1_models_model_id_getop.py +2 -0
  20. mistralai/models/systemmessage.py +6 -6
  21. mistralai/models/usermessage.py +6 -6
  22. mistralai/models/validationerror.py +6 -6
  23. mistralai/models_.py +10 -10
  24. mistralai/sdkconfiguration.py +3 -3
  25. {mistralai-1.0.0rc2.dist-info → mistralai-1.0.1.dist-info}/METADATA +36 -17
  26. {mistralai-1.0.0rc2.dist-info → mistralai-1.0.1.dist-info}/RECORD +48 -48
  27. mistralai_azure/models/__init__.py +2 -2
  28. mistralai_azure/models/chatcompletionrequest.py +14 -14
  29. mistralai_azure/models/chatcompletionstreamrequest.py +14 -14
  30. mistralai_azure/models/deltamessage.py +3 -3
  31. mistralai_azure/models/functioncall.py +9 -3
  32. mistralai_azure/models/systemmessage.py +6 -6
  33. mistralai_azure/models/usermessage.py +6 -6
  34. mistralai_azure/models/validationerror.py +6 -6
  35. mistralai_azure/sdkconfiguration.py +3 -3
  36. mistralai_gcp/models/__init__.py +2 -2
  37. mistralai_gcp/models/chatcompletionrequest.py +14 -14
  38. mistralai_gcp/models/chatcompletionstreamrequest.py +14 -14
  39. mistralai_gcp/models/deltamessage.py +3 -3
  40. mistralai_gcp/models/fimcompletionrequest.py +8 -8
  41. mistralai_gcp/models/fimcompletionstreamrequest.py +8 -8
  42. mistralai_gcp/models/functioncall.py +9 -3
  43. mistralai_gcp/models/systemmessage.py +6 -6
  44. mistralai_gcp/models/usermessage.py +6 -6
  45. mistralai_gcp/models/validationerror.py +6 -6
  46. mistralai_gcp/sdkconfiguration.py +3 -3
  47. {mistralai-1.0.0rc2.dist-info → mistralai-1.0.1.dist-info}/LICENSE +0 -0
  48. {mistralai-1.0.0rc2.dist-info → mistralai-1.0.1.dist-info}/WHEEL +0 -0
@@ -7,6 +7,12 @@ from typing import List, Literal, Optional, TypedDict, Union
7
7
  from typing_extensions import NotRequired
8
8
 
9
9
 
10
+ UserMessageContentTypedDict = Union[str, List[TextChunkTypedDict]]
11
+
12
+
13
+ UserMessageContent = Union[str, List[TextChunk]]
14
+
15
+
10
16
  UserMessageRole = Literal["user"]
11
17
 
12
18
  class UserMessageTypedDict(TypedDict):
@@ -18,9 +24,3 @@ class UserMessage(BaseModel):
18
24
  content: UserMessageContent
19
25
  role: Optional[UserMessageRole] = "user"
20
26
 
21
-
22
- UserMessageContentTypedDict = Union[str, List[TextChunkTypedDict]]
23
-
24
-
25
- UserMessageContent = Union[str, List[TextChunk]]
26
-
@@ -5,6 +5,12 @@ from mistralai_azure.types import BaseModel
5
5
  from typing import List, TypedDict, Union
6
6
 
7
7
 
8
+ LocTypedDict = Union[str, int]
9
+
10
+
11
+ Loc = Union[str, int]
12
+
13
+
8
14
  class ValidationErrorTypedDict(TypedDict):
9
15
  loc: List[LocTypedDict]
10
16
  msg: str
@@ -16,9 +22,3 @@ class ValidationError(BaseModel):
16
22
  msg: str
17
23
  type: str
18
24
 
19
-
20
- LocTypedDict = Union[str, int]
21
-
22
-
23
- Loc = Union[str, int]
24
-
@@ -29,9 +29,9 @@ class SDKConfiguration:
29
29
  server: Optional[str] = ""
30
30
  language: str = "python"
31
31
  openapi_doc_version: str = "0.0.2"
32
- sdk_version: str = "1.0.0-alpha.13"
33
- gen_version: str = "2.386.0"
34
- user_agent: str = "speakeasy-sdk/python 1.0.0-alpha.13 2.386.0 0.0.2 mistralai_azure"
32
+ sdk_version: str = "1.0.0-rc.4"
33
+ gen_version: str = "2.390.6"
34
+ user_agent: str = "speakeasy-sdk/python 1.0.0-rc.4 2.390.6 0.0.2 mistralai_azure"
35
35
  retry_config: OptionalNullable[RetryConfig] = Field(default_factory=lambda: UNSET)
36
36
  timeout_ms: Optional[int] = None
37
37
 
@@ -14,7 +14,7 @@ from .fimcompletionrequest import FIMCompletionRequest, FIMCompletionRequestStop
14
14
  from .fimcompletionresponse import FIMCompletionResponse, FIMCompletionResponseTypedDict
15
15
  from .fimcompletionstreamrequest import FIMCompletionStreamRequest, FIMCompletionStreamRequestStop, FIMCompletionStreamRequestStopTypedDict, FIMCompletionStreamRequestTypedDict
16
16
  from .function import Function, FunctionTypedDict
17
- from .functioncall import FunctionCall, FunctionCallTypedDict
17
+ from .functioncall import Arguments, ArgumentsTypedDict, FunctionCall, FunctionCallTypedDict
18
18
  from .httpvalidationerror import HTTPValidationError, HTTPValidationErrorData
19
19
  from .responseformat import ResponseFormat, ResponseFormatTypedDict, ResponseFormats
20
20
  from .sdkerror import SDKError
@@ -28,4 +28,4 @@ from .usageinfo import UsageInfo, UsageInfoTypedDict
28
28
  from .usermessage import UserMessage, UserMessageContent, UserMessageContentTypedDict, UserMessageRole, UserMessageTypedDict
29
29
  from .validationerror import Loc, LocTypedDict, ValidationError, ValidationErrorTypedDict
30
30
 
31
- __all__ = ["AssistantMessage", "AssistantMessageRole", "AssistantMessageTypedDict", "ChatCompletionChoice", "ChatCompletionChoiceFinishReason", "ChatCompletionChoiceTypedDict", "ChatCompletionRequest", "ChatCompletionRequestMessages", "ChatCompletionRequestMessagesTypedDict", "ChatCompletionRequestStop", "ChatCompletionRequestStopTypedDict", "ChatCompletionRequestToolChoice", "ChatCompletionRequestTypedDict", "ChatCompletionResponse", "ChatCompletionResponseTypedDict", "ChatCompletionStreamRequest", "ChatCompletionStreamRequestTypedDict", "CompletionChunk", "CompletionChunkTypedDict", "CompletionEvent", "CompletionEventTypedDict", "CompletionResponseStreamChoice", "CompletionResponseStreamChoiceTypedDict", "Content", "ContentChunk", "ContentChunkTypedDict", "ContentTypedDict", "DeltaMessage", "DeltaMessageTypedDict", "FIMCompletionRequest", "FIMCompletionRequestStop", "FIMCompletionRequestStopTypedDict", "FIMCompletionRequestTypedDict", "FIMCompletionResponse", "FIMCompletionResponseTypedDict", "FIMCompletionStreamRequest", "FIMCompletionStreamRequestStop", "FIMCompletionStreamRequestStopTypedDict", "FIMCompletionStreamRequestTypedDict", "FinishReason", "Function", "FunctionCall", "FunctionCallTypedDict", "FunctionTypedDict", "HTTPValidationError", "HTTPValidationErrorData", "Loc", "LocTypedDict", "Messages", "MessagesTypedDict", "ResponseFormat", "ResponseFormatTypedDict", "ResponseFormats", "Role", "SDKError", "Security", "SecurityTypedDict", "Stop", "StopTypedDict", "SystemMessage", "SystemMessageTypedDict", "TextChunk", "TextChunkTypedDict", "Tool", "ToolCall", "ToolCallTypedDict", "ToolChoice", "ToolMessage", "ToolMessageRole", "ToolMessageTypedDict", "ToolTypedDict", "UsageInfo", "UsageInfoTypedDict", "UserMessage", "UserMessageContent", "UserMessageContentTypedDict", "UserMessageRole", "UserMessageTypedDict", "ValidationError", "ValidationErrorTypedDict"]
31
+ __all__ = ["Arguments", "ArgumentsTypedDict", "AssistantMessage", "AssistantMessageRole", "AssistantMessageTypedDict", "ChatCompletionChoice", "ChatCompletionChoiceFinishReason", "ChatCompletionChoiceTypedDict", "ChatCompletionRequest", "ChatCompletionRequestMessages", "ChatCompletionRequestMessagesTypedDict", "ChatCompletionRequestStop", "ChatCompletionRequestStopTypedDict", "ChatCompletionRequestToolChoice", "ChatCompletionRequestTypedDict", "ChatCompletionResponse", "ChatCompletionResponseTypedDict", "ChatCompletionStreamRequest", "ChatCompletionStreamRequestTypedDict", "CompletionChunk", "CompletionChunkTypedDict", "CompletionEvent", "CompletionEventTypedDict", "CompletionResponseStreamChoice", "CompletionResponseStreamChoiceTypedDict", "Content", "ContentChunk", "ContentChunkTypedDict", "ContentTypedDict", "DeltaMessage", "DeltaMessageTypedDict", "FIMCompletionRequest", "FIMCompletionRequestStop", "FIMCompletionRequestStopTypedDict", "FIMCompletionRequestTypedDict", "FIMCompletionResponse", "FIMCompletionResponseTypedDict", "FIMCompletionStreamRequest", "FIMCompletionStreamRequestStop", "FIMCompletionStreamRequestStopTypedDict", "FIMCompletionStreamRequestTypedDict", "FinishReason", "Function", "FunctionCall", "FunctionCallTypedDict", "FunctionTypedDict", "HTTPValidationError", "HTTPValidationErrorData", "Loc", "LocTypedDict", "Messages", "MessagesTypedDict", "ResponseFormat", "ResponseFormatTypedDict", "ResponseFormats", "Role", "SDKError", "Security", "SecurityTypedDict", "Stop", "StopTypedDict", "SystemMessage", "SystemMessageTypedDict", "TextChunk", "TextChunkTypedDict", "Tool", "ToolCall", "ToolCallTypedDict", "ToolChoice", "ToolMessage", "ToolMessageRole", "ToolMessageTypedDict", "ToolTypedDict", "UsageInfo", "UsageInfoTypedDict", "UserMessage", "UserMessageContent", "UserMessageContentTypedDict", "UserMessageRole", "UserMessageTypedDict", "ValidationError", "ValidationErrorTypedDict"]
@@ -14,6 +14,20 @@ from typing import List, Literal, Optional, TypedDict, Union
14
14
  from typing_extensions import Annotated, NotRequired
15
15
 
16
16
 
17
+ ChatCompletionRequestStopTypedDict = Union[str, List[str]]
18
+ r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
19
+
20
+
21
+ ChatCompletionRequestStop = Union[str, List[str]]
22
+ r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
23
+
24
+
25
+ ChatCompletionRequestMessagesTypedDict = Union[SystemMessageTypedDict, UserMessageTypedDict, AssistantMessageTypedDict, ToolMessageTypedDict]
26
+
27
+
28
+ ChatCompletionRequestMessages = Annotated[Union[Annotated[AssistantMessage, Tag("assistant")], Annotated[SystemMessage, Tag("system")], Annotated[ToolMessage, Tag("tool")], Annotated[UserMessage, Tag("user")]], Discriminator(lambda m: get_discriminator(m, "role", "role"))]
29
+
30
+
17
31
  ChatCompletionRequestToolChoice = Literal["auto", "none", "any"]
18
32
 
19
33
  class ChatCompletionRequestTypedDict(TypedDict):
@@ -89,17 +103,3 @@ class ChatCompletionRequest(BaseModel):
89
103
 
90
104
  return m
91
105
 
92
-
93
- ChatCompletionRequestStopTypedDict = Union[str, List[str]]
94
- r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
95
-
96
-
97
- ChatCompletionRequestStop = Union[str, List[str]]
98
- r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
99
-
100
-
101
- ChatCompletionRequestMessagesTypedDict = Union[SystemMessageTypedDict, UserMessageTypedDict, AssistantMessageTypedDict, ToolMessageTypedDict]
102
-
103
-
104
- ChatCompletionRequestMessages = Annotated[Union[Annotated[AssistantMessage, Tag("assistant")], Annotated[SystemMessage, Tag("system")], Annotated[ToolMessage, Tag("tool")], Annotated[UserMessage, Tag("user")]], Discriminator(lambda m: get_discriminator(m, "role", "role"))]
105
-
@@ -14,6 +14,20 @@ from typing import List, Literal, Optional, TypedDict, Union
14
14
  from typing_extensions import Annotated, NotRequired
15
15
 
16
16
 
17
+ StopTypedDict = Union[str, List[str]]
18
+ r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
19
+
20
+
21
+ Stop = Union[str, List[str]]
22
+ r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
23
+
24
+
25
+ MessagesTypedDict = Union[SystemMessageTypedDict, UserMessageTypedDict, AssistantMessageTypedDict, ToolMessageTypedDict]
26
+
27
+
28
+ Messages = Annotated[Union[Annotated[AssistantMessage, Tag("assistant")], Annotated[SystemMessage, Tag("system")], Annotated[ToolMessage, Tag("tool")], Annotated[UserMessage, Tag("user")]], Discriminator(lambda m: get_discriminator(m, "role", "role"))]
29
+
30
+
17
31
  ToolChoice = Literal["auto", "none", "any"]
18
32
 
19
33
  class ChatCompletionStreamRequestTypedDict(TypedDict):
@@ -87,17 +101,3 @@ class ChatCompletionStreamRequest(BaseModel):
87
101
 
88
102
  return m
89
103
 
90
-
91
- StopTypedDict = Union[str, List[str]]
92
- r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
93
-
94
-
95
- Stop = Union[str, List[str]]
96
- r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
97
-
98
-
99
- MessagesTypedDict = Union[SystemMessageTypedDict, UserMessageTypedDict, AssistantMessageTypedDict, ToolMessageTypedDict]
100
-
101
-
102
- Messages = Annotated[Union[Annotated[AssistantMessage, Tag("assistant")], Annotated[SystemMessage, Tag("system")], Annotated[ToolMessage, Tag("tool")], Annotated[UserMessage, Tag("user")]], Discriminator(lambda m: get_discriminator(m, "role", "role"))]
103
-
@@ -4,20 +4,20 @@ from __future__ import annotations
4
4
  from .toolcall import ToolCall, ToolCallTypedDict
5
5
  from mistralai_gcp.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
6
6
  from pydantic import model_serializer
7
- from typing import Optional, TypedDict
7
+ from typing import List, Optional, TypedDict
8
8
  from typing_extensions import NotRequired
9
9
 
10
10
 
11
11
  class DeltaMessageTypedDict(TypedDict):
12
12
  role: NotRequired[str]
13
13
  content: NotRequired[str]
14
- tool_calls: NotRequired[Nullable[ToolCallTypedDict]]
14
+ tool_calls: NotRequired[Nullable[List[ToolCallTypedDict]]]
15
15
 
16
16
 
17
17
  class DeltaMessage(BaseModel):
18
18
  role: Optional[str] = None
19
19
  content: Optional[str] = None
20
- tool_calls: OptionalNullable[ToolCall] = UNSET
20
+ tool_calls: OptionalNullable[List[ToolCall]] = UNSET
21
21
 
22
22
  @model_serializer(mode="wrap")
23
23
  def serialize_model(self, handler):
@@ -7,6 +7,14 @@ from typing import List, Optional, TypedDict, Union
7
7
  from typing_extensions import NotRequired
8
8
 
9
9
 
10
+ FIMCompletionRequestStopTypedDict = Union[str, List[str]]
11
+ r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
12
+
13
+
14
+ FIMCompletionRequestStop = Union[str, List[str]]
15
+ r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
16
+
17
+
10
18
  class FIMCompletionRequestTypedDict(TypedDict):
11
19
  model: Nullable[str]
12
20
  r"""ID of the model to use. Only compatible for now with:
@@ -84,11 +92,3 @@ class FIMCompletionRequest(BaseModel):
84
92
 
85
93
  return m
86
94
 
87
-
88
- FIMCompletionRequestStopTypedDict = Union[str, List[str]]
89
- r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
90
-
91
-
92
- FIMCompletionRequestStop = Union[str, List[str]]
93
- r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
94
-
@@ -7,6 +7,14 @@ from typing import List, Optional, TypedDict, Union
7
7
  from typing_extensions import NotRequired
8
8
 
9
9
 
10
+ FIMCompletionStreamRequestStopTypedDict = Union[str, List[str]]
11
+ r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
12
+
13
+
14
+ FIMCompletionStreamRequestStop = Union[str, List[str]]
15
+ r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
16
+
17
+
10
18
  class FIMCompletionStreamRequestTypedDict(TypedDict):
11
19
  model: Nullable[str]
12
20
  r"""ID of the model to use. Only compatible for now with:
@@ -82,11 +90,3 @@ class FIMCompletionStreamRequest(BaseModel):
82
90
 
83
91
  return m
84
92
 
85
-
86
- FIMCompletionStreamRequestStopTypedDict = Union[str, List[str]]
87
- r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
88
-
89
-
90
- FIMCompletionStreamRequestStop = Union[str, List[str]]
91
- r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
92
-
@@ -2,15 +2,21 @@
2
2
 
3
3
  from __future__ import annotations
4
4
  from mistralai_gcp.types import BaseModel
5
- from typing import TypedDict
5
+ from typing import Any, Dict, TypedDict, Union
6
+
7
+
8
+ ArgumentsTypedDict = Union[Dict[str, Any], str]
9
+
10
+
11
+ Arguments = Union[Dict[str, Any], str]
6
12
 
7
13
 
8
14
  class FunctionCallTypedDict(TypedDict):
9
15
  name: str
10
- arguments: str
16
+ arguments: ArgumentsTypedDict
11
17
 
12
18
 
13
19
  class FunctionCall(BaseModel):
14
20
  name: str
15
- arguments: str
21
+ arguments: Arguments
16
22
 
@@ -7,6 +7,12 @@ from typing import List, Literal, Optional, TypedDict, Union
7
7
  from typing_extensions import NotRequired
8
8
 
9
9
 
10
+ ContentTypedDict = Union[str, List[ContentChunkTypedDict]]
11
+
12
+
13
+ Content = Union[str, List[ContentChunk]]
14
+
15
+
10
16
  Role = Literal["system"]
11
17
 
12
18
  class SystemMessageTypedDict(TypedDict):
@@ -18,9 +24,3 @@ class SystemMessage(BaseModel):
18
24
  content: Content
19
25
  role: Optional[Role] = "system"
20
26
 
21
-
22
- ContentTypedDict = Union[str, List[ContentChunkTypedDict]]
23
-
24
-
25
- Content = Union[str, List[ContentChunk]]
26
-
@@ -7,6 +7,12 @@ from typing import List, Literal, Optional, TypedDict, Union
7
7
  from typing_extensions import NotRequired
8
8
 
9
9
 
10
+ UserMessageContentTypedDict = Union[str, List[TextChunkTypedDict]]
11
+
12
+
13
+ UserMessageContent = Union[str, List[TextChunk]]
14
+
15
+
10
16
  UserMessageRole = Literal["user"]
11
17
 
12
18
  class UserMessageTypedDict(TypedDict):
@@ -18,9 +24,3 @@ class UserMessage(BaseModel):
18
24
  content: UserMessageContent
19
25
  role: Optional[UserMessageRole] = "user"
20
26
 
21
-
22
- UserMessageContentTypedDict = Union[str, List[TextChunkTypedDict]]
23
-
24
-
25
- UserMessageContent = Union[str, List[TextChunk]]
26
-
@@ -5,6 +5,12 @@ from mistralai_gcp.types import BaseModel
5
5
  from typing import List, TypedDict, Union
6
6
 
7
7
 
8
+ LocTypedDict = Union[str, int]
9
+
10
+
11
+ Loc = Union[str, int]
12
+
13
+
8
14
  class ValidationErrorTypedDict(TypedDict):
9
15
  loc: List[LocTypedDict]
10
16
  msg: str
@@ -16,9 +22,3 @@ class ValidationError(BaseModel):
16
22
  msg: str
17
23
  type: str
18
24
 
19
-
20
- LocTypedDict = Union[str, int]
21
-
22
-
23
- Loc = Union[str, int]
24
-
@@ -29,9 +29,9 @@ class SDKConfiguration:
29
29
  server: Optional[str] = ""
30
30
  language: str = "python"
31
31
  openapi_doc_version: str = "0.0.2"
32
- sdk_version: str = "1.0.0-alpha.12"
33
- gen_version: str = "2.386.0"
34
- user_agent: str = "speakeasy-sdk/python 1.0.0-alpha.12 2.386.0 0.0.2 mistralai-gcp"
32
+ sdk_version: str = "1.0.0-rc.4"
33
+ gen_version: str = "2.390.6"
34
+ user_agent: str = "speakeasy-sdk/python 1.0.0-rc.4 2.390.6 0.0.2 mistralai-gcp"
35
35
  retry_config: OptionalNullable[RetryConfig] = Field(default_factory=lambda: UNSET)
36
36
  timeout_ms: Optional[int] = None
37
37