mistralai 1.2.2__py3-none-any.whl → 1.2.4__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 (89) hide show
  1. mistralai/_version.py +1 -1
  2. mistralai/agents.py +5 -5
  3. mistralai/chat.py +5 -5
  4. mistralai/files.py +166 -0
  5. mistralai/fim.py +5 -5
  6. mistralai/httpclient.py +6 -0
  7. mistralai/jobs.py +2 -2
  8. mistralai/models/__init__.py +22 -3
  9. mistralai/models/agentscompletionrequest.py +23 -11
  10. mistralai/models/agentscompletionstreamrequest.py +23 -13
  11. mistralai/models/apiendpoint.py +11 -3
  12. mistralai/models/assistantmessage.py +7 -3
  13. mistralai/models/batchjobin.py +4 -2
  14. mistralai/models/chatclassificationrequest.py +26 -17
  15. mistralai/models/chatcompletionrequest.py +19 -11
  16. mistralai/models/chatcompletionstreamrequest.py +23 -13
  17. mistralai/models/classificationrequest.py +7 -3
  18. mistralai/models/contentchunk.py +9 -3
  19. mistralai/models/deltamessage.py +5 -3
  20. mistralai/models/detailedjobout.py +2 -3
  21. mistralai/models/embeddingrequest.py +3 -3
  22. mistralai/models/files_api_routes_get_signed_urlop.py +25 -0
  23. mistralai/models/filesignedurl.py +13 -0
  24. mistralai/models/fimcompletionrequest.py +7 -3
  25. mistralai/models/fimcompletionstreamrequest.py +7 -3
  26. mistralai/models/functioncall.py +3 -3
  27. mistralai/models/imageurlchunk.py +9 -14
  28. mistralai/models/jobin.py +2 -3
  29. mistralai/models/jobout.py +2 -3
  30. mistralai/models/jobs_api_routes_fine_tuning_create_fine_tuning_jobop.py +9 -4
  31. mistralai/models/modellist.py +4 -2
  32. mistralai/models/referencechunk.py +20 -0
  33. mistralai/models/retrieve_model_v1_models_model_id_getop.py +5 -4
  34. mistralai/models/systemmessage.py +7 -3
  35. mistralai/models/textchunk.py +3 -9
  36. mistralai/models/toolmessage.py +14 -5
  37. mistralai/models/usermessage.py +5 -3
  38. mistralai/models/validationerror.py +3 -3
  39. mistralai/sdk.py +14 -0
  40. mistralai/sdkconfiguration.py +3 -3
  41. mistralai/utils/annotations.py +42 -17
  42. mistralai/utils/eventstreaming.py +61 -1
  43. {mistralai-1.2.2.dist-info → mistralai-1.2.4.dist-info}/METADATA +181 -176
  44. {mistralai-1.2.2.dist-info → mistralai-1.2.4.dist-info}/RECORD +88 -84
  45. mistralai_azure/_version.py +1 -1
  46. mistralai_azure/chat.py +5 -5
  47. mistralai_azure/httpclient.py +6 -0
  48. mistralai_azure/models/__init__.py +13 -1
  49. mistralai_azure/models/assistantmessage.py +7 -3
  50. mistralai_azure/models/chatcompletionrequest.py +23 -11
  51. mistralai_azure/models/chatcompletionstreamrequest.py +19 -13
  52. mistralai_azure/models/contentchunk.py +14 -2
  53. mistralai_azure/models/deltamessage.py +5 -3
  54. mistralai_azure/models/functioncall.py +3 -3
  55. mistralai_azure/models/referencechunk.py +20 -0
  56. mistralai_azure/models/systemmessage.py +7 -3
  57. mistralai_azure/models/textchunk.py +3 -9
  58. mistralai_azure/models/toolmessage.py +14 -5
  59. mistralai_azure/models/usermessage.py +5 -3
  60. mistralai_azure/models/validationerror.py +3 -3
  61. mistralai_azure/sdkconfiguration.py +3 -3
  62. mistralai_azure/utils/annotations.py +42 -17
  63. mistralai_azure/utils/eventstreaming.py +61 -1
  64. mistralai_gcp/_version.py +1 -1
  65. mistralai_gcp/chat.py +5 -5
  66. mistralai_gcp/fim.py +5 -5
  67. mistralai_gcp/httpclient.py +6 -0
  68. mistralai_gcp/models/__init__.py +13 -1
  69. mistralai_gcp/models/assistantmessage.py +7 -3
  70. mistralai_gcp/models/chatcompletionrequest.py +23 -11
  71. mistralai_gcp/models/chatcompletionstreamrequest.py +19 -13
  72. mistralai_gcp/models/contentchunk.py +14 -2
  73. mistralai_gcp/models/deltamessage.py +5 -3
  74. mistralai_gcp/models/fimcompletionrequest.py +7 -3
  75. mistralai_gcp/models/fimcompletionstreamrequest.py +7 -3
  76. mistralai_gcp/models/functioncall.py +3 -3
  77. mistralai_gcp/models/referencechunk.py +20 -0
  78. mistralai_gcp/models/systemmessage.py +7 -3
  79. mistralai_gcp/models/textchunk.py +3 -9
  80. mistralai_gcp/models/toolmessage.py +14 -5
  81. mistralai_gcp/models/usermessage.py +5 -3
  82. mistralai_gcp/models/validationerror.py +3 -3
  83. mistralai_gcp/sdk.py +5 -4
  84. mistralai_gcp/sdkconfiguration.py +3 -3
  85. mistralai_gcp/utils/annotations.py +42 -17
  86. mistralai_gcp/utils/eventstreaming.py +61 -1
  87. mistralai/models/finetuneablemodel.py +0 -14
  88. {mistralai-1.2.2.dist-info → mistralai-1.2.4.dist-info}/LICENSE +0 -0
  89. {mistralai-1.2.2.dist-info → mistralai-1.2.4.dist-info}/WHEEL +0 -0
@@ -4,49 +4,50 @@ mistralai_azure/_hooks/custom_user_agent.py,sha256=1VAY7_oknRQGL8QNO0uCRHg6r7XYl
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
7
- mistralai_azure/_version.py,sha256=xy-G9WR4FHliLcNY_yQRTHXr2SIawBzvaUAIakk182Y,319
7
+ mistralai_azure/_version.py,sha256=XUjO99pmQX1np8WHgz8F48bgOPxWExv6D_pZv0-mKks,319
8
8
  mistralai_azure/basesdk.py,sha256=ZtCvahXCyuWen9Jk7BC8BoeChQZCb04JAA6oG-0SOaE,11218
9
- mistralai_azure/chat.py,sha256=6nuvRX1Gf3HTVPCHN31RMsL1Vx_jgpu1Q6XmhaJ9r9M,31957
10
- mistralai_azure/httpclient.py,sha256=S_ItzEchFX-znIdHD6i5-a91H0Dn5QxpT0KhucdHBbI,2595
11
- mistralai_azure/models/__init__.py,sha256=HDWxNFXEXr4cCDrJ_QjDGEwNlf7CrneyoCd2Hrha8GI,5172
12
- mistralai_azure/models/assistantmessage.py,sha256=B8Q9_bgiIKne8sJzvWqiff2XehYkOBuBLWuGFKr_s1E,2115
9
+ mistralai_azure/chat.py,sha256=oXfpD7auiYPjeSsph-oDsaURnXOjO1WaILJB1fOB6Ng,31941
10
+ mistralai_azure/httpclient.py,sha256=WDbLpMzo7qmWki_ryOJcCAYNI1T4uyWKV08rRuCdNII,2688
11
+ mistralai_azure/models/__init__.py,sha256=tUGnhgKbxZMd8R7DK09Zy0TRlcp5fg32LaHgwE6wtRs,5474
12
+ mistralai_azure/models/assistantmessage.py,sha256=RfLzXdsZ0VIo-x5I10YveA5lRu0caYZ6ZYqqUvV8T5Y,2235
13
13
  mistralai_azure/models/chatcompletionchoice.py,sha256=-JE13p36mWnyc3zxnHLJp1Q43QVgj5QRurnZslXdJc0,935
14
- mistralai_azure/models/chatcompletionrequest.py,sha256=VrjROK0FypeHKfq-ppp1SPNXdH4Ut70BuKQisFV_VYQ,9231
14
+ mistralai_azure/models/chatcompletionrequest.py,sha256=28venygprMoniioY9EWDWM3QE32seOg8QN4KTiPLdQ0,9569
15
15
  mistralai_azure/models/chatcompletionresponse.py,sha256=sPmb4kih2DpE3r8Xem_HYj6o3E3i-6PyVROvm7Ysrfs,798
16
- mistralai_azure/models/chatcompletionstreamrequest.py,sha256=XugmBCTRm2IB183Wrz4de1DlpZIKdltGEnEnLCIushg,8436
16
+ mistralai_azure/models/chatcompletionstreamrequest.py,sha256=UB7SwnleDy5lnbTXsqisiEYBGF-gg1YJ6UfTLgKB5zE,8705
17
17
  mistralai_azure/models/completionchunk.py,sha256=yoA0tYoyK5RChQPbEvYUi1BVmuyH-QT5IYwEYJNtsXM,877
18
18
  mistralai_azure/models/completionevent.py,sha256=8wkRAMMpDFfhFSm7OEmli80lsK98Tir7R6IxW-KxeuE,405
19
19
  mistralai_azure/models/completionresponsestreamchoice.py,sha256=c6BncIEgKnK4HUPCeIhLfVc3RgxXKNcxp2JrlObUu9E,1834
20
- mistralai_azure/models/contentchunk.py,sha256=3lXlnRUBHMl2ZRiZ5DlaPU4SFh7-rLttRmbsokOgSMQ,233
21
- mistralai_azure/models/deltamessage.py,sha256=1Mou57fy_Fppdn9SA5mewAfZtwoN9-mtX1WMcI2Qeak,1888
20
+ mistralai_azure/models/contentchunk.py,sha256=k2eowNYDzv-DH6gyNqmCEtNPCNAgmWDswX15sFb7SQk,726
21
+ mistralai_azure/models/deltamessage.py,sha256=DvXCMs-P1i3QlUjCjJv4en2d04ydTrH6AjECpX9L2aw,1970
22
22
  mistralai_azure/models/function.py,sha256=dMKaqEIbvXIQWf6NROjlVaHGOU6weso7AVCIG691ULI,473
23
- mistralai_azure/models/functioncall.py,sha256=IZ4qiW94J282ooN4Kp96xE_jCt6q-RS3qhc0naCkl-Y,482
23
+ mistralai_azure/models/functioncall.py,sha256=H2eemkzk2Zm1LEm11atVh6PGvr6XJn9SWqNUziT_WK8,562
24
24
  mistralai_azure/models/functionname.py,sha256=4rGsO-FYjvLMRGDBbdZ3cLyiiwml_voRQQ924K2_S1M,473
25
25
  mistralai_azure/models/httpvalidationerror.py,sha256=tcUK2zfyCZ1TJjmvF93E9G2Ah-S2UUSpM-ZJBbR4hgc,616
26
+ mistralai_azure/models/referencechunk.py,sha256=uiouhIPrWpVEhpY_Cea1Som9XapC4mM3R82hhND-j-s,525
26
27
  mistralai_azure/models/responseformat.py,sha256=ObWSXbwECiJJ0j0Ra_Yh2kpcC0zB2IZoGk_HpWBTwtc,1051
27
28
  mistralai_azure/models/responseformats.py,sha256=oeXHoVUoZrZwrz-0cm-rHj5sHygv9MpeqgdummGa8ww,488
28
29
  mistralai_azure/models/sdkerror.py,sha256=kd75e3JYF2TXNgRZopcV-oGdBWoBZqRcvrwqn2fsFYs,528
29
30
  mistralai_azure/models/security.py,sha256=lPLcQ1OV2SA6ZJP5_lOFWUDVuPc-L90C3N127KMWdPo,627
30
- mistralai_azure/models/systemmessage.py,sha256=XGMROJsLI5lca94mv_zwwAWM08CMHPesVY8X0hUyMj4,678
31
- mistralai_azure/models/textchunk.py,sha256=39mTUtsxsndRrW-B88V0OmuJeUW9evWRHvRL8XhiY7c,650
31
+ mistralai_azure/models/systemmessage.py,sha256=8vcbWj6yaGEuDxsCqz4Hdarxt9mJKotFsoxCtoa93vA,792
32
+ mistralai_azure/models/textchunk.py,sha256=D12hZryrlifzFWP5D1W--7sor61Mstdp8fTOyrhK9_8,427
32
33
  mistralai_azure/models/tool.py,sha256=Li0qpB3KgGN0mtT8lKG1N_MfOOwGvzok0ZRK_J3Us80,693
33
34
  mistralai_azure/models/toolcall.py,sha256=RTTdfhcbbH6kq7L1MaQ0Z7bl7xdrE6DZxYPZaIe0lEo,778
34
35
  mistralai_azure/models/toolchoice.py,sha256=etDg86Frx-VoiccMlGP_Va3Vipy4UGMa9LMUGQFY6UY,1033
35
36
  mistralai_azure/models/toolchoiceenum.py,sha256=Ca4ileCwuOjfPzIXLRIxT3RkE5zR7oqV6nXU-UjW0w0,197
36
- mistralai_azure/models/toolmessage.py,sha256=rm7anBMBis06KkNMNxZklazxLoVPBaRmKw8fGkFehZU,1694
37
+ mistralai_azure/models/toolmessage.py,sha256=Vnq3QXhCYqECfGOjbkK8ZA2hJwbgxhxgZU_lpREyVhk,2069
37
38
  mistralai_azure/models/tooltypes.py,sha256=AGC_JaMGWyMRJ1rCIGhLh5DWbyohdiQkEeKoW5a97Ro,250
38
39
  mistralai_azure/models/usageinfo.py,sha256=jG6lRE1t4wDqD4Cote82IFLQtWA_eJmTwP66TI8botg,407
39
- mistralai_azure/models/usermessage.py,sha256=1nU5Hc6Mv3STaopLTcIwNMUACRbJh6kACURtBD11TYI,1695
40
- mistralai_azure/models/validationerror.py,sha256=du2NEF6bMVx_b10g741vkbUE1RIQnO4y4xnukvAsbAY,464
40
+ mistralai_azure/models/usermessage.py,sha256=U8b5KMT3b0j8AOLFjCMWjjCM3zBl54Vc-Rzc5qJz1sc,1799
41
+ mistralai_azure/models/validationerror.py,sha256=vghbUqW9H5AsbYmW5i0C56eHPFC054x8SJA-mJZPKak,532
41
42
  mistralai_azure/py.typed,sha256=zrp19r0G21lr2yRiMC0f8MFkQFGj9wMpSbboePMg8KM,59
42
43
  mistralai_azure/sdk.py,sha256=ZVPE9lFzswEMlHw_kiG_j5vzoJ5BwEYlVLCYGmYDjpA,3936
43
- mistralai_azure/sdkconfiguration.py,sha256=92hCh3g6MWBx0ojZN6ox1oaiUrWoSN3gC3muJuDd7NE,1706
44
+ mistralai_azure/sdkconfiguration.py,sha256=zGJfK4Ouyw18LTVtAQ44j6pm8w4fhS_641HRFHCw1FE,1706
44
45
  mistralai_azure/types/__init__.py,sha256=RArOwSgeeTIva6h-4ttjXwMUeCkz10nAFBL9D-QljI4,377
45
46
  mistralai_azure/types/basemodel.py,sha256=PexI39iKiOkIlobB8Ueo0yn8PLHp6_wb-WO-zelNDZY,1170
46
47
  mistralai_azure/utils/__init__.py,sha256=1x4KfQf2ULtO_aOk_M2RMCimoz8jgcW0RZd-f34KDAs,2365
47
- mistralai_azure/utils/annotations.py,sha256=KFt29x1HavmpmtFN8OGNxTje0gGb3ZTnGi1BcFXIdPc,861
48
+ mistralai_azure/utils/annotations.py,sha256=aR7mZG34FzgRdew7WZPYEu9QGBerpuKxCF4sek5Z_5Y,1699
48
49
  mistralai_azure/utils/enums.py,sha256=VzjeslROrAr2luZOTJlvu-4UlxgTaGOKlRYtJJ7IfyY,1006
49
- mistralai_azure/utils/eventstreaming.py,sha256=klSQ0FURc-hqYz1si1EG92VCFhfaxyBo_xP2mX3BBWo,4917
50
+ mistralai_azure/utils/eventstreaming.py,sha256=LtcrfJYw4nP2Oe4Wl0-cEURLzRGYReRGWNFY5wYECIE,6186
50
51
  mistralai_azure/utils/forms.py,sha256=L-y62DiD0FXnq7WIyw8BAMbdr7sZDwhZ_3YsdviBbaA,6325
51
52
  mistralai_azure/utils/headers.py,sha256=cPxWSmUILrefTGDzTH1Hdj7_Hlsj-EY6K5Tyc4iH4dk,3663
52
53
  mistralai_azure/utils/logger.py,sha256=9nUtlKHo3RFsIVyMw5jq3wEKZMVwHnZMSc6xLp-otC0,520
@@ -64,53 +65,54 @@ mistralai_gcp/_hooks/custom_user_agent.py,sha256=1VAY7_oknRQGL8QNO0uCRHg6r7XYlss
64
65
  mistralai_gcp/_hooks/registration.py,sha256=5BN-U92pwP5kUaN7EOso2vWrwZlLvRcU5Coccibqp20,741
65
66
  mistralai_gcp/_hooks/sdkhooks.py,sha256=nr_ACx8Rn5xvTkmZP6_EI-f_0hw8wMyPqPHNvjAWAxI,2563
66
67
  mistralai_gcp/_hooks/types.py,sha256=DJD5-sKZfAj0tKgSU5w0YNuJE_C8PDa5n1V27T_7SmE,2574
67
- mistralai_gcp/_version.py,sha256=I7PJyyd46iSOwCYkLhy2IMPfVA9cjFyjtLHbN30EInI,317
68
+ mistralai_gcp/_version.py,sha256=yKedfo-mdDZ2lh-brndG2TXcE4w37UWJhP4KIRqMnS0,317
68
69
  mistralai_gcp/basesdk.py,sha256=cJcOSsYxfNznSOMrDlh0z06_e8HXnlIFCfIYm0yOPlc,11212
69
- mistralai_gcp/chat.py,sha256=DXLTCQ159R7hlcMeBN63w3erZv__k36aVEsj6MnsZVg,31873
70
- mistralai_gcp/fim.py,sha256=qVr3jNIbWYXTYEZfj9GTKTAaLG2z7bGCADXTiyNiswk,25395
71
- mistralai_gcp/httpclient.py,sha256=S_ItzEchFX-znIdHD6i5-a91H0Dn5QxpT0KhucdHBbI,2595
72
- mistralai_gcp/models/__init__.py,sha256=UGlyP2Ef3fQH_8IiyHuc3kgD46Ykp-zBzH5TsorMd08,5994
73
- mistralai_gcp/models/assistantmessage.py,sha256=oESoVA3BJZMPmj8ojdPi-wqb0WzXCCToHa9cO4qzAlY,2113
70
+ mistralai_gcp/chat.py,sha256=z17ADwmfkzfCu7XLXDeqdOXzlemEGPluWD3QWvGYpkE,31857
71
+ mistralai_gcp/fim.py,sha256=nqFFDve7SJSF0pyXO2-JoEHDYBjQS5_Z12A5MkjkWqI,25379
72
+ mistralai_gcp/httpclient.py,sha256=WDbLpMzo7qmWki_ryOJcCAYNI1T4uyWKV08rRuCdNII,2688
73
+ mistralai_gcp/models/__init__.py,sha256=FgLN8dDWM4XLCS1EWH--nk90Fnnrypsa2EEu5Yv84iI,6296
74
+ mistralai_gcp/models/assistantmessage.py,sha256=dAdHmWc2sXzeDoP4jCLH5T4wqu7gRUzBlOVfAVmbO7Q,2233
74
75
  mistralai_gcp/models/chatcompletionchoice.py,sha256=1t3Sb_IICDH7gyyEMX-WuxHnSVV-PZTLfpUjkUVp3do,931
75
- mistralai_gcp/models/chatcompletionrequest.py,sha256=yKWizYia91JDpCm6UwRATlgmplrN7IeZM_tJdfEV_8Q,9278
76
+ mistralai_gcp/models/chatcompletionrequest.py,sha256=NKbYDJvnhMSf78-7-BoXY-NjAPa4aikGVAwQrDSvV0Q,9616
76
77
  mistralai_gcp/models/chatcompletionresponse.py,sha256=Ctvqs2ZjvWTycozqXn-fvucgqOn0dm4cOjUZ2BjD4BM,796
77
- mistralai_gcp/models/chatcompletionstreamrequest.py,sha256=UNQ9ioJKTNS2xqXkiMhORcW0sKXw8IuZ8kZMNaNWGEw,8483
78
+ mistralai_gcp/models/chatcompletionstreamrequest.py,sha256=Ste1UKJBSy8JXJ6rnEruHFqHt9B1H5UbMA-rOA6GP40,8752
78
79
  mistralai_gcp/models/completionchunk.py,sha256=0DBDcrqVWrUskHA3hHYtuWk2E4JcJy_zc_LiGyLHBlA,875
79
80
  mistralai_gcp/models/completionevent.py,sha256=cP7Q5dN4Z46FQTlyCYeIwvqt7pgN-22jNPD2bi7Eals,403
80
81
  mistralai_gcp/models/completionresponsestreamchoice.py,sha256=MdZaPMSqFbIbenEAdPyYMFemsFSZdPglEEt5ssZ3x7E,1830
81
- mistralai_gcp/models/contentchunk.py,sha256=3lXlnRUBHMl2ZRiZ5DlaPU4SFh7-rLttRmbsokOgSMQ,233
82
- mistralai_gcp/models/deltamessage.py,sha256=YFmrqPchctnEB-UQdr7Mmd2-UFE86SX47kOQVtYVv1U,1886
83
- mistralai_gcp/models/fimcompletionrequest.py,sha256=ZWpkEGtRmPgdhD9KwbnaixU3O8fD-gpArkDDQb9cn40,6513
82
+ mistralai_gcp/models/contentchunk.py,sha256=zzr7oENSG3rVjgKO_ebL-vmaX9Iy0ZiDygxyQtiWcMI,724
83
+ mistralai_gcp/models/deltamessage.py,sha256=6AcVFRWaW4mLFAyd7yOIJfKVroFe0htdclMlbv_R_iM,1968
84
+ mistralai_gcp/models/fimcompletionrequest.py,sha256=pKPGdwfd8KGUKi2th6836ZH-VR1hoKZodqB73swKjoM,6635
84
85
  mistralai_gcp/models/fimcompletionresponse.py,sha256=zUG83S6DchgEYsSG1dkOSuoOFHvlAR62gCoN9UzF06A,794
85
- mistralai_gcp/models/fimcompletionstreamrequest.py,sha256=osme14HgDjlJ4Pl_CARF7WQY_h4iSu1LyuiEKoa6qlM,5880
86
+ mistralai_gcp/models/fimcompletionstreamrequest.py,sha256=aW1JB_LF4fi-GFtlXS41wOZuc09wgCMrP8mRFpvuRjE,6014
86
87
  mistralai_gcp/models/function.py,sha256=5DkjJlymCiGz9GLfspaHAD8gxVXllSjySHeEhsgf3KI,471
87
- mistralai_gcp/models/functioncall.py,sha256=NOwWgvq86cdR6o0Y0wDalczc7aDtzxQaoqDsVSTSPSU,480
88
+ mistralai_gcp/models/functioncall.py,sha256=iIeo1sJUi1DJmASNUuqMq6iYwGLgM1fxC-mWgEiluQ4,560
88
89
  mistralai_gcp/models/functionname.py,sha256=Rp4TPQA1IvhnBZx-GwBF1fFyAd6w5Ys5A84waQ9fYKg,471
89
90
  mistralai_gcp/models/httpvalidationerror.py,sha256=wGmVyH_T7APhs_mCpOkumZ3x15FQ95cL-GH5M2iLst8,612
91
+ mistralai_gcp/models/referencechunk.py,sha256=NmajuCeC5caD70iUPL8P6DlTO44oivRnFaOhfLGBiE8,523
90
92
  mistralai_gcp/models/responseformat.py,sha256=2FG3eNkJyeSgh7YzQ-saib43ILvSVElV42h8hIffZ1g,1049
91
93
  mistralai_gcp/models/responseformats.py,sha256=oeXHoVUoZrZwrz-0cm-rHj5sHygv9MpeqgdummGa8ww,488
92
94
  mistralai_gcp/models/sdkerror.py,sha256=kd75e3JYF2TXNgRZopcV-oGdBWoBZqRcvrwqn2fsFYs,528
93
95
  mistralai_gcp/models/security.py,sha256=Z2MdVBo5vcSXMkFdCRHPJY-cNH9EqZYAK1Je5VGp4NU,623
94
- mistralai_gcp/models/systemmessage.py,sha256=0OxlU6tEeKPxjCKcFOp23xNIdZdUU9oD3xzR1MTYc-k,676
95
- mistralai_gcp/models/textchunk.py,sha256=Ni4GK43x-6L_wD5Rd1sS7yjHqh8s1m7V1S5-3aMeqB4,646
96
+ mistralai_gcp/models/systemmessage.py,sha256=cdWnQ4v7p3io9aOLFfpqx-n8c4UbOo5ghGEKpEihwSI,790
97
+ mistralai_gcp/models/textchunk.py,sha256=i3uNJmFq4W9Eg4SOUbTNRCS9bitizmooYOHhgVYkxy0,425
96
98
  mistralai_gcp/models/tool.py,sha256=u2mQpXPj38x4CfEIbx0TwTeQx5qmkjt1wUTWTZY2dak,689
97
99
  mistralai_gcp/models/toolcall.py,sha256=2qaVi_1HrjESewPBG3FdZrllVmbYwkcnBm2bnVCoVlM,774
98
100
  mistralai_gcp/models/toolchoice.py,sha256=GQcyKrGg6CwJC2Wx-hBfD8giDZiFoEuRJN3ZXmnkU1Q,1029
99
101
  mistralai_gcp/models/toolchoiceenum.py,sha256=Ca4ileCwuOjfPzIXLRIxT3RkE5zR7oqV6nXU-UjW0w0,197
100
- mistralai_gcp/models/toolmessage.py,sha256=Y-k19TSbvzdZoYsd81U2lePvTtRxgsFEQ0i79TrAPhw,1692
102
+ mistralai_gcp/models/toolmessage.py,sha256=z9BVNoFRqbK8N4kKKmzFNn8KgpxVrDW8sOR5Sc94XYI,2067
101
103
  mistralai_gcp/models/tooltypes.py,sha256=6vY1LVrp7xzXlidl1x-3SSwqdx9TBlecIeKd4sU7e6I,248
102
104
  mistralai_gcp/models/usageinfo.py,sha256=Uo2LJB58JMzlrmnfMUQnDxiMCINMS63ejp-sbOq9O-Q,405
103
- mistralai_gcp/models/usermessage.py,sha256=Wv_yQOP6K81KZZ13Jn0N4i7nM8qh3yuPO0t0pgq-JHQ,1693
104
- mistralai_gcp/models/validationerror.py,sha256=nPH4v5nEe7Vy-l4fYft3vvfb67yhGYHhoAUFDkupEZg,462
105
+ mistralai_gcp/models/usermessage.py,sha256=3OXMcPO3Tyje6wQuOfMVp35OD0EnfYZ2tkElVxOfXs8,1797
106
+ mistralai_gcp/models/validationerror.py,sha256=EVhyAndNY5aayJSNGv-W1XL7Wu9bS92JJe1yu9UmBSY,530
105
107
  mistralai_gcp/py.typed,sha256=zrp19r0G21lr2yRiMC0f8MFkQFGj9wMpSbboePMg8KM,59
106
- mistralai_gcp/sdk.py,sha256=qmgp6E5LTBMnNN3mc1aDzmnHnvRQNTR7JEJgjBkYGWg,6730
107
- mistralai_gcp/sdkconfiguration.py,sha256=3IS69cQcgicnG032IIM6FIFzZqfHY3d96ru6f5y651k,1700
108
+ mistralai_gcp/sdk.py,sha256=nioXPwRIiI0KJz8T14LwY4FR4pIHktdLr3u9qSL_y2M,6760
109
+ mistralai_gcp/sdkconfiguration.py,sha256=hvw-5lXbfz3t5grKuLAZ60jta8IRwnP2Wxu9UUKrU08,1700
108
110
  mistralai_gcp/types/__init__.py,sha256=RArOwSgeeTIva6h-4ttjXwMUeCkz10nAFBL9D-QljI4,377
109
111
  mistralai_gcp/types/basemodel.py,sha256=PexI39iKiOkIlobB8Ueo0yn8PLHp6_wb-WO-zelNDZY,1170
110
112
  mistralai_gcp/utils/__init__.py,sha256=1x4KfQf2ULtO_aOk_M2RMCimoz8jgcW0RZd-f34KDAs,2365
111
- mistralai_gcp/utils/annotations.py,sha256=KFt29x1HavmpmtFN8OGNxTje0gGb3ZTnGi1BcFXIdPc,861
113
+ mistralai_gcp/utils/annotations.py,sha256=aR7mZG34FzgRdew7WZPYEu9QGBerpuKxCF4sek5Z_5Y,1699
112
114
  mistralai_gcp/utils/enums.py,sha256=VzjeslROrAr2luZOTJlvu-4UlxgTaGOKlRYtJJ7IfyY,1006
113
- mistralai_gcp/utils/eventstreaming.py,sha256=klSQ0FURc-hqYz1si1EG92VCFhfaxyBo_xP2mX3BBWo,4917
115
+ mistralai_gcp/utils/eventstreaming.py,sha256=LtcrfJYw4nP2Oe4Wl0-cEURLzRGYReRGWNFY5wYECIE,6186
114
116
  mistralai_gcp/utils/forms.py,sha256=L-y62DiD0FXnq7WIyw8BAMbdr7sZDwhZ_3YsdviBbaA,6325
115
117
  mistralai_gcp/utils/headers.py,sha256=cPxWSmUILrefTGDzTH1Hdj7_Hlsj-EY6K5Tyc4iH4dk,3663
116
118
  mistralai_gcp/utils/logger.py,sha256=9nUtlKHo3RFsIVyMw5jq3wEKZMVwHnZMSc6xLp-otC0,520
@@ -130,86 +132,87 @@ mistralai/_hooks/deprecation_warning.py,sha256=eyEOf7-o9uqqNWJnufD2RXp3dYrGV4in9
130
132
  mistralai/_hooks/registration.py,sha256=ML0W-XbE4WYdJ4eGks_XxF2aLCJTaIWjQATFGzFwvyU,861
131
133
  mistralai/_hooks/sdkhooks.py,sha256=s-orhdvnV89TmI3QiPC2LWQtYeM9RrsG1CTll-fYZmQ,2559
132
134
  mistralai/_hooks/types.py,sha256=vUkTVk_TSaK10aEj368KYWvnd4T5EsawixMcTro_UHc,2570
133
- mistralai/_version.py,sha256=mAhPru1qHTLyCLquxNR7dXvQCNDz1KmngUDdg-1T0dQ,313
134
- mistralai/agents.py,sha256=62QnyHeUPmWtdBNCrq8MQEwpO2CtQMRa5_8VCbWU588,28795
135
+ mistralai/_version.py,sha256=evo3lX3nZiEzlnnF8oB-_9lRabkSZA8GtU9FCSTjjww,313
136
+ mistralai/agents.py,sha256=sAt7cSO2VJGmcUy-EA3Bz9rFB5EYb4tArAO8vLZoXBs,28779
135
137
  mistralai/async_client.py,sha256=KUdYxIIqoD6L7vB0EGwUR6lQ0NK5iCTHjnLVR9CVcJY,355
136
138
  mistralai/basesdk.py,sha256=MQIeNInArcIcDYgSy67EDf77ykys9JwjTlnboqdpEJ8,11273
137
139
  mistralai/batch.py,sha256=YN4D0Duwrap9Ysmp_lRpADYp1Znay7THE_z8ERGvDds,501
138
- mistralai/chat.py,sha256=YrYMOr5iJETs_rxiGaSv0x8PKLdmYVjVqWYLbNuYbhM,32970
140
+ mistralai/chat.py,sha256=sNZIDHGqaVeBesVCnAvGhh5PYN1i4n1ejtyZw-Yd0Ks,32954
139
141
  mistralai/classifiers.py,sha256=pHLWNynj1iiWvuRTKs5lcUNphX9Ko1eQ2Yi6wcnxu2Q,15101
140
142
  mistralai/client.py,sha256=hrPg-LciKMKiascF0WbRRmqQyCv1lb2yDh6j-aaKVNo,509
141
143
  mistralai/embeddings.py,sha256=S-_VG_djZ1oZOd5fQj4pYu0Y4INlgatl10dfMg5cmZw,7839
142
- mistralai/files.py,sha256=UDUxfJMkYRn-qU1OCV0yjbBCzoCk80dpPSkEpK4RkKU,33284
143
- mistralai/fim.py,sha256=Wa0aEKn4WL1mOxeqeDlmFnVL2KaOFEo-Drh4Nsm8Ns0,25732
144
+ mistralai/files.py,sha256=5Cb8zIvu5aQ7PTWBMNdXeU93FQ6Cyadj5BsyOcDEIyQ,39518
145
+ mistralai/fim.py,sha256=R1tJV1ycDmkhvTkR_pxybYz0u8g5LcKMWlB1euCE5tA,25716
144
146
  mistralai/fine_tuning.py,sha256=UENQqfE054VEsAYxdruV-TBLFIFfO-joXNznH08GUvE,477
145
- mistralai/httpclient.py,sha256=S_ItzEchFX-znIdHD6i5-a91H0Dn5QxpT0KhucdHBbI,2595
146
- mistralai/jobs.py,sha256=0XLeyOblCdEIq6tw3zkZQG1GtdxidRNNnnpfxpH0g5E,39538
147
+ mistralai/httpclient.py,sha256=WDbLpMzo7qmWki_ryOJcCAYNI1T4uyWKV08rRuCdNII,2688
148
+ mistralai/jobs.py,sha256=Ne0JBqO9c1jZINpVZN-VuvCeJyiN4fTQPSnb7b-NRVM,39496
147
149
  mistralai/mistral_jobs.py,sha256=M3N-A7dYNY_Zobo1xhZtq3EK0j1I8Ny7hnoYmlajeMM,26949
148
- mistralai/models/__init__.py,sha256=0oCW3dCrphA-TGt3VeG0HY22kf3SZI7HvtNj6X8Ecc8,21139
149
- mistralai/models/agentscompletionrequest.py,sha256=4enNzxh-QKJ53FNfCZbmS_2jPwEBC4Ys13eH0WkggH4,7270
150
- mistralai/models/agentscompletionstreamrequest.py,sha256=F_SM-vg3oi_kGmSIf9EKpRt3rhoc4cdnJa634RnyOGc,6691
151
- mistralai/models/apiendpoint.py,sha256=9gAlIzzFW9XhYtsKY-wBwjjDslwl-XYQTUSBT-GuEGs,249
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
153
+ mistralai/models/apiendpoint.py,sha256=Hvar5leWsJR_FYb0UzRlSw3vRdBZhk_6BR5r2pIb214,400
152
154
  mistralai/models/archiveftmodelout.py,sha256=dQx1J91UA06pjk2r7okhKMyBBePmHal7SPpn6Y_wEsY,820
153
- mistralai/models/assistantmessage.py,sha256=C_q6MFuEZafb9nP53iJdcF26jS-To_bhPKaz8lSq9T4,2084
155
+ mistralai/models/assistantmessage.py,sha256=BbvwC7W4Zk_8GN3w1QnEj7UygVGl0pfCtr6eosNUAgw,2204
154
156
  mistralai/models/basemodelcard.py,sha256=nv-xjoZFCuIdjKBl98dVH5puT7qh0AC2MaE7WTsMxfs,2985
155
157
  mistralai/models/batcherror.py,sha256=tThkO9B-g-6eDSBCm1Emd-zDI4B3mk2vAl0L1MI3pdQ,390
156
- mistralai/models/batchjobin.py,sha256=TQ9McRkWkja8Z_krF_OxGRORdgh8thKLLs-sgstl9F4,1680
158
+ mistralai/models/batchjobin.py,sha256=1GDaaHJeGVo71F4HVZkTdX92bmr3DieNB0ZuHFzBIeE,1850
157
159
  mistralai/models/batchjobout.py,sha256=AXFSgDFGY_1LFVHtVdp66y08DUZCxsZtZ_NzTAJYDWM,3067
158
160
  mistralai/models/batchjobsout.py,sha256=Tq6bcb4_-fcW8C9AOnfJN8_sTy1NQhDn82qFOKdFPcg,868
159
161
  mistralai/models/batchjobstatus.py,sha256=WlrIl5vWQGfLmgQA91_9CnCMKhWN6Lli458fT-4Asj4,294
160
- mistralai/models/chatclassificationrequest.py,sha256=H750wn8R8AANF8O6ifc__bsXAmv4JEHXzcplvZ3oR_U,2966
162
+ mistralai/models/chatclassificationrequest.py,sha256=WYthWiE7euFSeJYJRQJMXmxbWzGDl8kpF4yyLO9ZJaE,3225
161
163
  mistralai/models/chatcompletionchoice.py,sha256=6iIFLZj2KYx0HFfzS3-E3sNXG6mPEAlDyXxIA5iZI_U,849
162
- mistralai/models/chatcompletionrequest.py,sha256=TLvxIVwvGlDJlpNhad1loQyD48LJGCYb3AeYiRNOJmE,9322
164
+ mistralai/models/chatcompletionrequest.py,sha256=EGBBbr1TeP2NZasm5FV6XeoCayFiHdq3YYVRF4-5-Q8,9585
163
165
  mistralai/models/chatcompletionresponse.py,sha256=sLE-_Bx9W5rH2-HE2fBWPVbJbmBWx_jSY2mJ3KBEn6w,792
164
- mistralai/models/chatcompletionstreamrequest.py,sha256=uP3Y07Xzih6WzmvtkQaFMoOU-yFgkZf52ZudUcDDsUE,8911
166
+ mistralai/models/chatcompletionstreamrequest.py,sha256=8VaeQ5h5rKGcmB5xZVeasoEouz92s4I7Fqcik81rsL8,9273
165
167
  mistralai/models/checkpointout.py,sha256=A2kXS8-VT_1lbg3brifVjZD6tXdsET8vLqBm2a-yXgA,1109
166
168
  mistralai/models/classificationobject.py,sha256=JqaKo3AQD4t5X12ZnHjJ6K3Y6LXUn94uGdLJSoGr8vY,665
167
- mistralai/models/classificationrequest.py,sha256=Z5vpN9KbXm6a1dvAm25vOGY4ldHhWpvqOpQbJDZBRJg,1775
169
+ mistralai/models/classificationrequest.py,sha256=UFbjwNFtLZmylX3DqQSH3FoOvvD1UGBapK6f9b9LGYE,1903
168
170
  mistralai/models/classificationresponse.py,sha256=TeBV7mH0nvLswPdLhv4ZaDTsXvuVYJqSe5Uci7qVSMA,649
169
171
  mistralai/models/completionchunk.py,sha256=Cdq-FBWa1oBhrxapoOAj8qr6xmkeIPsfPQSbjeK6NLY,871
170
172
  mistralai/models/completionevent.py,sha256=rFc5dJBRnNOzspI95Jhkjd9WyM476u48cN0T1Vh-Cxw,399
171
173
  mistralai/models/completionresponsestreamchoice.py,sha256=gw5-_iOyznuimumDtBV65E3zwUW0KH1OHP55dGCAsAA,1927
172
- mistralai/models/contentchunk.py,sha256=gzhtNmAKX_CO22axRnTn4R6021wrK2zC5gmcXroszpA,654
174
+ mistralai/models/contentchunk.py,sha256=SR4M8E4u44_r5YNFVTuWmiDHCU3wp4kWBhVj_HRxRGs,875
173
175
  mistralai/models/delete_model_v1_models_model_id_deleteop.py,sha256=lnVRFX-G0jkn1dCFC89sXY2Pj_w4QfMDeF1tPjS4hWE,602
174
176
  mistralai/models/deletefileout.py,sha256=s3a-H2RgFQ9HX0kYSmP6GwmwE1jghz7dBj-3G0NBVSY,587
175
177
  mistralai/models/deletemodelout.py,sha256=W_crO0WtksoKUgq5s9Yh8zS8RxSuyKYQCBt1i8vB1sE,693
176
- mistralai/models/deltamessage.py,sha256=jioyJxnplv3uK-aA_0B8Pf_ZIVlYj1hgDuSu61m79Ec,1857
177
- mistralai/models/detailedjobout.py,sha256=5cQtC4p8bPCTo96abxsUyg8DibVRNPToRzoBEOgJROA,4939
178
- mistralai/models/embeddingrequest.py,sha256=7zAcQxo7ejae_iYFt3i1SvxaX8YIQFiXGYLaoy_-VOU,1972
178
+ mistralai/models/deltamessage.py,sha256=7NtvEjdmBOl86rwOx7x2fcCCJSzIF8K6-eu-G9Wr9PI,1939
179
+ mistralai/models/detailedjobout.py,sha256=aw3xmCM6E2kE1cPI5MtLZOdbtaP7FLBeHZG7ACtlEKg,4862
180
+ mistralai/models/embeddingrequest.py,sha256=9SZGc7QOMfYd_KRGSG-fXapaKisN4hZlqmtvo6sodX0,2046
179
181
  mistralai/models/embeddingresponse.py,sha256=te6E_LYEzRjHJ9QREmsFp5PeNP2J_8ALVjyb1T20pNA,663
180
182
  mistralai/models/embeddingresponsedata.py,sha256=fJ3mrZqyBBBE40a6iegOJX3DVDfgyMRq23ByeGSTLFk,534
181
183
  mistralai/models/eventout.py,sha256=TouRJeISBLphMTPHfgSOpuoOmbGDVohPOrdgHyExMpw,1633
182
184
  mistralai/models/filepurpose.py,sha256=SPaXJpAW93EozlCU63Wtpj_KkMtjwXk41n1_ohFccQw,256
183
185
  mistralai/models/files_api_routes_delete_fileop.py,sha256=HOx-hJxphSYF-JV3zOGe2eucWQUpfVqxG0IDaSa3dcE,500
184
186
  mistralai/models/files_api_routes_download_fileop.py,sha256=y3sLFZ-j4eUuxCyIP0L-exy0ViqskDLkghkOccElBQQ,504
187
+ mistralai/models/files_api_routes_get_signed_urlop.py,sha256=e_XczBgInaylmHJ9m5wJQ78hfCp2PrrTrT8bxGGi8BI,879
185
188
  mistralai/models/files_api_routes_list_filesop.py,sha256=ztzOPSCg7VfmRDxwxnquhTpLPRbI-wJAvkASOsv2g8w,3130
186
189
  mistralai/models/files_api_routes_retrieve_fileop.py,sha256=8DjSbYqUqFoPq-qcNXyVADeBVSsoCfHFlkRfmwYk-ns,504
187
190
  mistralai/models/files_api_routes_upload_fileop.py,sha256=w0KC9_GxbAXdXRlws7v9xezWepcBZVI-4gRzXXSG72w,2266
188
191
  mistralai/models/fileschema.py,sha256=cgHHp6H9HdIZqiFc-7BQwF5L6nzXz8aF6F7evEm-Hj4,2529
189
- mistralai/models/fimcompletionrequest.py,sha256=Sl_RrokeC_XC81B4-ZwFvz1961koIpryOzfrGeFtWok,6484
192
+ mistralai/models/filesignedurl.py,sha256=VwvuhzhJulAB99Qxz6zr-2F1aINosAfaSxU0IhytDSU,289
193
+ mistralai/models/fimcompletionrequest.py,sha256=DidQBt_s7o-VkfGNZpN6MUi19Qs_WXVCvA7ek45Hcnc,6606
190
194
  mistralai/models/fimcompletionresponse.py,sha256=_QwzRuL3KuKkyrA4Fxp366SW0H0EzOA7f4FLkWLm-PM,790
191
- mistralai/models/fimcompletionstreamrequest.py,sha256=buOtwz0xraey3IxIMYU2sF-LHn5SlP8WYD9qYnFCpY8,5851
192
- mistralai/models/finetuneablemodel.py,sha256=UvVJAugrRAc6elHPa9yM2ofde77cCmCYL-SWcemCJ50,339
195
+ mistralai/models/fimcompletionstreamrequest.py,sha256=n-uzWyaSk4-dfBLM8rbmhZOESLKG_JCcaoc1aLdrbqk,5985
193
196
  mistralai/models/ftmodelcapabilitiesout.py,sha256=H1kKEChUPgYT31ZQUz0tn9NRa7Z3hRZlh-sFfDYvBos,648
194
197
  mistralai/models/ftmodelcard.py,sha256=G3dioHDMOhbI5HIw5gCaxZDb1sCthBzkXIAYMNHwya8,3300
195
198
  mistralai/models/ftmodelout.py,sha256=dw-y8KKT_7rzW6tu10gfc1YKB8-Kpw4e4zy23z_U1d4,2530
196
199
  mistralai/models/function.py,sha256=lr0wmE5PW9RyNdLp6JPcVCIzUz6iBAvRie_1eg9WVj8,467
197
- mistralai/models/functioncall.py,sha256=VYeaxktyC4ygvZhBLz9Rt2ciwv2LRxFotjaZGnMbrZc,476
200
+ mistralai/models/functioncall.py,sha256=VvvBe4bVq1Irqo5t4_n1iq60UF7hLf8tE_GjkbyM8iE,556
198
201
  mistralai/models/functionname.py,sha256=jgd0moI9eORQtEAQI4ROiMSKpWSbCLmK6IhDn7uppKY,467
199
202
  mistralai/models/githubrepositoryin.py,sha256=kHU3QnOEJsdsyAt-74jrY2ztEao3aXcNdtj7aOgASRg,1956
200
203
  mistralai/models/githubrepositoryout.py,sha256=m29woSEL-vu05GU7Jva0bzzbvIykLTYUC_xSuHFVj50,1969
201
204
  mistralai/models/httpvalidationerror.py,sha256=l47dL2BTqauhRn4_GdSl3TC-QWsdL98HoloMvp6vtRQ,604
202
205
  mistralai/models/imageurl.py,sha256=6Fpt-8V3XYK-3u_Lw85gbMnyFWUdbNhOxjldqCvSpKs,1365
203
- mistralai/models/imageurlchunk.py,sha256=sMN4UJFKSRlcYg0UZnW_X5ywGhpSidvxqJHmlMyhtGE,1127
204
- mistralai/models/jobin.py,sha256=HHLWQgPv5z_EWW43O1dKzhHI8r7wl2ASEvpjkWcA8ys,4457
206
+ mistralai/models/imageurlchunk.py,sha256=yHgdAi_jOw-e5aXd4Dlr7YCtJcyw-W3QYol8-MAAT1Y,994
207
+ mistralai/models/jobin.py,sha256=FeH3zXK7B5-ux58JCXYlG37PK28G-jdz6zZsE-D6JsY,4380
205
208
  mistralai/models/jobmetadataout.py,sha256=wtKbig55Uheqwxp-VUr8Q3KH8eSSB0Vt067Ux7_caEo,2412
206
- mistralai/models/jobout.py,sha256=2QZe7WfTcFTt_osjD1VLmXRLe2tfNJu8bCk_4YIjQ1A,6180
209
+ mistralai/models/jobout.py,sha256=px44vM7dbJ5ZD5aCHOAMeSW2pZ3DPV2TMR0dVsRiVvs,6103
207
210
  mistralai/models/jobs_api_routes_batch_cancel_batch_jobop.py,sha256=3Q-YaI2zAX550v--wedoh9XoWg2rRSVFIYOrv2SjhG8,514
208
211
  mistralai/models/jobs_api_routes_batch_get_batch_jobop.py,sha256=8kCUZZZKrkDCXFtvWZVcF1XObl8QhLEajBjjfZrd12o,508
209
212
  mistralai/models/jobs_api_routes_batch_get_batch_jobsop.py,sha256=p2Sf2S04SU2tkO0HluuiaWqYffmnes728NKZxEwxiZ0,3031
210
213
  mistralai/models/jobs_api_routes_fine_tuning_archive_fine_tuned_modelop.py,sha256=pevOFk9Ji8iebXVadr5d882kKjrvT6_R6b8qBTYkQAU,628
211
214
  mistralai/models/jobs_api_routes_fine_tuning_cancel_fine_tuning_jobop.py,sha256=Q75l-kuKesb_x7uBkQNumTVBqzD9DmeTiWpOGSmtFwk,614
212
- mistralai/models/jobs_api_routes_fine_tuning_create_fine_tuning_jobop.py,sha256=GaXfilzcfQnGEN-uIolvOBIbQKyROwrY7tcYIRFKeek,498
215
+ mistralai/models/jobs_api_routes_fine_tuning_create_fine_tuning_jobop.py,sha256=FlWx7DxCdxgWrc3qTwc6XwjonzPG4BM1WDUGW2Wqij8,705
213
216
  mistralai/models/jobs_api_routes_fine_tuning_get_fine_tuning_jobop.py,sha256=742v4h88GSbgP2M-5J_0hyNSdtbZ9_awaEWXCL1hbAw,610
214
217
  mistralai/models/jobs_api_routes_fine_tuning_get_fine_tuning_jobsop.py,sha256=WQMJ7nmQ8G2W-K0dGGUOP4cSVou_OlUwzUBWOP3EmPE,5491
215
218
  mistralai/models/jobs_api_routes_fine_tuning_start_fine_tuning_jobop.py,sha256=h7d8AbRonPwHyrPDIhrrawDEeyxDqXEKtgYL-Jvg7SU,532
@@ -220,22 +223,23 @@ mistralai/models/legacyjobmetadataout.py,sha256=08zAGNTSrICsK8u2SFFUXiNWF7MCQvez
220
223
  mistralai/models/listfilesout.py,sha256=tW2fNabLKcftc5kytkjwVaChlOzWRL4FKtNzDak9MNs,468
221
224
  mistralai/models/metricout.py,sha256=dXQMMU4Nk6-Zr06Jx1TWilFi6cOwiVLjSanCFn0cPxo,2034
222
225
  mistralai/models/modelcapabilities.py,sha256=No-Dl09zT1sG4MxsWnx4s8Yo1tUeMQ7k-HR_iQFIMFc,703
223
- mistralai/models/modellist.py,sha256=uNR7bXKdsJ7IBAqTOMP2gurqIRUrGujmxf5pYwdp7wA,942
226
+ mistralai/models/modellist.py,sha256=D4Y784kQkx0ARhofFrpEqGLfxa-jTY8ev0TQMrD_n8I,995
227
+ mistralai/models/referencechunk.py,sha256=A9vV5pZv-tUqGlswdu0HOyCYy0Q-UIJY0Oc9ZfM6XJA,519
224
228
  mistralai/models/responseformat.py,sha256=C_zO6X4cbT1qSS_q9Qxq64AmtfK10i9tqEz_39ZcFzo,1045
225
229
  mistralai/models/responseformats.py,sha256=oeXHoVUoZrZwrz-0cm-rHj5sHygv9MpeqgdummGa8ww,488
226
- mistralai/models/retrieve_model_v1_models_model_id_getop.py,sha256=tvwm2ldwj-_Wo3dB_YLZF5zWHnLcwH768nvjgImoLRA,1270
230
+ mistralai/models/retrieve_model_v1_models_model_id_getop.py,sha256=N9_JFwiz9tz4zRXJ9c1V0c_anFEVxVzPDoFt2Wrer4M,1388
227
231
  mistralai/models/retrievefileout.py,sha256=LfeXEjmn3xQLRzunWGW-0V_zsZIqtS6JEwQwz1s06DU,2576
228
232
  mistralai/models/sampletype.py,sha256=zowUiTFxum8fltBs6j__BrFPio-dQdG0CIyLj-5icG8,316
229
233
  mistralai/models/sdkerror.py,sha256=kd75e3JYF2TXNgRZopcV-oGdBWoBZqRcvrwqn2fsFYs,528
230
234
  mistralai/models/security.py,sha256=RQn-xHLq3q4OEzrx9BcJMuT49UaCvwABXrqBEcqyKmA,686
231
235
  mistralai/models/source.py,sha256=_MSV-LRL2fL7wCUTXEvvsOUIWlOKqPvdZS4rm2Xhs0o,264
232
- mistralai/models/systemmessage.py,sha256=p-udUGsvM6mj960FVq-pmDXvmu43ihbQXKub-SlMrrY,672
233
- mistralai/models/textchunk.py,sha256=_bgpR2Atnwpc4LIjERfE_htPFBypJ7o6jjEtPUurX88,665
236
+ mistralai/models/systemmessage.py,sha256=ZTDim1ZLHLiCe2zd70PuxcGyUrNhB_aFSMOpOzl5eCI,786
237
+ mistralai/models/textchunk.py,sha256=2VD-TR3NOOWJ9Jzcw_E3ryq0GWz6b5XSP3k5o7oVlnc,448
234
238
  mistralai/models/tool.py,sha256=qLY0XE3uk79v3RsJqVpA81A0K9OWtmX6rwVeKal5ARk,681
235
239
  mistralai/models/toolcall.py,sha256=DYYZ4KEwwJWrN0zLJtXkCd35es8f-LVzE7kVfFe_ZkE,766
236
240
  mistralai/models/toolchoice.py,sha256=dGeb5koPp9eqHQuG1u-kP7T5Od6-cPL2rEe06-dqzcs,1021
237
241
  mistralai/models/toolchoiceenum.py,sha256=Ca4ileCwuOjfPzIXLRIxT3RkE5zR7oqV6nXU-UjW0w0,197
238
- mistralai/models/toolmessage.py,sha256=witTNk7P5w7PqO5rVTHWJEbzqRXTXFd7SCokyLlZOqE,1663
242
+ mistralai/models/toolmessage.py,sha256=zcu054y_vBaGbsCWmq58DsY8aiRrSNwzC4LJz_5kUVg,2038
239
243
  mistralai/models/tooltypes.py,sha256=NcvRsZA_ORf4WY_gn6WjgX6eEXmR2faVG3Q1sLlzdG8,244
240
244
  mistralai/models/trainingfile.py,sha256=IlwKP2GL8gL0VHVJ_zUDV-Q0F7obdLzMMRDDJatSjwo,400
241
245
  mistralai/models/trainingparameters.py,sha256=GA7OFskn6BxHs_N6YIv0s6e1j_xmqrwcuCioqIcPZK8,2298
@@ -244,20 +248,20 @@ mistralai/models/unarchiveftmodelout.py,sha256=8sn1THvCNnZpPcw3m3avWiBHe357D_nSr
244
248
  mistralai/models/updateftmodelin.py,sha256=Slabh0wwDFP8bzXWFqDzGoLh3KPnOAEyc7vttabSqX8,1466
245
249
  mistralai/models/uploadfileout.py,sha256=cLMFKbAH6sEgH1G8SVgX4-C3KpIZEc6B39LGySi3XTI,2535
246
250
  mistralai/models/usageinfo.py,sha256=66AzKK8cOFft498eUOUx6C_mCFAt5LmIFrYthJfLWpU,401
247
- mistralai/models/usermessage.py,sha256=BHV6A1VQx9GwZYq0K_jHlKPnWP-SRCaje1-BKbztPGA,1689
248
- mistralai/models/validationerror.py,sha256=obz0Yp8TU5gV3B5Auw8WOdL0ZgunetRTAJ3xVMFEXjs,458
251
+ mistralai/models/usermessage.py,sha256=kjS5HudMsaRNfeUhFFBCyY8l-umlHvGk8wvTIq6-qzY,1793
252
+ mistralai/models/validationerror.py,sha256=DouDBJmBhbW4KPsF5rZEyBdnB_adC-l32kuHC0bvO2I,526
249
253
  mistralai/models/wandbintegration.py,sha256=BkLD3r08ToZkMAhPXdnC7bfOGr3banKqt1wVKMGehUQ,2406
250
254
  mistralai/models/wandbintegrationout.py,sha256=C0HpS8jJGnACs7eWnuIq0qJEroIUAbjkvzfSSkSKS7Q,2274
251
255
  mistralai/models_.py,sha256=Kj5U6ODBgKiKs93Miaq11dlV3MDvaz--nI8ucvZVark,39797
252
256
  mistralai/py.typed,sha256=zrp19r0G21lr2yRiMC0f8MFkQFGj9wMpSbboePMg8KM,59
253
- mistralai/sdk.py,sha256=JBssVtpwLDcni_Q-IHNs62Zz8-ZxFldiOj3e42-Cb8k,5007
254
- mistralai/sdkconfiguration.py,sha256=iHOJ9SRgNMG7Z_Bl9y-fPk9TFWJru38eHCBT9W3dqJI,1688
257
+ mistralai/sdk.py,sha256=lLJOfh9im-TDBsXX9_qM-JPGXjQOOSBDGOrmlJWaZBY,5444
258
+ mistralai/sdkconfiguration.py,sha256=DgRkl_w6gaKdTlhXbXwLenGlNzWaihGuy62Kwv0ozQg,1688
255
259
  mistralai/types/__init__.py,sha256=RArOwSgeeTIva6h-4ttjXwMUeCkz10nAFBL9D-QljI4,377
256
260
  mistralai/types/basemodel.py,sha256=PexI39iKiOkIlobB8Ueo0yn8PLHp6_wb-WO-zelNDZY,1170
257
261
  mistralai/utils/__init__.py,sha256=8npwwHS-7zjVrbkzBGO-Uk4GkjC240PCleMbgPK1Axs,2418
258
- mistralai/utils/annotations.py,sha256=KFt29x1HavmpmtFN8OGNxTje0gGb3ZTnGi1BcFXIdPc,861
262
+ mistralai/utils/annotations.py,sha256=aR7mZG34FzgRdew7WZPYEu9QGBerpuKxCF4sek5Z_5Y,1699
259
263
  mistralai/utils/enums.py,sha256=VzjeslROrAr2luZOTJlvu-4UlxgTaGOKlRYtJJ7IfyY,1006
260
- mistralai/utils/eventstreaming.py,sha256=klSQ0FURc-hqYz1si1EG92VCFhfaxyBo_xP2mX3BBWo,4917
264
+ mistralai/utils/eventstreaming.py,sha256=LtcrfJYw4nP2Oe4Wl0-cEURLzRGYReRGWNFY5wYECIE,6186
261
265
  mistralai/utils/forms.py,sha256=L-y62DiD0FXnq7WIyw8BAMbdr7sZDwhZ_3YsdviBbaA,6325
262
266
  mistralai/utils/headers.py,sha256=cPxWSmUILrefTGDzTH1Hdj7_Hlsj-EY6K5Tyc4iH4dk,3663
263
267
  mistralai/utils/logger.py,sha256=TOF0Mqsua4GlsDhmrZz9hgMRvwd9zK7ytuqly3Vevxo,675
@@ -270,7 +274,7 @@ mistralai/utils/serializers.py,sha256=BSJT7kBOkNBFyP7KREyMoe14JGbgijD1M6AXFMbdmc
270
274
  mistralai/utils/url.py,sha256=BgGPgcTA6MRK4bF8fjP2dUopN3NzEzxWMXPBVg8NQUA,5254
271
275
  mistralai/utils/values.py,sha256=_89YXPTI_BU6SXJBzFR4pIzTCBPQW9tsOTN1jeBBIDs,3428
272
276
  mistralai/version.py,sha256=iosXhlXclBwBqlADFKEilxAC2wWKbtuBKi87AmPi7s8,196
273
- mistralai-1.2.2.dist-info/LICENSE,sha256=rUtQ_9GD0OyLPlb-2uWVdfE87hzudMRmsW-tS-0DK-0,11340
274
- mistralai-1.2.2.dist-info/METADATA,sha256=z0jXcLTzTc0HhjbFArDs_aqEzb5HDKd5wGY3q3nh5xI,26299
275
- mistralai-1.2.2.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
276
- mistralai-1.2.2.dist-info/RECORD,,
277
+ mistralai-1.2.4.dist-info/LICENSE,sha256=rUtQ_9GD0OyLPlb-2uWVdfE87hzudMRmsW-tS-0DK-0,11340
278
+ mistralai-1.2.4.dist-info/METADATA,sha256=jyoOgFa2nqCIxRpILeOF4SMJxBKzNC-7tIK7XcGmZUY,27984
279
+ mistralai-1.2.4.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
280
+ mistralai-1.2.4.dist-info/RECORD,,
@@ -3,7 +3,7 @@
3
3
  import importlib.metadata
4
4
 
5
5
  __title__: str = "mistralai_azure"
6
- __version__: str = "1.2.0"
6
+ __version__: str = "1.2.3"
7
7
 
8
8
  try:
9
9
  if __package__ is not None:
mistralai_azure/chat.py CHANGED
@@ -5,7 +5,7 @@ from mistralai_azure import models, utils
5
5
  from mistralai_azure._hooks import HookContext
6
6
  from mistralai_azure.types import OptionalNullable, UNSET
7
7
  from mistralai_azure.utils import eventstreaming
8
- from typing import Any, AsyncGenerator, Generator, List, Optional, Union
8
+ from typing import Any, List, Optional, Union
9
9
 
10
10
 
11
11
  class Chat(BaseSDK):
@@ -41,7 +41,7 @@ class Chat(BaseSDK):
41
41
  retries: OptionalNullable[utils.RetryConfig] = UNSET,
42
42
  server_url: Optional[str] = None,
43
43
  timeout_ms: Optional[int] = None,
44
- ) -> Optional[Generator[models.CompletionEvent, None, None]]:
44
+ ) -> Optional[eventstreaming.EventStream[models.CompletionEvent]]:
45
45
  r"""Stream chat completion
46
46
 
47
47
  Mistral AI provides the ability to stream responses back to a client in order to allow partial results for certain requests. Tokens will be sent as data-only server-sent events as they become available, with the stream terminated by a data: [DONE] message. Otherwise, the server will hold the request open until the timeout or until completion, with the response containing the full result as JSON.
@@ -135,7 +135,7 @@ class Chat(BaseSDK):
135
135
 
136
136
  data: Any = None
137
137
  if utils.match_response(http_res, "200", "text/event-stream"):
138
- return eventstreaming.stream_events(
138
+ return eventstreaming.EventStream(
139
139
  http_res,
140
140
  lambda raw: utils.unmarshal_json(raw, models.CompletionEvent),
141
141
  sentinel="[DONE]",
@@ -189,7 +189,7 @@ class Chat(BaseSDK):
189
189
  retries: OptionalNullable[utils.RetryConfig] = UNSET,
190
190
  server_url: Optional[str] = None,
191
191
  timeout_ms: Optional[int] = None,
192
- ) -> Optional[AsyncGenerator[models.CompletionEvent, None]]:
192
+ ) -> Optional[eventstreaming.EventStreamAsync[models.CompletionEvent]]:
193
193
  r"""Stream chat completion
194
194
 
195
195
  Mistral AI provides the ability to stream responses back to a client in order to allow partial results for certain requests. Tokens will be sent as data-only server-sent events as they become available, with the stream terminated by a data: [DONE] message. Otherwise, the server will hold the request open until the timeout or until completion, with the response containing the full result as JSON.
@@ -283,7 +283,7 @@ class Chat(BaseSDK):
283
283
 
284
284
  data: Any = None
285
285
  if utils.match_response(http_res, "200", "text/event-stream"):
286
- return eventstreaming.stream_events_async(
286
+ return eventstreaming.EventStreamAsync(
287
287
  http_res,
288
288
  lambda raw: utils.unmarshal_json(raw, models.CompletionEvent),
289
289
  sentinel="[DONE]",
@@ -41,6 +41,9 @@ class HttpClient(Protocol):
41
41
  ) -> httpx.Request:
42
42
  pass
43
43
 
44
+ def close(self) -> None:
45
+ pass
46
+
44
47
 
45
48
  @runtime_checkable
46
49
  class AsyncHttpClient(Protocol):
@@ -76,3 +79,6 @@ class AsyncHttpClient(Protocol):
76
79
  extensions: Optional[httpx._types.RequestExtensions] = None,
77
80
  ) -> httpx.Request:
78
81
  pass
82
+
83
+ async def aclose(self) -> None:
84
+ pass
@@ -54,6 +54,7 @@ from .functioncall import (
54
54
  )
55
55
  from .functionname import FunctionName, FunctionNameTypedDict
56
56
  from .httpvalidationerror import HTTPValidationError, HTTPValidationErrorData
57
+ from .referencechunk import ReferenceChunk, ReferenceChunkType, ReferenceChunkTypedDict
57
58
  from .responseformat import ResponseFormat, ResponseFormatTypedDict
58
59
  from .responseformats import ResponseFormats
59
60
  from .sdkerror import SDKError
@@ -70,7 +71,13 @@ from .tool import Tool, ToolTypedDict
70
71
  from .toolcall import ToolCall, ToolCallTypedDict
71
72
  from .toolchoice import ToolChoice, ToolChoiceTypedDict
72
73
  from .toolchoiceenum import ToolChoiceEnum
73
- from .toolmessage import ToolMessage, ToolMessageRole, ToolMessageTypedDict
74
+ from .toolmessage import (
75
+ ToolMessage,
76
+ ToolMessageContent,
77
+ ToolMessageContentTypedDict,
78
+ ToolMessageRole,
79
+ ToolMessageTypedDict,
80
+ )
74
81
  from .tooltypes import ToolTypes
75
82
  from .usageinfo import UsageInfo, UsageInfoTypedDict
76
83
  from .usermessage import (
@@ -137,6 +144,9 @@ __all__ = [
137
144
  "LocTypedDict",
138
145
  "Messages",
139
146
  "MessagesTypedDict",
147
+ "ReferenceChunk",
148
+ "ReferenceChunkType",
149
+ "ReferenceChunkTypedDict",
140
150
  "ResponseFormat",
141
151
  "ResponseFormatTypedDict",
142
152
  "ResponseFormats",
@@ -159,6 +169,8 @@ __all__ = [
159
169
  "ToolChoiceEnum",
160
170
  "ToolChoiceTypedDict",
161
171
  "ToolMessage",
172
+ "ToolMessageContent",
173
+ "ToolMessageContentTypedDict",
162
174
  "ToolMessageRole",
163
175
  "ToolMessageTypedDict",
164
176
  "ToolTypedDict",
@@ -12,13 +12,17 @@ from mistralai_azure.types import (
12
12
  )
13
13
  from pydantic import model_serializer
14
14
  from typing import List, Literal, Optional, Union
15
- from typing_extensions import NotRequired, TypedDict
15
+ from typing_extensions import NotRequired, TypeAliasType, TypedDict
16
16
 
17
17
 
18
- AssistantMessageContentTypedDict = Union[str, List[ContentChunkTypedDict]]
18
+ AssistantMessageContentTypedDict = TypeAliasType(
19
+ "AssistantMessageContentTypedDict", Union[str, List[ContentChunkTypedDict]]
20
+ )
19
21
 
20
22
 
21
- AssistantMessageContent = Union[str, List[ContentChunk]]
23
+ AssistantMessageContent = TypeAliasType(
24
+ "AssistantMessageContent", Union[str, List[ContentChunk]]
25
+ )
22
26
 
23
27
 
24
28
  AssistantMessageRole = Literal["assistant"]
@@ -19,23 +19,30 @@ from mistralai_azure.types import (
19
19
  from mistralai_azure.utils import get_discriminator
20
20
  from pydantic import Discriminator, Tag, model_serializer
21
21
  from typing import List, Optional, Union
22
- from typing_extensions import Annotated, NotRequired, TypedDict
22
+ from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
23
23
 
24
24
 
25
- ChatCompletionRequestStopTypedDict = Union[str, List[str]]
25
+ ChatCompletionRequestStopTypedDict = TypeAliasType(
26
+ "ChatCompletionRequestStopTypedDict", Union[str, List[str]]
27
+ )
26
28
  r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
27
29
 
28
30
 
29
- ChatCompletionRequestStop = Union[str, List[str]]
31
+ ChatCompletionRequestStop = TypeAliasType(
32
+ "ChatCompletionRequestStop", Union[str, List[str]]
33
+ )
30
34
  r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
31
35
 
32
36
 
33
- ChatCompletionRequestMessagesTypedDict = Union[
34
- SystemMessageTypedDict,
35
- UserMessageTypedDict,
36
- AssistantMessageTypedDict,
37
- ToolMessageTypedDict,
38
- ]
37
+ ChatCompletionRequestMessagesTypedDict = TypeAliasType(
38
+ "ChatCompletionRequestMessagesTypedDict",
39
+ Union[
40
+ SystemMessageTypedDict,
41
+ UserMessageTypedDict,
42
+ AssistantMessageTypedDict,
43
+ ToolMessageTypedDict,
44
+ ],
45
+ )
39
46
 
40
47
 
41
48
  ChatCompletionRequestMessages = Annotated[
@@ -49,10 +56,15 @@ ChatCompletionRequestMessages = Annotated[
49
56
  ]
50
57
 
51
58
 
52
- ChatCompletionRequestToolChoiceTypedDict = Union[ToolChoiceTypedDict, ToolChoiceEnum]
59
+ ChatCompletionRequestToolChoiceTypedDict = TypeAliasType(
60
+ "ChatCompletionRequestToolChoiceTypedDict",
61
+ Union[ToolChoiceTypedDict, ToolChoiceEnum],
62
+ )
53
63
 
54
64
 
55
- ChatCompletionRequestToolChoice = Union[ToolChoice, ToolChoiceEnum]
65
+ ChatCompletionRequestToolChoice = TypeAliasType(
66
+ "ChatCompletionRequestToolChoice", Union[ToolChoice, ToolChoiceEnum]
67
+ )
56
68
 
57
69
 
58
70
  class ChatCompletionRequestTypedDict(TypedDict):
@@ -19,23 +19,26 @@ from mistralai_azure.types import (
19
19
  from mistralai_azure.utils import get_discriminator
20
20
  from pydantic import Discriminator, Tag, model_serializer
21
21
  from typing import List, Optional, Union
22
- from typing_extensions import Annotated, NotRequired, TypedDict
22
+ from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
23
23
 
24
24
 
25
- StopTypedDict = Union[str, List[str]]
25
+ StopTypedDict = TypeAliasType("StopTypedDict", Union[str, List[str]])
26
26
  r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
27
27
 
28
28
 
29
- Stop = Union[str, List[str]]
29
+ Stop = TypeAliasType("Stop", Union[str, List[str]])
30
30
  r"""Stop generation if this token is detected. Or if one of these tokens is detected when providing an array"""
31
31
 
32
32
 
33
- MessagesTypedDict = Union[
34
- SystemMessageTypedDict,
35
- UserMessageTypedDict,
36
- AssistantMessageTypedDict,
37
- ToolMessageTypedDict,
38
- ]
33
+ MessagesTypedDict = TypeAliasType(
34
+ "MessagesTypedDict",
35
+ Union[
36
+ SystemMessageTypedDict,
37
+ UserMessageTypedDict,
38
+ AssistantMessageTypedDict,
39
+ ToolMessageTypedDict,
40
+ ],
41
+ )
39
42
 
40
43
 
41
44
  Messages = Annotated[
@@ -49,12 +52,15 @@ Messages = Annotated[
49
52
  ]
50
53
 
51
54
 
52
- ChatCompletionStreamRequestToolChoiceTypedDict = Union[
53
- ToolChoiceTypedDict, ToolChoiceEnum
54
- ]
55
+ ChatCompletionStreamRequestToolChoiceTypedDict = TypeAliasType(
56
+ "ChatCompletionStreamRequestToolChoiceTypedDict",
57
+ Union[ToolChoiceTypedDict, ToolChoiceEnum],
58
+ )
55
59
 
56
60
 
57
- ChatCompletionStreamRequestToolChoice = Union[ToolChoice, ToolChoiceEnum]
61
+ ChatCompletionStreamRequestToolChoice = TypeAliasType(
62
+ "ChatCompletionStreamRequestToolChoice", Union[ToolChoice, ToolChoiceEnum]
63
+ )
58
64
 
59
65
 
60
66
  class ChatCompletionStreamRequestTypedDict(TypedDict):
@@ -1,10 +1,22 @@
1
1
  """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
2
2
 
3
3
  from __future__ import annotations
4
+ from .referencechunk import ReferenceChunk, ReferenceChunkTypedDict
4
5
  from .textchunk import TextChunk, TextChunkTypedDict
6
+ from mistralai_azure.utils import get_discriminator
7
+ from pydantic import Discriminator, Tag
8
+ from typing import Union
9
+ from typing_extensions import Annotated, TypeAliasType
5
10
 
6
11
 
7
- ContentChunkTypedDict = TextChunkTypedDict
12
+ ContentChunkTypedDict = TypeAliasType(
13
+ "ContentChunkTypedDict", Union[TextChunkTypedDict, ReferenceChunkTypedDict]
14
+ )
8
15
 
9
16
 
10
- ContentChunk = TextChunk
17
+ ContentChunk = Annotated[
18
+ Union[
19
+ Annotated[TextChunk, Tag("text")], Annotated[ReferenceChunk, Tag("reference")]
20
+ ],
21
+ Discriminator(lambda m: get_discriminator(m, "type", "type")),
22
+ ]