mistralai 1.7.1__py3-none-any.whl → 1.8.1__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 (91) hide show
  1. mistralai/_version.py +2 -2
  2. mistralai/beta.py +22 -0
  3. mistralai/conversations.py +2660 -0
  4. mistralai/embeddings.py +12 -0
  5. mistralai/extra/__init__.py +10 -2
  6. mistralai/extra/exceptions.py +14 -0
  7. mistralai/extra/mcp/__init__.py +0 -0
  8. mistralai/extra/mcp/auth.py +166 -0
  9. mistralai/extra/mcp/base.py +155 -0
  10. mistralai/extra/mcp/sse.py +165 -0
  11. mistralai/extra/mcp/stdio.py +22 -0
  12. mistralai/extra/run/__init__.py +0 -0
  13. mistralai/extra/run/context.py +295 -0
  14. mistralai/extra/run/result.py +212 -0
  15. mistralai/extra/run/tools.py +225 -0
  16. mistralai/extra/run/utils.py +36 -0
  17. mistralai/extra/tests/test_struct_chat.py +1 -1
  18. mistralai/mistral_agents.py +1160 -0
  19. mistralai/models/__init__.py +472 -1
  20. mistralai/models/agent.py +129 -0
  21. mistralai/models/agentconversation.py +71 -0
  22. mistralai/models/agentcreationrequest.py +109 -0
  23. mistralai/models/agenthandoffdoneevent.py +33 -0
  24. mistralai/models/agenthandoffentry.py +75 -0
  25. mistralai/models/agenthandoffstartedevent.py +33 -0
  26. mistralai/models/agents_api_v1_agents_getop.py +16 -0
  27. mistralai/models/agents_api_v1_agents_listop.py +24 -0
  28. mistralai/models/agents_api_v1_agents_update_versionop.py +21 -0
  29. mistralai/models/agents_api_v1_agents_updateop.py +23 -0
  30. mistralai/models/agents_api_v1_conversations_append_streamop.py +28 -0
  31. mistralai/models/agents_api_v1_conversations_appendop.py +28 -0
  32. mistralai/models/agents_api_v1_conversations_getop.py +33 -0
  33. mistralai/models/agents_api_v1_conversations_historyop.py +16 -0
  34. mistralai/models/agents_api_v1_conversations_listop.py +37 -0
  35. mistralai/models/agents_api_v1_conversations_messagesop.py +16 -0
  36. mistralai/models/agents_api_v1_conversations_restart_streamop.py +26 -0
  37. mistralai/models/agents_api_v1_conversations_restartop.py +26 -0
  38. mistralai/models/agentupdaterequest.py +111 -0
  39. mistralai/models/builtinconnectors.py +13 -0
  40. mistralai/models/chatcompletionresponse.py +6 -6
  41. mistralai/models/codeinterpretertool.py +17 -0
  42. mistralai/models/completionargs.py +100 -0
  43. mistralai/models/completionargsstop.py +13 -0
  44. mistralai/models/completionjobout.py +3 -3
  45. mistralai/models/conversationappendrequest.py +35 -0
  46. mistralai/models/conversationappendstreamrequest.py +37 -0
  47. mistralai/models/conversationevents.py +72 -0
  48. mistralai/models/conversationhistory.py +58 -0
  49. mistralai/models/conversationinputs.py +14 -0
  50. mistralai/models/conversationmessages.py +28 -0
  51. mistralai/models/conversationrequest.py +133 -0
  52. mistralai/models/conversationresponse.py +51 -0
  53. mistralai/models/conversationrestartrequest.py +42 -0
  54. mistralai/models/conversationrestartstreamrequest.py +44 -0
  55. mistralai/models/conversationstreamrequest.py +135 -0
  56. mistralai/models/conversationusageinfo.py +63 -0
  57. mistralai/models/documentlibrarytool.py +22 -0
  58. mistralai/models/embeddingdtype.py +7 -0
  59. mistralai/models/embeddingrequest.py +43 -3
  60. mistralai/models/fimcompletionresponse.py +6 -6
  61. mistralai/models/functioncallentry.py +76 -0
  62. mistralai/models/functioncallentryarguments.py +15 -0
  63. mistralai/models/functioncallevent.py +36 -0
  64. mistralai/models/functionresultentry.py +69 -0
  65. mistralai/models/functiontool.py +21 -0
  66. mistralai/models/imagegenerationtool.py +17 -0
  67. mistralai/models/inputentries.py +18 -0
  68. mistralai/models/messageentries.py +18 -0
  69. mistralai/models/messageinputcontentchunks.py +26 -0
  70. mistralai/models/messageinputentry.py +89 -0
  71. mistralai/models/messageoutputcontentchunks.py +30 -0
  72. mistralai/models/messageoutputentry.py +100 -0
  73. mistralai/models/messageoutputevent.py +93 -0
  74. mistralai/models/modelconversation.py +127 -0
  75. mistralai/models/outputcontentchunks.py +30 -0
  76. mistralai/models/responsedoneevent.py +25 -0
  77. mistralai/models/responseerrorevent.py +27 -0
  78. mistralai/models/responsestartedevent.py +24 -0
  79. mistralai/models/ssetypes.py +18 -0
  80. mistralai/models/toolexecutiondoneevent.py +34 -0
  81. mistralai/models/toolexecutionentry.py +70 -0
  82. mistralai/models/toolexecutionstartedevent.py +31 -0
  83. mistralai/models/toolfilechunk.py +61 -0
  84. mistralai/models/toolreferencechunk.py +61 -0
  85. mistralai/models/websearchpremiumtool.py +17 -0
  86. mistralai/models/websearchtool.py +17 -0
  87. mistralai/sdk.py +3 -0
  88. {mistralai-1.7.1.dist-info → mistralai-1.8.1.dist-info}/METADATA +42 -7
  89. {mistralai-1.7.1.dist-info → mistralai-1.8.1.dist-info}/RECORD +91 -14
  90. {mistralai-1.7.1.dist-info → mistralai-1.8.1.dist-info}/LICENSE +0 -0
  91. {mistralai-1.7.1.dist-info → mistralai-1.8.1.dist-info}/WHEEL +0 -0
@@ -1,5 +1,85 @@
1
1
  """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
2
2
 
3
+ from .agent import Agent, AgentObject, AgentTools, AgentToolsTypedDict, AgentTypedDict
4
+ from .agentconversation import (
5
+ AgentConversation,
6
+ AgentConversationObject,
7
+ AgentConversationTypedDict,
8
+ )
9
+ from .agentcreationrequest import (
10
+ AgentCreationRequest,
11
+ AgentCreationRequestTools,
12
+ AgentCreationRequestToolsTypedDict,
13
+ AgentCreationRequestTypedDict,
14
+ )
15
+ from .agenthandoffdoneevent import (
16
+ AgentHandoffDoneEvent,
17
+ AgentHandoffDoneEventType,
18
+ AgentHandoffDoneEventTypedDict,
19
+ )
20
+ from .agenthandoffentry import (
21
+ AgentHandoffEntry,
22
+ AgentHandoffEntryObject,
23
+ AgentHandoffEntryType,
24
+ AgentHandoffEntryTypedDict,
25
+ )
26
+ from .agenthandoffstartedevent import (
27
+ AgentHandoffStartedEvent,
28
+ AgentHandoffStartedEventType,
29
+ AgentHandoffStartedEventTypedDict,
30
+ )
31
+ from .agents_api_v1_agents_getop import (
32
+ AgentsAPIV1AgentsGetRequest,
33
+ AgentsAPIV1AgentsGetRequestTypedDict,
34
+ )
35
+ from .agents_api_v1_agents_listop import (
36
+ AgentsAPIV1AgentsListRequest,
37
+ AgentsAPIV1AgentsListRequestTypedDict,
38
+ )
39
+ from .agents_api_v1_agents_update_versionop import (
40
+ AgentsAPIV1AgentsUpdateVersionRequest,
41
+ AgentsAPIV1AgentsUpdateVersionRequestTypedDict,
42
+ )
43
+ from .agents_api_v1_agents_updateop import (
44
+ AgentsAPIV1AgentsUpdateRequest,
45
+ AgentsAPIV1AgentsUpdateRequestTypedDict,
46
+ )
47
+ from .agents_api_v1_conversations_append_streamop import (
48
+ AgentsAPIV1ConversationsAppendStreamRequest,
49
+ AgentsAPIV1ConversationsAppendStreamRequestTypedDict,
50
+ )
51
+ from .agents_api_v1_conversations_appendop import (
52
+ AgentsAPIV1ConversationsAppendRequest,
53
+ AgentsAPIV1ConversationsAppendRequestTypedDict,
54
+ )
55
+ from .agents_api_v1_conversations_getop import (
56
+ AgentsAPIV1ConversationsGetRequest,
57
+ AgentsAPIV1ConversationsGetRequestTypedDict,
58
+ AgentsAPIV1ConversationsGetResponseV1ConversationsGet,
59
+ AgentsAPIV1ConversationsGetResponseV1ConversationsGetTypedDict,
60
+ )
61
+ from .agents_api_v1_conversations_historyop import (
62
+ AgentsAPIV1ConversationsHistoryRequest,
63
+ AgentsAPIV1ConversationsHistoryRequestTypedDict,
64
+ )
65
+ from .agents_api_v1_conversations_listop import (
66
+ AgentsAPIV1ConversationsListRequest,
67
+ AgentsAPIV1ConversationsListRequestTypedDict,
68
+ ResponseBody,
69
+ ResponseBodyTypedDict,
70
+ )
71
+ from .agents_api_v1_conversations_messagesop import (
72
+ AgentsAPIV1ConversationsMessagesRequest,
73
+ AgentsAPIV1ConversationsMessagesRequestTypedDict,
74
+ )
75
+ from .agents_api_v1_conversations_restart_streamop import (
76
+ AgentsAPIV1ConversationsRestartStreamRequest,
77
+ AgentsAPIV1ConversationsRestartStreamRequestTypedDict,
78
+ )
79
+ from .agents_api_v1_conversations_restartop import (
80
+ AgentsAPIV1ConversationsRestartRequest,
81
+ AgentsAPIV1ConversationsRestartRequestTypedDict,
82
+ )
3
83
  from .agentscompletionrequest import (
4
84
  AgentsCompletionRequest,
5
85
  AgentsCompletionRequestMessages,
@@ -20,6 +100,12 @@ from .agentscompletionstreamrequest import (
20
100
  AgentsCompletionStreamRequestToolChoiceTypedDict,
21
101
  AgentsCompletionStreamRequestTypedDict,
22
102
  )
103
+ from .agentupdaterequest import (
104
+ AgentUpdateRequest,
105
+ AgentUpdateRequestTools,
106
+ AgentUpdateRequestToolsTypedDict,
107
+ AgentUpdateRequestTypedDict,
108
+ )
23
109
  from .apiendpoint import APIEndpoint
24
110
  from .archiveftmodelout import (
25
111
  ArchiveFTModelOut,
@@ -39,6 +125,7 @@ from .batchjobin import BatchJobIn, BatchJobInTypedDict
39
125
  from .batchjobout import BatchJobOut, BatchJobOutObject, BatchJobOutTypedDict
40
126
  from .batchjobsout import BatchJobsOut, BatchJobsOutObject, BatchJobsOutTypedDict
41
127
  from .batchjobstatus import BatchJobStatus
128
+ from .builtinconnectors import BuiltInConnectors
42
129
  from .chatclassificationrequest import (
43
130
  ChatClassificationRequest,
44
131
  ChatClassificationRequestTypedDict,
@@ -131,6 +218,13 @@ from .classifiertrainingparametersin import (
131
218
  ClassifierTrainingParametersIn,
132
219
  ClassifierTrainingParametersInTypedDict,
133
220
  )
221
+ from .codeinterpretertool import (
222
+ CodeInterpreterTool,
223
+ CodeInterpreterToolType,
224
+ CodeInterpreterToolTypedDict,
225
+ )
226
+ from .completionargs import CompletionArgs, CompletionArgsTypedDict
227
+ from .completionargsstop import CompletionArgsStop, CompletionArgsStopTypedDict
134
228
  from .completionchunk import CompletionChunk, CompletionChunkTypedDict
135
229
  from .completiondetailedjobout import (
136
230
  CompletionDetailedJobOut,
@@ -152,11 +246,11 @@ from .completionftmodelout import (
152
246
  )
153
247
  from .completionjobout import (
154
248
  CompletionJobOut,
249
+ CompletionJobOutObject,
155
250
  CompletionJobOutTypedDict,
156
251
  Integrations,
157
252
  IntegrationsTypedDict,
158
253
  JobType,
159
- Object,
160
254
  Repositories,
161
255
  RepositoriesTypedDict,
162
256
  Status,
@@ -175,6 +269,67 @@ from .completiontrainingparametersin import (
175
269
  CompletionTrainingParametersInTypedDict,
176
270
  )
177
271
  from .contentchunk import ContentChunk, ContentChunkTypedDict
272
+ from .conversationappendrequest import (
273
+ ConversationAppendRequest,
274
+ ConversationAppendRequestHandoffExecution,
275
+ ConversationAppendRequestTypedDict,
276
+ )
277
+ from .conversationappendstreamrequest import (
278
+ ConversationAppendStreamRequest,
279
+ ConversationAppendStreamRequestHandoffExecution,
280
+ ConversationAppendStreamRequestTypedDict,
281
+ )
282
+ from .conversationevents import (
283
+ ConversationEvents,
284
+ ConversationEventsData,
285
+ ConversationEventsDataTypedDict,
286
+ ConversationEventsTypedDict,
287
+ )
288
+ from .conversationhistory import (
289
+ ConversationHistory,
290
+ ConversationHistoryObject,
291
+ ConversationHistoryTypedDict,
292
+ Entries,
293
+ EntriesTypedDict,
294
+ )
295
+ from .conversationinputs import ConversationInputs, ConversationInputsTypedDict
296
+ from .conversationmessages import (
297
+ ConversationMessages,
298
+ ConversationMessagesObject,
299
+ ConversationMessagesTypedDict,
300
+ )
301
+ from .conversationrequest import (
302
+ ConversationRequest,
303
+ ConversationRequestTypedDict,
304
+ HandoffExecution,
305
+ Tools,
306
+ ToolsTypedDict,
307
+ )
308
+ from .conversationresponse import (
309
+ ConversationResponse,
310
+ ConversationResponseObject,
311
+ ConversationResponseTypedDict,
312
+ Outputs,
313
+ OutputsTypedDict,
314
+ )
315
+ from .conversationrestartrequest import (
316
+ ConversationRestartRequest,
317
+ ConversationRestartRequestHandoffExecution,
318
+ ConversationRestartRequestTypedDict,
319
+ )
320
+ from .conversationrestartstreamrequest import (
321
+ ConversationRestartStreamRequest,
322
+ ConversationRestartStreamRequestHandoffExecution,
323
+ ConversationRestartStreamRequestTypedDict,
324
+ )
325
+ from .conversationstreamrequest import (
326
+ ConversationStreamRequest,
327
+ ConversationStreamRequestHandoffExecution,
328
+ ConversationStreamRequestTools,
329
+ ConversationStreamRequestToolsTypedDict,
330
+ ConversationStreamRequestTypedDict,
331
+ )
332
+ from .conversationusageinfo import ConversationUsageInfo, ConversationUsageInfoTypedDict
178
333
  from .delete_model_v1_models_model_id_deleteop import (
179
334
  DeleteModelV1ModelsModelIDDeleteRequest,
180
335
  DeleteModelV1ModelsModelIDDeleteRequestTypedDict,
@@ -182,11 +337,17 @@ from .delete_model_v1_models_model_id_deleteop import (
182
337
  from .deletefileout import DeleteFileOut, DeleteFileOutTypedDict
183
338
  from .deletemodelout import DeleteModelOut, DeleteModelOutTypedDict
184
339
  from .deltamessage import Content, ContentTypedDict, DeltaMessage, DeltaMessageTypedDict
340
+ from .documentlibrarytool import (
341
+ DocumentLibraryTool,
342
+ DocumentLibraryToolType,
343
+ DocumentLibraryToolTypedDict,
344
+ )
185
345
  from .documenturlchunk import (
186
346
  DocumentURLChunk,
187
347
  DocumentURLChunkType,
188
348
  DocumentURLChunkTypedDict,
189
349
  )
350
+ from .embeddingdtype import EmbeddingDtype
190
351
  from .embeddingrequest import (
191
352
  EmbeddingRequest,
192
353
  EmbeddingRequestInputs,
@@ -252,7 +413,29 @@ from .functioncall import (
252
413
  FunctionCall,
253
414
  FunctionCallTypedDict,
254
415
  )
416
+ from .functioncallentry import (
417
+ FunctionCallEntry,
418
+ FunctionCallEntryObject,
419
+ FunctionCallEntryType,
420
+ FunctionCallEntryTypedDict,
421
+ )
422
+ from .functioncallentryarguments import (
423
+ FunctionCallEntryArguments,
424
+ FunctionCallEntryArgumentsTypedDict,
425
+ )
426
+ from .functioncallevent import (
427
+ FunctionCallEvent,
428
+ FunctionCallEventType,
429
+ FunctionCallEventTypedDict,
430
+ )
255
431
  from .functionname import FunctionName, FunctionNameTypedDict
432
+ from .functionresultentry import (
433
+ FunctionResultEntry,
434
+ FunctionResultEntryObject,
435
+ FunctionResultEntryType,
436
+ FunctionResultEntryTypedDict,
437
+ )
438
+ from .functiontool import FunctionTool, FunctionToolType, FunctionToolTypedDict
256
439
  from .githubrepositoryin import (
257
440
  GithubRepositoryIn,
258
441
  GithubRepositoryInType,
@@ -264,6 +447,11 @@ from .githubrepositoryout import (
264
447
  GithubRepositoryOutTypedDict,
265
448
  )
266
449
  from .httpvalidationerror import HTTPValidationError, HTTPValidationErrorData
450
+ from .imagegenerationtool import (
451
+ ImageGenerationTool,
452
+ ImageGenerationToolType,
453
+ ImageGenerationToolTypedDict,
454
+ )
267
455
  from .imageurl import ImageURL, ImageURLTypedDict
268
456
  from .imageurlchunk import (
269
457
  ImageURLChunk,
@@ -272,6 +460,7 @@ from .imageurlchunk import (
272
460
  ImageURLChunkType,
273
461
  ImageURLChunkTypedDict,
274
462
  )
463
+ from .inputentries import InputEntries, InputEntriesTypedDict
275
464
  from .inputs import (
276
465
  Inputs,
277
466
  InputsTypedDict,
@@ -366,8 +555,50 @@ from .legacyjobmetadataout import (
366
555
  LegacyJobMetadataOutTypedDict,
367
556
  )
368
557
  from .listfilesout import ListFilesOut, ListFilesOutTypedDict
558
+ from .messageentries import MessageEntries, MessageEntriesTypedDict
559
+ from .messageinputcontentchunks import (
560
+ MessageInputContentChunks,
561
+ MessageInputContentChunksTypedDict,
562
+ )
563
+ from .messageinputentry import (
564
+ MessageInputEntry,
565
+ MessageInputEntryContent,
566
+ MessageInputEntryContentTypedDict,
567
+ MessageInputEntryRole,
568
+ MessageInputEntryType,
569
+ MessageInputEntryTypedDict,
570
+ Object,
571
+ )
572
+ from .messageoutputcontentchunks import (
573
+ MessageOutputContentChunks,
574
+ MessageOutputContentChunksTypedDict,
575
+ )
576
+ from .messageoutputentry import (
577
+ MessageOutputEntry,
578
+ MessageOutputEntryContent,
579
+ MessageOutputEntryContentTypedDict,
580
+ MessageOutputEntryObject,
581
+ MessageOutputEntryRole,
582
+ MessageOutputEntryType,
583
+ MessageOutputEntryTypedDict,
584
+ )
585
+ from .messageoutputevent import (
586
+ MessageOutputEvent,
587
+ MessageOutputEventContent,
588
+ MessageOutputEventContentTypedDict,
589
+ MessageOutputEventRole,
590
+ MessageOutputEventType,
591
+ MessageOutputEventTypedDict,
592
+ )
369
593
  from .metricout import MetricOut, MetricOutTypedDict
370
594
  from .modelcapabilities import ModelCapabilities, ModelCapabilitiesTypedDict
595
+ from .modelconversation import (
596
+ ModelConversation,
597
+ ModelConversationObject,
598
+ ModelConversationTools,
599
+ ModelConversationToolsTypedDict,
600
+ ModelConversationTypedDict,
601
+ )
371
602
  from .modellist import Data, DataTypedDict, ModelList, ModelListTypedDict
372
603
  from .moderationobject import ModerationObject, ModerationObjectTypedDict
373
604
  from .moderationresponse import ModerationResponse, ModerationResponseTypedDict
@@ -377,10 +608,26 @@ from .ocrpageobject import OCRPageObject, OCRPageObjectTypedDict
377
608
  from .ocrrequest import Document, DocumentTypedDict, OCRRequest, OCRRequestTypedDict
378
609
  from .ocrresponse import OCRResponse, OCRResponseTypedDict
379
610
  from .ocrusageinfo import OCRUsageInfo, OCRUsageInfoTypedDict
611
+ from .outputcontentchunks import OutputContentChunks, OutputContentChunksTypedDict
380
612
  from .prediction import Prediction, PredictionTypedDict
381
613
  from .referencechunk import ReferenceChunk, ReferenceChunkType, ReferenceChunkTypedDict
614
+ from .responsedoneevent import (
615
+ ResponseDoneEvent,
616
+ ResponseDoneEventType,
617
+ ResponseDoneEventTypedDict,
618
+ )
619
+ from .responseerrorevent import (
620
+ ResponseErrorEvent,
621
+ ResponseErrorEventType,
622
+ ResponseErrorEventTypedDict,
623
+ )
382
624
  from .responseformat import ResponseFormat, ResponseFormatTypedDict
383
625
  from .responseformats import ResponseFormats
626
+ from .responsestartedevent import (
627
+ ResponseStartedEvent,
628
+ ResponseStartedEventType,
629
+ ResponseStartedEventTypedDict,
630
+ )
384
631
  from .retrieve_model_v1_models_model_id_getop import (
385
632
  RetrieveModelV1ModelsModelIDGetRequest,
386
633
  RetrieveModelV1ModelsModelIDGetRequestTypedDict,
@@ -392,6 +639,7 @@ from .sampletype import SampleType
392
639
  from .sdkerror import SDKError
393
640
  from .security import Security, SecurityTypedDict
394
641
  from .source import Source
642
+ from .ssetypes import SSETypes
395
643
  from .systemmessage import (
396
644
  Role,
397
645
  SystemMessage,
@@ -404,6 +652,23 @@ from .tool import Tool, ToolTypedDict
404
652
  from .toolcall import ToolCall, ToolCallTypedDict
405
653
  from .toolchoice import ToolChoice, ToolChoiceTypedDict
406
654
  from .toolchoiceenum import ToolChoiceEnum
655
+ from .toolexecutiondoneevent import (
656
+ ToolExecutionDoneEvent,
657
+ ToolExecutionDoneEventType,
658
+ ToolExecutionDoneEventTypedDict,
659
+ )
660
+ from .toolexecutionentry import (
661
+ ToolExecutionEntry,
662
+ ToolExecutionEntryObject,
663
+ ToolExecutionEntryType,
664
+ ToolExecutionEntryTypedDict,
665
+ )
666
+ from .toolexecutionstartedevent import (
667
+ ToolExecutionStartedEvent,
668
+ ToolExecutionStartedEventType,
669
+ ToolExecutionStartedEventTypedDict,
670
+ )
671
+ from .toolfilechunk import ToolFileChunk, ToolFileChunkType, ToolFileChunkTypedDict
407
672
  from .toolmessage import (
408
673
  ToolMessage,
409
674
  ToolMessageContent,
@@ -411,6 +676,11 @@ from .toolmessage import (
411
676
  ToolMessageRole,
412
677
  ToolMessageTypedDict,
413
678
  )
679
+ from .toolreferencechunk import (
680
+ ToolReferenceChunk,
681
+ ToolReferenceChunkType,
682
+ ToolReferenceChunkTypedDict,
683
+ )
414
684
  from .tooltypes import ToolTypes
415
685
  from .trainingfile import TrainingFile, TrainingFileTypedDict
416
686
  from .unarchiveftmodelout import (
@@ -444,10 +714,68 @@ from .wandbintegrationout import (
444
714
  WandbIntegrationOutType,
445
715
  WandbIntegrationOutTypedDict,
446
716
  )
717
+ from .websearchpremiumtool import (
718
+ WebSearchPremiumTool,
719
+ WebSearchPremiumToolType,
720
+ WebSearchPremiumToolTypedDict,
721
+ )
722
+ from .websearchtool import WebSearchTool, WebSearchToolType, WebSearchToolTypedDict
447
723
 
448
724
 
449
725
  __all__ = [
450
726
  "APIEndpoint",
727
+ "Agent",
728
+ "AgentConversation",
729
+ "AgentConversationObject",
730
+ "AgentConversationTypedDict",
731
+ "AgentCreationRequest",
732
+ "AgentCreationRequestTools",
733
+ "AgentCreationRequestToolsTypedDict",
734
+ "AgentCreationRequestTypedDict",
735
+ "AgentHandoffDoneEvent",
736
+ "AgentHandoffDoneEventType",
737
+ "AgentHandoffDoneEventTypedDict",
738
+ "AgentHandoffEntry",
739
+ "AgentHandoffEntryObject",
740
+ "AgentHandoffEntryType",
741
+ "AgentHandoffEntryTypedDict",
742
+ "AgentHandoffStartedEvent",
743
+ "AgentHandoffStartedEventType",
744
+ "AgentHandoffStartedEventTypedDict",
745
+ "AgentObject",
746
+ "AgentTools",
747
+ "AgentToolsTypedDict",
748
+ "AgentTypedDict",
749
+ "AgentUpdateRequest",
750
+ "AgentUpdateRequestTools",
751
+ "AgentUpdateRequestToolsTypedDict",
752
+ "AgentUpdateRequestTypedDict",
753
+ "AgentsAPIV1AgentsGetRequest",
754
+ "AgentsAPIV1AgentsGetRequestTypedDict",
755
+ "AgentsAPIV1AgentsListRequest",
756
+ "AgentsAPIV1AgentsListRequestTypedDict",
757
+ "AgentsAPIV1AgentsUpdateRequest",
758
+ "AgentsAPIV1AgentsUpdateRequestTypedDict",
759
+ "AgentsAPIV1AgentsUpdateVersionRequest",
760
+ "AgentsAPIV1AgentsUpdateVersionRequestTypedDict",
761
+ "AgentsAPIV1ConversationsAppendRequest",
762
+ "AgentsAPIV1ConversationsAppendRequestTypedDict",
763
+ "AgentsAPIV1ConversationsAppendStreamRequest",
764
+ "AgentsAPIV1ConversationsAppendStreamRequestTypedDict",
765
+ "AgentsAPIV1ConversationsGetRequest",
766
+ "AgentsAPIV1ConversationsGetRequestTypedDict",
767
+ "AgentsAPIV1ConversationsGetResponseV1ConversationsGet",
768
+ "AgentsAPIV1ConversationsGetResponseV1ConversationsGetTypedDict",
769
+ "AgentsAPIV1ConversationsHistoryRequest",
770
+ "AgentsAPIV1ConversationsHistoryRequestTypedDict",
771
+ "AgentsAPIV1ConversationsListRequest",
772
+ "AgentsAPIV1ConversationsListRequestTypedDict",
773
+ "AgentsAPIV1ConversationsMessagesRequest",
774
+ "AgentsAPIV1ConversationsMessagesRequestTypedDict",
775
+ "AgentsAPIV1ConversationsRestartRequest",
776
+ "AgentsAPIV1ConversationsRestartRequestTypedDict",
777
+ "AgentsAPIV1ConversationsRestartStreamRequest",
778
+ "AgentsAPIV1ConversationsRestartStreamRequestTypedDict",
451
779
  "AgentsCompletionRequest",
452
780
  "AgentsCompletionRequestMessages",
453
781
  "AgentsCompletionRequestMessagesTypedDict",
@@ -487,6 +815,7 @@ __all__ = [
487
815
  "BatchJobsOut",
488
816
  "BatchJobsOutObject",
489
817
  "BatchJobsOutTypedDict",
818
+ "BuiltInConnectors",
490
819
  "ChatClassificationRequest",
491
820
  "ChatClassificationRequestTypedDict",
492
821
  "ChatCompletionChoice",
@@ -545,6 +874,13 @@ __all__ = [
545
874
  "ClassifierTrainingParametersIn",
546
875
  "ClassifierTrainingParametersInTypedDict",
547
876
  "ClassifierTrainingParametersTypedDict",
877
+ "CodeInterpreterTool",
878
+ "CodeInterpreterToolType",
879
+ "CodeInterpreterToolTypedDict",
880
+ "CompletionArgs",
881
+ "CompletionArgsStop",
882
+ "CompletionArgsStopTypedDict",
883
+ "CompletionArgsTypedDict",
548
884
  "CompletionChunk",
549
885
  "CompletionChunkTypedDict",
550
886
  "CompletionDetailedJobOut",
@@ -562,6 +898,7 @@ __all__ = [
562
898
  "CompletionFTModelOutObject",
563
899
  "CompletionFTModelOutTypedDict",
564
900
  "CompletionJobOut",
901
+ "CompletionJobOutObject",
565
902
  "CompletionJobOutTypedDict",
566
903
  "CompletionResponseStreamChoice",
567
904
  "CompletionResponseStreamChoiceFinishReason",
@@ -574,6 +911,42 @@ __all__ = [
574
911
  "ContentChunk",
575
912
  "ContentChunkTypedDict",
576
913
  "ContentTypedDict",
914
+ "ConversationAppendRequest",
915
+ "ConversationAppendRequestHandoffExecution",
916
+ "ConversationAppendRequestTypedDict",
917
+ "ConversationAppendStreamRequest",
918
+ "ConversationAppendStreamRequestHandoffExecution",
919
+ "ConversationAppendStreamRequestTypedDict",
920
+ "ConversationEvents",
921
+ "ConversationEventsData",
922
+ "ConversationEventsDataTypedDict",
923
+ "ConversationEventsTypedDict",
924
+ "ConversationHistory",
925
+ "ConversationHistoryObject",
926
+ "ConversationHistoryTypedDict",
927
+ "ConversationInputs",
928
+ "ConversationInputsTypedDict",
929
+ "ConversationMessages",
930
+ "ConversationMessagesObject",
931
+ "ConversationMessagesTypedDict",
932
+ "ConversationRequest",
933
+ "ConversationRequestTypedDict",
934
+ "ConversationResponse",
935
+ "ConversationResponseObject",
936
+ "ConversationResponseTypedDict",
937
+ "ConversationRestartRequest",
938
+ "ConversationRestartRequestHandoffExecution",
939
+ "ConversationRestartRequestTypedDict",
940
+ "ConversationRestartStreamRequest",
941
+ "ConversationRestartStreamRequestHandoffExecution",
942
+ "ConversationRestartStreamRequestTypedDict",
943
+ "ConversationStreamRequest",
944
+ "ConversationStreamRequestHandoffExecution",
945
+ "ConversationStreamRequestTools",
946
+ "ConversationStreamRequestToolsTypedDict",
947
+ "ConversationStreamRequestTypedDict",
948
+ "ConversationUsageInfo",
949
+ "ConversationUsageInfoTypedDict",
577
950
  "Data",
578
951
  "DataTypedDict",
579
952
  "DeleteFileOut",
@@ -585,10 +958,14 @@ __all__ = [
585
958
  "DeltaMessage",
586
959
  "DeltaMessageTypedDict",
587
960
  "Document",
961
+ "DocumentLibraryTool",
962
+ "DocumentLibraryToolType",
963
+ "DocumentLibraryToolTypedDict",
588
964
  "DocumentTypedDict",
589
965
  "DocumentURLChunk",
590
966
  "DocumentURLChunkType",
591
967
  "DocumentURLChunkTypedDict",
968
+ "EmbeddingDtype",
592
969
  "EmbeddingRequest",
593
970
  "EmbeddingRequestInputs",
594
971
  "EmbeddingRequestInputsTypedDict",
@@ -597,6 +974,8 @@ __all__ = [
597
974
  "EmbeddingResponseData",
598
975
  "EmbeddingResponseDataTypedDict",
599
976
  "EmbeddingResponseTypedDict",
977
+ "Entries",
978
+ "EntriesTypedDict",
600
979
  "EventOut",
601
980
  "EventOutTypedDict",
602
981
  "FIMCompletionRequest",
@@ -638,9 +1017,25 @@ __all__ = [
638
1017
  "FinishReason",
639
1018
  "Function",
640
1019
  "FunctionCall",
1020
+ "FunctionCallEntry",
1021
+ "FunctionCallEntryArguments",
1022
+ "FunctionCallEntryArgumentsTypedDict",
1023
+ "FunctionCallEntryObject",
1024
+ "FunctionCallEntryType",
1025
+ "FunctionCallEntryTypedDict",
1026
+ "FunctionCallEvent",
1027
+ "FunctionCallEventType",
1028
+ "FunctionCallEventTypedDict",
641
1029
  "FunctionCallTypedDict",
642
1030
  "FunctionName",
643
1031
  "FunctionNameTypedDict",
1032
+ "FunctionResultEntry",
1033
+ "FunctionResultEntryObject",
1034
+ "FunctionResultEntryType",
1035
+ "FunctionResultEntryTypedDict",
1036
+ "FunctionTool",
1037
+ "FunctionToolType",
1038
+ "FunctionToolTypedDict",
644
1039
  "FunctionTypedDict",
645
1040
  "GithubRepositoryIn",
646
1041
  "GithubRepositoryInType",
@@ -650,8 +1045,12 @@ __all__ = [
650
1045
  "GithubRepositoryOutTypedDict",
651
1046
  "HTTPValidationError",
652
1047
  "HTTPValidationErrorData",
1048
+ "HandoffExecution",
653
1049
  "Hyperparameters",
654
1050
  "HyperparametersTypedDict",
1051
+ "ImageGenerationTool",
1052
+ "ImageGenerationToolType",
1053
+ "ImageGenerationToolTypedDict",
655
1054
  "ImageURL",
656
1055
  "ImageURLChunk",
657
1056
  "ImageURLChunkImageURL",
@@ -659,6 +1058,8 @@ __all__ = [
659
1058
  "ImageURLChunkType",
660
1059
  "ImageURLChunkTypedDict",
661
1060
  "ImageURLTypedDict",
1061
+ "InputEntries",
1062
+ "InputEntriesTypedDict",
662
1063
  "Inputs",
663
1064
  "InputsTypedDict",
664
1065
  "InstructRequest",
@@ -724,12 +1125,42 @@ __all__ = [
724
1125
  "ListFilesOutTypedDict",
725
1126
  "Loc",
726
1127
  "LocTypedDict",
1128
+ "MessageEntries",
1129
+ "MessageEntriesTypedDict",
1130
+ "MessageInputContentChunks",
1131
+ "MessageInputContentChunksTypedDict",
1132
+ "MessageInputEntry",
1133
+ "MessageInputEntryContent",
1134
+ "MessageInputEntryContentTypedDict",
1135
+ "MessageInputEntryRole",
1136
+ "MessageInputEntryType",
1137
+ "MessageInputEntryTypedDict",
1138
+ "MessageOutputContentChunks",
1139
+ "MessageOutputContentChunksTypedDict",
1140
+ "MessageOutputEntry",
1141
+ "MessageOutputEntryContent",
1142
+ "MessageOutputEntryContentTypedDict",
1143
+ "MessageOutputEntryObject",
1144
+ "MessageOutputEntryRole",
1145
+ "MessageOutputEntryType",
1146
+ "MessageOutputEntryTypedDict",
1147
+ "MessageOutputEvent",
1148
+ "MessageOutputEventContent",
1149
+ "MessageOutputEventContentTypedDict",
1150
+ "MessageOutputEventRole",
1151
+ "MessageOutputEventType",
1152
+ "MessageOutputEventTypedDict",
727
1153
  "Messages",
728
1154
  "MessagesTypedDict",
729
1155
  "MetricOut",
730
1156
  "MetricOutTypedDict",
731
1157
  "ModelCapabilities",
732
1158
  "ModelCapabilitiesTypedDict",
1159
+ "ModelConversation",
1160
+ "ModelConversationObject",
1161
+ "ModelConversationTools",
1162
+ "ModelConversationToolsTypedDict",
1163
+ "ModelConversationTypedDict",
733
1164
  "ModelList",
734
1165
  "ModelListTypedDict",
735
1166
  "ModelType",
@@ -752,6 +1183,10 @@ __all__ = [
752
1183
  "Object",
753
1184
  "One",
754
1185
  "OneTypedDict",
1186
+ "OutputContentChunks",
1187
+ "OutputContentChunksTypedDict",
1188
+ "Outputs",
1189
+ "OutputsTypedDict",
755
1190
  "Prediction",
756
1191
  "PredictionTypedDict",
757
1192
  "QueryParamStatus",
@@ -762,9 +1197,20 @@ __all__ = [
762
1197
  "RepositoriesTypedDict",
763
1198
  "Response1",
764
1199
  "Response1TypedDict",
1200
+ "ResponseBody",
1201
+ "ResponseBodyTypedDict",
1202
+ "ResponseDoneEvent",
1203
+ "ResponseDoneEventType",
1204
+ "ResponseDoneEventTypedDict",
1205
+ "ResponseErrorEvent",
1206
+ "ResponseErrorEventType",
1207
+ "ResponseErrorEventTypedDict",
765
1208
  "ResponseFormat",
766
1209
  "ResponseFormatTypedDict",
767
1210
  "ResponseFormats",
1211
+ "ResponseStartedEvent",
1212
+ "ResponseStartedEventType",
1213
+ "ResponseStartedEventTypedDict",
768
1214
  "RetrieveFileOut",
769
1215
  "RetrieveFileOutTypedDict",
770
1216
  "RetrieveModelV1ModelsModelIDGetRequest",
@@ -773,6 +1219,7 @@ __all__ = [
773
1219
  "RetrieveModelV1ModelsModelIDGetResponseRetrieveModelV1ModelsModelIDGetTypedDict",
774
1220
  "Role",
775
1221
  "SDKError",
1222
+ "SSETypes",
776
1223
  "SampleType",
777
1224
  "Security",
778
1225
  "SecurityTypedDict",
@@ -793,13 +1240,31 @@ __all__ = [
793
1240
  "ToolChoice",
794
1241
  "ToolChoiceEnum",
795
1242
  "ToolChoiceTypedDict",
1243
+ "ToolExecutionDoneEvent",
1244
+ "ToolExecutionDoneEventType",
1245
+ "ToolExecutionDoneEventTypedDict",
1246
+ "ToolExecutionEntry",
1247
+ "ToolExecutionEntryObject",
1248
+ "ToolExecutionEntryType",
1249
+ "ToolExecutionEntryTypedDict",
1250
+ "ToolExecutionStartedEvent",
1251
+ "ToolExecutionStartedEventType",
1252
+ "ToolExecutionStartedEventTypedDict",
1253
+ "ToolFileChunk",
1254
+ "ToolFileChunkType",
1255
+ "ToolFileChunkTypedDict",
796
1256
  "ToolMessage",
797
1257
  "ToolMessageContent",
798
1258
  "ToolMessageContentTypedDict",
799
1259
  "ToolMessageRole",
800
1260
  "ToolMessageTypedDict",
1261
+ "ToolReferenceChunk",
1262
+ "ToolReferenceChunkType",
1263
+ "ToolReferenceChunkTypedDict",
801
1264
  "ToolTypedDict",
802
1265
  "ToolTypes",
1266
+ "Tools",
1267
+ "ToolsTypedDict",
803
1268
  "TrainingFile",
804
1269
  "TrainingFileTypedDict",
805
1270
  "Two",
@@ -827,4 +1292,10 @@ __all__ = [
827
1292
  "WandbIntegrationOutTypedDict",
828
1293
  "WandbIntegrationType",
829
1294
  "WandbIntegrationTypedDict",
1295
+ "WebSearchPremiumTool",
1296
+ "WebSearchPremiumToolType",
1297
+ "WebSearchPremiumToolTypedDict",
1298
+ "WebSearchTool",
1299
+ "WebSearchToolType",
1300
+ "WebSearchToolTypedDict",
830
1301
  ]