mistralai 1.2.4__py3-none-any.whl → 1.2.6__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.
- mistralai/_hooks/custom_user_agent.py +7 -1
- mistralai/_version.py +1 -1
- mistralai/agents.py +17 -5
- mistralai/basesdk.py +25 -9
- mistralai/chat.py +25 -13
- mistralai/classifiers.py +17 -5
- mistralai/embeddings.py +9 -3
- mistralai/files.py +49 -13
- mistralai/fim.py +17 -5
- mistralai/jobs.py +41 -11
- mistralai/mistral_jobs.py +33 -9
- mistralai/models/chatcompletionrequest.py +2 -2
- mistralai/models/chatcompletionstreamrequest.py +2 -2
- mistralai/models/files_api_routes_upload_fileop.py +2 -6
- mistralai/models_.py +49 -13
- mistralai/sdkconfiguration.py +3 -3
- mistralai/utils/forms.py +4 -10
- mistralai/utils/requestbodies.py +1 -1
- {mistralai-1.2.4.dist-info → mistralai-1.2.6.dist-info}/METADATA +117 -86
- {mistralai-1.2.4.dist-info → mistralai-1.2.6.dist-info}/RECORD +41 -41
- {mistralai-1.2.4.dist-info → mistralai-1.2.6.dist-info}/WHEEL +1 -1
- mistralai_azure/_hooks/custom_user_agent.py +8 -2
- mistralai_azure/_version.py +1 -1
- mistralai_azure/basesdk.py +25 -9
- mistralai_azure/chat.py +25 -13
- mistralai_azure/models/chatcompletionrequest.py +2 -2
- mistralai_azure/models/chatcompletionstreamrequest.py +2 -2
- mistralai_azure/sdkconfiguration.py +3 -3
- mistralai_azure/utils/forms.py +4 -10
- mistralai_azure/utils/requestbodies.py +1 -1
- mistralai_gcp/_hooks/custom_user_agent.py +7 -1
- mistralai_gcp/_version.py +1 -1
- mistralai_gcp/basesdk.py +25 -9
- mistralai_gcp/chat.py +21 -9
- mistralai_gcp/fim.py +17 -5
- mistralai_gcp/models/chatcompletionrequest.py +1 -1
- mistralai_gcp/models/chatcompletionstreamrequest.py +1 -1
- mistralai_gcp/sdkconfiguration.py +3 -3
- mistralai_gcp/utils/forms.py +4 -10
- mistralai_gcp/utils/requestbodies.py +1 -1
- {mistralai-1.2.4.dist-info → mistralai-1.2.6.dist-info}/LICENSE +0 -0
mistralai/jobs.py
CHANGED
|
@@ -6,7 +6,7 @@ from mistralai import models, utils
|
|
|
6
6
|
from mistralai._hooks import HookContext
|
|
7
7
|
from mistralai.types import OptionalNullable, UNSET
|
|
8
8
|
from mistralai.utils import get_security_from_env
|
|
9
|
-
from typing import List, Optional, Union
|
|
9
|
+
from typing import List, Mapping, Optional, Union
|
|
10
10
|
|
|
11
11
|
|
|
12
12
|
class Jobs(BaseSDK):
|
|
@@ -25,6 +25,7 @@ class Jobs(BaseSDK):
|
|
|
25
25
|
retries: OptionalNullable[utils.RetryConfig] = UNSET,
|
|
26
26
|
server_url: Optional[str] = None,
|
|
27
27
|
timeout_ms: Optional[int] = None,
|
|
28
|
+
http_headers: Optional[Mapping[str, str]] = None,
|
|
28
29
|
) -> Optional[models.JobsOut]:
|
|
29
30
|
r"""Get Fine Tuning Jobs
|
|
30
31
|
|
|
@@ -42,6 +43,7 @@ class Jobs(BaseSDK):
|
|
|
42
43
|
:param retries: Override the default retry configuration for this method
|
|
43
44
|
:param server_url: Override the default server URL for this method
|
|
44
45
|
:param timeout_ms: Override the default request timeout configuration for this method in milliseconds
|
|
46
|
+
:param http_headers: Additional headers to set or replace on requests.
|
|
45
47
|
"""
|
|
46
48
|
base_url = None
|
|
47
49
|
url_variables = None
|
|
@@ -63,7 +65,7 @@ class Jobs(BaseSDK):
|
|
|
63
65
|
suffix=suffix,
|
|
64
66
|
)
|
|
65
67
|
|
|
66
|
-
req = self.
|
|
68
|
+
req = self._build_request(
|
|
67
69
|
method="GET",
|
|
68
70
|
path="/v1/fine_tuning/jobs",
|
|
69
71
|
base_url=base_url,
|
|
@@ -74,6 +76,7 @@ class Jobs(BaseSDK):
|
|
|
74
76
|
request_has_query_params=True,
|
|
75
77
|
user_agent_header="user-agent",
|
|
76
78
|
accept_header_value="application/json",
|
|
79
|
+
http_headers=http_headers,
|
|
77
80
|
security=self.sdk_configuration.security,
|
|
78
81
|
timeout_ms=timeout_ms,
|
|
79
82
|
)
|
|
@@ -131,6 +134,7 @@ class Jobs(BaseSDK):
|
|
|
131
134
|
retries: OptionalNullable[utils.RetryConfig] = UNSET,
|
|
132
135
|
server_url: Optional[str] = None,
|
|
133
136
|
timeout_ms: Optional[int] = None,
|
|
137
|
+
http_headers: Optional[Mapping[str, str]] = None,
|
|
134
138
|
) -> Optional[models.JobsOut]:
|
|
135
139
|
r"""Get Fine Tuning Jobs
|
|
136
140
|
|
|
@@ -148,6 +152,7 @@ class Jobs(BaseSDK):
|
|
|
148
152
|
:param retries: Override the default retry configuration for this method
|
|
149
153
|
:param server_url: Override the default server URL for this method
|
|
150
154
|
:param timeout_ms: Override the default request timeout configuration for this method in milliseconds
|
|
155
|
+
:param http_headers: Additional headers to set or replace on requests.
|
|
151
156
|
"""
|
|
152
157
|
base_url = None
|
|
153
158
|
url_variables = None
|
|
@@ -169,7 +174,7 @@ class Jobs(BaseSDK):
|
|
|
169
174
|
suffix=suffix,
|
|
170
175
|
)
|
|
171
176
|
|
|
172
|
-
req = self.
|
|
177
|
+
req = self._build_request_async(
|
|
173
178
|
method="GET",
|
|
174
179
|
path="/v1/fine_tuning/jobs",
|
|
175
180
|
base_url=base_url,
|
|
@@ -180,6 +185,7 @@ class Jobs(BaseSDK):
|
|
|
180
185
|
request_has_query_params=True,
|
|
181
186
|
user_agent_header="user-agent",
|
|
182
187
|
accept_header_value="application/json",
|
|
188
|
+
http_headers=http_headers,
|
|
183
189
|
security=self.sdk_configuration.security,
|
|
184
190
|
timeout_ms=timeout_ms,
|
|
185
191
|
)
|
|
@@ -248,6 +254,7 @@ class Jobs(BaseSDK):
|
|
|
248
254
|
retries: OptionalNullable[utils.RetryConfig] = UNSET,
|
|
249
255
|
server_url: Optional[str] = None,
|
|
250
256
|
timeout_ms: Optional[int] = None,
|
|
257
|
+
http_headers: Optional[Mapping[str, str]] = None,
|
|
251
258
|
) -> Optional[models.JobsAPIRoutesFineTuningCreateFineTuningJobResponse]:
|
|
252
259
|
r"""Create Fine Tuning Job
|
|
253
260
|
|
|
@@ -264,6 +271,7 @@ class Jobs(BaseSDK):
|
|
|
264
271
|
:param retries: Override the default retry configuration for this method
|
|
265
272
|
:param server_url: Override the default server URL for this method
|
|
266
273
|
:param timeout_ms: Override the default request timeout configuration for this method in milliseconds
|
|
274
|
+
:param http_headers: Additional headers to set or replace on requests.
|
|
267
275
|
"""
|
|
268
276
|
base_url = None
|
|
269
277
|
url_variables = None
|
|
@@ -292,7 +300,7 @@ class Jobs(BaseSDK):
|
|
|
292
300
|
auto_start=auto_start,
|
|
293
301
|
)
|
|
294
302
|
|
|
295
|
-
req = self.
|
|
303
|
+
req = self._build_request(
|
|
296
304
|
method="POST",
|
|
297
305
|
path="/v1/fine_tuning/jobs",
|
|
298
306
|
base_url=base_url,
|
|
@@ -303,6 +311,7 @@ class Jobs(BaseSDK):
|
|
|
303
311
|
request_has_query_params=True,
|
|
304
312
|
user_agent_header="user-agent",
|
|
305
313
|
accept_header_value="application/json",
|
|
314
|
+
http_headers=http_headers,
|
|
306
315
|
security=self.sdk_configuration.security,
|
|
307
316
|
get_serialized_body=lambda: utils.serialize_request_body(
|
|
308
317
|
request, False, False, "json", models.JobIn
|
|
@@ -377,6 +386,7 @@ class Jobs(BaseSDK):
|
|
|
377
386
|
retries: OptionalNullable[utils.RetryConfig] = UNSET,
|
|
378
387
|
server_url: Optional[str] = None,
|
|
379
388
|
timeout_ms: Optional[int] = None,
|
|
389
|
+
http_headers: Optional[Mapping[str, str]] = None,
|
|
380
390
|
) -> Optional[models.JobsAPIRoutesFineTuningCreateFineTuningJobResponse]:
|
|
381
391
|
r"""Create Fine Tuning Job
|
|
382
392
|
|
|
@@ -393,6 +403,7 @@ class Jobs(BaseSDK):
|
|
|
393
403
|
:param retries: Override the default retry configuration for this method
|
|
394
404
|
:param server_url: Override the default server URL for this method
|
|
395
405
|
:param timeout_ms: Override the default request timeout configuration for this method in milliseconds
|
|
406
|
+
:param http_headers: Additional headers to set or replace on requests.
|
|
396
407
|
"""
|
|
397
408
|
base_url = None
|
|
398
409
|
url_variables = None
|
|
@@ -421,7 +432,7 @@ class Jobs(BaseSDK):
|
|
|
421
432
|
auto_start=auto_start,
|
|
422
433
|
)
|
|
423
434
|
|
|
424
|
-
req = self.
|
|
435
|
+
req = self._build_request_async(
|
|
425
436
|
method="POST",
|
|
426
437
|
path="/v1/fine_tuning/jobs",
|
|
427
438
|
base_url=base_url,
|
|
@@ -432,6 +443,7 @@ class Jobs(BaseSDK):
|
|
|
432
443
|
request_has_query_params=True,
|
|
433
444
|
user_agent_header="user-agent",
|
|
434
445
|
accept_header_value="application/json",
|
|
446
|
+
http_headers=http_headers,
|
|
435
447
|
security=self.sdk_configuration.security,
|
|
436
448
|
get_serialized_body=lambda: utils.serialize_request_body(
|
|
437
449
|
request, False, False, "json", models.JobIn
|
|
@@ -487,6 +499,7 @@ class Jobs(BaseSDK):
|
|
|
487
499
|
retries: OptionalNullable[utils.RetryConfig] = UNSET,
|
|
488
500
|
server_url: Optional[str] = None,
|
|
489
501
|
timeout_ms: Optional[int] = None,
|
|
502
|
+
http_headers: Optional[Mapping[str, str]] = None,
|
|
490
503
|
) -> Optional[models.DetailedJobOut]:
|
|
491
504
|
r"""Get Fine Tuning Job
|
|
492
505
|
|
|
@@ -496,6 +509,7 @@ class Jobs(BaseSDK):
|
|
|
496
509
|
:param retries: Override the default retry configuration for this method
|
|
497
510
|
:param server_url: Override the default server URL for this method
|
|
498
511
|
:param timeout_ms: Override the default request timeout configuration for this method in milliseconds
|
|
512
|
+
:param http_headers: Additional headers to set or replace on requests.
|
|
499
513
|
"""
|
|
500
514
|
base_url = None
|
|
501
515
|
url_variables = None
|
|
@@ -509,7 +523,7 @@ class Jobs(BaseSDK):
|
|
|
509
523
|
job_id=job_id,
|
|
510
524
|
)
|
|
511
525
|
|
|
512
|
-
req = self.
|
|
526
|
+
req = self._build_request(
|
|
513
527
|
method="GET",
|
|
514
528
|
path="/v1/fine_tuning/jobs/{job_id}",
|
|
515
529
|
base_url=base_url,
|
|
@@ -520,6 +534,7 @@ class Jobs(BaseSDK):
|
|
|
520
534
|
request_has_query_params=True,
|
|
521
535
|
user_agent_header="user-agent",
|
|
522
536
|
accept_header_value="application/json",
|
|
537
|
+
http_headers=http_headers,
|
|
523
538
|
security=self.sdk_configuration.security,
|
|
524
539
|
timeout_ms=timeout_ms,
|
|
525
540
|
)
|
|
@@ -569,6 +584,7 @@ class Jobs(BaseSDK):
|
|
|
569
584
|
retries: OptionalNullable[utils.RetryConfig] = UNSET,
|
|
570
585
|
server_url: Optional[str] = None,
|
|
571
586
|
timeout_ms: Optional[int] = None,
|
|
587
|
+
http_headers: Optional[Mapping[str, str]] = None,
|
|
572
588
|
) -> Optional[models.DetailedJobOut]:
|
|
573
589
|
r"""Get Fine Tuning Job
|
|
574
590
|
|
|
@@ -578,6 +594,7 @@ class Jobs(BaseSDK):
|
|
|
578
594
|
:param retries: Override the default retry configuration for this method
|
|
579
595
|
:param server_url: Override the default server URL for this method
|
|
580
596
|
:param timeout_ms: Override the default request timeout configuration for this method in milliseconds
|
|
597
|
+
:param http_headers: Additional headers to set or replace on requests.
|
|
581
598
|
"""
|
|
582
599
|
base_url = None
|
|
583
600
|
url_variables = None
|
|
@@ -591,7 +608,7 @@ class Jobs(BaseSDK):
|
|
|
591
608
|
job_id=job_id,
|
|
592
609
|
)
|
|
593
610
|
|
|
594
|
-
req = self.
|
|
611
|
+
req = self._build_request_async(
|
|
595
612
|
method="GET",
|
|
596
613
|
path="/v1/fine_tuning/jobs/{job_id}",
|
|
597
614
|
base_url=base_url,
|
|
@@ -602,6 +619,7 @@ class Jobs(BaseSDK):
|
|
|
602
619
|
request_has_query_params=True,
|
|
603
620
|
user_agent_header="user-agent",
|
|
604
621
|
accept_header_value="application/json",
|
|
622
|
+
http_headers=http_headers,
|
|
605
623
|
security=self.sdk_configuration.security,
|
|
606
624
|
timeout_ms=timeout_ms,
|
|
607
625
|
)
|
|
@@ -651,6 +669,7 @@ class Jobs(BaseSDK):
|
|
|
651
669
|
retries: OptionalNullable[utils.RetryConfig] = UNSET,
|
|
652
670
|
server_url: Optional[str] = None,
|
|
653
671
|
timeout_ms: Optional[int] = None,
|
|
672
|
+
http_headers: Optional[Mapping[str, str]] = None,
|
|
654
673
|
) -> Optional[models.DetailedJobOut]:
|
|
655
674
|
r"""Cancel Fine Tuning Job
|
|
656
675
|
|
|
@@ -660,6 +679,7 @@ class Jobs(BaseSDK):
|
|
|
660
679
|
:param retries: Override the default retry configuration for this method
|
|
661
680
|
:param server_url: Override the default server URL for this method
|
|
662
681
|
:param timeout_ms: Override the default request timeout configuration for this method in milliseconds
|
|
682
|
+
:param http_headers: Additional headers to set or replace on requests.
|
|
663
683
|
"""
|
|
664
684
|
base_url = None
|
|
665
685
|
url_variables = None
|
|
@@ -673,7 +693,7 @@ class Jobs(BaseSDK):
|
|
|
673
693
|
job_id=job_id,
|
|
674
694
|
)
|
|
675
695
|
|
|
676
|
-
req = self.
|
|
696
|
+
req = self._build_request(
|
|
677
697
|
method="POST",
|
|
678
698
|
path="/v1/fine_tuning/jobs/{job_id}/cancel",
|
|
679
699
|
base_url=base_url,
|
|
@@ -684,6 +704,7 @@ class Jobs(BaseSDK):
|
|
|
684
704
|
request_has_query_params=True,
|
|
685
705
|
user_agent_header="user-agent",
|
|
686
706
|
accept_header_value="application/json",
|
|
707
|
+
http_headers=http_headers,
|
|
687
708
|
security=self.sdk_configuration.security,
|
|
688
709
|
timeout_ms=timeout_ms,
|
|
689
710
|
)
|
|
@@ -733,6 +754,7 @@ class Jobs(BaseSDK):
|
|
|
733
754
|
retries: OptionalNullable[utils.RetryConfig] = UNSET,
|
|
734
755
|
server_url: Optional[str] = None,
|
|
735
756
|
timeout_ms: Optional[int] = None,
|
|
757
|
+
http_headers: Optional[Mapping[str, str]] = None,
|
|
736
758
|
) -> Optional[models.DetailedJobOut]:
|
|
737
759
|
r"""Cancel Fine Tuning Job
|
|
738
760
|
|
|
@@ -742,6 +764,7 @@ class Jobs(BaseSDK):
|
|
|
742
764
|
:param retries: Override the default retry configuration for this method
|
|
743
765
|
:param server_url: Override the default server URL for this method
|
|
744
766
|
:param timeout_ms: Override the default request timeout configuration for this method in milliseconds
|
|
767
|
+
:param http_headers: Additional headers to set or replace on requests.
|
|
745
768
|
"""
|
|
746
769
|
base_url = None
|
|
747
770
|
url_variables = None
|
|
@@ -755,7 +778,7 @@ class Jobs(BaseSDK):
|
|
|
755
778
|
job_id=job_id,
|
|
756
779
|
)
|
|
757
780
|
|
|
758
|
-
req = self.
|
|
781
|
+
req = self._build_request_async(
|
|
759
782
|
method="POST",
|
|
760
783
|
path="/v1/fine_tuning/jobs/{job_id}/cancel",
|
|
761
784
|
base_url=base_url,
|
|
@@ -766,6 +789,7 @@ class Jobs(BaseSDK):
|
|
|
766
789
|
request_has_query_params=True,
|
|
767
790
|
user_agent_header="user-agent",
|
|
768
791
|
accept_header_value="application/json",
|
|
792
|
+
http_headers=http_headers,
|
|
769
793
|
security=self.sdk_configuration.security,
|
|
770
794
|
timeout_ms=timeout_ms,
|
|
771
795
|
)
|
|
@@ -815,6 +839,7 @@ class Jobs(BaseSDK):
|
|
|
815
839
|
retries: OptionalNullable[utils.RetryConfig] = UNSET,
|
|
816
840
|
server_url: Optional[str] = None,
|
|
817
841
|
timeout_ms: Optional[int] = None,
|
|
842
|
+
http_headers: Optional[Mapping[str, str]] = None,
|
|
818
843
|
) -> Optional[models.DetailedJobOut]:
|
|
819
844
|
r"""Start Fine Tuning Job
|
|
820
845
|
|
|
@@ -824,6 +849,7 @@ class Jobs(BaseSDK):
|
|
|
824
849
|
:param retries: Override the default retry configuration for this method
|
|
825
850
|
:param server_url: Override the default server URL for this method
|
|
826
851
|
:param timeout_ms: Override the default request timeout configuration for this method in milliseconds
|
|
852
|
+
:param http_headers: Additional headers to set or replace on requests.
|
|
827
853
|
"""
|
|
828
854
|
base_url = None
|
|
829
855
|
url_variables = None
|
|
@@ -837,7 +863,7 @@ class Jobs(BaseSDK):
|
|
|
837
863
|
job_id=job_id,
|
|
838
864
|
)
|
|
839
865
|
|
|
840
|
-
req = self.
|
|
866
|
+
req = self._build_request(
|
|
841
867
|
method="POST",
|
|
842
868
|
path="/v1/fine_tuning/jobs/{job_id}/start",
|
|
843
869
|
base_url=base_url,
|
|
@@ -848,6 +874,7 @@ class Jobs(BaseSDK):
|
|
|
848
874
|
request_has_query_params=True,
|
|
849
875
|
user_agent_header="user-agent",
|
|
850
876
|
accept_header_value="application/json",
|
|
877
|
+
http_headers=http_headers,
|
|
851
878
|
security=self.sdk_configuration.security,
|
|
852
879
|
timeout_ms=timeout_ms,
|
|
853
880
|
)
|
|
@@ -897,6 +924,7 @@ class Jobs(BaseSDK):
|
|
|
897
924
|
retries: OptionalNullable[utils.RetryConfig] = UNSET,
|
|
898
925
|
server_url: Optional[str] = None,
|
|
899
926
|
timeout_ms: Optional[int] = None,
|
|
927
|
+
http_headers: Optional[Mapping[str, str]] = None,
|
|
900
928
|
) -> Optional[models.DetailedJobOut]:
|
|
901
929
|
r"""Start Fine Tuning Job
|
|
902
930
|
|
|
@@ -906,6 +934,7 @@ class Jobs(BaseSDK):
|
|
|
906
934
|
:param retries: Override the default retry configuration for this method
|
|
907
935
|
:param server_url: Override the default server URL for this method
|
|
908
936
|
:param timeout_ms: Override the default request timeout configuration for this method in milliseconds
|
|
937
|
+
:param http_headers: Additional headers to set or replace on requests.
|
|
909
938
|
"""
|
|
910
939
|
base_url = None
|
|
911
940
|
url_variables = None
|
|
@@ -919,7 +948,7 @@ class Jobs(BaseSDK):
|
|
|
919
948
|
job_id=job_id,
|
|
920
949
|
)
|
|
921
950
|
|
|
922
|
-
req = self.
|
|
951
|
+
req = self._build_request_async(
|
|
923
952
|
method="POST",
|
|
924
953
|
path="/v1/fine_tuning/jobs/{job_id}/start",
|
|
925
954
|
base_url=base_url,
|
|
@@ -930,6 +959,7 @@ class Jobs(BaseSDK):
|
|
|
930
959
|
request_has_query_params=True,
|
|
931
960
|
user_agent_header="user-agent",
|
|
932
961
|
accept_header_value="application/json",
|
|
962
|
+
http_headers=http_headers,
|
|
933
963
|
security=self.sdk_configuration.security,
|
|
934
964
|
timeout_ms=timeout_ms,
|
|
935
965
|
)
|
mistralai/mistral_jobs.py
CHANGED
|
@@ -6,7 +6,7 @@ from mistralai import models, utils
|
|
|
6
6
|
from mistralai._hooks import HookContext
|
|
7
7
|
from mistralai.types import OptionalNullable, UNSET
|
|
8
8
|
from mistralai.utils import get_security_from_env
|
|
9
|
-
from typing import Any, Dict, List, Optional
|
|
9
|
+
from typing import Any, Dict, List, Mapping, Optional
|
|
10
10
|
|
|
11
11
|
|
|
12
12
|
class MistralJobs(BaseSDK):
|
|
@@ -23,6 +23,7 @@ class MistralJobs(BaseSDK):
|
|
|
23
23
|
retries: OptionalNullable[utils.RetryConfig] = UNSET,
|
|
24
24
|
server_url: Optional[str] = None,
|
|
25
25
|
timeout_ms: Optional[int] = None,
|
|
26
|
+
http_headers: Optional[Mapping[str, str]] = None,
|
|
26
27
|
) -> Optional[models.BatchJobsOut]:
|
|
27
28
|
r"""Get Batch Jobs
|
|
28
29
|
|
|
@@ -38,6 +39,7 @@ class MistralJobs(BaseSDK):
|
|
|
38
39
|
:param retries: Override the default retry configuration for this method
|
|
39
40
|
:param server_url: Override the default server URL for this method
|
|
40
41
|
:param timeout_ms: Override the default request timeout configuration for this method in milliseconds
|
|
42
|
+
:param http_headers: Additional headers to set or replace on requests.
|
|
41
43
|
"""
|
|
42
44
|
base_url = None
|
|
43
45
|
url_variables = None
|
|
@@ -57,7 +59,7 @@ class MistralJobs(BaseSDK):
|
|
|
57
59
|
status=status,
|
|
58
60
|
)
|
|
59
61
|
|
|
60
|
-
req = self.
|
|
62
|
+
req = self._build_request(
|
|
61
63
|
method="GET",
|
|
62
64
|
path="/v1/batch/jobs",
|
|
63
65
|
base_url=base_url,
|
|
@@ -68,6 +70,7 @@ class MistralJobs(BaseSDK):
|
|
|
68
70
|
request_has_query_params=True,
|
|
69
71
|
user_agent_header="user-agent",
|
|
70
72
|
accept_header_value="application/json",
|
|
73
|
+
http_headers=http_headers,
|
|
71
74
|
security=self.sdk_configuration.security,
|
|
72
75
|
timeout_ms=timeout_ms,
|
|
73
76
|
)
|
|
@@ -123,6 +126,7 @@ class MistralJobs(BaseSDK):
|
|
|
123
126
|
retries: OptionalNullable[utils.RetryConfig] = UNSET,
|
|
124
127
|
server_url: Optional[str] = None,
|
|
125
128
|
timeout_ms: Optional[int] = None,
|
|
129
|
+
http_headers: Optional[Mapping[str, str]] = None,
|
|
126
130
|
) -> Optional[models.BatchJobsOut]:
|
|
127
131
|
r"""Get Batch Jobs
|
|
128
132
|
|
|
@@ -138,6 +142,7 @@ class MistralJobs(BaseSDK):
|
|
|
138
142
|
:param retries: Override the default retry configuration for this method
|
|
139
143
|
:param server_url: Override the default server URL for this method
|
|
140
144
|
:param timeout_ms: Override the default request timeout configuration for this method in milliseconds
|
|
145
|
+
:param http_headers: Additional headers to set or replace on requests.
|
|
141
146
|
"""
|
|
142
147
|
base_url = None
|
|
143
148
|
url_variables = None
|
|
@@ -157,7 +162,7 @@ class MistralJobs(BaseSDK):
|
|
|
157
162
|
status=status,
|
|
158
163
|
)
|
|
159
164
|
|
|
160
|
-
req = self.
|
|
165
|
+
req = self._build_request_async(
|
|
161
166
|
method="GET",
|
|
162
167
|
path="/v1/batch/jobs",
|
|
163
168
|
base_url=base_url,
|
|
@@ -168,6 +173,7 @@ class MistralJobs(BaseSDK):
|
|
|
168
173
|
request_has_query_params=True,
|
|
169
174
|
user_agent_header="user-agent",
|
|
170
175
|
accept_header_value="application/json",
|
|
176
|
+
http_headers=http_headers,
|
|
171
177
|
security=self.sdk_configuration.security,
|
|
172
178
|
timeout_ms=timeout_ms,
|
|
173
179
|
)
|
|
@@ -221,6 +227,7 @@ class MistralJobs(BaseSDK):
|
|
|
221
227
|
retries: OptionalNullable[utils.RetryConfig] = UNSET,
|
|
222
228
|
server_url: Optional[str] = None,
|
|
223
229
|
timeout_ms: Optional[int] = None,
|
|
230
|
+
http_headers: Optional[Mapping[str, str]] = None,
|
|
224
231
|
) -> Optional[models.BatchJobOut]:
|
|
225
232
|
r"""Create Batch Job
|
|
226
233
|
|
|
@@ -234,6 +241,7 @@ class MistralJobs(BaseSDK):
|
|
|
234
241
|
:param retries: Override the default retry configuration for this method
|
|
235
242
|
:param server_url: Override the default server URL for this method
|
|
236
243
|
:param timeout_ms: Override the default request timeout configuration for this method in milliseconds
|
|
244
|
+
:param http_headers: Additional headers to set or replace on requests.
|
|
237
245
|
"""
|
|
238
246
|
base_url = None
|
|
239
247
|
url_variables = None
|
|
@@ -251,7 +259,7 @@ class MistralJobs(BaseSDK):
|
|
|
251
259
|
timeout_hours=timeout_hours,
|
|
252
260
|
)
|
|
253
261
|
|
|
254
|
-
req = self.
|
|
262
|
+
req = self._build_request(
|
|
255
263
|
method="POST",
|
|
256
264
|
path="/v1/batch/jobs",
|
|
257
265
|
base_url=base_url,
|
|
@@ -262,6 +270,7 @@ class MistralJobs(BaseSDK):
|
|
|
262
270
|
request_has_query_params=True,
|
|
263
271
|
user_agent_header="user-agent",
|
|
264
272
|
accept_header_value="application/json",
|
|
273
|
+
http_headers=http_headers,
|
|
265
274
|
security=self.sdk_configuration.security,
|
|
266
275
|
get_serialized_body=lambda: utils.serialize_request_body(
|
|
267
276
|
request, False, False, "json", models.BatchJobIn
|
|
@@ -318,6 +327,7 @@ class MistralJobs(BaseSDK):
|
|
|
318
327
|
retries: OptionalNullable[utils.RetryConfig] = UNSET,
|
|
319
328
|
server_url: Optional[str] = None,
|
|
320
329
|
timeout_ms: Optional[int] = None,
|
|
330
|
+
http_headers: Optional[Mapping[str, str]] = None,
|
|
321
331
|
) -> Optional[models.BatchJobOut]:
|
|
322
332
|
r"""Create Batch Job
|
|
323
333
|
|
|
@@ -331,6 +341,7 @@ class MistralJobs(BaseSDK):
|
|
|
331
341
|
:param retries: Override the default retry configuration for this method
|
|
332
342
|
:param server_url: Override the default server URL for this method
|
|
333
343
|
:param timeout_ms: Override the default request timeout configuration for this method in milliseconds
|
|
344
|
+
:param http_headers: Additional headers to set or replace on requests.
|
|
334
345
|
"""
|
|
335
346
|
base_url = None
|
|
336
347
|
url_variables = None
|
|
@@ -348,7 +359,7 @@ class MistralJobs(BaseSDK):
|
|
|
348
359
|
timeout_hours=timeout_hours,
|
|
349
360
|
)
|
|
350
361
|
|
|
351
|
-
req = self.
|
|
362
|
+
req = self._build_request_async(
|
|
352
363
|
method="POST",
|
|
353
364
|
path="/v1/batch/jobs",
|
|
354
365
|
base_url=base_url,
|
|
@@ -359,6 +370,7 @@ class MistralJobs(BaseSDK):
|
|
|
359
370
|
request_has_query_params=True,
|
|
360
371
|
user_agent_header="user-agent",
|
|
361
372
|
accept_header_value="application/json",
|
|
373
|
+
http_headers=http_headers,
|
|
362
374
|
security=self.sdk_configuration.security,
|
|
363
375
|
get_serialized_body=lambda: utils.serialize_request_body(
|
|
364
376
|
request, False, False, "json", models.BatchJobIn
|
|
@@ -411,6 +423,7 @@ class MistralJobs(BaseSDK):
|
|
|
411
423
|
retries: OptionalNullable[utils.RetryConfig] = UNSET,
|
|
412
424
|
server_url: Optional[str] = None,
|
|
413
425
|
timeout_ms: Optional[int] = None,
|
|
426
|
+
http_headers: Optional[Mapping[str, str]] = None,
|
|
414
427
|
) -> Optional[models.BatchJobOut]:
|
|
415
428
|
r"""Get Batch Job
|
|
416
429
|
|
|
@@ -420,6 +433,7 @@ class MistralJobs(BaseSDK):
|
|
|
420
433
|
:param retries: Override the default retry configuration for this method
|
|
421
434
|
:param server_url: Override the default server URL for this method
|
|
422
435
|
:param timeout_ms: Override the default request timeout configuration for this method in milliseconds
|
|
436
|
+
:param http_headers: Additional headers to set or replace on requests.
|
|
423
437
|
"""
|
|
424
438
|
base_url = None
|
|
425
439
|
url_variables = None
|
|
@@ -433,7 +447,7 @@ class MistralJobs(BaseSDK):
|
|
|
433
447
|
job_id=job_id,
|
|
434
448
|
)
|
|
435
449
|
|
|
436
|
-
req = self.
|
|
450
|
+
req = self._build_request(
|
|
437
451
|
method="GET",
|
|
438
452
|
path="/v1/batch/jobs/{job_id}",
|
|
439
453
|
base_url=base_url,
|
|
@@ -444,6 +458,7 @@ class MistralJobs(BaseSDK):
|
|
|
444
458
|
request_has_query_params=True,
|
|
445
459
|
user_agent_header="user-agent",
|
|
446
460
|
accept_header_value="application/json",
|
|
461
|
+
http_headers=http_headers,
|
|
447
462
|
security=self.sdk_configuration.security,
|
|
448
463
|
timeout_ms=timeout_ms,
|
|
449
464
|
)
|
|
@@ -493,6 +508,7 @@ class MistralJobs(BaseSDK):
|
|
|
493
508
|
retries: OptionalNullable[utils.RetryConfig] = UNSET,
|
|
494
509
|
server_url: Optional[str] = None,
|
|
495
510
|
timeout_ms: Optional[int] = None,
|
|
511
|
+
http_headers: Optional[Mapping[str, str]] = None,
|
|
496
512
|
) -> Optional[models.BatchJobOut]:
|
|
497
513
|
r"""Get Batch Job
|
|
498
514
|
|
|
@@ -502,6 +518,7 @@ class MistralJobs(BaseSDK):
|
|
|
502
518
|
:param retries: Override the default retry configuration for this method
|
|
503
519
|
:param server_url: Override the default server URL for this method
|
|
504
520
|
:param timeout_ms: Override the default request timeout configuration for this method in milliseconds
|
|
521
|
+
:param http_headers: Additional headers to set or replace on requests.
|
|
505
522
|
"""
|
|
506
523
|
base_url = None
|
|
507
524
|
url_variables = None
|
|
@@ -515,7 +532,7 @@ class MistralJobs(BaseSDK):
|
|
|
515
532
|
job_id=job_id,
|
|
516
533
|
)
|
|
517
534
|
|
|
518
|
-
req = self.
|
|
535
|
+
req = self._build_request_async(
|
|
519
536
|
method="GET",
|
|
520
537
|
path="/v1/batch/jobs/{job_id}",
|
|
521
538
|
base_url=base_url,
|
|
@@ -526,6 +543,7 @@ class MistralJobs(BaseSDK):
|
|
|
526
543
|
request_has_query_params=True,
|
|
527
544
|
user_agent_header="user-agent",
|
|
528
545
|
accept_header_value="application/json",
|
|
546
|
+
http_headers=http_headers,
|
|
529
547
|
security=self.sdk_configuration.security,
|
|
530
548
|
timeout_ms=timeout_ms,
|
|
531
549
|
)
|
|
@@ -575,6 +593,7 @@ class MistralJobs(BaseSDK):
|
|
|
575
593
|
retries: OptionalNullable[utils.RetryConfig] = UNSET,
|
|
576
594
|
server_url: Optional[str] = None,
|
|
577
595
|
timeout_ms: Optional[int] = None,
|
|
596
|
+
http_headers: Optional[Mapping[str, str]] = None,
|
|
578
597
|
) -> Optional[models.BatchJobOut]:
|
|
579
598
|
r"""Cancel Batch Job
|
|
580
599
|
|
|
@@ -584,6 +603,7 @@ class MistralJobs(BaseSDK):
|
|
|
584
603
|
:param retries: Override the default retry configuration for this method
|
|
585
604
|
:param server_url: Override the default server URL for this method
|
|
586
605
|
:param timeout_ms: Override the default request timeout configuration for this method in milliseconds
|
|
606
|
+
:param http_headers: Additional headers to set or replace on requests.
|
|
587
607
|
"""
|
|
588
608
|
base_url = None
|
|
589
609
|
url_variables = None
|
|
@@ -597,7 +617,7 @@ class MistralJobs(BaseSDK):
|
|
|
597
617
|
job_id=job_id,
|
|
598
618
|
)
|
|
599
619
|
|
|
600
|
-
req = self.
|
|
620
|
+
req = self._build_request(
|
|
601
621
|
method="POST",
|
|
602
622
|
path="/v1/batch/jobs/{job_id}/cancel",
|
|
603
623
|
base_url=base_url,
|
|
@@ -608,6 +628,7 @@ class MistralJobs(BaseSDK):
|
|
|
608
628
|
request_has_query_params=True,
|
|
609
629
|
user_agent_header="user-agent",
|
|
610
630
|
accept_header_value="application/json",
|
|
631
|
+
http_headers=http_headers,
|
|
611
632
|
security=self.sdk_configuration.security,
|
|
612
633
|
timeout_ms=timeout_ms,
|
|
613
634
|
)
|
|
@@ -657,6 +678,7 @@ class MistralJobs(BaseSDK):
|
|
|
657
678
|
retries: OptionalNullable[utils.RetryConfig] = UNSET,
|
|
658
679
|
server_url: Optional[str] = None,
|
|
659
680
|
timeout_ms: Optional[int] = None,
|
|
681
|
+
http_headers: Optional[Mapping[str, str]] = None,
|
|
660
682
|
) -> Optional[models.BatchJobOut]:
|
|
661
683
|
r"""Cancel Batch Job
|
|
662
684
|
|
|
@@ -666,6 +688,7 @@ class MistralJobs(BaseSDK):
|
|
|
666
688
|
:param retries: Override the default retry configuration for this method
|
|
667
689
|
:param server_url: Override the default server URL for this method
|
|
668
690
|
:param timeout_ms: Override the default request timeout configuration for this method in milliseconds
|
|
691
|
+
:param http_headers: Additional headers to set or replace on requests.
|
|
669
692
|
"""
|
|
670
693
|
base_url = None
|
|
671
694
|
url_variables = None
|
|
@@ -679,7 +702,7 @@ class MistralJobs(BaseSDK):
|
|
|
679
702
|
job_id=job_id,
|
|
680
703
|
)
|
|
681
704
|
|
|
682
|
-
req = self.
|
|
705
|
+
req = self._build_request_async(
|
|
683
706
|
method="POST",
|
|
684
707
|
path="/v1/batch/jobs/{job_id}/cancel",
|
|
685
708
|
base_url=base_url,
|
|
@@ -690,6 +713,7 @@ class MistralJobs(BaseSDK):
|
|
|
690
713
|
request_has_query_params=True,
|
|
691
714
|
user_agent_header="user-agent",
|
|
692
715
|
accept_header_value="application/json",
|
|
716
|
+
http_headers=http_headers,
|
|
693
717
|
security=self.sdk_configuration.security,
|
|
694
718
|
timeout_ms=timeout_ms,
|
|
695
719
|
)
|
|
@@ -97,7 +97,7 @@ class ChatCompletionRequest(BaseModel):
|
|
|
97
97
|
temperature: OptionalNullable[float] = UNSET
|
|
98
98
|
r"""What sampling temperature to use, we recommend between 0.0 and 0.7. Higher values like 0.7 will make the output more random, while lower values like 0.2 will make it more focused and deterministic. We generally recommend altering this or `top_p` but not both. The default value varies depending on the model you are targeting. Call the `/models` endpoint to retrieve the appropriate value."""
|
|
99
99
|
|
|
100
|
-
top_p: Optional[float] =
|
|
100
|
+
top_p: Optional[float] = None
|
|
101
101
|
r"""Nucleus sampling, where the model considers the results of the tokens with `top_p` probability mass. So 0.1 means only the tokens comprising the top 10% probability mass are considered. We generally recommend altering this or `temperature` but not both."""
|
|
102
102
|
|
|
103
103
|
max_tokens: OptionalNullable[int] = UNSET
|
|
@@ -127,7 +127,7 @@ class ChatCompletionRequest(BaseModel):
|
|
|
127
127
|
n: OptionalNullable[int] = UNSET
|
|
128
128
|
r"""Number of completions to return for each request, input tokens are only billed once."""
|
|
129
129
|
|
|
130
|
-
safe_prompt: Optional[bool] =
|
|
130
|
+
safe_prompt: Optional[bool] = None
|
|
131
131
|
r"""Whether to inject a safety prompt before all conversations."""
|
|
132
132
|
|
|
133
133
|
@model_serializer(mode="wrap")
|
|
@@ -100,7 +100,7 @@ class ChatCompletionStreamRequest(BaseModel):
|
|
|
100
100
|
temperature: OptionalNullable[float] = UNSET
|
|
101
101
|
r"""What sampling temperature to use, we recommend between 0.0 and 0.7. Higher values like 0.7 will make the output more random, while lower values like 0.2 will make it more focused and deterministic. We generally recommend altering this or `top_p` but not both. The default value varies depending on the model you are targeting. Call the `/models` endpoint to retrieve the appropriate value."""
|
|
102
102
|
|
|
103
|
-
top_p: Optional[float] =
|
|
103
|
+
top_p: Optional[float] = None
|
|
104
104
|
r"""Nucleus sampling, where the model considers the results of the tokens with `top_p` probability mass. So 0.1 means only the tokens comprising the top 10% probability mass are considered. We generally recommend altering this or `temperature` but not both."""
|
|
105
105
|
|
|
106
106
|
max_tokens: OptionalNullable[int] = UNSET
|
|
@@ -129,7 +129,7 @@ class ChatCompletionStreamRequest(BaseModel):
|
|
|
129
129
|
n: OptionalNullable[int] = UNSET
|
|
130
130
|
r"""Number of completions to return for each request, input tokens are only billed once."""
|
|
131
131
|
|
|
132
|
-
safe_prompt: Optional[bool] =
|
|
132
|
+
safe_prompt: Optional[bool] = None
|
|
133
133
|
r"""Whether to inject a safety prompt before all conversations."""
|
|
134
134
|
|
|
135
135
|
@model_serializer(mode="wrap")
|
|
@@ -19,7 +19,7 @@ class FileTypedDict(TypedDict):
|
|
|
19
19
|
|
|
20
20
|
class File(BaseModel):
|
|
21
21
|
file_name: Annotated[
|
|
22
|
-
str, pydantic.Field(alias="
|
|
22
|
+
str, pydantic.Field(alias="fileName"), FieldMetadata(multipart=True)
|
|
23
23
|
]
|
|
24
24
|
|
|
25
25
|
content: Annotated[
|
|
@@ -51,11 +51,7 @@ class FilesAPIRoutesUploadFileMultiPartBodyParamsTypedDict(TypedDict):
|
|
|
51
51
|
|
|
52
52
|
|
|
53
53
|
class FilesAPIRoutesUploadFileMultiPartBodyParams(BaseModel):
|
|
54
|
-
file: Annotated[
|
|
55
|
-
File,
|
|
56
|
-
pydantic.Field(alias=""),
|
|
57
|
-
FieldMetadata(multipart=MultipartFormMetadata(file=True)),
|
|
58
|
-
]
|
|
54
|
+
file: Annotated[File, FieldMetadata(multipart=MultipartFormMetadata(file=True))]
|
|
59
55
|
r"""The File object (not file name) to be uploaded.
|
|
60
56
|
To upload a file and specify a custom file name you should format your request as such:
|
|
61
57
|
```bash
|