mistralai 1.5.2rc1__py3-none-any.whl → 1.7.0__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 (81) hide show
  1. mistralai/_version.py +2 -2
  2. mistralai/agents.py +12 -0
  3. mistralai/chat.py +12 -0
  4. mistralai/classifiers.py +435 -23
  5. mistralai/embeddings.py +6 -2
  6. mistralai/jobs.py +84 -38
  7. mistralai/mistral_jobs.py +2 -2
  8. mistralai/models/__init__.py +197 -46
  9. mistralai/models/agentscompletionrequest.py +4 -0
  10. mistralai/models/agentscompletionstreamrequest.py +4 -0
  11. mistralai/models/archiveftmodelout.py +3 -11
  12. mistralai/models/batchjobout.py +3 -9
  13. mistralai/models/batchjobsout.py +3 -9
  14. mistralai/models/chatclassificationrequest.py +20 -0
  15. mistralai/models/chatcompletionrequest.py +4 -0
  16. mistralai/models/chatcompletionstreamrequest.py +4 -0
  17. mistralai/models/chatmoderationrequest.py +4 -7
  18. mistralai/models/classificationresponse.py +12 -9
  19. mistralai/models/classificationtargetresult.py +14 -0
  20. mistralai/models/classifierdetailedjobout.py +156 -0
  21. mistralai/models/classifierftmodelout.py +101 -0
  22. mistralai/models/classifierjobout.py +165 -0
  23. mistralai/models/classifiertargetin.py +55 -0
  24. mistralai/models/classifiertargetout.py +24 -0
  25. mistralai/models/classifiertrainingparameters.py +73 -0
  26. mistralai/models/classifiertrainingparametersin.py +85 -0
  27. mistralai/models/{detailedjobout.py → completiondetailedjobout.py} +34 -34
  28. mistralai/models/{ftmodelout.py → completionftmodelout.py} +12 -12
  29. mistralai/models/{jobout.py → completionjobout.py} +25 -24
  30. mistralai/models/{trainingparameters.py → completiontrainingparameters.py} +7 -7
  31. mistralai/models/{trainingparametersin.py → completiontrainingparametersin.py} +7 -7
  32. mistralai/models/embeddingrequest.py +6 -4
  33. mistralai/models/finetuneablemodeltype.py +7 -0
  34. mistralai/models/ftclassifierlossfunction.py +7 -0
  35. mistralai/models/ftmodelcapabilitiesout.py +3 -0
  36. mistralai/models/function.py +2 -2
  37. mistralai/models/githubrepositoryin.py +3 -11
  38. mistralai/models/githubrepositoryout.py +3 -11
  39. mistralai/models/inputs.py +54 -0
  40. mistralai/models/instructrequest.py +42 -0
  41. mistralai/models/jobin.py +52 -12
  42. mistralai/models/jobs_api_routes_batch_get_batch_jobsop.py +3 -3
  43. mistralai/models/jobs_api_routes_fine_tuning_cancel_fine_tuning_jobop.py +29 -2
  44. mistralai/models/jobs_api_routes_fine_tuning_create_fine_tuning_jobop.py +21 -4
  45. mistralai/models/jobs_api_routes_fine_tuning_get_fine_tuning_jobop.py +29 -2
  46. mistralai/models/jobs_api_routes_fine_tuning_get_fine_tuning_jobsop.py +8 -0
  47. mistralai/models/jobs_api_routes_fine_tuning_start_fine_tuning_jobop.py +29 -2
  48. mistralai/models/jobs_api_routes_fine_tuning_update_fine_tuned_modelop.py +28 -2
  49. mistralai/models/jobsout.py +24 -13
  50. mistralai/models/jsonschema.py +1 -1
  51. mistralai/models/legacyjobmetadataout.py +3 -12
  52. mistralai/models/{classificationobject.py → moderationobject.py} +6 -6
  53. mistralai/models/moderationresponse.py +21 -0
  54. mistralai/models/unarchiveftmodelout.py +3 -11
  55. mistralai/models/wandbintegration.py +3 -11
  56. mistralai/models/wandbintegrationout.py +8 -13
  57. mistralai/models_.py +10 -4
  58. {mistralai-1.5.2rc1.dist-info → mistralai-1.7.0.dist-info}/METADATA +4 -2
  59. {mistralai-1.5.2rc1.dist-info → mistralai-1.7.0.dist-info}/RECORD +81 -63
  60. {mistralai-1.5.2rc1.dist-info → mistralai-1.7.0.dist-info}/WHEEL +1 -1
  61. mistralai_azure/_version.py +2 -2
  62. mistralai_azure/chat.py +12 -0
  63. mistralai_azure/models/__init__.py +15 -0
  64. mistralai_azure/models/chatcompletionrequest.py +4 -0
  65. mistralai_azure/models/chatcompletionstreamrequest.py +4 -0
  66. mistralai_azure/models/contentchunk.py +6 -2
  67. mistralai_azure/models/function.py +2 -2
  68. mistralai_azure/models/imageurl.py +53 -0
  69. mistralai_azure/models/imageurlchunk.py +33 -0
  70. mistralai_azure/models/jsonschema.py +1 -1
  71. mistralai_gcp/_version.py +2 -2
  72. mistralai_gcp/chat.py +12 -0
  73. mistralai_gcp/models/__init__.py +15 -0
  74. mistralai_gcp/models/chatcompletionrequest.py +4 -0
  75. mistralai_gcp/models/chatcompletionstreamrequest.py +4 -0
  76. mistralai_gcp/models/contentchunk.py +6 -2
  77. mistralai_gcp/models/function.py +2 -2
  78. mistralai_gcp/models/imageurl.py +53 -0
  79. mistralai_gcp/models/imageurlchunk.py +33 -0
  80. mistralai_gcp/models/jsonschema.py +1 -1
  81. {mistralai-1.5.2rc1.dist-info → mistralai-1.7.0.dist-info}/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.5.2-rc.1"
6
+ __version__: str = "1.7.0"
7
7
  __openapi_doc_version__: str = "0.0.2"
8
8
  __gen_version__: str = "2.548.6"
9
- __user_agent__: str = "speakeasy-sdk/python 1.5.2-rc.1 2.548.6 0.0.2 mistralai"
9
+ __user_agent__: str = "speakeasy-sdk/python 1.7.0 2.548.6 0.0.2 mistralai"
10
10
 
11
11
  try:
12
12
  if __package__ is not None:
mistralai/agents.py CHANGED
@@ -46,6 +46,7 @@ class Agents(BaseSDK):
46
46
  prediction: Optional[
47
47
  Union[models.Prediction, models.PredictionTypedDict]
48
48
  ] = None,
49
+ parallel_tool_calls: Optional[bool] = None,
49
50
  retries: OptionalNullable[utils.RetryConfig] = UNSET,
50
51
  server_url: Optional[str] = None,
51
52
  timeout_ms: Optional[int] = None,
@@ -66,6 +67,7 @@ class Agents(BaseSDK):
66
67
  :param frequency_penalty: frequency_penalty penalizes the repetition of words based on their frequency in the generated text. A higher frequency penalty discourages the model from repeating words that have already appeared frequently in the output, promoting diversity and reducing repetition.
67
68
  :param n: Number of completions to return for each request, input tokens are only billed once.
68
69
  :param prediction:
70
+ :param parallel_tool_calls:
69
71
  :param retries: Override the default retry configuration for this method
70
72
  :param server_url: Override the default server URL for this method
71
73
  :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
@@ -102,6 +104,7 @@ class Agents(BaseSDK):
102
104
  prediction=utils.get_pydantic_model(
103
105
  prediction, Optional[models.Prediction]
104
106
  ),
107
+ parallel_tool_calls=parallel_tool_calls,
105
108
  agent_id=agent_id,
106
109
  )
107
110
 
@@ -209,6 +212,7 @@ class Agents(BaseSDK):
209
212
  prediction: Optional[
210
213
  Union[models.Prediction, models.PredictionTypedDict]
211
214
  ] = None,
215
+ parallel_tool_calls: Optional[bool] = None,
212
216
  retries: OptionalNullable[utils.RetryConfig] = UNSET,
213
217
  server_url: Optional[str] = None,
214
218
  timeout_ms: Optional[int] = None,
@@ -229,6 +233,7 @@ class Agents(BaseSDK):
229
233
  :param frequency_penalty: frequency_penalty penalizes the repetition of words based on their frequency in the generated text. A higher frequency penalty discourages the model from repeating words that have already appeared frequently in the output, promoting diversity and reducing repetition.
230
234
  :param n: Number of completions to return for each request, input tokens are only billed once.
231
235
  :param prediction:
236
+ :param parallel_tool_calls:
232
237
  :param retries: Override the default retry configuration for this method
233
238
  :param server_url: Override the default server URL for this method
234
239
  :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
@@ -265,6 +270,7 @@ class Agents(BaseSDK):
265
270
  prediction=utils.get_pydantic_model(
266
271
  prediction, Optional[models.Prediction]
267
272
  ),
273
+ parallel_tool_calls=parallel_tool_calls,
268
274
  agent_id=agent_id,
269
275
  )
270
276
 
@@ -372,6 +378,7 @@ class Agents(BaseSDK):
372
378
  prediction: Optional[
373
379
  Union[models.Prediction, models.PredictionTypedDict]
374
380
  ] = None,
381
+ parallel_tool_calls: Optional[bool] = None,
375
382
  retries: OptionalNullable[utils.RetryConfig] = UNSET,
376
383
  server_url: Optional[str] = None,
377
384
  timeout_ms: Optional[int] = None,
@@ -394,6 +401,7 @@ class Agents(BaseSDK):
394
401
  :param frequency_penalty: frequency_penalty penalizes the repetition of words based on their frequency in the generated text. A higher frequency penalty discourages the model from repeating words that have already appeared frequently in the output, promoting diversity and reducing repetition.
395
402
  :param n: Number of completions to return for each request, input tokens are only billed once.
396
403
  :param prediction:
404
+ :param parallel_tool_calls:
397
405
  :param retries: Override the default retry configuration for this method
398
406
  :param server_url: Override the default server URL for this method
399
407
  :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
@@ -430,6 +438,7 @@ class Agents(BaseSDK):
430
438
  prediction=utils.get_pydantic_model(
431
439
  prediction, Optional[models.Prediction]
432
440
  ),
441
+ parallel_tool_calls=parallel_tool_calls,
433
442
  agent_id=agent_id,
434
443
  )
435
444
 
@@ -543,6 +552,7 @@ class Agents(BaseSDK):
543
552
  prediction: Optional[
544
553
  Union[models.Prediction, models.PredictionTypedDict]
545
554
  ] = None,
555
+ parallel_tool_calls: Optional[bool] = None,
546
556
  retries: OptionalNullable[utils.RetryConfig] = UNSET,
547
557
  server_url: Optional[str] = None,
548
558
  timeout_ms: Optional[int] = None,
@@ -565,6 +575,7 @@ class Agents(BaseSDK):
565
575
  :param frequency_penalty: frequency_penalty penalizes the repetition of words based on their frequency in the generated text. A higher frequency penalty discourages the model from repeating words that have already appeared frequently in the output, promoting diversity and reducing repetition.
566
576
  :param n: Number of completions to return for each request, input tokens are only billed once.
567
577
  :param prediction:
578
+ :param parallel_tool_calls:
568
579
  :param retries: Override the default retry configuration for this method
569
580
  :param server_url: Override the default server URL for this method
570
581
  :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
@@ -601,6 +612,7 @@ class Agents(BaseSDK):
601
612
  prediction=utils.get_pydantic_model(
602
613
  prediction, Optional[models.Prediction]
603
614
  ),
615
+ parallel_tool_calls=parallel_tool_calls,
604
616
  agent_id=agent_id,
605
617
  )
606
618
 
mistralai/chat.py CHANGED
@@ -122,6 +122,7 @@ class Chat(BaseSDK):
122
122
  prediction: Optional[
123
123
  Union[models.Prediction, models.PredictionTypedDict]
124
124
  ] = None,
125
+ parallel_tool_calls: Optional[bool] = None,
125
126
  safe_prompt: Optional[bool] = None,
126
127
  retries: OptionalNullable[utils.RetryConfig] = UNSET,
127
128
  server_url: Optional[str] = None,
@@ -145,6 +146,7 @@ class Chat(BaseSDK):
145
146
  :param frequency_penalty: frequency_penalty penalizes the repetition of words based on their frequency in the generated text. A higher frequency penalty discourages the model from repeating words that have already appeared frequently in the output, promoting diversity and reducing repetition.
146
147
  :param n: Number of completions to return for each request, input tokens are only billed once.
147
148
  :param prediction:
149
+ :param parallel_tool_calls:
148
150
  :param safe_prompt: Whether to inject a safety prompt before all conversations.
149
151
  :param retries: Override the default retry configuration for this method
150
152
  :param server_url: Override the default server URL for this method
@@ -183,6 +185,7 @@ class Chat(BaseSDK):
183
185
  prediction=utils.get_pydantic_model(
184
186
  prediction, Optional[models.Prediction]
185
187
  ),
188
+ parallel_tool_calls=parallel_tool_calls,
186
189
  safe_prompt=safe_prompt,
187
190
  )
188
191
 
@@ -284,6 +287,7 @@ class Chat(BaseSDK):
284
287
  prediction: Optional[
285
288
  Union[models.Prediction, models.PredictionTypedDict]
286
289
  ] = None,
290
+ parallel_tool_calls: Optional[bool] = None,
287
291
  safe_prompt: Optional[bool] = None,
288
292
  retries: OptionalNullable[utils.RetryConfig] = UNSET,
289
293
  server_url: Optional[str] = None,
@@ -307,6 +311,7 @@ class Chat(BaseSDK):
307
311
  :param frequency_penalty: frequency_penalty penalizes the repetition of words based on their frequency in the generated text. A higher frequency penalty discourages the model from repeating words that have already appeared frequently in the output, promoting diversity and reducing repetition.
308
312
  :param n: Number of completions to return for each request, input tokens are only billed once.
309
313
  :param prediction:
314
+ :param parallel_tool_calls:
310
315
  :param safe_prompt: Whether to inject a safety prompt before all conversations.
311
316
  :param retries: Override the default retry configuration for this method
312
317
  :param server_url: Override the default server URL for this method
@@ -345,6 +350,7 @@ class Chat(BaseSDK):
345
350
  prediction=utils.get_pydantic_model(
346
351
  prediction, Optional[models.Prediction]
347
352
  ),
353
+ parallel_tool_calls=parallel_tool_calls,
348
354
  safe_prompt=safe_prompt,
349
355
  )
350
356
 
@@ -454,6 +460,7 @@ class Chat(BaseSDK):
454
460
  prediction: Optional[
455
461
  Union[models.Prediction, models.PredictionTypedDict]
456
462
  ] = None,
463
+ parallel_tool_calls: Optional[bool] = None,
457
464
  safe_prompt: Optional[bool] = None,
458
465
  retries: OptionalNullable[utils.RetryConfig] = UNSET,
459
466
  server_url: Optional[str] = None,
@@ -479,6 +486,7 @@ class Chat(BaseSDK):
479
486
  :param frequency_penalty: frequency_penalty penalizes the repetition of words based on their frequency in the generated text. A higher frequency penalty discourages the model from repeating words that have already appeared frequently in the output, promoting diversity and reducing repetition.
480
487
  :param n: Number of completions to return for each request, input tokens are only billed once.
481
488
  :param prediction:
489
+ :param parallel_tool_calls:
482
490
  :param safe_prompt: Whether to inject a safety prompt before all conversations.
483
491
  :param retries: Override the default retry configuration for this method
484
492
  :param server_url: Override the default server URL for this method
@@ -519,6 +527,7 @@ class Chat(BaseSDK):
519
527
  prediction=utils.get_pydantic_model(
520
528
  prediction, Optional[models.Prediction]
521
529
  ),
530
+ parallel_tool_calls=parallel_tool_calls,
522
531
  safe_prompt=safe_prompt,
523
532
  )
524
533
 
@@ -634,6 +643,7 @@ class Chat(BaseSDK):
634
643
  prediction: Optional[
635
644
  Union[models.Prediction, models.PredictionTypedDict]
636
645
  ] = None,
646
+ parallel_tool_calls: Optional[bool] = None,
637
647
  safe_prompt: Optional[bool] = None,
638
648
  retries: OptionalNullable[utils.RetryConfig] = UNSET,
639
649
  server_url: Optional[str] = None,
@@ -659,6 +669,7 @@ class Chat(BaseSDK):
659
669
  :param frequency_penalty: frequency_penalty penalizes the repetition of words based on their frequency in the generated text. A higher frequency penalty discourages the model from repeating words that have already appeared frequently in the output, promoting diversity and reducing repetition.
660
670
  :param n: Number of completions to return for each request, input tokens are only billed once.
661
671
  :param prediction:
672
+ :param parallel_tool_calls:
662
673
  :param safe_prompt: Whether to inject a safety prompt before all conversations.
663
674
  :param retries: Override the default retry configuration for this method
664
675
  :param server_url: Override the default server URL for this method
@@ -699,6 +710,7 @@ class Chat(BaseSDK):
699
710
  prediction=utils.get_pydantic_model(
700
711
  prediction, Optional[models.Prediction]
701
712
  ),
713
+ parallel_tool_calls=parallel_tool_calls,
702
714
  safe_prompt=safe_prompt,
703
715
  )
704
716