mistralai 1.9.7__py3-none-any.whl → 1.9.8__py3-none-any.whl

mistralai/_version.py

@@ -3,10 +3,10 @@
 import importlib.metadata
 
 __title__: str = "mistralai"
-__version__: str = "1.9.7"
+__version__: str = "1.9.8"
 __openapi_doc_version__: str = "1.0.0"
 __gen_version__: str = "2.634.2"
-__user_agent__: str = "speakeasy-sdk/python 1.9.7 2.634.2 1.0.0 mistralai"
+__user_agent__: str = "speakeasy-sdk/python 1.9.8 2.634.2 1.0.0 mistralai"
 
 try:
     if __package__ is not None:

{mistralai-1.9.7.dist-info → mistralai-1.9.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: mistralai
-Version: 1.9.7
+Version: 1.9.8
 Summary: Python Client SDK for the Mistral AI API.
 Author: Mistral
 Requires-Python: >=3.9

{mistralai-1.9.7.dist-info → mistralai-1.9.8.dist-info}/RECORD

@@ -3,32 +3,41 @@ mistralai_azure/_hooks/__init__.py,sha256=9_7W5jAYw8rcO8Kfc-Ty-lB82BHfksAJJpVFb_
 mistralai_azure/_hooks/custom_user_agent.py,sha256=0m-1JzJxOT42rvRTEuCiFLqbOMriOlsraSrAGaXAbyo,656
 mistralai_azure/_hooks/registration.py,sha256=5BN-U92pwP5kUaN7EOso2vWrwZlLvRcU5Coccibqp20,741
 mistralai_azure/_hooks/sdkhooks.py,sha256=urOhVMYX_n5KgMoNDNmGs4fsgUWoeSG6_GarhPxH-YU,2565
-mistralai_azure/_hooks/types.py,sha256=5lbjAtBy4DcEmoFjepuZA4l3vjE73G_NW5izQHi3DK0,2818
-mistralai_azure/_version.py,sha256=qa1zEa5rDcZbGrfBSLYlvEl14a0nG1VfMvKuRuXKSwQ,472
-mistralai_azure/basesdk.py,sha256=95JOT11O1oU74EcPvwu3lj9o32a7IPInYIveTVjV8pE,12136
-mistralai_azure/chat.py,sha256=6zTDL9fbR7v4n23Awgu7P_9nPKeum2W5IoX7ubumn44,35910
-mistralai_azure/httpclient.py,sha256=lC-YQ7q4yiJGKElxBeb3aZnr-4aYxjgEpZ6roeXYlyg,4318
-mistralai_azure/models/__init__.py,sha256=zByHopg2jAg0Sfj7cfw2tD7ioV6bjK7s61KqZkiDnfA,6075
-mistralai_azure/models/assistantmessage.py,sha256=OmHqIM8Cnp4gW6_NbEGMam-_-XBDqMOdskb4BejEBZY,2655
+mistralai_azure/_hooks/types.py,sha256=F5N_UYekLG4DK9X22awmKxTefZCtmJ8JoH7MQ_1vvq8,3059
+mistralai_azure/_version.py,sha256=v9Stuq-LTWVt4os7IEen_TdltjSyuWmz27kACZK3WmY,472
+mistralai_azure/basesdk.py,sha256=GDFjY61iCGIMrrhM_m5DSeyMZV27X2MEwCb4QG7SK0s,11810
+mistralai_azure/chat.py,sha256=GN3z1pP9Cq8W2l5GNS4lgvE1Jtmj_wDZI-5-ALTozro,37218
+mistralai_azure/httpclient.py,sha256=Eu73urOAiZQtdUIyOUnPccxCiBbWEKrXG-JrRG3SLM4,3946
+mistralai_azure/models/__init__.py,sha256=mEq5qJ0R8y0G28Hsnme4xFZ6ZsN9TAfJkVpV-9gecWU,14188
+mistralai_azure/models/assistantmessage.py,sha256=aR4_dCNMC5gaViAGN1g-HvBdvkJoe3LMLG1xKbPQU9U,2661
 mistralai_azure/models/chatcompletionchoice.py,sha256=-JE13p36mWnyc3zxnHLJp1Q43QVgj5QRurnZslXdJc0,935
-mistralai_azure/models/chatcompletionrequest.py,sha256=ixQP91BFZv1Zebxq0ephBmHyI5W0_yeEQoNR13Z5QdU,9763
-mistralai_azure/models/chatcompletionresponse.py,sha256=sPmb4kih2DpE3r8Xem_HYj6o3E3i-6PyVROvm7Ysrfs,798
-mistralai_azure/models/chatcompletionstreamrequest.py,sha256=o8cbpKDS0scF_B3dfojOUhnLTd7D2G6AT-anOVe9ZFo,8899
+mistralai_azure/models/chatcompletionrequest.py,sha256=w2VNBLoNOT55acEBt9STL6t3YnKxTIG8ZO1TjPSWHMM,10515
+mistralai_azure/models/chatcompletionresponse.py,sha256=Yn7xcGIzmc2i4jSpq53YgmwOSKYk8oXKyIYWo8QjYfo,715
+mistralai_azure/models/chatcompletionstreamrequest.py,sha256=ewycVKlbjL6Y-DY-99xw3-WtwAr7Z67Q5oCNz0cUs6s,9651
 mistralai_azure/models/completionchunk.py,sha256=yoA0tYoyK5RChQPbEvYUi1BVmuyH-QT5IYwEYJNtsXM,877
 mistralai_azure/models/completionevent.py,sha256=8wkRAMMpDFfhFSm7OEmli80lsK98Tir7R6IxW-KxeuE,405
-mistralai_azure/models/completionresponsestreamchoice.py,sha256=c6BncIEgKnK4HUPCeIhLfVc3RgxXKNcxp2JrlObUu9E,1834
+mistralai_azure/models/completionresponsestreamchoice.py,sha256=ysp8H1ztx8B_jlnqKgcwEb7St0BCEWKR1QhnPRhxBvU,1840
 mistralai_azure/models/contentchunk.py,sha256=a7A9ymr1Qvg4am-uqrGxqrmTf9NBMPiGbVncuOevchE,881
-mistralai_azure/models/deltamessage.py,sha256=DvXCMs-P1i3QlUjCjJv4en2d04ydTrH6AjECpX9L2aw,1970
+mistralai_azure/models/deltamessage.py,sha256=uDjAlgTy_W9wIX7qQUWbwsf7zSLn0GkBvYyIg7d_sto,1976
+mistralai_azure/models/documenturlchunk.py,sha256=bxo2zp8MmL-be5AMGLD2XBCmgfzJMVc7-relOfKsz3o,1741
+mistralai_azure/models/filechunk.py,sha256=wR69lE1QRNUwYDgKOM1ZXg5-8qQjomZ0tPzSUXgxqGI,653
 mistralai_azure/models/function.py,sha256=VKcPB1oJ8_jvfXRfqufa2Y9to5WdxS-hi9OLu78GNpM,540
 mistralai_azure/models/functioncall.py,sha256=H2eemkzk2Zm1LEm11atVh6PGvr6XJn9SWqNUziT_WK8,562
 mistralai_azure/models/functionname.py,sha256=4rGsO-FYjvLMRGDBbdZ3cLyiiwml_voRQQ924K2_S1M,473
 mistralai_azure/models/httpvalidationerror.py,sha256=tcUK2zfyCZ1TJjmvF93E9G2Ah-S2UUSpM-ZJBbR4hgc,616
-mistralai_azure/models/imageurl.py,sha256=Dm3S96XCb-F11vx3HYWnKG5GOm246q21vDJ81ywVDVQ,1396
+mistralai_azure/models/imageurl.py,sha256=TUOpT-mTQJ9VoMoVcN303592gGBgrx6NZmmDhui-8h8,1402
 mistralai_azure/models/imageurlchunk.py,sha256=JWfOtcxm-AEzRdNny-KWAWXV275hSnWFfn_Ux6OjrYA,1000
-mistralai_azure/models/jsonschema.py,sha256=Hg6iOf3AiR55dX_-4nb0DMcA4TFJQac-51QtjmrcTBE,1683
+mistralai_azure/models/jsonschema.py,sha256=Cu_coSyKmQB3tB_6PLKmcsdaOeyuAJDx5Z6Y1POuJ0Q,1689
+mistralai_azure/models/mistralpromptmode.py,sha256=WDDqWnoJyI_krKYEOWNbWNYByFnLEYED_6hoYUZY_AE,259
+mistralai_azure/models/ocrimageobject.py,sha256=FmmDchJs3dB7ctG_S-pFoUBhHvaLs20fJl3oya32VOo,2839
+mistralai_azure/models/ocrpagedimensions.py,sha256=y3tkCQcdNUd_QWi3rzH6iROyVzsr4f-hQ8qh8bbbmgI,615
+mistralai_azure/models/ocrpageobject.py,sha256=wF0iD_OpB-FFsLyQTuf7BExUMtAw0hMPGX4l1z3rXo8,2097
+mistralai_azure/models/ocrrequest.py,sha256=N56QyuUst9uYasdioz6607RPEQuUs7xPNWtpZsJvHGo,4368
+mistralai_azure/models/ocrresponse.py,sha256=BXz6SzdBWPELQoNd6dp6o8E3b3KlN1oUNMq4-JHquE4,2079
+mistralai_azure/models/ocrusageinfo.py,sha256=mWKzoh9WQKeQOuQrSagn2LlXj3rvPHml_84UtEgalp0,1608
 mistralai_azure/models/prediction.py,sha256=GERxBI8NoS9Fc14FD4ityVfJfXNts1dxjoK3XIVHHc0,730
 mistralai_azure/models/referencechunk.py,sha256=uiouhIPrWpVEhpY_Cea1Som9XapC4mM3R82hhND-j-s,525
-mistralai_azure/models/responseformat.py,sha256=n0aKQE1girltBvrih5w4bbfp_C7_ban4KTrGpS4bAFM,2256
+mistralai_azure/models/responseformat.py,sha256=A82NwuTKo-xI7b6VPkOwgQnIkGpzRQAo4zbSjdyXhzw,2262
 mistralai_azure/models/responseformats.py,sha256=O9lwS2M9m53DsRxTC4uRP12SvRhgaQoMjIYsDys5A7s,503
 mistralai_azure/models/sdkerror.py,sha256=kd75e3JYF2TXNgRZopcV-oGdBWoBZqRcvrwqn2fsFYs,528
 mistralai_azure/models/security.py,sha256=lPLcQ1OV2SA6ZJP5_lOFWUDVuPc-L90C3N127KMWdPo,627
@@ -38,21 +47,23 @@ mistralai_azure/models/tool.py,sha256=Li0qpB3KgGN0mtT8lKG1N_MfOOwGvzok0ZRK_J3Us8
 mistralai_azure/models/toolcall.py,sha256=MYHTegL2wzO23cG9AyPS9YhomXWh8ekULwzIeGt31Pw,836
 mistralai_azure/models/toolchoice.py,sha256=etDg86Frx-VoiccMlGP_Va3Vipy4UGMa9LMUGQFY6UY,1033
 mistralai_azure/models/toolchoiceenum.py,sha256=Ca4ileCwuOjfPzIXLRIxT3RkE5zR7oqV6nXU-UjW0w0,197
-mistralai_azure/models/toolmessage.py,sha256=Vnq3QXhCYqECfGOjbkK8ZA2hJwbgxhxgZU_lpREyVhk,2069
+mistralai_azure/models/toolmessage.py,sha256=I-zRrc1Qs_l0lfNTeGSP90SohNUuzv1q3TbigaB7y8w,2075
 mistralai_azure/models/tooltypes.py,sha256=AGC_JaMGWyMRJ1rCIGhLh5DWbyohdiQkEeKoW5a97Ro,250
-mistralai_azure/models/usageinfo.py,sha256=jG6lRE1t4wDqD4Cote82IFLQtWA_eJmTwP66TI8botg,407
-mistralai_azure/models/usermessage.py,sha256=U8b5KMT3b0j8AOLFjCMWjjCM3zBl54Vc-Rzc5qJz1sc,1799
+mistralai_azure/models/usageinfo.py,sha256=u9Ds73z87KyluvVCLjlTXnzCVjbBGlLWTBBPpGK6AcM,2349
+mistralai_azure/models/usermessage.py,sha256=tzTfEH0wzDHf96C8AJanTxckrDJOpxtMOa3tiLHDxs4,1805
 mistralai_azure/models/validationerror.py,sha256=vghbUqW9H5AsbYmW5i0C56eHPFC054x8SJA-mJZPKak,532
+mistralai_azure/ocr.py,sha256=hHX6-72A_nSRt8aPOaBf5DPP9NVKdqddRkkWObK8KeM,11911
 mistralai_azure/py.typed,sha256=zrp19r0G21lr2yRiMC0f8MFkQFGj9wMpSbboePMg8KM,59
 mistralai_azure/sdk.py,sha256=CVTTW027gX8y4rhprAbmkB5W7r3afru01yRSPwxpMk4,5437
-mistralai_azure/sdkconfiguration.py,sha256=t2a28Th0mVQ2C1R2ljPi8OJxpQ9xmb3hVgWzHXelq_o,1885
+mistralai_azure/sdkconfiguration.py,sha256=tqLLVewof7YZcBIykjMjGZIEtX_b_JKaMP9KHNdrRsM,1728
 mistralai_azure/types/__init__.py,sha256=RArOwSgeeTIva6h-4ttjXwMUeCkz10nAFBL9D-QljI4,377
-mistralai_azure/types/basemodel.py,sha256=PexI39iKiOkIlobB8Ueo0yn8PLHp6_wb-WO-zelNDZY,1170
-mistralai_azure/utils/__init__.py,sha256=Q7llS9EohG8aiwH3X_YC3Ia1erz5qKWHVxfHE6L1_tQ,2403
+mistralai_azure/types/basemodel.py,sha256=L79WXvTECbSqaJzs8D3ud_KdIWkU7Cx2wbohDAktE9E,1127
+mistralai_azure/utils/__init__.py,sha256=811KKdkxMaWDfz2lMohUWqrR4JnIWxqeNQ1lRGQU4DM,5341
 mistralai_azure/utils/annotations.py,sha256=aR7mZG34FzgRdew7WZPYEu9QGBerpuKxCF4sek5Z_5Y,1699
-mistralai_azure/utils/enums.py,sha256=VzjeslROrAr2luZOTJlvu-4UlxgTaGOKlRYtJJ7IfyY,1006
+mistralai_azure/utils/datetimes.py,sha256=oppAA5e3V35pQov1-FNLKxAaNF1_XWi-bQtyjjql3H8,855
+mistralai_azure/utils/enums.py,sha256=REU6ydF8gsVL3xaeGX4sMNyiL3q5P9h29-f6Sa6luAE,2633
 mistralai_azure/utils/eventstreaming.py,sha256=LtcrfJYw4nP2Oe4Wl0-cEURLzRGYReRGWNFY5wYECIE,6186
-mistralai_azure/utils/forms.py,sha256=YSSijXrsM2nfrRHlPQejh1uRRKfoILomHL3d9xpJiy8,6058
+mistralai_azure/utils/forms.py,sha256=EJdnrfIkuwpDtekyHutla0HjI_FypTYcmYNyPKEu_W0,6874
 mistralai_azure/utils/headers.py,sha256=cPxWSmUILrefTGDzTH1Hdj7_Hlsj-EY6K5Tyc4iH4dk,3663
 mistralai_azure/utils/logger.py,sha256=9nUtlKHo3RFsIVyMw5jq3wEKZMVwHnZMSc6xLp-otC0,520
 mistralai_azure/utils/metadata.py,sha256=Per2KFXXOqOtoUWXrlIfjrSrBg199KrRW0nKQDgHIBU,3136
@@ -60,7 +71,7 @@ mistralai_azure/utils/queryparams.py,sha256=MTK6inMS1_WwjmMJEJmAn67tSHHJyarpdGRl
 mistralai_azure/utils/requestbodies.py,sha256=ySjEyjcLi731LNUahWvLOrES2HihuA8VrOJx4eQ7Qzg,2101
 mistralai_azure/utils/retries.py,sha256=6yhfZifqIat9i76xF0lTR2jLj1IN9BNGyqqxATlEFPU,6348
 mistralai_azure/utils/security.py,sha256=ktep3HKwbFs-MLxUYTM8Jd4v-ZBum5_Z0u1PFIdYBX0,5516
-mistralai_azure/utils/serializers.py,sha256=EGH40Pgp3sSK9uM4PxL7_SYzSHtmo-Uy6QIE5xLVg68,5198
+mistralai_azure/utils/serializers.py,sha256=hiHBXM1AY8_N2Z_rvFfNSYwvLBkSQlPGFp8poasdU4s,5986
 mistralai_azure/utils/url.py,sha256=BgGPgcTA6MRK4bF8fjP2dUopN3NzEzxWMXPBVg8NQUA,5254
 mistralai_azure/utils/values.py,sha256=CcaCXEa3xHhkUDROyXZocN8f0bdITftv9Y0P9lTf0YM,3517
 mistralai_gcp/__init__.py,sha256=Tz5Y5FzbIUT1AmaYiTwJI56XTmuldo9AalaAm4h_FdE,423
@@ -139,7 +150,7 @@ mistralai/_hooks/deprecation_warning.py,sha256=eyEOf7-o9uqqNWJnufD2RXp3dYrGV4in9
 mistralai/_hooks/registration.py,sha256=ML0W-XbE4WYdJ4eGks_XxF2aLCJTaIWjQATFGzFwvyU,861
 mistralai/_hooks/sdkhooks.py,sha256=s-orhdvnV89TmI3QiPC2LWQtYeM9RrsG1CTll-fYZmQ,2559
 mistralai/_hooks/types.py,sha256=70IiFr5bfsJYafuDkXQWVfl6nY4dQkA5SZoEBCircqs,3047
-mistralai/_version.py,sha256=oCbrLM0xwhq7nYsvgqHLY0N9qeBd3fWSRVuToJh4IME,460
+mistralai/_version.py,sha256=6FjB8zY1tFIrap8yREauj7OwtQdW0EpeYdqcdGIlUwA,460
 mistralai/accesses.py,sha256=o5VCW-nyOfwzopHo0jSXrPKJ1fqJL88PZto1RwysXPA,27320
 mistralai/agents.py,sha256=3E-c1YZOp3mS1PqA6OYekZmOcAdLCqWfq1o-hqUMsIw,33960
 mistralai/async_client.py,sha256=KUdYxIIqoD6L7vB0EGwUR6lQ0NK5iCTHjnLVR9CVcJY,355
@@ -445,7 +456,7 @@ mistralai/utils/security.py,sha256=vWlpkikOnGN_HRRhJ7Pb8ywVAjiM3d3ey3oTWtM6jTU,6
 mistralai/utils/serializers.py,sha256=hiHBXM1AY8_N2Z_rvFfNSYwvLBkSQlPGFp8poasdU4s,5986
 mistralai/utils/url.py,sha256=BgGPgcTA6MRK4bF8fjP2dUopN3NzEzxWMXPBVg8NQUA,5254
 mistralai/utils/values.py,sha256=CcaCXEa3xHhkUDROyXZocN8f0bdITftv9Y0P9lTf0YM,3517
-mistralai-1.9.7.dist-info/LICENSE,sha256=rUtQ_9GD0OyLPlb-2uWVdfE87hzudMRmsW-tS-0DK-0,11340
-mistralai-1.9.7.dist-info/METADATA,sha256=AFE0MZduqkuOApBCvcJpWvdSbQzAc4Naj6XANpiDHQw,37249
-mistralai-1.9.7.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-mistralai-1.9.7.dist-info/RECORD,,
+mistralai-1.9.8.dist-info/LICENSE,sha256=rUtQ_9GD0OyLPlb-2uWVdfE87hzudMRmsW-tS-0DK-0,11340
+mistralai-1.9.8.dist-info/METADATA,sha256=1Dbq0gOz5mYZMZq-iWpXAabW8wXQ04G8GrRQe1Il8Oc,37249
+mistralai-1.9.8.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+mistralai-1.9.8.dist-info/RECORD,,

mistralai_azure/_hooks/types.py

@@ -3,10 +3,12 @@
 from abc import ABC, abstractmethod
 import httpx
 from mistralai_azure.httpclient import HttpClient
+from mistralai_azure.sdkconfiguration import SDKConfiguration
 from typing import Any, Callable, List, Optional, Tuple, Union
 
 
 class HookContext:
+    config: SDKConfiguration
     base_url: str
     operation_id: str
     oauth2_scopes: Optional[List[str]] = None
@@ -14,11 +16,13 @@ class HookContext:
 
     def __init__(
         self,
+        config: SDKConfiguration,
         base_url: str,
         operation_id: str,
         oauth2_scopes: Optional[List[str]],
         security_source: Optional[Union[Any, Callable[[], Any]]],
     ):
+        self.config = config
         self.base_url = base_url
         self.operation_id = operation_id
         self.oauth2_scopes = oauth2_scopes
@@ -28,6 +32,7 @@ class HookContext:
 class BeforeRequestContext(HookContext):
     def __init__(self, hook_ctx: HookContext):
         super().__init__(
+            hook_ctx.config,
             hook_ctx.base_url,
             hook_ctx.operation_id,
             hook_ctx.oauth2_scopes,
@@ -38,6 +43,7 @@ class BeforeRequestContext(HookContext):
 class AfterSuccessContext(HookContext):
     def __init__(self, hook_ctx: HookContext):
         super().__init__(
+            hook_ctx.config,
             hook_ctx.base_url,
             hook_ctx.operation_id,
             hook_ctx.oauth2_scopes,
@@ -48,6 +54,7 @@ class AfterSuccessContext(HookContext):
 class AfterErrorContext(HookContext):
     def __init__(self, hook_ctx: HookContext):
         super().__init__(
+            hook_ctx.config,
             hook_ctx.base_url,
             hook_ctx.operation_id,
             hook_ctx.oauth2_scopes,

mistralai_azure/_version.py

@@ -4,9 +4,9 @@ import importlib.metadata
 
 __title__: str = "mistralai_azure"
 __version__: str = "1.6.0"
-__openapi_doc_version__: str = "0.0.2"
-__gen_version__: str = "2.548.6"
-__user_agent__: str = "speakeasy-sdk/python 1.6.0 2.548.6 0.0.2 mistralai_azure"
+__openapi_doc_version__: str = "1.0.0"
+__gen_version__: str = "2.634.2"
+__user_agent__: str = "speakeasy-sdk/python 1.6.0 2.634.2 1.0.0 mistralai_azure"
 
 try:
     if __package__ is not None:

mistralai_azure/basesdk.py

@@ -218,12 +218,12 @@ class BaseSDK:
         client = self.sdk_configuration.client
         logger = self.sdk_configuration.debug_logger
 
+        hooks = self.sdk_configuration.__dict__["_hooks"]
+
         def do():
             http_res = None
             try:
-                req = self.sdk_configuration.get_hooks().before_request(
-                    BeforeRequestContext(hook_ctx), request
-                )
+                req = hooks.before_request(BeforeRequestContext(hook_ctx), request)
                 logger.debug(
                     "Request:\nMethod: %s\nURL: %s\nHeaders: %s\nBody: %s",
                     req.method,
@@ -237,9 +237,7 @@ class BaseSDK:
 
                 http_res = client.send(req, stream=stream)
             except Exception as e:
-                _, e = self.sdk_configuration.get_hooks().after_error(
-                    AfterErrorContext(hook_ctx), None, e
-                )
+                _, e = hooks.after_error(AfterErrorContext(hook_ctx), None, e)
                 if e is not None:
                     logger.debug("Request Exception", exc_info=True)
                     raise e
@@ -257,7 +255,7 @@ class BaseSDK:
             )
 
             if utils.match_status_codes(error_status_codes, http_res.status_code):
-                result, err = self.sdk_configuration.get_hooks().after_error(
+                result, err = hooks.after_error(
                     AfterErrorContext(hook_ctx), http_res, None
                 )
                 if err is not None:
@@ -277,9 +275,7 @@ class BaseSDK:
             http_res = do()
 
         if not utils.match_status_codes(error_status_codes, http_res.status_code):
-            http_res = self.sdk_configuration.get_hooks().after_success(
-                AfterSuccessContext(hook_ctx), http_res
-            )
+            http_res = hooks.after_success(AfterSuccessContext(hook_ctx), http_res)
 
         return http_res
 
@@ -294,12 +290,12 @@ class BaseSDK:
         client = self.sdk_configuration.async_client
         logger = self.sdk_configuration.debug_logger
 
+        hooks = self.sdk_configuration.__dict__["_hooks"]
+
         async def do():
             http_res = None
             try:
-                req = self.sdk_configuration.get_hooks().before_request(
-                    BeforeRequestContext(hook_ctx), request
-                )
+                req = hooks.before_request(BeforeRequestContext(hook_ctx), request)
                 logger.debug(
                     "Request:\nMethod: %s\nURL: %s\nHeaders: %s\nBody: %s",
                     req.method,
@@ -313,9 +309,7 @@ class BaseSDK:
 
                 http_res = await client.send(req, stream=stream)
             except Exception as e:
-                _, e = self.sdk_configuration.get_hooks().after_error(
-                    AfterErrorContext(hook_ctx), None, e
-                )
+                _, e = hooks.after_error(AfterErrorContext(hook_ctx), None, e)
                 if e is not None:
                     logger.debug("Request Exception", exc_info=True)
                     raise e
@@ -333,7 +327,7 @@ class BaseSDK:
             )
 
             if utils.match_status_codes(error_status_codes, http_res.status_code):
-                result, err = self.sdk_configuration.get_hooks().after_error(
+                result, err = hooks.after_error(
                     AfterErrorContext(hook_ctx), http_res, None
                 )
                 if err is not None:
@@ -355,8 +349,6 @@ class BaseSDK:
             http_res = await do()
 
         if not utils.match_status_codes(error_status_codes, http_res.status_code):
-            http_res = self.sdk_configuration.get_hooks().after_success(
-                AfterSuccessContext(hook_ctx), http_res
-            )
+            http_res = hooks.after_success(AfterSuccessContext(hook_ctx), http_res)
 
         return http_res

mistralai_azure/chat.py

@@ -41,6 +41,7 @@ class Chat(BaseSDK):
             Union[models.Prediction, models.PredictionTypedDict]
         ] = None,
         parallel_tool_calls: Optional[bool] = None,
+        prompt_mode: OptionalNullable[models.MistralPromptMode] = UNSET,
         safe_prompt: Optional[bool] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
@@ -67,6 +68,7 @@ class Chat(BaseSDK):
         :param n: Number of completions to return for each request, input tokens are only billed once.
         :param prediction:
         :param parallel_tool_calls:
+        :param prompt_mode: Allows toggling between the reasoning mode and no system prompt. When set to `reasoning` the system prompt for reasoning models will be used.
         :param safe_prompt: Whether to inject a safety prompt before all conversations.
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
@@ -106,6 +108,7 @@ class Chat(BaseSDK):
                 prediction, Optional[models.Prediction]
             ),
             parallel_tool_calls=parallel_tool_calls,
+            prompt_mode=prompt_mode,
             safe_prompt=safe_prompt,
         )
 
@@ -138,6 +141,7 @@ class Chat(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="stream_chat",
                 oauth2_scopes=[],
@@ -212,6 +216,7 @@ class Chat(BaseSDK):
             Union[models.Prediction, models.PredictionTypedDict]
         ] = None,
         parallel_tool_calls: Optional[bool] = None,
+        prompt_mode: OptionalNullable[models.MistralPromptMode] = UNSET,
         safe_prompt: Optional[bool] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
@@ -238,6 +243,7 @@ class Chat(BaseSDK):
         :param n: Number of completions to return for each request, input tokens are only billed once.
         :param prediction:
         :param parallel_tool_calls:
+        :param prompt_mode: Allows toggling between the reasoning mode and no system prompt. When set to `reasoning` the system prompt for reasoning models will be used.
         :param safe_prompt: Whether to inject a safety prompt before all conversations.
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
@@ -277,6 +283,7 @@ class Chat(BaseSDK):
                 prediction, Optional[models.Prediction]
             ),
             parallel_tool_calls=parallel_tool_calls,
+            prompt_mode=prompt_mode,
             safe_prompt=safe_prompt,
         )
 
@@ -309,6 +316,7 @@ class Chat(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="stream_chat",
                 oauth2_scopes=[],
@@ -391,6 +399,7 @@ class Chat(BaseSDK):
             Union[models.Prediction, models.PredictionTypedDict]
         ] = None,
         parallel_tool_calls: Optional[bool] = None,
+        prompt_mode: OptionalNullable[models.MistralPromptMode] = UNSET,
         safe_prompt: Optional[bool] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
@@ -415,6 +424,7 @@ class Chat(BaseSDK):
         :param n: Number of completions to return for each request, input tokens are only billed once.
         :param prediction:
         :param parallel_tool_calls:
+        :param prompt_mode: Allows toggling between the reasoning mode and no system prompt. When set to `reasoning` the system prompt for reasoning models will be used.
         :param safe_prompt: Whether to inject a safety prompt before all conversations.
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
@@ -456,6 +466,7 @@ class Chat(BaseSDK):
                 prediction, Optional[models.Prediction]
             ),
             parallel_tool_calls=parallel_tool_calls,
+            prompt_mode=prompt_mode,
             safe_prompt=safe_prompt,
         )
 
@@ -488,6 +499,7 @@ class Chat(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="chat_completion_v1_chat_completions_post",
                 oauth2_scopes=[],
@@ -566,6 +578,7 @@ class Chat(BaseSDK):
             Union[models.Prediction, models.PredictionTypedDict]
         ] = None,
         parallel_tool_calls: Optional[bool] = None,
+        prompt_mode: OptionalNullable[models.MistralPromptMode] = UNSET,
         safe_prompt: Optional[bool] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
@@ -590,6 +603,7 @@ class Chat(BaseSDK):
         :param n: Number of completions to return for each request, input tokens are only billed once.
         :param prediction:
         :param parallel_tool_calls:
+        :param prompt_mode: Allows toggling between the reasoning mode and no system prompt. When set to `reasoning` the system prompt for reasoning models will be used.
         :param safe_prompt: Whether to inject a safety prompt before all conversations.
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
@@ -631,6 +645,7 @@ class Chat(BaseSDK):
                 prediction, Optional[models.Prediction]
             ),
             parallel_tool_calls=parallel_tool_calls,
+            prompt_mode=prompt_mode,
             safe_prompt=safe_prompt,
         )
 
@@ -663,6 +678,7 @@ class Chat(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="chat_completion_v1_chat_completions_post",
                 oauth2_scopes=[],

mistralai_azure/httpclient.py

@@ -2,7 +2,6 @@
 
 # pyright: reportReturnType = false
 import asyncio
-from concurrent.futures import ThreadPoolExecutor
 from typing_extensions import Protocol, runtime_checkable
 import httpx
 from typing import Any, Optional, Union
@@ -116,21 +115,12 @@ def close_clients(
             pass
 
     if async_client is not None and not async_client_supplied:
-        is_async = False
         try:
-            asyncio.get_running_loop()
-            is_async = True
+            loop = asyncio.get_running_loop()
+            asyncio.run_coroutine_threadsafe(async_client.aclose(), loop)
         except RuntimeError:
-            pass
-
-        try:
-            # If this function is called in an async loop then start another
-            # loop in a separate thread to close the async http client.
-            if is_async:
-                with ThreadPoolExecutor(max_workers=1) as executor:
-                    future = executor.submit(asyncio.run, async_client.aclose())
-                    future.result()
-            else:
+            try:
                 asyncio.run(async_client.aclose())
-        except Exception:
-            pass
+            except RuntimeError:
+                # best effort
+                pass