mistralai 1.10.1__py3-none-any.whl → 1.11.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 (233) hide show
  1. mistralai/_version.py +3 -3
  2. mistralai/accesses.py +22 -12
  3. mistralai/agents.py +88 -44
  4. mistralai/basesdk.py +6 -0
  5. mistralai/chat.py +96 -40
  6. mistralai/classifiers.py +35 -22
  7. mistralai/conversations.py +186 -64
  8. mistralai/documents.py +72 -26
  9. mistralai/embeddings.py +17 -8
  10. mistralai/files.py +58 -24
  11. mistralai/fim.py +20 -12
  12. mistralai/httpclient.py +0 -1
  13. mistralai/jobs.py +65 -26
  14. mistralai/libraries.py +20 -10
  15. mistralai/mistral_agents.py +438 -30
  16. mistralai/mistral_jobs.py +33 -14
  17. mistralai/models/__init__.py +16 -0
  18. mistralai/models/agent.py +1 -1
  19. mistralai/models/agentconversation.py +1 -1
  20. mistralai/models/agenthandoffdoneevent.py +1 -1
  21. mistralai/models/agenthandoffentry.py +3 -2
  22. mistralai/models/agenthandoffstartedevent.py +1 -1
  23. mistralai/models/agents_api_v1_agents_get_versionop.py +21 -0
  24. mistralai/models/agents_api_v1_agents_list_versionsop.py +33 -0
  25. mistralai/models/agents_api_v1_agents_listop.py +4 -0
  26. mistralai/models/agentscompletionrequest.py +2 -5
  27. mistralai/models/agentscompletionstreamrequest.py +2 -5
  28. mistralai/models/archiveftmodelout.py +1 -1
  29. mistralai/models/assistantmessage.py +1 -1
  30. mistralai/models/audiochunk.py +1 -1
  31. mistralai/models/audioencoding.py +6 -1
  32. mistralai/models/audioformat.py +2 -4
  33. mistralai/models/basemodelcard.py +1 -1
  34. mistralai/models/batchjobin.py +2 -4
  35. mistralai/models/batchjobout.py +1 -1
  36. mistralai/models/batchjobsout.py +1 -1
  37. mistralai/models/chatcompletionchoice.py +10 -5
  38. mistralai/models/chatcompletionrequest.py +2 -5
  39. mistralai/models/chatcompletionstreamrequest.py +2 -5
  40. mistralai/models/classifierdetailedjobout.py +4 -2
  41. mistralai/models/classifierftmodelout.py +3 -2
  42. mistralai/models/classifierjobout.py +4 -2
  43. mistralai/models/codeinterpretertool.py +1 -1
  44. mistralai/models/completiondetailedjobout.py +5 -2
  45. mistralai/models/completionftmodelout.py +3 -2
  46. mistralai/models/completionjobout.py +5 -2
  47. mistralai/models/completionresponsestreamchoice.py +9 -8
  48. mistralai/models/conversationappendrequest.py +4 -1
  49. mistralai/models/conversationappendstreamrequest.py +4 -1
  50. mistralai/models/conversationhistory.py +2 -1
  51. mistralai/models/conversationmessages.py +1 -1
  52. mistralai/models/conversationrequest.py +5 -1
  53. mistralai/models/conversationresponse.py +2 -1
  54. mistralai/models/conversationrestartrequest.py +4 -1
  55. mistralai/models/conversationrestartstreamrequest.py +4 -1
  56. mistralai/models/conversationstreamrequest.py +5 -1
  57. mistralai/models/documentlibrarytool.py +1 -1
  58. mistralai/models/documenturlchunk.py +1 -1
  59. mistralai/models/embeddingdtype.py +7 -1
  60. mistralai/models/encodingformat.py +4 -1
  61. mistralai/models/entitytype.py +8 -1
  62. mistralai/models/filepurpose.py +8 -1
  63. mistralai/models/files_api_routes_list_filesop.py +4 -11
  64. mistralai/models/files_api_routes_upload_fileop.py +2 -6
  65. mistralai/models/fileschema.py +3 -5
  66. mistralai/models/finetuneablemodeltype.py +4 -1
  67. mistralai/models/ftclassifierlossfunction.py +4 -1
  68. mistralai/models/ftmodelcard.py +1 -1
  69. mistralai/models/functioncallentry.py +3 -2
  70. mistralai/models/functioncallevent.py +1 -1
  71. mistralai/models/functionresultentry.py +3 -2
  72. mistralai/models/functiontool.py +1 -1
  73. mistralai/models/githubrepositoryin.py +1 -1
  74. mistralai/models/githubrepositoryout.py +1 -1
  75. mistralai/models/httpvalidationerror.py +4 -2
  76. mistralai/models/imagegenerationtool.py +1 -1
  77. mistralai/models/imageurlchunk.py +1 -1
  78. mistralai/models/jobsout.py +1 -1
  79. mistralai/models/legacyjobmetadataout.py +1 -1
  80. mistralai/models/messageinputentry.py +9 -3
  81. mistralai/models/messageoutputentry.py +6 -3
  82. mistralai/models/messageoutputevent.py +4 -2
  83. mistralai/models/mistralerror.py +11 -7
  84. mistralai/models/mistralpromptmode.py +1 -1
  85. mistralai/models/modelconversation.py +1 -1
  86. mistralai/models/no_response_error.py +5 -1
  87. mistralai/models/ocrrequest.py +11 -1
  88. mistralai/models/ocrtableobject.py +4 -1
  89. mistralai/models/referencechunk.py +1 -1
  90. mistralai/models/requestsource.py +5 -1
  91. mistralai/models/responsedoneevent.py +1 -1
  92. mistralai/models/responseerrorevent.py +1 -1
  93. mistralai/models/responseformats.py +5 -1
  94. mistralai/models/responsestartedevent.py +1 -1
  95. mistralai/models/responsevalidationerror.py +2 -0
  96. mistralai/models/retrievefileout.py +3 -5
  97. mistralai/models/sampletype.py +7 -1
  98. mistralai/models/sdkerror.py +2 -0
  99. mistralai/models/shareenum.py +7 -1
  100. mistralai/models/sharingdelete.py +2 -4
  101. mistralai/models/sharingin.py +3 -5
  102. mistralai/models/source.py +8 -1
  103. mistralai/models/systemmessage.py +1 -1
  104. mistralai/models/textchunk.py +1 -1
  105. mistralai/models/thinkchunk.py +1 -1
  106. mistralai/models/timestampgranularity.py +1 -1
  107. mistralai/models/tool.py +2 -6
  108. mistralai/models/toolcall.py +2 -6
  109. mistralai/models/toolchoice.py +2 -6
  110. mistralai/models/toolchoiceenum.py +6 -1
  111. mistralai/models/toolexecutiondeltaevent.py +2 -1
  112. mistralai/models/toolexecutiondoneevent.py +2 -1
  113. mistralai/models/toolexecutionentry.py +4 -2
  114. mistralai/models/toolexecutionstartedevent.py +2 -1
  115. mistralai/models/toolfilechunk.py +2 -1
  116. mistralai/models/toolmessage.py +1 -1
  117. mistralai/models/toolreferencechunk.py +2 -1
  118. mistralai/models/tooltypes.py +1 -1
  119. mistralai/models/transcriptionsegmentchunk.py +1 -1
  120. mistralai/models/transcriptionstreamdone.py +1 -1
  121. mistralai/models/transcriptionstreamlanguage.py +1 -1
  122. mistralai/models/transcriptionstreamsegmentdelta.py +1 -1
  123. mistralai/models/transcriptionstreamtextdelta.py +1 -1
  124. mistralai/models/unarchiveftmodelout.py +1 -1
  125. mistralai/models/uploadfileout.py +3 -5
  126. mistralai/models/usermessage.py +1 -1
  127. mistralai/models/wandbintegration.py +1 -1
  128. mistralai/models/wandbintegrationout.py +1 -1
  129. mistralai/models/websearchpremiumtool.py +1 -1
  130. mistralai/models/websearchtool.py +1 -1
  131. mistralai/models_.py +24 -12
  132. mistralai/ocr.py +38 -10
  133. mistralai/sdk.py +2 -2
  134. mistralai/transcriptions.py +28 -12
  135. mistralai/types/basemodel.py +41 -3
  136. mistralai/utils/__init__.py +0 -3
  137. mistralai/utils/annotations.py +32 -8
  138. mistralai/utils/enums.py +60 -0
  139. mistralai/utils/forms.py +21 -10
  140. mistralai/utils/queryparams.py +14 -2
  141. mistralai/utils/requestbodies.py +3 -3
  142. mistralai/utils/retries.py +69 -5
  143. mistralai/utils/serializers.py +0 -20
  144. mistralai/utils/unmarshal_json_response.py +15 -1
  145. {mistralai-1.10.1.dist-info → mistralai-1.11.1.dist-info}/METADATA +24 -31
  146. {mistralai-1.10.1.dist-info → mistralai-1.11.1.dist-info}/RECORD +233 -230
  147. mistralai_azure/_version.py +3 -3
  148. mistralai_azure/basesdk.py +6 -0
  149. mistralai_azure/chat.py +27 -15
  150. mistralai_azure/httpclient.py +0 -1
  151. mistralai_azure/models/__init__.py +16 -1
  152. mistralai_azure/models/assistantmessage.py +1 -1
  153. mistralai_azure/models/chatcompletionchoice.py +10 -7
  154. mistralai_azure/models/chatcompletionrequest.py +8 -6
  155. mistralai_azure/models/chatcompletionstreamrequest.py +8 -6
  156. mistralai_azure/models/completionresponsestreamchoice.py +11 -7
  157. mistralai_azure/models/documenturlchunk.py +1 -1
  158. mistralai_azure/models/httpvalidationerror.py +4 -2
  159. mistralai_azure/models/imageurlchunk.py +1 -1
  160. mistralai_azure/models/mistralazureerror.py +11 -7
  161. mistralai_azure/models/mistralpromptmode.py +1 -1
  162. mistralai_azure/models/no_response_error.py +5 -1
  163. mistralai_azure/models/ocrpageobject.py +32 -5
  164. mistralai_azure/models/ocrrequest.py +20 -1
  165. mistralai_azure/models/ocrtableobject.py +34 -0
  166. mistralai_azure/models/referencechunk.py +1 -1
  167. mistralai_azure/models/responseformats.py +5 -1
  168. mistralai_azure/models/responsevalidationerror.py +2 -0
  169. mistralai_azure/models/sdkerror.py +2 -0
  170. mistralai_azure/models/systemmessage.py +1 -1
  171. mistralai_azure/models/textchunk.py +1 -1
  172. mistralai_azure/models/thinkchunk.py +1 -1
  173. mistralai_azure/models/tool.py +2 -6
  174. mistralai_azure/models/toolcall.py +2 -6
  175. mistralai_azure/models/toolchoice.py +2 -6
  176. mistralai_azure/models/toolchoiceenum.py +6 -1
  177. mistralai_azure/models/toolmessage.py +1 -1
  178. mistralai_azure/models/tooltypes.py +1 -1
  179. mistralai_azure/models/usermessage.py +1 -1
  180. mistralai_azure/ocr.py +26 -6
  181. mistralai_azure/types/basemodel.py +41 -3
  182. mistralai_azure/utils/__init__.py +0 -3
  183. mistralai_azure/utils/annotations.py +32 -8
  184. mistralai_azure/utils/enums.py +60 -0
  185. mistralai_azure/utils/forms.py +21 -10
  186. mistralai_azure/utils/queryparams.py +14 -2
  187. mistralai_azure/utils/requestbodies.py +3 -3
  188. mistralai_azure/utils/retries.py +69 -5
  189. mistralai_azure/utils/serializers.py +0 -20
  190. mistralai_azure/utils/unmarshal_json_response.py +15 -1
  191. mistralai_gcp/_version.py +3 -3
  192. mistralai_gcp/basesdk.py +6 -0
  193. mistralai_gcp/chat.py +27 -15
  194. mistralai_gcp/fim.py +27 -15
  195. mistralai_gcp/httpclient.py +0 -1
  196. mistralai_gcp/models/assistantmessage.py +1 -1
  197. mistralai_gcp/models/chatcompletionchoice.py +10 -7
  198. mistralai_gcp/models/chatcompletionrequest.py +8 -6
  199. mistralai_gcp/models/chatcompletionstreamrequest.py +8 -6
  200. mistralai_gcp/models/completionresponsestreamchoice.py +11 -7
  201. mistralai_gcp/models/fimcompletionrequest.py +6 -1
  202. mistralai_gcp/models/fimcompletionstreamrequest.py +6 -1
  203. mistralai_gcp/models/httpvalidationerror.py +4 -2
  204. mistralai_gcp/models/imageurlchunk.py +1 -1
  205. mistralai_gcp/models/mistralgcperror.py +11 -7
  206. mistralai_gcp/models/mistralpromptmode.py +1 -1
  207. mistralai_gcp/models/no_response_error.py +5 -1
  208. mistralai_gcp/models/referencechunk.py +1 -1
  209. mistralai_gcp/models/responseformats.py +5 -1
  210. mistralai_gcp/models/responsevalidationerror.py +2 -0
  211. mistralai_gcp/models/sdkerror.py +2 -0
  212. mistralai_gcp/models/systemmessage.py +1 -1
  213. mistralai_gcp/models/textchunk.py +1 -1
  214. mistralai_gcp/models/thinkchunk.py +1 -1
  215. mistralai_gcp/models/tool.py +2 -6
  216. mistralai_gcp/models/toolcall.py +2 -6
  217. mistralai_gcp/models/toolchoice.py +2 -6
  218. mistralai_gcp/models/toolchoiceenum.py +6 -1
  219. mistralai_gcp/models/toolmessage.py +1 -1
  220. mistralai_gcp/models/tooltypes.py +1 -1
  221. mistralai_gcp/models/usermessage.py +1 -1
  222. mistralai_gcp/types/basemodel.py +41 -3
  223. mistralai_gcp/utils/__init__.py +0 -3
  224. mistralai_gcp/utils/annotations.py +32 -8
  225. mistralai_gcp/utils/enums.py +60 -0
  226. mistralai_gcp/utils/forms.py +21 -10
  227. mistralai_gcp/utils/queryparams.py +14 -2
  228. mistralai_gcp/utils/requestbodies.py +3 -3
  229. mistralai_gcp/utils/retries.py +69 -5
  230. mistralai_gcp/utils/serializers.py +0 -20
  231. mistralai_gcp/utils/unmarshal_json_response.py +15 -1
  232. {mistralai-1.10.1.dist-info → mistralai-1.11.1.dist-info}/WHEEL +0 -0
  233. {mistralai-1.10.1.dist-info → mistralai-1.11.1.dist-info}/licenses/LICENSE +0 -0
mistralai/_version.py CHANGED
@@ -3,10 +3,10 @@
3
3
  import importlib.metadata
4
4
 
5
5
  __title__: str = "mistralai"
6
- __version__: str = "1.10.1"
6
+ __version__: str = "1.11.1"
7
7
  __openapi_doc_version__: str = "1.0.0"
8
- __gen_version__: str = "2.687.13"
9
- __user_agent__: str = "speakeasy-sdk/python 1.10.1 2.687.13 1.0.0 mistralai"
8
+ __gen_version__: str = "2.794.1"
9
+ __user_agent__: str = "speakeasy-sdk/python 1.11.1 2.794.1 1.0.0 mistralai"
10
10
 
11
11
  try:
12
12
  if __package__ is not None:
mistralai/accesses.py CHANGED
@@ -3,6 +3,10 @@
3
3
  from .basesdk import BaseSDK
4
4
  from mistralai import models, utils
5
5
  from mistralai._hooks import HookContext
6
+ from mistralai.models import (
7
+ entitytype as models_entitytype,
8
+ shareenum as models_shareenum,
9
+ )
6
10
  from mistralai.types import OptionalNullable, UNSET
7
11
  from mistralai.utils import get_security_from_env
8
12
  from mistralai.utils.unmarshal_json_response import unmarshal_json_response
@@ -58,6 +62,7 @@ class Accesses(BaseSDK):
58
62
  accept_header_value="application/json",
59
63
  http_headers=http_headers,
60
64
  security=self.sdk_configuration.security,
65
+ allow_empty_value=None,
61
66
  timeout_ms=timeout_ms,
62
67
  )
63
68
 
@@ -74,7 +79,7 @@ class Accesses(BaseSDK):
74
79
  config=self.sdk_configuration,
75
80
  base_url=base_url or "",
76
81
  operation_id="libraries_share_list_v1",
77
- oauth2_scopes=[],
82
+ oauth2_scopes=None,
78
83
  security_source=get_security_from_env(
79
84
  self.sdk_configuration.security, models.Security
80
85
  ),
@@ -147,6 +152,7 @@ class Accesses(BaseSDK):
147
152
  accept_header_value="application/json",
148
153
  http_headers=http_headers,
149
154
  security=self.sdk_configuration.security,
155
+ allow_empty_value=None,
150
156
  timeout_ms=timeout_ms,
151
157
  )
152
158
 
@@ -163,7 +169,7 @@ class Accesses(BaseSDK):
163
169
  config=self.sdk_configuration,
164
170
  base_url=base_url or "",
165
171
  operation_id="libraries_share_list_v1",
166
- oauth2_scopes=[],
172
+ oauth2_scopes=None,
167
173
  security_source=get_security_from_env(
168
174
  self.sdk_configuration.security, models.Security
169
175
  ),
@@ -194,9 +200,9 @@ class Accesses(BaseSDK):
194
200
  self,
195
201
  *,
196
202
  library_id: str,
197
- level: models.ShareEnum,
203
+ level: models_shareenum.ShareEnum,
198
204
  share_with_uuid: str,
199
- share_with_type: models.EntityType,
205
+ share_with_type: models_entitytype.EntityType,
200
206
  org_id: OptionalNullable[str] = UNSET,
201
207
  retries: OptionalNullable[utils.RetryConfig] = UNSET,
202
208
  server_url: Optional[str] = None,
@@ -253,6 +259,7 @@ class Accesses(BaseSDK):
253
259
  get_serialized_body=lambda: utils.serialize_request_body(
254
260
  request.sharing_in, False, False, "json", models.SharingIn
255
261
  ),
262
+ allow_empty_value=None,
256
263
  timeout_ms=timeout_ms,
257
264
  )
258
265
 
@@ -269,7 +276,7 @@ class Accesses(BaseSDK):
269
276
  config=self.sdk_configuration,
270
277
  base_url=base_url or "",
271
278
  operation_id="libraries_share_create_v1",
272
- oauth2_scopes=[],
279
+ oauth2_scopes=None,
273
280
  security_source=get_security_from_env(
274
281
  self.sdk_configuration.security, models.Security
275
282
  ),
@@ -300,9 +307,9 @@ class Accesses(BaseSDK):
300
307
  self,
301
308
  *,
302
309
  library_id: str,
303
- level: models.ShareEnum,
310
+ level: models_shareenum.ShareEnum,
304
311
  share_with_uuid: str,
305
- share_with_type: models.EntityType,
312
+ share_with_type: models_entitytype.EntityType,
306
313
  org_id: OptionalNullable[str] = UNSET,
307
314
  retries: OptionalNullable[utils.RetryConfig] = UNSET,
308
315
  server_url: Optional[str] = None,
@@ -359,6 +366,7 @@ class Accesses(BaseSDK):
359
366
  get_serialized_body=lambda: utils.serialize_request_body(
360
367
  request.sharing_in, False, False, "json", models.SharingIn
361
368
  ),
369
+ allow_empty_value=None,
362
370
  timeout_ms=timeout_ms,
363
371
  )
364
372
 
@@ -375,7 +383,7 @@ class Accesses(BaseSDK):
375
383
  config=self.sdk_configuration,
376
384
  base_url=base_url or "",
377
385
  operation_id="libraries_share_create_v1",
378
- oauth2_scopes=[],
386
+ oauth2_scopes=None,
379
387
  security_source=get_security_from_env(
380
388
  self.sdk_configuration.security, models.Security
381
389
  ),
@@ -407,7 +415,7 @@ class Accesses(BaseSDK):
407
415
  *,
408
416
  library_id: str,
409
417
  share_with_uuid: str,
410
- share_with_type: models.EntityType,
418
+ share_with_type: models_entitytype.EntityType,
411
419
  org_id: OptionalNullable[str] = UNSET,
412
420
  retries: OptionalNullable[utils.RetryConfig] = UNSET,
413
421
  server_url: Optional[str] = None,
@@ -462,6 +470,7 @@ class Accesses(BaseSDK):
462
470
  get_serialized_body=lambda: utils.serialize_request_body(
463
471
  request.sharing_delete, False, False, "json", models.SharingDelete
464
472
  ),
473
+ allow_empty_value=None,
465
474
  timeout_ms=timeout_ms,
466
475
  )
467
476
 
@@ -478,7 +487,7 @@ class Accesses(BaseSDK):
478
487
  config=self.sdk_configuration,
479
488
  base_url=base_url or "",
480
489
  operation_id="libraries_share_delete_v1",
481
- oauth2_scopes=[],
490
+ oauth2_scopes=None,
482
491
  security_source=get_security_from_env(
483
492
  self.sdk_configuration.security, models.Security
484
493
  ),
@@ -510,7 +519,7 @@ class Accesses(BaseSDK):
510
519
  *,
511
520
  library_id: str,
512
521
  share_with_uuid: str,
513
- share_with_type: models.EntityType,
522
+ share_with_type: models_entitytype.EntityType,
514
523
  org_id: OptionalNullable[str] = UNSET,
515
524
  retries: OptionalNullable[utils.RetryConfig] = UNSET,
516
525
  server_url: Optional[str] = None,
@@ -565,6 +574,7 @@ class Accesses(BaseSDK):
565
574
  get_serialized_body=lambda: utils.serialize_request_body(
566
575
  request.sharing_delete, False, False, "json", models.SharingDelete
567
576
  ),
577
+ allow_empty_value=None,
568
578
  timeout_ms=timeout_ms,
569
579
  )
570
580
 
@@ -581,7 +591,7 @@ class Accesses(BaseSDK):
581
591
  config=self.sdk_configuration,
582
592
  base_url=base_url or "",
583
593
  operation_id="libraries_share_delete_v1",
584
- oauth2_scopes=[],
594
+ oauth2_scopes=None,
585
595
  security_source=get_security_from_env(
586
596
  self.sdk_configuration.security, models.Security
587
597
  ),
mistralai/agents.py CHANGED
@@ -3,6 +3,14 @@
3
3
  from .basesdk import BaseSDK
4
4
  from mistralai import models, utils
5
5
  from mistralai._hooks import HookContext
6
+ from mistralai.models import (
7
+ agentscompletionrequest as models_agentscompletionrequest,
8
+ agentscompletionstreamrequest as models_agentscompletionstreamrequest,
9
+ mistralpromptmode as models_mistralpromptmode,
10
+ prediction as models_prediction,
11
+ responseformat as models_responseformat,
12
+ tool as models_tool,
13
+ )
6
14
  from mistralai.types import OptionalNullable, UNSET
7
15
  from mistralai.utils import eventstreaming, get_security_from_env
8
16
  from mistralai.utils.unmarshal_json_response import unmarshal_json_response
@@ -16,40 +24,47 @@ class Agents(BaseSDK):
16
24
  self,
17
25
  *,
18
26
  messages: Union[
19
- List[models.AgentsCompletionRequestMessages],
20
- List[models.AgentsCompletionRequestMessagesTypedDict],
27
+ List[models_agentscompletionrequest.AgentsCompletionRequestMessages],
28
+ List[
29
+ models_agentscompletionrequest.AgentsCompletionRequestMessagesTypedDict
30
+ ],
21
31
  ],
22
32
  agent_id: str,
23
33
  max_tokens: OptionalNullable[int] = UNSET,
24
34
  stream: Optional[bool] = False,
25
35
  stop: Optional[
26
36
  Union[
27
- models.AgentsCompletionRequestStop,
28
- models.AgentsCompletionRequestStopTypedDict,
37
+ models_agentscompletionrequest.AgentsCompletionRequestStop,
38
+ models_agentscompletionrequest.AgentsCompletionRequestStopTypedDict,
29
39
  ]
30
40
  ] = None,
31
41
  random_seed: OptionalNullable[int] = UNSET,
32
42
  metadata: OptionalNullable[Dict[str, Any]] = UNSET,
33
43
  response_format: Optional[
34
- Union[models.ResponseFormat, models.ResponseFormatTypedDict]
44
+ Union[
45
+ models_responseformat.ResponseFormat,
46
+ models_responseformat.ResponseFormatTypedDict,
47
+ ]
35
48
  ] = None,
36
49
  tools: OptionalNullable[
37
- Union[List[models.Tool], List[models.ToolTypedDict]]
50
+ Union[List[models_tool.Tool], List[models_tool.ToolTypedDict]]
38
51
  ] = UNSET,
39
52
  tool_choice: Optional[
40
53
  Union[
41
- models.AgentsCompletionRequestToolChoice,
42
- models.AgentsCompletionRequestToolChoiceTypedDict,
54
+ models_agentscompletionrequest.AgentsCompletionRequestToolChoice,
55
+ models_agentscompletionrequest.AgentsCompletionRequestToolChoiceTypedDict,
43
56
  ]
44
57
  ] = None,
45
58
  presence_penalty: Optional[float] = None,
46
59
  frequency_penalty: Optional[float] = None,
47
60
  n: OptionalNullable[int] = UNSET,
48
61
  prediction: Optional[
49
- Union[models.Prediction, models.PredictionTypedDict]
62
+ Union[models_prediction.Prediction, models_prediction.PredictionTypedDict]
50
63
  ] = None,
51
64
  parallel_tool_calls: Optional[bool] = None,
52
- prompt_mode: OptionalNullable[models.MistralPromptMode] = UNSET,
65
+ prompt_mode: OptionalNullable[
66
+ models_mistralpromptmode.MistralPromptMode
67
+ ] = UNSET,
53
68
  retries: OptionalNullable[utils.RetryConfig] = UNSET,
54
69
  server_url: Optional[str] = None,
55
70
  timeout_ms: Optional[int] = None,
@@ -131,6 +146,7 @@ class Agents(BaseSDK):
131
146
  get_serialized_body=lambda: utils.serialize_request_body(
132
147
  request, False, False, "json", models.AgentsCompletionRequest
133
148
  ),
149
+ allow_empty_value=None,
134
150
  timeout_ms=timeout_ms,
135
151
  )
136
152
 
@@ -147,7 +163,7 @@ class Agents(BaseSDK):
147
163
  config=self.sdk_configuration,
148
164
  base_url=base_url or "",
149
165
  operation_id="agents_completion_v1_agents_completions_post",
150
- oauth2_scopes=[],
166
+ oauth2_scopes=None,
151
167
  security_source=get_security_from_env(
152
168
  self.sdk_configuration.security, models.Security
153
169
  ),
@@ -178,40 +194,47 @@ class Agents(BaseSDK):
178
194
  self,
179
195
  *,
180
196
  messages: Union[
181
- List[models.AgentsCompletionRequestMessages],
182
- List[models.AgentsCompletionRequestMessagesTypedDict],
197
+ List[models_agentscompletionrequest.AgentsCompletionRequestMessages],
198
+ List[
199
+ models_agentscompletionrequest.AgentsCompletionRequestMessagesTypedDict
200
+ ],
183
201
  ],
184
202
  agent_id: str,
185
203
  max_tokens: OptionalNullable[int] = UNSET,
186
204
  stream: Optional[bool] = False,
187
205
  stop: Optional[
188
206
  Union[
189
- models.AgentsCompletionRequestStop,
190
- models.AgentsCompletionRequestStopTypedDict,
207
+ models_agentscompletionrequest.AgentsCompletionRequestStop,
208
+ models_agentscompletionrequest.AgentsCompletionRequestStopTypedDict,
191
209
  ]
192
210
  ] = None,
193
211
  random_seed: OptionalNullable[int] = UNSET,
194
212
  metadata: OptionalNullable[Dict[str, Any]] = UNSET,
195
213
  response_format: Optional[
196
- Union[models.ResponseFormat, models.ResponseFormatTypedDict]
214
+ Union[
215
+ models_responseformat.ResponseFormat,
216
+ models_responseformat.ResponseFormatTypedDict,
217
+ ]
197
218
  ] = None,
198
219
  tools: OptionalNullable[
199
- Union[List[models.Tool], List[models.ToolTypedDict]]
220
+ Union[List[models_tool.Tool], List[models_tool.ToolTypedDict]]
200
221
  ] = UNSET,
201
222
  tool_choice: Optional[
202
223
  Union[
203
- models.AgentsCompletionRequestToolChoice,
204
- models.AgentsCompletionRequestToolChoiceTypedDict,
224
+ models_agentscompletionrequest.AgentsCompletionRequestToolChoice,
225
+ models_agentscompletionrequest.AgentsCompletionRequestToolChoiceTypedDict,
205
226
  ]
206
227
  ] = None,
207
228
  presence_penalty: Optional[float] = None,
208
229
  frequency_penalty: Optional[float] = None,
209
230
  n: OptionalNullable[int] = UNSET,
210
231
  prediction: Optional[
211
- Union[models.Prediction, models.PredictionTypedDict]
232
+ Union[models_prediction.Prediction, models_prediction.PredictionTypedDict]
212
233
  ] = None,
213
234
  parallel_tool_calls: Optional[bool] = None,
214
- prompt_mode: OptionalNullable[models.MistralPromptMode] = UNSET,
235
+ prompt_mode: OptionalNullable[
236
+ models_mistralpromptmode.MistralPromptMode
237
+ ] = UNSET,
215
238
  retries: OptionalNullable[utils.RetryConfig] = UNSET,
216
239
  server_url: Optional[str] = None,
217
240
  timeout_ms: Optional[int] = None,
@@ -293,6 +316,7 @@ class Agents(BaseSDK):
293
316
  get_serialized_body=lambda: utils.serialize_request_body(
294
317
  request, False, False, "json", models.AgentsCompletionRequest
295
318
  ),
319
+ allow_empty_value=None,
296
320
  timeout_ms=timeout_ms,
297
321
  )
298
322
 
@@ -309,7 +333,7 @@ class Agents(BaseSDK):
309
333
  config=self.sdk_configuration,
310
334
  base_url=base_url or "",
311
335
  operation_id="agents_completion_v1_agents_completions_post",
312
- oauth2_scopes=[],
336
+ oauth2_scopes=None,
313
337
  security_source=get_security_from_env(
314
338
  self.sdk_configuration.security, models.Security
315
339
  ),
@@ -340,40 +364,49 @@ class Agents(BaseSDK):
340
364
  self,
341
365
  *,
342
366
  messages: Union[
343
- List[models.AgentsCompletionStreamRequestMessages],
344
- List[models.AgentsCompletionStreamRequestMessagesTypedDict],
367
+ List[
368
+ models_agentscompletionstreamrequest.AgentsCompletionStreamRequestMessages
369
+ ],
370
+ List[
371
+ models_agentscompletionstreamrequest.AgentsCompletionStreamRequestMessagesTypedDict
372
+ ],
345
373
  ],
346
374
  agent_id: str,
347
375
  max_tokens: OptionalNullable[int] = UNSET,
348
376
  stream: Optional[bool] = True,
349
377
  stop: Optional[
350
378
  Union[
351
- models.AgentsCompletionStreamRequestStop,
352
- models.AgentsCompletionStreamRequestStopTypedDict,
379
+ models_agentscompletionstreamrequest.AgentsCompletionStreamRequestStop,
380
+ models_agentscompletionstreamrequest.AgentsCompletionStreamRequestStopTypedDict,
353
381
  ]
354
382
  ] = None,
355
383
  random_seed: OptionalNullable[int] = UNSET,
356
384
  metadata: OptionalNullable[Dict[str, Any]] = UNSET,
357
385
  response_format: Optional[
358
- Union[models.ResponseFormat, models.ResponseFormatTypedDict]
386
+ Union[
387
+ models_responseformat.ResponseFormat,
388
+ models_responseformat.ResponseFormatTypedDict,
389
+ ]
359
390
  ] = None,
360
391
  tools: OptionalNullable[
361
- Union[List[models.Tool], List[models.ToolTypedDict]]
392
+ Union[List[models_tool.Tool], List[models_tool.ToolTypedDict]]
362
393
  ] = UNSET,
363
394
  tool_choice: Optional[
364
395
  Union[
365
- models.AgentsCompletionStreamRequestToolChoice,
366
- models.AgentsCompletionStreamRequestToolChoiceTypedDict,
396
+ models_agentscompletionstreamrequest.AgentsCompletionStreamRequestToolChoice,
397
+ models_agentscompletionstreamrequest.AgentsCompletionStreamRequestToolChoiceTypedDict,
367
398
  ]
368
399
  ] = None,
369
400
  presence_penalty: Optional[float] = None,
370
401
  frequency_penalty: Optional[float] = None,
371
402
  n: OptionalNullable[int] = UNSET,
372
403
  prediction: Optional[
373
- Union[models.Prediction, models.PredictionTypedDict]
404
+ Union[models_prediction.Prediction, models_prediction.PredictionTypedDict]
374
405
  ] = None,
375
406
  parallel_tool_calls: Optional[bool] = None,
376
- prompt_mode: OptionalNullable[models.MistralPromptMode] = UNSET,
407
+ prompt_mode: OptionalNullable[
408
+ models_mistralpromptmode.MistralPromptMode
409
+ ] = UNSET,
377
410
  retries: OptionalNullable[utils.RetryConfig] = UNSET,
378
411
  server_url: Optional[str] = None,
379
412
  timeout_ms: Optional[int] = None,
@@ -457,6 +490,7 @@ class Agents(BaseSDK):
457
490
  get_serialized_body=lambda: utils.serialize_request_body(
458
491
  request, False, False, "json", models.AgentsCompletionStreamRequest
459
492
  ),
493
+ allow_empty_value=None,
460
494
  timeout_ms=timeout_ms,
461
495
  )
462
496
 
@@ -473,7 +507,7 @@ class Agents(BaseSDK):
473
507
  config=self.sdk_configuration,
474
508
  base_url=base_url or "",
475
509
  operation_id="stream_agents",
476
- oauth2_scopes=[],
510
+ oauth2_scopes=None,
477
511
  security_source=get_security_from_env(
478
512
  self.sdk_configuration.security, models.Security
479
513
  ),
@@ -512,40 +546,49 @@ class Agents(BaseSDK):
512
546
  self,
513
547
  *,
514
548
  messages: Union[
515
- List[models.AgentsCompletionStreamRequestMessages],
516
- List[models.AgentsCompletionStreamRequestMessagesTypedDict],
549
+ List[
550
+ models_agentscompletionstreamrequest.AgentsCompletionStreamRequestMessages
551
+ ],
552
+ List[
553
+ models_agentscompletionstreamrequest.AgentsCompletionStreamRequestMessagesTypedDict
554
+ ],
517
555
  ],
518
556
  agent_id: str,
519
557
  max_tokens: OptionalNullable[int] = UNSET,
520
558
  stream: Optional[bool] = True,
521
559
  stop: Optional[
522
560
  Union[
523
- models.AgentsCompletionStreamRequestStop,
524
- models.AgentsCompletionStreamRequestStopTypedDict,
561
+ models_agentscompletionstreamrequest.AgentsCompletionStreamRequestStop,
562
+ models_agentscompletionstreamrequest.AgentsCompletionStreamRequestStopTypedDict,
525
563
  ]
526
564
  ] = None,
527
565
  random_seed: OptionalNullable[int] = UNSET,
528
566
  metadata: OptionalNullable[Dict[str, Any]] = UNSET,
529
567
  response_format: Optional[
530
- Union[models.ResponseFormat, models.ResponseFormatTypedDict]
568
+ Union[
569
+ models_responseformat.ResponseFormat,
570
+ models_responseformat.ResponseFormatTypedDict,
571
+ ]
531
572
  ] = None,
532
573
  tools: OptionalNullable[
533
- Union[List[models.Tool], List[models.ToolTypedDict]]
574
+ Union[List[models_tool.Tool], List[models_tool.ToolTypedDict]]
534
575
  ] = UNSET,
535
576
  tool_choice: Optional[
536
577
  Union[
537
- models.AgentsCompletionStreamRequestToolChoice,
538
- models.AgentsCompletionStreamRequestToolChoiceTypedDict,
578
+ models_agentscompletionstreamrequest.AgentsCompletionStreamRequestToolChoice,
579
+ models_agentscompletionstreamrequest.AgentsCompletionStreamRequestToolChoiceTypedDict,
539
580
  ]
540
581
  ] = None,
541
582
  presence_penalty: Optional[float] = None,
542
583
  frequency_penalty: Optional[float] = None,
543
584
  n: OptionalNullable[int] = UNSET,
544
585
  prediction: Optional[
545
- Union[models.Prediction, models.PredictionTypedDict]
586
+ Union[models_prediction.Prediction, models_prediction.PredictionTypedDict]
546
587
  ] = None,
547
588
  parallel_tool_calls: Optional[bool] = None,
548
- prompt_mode: OptionalNullable[models.MistralPromptMode] = UNSET,
589
+ prompt_mode: OptionalNullable[
590
+ models_mistralpromptmode.MistralPromptMode
591
+ ] = UNSET,
549
592
  retries: OptionalNullable[utils.RetryConfig] = UNSET,
550
593
  server_url: Optional[str] = None,
551
594
  timeout_ms: Optional[int] = None,
@@ -629,6 +672,7 @@ class Agents(BaseSDK):
629
672
  get_serialized_body=lambda: utils.serialize_request_body(
630
673
  request, False, False, "json", models.AgentsCompletionStreamRequest
631
674
  ),
675
+ allow_empty_value=None,
632
676
  timeout_ms=timeout_ms,
633
677
  )
634
678
 
@@ -645,7 +689,7 @@ class Agents(BaseSDK):
645
689
  config=self.sdk_configuration,
646
690
  base_url=base_url or "",
647
691
  operation_id="stream_agents",
648
- oauth2_scopes=[],
692
+ oauth2_scopes=None,
649
693
  security_source=get_security_from_env(
650
694
  self.sdk_configuration.security, models.Security
651
695
  ),
mistralai/basesdk.py CHANGED
@@ -60,6 +60,7 @@ class BaseSDK:
60
60
  ] = None,
61
61
  url_override: Optional[str] = None,
62
62
  http_headers: Optional[Mapping[str, str]] = None,
63
+ allow_empty_value: Optional[List[str]] = None,
63
64
  ) -> httpx.Request:
64
65
  client = self.sdk_configuration.async_client
65
66
  return self._build_request_with_client(
@@ -80,6 +81,7 @@ class BaseSDK:
80
81
  get_serialized_body,
81
82
  url_override,
82
83
  http_headers,
84
+ allow_empty_value,
83
85
  )
84
86
 
85
87
  def _build_request(
@@ -102,6 +104,7 @@ class BaseSDK:
102
104
  ] = None,
103
105
  url_override: Optional[str] = None,
104
106
  http_headers: Optional[Mapping[str, str]] = None,
107
+ allow_empty_value: Optional[List[str]] = None,
105
108
  ) -> httpx.Request:
106
109
  client = self.sdk_configuration.client
107
110
  return self._build_request_with_client(
@@ -122,6 +125,7 @@ class BaseSDK:
122
125
  get_serialized_body,
123
126
  url_override,
124
127
  http_headers,
128
+ allow_empty_value,
125
129
  )
126
130
 
127
131
  def _build_request_with_client(
@@ -145,6 +149,7 @@ class BaseSDK:
145
149
  ] = None,
146
150
  url_override: Optional[str] = None,
147
151
  http_headers: Optional[Mapping[str, str]] = None,
152
+ allow_empty_value: Optional[List[str]] = None,
148
153
  ) -> httpx.Request:
149
154
  query_params = {}
150
155
 
@@ -160,6 +165,7 @@ class BaseSDK:
160
165
  query_params = utils.get_query_params(
161
166
  request if request_has_query_params else None,
162
167
  _globals if request_has_query_params else None,
168
+ allow_empty_value,
163
169
  )
164
170
  else:
165
171
  # Pick up the query parameter from the override so they can be