vellum-ai 1.1.2__py3-none-any.whl → 1.1.3__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 (84) hide show
  1. vellum/__init__.py +16 -0
  2. vellum/client/README.md +55 -0
  3. vellum/client/__init__.py +66 -507
  4. vellum/client/core/client_wrapper.py +2 -2
  5. vellum/client/raw_client.py +844 -0
  6. vellum/client/reference.md +692 -19
  7. vellum/client/resources/ad_hoc/client.py +23 -180
  8. vellum/client/resources/ad_hoc/raw_client.py +276 -0
  9. vellum/client/resources/container_images/client.py +10 -36
  10. vellum/client/resources/deployments/client.py +16 -62
  11. vellum/client/resources/document_indexes/client.py +16 -72
  12. vellum/client/resources/documents/client.py +8 -30
  13. vellum/client/resources/folder_entities/client.py +4 -8
  14. vellum/client/resources/metric_definitions/client.py +4 -14
  15. vellum/client/resources/ml_models/client.py +2 -8
  16. vellum/client/resources/organizations/client.py +2 -6
  17. vellum/client/resources/prompts/client.py +2 -10
  18. vellum/client/resources/sandboxes/client.py +4 -20
  19. vellum/client/resources/test_suite_runs/client.py +4 -18
  20. vellum/client/resources/test_suites/client.py +11 -86
  21. vellum/client/resources/test_suites/raw_client.py +136 -0
  22. vellum/client/resources/workflow_deployments/client.py +20 -78
  23. vellum/client/resources/workflow_executions/client.py +2 -6
  24. vellum/client/resources/workflow_sandboxes/client.py +2 -10
  25. vellum/client/resources/workflows/client.py +7 -6
  26. vellum/client/resources/workflows/raw_client.py +58 -47
  27. vellum/client/resources/workspace_secrets/client.py +4 -20
  28. vellum/client/resources/workspaces/client.py +2 -6
  29. vellum/client/types/__init__.py +16 -0
  30. vellum/client/types/array_chat_message_content_item.py +4 -2
  31. vellum/client/types/array_chat_message_content_item_request.py +4 -2
  32. vellum/client/types/chat_message_content.py +4 -2
  33. vellum/client/types/chat_message_content_request.py +4 -2
  34. vellum/client/types/node_execution_span.py +2 -0
  35. vellum/client/types/prompt_block.py +4 -2
  36. vellum/client/types/vellum_value.py +4 -2
  37. vellum/client/types/vellum_value_request.py +4 -2
  38. vellum/client/types/vellum_variable_type.py +2 -1
  39. vellum/client/types/vellum_video.py +24 -0
  40. vellum/client/types/vellum_video_request.py +24 -0
  41. vellum/client/types/video_chat_message_content.py +25 -0
  42. vellum/client/types/video_chat_message_content_request.py +25 -0
  43. vellum/client/types/video_prompt_block.py +29 -0
  44. vellum/client/types/video_vellum_value.py +25 -0
  45. vellum/client/types/video_vellum_value_request.py +25 -0
  46. vellum/client/types/workflow_execution_span.py +2 -0
  47. vellum/client/types/workflow_execution_usage_calculation_fulfilled_body.py +22 -0
  48. vellum/prompts/blocks/compilation.py +22 -10
  49. vellum/types/vellum_video.py +3 -0
  50. vellum/types/vellum_video_request.py +3 -0
  51. vellum/types/video_chat_message_content.py +3 -0
  52. vellum/types/video_chat_message_content_request.py +3 -0
  53. vellum/types/video_prompt_block.py +3 -0
  54. vellum/types/video_vellum_value.py +3 -0
  55. vellum/types/video_vellum_value_request.py +3 -0
  56. vellum/types/workflow_execution_usage_calculation_fulfilled_body.py +3 -0
  57. vellum/workflows/events/workflow.py +11 -0
  58. vellum/workflows/graph/graph.py +103 -1
  59. vellum/workflows/graph/tests/test_graph.py +99 -0
  60. vellum/workflows/nodes/bases/base.py +9 -1
  61. vellum/workflows/nodes/displayable/bases/utils.py +4 -2
  62. vellum/workflows/nodes/displayable/tool_calling_node/node.py +19 -18
  63. vellum/workflows/nodes/displayable/tool_calling_node/tests/test_node.py +17 -7
  64. vellum/workflows/nodes/displayable/tool_calling_node/tests/test_utils.py +7 -7
  65. vellum/workflows/nodes/displayable/tool_calling_node/utils.py +47 -80
  66. vellum/workflows/references/environment_variable.py +10 -0
  67. vellum/workflows/runner/runner.py +18 -2
  68. vellum/workflows/state/context.py +101 -12
  69. vellum/workflows/types/definition.py +11 -1
  70. vellum/workflows/types/tests/test_definition.py +19 -0
  71. vellum/workflows/utils/vellum_variables.py +9 -5
  72. vellum/workflows/workflows/base.py +12 -5
  73. {vellum_ai-1.1.2.dist-info → vellum_ai-1.1.3.dist-info}/METADATA +1 -1
  74. {vellum_ai-1.1.2.dist-info → vellum_ai-1.1.3.dist-info}/RECORD +84 -68
  75. vellum_ee/workflows/display/nodes/vellum/code_execution_node.py +1 -1
  76. vellum_ee/workflows/display/nodes/vellum/tests/test_code_execution_node.py +55 -1
  77. vellum_ee/workflows/display/nodes/vellum/tests/test_tool_calling_node.py +15 -52
  78. vellum_ee/workflows/display/tests/workflow_serialization/test_basic_tool_calling_node_mcp_serialization.py +15 -49
  79. vellum_ee/workflows/display/types.py +14 -1
  80. vellum_ee/workflows/display/utils/expressions.py +13 -4
  81. vellum_ee/workflows/display/workflows/base_workflow_display.py +6 -19
  82. {vellum_ai-1.1.2.dist-info → vellum_ai-1.1.3.dist-info}/LICENSE +0 -0
  83. {vellum_ai-1.1.2.dist-info → vellum_ai-1.1.3.dist-info}/WHEEL +0 -0
  84. {vellum_ai-1.1.2.dist-info → vellum_ai-1.1.3.dist-info}/entry_points.txt +0 -0
@@ -41,7 +41,7 @@ class DeploymentsClient:
41
41
  """
42
42
  return self._raw_client
43
43
 
44
- def list_(
44
+ def list(
45
45
  self,
46
46
  *,
47
47
  limit: typing.Optional[int] = None,
@@ -86,11 +86,7 @@ class DeploymentsClient:
86
86
  client.deployments.list()
87
87
  """
88
88
  response = self._raw_client.list(
89
- limit=limit,
90
- offset=offset,
91
- ordering=ordering,
92
- status=status,
93
- request_options=request_options,
89
+ limit=limit, offset=offset, ordering=ordering, status=status, request_options=request_options
94
90
  )
95
91
  return response.data
96
92
 
@@ -123,10 +119,7 @@ class DeploymentsClient:
123
119
  id="id",
124
120
  )
125
121
  """
126
- response = self._raw_client.retrieve(
127
- id,
128
- request_options=request_options,
129
- )
122
+ response = self._raw_client.retrieve(id, request_options=request_options)
130
123
  return response.data
131
124
 
132
125
  def deployment_history_item_retrieve(
@@ -166,9 +159,7 @@ class DeploymentsClient:
166
159
  )
167
160
  """
168
161
  response = self._raw_client.deployment_history_item_retrieve(
169
- history_id_or_release_tag,
170
- id,
171
- request_options=request_options,
162
+ history_id_or_release_tag, id, request_options=request_options
172
163
  )
173
164
  return response.data
174
165
 
@@ -222,12 +213,7 @@ class DeploymentsClient:
222
213
  )
223
214
  """
224
215
  response = self._raw_client.list_deployment_release_tags(
225
- id,
226
- limit=limit,
227
- offset=offset,
228
- ordering=ordering,
229
- source=source,
230
- request_options=request_options,
216
+ id, limit=limit, offset=offset, ordering=ordering, source=source, request_options=request_options
231
217
  )
232
218
  return response.data
233
219
 
@@ -266,11 +252,7 @@ class DeploymentsClient:
266
252
  name="name",
267
253
  )
268
254
  """
269
- response = self._raw_client.retrieve_deployment_release_tag(
270
- id,
271
- name,
272
- request_options=request_options,
273
- )
255
+ response = self._raw_client.retrieve_deployment_release_tag(id, name, request_options=request_options)
274
256
  return response.data
275
257
 
276
258
  def update_deployment_release_tag(
@@ -317,10 +299,7 @@ class DeploymentsClient:
317
299
  )
318
300
  """
319
301
  response = self._raw_client.update_deployment_release_tag(
320
- id,
321
- name,
322
- history_item_id=history_item_id,
323
- request_options=request_options,
302
+ id, name, history_item_id=history_item_id, request_options=request_options
324
303
  )
325
304
  return response.data
326
305
 
@@ -360,9 +339,7 @@ class DeploymentsClient:
360
339
  )
361
340
  """
362
341
  response = self._raw_client.retrieve_prompt_deployment_release(
363
- id,
364
- release_id_or_release_tag,
365
- request_options=request_options,
342
+ id, release_id_or_release_tag, request_options=request_options
366
343
  )
367
344
  return response.data
368
345
 
@@ -459,7 +436,7 @@ class AsyncDeploymentsClient:
459
436
  """
460
437
  return self._raw_client
461
438
 
462
- async def list_(
439
+ async def list(
463
440
  self,
464
441
  *,
465
442
  limit: typing.Optional[int] = None,
@@ -512,11 +489,7 @@ class AsyncDeploymentsClient:
512
489
  asyncio.run(main())
513
490
  """
514
491
  response = await self._raw_client.list(
515
- limit=limit,
516
- offset=offset,
517
- ordering=ordering,
518
- status=status,
519
- request_options=request_options,
492
+ limit=limit, offset=offset, ordering=ordering, status=status, request_options=request_options
520
493
  )
521
494
  return response.data
522
495
 
@@ -557,10 +530,7 @@ class AsyncDeploymentsClient:
557
530
 
558
531
  asyncio.run(main())
559
532
  """
560
- response = await self._raw_client.retrieve(
561
- id,
562
- request_options=request_options,
563
- )
533
+ response = await self._raw_client.retrieve(id, request_options=request_options)
564
534
  return response.data
565
535
 
566
536
  async def deployment_history_item_retrieve(
@@ -608,9 +578,7 @@ class AsyncDeploymentsClient:
608
578
  asyncio.run(main())
609
579
  """
610
580
  response = await self._raw_client.deployment_history_item_retrieve(
611
- history_id_or_release_tag,
612
- id,
613
- request_options=request_options,
581
+ history_id_or_release_tag, id, request_options=request_options
614
582
  )
615
583
  return response.data
616
584
 
@@ -672,12 +640,7 @@ class AsyncDeploymentsClient:
672
640
  asyncio.run(main())
673
641
  """
674
642
  response = await self._raw_client.list_deployment_release_tags(
675
- id,
676
- limit=limit,
677
- offset=offset,
678
- ordering=ordering,
679
- source=source,
680
- request_options=request_options,
643
+ id, limit=limit, offset=offset, ordering=ordering, source=source, request_options=request_options
681
644
  )
682
645
  return response.data
683
646
 
@@ -724,11 +687,7 @@ class AsyncDeploymentsClient:
724
687
 
725
688
  asyncio.run(main())
726
689
  """
727
- response = await self._raw_client.retrieve_deployment_release_tag(
728
- id,
729
- name,
730
- request_options=request_options,
731
- )
690
+ response = await self._raw_client.retrieve_deployment_release_tag(id, name, request_options=request_options)
732
691
  return response.data
733
692
 
734
693
  async def update_deployment_release_tag(
@@ -783,10 +742,7 @@ class AsyncDeploymentsClient:
783
742
  asyncio.run(main())
784
743
  """
785
744
  response = await self._raw_client.update_deployment_release_tag(
786
- id,
787
- name,
788
- history_item_id=history_item_id,
789
- request_options=request_options,
745
+ id, name, history_item_id=history_item_id, request_options=request_options
790
746
  )
791
747
  return response.data
792
748
 
@@ -834,9 +790,7 @@ class AsyncDeploymentsClient:
834
790
  asyncio.run(main())
835
791
  """
836
792
  response = await self._raw_client.retrieve_prompt_deployment_release(
837
- id,
838
- release_id_or_release_tag,
839
- request_options=request_options,
793
+ id, release_id_or_release_tag, request_options=request_options
840
794
  )
841
795
  return response.data
842
796
 
@@ -35,7 +35,7 @@ class DocumentIndexesClient:
35
35
  """
36
36
  return self._raw_client
37
37
 
38
- def list_(
38
+ def list(
39
39
  self,
40
40
  *,
41
41
  limit: typing.Optional[int] = None,
@@ -87,12 +87,7 @@ class DocumentIndexesClient:
87
87
  client.document_indexes.list()
88
88
  """
89
89
  response = self._raw_client.list(
90
- limit=limit,
91
- offset=offset,
92
- ordering=ordering,
93
- search=search,
94
- status=status,
95
- request_options=request_options,
90
+ limit=limit, offset=offset, ordering=ordering, search=search, status=status, request_options=request_options
96
91
  )
97
92
  return response.data
98
93
 
@@ -198,10 +193,7 @@ class DocumentIndexesClient:
198
193
  id="id",
199
194
  )
200
195
  """
201
- response = self._raw_client.retrieve(
202
- id,
203
- request_options=request_options,
204
- )
196
+ response = self._raw_client.retrieve(id, request_options=request_options)
205
197
  return response.data
206
198
 
207
199
  def update(
@@ -250,12 +242,7 @@ class DocumentIndexesClient:
250
242
  label="x",
251
243
  )
252
244
  """
253
- response = self._raw_client.update(
254
- id,
255
- label=label,
256
- status=status,
257
- request_options=request_options,
258
- )
245
+ response = self._raw_client.update(id, label=label, status=status, request_options=request_options)
259
246
  return response.data
260
247
 
261
248
  def destroy(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> None:
@@ -286,10 +273,7 @@ class DocumentIndexesClient:
286
273
  id="id",
287
274
  )
288
275
  """
289
- response = self._raw_client.destroy(
290
- id,
291
- request_options=request_options,
292
- )
276
+ response = self._raw_client.destroy(id, request_options=request_options)
293
277
  return response.data
294
278
 
295
279
  def partial_update(
@@ -337,12 +321,7 @@ class DocumentIndexesClient:
337
321
  id="id",
338
322
  )
339
323
  """
340
- response = self._raw_client.partial_update(
341
- id,
342
- label=label,
343
- status=status,
344
- request_options=request_options,
345
- )
324
+ response = self._raw_client.partial_update(id, label=label, status=status, request_options=request_options)
346
325
  return response.data
347
326
 
348
327
  def add_document(
@@ -379,11 +358,7 @@ class DocumentIndexesClient:
379
358
  id="id",
380
359
  )
381
360
  """
382
- response = self._raw_client.add_document(
383
- document_id,
384
- id,
385
- request_options=request_options,
386
- )
361
+ response = self._raw_client.add_document(document_id, id, request_options=request_options)
387
362
  return response.data
388
363
 
389
364
  def remove_document(
@@ -420,11 +395,7 @@ class DocumentIndexesClient:
420
395
  id="id",
421
396
  )
422
397
  """
423
- response = self._raw_client.remove_document(
424
- document_id,
425
- id,
426
- request_options=request_options,
427
- )
398
+ response = self._raw_client.remove_document(document_id, id, request_options=request_options)
428
399
  return response.data
429
400
 
430
401
 
@@ -443,7 +414,7 @@ class AsyncDocumentIndexesClient:
443
414
  """
444
415
  return self._raw_client
445
416
 
446
- async def list_(
417
+ async def list(
447
418
  self,
448
419
  *,
449
420
  limit: typing.Optional[int] = None,
@@ -503,12 +474,7 @@ class AsyncDocumentIndexesClient:
503
474
  asyncio.run(main())
504
475
  """
505
476
  response = await self._raw_client.list(
506
- limit=limit,
507
- offset=offset,
508
- ordering=ordering,
509
- search=search,
510
- status=status,
511
- request_options=request_options,
477
+ limit=limit, offset=offset, ordering=ordering, search=search, status=status, request_options=request_options
512
478
  )
513
479
  return response.data
514
480
 
@@ -630,10 +596,7 @@ class AsyncDocumentIndexesClient:
630
596
 
631
597
  asyncio.run(main())
632
598
  """
633
- response = await self._raw_client.retrieve(
634
- id,
635
- request_options=request_options,
636
- )
599
+ response = await self._raw_client.retrieve(id, request_options=request_options)
637
600
  return response.data
638
601
 
639
602
  async def update(
@@ -690,12 +653,7 @@ class AsyncDocumentIndexesClient:
690
653
 
691
654
  asyncio.run(main())
692
655
  """
693
- response = await self._raw_client.update(
694
- id,
695
- label=label,
696
- status=status,
697
- request_options=request_options,
698
- )
656
+ response = await self._raw_client.update(id, label=label, status=status, request_options=request_options)
699
657
  return response.data
700
658
 
701
659
  async def destroy(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> None:
@@ -734,10 +692,7 @@ class AsyncDocumentIndexesClient:
734
692
 
735
693
  asyncio.run(main())
736
694
  """
737
- response = await self._raw_client.destroy(
738
- id,
739
- request_options=request_options,
740
- )
695
+ response = await self._raw_client.destroy(id, request_options=request_options)
741
696
  return response.data
742
697
 
743
698
  async def partial_update(
@@ -794,10 +749,7 @@ class AsyncDocumentIndexesClient:
794
749
  asyncio.run(main())
795
750
  """
796
751
  response = await self._raw_client.partial_update(
797
- id,
798
- label=label,
799
- status=status,
800
- request_options=request_options,
752
+ id, label=label, status=status, request_options=request_options
801
753
  )
802
754
  return response.data
803
755
 
@@ -843,11 +795,7 @@ class AsyncDocumentIndexesClient:
843
795
 
844
796
  asyncio.run(main())
845
797
  """
846
- response = await self._raw_client.add_document(
847
- document_id,
848
- id,
849
- request_options=request_options,
850
- )
798
+ response = await self._raw_client.add_document(document_id, id, request_options=request_options)
851
799
  return response.data
852
800
 
853
801
  async def remove_document(
@@ -892,9 +840,5 @@ class AsyncDocumentIndexesClient:
892
840
 
893
841
  asyncio.run(main())
894
842
  """
895
- response = await self._raw_client.remove_document(
896
- document_id,
897
- id,
898
- request_options=request_options,
899
- )
843
+ response = await self._raw_client.remove_document(document_id, id, request_options=request_options)
900
844
  return response.data
@@ -35,7 +35,7 @@ class DocumentsClient:
35
35
  """
36
36
  return self._raw_client
37
37
 
38
- def list_(
38
+ def list(
39
39
  self,
40
40
  *,
41
41
  document_index_id: typing.Optional[str] = None,
@@ -121,10 +121,7 @@ class DocumentsClient:
121
121
  id="id",
122
122
  )
123
123
  """
124
- response = self._raw_client.retrieve(
125
- id,
126
- request_options=request_options,
127
- )
124
+ response = self._raw_client.retrieve(id, request_options=request_options)
128
125
  return response.data
129
126
 
130
127
  def destroy(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> None:
@@ -154,10 +151,7 @@ class DocumentsClient:
154
151
  id="id",
155
152
  )
156
153
  """
157
- response = self._raw_client.destroy(
158
- id,
159
- request_options=request_options,
160
- )
154
+ response = self._raw_client.destroy(id, request_options=request_options)
161
155
  return response.data
162
156
 
163
157
  def partial_update(
@@ -212,12 +206,7 @@ class DocumentsClient:
212
206
  )
213
207
  """
214
208
  response = self._raw_client.partial_update(
215
- id,
216
- label=label,
217
- status=status,
218
- keywords=keywords,
219
- metadata=metadata,
220
- request_options=request_options,
209
+ id, label=label, status=status, keywords=keywords, metadata=metadata, request_options=request_options
221
210
  )
222
211
  return response.data
223
212
 
@@ -292,7 +281,7 @@ class AsyncDocumentsClient:
292
281
  """
293
282
  return self._raw_client
294
283
 
295
- async def list_(
284
+ async def list(
296
285
  self,
297
286
  *,
298
287
  document_index_id: typing.Optional[str] = None,
@@ -394,10 +383,7 @@ class AsyncDocumentsClient:
394
383
 
395
384
  asyncio.run(main())
396
385
  """
397
- response = await self._raw_client.retrieve(
398
- id,
399
- request_options=request_options,
400
- )
386
+ response = await self._raw_client.retrieve(id, request_options=request_options)
401
387
  return response.data
402
388
 
403
389
  async def destroy(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> None:
@@ -435,10 +421,7 @@ class AsyncDocumentsClient:
435
421
 
436
422
  asyncio.run(main())
437
423
  """
438
- response = await self._raw_client.destroy(
439
- id,
440
- request_options=request_options,
441
- )
424
+ response = await self._raw_client.destroy(id, request_options=request_options)
442
425
  return response.data
443
426
 
444
427
  async def partial_update(
@@ -501,12 +484,7 @@ class AsyncDocumentsClient:
501
484
  asyncio.run(main())
502
485
  """
503
486
  response = await self._raw_client.partial_update(
504
- id,
505
- label=label,
506
- status=status,
507
- keywords=keywords,
508
- metadata=metadata,
509
- request_options=request_options,
487
+ id, label=label, status=status, keywords=keywords, metadata=metadata, request_options=request_options
510
488
  )
511
489
  return response.data
512
490
 
@@ -32,7 +32,7 @@ class FolderEntitiesClient:
32
32
  """
33
33
  return self._raw_client
34
34
 
35
- def list_(
35
+ def list(
36
36
  self,
37
37
  *,
38
38
  parent_folder_id: str,
@@ -144,9 +144,7 @@ class FolderEntitiesClient:
144
144
  )
145
145
  """
146
146
  response = self._raw_client.add_entity_to_folder(
147
- folder_id,
148
- entity_id=entity_id,
149
- request_options=request_options,
147
+ folder_id, entity_id=entity_id, request_options=request_options
150
148
  )
151
149
  return response.data
152
150
 
@@ -166,7 +164,7 @@ class AsyncFolderEntitiesClient:
166
164
  """
167
165
  return self._raw_client
168
166
 
169
- async def list_(
167
+ async def list(
170
168
  self,
171
169
  *,
172
170
  parent_folder_id: str,
@@ -294,8 +292,6 @@ class AsyncFolderEntitiesClient:
294
292
  asyncio.run(main())
295
293
  """
296
294
  response = await self._raw_client.add_entity_to_folder(
297
- folder_id,
298
- entity_id=entity_id,
299
- request_options=request_options,
295
+ folder_id, entity_id=entity_id, request_options=request_options
300
296
  )
301
297
  return response.data
@@ -82,10 +82,7 @@ class MetricDefinitionsClient:
82
82
  )
83
83
  """
84
84
  response = self._raw_client.execute_metric_definition(
85
- id,
86
- inputs=inputs,
87
- release_tag=release_tag,
88
- request_options=request_options,
85
+ id, inputs=inputs, release_tag=release_tag, request_options=request_options
89
86
  )
90
87
  return response.data
91
88
 
@@ -123,9 +120,7 @@ class MetricDefinitionsClient:
123
120
  )
124
121
  """
125
122
  response = self._raw_client.metric_definition_history_item_retrieve(
126
- history_id_or_release_tag,
127
- id,
128
- request_options=request_options,
123
+ history_id_or_release_tag, id, request_options=request_options
129
124
  )
130
125
  return response.data
131
126
 
@@ -202,10 +197,7 @@ class AsyncMetricDefinitionsClient:
202
197
  asyncio.run(main())
203
198
  """
204
199
  response = await self._raw_client.execute_metric_definition(
205
- id,
206
- inputs=inputs,
207
- release_tag=release_tag,
208
- request_options=request_options,
200
+ id, inputs=inputs, release_tag=release_tag, request_options=request_options
209
201
  )
210
202
  return response.data
211
203
 
@@ -251,8 +243,6 @@ class AsyncMetricDefinitionsClient:
251
243
  asyncio.run(main())
252
244
  """
253
245
  response = await self._raw_client.metric_definition_history_item_retrieve(
254
- history_id_or_release_tag,
255
- id,
256
- request_options=request_options,
246
+ history_id_or_release_tag, id, request_options=request_options
257
247
  )
258
248
  return response.data
@@ -57,10 +57,7 @@ class MlModelsClient:
57
57
  id="id",
58
58
  )
59
59
  """
60
- response = self._raw_client.retrieve(
61
- id,
62
- request_options=request_options,
63
- )
60
+ response = self._raw_client.retrieve(id, request_options=request_options)
64
61
  return response.data
65
62
 
66
63
 
@@ -116,8 +113,5 @@ class AsyncMlModelsClient:
116
113
 
117
114
  asyncio.run(main())
118
115
  """
119
- response = await self._raw_client.retrieve(
120
- id,
121
- request_options=request_options,
122
- )
116
+ response = await self._raw_client.retrieve(id, request_options=request_options)
123
117
  return response.data
@@ -52,9 +52,7 @@ class OrganizationsClient:
52
52
  )
53
53
  client.organizations.organization_identity()
54
54
  """
55
- response = self._raw_client.organization_identity(
56
- request_options=request_options,
57
- )
55
+ response = self._raw_client.organization_identity(request_options=request_options)
58
56
  return response.data
59
57
 
60
58
 
@@ -107,7 +105,5 @@ class AsyncOrganizationsClient:
107
105
 
108
106
  asyncio.run(main())
109
107
  """
110
- response = await self._raw_client.organization_identity(
111
- request_options=request_options,
112
- )
108
+ response = await self._raw_client.organization_identity(request_options=request_options)
113
109
  return response.data
@@ -70,11 +70,7 @@ class PromptsClient:
70
70
  id="id",
71
71
  )
72
72
  """
73
- response = self._raw_client.pull(
74
- id,
75
- prompt_variant_id=prompt_variant_id,
76
- request_options=request_options,
77
- )
73
+ response = self._raw_client.pull(id, prompt_variant_id=prompt_variant_id, request_options=request_options)
78
74
  return response.data
79
75
 
80
76
  def push(
@@ -221,11 +217,7 @@ class AsyncPromptsClient:
221
217
 
222
218
  asyncio.run(main())
223
219
  """
224
- response = await self._raw_client.pull(
225
- id,
226
- prompt_variant_id=prompt_variant_id,
227
- request_options=request_options,
228
- )
220
+ response = await self._raw_client.pull(id, prompt_variant_id=prompt_variant_id, request_options=request_options)
229
221
  return response.data
230
222
 
231
223
  async def push(
@@ -162,11 +162,7 @@ class SandboxesClient:
162
162
  )
163
163
  """
164
164
  response = self._raw_client.upsert_sandbox_scenario(
165
- id,
166
- inputs=inputs,
167
- label=label,
168
- scenario_id=scenario_id,
169
- request_options=request_options,
165
+ id, inputs=inputs, label=label, scenario_id=scenario_id, request_options=request_options
170
166
  )
171
167
  return response.data
172
168
 
@@ -204,11 +200,7 @@ class SandboxesClient:
204
200
  scenario_id="scenario_id",
205
201
  )
206
202
  """
207
- response = self._raw_client.delete_sandbox_scenario(
208
- id,
209
- scenario_id,
210
- request_options=request_options,
211
- )
203
+ response = self._raw_client.delete_sandbox_scenario(id, scenario_id, request_options=request_options)
212
204
  return response.data
213
205
 
214
206
 
@@ -372,11 +364,7 @@ class AsyncSandboxesClient:
372
364
  asyncio.run(main())
373
365
  """
374
366
  response = await self._raw_client.upsert_sandbox_scenario(
375
- id,
376
- inputs=inputs,
377
- label=label,
378
- scenario_id=scenario_id,
379
- request_options=request_options,
367
+ id, inputs=inputs, label=label, scenario_id=scenario_id, request_options=request_options
380
368
  )
381
369
  return response.data
382
370
 
@@ -422,9 +410,5 @@ class AsyncSandboxesClient:
422
410
 
423
411
  asyncio.run(main())
424
412
  """
425
- response = await self._raw_client.delete_sandbox_scenario(
426
- id,
427
- scenario_id,
428
- request_options=request_options,
429
- )
413
+ response = await self._raw_client.delete_sandbox_scenario(id, scenario_id, request_options=request_options)
430
414
  return response.data