mistralai 1.3.1__py3-none-any.whl → 1.5.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 (44) hide show
  1. mistralai/__init__.py +10 -1
  2. mistralai/_version.py +4 -1
  3. mistralai/agents.py +58 -14
  4. mistralai/chat.py +140 -14
  5. mistralai/classifiers.py +32 -20
  6. mistralai/embeddings.py +16 -10
  7. mistralai/extra/README.md +56 -0
  8. mistralai/extra/__init__.py +5 -0
  9. mistralai/extra/struct_chat.py +41 -0
  10. mistralai/extra/tests/__init__.py +0 -0
  11. mistralai/extra/tests/test_struct_chat.py +103 -0
  12. mistralai/extra/tests/test_utils.py +162 -0
  13. mistralai/extra/utils/__init__.py +3 -0
  14. mistralai/extra/utils/_pydantic_helper.py +20 -0
  15. mistralai/extra/utils/response_format.py +24 -0
  16. mistralai/files.py +94 -34
  17. mistralai/fim.py +30 -14
  18. mistralai/httpclient.py +50 -0
  19. mistralai/jobs.py +80 -32
  20. mistralai/mistral_jobs.py +64 -24
  21. mistralai/models/__init__.py +8 -0
  22. mistralai/models/agentscompletionrequest.py +5 -0
  23. mistralai/models/agentscompletionstreamrequest.py +5 -0
  24. mistralai/models/chatcompletionrequest.py +5 -0
  25. mistralai/models/chatcompletionstreamrequest.py +5 -0
  26. mistralai/models/fileschema.py +3 -2
  27. mistralai/models/function.py +3 -0
  28. mistralai/models/jsonschema.py +55 -0
  29. mistralai/models/prediction.py +26 -0
  30. mistralai/models/responseformat.py +36 -1
  31. mistralai/models/responseformats.py +1 -1
  32. mistralai/models/retrievefileout.py +3 -2
  33. mistralai/models/toolcall.py +3 -0
  34. mistralai/models/uploadfileout.py +3 -2
  35. mistralai/models_.py +92 -48
  36. mistralai/sdk.py +13 -3
  37. mistralai/sdkconfiguration.py +10 -4
  38. {mistralai-1.3.1.dist-info → mistralai-1.5.0.dist-info}/METADATA +41 -42
  39. {mistralai-1.3.1.dist-info → mistralai-1.5.0.dist-info}/RECORD +43 -33
  40. {mistralai-1.3.1.dist-info → mistralai-1.5.0.dist-info}/WHEEL +1 -1
  41. mistralai_azure/_hooks/custom_user_agent.py +1 -1
  42. mistralai_gcp/sdk.py +1 -2
  43. py.typed +0 -1
  44. {mistralai-1.3.1.dist-info → mistralai-1.5.0.dist-info}/LICENSE +0 -0
@@ -1,8 +1,7 @@
1
- Metadata-Version: 2.1
1
+ Metadata-Version: 2.3
2
2
  Name: mistralai
3
- Version: 1.3.1
3
+ Version: 1.5.0
4
4
  Summary: Python Client SDK for the Mistral AI API.
5
- Home-page: https://github.com/mistralai/client-python.git
6
5
  Author: Mistral
7
6
  Requires-Python: >=3.8
8
7
  Classifier: Programming Language :: Python :: 3
@@ -68,6 +67,7 @@ Mistral AI API: Our Chat Completion and Embeddings APIs specification. Create yo
68
67
  * [Server Selection](https://github.com/mistralai/client-python/blob/master/#server-selection)
69
68
  * [Custom HTTP Client](https://github.com/mistralai/client-python/blob/master/#custom-http-client)
70
69
  * [Authentication](https://github.com/mistralai/client-python/blob/master/#authentication)
70
+ * [Resource Management](https://github.com/mistralai/client-python/blob/master/#resource-management)
71
71
  * [Debugging](https://github.com/mistralai/client-python/blob/master/#debugging)
72
72
  * [IDE Support](https://github.com/mistralai/client-python/blob/master/#ide-support)
73
73
  * [Development](https://github.com/mistralai/client-python/blob/master/#development)
@@ -78,6 +78,11 @@ Mistral AI API: Our Chat Completion and Embeddings APIs specification. Create yo
78
78
  <!-- Start SDK Installation [installation] -->
79
79
  ## SDK Installation
80
80
 
81
+ > [!NOTE]
82
+ > **Python version upgrade policy**
83
+ >
84
+ > Once a Python version reaches its [official end of life date](https://devguide.python.org/versions/), a 3-month grace period is provided for users to upgrade. Following this grace period, the minimum python version supported in the SDK will be updated.
85
+
81
86
  The SDK can be installed with either *pip* or *poetry* package managers.
82
87
 
83
88
  ### PIP
@@ -118,9 +123,7 @@ with Mistral(
118
123
  "content": "Who is the best French painter? Answer in one short sentence.",
119
124
  "role": "user",
120
125
  },
121
- ])
122
-
123
- assert res is not None
126
+ ], stream=False)
124
127
 
125
128
  # Handle response
126
129
  print(res)
@@ -145,9 +148,7 @@ async def main():
145
148
  "content": "Who is the best French painter? Answer in one short sentence.",
146
149
  "role": "user",
147
150
  },
148
- ])
149
-
150
- assert res is not None
151
+ ], stream=False)
151
152
 
152
153
  # Handle response
153
154
  print(res)
@@ -173,8 +174,6 @@ with Mistral(
173
174
  "content": open("example.file", "rb"),
174
175
  })
175
176
 
176
- assert res is not None
177
-
178
177
  # Handle response
179
178
  print(res)
180
179
  ```
@@ -198,8 +197,6 @@ async def main():
198
197
  "content": open("example.file", "rb"),
199
198
  })
200
199
 
201
- assert res is not None
202
-
203
200
  # Handle response
204
201
  print(res)
205
202
 
@@ -224,9 +221,7 @@ with Mistral(
224
221
  "content": "Who is the best French painter? Answer in one short sentence.",
225
222
  "role": "user",
226
223
  },
227
- ], agent_id="<value>")
228
-
229
- assert res is not None
224
+ ], agent_id="<id>", stream=False)
230
225
 
231
226
  # Handle response
232
227
  print(res)
@@ -251,9 +246,7 @@ async def main():
251
246
  "content": "Who is the best French painter? Answer in one short sentence.",
252
247
  "role": "user",
253
248
  },
254
- ], agent_id="<value>")
255
-
256
- assert res is not None
249
+ ], agent_id="<id>", stream=False)
257
250
 
258
251
  # Handle response
259
252
  print(res)
@@ -277,9 +270,7 @@ with Mistral(
277
270
  res = mistral.embeddings.create(inputs=[
278
271
  "Embed this sentence.",
279
272
  "As well as this one.",
280
- ], model="Wrangler")
281
-
282
- assert res is not None
273
+ ], model="mistral-embed")
283
274
 
284
275
  # Handle response
285
276
  print(res)
@@ -302,9 +293,7 @@ async def main():
302
293
  res = await mistral.embeddings.create_async(inputs=[
303
294
  "Embed this sentence.",
304
295
  "As well as this one.",
305
- ], model="Wrangler")
306
-
307
- assert res is not None
296
+ ], model="mistral-embed")
308
297
 
309
298
  # Handle response
310
299
  print(res)
@@ -506,9 +495,7 @@ with Mistral(
506
495
  "content": "Who is the best French painter? Answer in one short sentence.",
507
496
  "role": "user",
508
497
  },
509
- ])
510
-
511
- assert res is not None
498
+ ], stream=True)
512
499
 
513
500
  with res as event_stream:
514
501
  for event in event_stream:
@@ -545,8 +532,6 @@ with Mistral(
545
532
  "content": open("example.file", "rb"),
546
533
  })
547
534
 
548
- assert res is not None
549
-
550
535
  # Handle response
551
536
  print(res)
552
537
 
@@ -571,8 +556,6 @@ with Mistral(
571
556
  res = mistral.models.list(,
572
557
  RetryConfig("backoff", BackoffStrategy(1, 50, 1.1, 100), False))
573
558
 
574
- assert res is not None
575
-
576
559
  # Handle response
577
560
  print(res)
578
561
 
@@ -591,8 +574,6 @@ with Mistral(
591
574
 
592
575
  res = mistral.models.list()
593
576
 
594
- assert res is not None
595
-
596
577
  # Handle response
597
578
  print(res)
598
579
 
@@ -634,8 +615,6 @@ with Mistral(
634
615
 
635
616
  res = mistral.models.list()
636
617
 
637
- assert res is not None
638
-
639
618
  # Handle response
640
619
  print(res)
641
620
 
@@ -672,8 +651,6 @@ with Mistral(
672
651
 
673
652
  res = mistral.models.list()
674
653
 
675
- assert res is not None
676
-
677
654
  # Handle response
678
655
  print(res)
679
656
 
@@ -693,8 +670,6 @@ with Mistral(
693
670
 
694
671
  res = mistral.models.list()
695
672
 
696
- assert res is not None
697
-
698
673
  # Handle response
699
674
  print(res)
700
675
 
@@ -804,14 +779,38 @@ with Mistral(
804
779
 
805
780
  res = mistral.models.list()
806
781
 
807
- assert res is not None
808
-
809
782
  # Handle response
810
783
  print(res)
811
784
 
812
785
  ```
813
786
  <!-- End Authentication [security] -->
814
787
 
788
+ <!-- Start Resource Management [resource-management] -->
789
+ ## Resource Management
790
+
791
+ The `Mistral` class implements the context manager protocol and registers a finalizer function to close the underlying sync and async HTTPX clients it uses under the hood. This will close HTTP connections, release memory and free up other resources held by the SDK. In short-lived Python programs and notebooks that make a few SDK method calls, resource management may not be a concern. However, in longer-lived programs, it is beneficial to create a single SDK instance via a [context manager][context-manager] and reuse it across the application.
792
+
793
+ [context-manager]: https://docs.python.org/3/reference/datamodel.html#context-managers
794
+
795
+ ```python
796
+ from mistralai import Mistral
797
+ import os
798
+ def main():
799
+ with Mistral(
800
+ api_key=os.getenv("MISTRAL_API_KEY", ""),
801
+ ) as mistral:
802
+ # Rest of application here...
803
+
804
+
805
+ # Or when using async:
806
+ async def amain():
807
+ async with Mistral(
808
+ api_key=os.getenv("MISTRAL_API_KEY", ""),
809
+ ) as mistral:
810
+ # Rest of application here...
811
+ ```
812
+ <!-- End Resource Management [resource-management] -->
813
+
815
814
  <!-- Start Debugging [debug] -->
816
815
  ## Debugging
817
816
 
@@ -1,6 +1,6 @@
1
1
  mistralai_azure/__init__.py,sha256=ybmMeI8Pa2HEQCdADbpyIdm3JWu8AU61HxmDupOlTws,220
2
2
  mistralai_azure/_hooks/__init__.py,sha256=9_7W5jAYw8rcO8Kfc-Ty-lB82BHfksAJJpVFb_UeU1c,146
3
- mistralai_azure/_hooks/custom_user_agent.py,sha256=CnX-8LjyTyOgDcvurQPK2suWBqwQx5PuTT5BBShBBhk,655
3
+ mistralai_azure/_hooks/custom_user_agent.py,sha256=0m-1JzJxOT42rvRTEuCiFLqbOMriOlsraSrAGaXAbyo,656
4
4
  mistralai_azure/_hooks/registration.py,sha256=5BN-U92pwP5kUaN7EOso2vWrwZlLvRcU5Coccibqp20,741
5
5
  mistralai_azure/_hooks/sdkhooks.py,sha256=urOhVMYX_n5KgMoNDNmGs4fsgUWoeSG6_GarhPxH-YU,2565
6
6
  mistralai_azure/_hooks/types.py,sha256=ealwk4q-4oIauxj5Nyx9xmu9vX0nm4vcDALDy4qydgE,2576
@@ -105,7 +105,7 @@ mistralai_gcp/models/usageinfo.py,sha256=Uo2LJB58JMzlrmnfMUQnDxiMCINMS63ejp-sbOq
105
105
  mistralai_gcp/models/usermessage.py,sha256=3OXMcPO3Tyje6wQuOfMVp35OD0EnfYZ2tkElVxOfXs8,1797
106
106
  mistralai_gcp/models/validationerror.py,sha256=EVhyAndNY5aayJSNGv-W1XL7Wu9bS92JJe1yu9UmBSY,530
107
107
  mistralai_gcp/py.typed,sha256=zrp19r0G21lr2yRiMC0f8MFkQFGj9wMpSbboePMg8KM,59
108
- mistralai_gcp/sdk.py,sha256=nioXPwRIiI0KJz8T14LwY4FR4pIHktdLr3u9qSL_y2M,6760
108
+ mistralai_gcp/sdk.py,sha256=c38PFI_NTMYbLtVGdq7bdf1Pbj-wgGJfplDmY_vkhIE,6746
109
109
  mistralai_gcp/sdkconfiguration.py,sha256=3EvSsy_Jk4G-Q9-1veLVFqbgVO-bkCtgi4St7aA-KlU,1700
110
110
  mistralai_gcp/types/__init__.py,sha256=RArOwSgeeTIva6h-4ttjXwMUeCkz10nAFBL9D-QljI4,377
111
111
  mistralai_gcp/types/basemodel.py,sha256=PexI39iKiOkIlobB8Ueo0yn8PLHp6_wb-WO-zelNDZY,1170
@@ -124,32 +124,40 @@ mistralai_gcp/utils/security.py,sha256=ktep3HKwbFs-MLxUYTM8Jd4v-ZBum5_Z0u1PFIdYB
124
124
  mistralai_gcp/utils/serializers.py,sha256=BSJT7kBOkNBFyP7KREyMoe14JGbgijD1M6AXFMbdmco,4924
125
125
  mistralai_gcp/utils/url.py,sha256=BgGPgcTA6MRK4bF8fjP2dUopN3NzEzxWMXPBVg8NQUA,5254
126
126
  mistralai_gcp/utils/values.py,sha256=_89YXPTI_BU6SXJBzFR4pIzTCBPQW9tsOTN1jeBBIDs,3428
127
- py.typed,sha256=zrp19r0G21lr2yRiMC0f8MFkQFGj9wMpSbboePMg8KM,59
128
- mistralai/__init__.py,sha256=ybmMeI8Pa2HEQCdADbpyIdm3JWu8AU61HxmDupOlTws,220
127
+ mistralai/__init__.py,sha256=Tz5Y5FzbIUT1AmaYiTwJI56XTmuldo9AalaAm4h_FdE,423
129
128
  mistralai/_hooks/__init__.py,sha256=9_7W5jAYw8rcO8Kfc-Ty-lB82BHfksAJJpVFb_UeU1c,146
130
129
  mistralai/_hooks/custom_user_agent.py,sha256=cHfp43RcsNvHusq8WVxWrCS3w-pmzJ8uNuvaMZKdtJ8,661
131
130
  mistralai/_hooks/deprecation_warning.py,sha256=eyEOf7-o9uqqNWJnufD2RXp3dYrGV4in9q76yLC1zog,921
132
131
  mistralai/_hooks/registration.py,sha256=ML0W-XbE4WYdJ4eGks_XxF2aLCJTaIWjQATFGzFwvyU,861
133
132
  mistralai/_hooks/sdkhooks.py,sha256=s-orhdvnV89TmI3QiPC2LWQtYeM9RrsG1CTll-fYZmQ,2559
134
133
  mistralai/_hooks/types.py,sha256=vUkTVk_TSaK10aEj368KYWvnd4T5EsawixMcTro_UHc,2570
135
- mistralai/_version.py,sha256=n77-fYLhudMxqUWevuHvbOKcsKC47q8F7a4ys7HPLoM,313
136
- mistralai/agents.py,sha256=fR0i6638GqTptSr2rNCQG6WKH4_AiiPUA16tzqiRQBA,29496
134
+ mistralai/_version.py,sha256=GRgL4RQk2kYt7mohgi1PRAMqEbZSHKMkxC4qZQ_m4Ak,460
135
+ mistralai/agents.py,sha256=63-FJfMJ_9Qb0O0-uorAJM8km4FHpCEjjIk14margM8,31392
137
136
  mistralai/async_client.py,sha256=KUdYxIIqoD6L7vB0EGwUR6lQ0NK5iCTHjnLVR9CVcJY,355
138
137
  mistralai/basesdk.py,sha256=da0sFeLR-ztU5-fuGJ4TMqUnnqSzXbPAjpNgKI52tBk,11999
139
138
  mistralai/batch.py,sha256=YN4D0Duwrap9Ysmp_lRpADYp1Znay7THE_z8ERGvDds,501
140
- mistralai/chat.py,sha256=gaTrfuLoJ7sH9Ku6SxCiZyswQE6XlGmdHAU4qJ2mvr0,33679
141
- mistralai/classifiers.py,sha256=E2gQl8R4p-aVpvMzTInQ0MwEZn91Jj59Nub6rk6Xz1E,15818
139
+ mistralai/chat.py,sha256=n_FpKg-yN0CbAcnVRAkkwcXQuBPwiZBe3e1YSg2hfq8,39339
140
+ mistralai/classifiers.py,sha256=oxGQEj6QfIWvV2GIAhfev9DVbFTG79uR5ODKTmOiJxs,16594
142
141
  mistralai/client.py,sha256=hrPg-LciKMKiascF0WbRRmqQyCv1lb2yDh6j-aaKVNo,509
143
- mistralai/embeddings.py,sha256=SBP-AeaiN2CMcznRkSli_F-YGABJXCe3MP7FDvNGRrs,8202
144
- mistralai/files.py,sha256=TLvxzY4uKNOaMpQieBNb7HDIwzZ0yP8HHgkdCruIzjs,41651
145
- mistralai/fim.py,sha256=55J_vlD0_mlcEW-a57vRU4oYMT-KgVNQ1BSSHDiB_Tg,26433
142
+ mistralai/embeddings.py,sha256=oT6SgsC3ODtn9mAWfpHN9Eli6puLmz9dEe-ZrY0QiPA,8590
143
+ mistralai/extra/README.md,sha256=BTS9fy0ijkiUP7ZVoFQ7FVBxHtXIXqucYZyy_ucFjo4,1739
144
+ mistralai/extra/__init__.py,sha256=MHf0pUgLc9Sb7eTUE31JlE2FKMxfQupmJ_iR8UkgQ9w,360
145
+ mistralai/extra/struct_chat.py,sha256=ZkpdExC5rgC-nBZ44hQIVhQmK6lYMk36RBSFPZMFaIg,2157
146
+ mistralai/extra/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
147
+ mistralai/extra/tests/test_struct_chat.py,sha256=WT6GGfcbXCok8UkEny19u4q1g2QOgkekvmAb3ZinQZ8,4343
148
+ mistralai/extra/tests/test_utils.py,sha256=VesGDR_IiE6u0iY7yOi1iERd7esdJgi2aL4xZp0vKVI,5113
149
+ mistralai/extra/utils/__init__.py,sha256=SExo5t_hx0ybiQhVJIG3r3hOA-Pfny3lIO_WsqNXlN8,116
150
+ mistralai/extra/utils/_pydantic_helper.py,sha256=kU_HbsSl1qGXnrrHnBcxun2MtHowu8eBp3jYMyFsPWw,859
151
+ mistralai/extra/utils/response_format.py,sha256=OiIpNXMODKJ6U2QDCXxPHBoVtXzXF7jtBzCLmI4t_CU,907
152
+ mistralai/files.py,sha256=E1-MxJc-84IdKrc-0k-bvYNa7OSoNFCQ7wBX9tMnFb8,44359
153
+ mistralai/fim.py,sha256=WFlELwTAT_dT7HcBRuby8xqoQsUsw-IXqzqufyyqz_g,27289
146
154
  mistralai/fine_tuning.py,sha256=UENQqfE054VEsAYxdruV-TBLFIFfO-joXNznH08GUvE,477
147
- mistralai/httpclient.py,sha256=WDbLpMzo7qmWki_ryOJcCAYNI1T4uyWKV08rRuCdNII,2688
148
- mistralai/jobs.py,sha256=ftNZ6nzQZPQoEECUrma4jLSdkV4GwA9Q63sFZSI6bnY,41275
149
- mistralai/mistral_jobs.py,sha256=UAjO2qL1zdfrTQ0WYXVsKtr9cqg3Lf6hgtc-aiL-Zno,28374
150
- mistralai/models/__init__.py,sha256=r-GIWDriQyAFaWvuazQhJPB6ZsCegcBVjoAcZyVu1ho,21712
151
- mistralai/models/agentscompletionrequest.py,sha256=iYCRAR6IQJR1WdIt6MjOkLlIdQUB8vCHf-utbGzU3oI,7618
152
- mistralai/models/agentscompletionstreamrequest.py,sha256=QLbh0bGxH-J_1A9isXJkwsx4IzeMWYZxw6Jrl0Kpfdk,7063
155
+ mistralai/httpclient.py,sha256=N-D-srtDBykpfyVKacTY4upDGvNLqdWlEYqhJvta99E,4194
156
+ mistralai/jobs.py,sha256=SVNlvn8XGaCkNJ5tKOKci5QLavacmkNqoYeIGF4ik0Q,43481
157
+ mistralai/mistral_jobs.py,sha256=2ScKd2Tv79-MWxEQkrqr53Ikya8rmTbSiJ96judp7DY,30166
158
+ mistralai/models/__init__.py,sha256=IGlVf2t6ZzB5HvRNQpFaf5WmKJOemcrCU9Uvsf9IAz0,21953
159
+ mistralai/models/agentscompletionrequest.py,sha256=2tV6_p33zLCfBD7EzlAVeSjG0E_pknbVZlbQFM3Lulc,7794
160
+ mistralai/models/agentscompletionstreamrequest.py,sha256=D1fla5nnnwNKXwHG1w4XVGnqaEvx6sOuhTXdP_e65sM,7239
153
161
  mistralai/models/apiendpoint.py,sha256=Hvar5leWsJR_FYb0UzRlSw3vRdBZhk_6BR5r2pIb214,400
154
162
  mistralai/models/archiveftmodelout.py,sha256=dQx1J91UA06pjk2r7okhKMyBBePmHal7SPpn6Y_wEsY,820
155
163
  mistralai/models/assistantmessage.py,sha256=BbvwC7W4Zk_8GN3w1QnEj7UygVGl0pfCtr6eosNUAgw,2204
@@ -161,9 +169,9 @@ mistralai/models/batchjobsout.py,sha256=Tq6bcb4_-fcW8C9AOnfJN8_sTy1NQhDn82qFOKdF
161
169
  mistralai/models/batchjobstatus.py,sha256=WlrIl5vWQGfLmgQA91_9CnCMKhWN6Lli458fT-4Asj4,294
162
170
  mistralai/models/chatclassificationrequest.py,sha256=WYthWiE7euFSeJYJRQJMXmxbWzGDl8kpF4yyLO9ZJaE,3225
163
171
  mistralai/models/chatcompletionchoice.py,sha256=6iIFLZj2KYx0HFfzS3-E3sNXG6mPEAlDyXxIA5iZI_U,849
164
- mistralai/models/chatcompletionrequest.py,sha256=E93EOWOm0uOjWqAoD9d0MAllzhwlFc23YD3bnzoXOA0,9587
172
+ mistralai/models/chatcompletionrequest.py,sha256=MlAYYz5-VK9i81Teb89D_KxqQ34lip0eU3DlHkv05aM,9763
165
173
  mistralai/models/chatcompletionresponse.py,sha256=sLE-_Bx9W5rH2-HE2fBWPVbJbmBWx_jSY2mJ3KBEn6w,792
166
- mistralai/models/chatcompletionstreamrequest.py,sha256=YSdqSpIDj5YcPxvQG7o7Beu36aPj4_-d4-edN9rIp6o,9275
174
+ mistralai/models/chatcompletionstreamrequest.py,sha256=3Td59jU_GDoioTjcrx5pteOqiRJxlFF0lshVpRsx7V0,9451
167
175
  mistralai/models/checkpointout.py,sha256=A2kXS8-VT_1lbg3brifVjZD6tXdsET8vLqBm2a-yXgA,1109
168
176
  mistralai/models/classificationobject.py,sha256=JqaKo3AQD4t5X12ZnHjJ6K3Y6LXUn94uGdLJSoGr8vY,665
169
177
  mistralai/models/classificationrequest.py,sha256=UFbjwNFtLZmylX3DqQSH3FoOvvD1UGBapK6f9b9LGYE,1903
@@ -188,7 +196,7 @@ mistralai/models/files_api_routes_get_signed_urlop.py,sha256=e_XczBgInaylmHJ9m5w
188
196
  mistralai/models/files_api_routes_list_filesop.py,sha256=ztzOPSCg7VfmRDxwxnquhTpLPRbI-wJAvkASOsv2g8w,3130
189
197
  mistralai/models/files_api_routes_retrieve_fileop.py,sha256=8DjSbYqUqFoPq-qcNXyVADeBVSsoCfHFlkRfmwYk-ns,504
190
198
  mistralai/models/files_api_routes_upload_fileop.py,sha256=gIGH5xcPryWYkj1FmNv_0-9He-QB_rRf1a5nUYBNDTU,2213
191
- mistralai/models/fileschema.py,sha256=cgHHp6H9HdIZqiFc-7BQwF5L6nzXz8aF6F7evEm-Hj4,2529
199
+ mistralai/models/fileschema.py,sha256=n_IjCdNOrC2fuzkv75wJn01XvqGTmPK3JqAFSHaOiMA,2597
192
200
  mistralai/models/filesignedurl.py,sha256=VwvuhzhJulAB99Qxz6zr-2F1aINosAfaSxU0IhytDSU,289
193
201
  mistralai/models/fimcompletionrequest.py,sha256=DidQBt_s7o-VkfGNZpN6MUi19Qs_WXVCvA7ek45Hcnc,6606
194
202
  mistralai/models/fimcompletionresponse.py,sha256=_QwzRuL3KuKkyrA4Fxp366SW0H0EzOA7f4FLkWLm-PM,790
@@ -196,7 +204,7 @@ mistralai/models/fimcompletionstreamrequest.py,sha256=n-uzWyaSk4-dfBLM8rbmhZOESL
196
204
  mistralai/models/ftmodelcapabilitiesout.py,sha256=H1kKEChUPgYT31ZQUz0tn9NRa7Z3hRZlh-sFfDYvBos,648
197
205
  mistralai/models/ftmodelcard.py,sha256=G3dioHDMOhbI5HIw5gCaxZDb1sCthBzkXIAYMNHwya8,3300
198
206
  mistralai/models/ftmodelout.py,sha256=dw-y8KKT_7rzW6tu10gfc1YKB8-Kpw4e4zy23z_U1d4,2530
199
- mistralai/models/function.py,sha256=lr0wmE5PW9RyNdLp6JPcVCIzUz6iBAvRie_1eg9WVj8,467
207
+ mistralai/models/function.py,sha256=MTJfwFM17CPKhzMh4zwOL3zM6INCieLw8_rOT-zuDw8,533
200
208
  mistralai/models/functioncall.py,sha256=VvvBe4bVq1Irqo5t4_n1iq60UF7hLf8tE_GjkbyM8iE,556
201
209
  mistralai/models/functionname.py,sha256=jgd0moI9eORQtEAQI4ROiMSKpWSbCLmK6IhDn7uppKY,467
202
210
  mistralai/models/githubrepositoryin.py,sha256=kHU3QnOEJsdsyAt-74jrY2ztEao3aXcNdtj7aOgASRg,1956
@@ -219,16 +227,18 @@ mistralai/models/jobs_api_routes_fine_tuning_start_fine_tuning_jobop.py,sha256=h
219
227
  mistralai/models/jobs_api_routes_fine_tuning_unarchive_fine_tuned_modelop.py,sha256=_pkyhD7OzG-59fgcajI9NmSLTLDktkCxXo_IuvWeyfs,636
220
228
  mistralai/models/jobs_api_routes_fine_tuning_update_fine_tuned_modelop.py,sha256=s-EYS-Hw0NExYeIyN-3JlHbKmnTmtyB8ljVSfOylqYk,907
221
229
  mistralai/models/jobsout.py,sha256=uCKt0aw7yXzI4oLDGeAAEhsRjdRg3g7lPopg0__czTA,818
230
+ mistralai/models/jsonschema.py,sha256=Sh6VCcRxJffd-vXX61GVElbxKNi-j2MKvMUZQcW5EUs,1653
222
231
  mistralai/models/legacyjobmetadataout.py,sha256=08zAGNTSrICsK8u2SFFUXiNWF7MCQvezmFQeMQzxsys,4762
223
232
  mistralai/models/listfilesout.py,sha256=tW2fNabLKcftc5kytkjwVaChlOzWRL4FKtNzDak9MNs,468
224
233
  mistralai/models/metricout.py,sha256=dXQMMU4Nk6-Zr06Jx1TWilFi6cOwiVLjSanCFn0cPxo,2034
225
234
  mistralai/models/modelcapabilities.py,sha256=No-Dl09zT1sG4MxsWnx4s8Yo1tUeMQ7k-HR_iQFIMFc,703
226
235
  mistralai/models/modellist.py,sha256=D4Y784kQkx0ARhofFrpEqGLfxa-jTY8ev0TQMrD_n8I,995
236
+ mistralai/models/prediction.py,sha256=54P1n4Y5pXu4YsFKAcmThj0GyUrWxyOlIfjG5K2htB8,726
227
237
  mistralai/models/referencechunk.py,sha256=A9vV5pZv-tUqGlswdu0HOyCYy0Q-UIJY0Oc9ZfM6XJA,519
228
- mistralai/models/responseformat.py,sha256=C_zO6X4cbT1qSS_q9Qxq64AmtfK10i9tqEz_39ZcFzo,1045
229
- mistralai/models/responseformats.py,sha256=oeXHoVUoZrZwrz-0cm-rHj5sHygv9MpeqgdummGa8ww,488
238
+ mistralai/models/responseformat.py,sha256=-TAPGth3_FAiNl-kuE4COI5cSP5fxQ7xewFSV989i58,2225
239
+ mistralai/models/responseformats.py,sha256=O9lwS2M9m53DsRxTC4uRP12SvRhgaQoMjIYsDys5A7s,503
230
240
  mistralai/models/retrieve_model_v1_models_model_id_getop.py,sha256=N9_JFwiz9tz4zRXJ9c1V0c_anFEVxVzPDoFt2Wrer4M,1388
231
- mistralai/models/retrievefileout.py,sha256=LfeXEjmn3xQLRzunWGW-0V_zsZIqtS6JEwQwz1s06DU,2576
241
+ mistralai/models/retrievefileout.py,sha256=nAjSITJCHj0daChhpwOZTmps-74mmYZO4IckGA0yIvQ,2644
232
242
  mistralai/models/sampletype.py,sha256=zowUiTFxum8fltBs6j__BrFPio-dQdG0CIyLj-5icG8,316
233
243
  mistralai/models/sdkerror.py,sha256=kd75e3JYF2TXNgRZopcV-oGdBWoBZqRcvrwqn2fsFYs,528
234
244
  mistralai/models/security.py,sha256=RQn-xHLq3q4OEzrx9BcJMuT49UaCvwABXrqBEcqyKmA,686
@@ -236,7 +246,7 @@ mistralai/models/source.py,sha256=_MSV-LRL2fL7wCUTXEvvsOUIWlOKqPvdZS4rm2Xhs0o,26
236
246
  mistralai/models/systemmessage.py,sha256=ZTDim1ZLHLiCe2zd70PuxcGyUrNhB_aFSMOpOzl5eCI,786
237
247
  mistralai/models/textchunk.py,sha256=2VD-TR3NOOWJ9Jzcw_E3ryq0GWz6b5XSP3k5o7oVlnc,448
238
248
  mistralai/models/tool.py,sha256=qLY0XE3uk79v3RsJqVpA81A0K9OWtmX6rwVeKal5ARk,681
239
- mistralai/models/toolcall.py,sha256=DYYZ4KEwwJWrN0zLJtXkCd35es8f-LVzE7kVfFe_ZkE,766
249
+ mistralai/models/toolcall.py,sha256=T5-3XQ-CKduBKTWwOeSBuaF90yk4yBgqmyLuXVB5uXQ,824
240
250
  mistralai/models/toolchoice.py,sha256=dGeb5koPp9eqHQuG1u-kP7T5Od6-cPL2rEe06-dqzcs,1021
241
251
  mistralai/models/toolchoiceenum.py,sha256=Ca4ileCwuOjfPzIXLRIxT3RkE5zR7oqV6nXU-UjW0w0,197
242
252
  mistralai/models/toolmessage.py,sha256=zcu054y_vBaGbsCWmq58DsY8aiRrSNwzC4LJz_5kUVg,2038
@@ -246,16 +256,16 @@ mistralai/models/trainingparameters.py,sha256=GA7OFskn6BxHs_N6YIv0s6e1j_xmqrwcuC
246
256
  mistralai/models/trainingparametersin.py,sha256=tKiFYm9RUOegdRX_kka_On-mOipX5fEv1sM_7wZ2leY,4556
247
257
  mistralai/models/unarchiveftmodelout.py,sha256=8sn1THvCNnZpPcw3m3avWiBHe357D_nSrp0LcFFYVsY,831
248
258
  mistralai/models/updateftmodelin.py,sha256=Slabh0wwDFP8bzXWFqDzGoLh3KPnOAEyc7vttabSqX8,1466
249
- mistralai/models/uploadfileout.py,sha256=cLMFKbAH6sEgH1G8SVgX4-C3KpIZEc6B39LGySi3XTI,2535
259
+ mistralai/models/uploadfileout.py,sha256=q05j3XeaoF-AHqIshXJ73dUL5zOMxxQAF-w-u2tmVNU,2603
250
260
  mistralai/models/usageinfo.py,sha256=66AzKK8cOFft498eUOUx6C_mCFAt5LmIFrYthJfLWpU,401
251
261
  mistralai/models/usermessage.py,sha256=kjS5HudMsaRNfeUhFFBCyY8l-umlHvGk8wvTIq6-qzY,1793
252
262
  mistralai/models/validationerror.py,sha256=DouDBJmBhbW4KPsF5rZEyBdnB_adC-l32kuHC0bvO2I,526
253
263
  mistralai/models/wandbintegration.py,sha256=BkLD3r08ToZkMAhPXdnC7bfOGr3banKqt1wVKMGehUQ,2406
254
264
  mistralai/models/wandbintegrationout.py,sha256=C0HpS8jJGnACs7eWnuIq0qJEroIUAbjkvzfSSkSKS7Q,2274
255
- mistralai/models_.py,sha256=kRnakzM3wQH8o99PZbMLXm7prVp2maguKJZab6HWzdM,41930
265
+ mistralai/models_.py,sha256=r0etSSUChK7hxxf7ZyhoeloyE8TyPRL1s5Jh4Jgukbw,44394
256
266
  mistralai/py.typed,sha256=zrp19r0G21lr2yRiMC0f8MFkQFGj9wMpSbboePMg8KM,59
257
- mistralai/sdk.py,sha256=lLJOfh9im-TDBsXX9_qM-JPGXjQOOSBDGOrmlJWaZBY,5444
258
- mistralai/sdkconfiguration.py,sha256=4u44yB75Pw--81TaTZCZRB-K_E1UdOgavCEizHJ17ho,1688
267
+ mistralai/sdk.py,sha256=itKygBxr6JLO8f9u0v0m-bsHLOjKutuLalZsg0OLISU,5733
268
+ mistralai/sdkconfiguration.py,sha256=sx6U1xgxFbflhJdOBAeI7isId-8wlMd8NK9pi-JGs1o,1789
259
269
  mistralai/types/__init__.py,sha256=RArOwSgeeTIva6h-4ttjXwMUeCkz10nAFBL9D-QljI4,377
260
270
  mistralai/types/basemodel.py,sha256=PexI39iKiOkIlobB8Ueo0yn8PLHp6_wb-WO-zelNDZY,1170
261
271
  mistralai/utils/__init__.py,sha256=8npwwHS-7zjVrbkzBGO-Uk4GkjC240PCleMbgPK1Axs,2418
@@ -274,7 +284,7 @@ mistralai/utils/serializers.py,sha256=BSJT7kBOkNBFyP7KREyMoe14JGbgijD1M6AXFMbdmc
274
284
  mistralai/utils/url.py,sha256=BgGPgcTA6MRK4bF8fjP2dUopN3NzEzxWMXPBVg8NQUA,5254
275
285
  mistralai/utils/values.py,sha256=_89YXPTI_BU6SXJBzFR4pIzTCBPQW9tsOTN1jeBBIDs,3428
276
286
  mistralai/version.py,sha256=iosXhlXclBwBqlADFKEilxAC2wWKbtuBKi87AmPi7s8,196
277
- mistralai-1.3.1.dist-info/LICENSE,sha256=rUtQ_9GD0OyLPlb-2uWVdfE87hzudMRmsW-tS-0DK-0,11340
278
- mistralai-1.3.1.dist-info/METADATA,sha256=4cx2LjFgsvPdtZoNLamwan0Tv2LRUaP5lm0AZLJGEnM,28158
279
- mistralai-1.3.1.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
280
- mistralai-1.3.1.dist-info/RECORD,,
287
+ mistralai-1.5.0.dist-info/LICENSE,sha256=rUtQ_9GD0OyLPlb-2uWVdfE87hzudMRmsW-tS-0DK-0,11340
288
+ mistralai-1.5.0.dist-info/METADATA,sha256=drBwjaz85HBcWlgOsNldybLkbE3KitbeGUsThLU8etU,29251
289
+ mistralai-1.5.0.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
290
+ mistralai-1.5.0.dist-info/RECORD,,
@@ -1,4 +1,4 @@
1
1
  Wheel-Version: 1.0
2
- Generator: poetry-core 1.9.1
2
+ Generator: poetry-core 2.0.1
3
3
  Root-Is-Purelib: true
4
4
  Tag: py3-none-any
@@ -19,4 +19,4 @@ class CustomUserAgentHook(BeforeRequestHook):
19
19
  PREFIX + current.split(" ")[1]
20
20
  )
21
21
 
22
- return request
22
+ return request
mistralai_gcp/sdk.py CHANGED
@@ -30,8 +30,7 @@ def get_model_info(model: str) -> Tuple[str, str]:
30
30
  # if the model requiers the legacy fomat, use it, else do nothing.
31
31
  if model in LEGACY_MODEL_ID_FORMAT:
32
32
  return "-".join(model.split("-")[:-1]), LEGACY_MODEL_ID_FORMAT[model]
33
- else:
34
- return model, model
33
+ return model, model
35
34
 
36
35
 
37
36
 
py.typed DELETED
@@ -1 +0,0 @@
1
- # Marker file for PEP 561. The package enables type hints.