asteroid-odyssey 1.3.3__py3-none-any.whl → 1.3.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 (53) hide show
  1. asteroid_odyssey/agents_v1_gen/models/agent_profile.py +3 -3
  2. asteroid_odyssey/agents_v2_gen/__init__.py +58 -108
  3. asteroid_odyssey/agents_v2_gen/api/__init__.py +2 -17
  4. asteroid_odyssey/agents_v2_gen/api/{execution_api.py → default_api.py} +376 -66
  5. asteroid_odyssey/agents_v2_gen/api_client.py +2 -5
  6. asteroid_odyssey/agents_v2_gen/configuration.py +4 -14
  7. asteroid_odyssey/agents_v2_gen/exceptions.py +1 -1
  8. asteroid_odyssey/agents_v2_gen/models/__init__.py +29 -69
  9. asteroid_odyssey/agents_v2_gen/models/activity_payload_union_action_completed.py +4 -4
  10. asteroid_odyssey/agents_v2_gen/models/activity_payload_union_action_failed.py +4 -4
  11. asteroid_odyssey/agents_v2_gen/models/activity_payload_union_action_started.py +4 -4
  12. asteroid_odyssey/agents_v2_gen/models/activity_payload_union_file_added.py +100 -0
  13. asteroid_odyssey/agents_v2_gen/models/activity_payload_union_generic.py +4 -4
  14. asteroid_odyssey/agents_v2_gen/models/activity_payload_union_status_changed.py +4 -4
  15. asteroid_odyssey/agents_v2_gen/models/activity_payload_union_step_completed.py +4 -4
  16. asteroid_odyssey/agents_v2_gen/models/activity_payload_union_step_started.py +4 -4
  17. asteroid_odyssey/agents_v2_gen/models/activity_payload_union_terminal.py +4 -4
  18. asteroid_odyssey/agents_v2_gen/models/activity_payload_union_transitioned_node.py +4 -4
  19. asteroid_odyssey/agents_v2_gen/models/activity_payload_union_user_message_received.py +4 -4
  20. asteroid_odyssey/agents_v2_gen/models/{execution_activity.py → agents_execution_activity.py} +11 -12
  21. asteroid_odyssey/agents_v2_gen/models/agents_execution_activity_action_completed_payload.py +87 -0
  22. asteroid_odyssey/agents_v2_gen/models/{execution_activity_action_completed_payload.py → agents_execution_activity_action_failed_payload.py} +5 -5
  23. asteroid_odyssey/agents_v2_gen/models/{execution_activity_action_failed_payload.py → agents_execution_activity_action_started_payload.py} +5 -5
  24. asteroid_odyssey/agents_v2_gen/models/agents_execution_activity_file_added_payload.py +104 -0
  25. asteroid_odyssey/agents_v2_gen/models/{execution_activity_action_started_payload.py → agents_execution_activity_generic_payload.py} +5 -5
  26. asteroid_odyssey/agents_v2_gen/models/{execution_activity_payload_union.py → agents_execution_activity_payload_union.py} +26 -12
  27. asteroid_odyssey/agents_v2_gen/models/{execution_activity_status_changed_payload.py → agents_execution_activity_status_changed_payload.py} +26 -20
  28. asteroid_odyssey/agents_v2_gen/models/{execution_activity_step_started_payload.py → agents_execution_activity_step_completed_payload.py} +5 -5
  29. asteroid_odyssey/agents_v2_gen/models/{execution_activity_step_completed_payload.py → agents_execution_activity_step_started_payload.py} +5 -5
  30. asteroid_odyssey/agents_v2_gen/models/{execution_activity_transitioned_node_payload.py → agents_execution_activity_transitioned_node_payload.py} +6 -7
  31. asteroid_odyssey/agents_v2_gen/models/{execution_activity_user_message_received_payload.py → agents_execution_activity_user_message_received_payload.py} +6 -7
  32. asteroid_odyssey/agents_v2_gen/models/{execution_awaiting_confirmation_payload.py → agents_execution_awaiting_confirmation_payload.py} +5 -5
  33. asteroid_odyssey/agents_v2_gen/models/agents_execution_cancel_reason.py +38 -0
  34. asteroid_odyssey/agents_v2_gen/models/agents_execution_cancelled_payload.py +88 -0
  35. asteroid_odyssey/agents_v2_gen/models/{execution_completed_payload.py → agents_execution_completed_payload.py} +5 -5
  36. asteroid_odyssey/agents_v2_gen/models/{execution_failed_payload.py → agents_execution_failed_payload.py} +5 -5
  37. asteroid_odyssey/agents_v2_gen/models/{execution_paused_payload.py → agents_execution_paused_payload.py} +5 -5
  38. asteroid_odyssey/agents_v2_gen/models/{execution_status.py → agents_execution_status.py} +4 -4
  39. asteroid_odyssey/agents_v2_gen/models/{execution_terminal_payload.py → agents_execution_terminal_payload.py} +5 -5
  40. asteroid_odyssey/agents_v2_gen/models/{execution_activity_generic_payload.py → agents_execution_user_messages_add_text_body.py} +5 -5
  41. asteroid_odyssey/agents_v2_gen/models/{file.py → agents_files_file.py} +8 -9
  42. asteroid_odyssey/agents_v2_gen/models/common_bad_request_error_body.py +96 -0
  43. asteroid_odyssey/agents_v2_gen/models/common_forbidden_error_body.py +96 -0
  44. asteroid_odyssey/agents_v2_gen/models/{error.py → common_not_found_error_body.py} +15 -8
  45. asteroid_odyssey/agents_v2_gen/models/{versions.py → version.py} +5 -6
  46. asteroid_odyssey/agents_v2_gen/rest.py +1 -1
  47. asteroid_odyssey/client.py +28 -29
  48. {asteroid_odyssey-1.3.3.dist-info → asteroid_odyssey-1.3.4.dist-info}/METADATA +1 -1
  49. asteroid_odyssey-1.3.4.dist-info/RECORD +84 -0
  50. asteroid_odyssey/agents_v2_gen/models/execution_user_messages_add_text_body.py +0 -87
  51. asteroid_odyssey-1.3.3.dist-info/RECORD +0 -78
  52. {asteroid_odyssey-1.3.3.dist-info → asteroid_odyssey-1.3.4.dist-info}/WHEEL +0 -0
  53. {asteroid_odyssey-1.3.3.dist-info → asteroid_odyssey-1.3.4.dist-info}/top_level.txt +0 -0
@@ -5,7 +5,7 @@
5
5
 
6
6
  No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
7
 
8
- The version of the OpenAPI document: v2
8
+ The version of the OpenAPI document: v1
9
9
  Generated by OpenAPI Generator (https://openapi-generator.tech)
10
10
 
11
11
  Do not edit the class manually.
@@ -16,20 +16,19 @@ from pydantic import validate_call, Field, StrictFloat, StrictStr, StrictInt
16
16
  from typing import Any, Dict, List, Optional, Tuple, Union
17
17
  from typing_extensions import Annotated
18
18
 
19
- from pydantic import Field, StrictInt, StrictStr, field_validator
20
- from typing import List, Optional
19
+ from pydantic import Field, StrictBytes, StrictInt, StrictStr, field_validator
20
+ from typing import List, Optional, Tuple, Union
21
21
  from typing_extensions import Annotated
22
- from uuid import UUID
23
- from asteroid_odyssey.agents_v2_gen.models.execution_activity import ExecutionActivity
24
- from asteroid_odyssey.agents_v2_gen.models.execution_user_messages_add_text_body import ExecutionUserMessagesAddTextBody
25
- from asteroid_odyssey.agents_v2_gen.models.file import File
22
+ from asteroid_odyssey.agents_v2_gen.models.agents_execution_activity import AgentsExecutionActivity
23
+ from asteroid_odyssey.agents_v2_gen.models.agents_execution_user_messages_add_text_body import AgentsExecutionUserMessagesAddTextBody
24
+ from asteroid_odyssey.agents_v2_gen.models.agents_files_file import AgentsFilesFile
26
25
 
27
26
  from asteroid_odyssey.agents_v2_gen.api_client import ApiClient, RequestSerialized
28
27
  from asteroid_odyssey.agents_v2_gen.api_response import ApiResponse
29
28
  from asteroid_odyssey.agents_v2_gen.rest import RESTResponseType
30
29
 
31
30
 
32
- class ExecutionApi:
31
+ class DefaultApi:
33
32
  """NOTE: This class is auto generated by OpenAPI Generator
34
33
  Ref: https://openapi-generator.tech
35
34
 
@@ -43,9 +42,9 @@ class ExecutionApi:
43
42
 
44
43
 
45
44
  @validate_call
46
- def activities_get(
45
+ def execution_activities_get(
47
46
  self,
48
- execution_id: Annotated[UUID, Field(description="The unique identifier of the execution")],
47
+ execution_id: Annotated[StrictStr, Field(description="The unique identifier of the execution")],
49
48
  order: Annotated[Optional[StrictStr], Field(description="Sort order for activities by timestamp")] = None,
50
49
  limit: Annotated[Optional[StrictInt], Field(description="Maximum number of activities to return")] = None,
51
50
  _request_timeout: Union[
@@ -60,7 +59,7 @@ class ExecutionApi:
60
59
  _content_type: Optional[StrictStr] = None,
61
60
  _headers: Optional[Dict[StrictStr, Any]] = None,
62
61
  _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
63
- ) -> List[ExecutionActivity]:
62
+ ) -> List[AgentsExecutionActivity]:
64
63
  """Retrieve execution activities
65
64
 
66
65
  Get activities for an execution
@@ -93,7 +92,7 @@ class ExecutionApi:
93
92
  :return: Returns the result object.
94
93
  """ # noqa: E501
95
94
 
96
- _param = self._activities_get_serialize(
95
+ _param = self._execution_activities_get_serialize(
97
96
  execution_id=execution_id,
98
97
  order=order,
99
98
  limit=limit,
@@ -104,7 +103,10 @@ class ExecutionApi:
104
103
  )
105
104
 
106
105
  _response_types_map: Dict[str, Optional[str]] = {
107
- '200': "List[ExecutionActivity]",
106
+ '200': "List[AgentsExecutionActivity]",
107
+ '400': "CommonBadRequestErrorBody",
108
+ '403': "CommonForbiddenErrorBody",
109
+ '404': "CommonNotFoundErrorBody",
108
110
  }
109
111
  response_data = self.api_client.call_api(
110
112
  *_param,
@@ -118,9 +120,9 @@ class ExecutionApi:
118
120
 
119
121
 
120
122
  @validate_call
121
- def activities_get_with_http_info(
123
+ def execution_activities_get_with_http_info(
122
124
  self,
123
- execution_id: Annotated[UUID, Field(description="The unique identifier of the execution")],
125
+ execution_id: Annotated[StrictStr, Field(description="The unique identifier of the execution")],
124
126
  order: Annotated[Optional[StrictStr], Field(description="Sort order for activities by timestamp")] = None,
125
127
  limit: Annotated[Optional[StrictInt], Field(description="Maximum number of activities to return")] = None,
126
128
  _request_timeout: Union[
@@ -135,7 +137,7 @@ class ExecutionApi:
135
137
  _content_type: Optional[StrictStr] = None,
136
138
  _headers: Optional[Dict[StrictStr, Any]] = None,
137
139
  _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
138
- ) -> ApiResponse[List[ExecutionActivity]]:
140
+ ) -> ApiResponse[List[AgentsExecutionActivity]]:
139
141
  """Retrieve execution activities
140
142
 
141
143
  Get activities for an execution
@@ -168,7 +170,7 @@ class ExecutionApi:
168
170
  :return: Returns the result object.
169
171
  """ # noqa: E501
170
172
 
171
- _param = self._activities_get_serialize(
173
+ _param = self._execution_activities_get_serialize(
172
174
  execution_id=execution_id,
173
175
  order=order,
174
176
  limit=limit,
@@ -179,7 +181,10 @@ class ExecutionApi:
179
181
  )
180
182
 
181
183
  _response_types_map: Dict[str, Optional[str]] = {
182
- '200': "List[ExecutionActivity]",
184
+ '200': "List[AgentsExecutionActivity]",
185
+ '400': "CommonBadRequestErrorBody",
186
+ '403': "CommonForbiddenErrorBody",
187
+ '404': "CommonNotFoundErrorBody",
183
188
  }
184
189
  response_data = self.api_client.call_api(
185
190
  *_param,
@@ -193,9 +198,9 @@ class ExecutionApi:
193
198
 
194
199
 
195
200
  @validate_call
196
- def activities_get_without_preload_content(
201
+ def execution_activities_get_without_preload_content(
197
202
  self,
198
- execution_id: Annotated[UUID, Field(description="The unique identifier of the execution")],
203
+ execution_id: Annotated[StrictStr, Field(description="The unique identifier of the execution")],
199
204
  order: Annotated[Optional[StrictStr], Field(description="Sort order for activities by timestamp")] = None,
200
205
  limit: Annotated[Optional[StrictInt], Field(description="Maximum number of activities to return")] = None,
201
206
  _request_timeout: Union[
@@ -243,7 +248,7 @@ class ExecutionApi:
243
248
  :return: Returns the result object.
244
249
  """ # noqa: E501
245
250
 
246
- _param = self._activities_get_serialize(
251
+ _param = self._execution_activities_get_serialize(
247
252
  execution_id=execution_id,
248
253
  order=order,
249
254
  limit=limit,
@@ -254,7 +259,10 @@ class ExecutionApi:
254
259
  )
255
260
 
256
261
  _response_types_map: Dict[str, Optional[str]] = {
257
- '200': "List[ExecutionActivity]",
262
+ '200': "List[AgentsExecutionActivity]",
263
+ '400': "CommonBadRequestErrorBody",
264
+ '403': "CommonForbiddenErrorBody",
265
+ '404': "CommonNotFoundErrorBody",
258
266
  }
259
267
  response_data = self.api_client.call_api(
260
268
  *_param,
@@ -263,7 +271,7 @@ class ExecutionApi:
263
271
  return response_data.response
264
272
 
265
273
 
266
- def _activities_get_serialize(
274
+ def _execution_activities_get_serialize(
267
275
  self,
268
276
  execution_id,
269
277
  order,
@@ -338,9 +346,9 @@ class ExecutionApi:
338
346
 
339
347
 
340
348
  @validate_call
341
- def context_files_get(
349
+ def execution_context_files_get(
342
350
  self,
343
- execution_id: UUID,
351
+ execution_id: StrictStr,
344
352
  _request_timeout: Union[
345
353
  None,
346
354
  Annotated[StrictFloat, Field(gt=0)],
@@ -353,8 +361,8 @@ class ExecutionApi:
353
361
  _content_type: Optional[StrictStr] = None,
354
362
  _headers: Optional[Dict[StrictStr, Any]] = None,
355
363
  _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
356
- ) -> List[File]:
357
- """context_files_get
364
+ ) -> List[AgentsFilesFile]:
365
+ """execution_context_files_get
358
366
 
359
367
 
360
368
  :param execution_id: (required)
@@ -381,7 +389,7 @@ class ExecutionApi:
381
389
  :return: Returns the result object.
382
390
  """ # noqa: E501
383
391
 
384
- _param = self._context_files_get_serialize(
392
+ _param = self._execution_context_files_get_serialize(
385
393
  execution_id=execution_id,
386
394
  _request_auth=_request_auth,
387
395
  _content_type=_content_type,
@@ -390,7 +398,7 @@ class ExecutionApi:
390
398
  )
391
399
 
392
400
  _response_types_map: Dict[str, Optional[str]] = {
393
- '200': "List[File]",
401
+ '200': "List[AgentsFilesFile]",
394
402
  '404': "str",
395
403
  }
396
404
  response_data = self.api_client.call_api(
@@ -405,9 +413,9 @@ class ExecutionApi:
405
413
 
406
414
 
407
415
  @validate_call
408
- def context_files_get_with_http_info(
416
+ def execution_context_files_get_with_http_info(
409
417
  self,
410
- execution_id: UUID,
418
+ execution_id: StrictStr,
411
419
  _request_timeout: Union[
412
420
  None,
413
421
  Annotated[StrictFloat, Field(gt=0)],
@@ -420,8 +428,8 @@ class ExecutionApi:
420
428
  _content_type: Optional[StrictStr] = None,
421
429
  _headers: Optional[Dict[StrictStr, Any]] = None,
422
430
  _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
423
- ) -> ApiResponse[List[File]]:
424
- """context_files_get
431
+ ) -> ApiResponse[List[AgentsFilesFile]]:
432
+ """execution_context_files_get
425
433
 
426
434
 
427
435
  :param execution_id: (required)
@@ -448,7 +456,7 @@ class ExecutionApi:
448
456
  :return: Returns the result object.
449
457
  """ # noqa: E501
450
458
 
451
- _param = self._context_files_get_serialize(
459
+ _param = self._execution_context_files_get_serialize(
452
460
  execution_id=execution_id,
453
461
  _request_auth=_request_auth,
454
462
  _content_type=_content_type,
@@ -457,7 +465,7 @@ class ExecutionApi:
457
465
  )
458
466
 
459
467
  _response_types_map: Dict[str, Optional[str]] = {
460
- '200': "List[File]",
468
+ '200': "List[AgentsFilesFile]",
461
469
  '404': "str",
462
470
  }
463
471
  response_data = self.api_client.call_api(
@@ -472,9 +480,9 @@ class ExecutionApi:
472
480
 
473
481
 
474
482
  @validate_call
475
- def context_files_get_without_preload_content(
483
+ def execution_context_files_get_without_preload_content(
476
484
  self,
477
- execution_id: UUID,
485
+ execution_id: StrictStr,
478
486
  _request_timeout: Union[
479
487
  None,
480
488
  Annotated[StrictFloat, Field(gt=0)],
@@ -488,7 +496,7 @@ class ExecutionApi:
488
496
  _headers: Optional[Dict[StrictStr, Any]] = None,
489
497
  _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
490
498
  ) -> RESTResponseType:
491
- """context_files_get
499
+ """execution_context_files_get
492
500
 
493
501
 
494
502
  :param execution_id: (required)
@@ -515,7 +523,7 @@ class ExecutionApi:
515
523
  :return: Returns the result object.
516
524
  """ # noqa: E501
517
525
 
518
- _param = self._context_files_get_serialize(
526
+ _param = self._execution_context_files_get_serialize(
519
527
  execution_id=execution_id,
520
528
  _request_auth=_request_auth,
521
529
  _content_type=_content_type,
@@ -524,7 +532,7 @@ class ExecutionApi:
524
532
  )
525
533
 
526
534
  _response_types_map: Dict[str, Optional[str]] = {
527
- '200': "List[File]",
535
+ '200': "List[AgentsFilesFile]",
528
536
  '404': "str",
529
537
  }
530
538
  response_data = self.api_client.call_api(
@@ -534,7 +542,7 @@ class ExecutionApi:
534
542
  return response_data.response
535
543
 
536
544
 
537
- def _context_files_get_serialize(
545
+ def _execution_context_files_get_serialize(
538
546
  self,
539
547
  execution_id,
540
548
  _request_auth,
@@ -600,10 +608,303 @@ class ExecutionApi:
600
608
 
601
609
 
602
610
  @validate_call
603
- def user_messages_add(
611
+ def execution_context_files_upload(
604
612
  self,
605
- execution_id: Annotated[UUID, Field(description="The unique identifier of the execution")],
606
- execution_user_messages_add_text_body: Annotated[ExecutionUserMessagesAddTextBody, Field(description="The message content to send")],
613
+ execution_id: StrictStr,
614
+ files: List[Union[StrictBytes, StrictStr, Tuple[StrictStr, StrictBytes]]],
615
+ _request_timeout: Union[
616
+ None,
617
+ Annotated[StrictFloat, Field(gt=0)],
618
+ Tuple[
619
+ Annotated[StrictFloat, Field(gt=0)],
620
+ Annotated[StrictFloat, Field(gt=0)]
621
+ ]
622
+ ] = None,
623
+ _request_auth: Optional[Dict[StrictStr, Any]] = None,
624
+ _content_type: Optional[StrictStr] = None,
625
+ _headers: Optional[Dict[StrictStr, Any]] = None,
626
+ _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
627
+ ) -> str:
628
+ """execution_context_files_upload
629
+
630
+
631
+ :param execution_id: (required)
632
+ :type execution_id: str
633
+ :param files: (required)
634
+ :type files: List[bytearray]
635
+ :param _request_timeout: timeout setting for this request. If one
636
+ number provided, it will be total request
637
+ timeout. It can also be a pair (tuple) of
638
+ (connection, read) timeouts.
639
+ :type _request_timeout: int, tuple(int, int), optional
640
+ :param _request_auth: set to override the auth_settings for an a single
641
+ request; this effectively ignores the
642
+ authentication in the spec for a single request.
643
+ :type _request_auth: dict, optional
644
+ :param _content_type: force content-type for the request.
645
+ :type _content_type: str, Optional
646
+ :param _headers: set to override the headers for a single
647
+ request; this effectively ignores the headers
648
+ in the spec for a single request.
649
+ :type _headers: dict, optional
650
+ :param _host_index: set to override the host_index for a single
651
+ request; this effectively ignores the host_index
652
+ in the spec for a single request.
653
+ :type _host_index: int, optional
654
+ :return: Returns the result object.
655
+ """ # noqa: E501
656
+
657
+ _param = self._execution_context_files_upload_serialize(
658
+ execution_id=execution_id,
659
+ files=files,
660
+ _request_auth=_request_auth,
661
+ _content_type=_content_type,
662
+ _headers=_headers,
663
+ _host_index=_host_index
664
+ )
665
+
666
+ _response_types_map: Dict[str, Optional[str]] = {
667
+ '200': "str",
668
+ '400': "str",
669
+ '404': "str",
670
+ }
671
+ response_data = self.api_client.call_api(
672
+ *_param,
673
+ _request_timeout=_request_timeout
674
+ )
675
+ response_data.read()
676
+ return self.api_client.response_deserialize(
677
+ response_data=response_data,
678
+ response_types_map=_response_types_map,
679
+ ).data
680
+
681
+
682
+ @validate_call
683
+ def execution_context_files_upload_with_http_info(
684
+ self,
685
+ execution_id: StrictStr,
686
+ files: List[Union[StrictBytes, StrictStr, Tuple[StrictStr, StrictBytes]]],
687
+ _request_timeout: Union[
688
+ None,
689
+ Annotated[StrictFloat, Field(gt=0)],
690
+ Tuple[
691
+ Annotated[StrictFloat, Field(gt=0)],
692
+ Annotated[StrictFloat, Field(gt=0)]
693
+ ]
694
+ ] = None,
695
+ _request_auth: Optional[Dict[StrictStr, Any]] = None,
696
+ _content_type: Optional[StrictStr] = None,
697
+ _headers: Optional[Dict[StrictStr, Any]] = None,
698
+ _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
699
+ ) -> ApiResponse[str]:
700
+ """execution_context_files_upload
701
+
702
+
703
+ :param execution_id: (required)
704
+ :type execution_id: str
705
+ :param files: (required)
706
+ :type files: List[bytearray]
707
+ :param _request_timeout: timeout setting for this request. If one
708
+ number provided, it will be total request
709
+ timeout. It can also be a pair (tuple) of
710
+ (connection, read) timeouts.
711
+ :type _request_timeout: int, tuple(int, int), optional
712
+ :param _request_auth: set to override the auth_settings for an a single
713
+ request; this effectively ignores the
714
+ authentication in the spec for a single request.
715
+ :type _request_auth: dict, optional
716
+ :param _content_type: force content-type for the request.
717
+ :type _content_type: str, Optional
718
+ :param _headers: set to override the headers for a single
719
+ request; this effectively ignores the headers
720
+ in the spec for a single request.
721
+ :type _headers: dict, optional
722
+ :param _host_index: set to override the host_index for a single
723
+ request; this effectively ignores the host_index
724
+ in the spec for a single request.
725
+ :type _host_index: int, optional
726
+ :return: Returns the result object.
727
+ """ # noqa: E501
728
+
729
+ _param = self._execution_context_files_upload_serialize(
730
+ execution_id=execution_id,
731
+ files=files,
732
+ _request_auth=_request_auth,
733
+ _content_type=_content_type,
734
+ _headers=_headers,
735
+ _host_index=_host_index
736
+ )
737
+
738
+ _response_types_map: Dict[str, Optional[str]] = {
739
+ '200': "str",
740
+ '400': "str",
741
+ '404': "str",
742
+ }
743
+ response_data = self.api_client.call_api(
744
+ *_param,
745
+ _request_timeout=_request_timeout
746
+ )
747
+ response_data.read()
748
+ return self.api_client.response_deserialize(
749
+ response_data=response_data,
750
+ response_types_map=_response_types_map,
751
+ )
752
+
753
+
754
+ @validate_call
755
+ def execution_context_files_upload_without_preload_content(
756
+ self,
757
+ execution_id: StrictStr,
758
+ files: List[Union[StrictBytes, StrictStr, Tuple[StrictStr, StrictBytes]]],
759
+ _request_timeout: Union[
760
+ None,
761
+ Annotated[StrictFloat, Field(gt=0)],
762
+ Tuple[
763
+ Annotated[StrictFloat, Field(gt=0)],
764
+ Annotated[StrictFloat, Field(gt=0)]
765
+ ]
766
+ ] = None,
767
+ _request_auth: Optional[Dict[StrictStr, Any]] = None,
768
+ _content_type: Optional[StrictStr] = None,
769
+ _headers: Optional[Dict[StrictStr, Any]] = None,
770
+ _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
771
+ ) -> RESTResponseType:
772
+ """execution_context_files_upload
773
+
774
+
775
+ :param execution_id: (required)
776
+ :type execution_id: str
777
+ :param files: (required)
778
+ :type files: List[bytearray]
779
+ :param _request_timeout: timeout setting for this request. If one
780
+ number provided, it will be total request
781
+ timeout. It can also be a pair (tuple) of
782
+ (connection, read) timeouts.
783
+ :type _request_timeout: int, tuple(int, int), optional
784
+ :param _request_auth: set to override the auth_settings for an a single
785
+ request; this effectively ignores the
786
+ authentication in the spec for a single request.
787
+ :type _request_auth: dict, optional
788
+ :param _content_type: force content-type for the request.
789
+ :type _content_type: str, Optional
790
+ :param _headers: set to override the headers for a single
791
+ request; this effectively ignores the headers
792
+ in the spec for a single request.
793
+ :type _headers: dict, optional
794
+ :param _host_index: set to override the host_index for a single
795
+ request; this effectively ignores the host_index
796
+ in the spec for a single request.
797
+ :type _host_index: int, optional
798
+ :return: Returns the result object.
799
+ """ # noqa: E501
800
+
801
+ _param = self._execution_context_files_upload_serialize(
802
+ execution_id=execution_id,
803
+ files=files,
804
+ _request_auth=_request_auth,
805
+ _content_type=_content_type,
806
+ _headers=_headers,
807
+ _host_index=_host_index
808
+ )
809
+
810
+ _response_types_map: Dict[str, Optional[str]] = {
811
+ '200': "str",
812
+ '400': "str",
813
+ '404': "str",
814
+ }
815
+ response_data = self.api_client.call_api(
816
+ *_param,
817
+ _request_timeout=_request_timeout
818
+ )
819
+ return response_data.response
820
+
821
+
822
+ def _execution_context_files_upload_serialize(
823
+ self,
824
+ execution_id,
825
+ files,
826
+ _request_auth,
827
+ _content_type,
828
+ _headers,
829
+ _host_index,
830
+ ) -> RequestSerialized:
831
+
832
+ _host = None
833
+
834
+ _collection_formats: Dict[str, str] = {
835
+ 'files': 'multi',
836
+ }
837
+
838
+ _path_params: Dict[str, str] = {}
839
+ _query_params: List[Tuple[str, str]] = []
840
+ _header_params: Dict[str, Optional[str]] = _headers or {}
841
+ _form_params: List[Tuple[str, str]] = []
842
+ _files: Dict[
843
+ str, Union[str, bytes, List[str], List[bytes], List[Tuple[str, bytes]]]
844
+ ] = {}
845
+ _body_params: Optional[bytes] = None
846
+
847
+ # process the path parameters
848
+ if execution_id is not None:
849
+ _path_params['executionId'] = execution_id
850
+ # process the query parameters
851
+ # process the header parameters
852
+ # process the form parameters
853
+ if files is not None:
854
+ _files['files'] = files
855
+ # process the body parameter
856
+
857
+
858
+ # set the HTTP header `Accept`
859
+ if 'Accept' not in _header_params:
860
+ _header_params['Accept'] = self.api_client.select_header_accept(
861
+ [
862
+ 'text/plain'
863
+ ]
864
+ )
865
+
866
+ # set the HTTP header `Content-Type`
867
+ if _content_type:
868
+ _header_params['Content-Type'] = _content_type
869
+ else:
870
+ _default_content_type = (
871
+ self.api_client.select_header_content_type(
872
+ [
873
+ 'multipart/form-data'
874
+ ]
875
+ )
876
+ )
877
+ if _default_content_type is not None:
878
+ _header_params['Content-Type'] = _default_content_type
879
+
880
+ # authentication setting
881
+ _auth_settings: List[str] = [
882
+ 'ApiKeyAuth'
883
+ ]
884
+
885
+ return self.api_client.param_serialize(
886
+ method='POST',
887
+ resource_path='/executions/{executionId}/context-files',
888
+ path_params=_path_params,
889
+ query_params=_query_params,
890
+ header_params=_header_params,
891
+ body=_body_params,
892
+ post_params=_form_params,
893
+ files=_files,
894
+ auth_settings=_auth_settings,
895
+ collection_formats=_collection_formats,
896
+ _host=_host,
897
+ _request_auth=_request_auth
898
+ )
899
+
900
+
901
+
902
+
903
+ @validate_call
904
+ def execution_user_messages_add(
905
+ self,
906
+ execution_id: Annotated[StrictStr, Field(description="The unique identifier of the execution")],
907
+ agents_execution_user_messages_add_text_body: Annotated[AgentsExecutionUserMessagesAddTextBody, Field(description="The message content to send")],
607
908
  _request_timeout: Union[
608
909
  None,
609
910
  Annotated[StrictFloat, Field(gt=0)],
@@ -623,8 +924,8 @@ class ExecutionApi:
623
924
 
624
925
  :param execution_id: The unique identifier of the execution (required)
625
926
  :type execution_id: str
626
- :param execution_user_messages_add_text_body: The message content to send (required)
627
- :type execution_user_messages_add_text_body: ExecutionUserMessagesAddTextBody
927
+ :param agents_execution_user_messages_add_text_body: The message content to send (required)
928
+ :type agents_execution_user_messages_add_text_body: AgentsExecutionUserMessagesAddTextBody
628
929
  :param _request_timeout: timeout setting for this request. If one
629
930
  number provided, it will be total request
630
931
  timeout. It can also be a pair (tuple) of
@@ -647,9 +948,9 @@ class ExecutionApi:
647
948
  :return: Returns the result object.
648
949
  """ # noqa: E501
649
950
 
650
- _param = self._user_messages_add_serialize(
951
+ _param = self._execution_user_messages_add_serialize(
651
952
  execution_id=execution_id,
652
- execution_user_messages_add_text_body=execution_user_messages_add_text_body,
953
+ agents_execution_user_messages_add_text_body=agents_execution_user_messages_add_text_body,
653
954
  _request_auth=_request_auth,
654
955
  _content_type=_content_type,
655
956
  _headers=_headers,
@@ -658,6 +959,9 @@ class ExecutionApi:
658
959
 
659
960
  _response_types_map: Dict[str, Optional[str]] = {
660
961
  '201': "str",
962
+ '400': "CommonBadRequestErrorBody",
963
+ '403': "CommonForbiddenErrorBody",
964
+ '404': "CommonNotFoundErrorBody",
661
965
  }
662
966
  response_data = self.api_client.call_api(
663
967
  *_param,
@@ -671,10 +975,10 @@ class ExecutionApi:
671
975
 
672
976
 
673
977
  @validate_call
674
- def user_messages_add_with_http_info(
978
+ def execution_user_messages_add_with_http_info(
675
979
  self,
676
- execution_id: Annotated[UUID, Field(description="The unique identifier of the execution")],
677
- execution_user_messages_add_text_body: Annotated[ExecutionUserMessagesAddTextBody, Field(description="The message content to send")],
980
+ execution_id: Annotated[StrictStr, Field(description="The unique identifier of the execution")],
981
+ agents_execution_user_messages_add_text_body: Annotated[AgentsExecutionUserMessagesAddTextBody, Field(description="The message content to send")],
678
982
  _request_timeout: Union[
679
983
  None,
680
984
  Annotated[StrictFloat, Field(gt=0)],
@@ -694,8 +998,8 @@ class ExecutionApi:
694
998
 
695
999
  :param execution_id: The unique identifier of the execution (required)
696
1000
  :type execution_id: str
697
- :param execution_user_messages_add_text_body: The message content to send (required)
698
- :type execution_user_messages_add_text_body: ExecutionUserMessagesAddTextBody
1001
+ :param agents_execution_user_messages_add_text_body: The message content to send (required)
1002
+ :type agents_execution_user_messages_add_text_body: AgentsExecutionUserMessagesAddTextBody
699
1003
  :param _request_timeout: timeout setting for this request. If one
700
1004
  number provided, it will be total request
701
1005
  timeout. It can also be a pair (tuple) of
@@ -718,9 +1022,9 @@ class ExecutionApi:
718
1022
  :return: Returns the result object.
719
1023
  """ # noqa: E501
720
1024
 
721
- _param = self._user_messages_add_serialize(
1025
+ _param = self._execution_user_messages_add_serialize(
722
1026
  execution_id=execution_id,
723
- execution_user_messages_add_text_body=execution_user_messages_add_text_body,
1027
+ agents_execution_user_messages_add_text_body=agents_execution_user_messages_add_text_body,
724
1028
  _request_auth=_request_auth,
725
1029
  _content_type=_content_type,
726
1030
  _headers=_headers,
@@ -729,6 +1033,9 @@ class ExecutionApi:
729
1033
 
730
1034
  _response_types_map: Dict[str, Optional[str]] = {
731
1035
  '201': "str",
1036
+ '400': "CommonBadRequestErrorBody",
1037
+ '403': "CommonForbiddenErrorBody",
1038
+ '404': "CommonNotFoundErrorBody",
732
1039
  }
733
1040
  response_data = self.api_client.call_api(
734
1041
  *_param,
@@ -742,10 +1049,10 @@ class ExecutionApi:
742
1049
 
743
1050
 
744
1051
  @validate_call
745
- def user_messages_add_without_preload_content(
1052
+ def execution_user_messages_add_without_preload_content(
746
1053
  self,
747
- execution_id: Annotated[UUID, Field(description="The unique identifier of the execution")],
748
- execution_user_messages_add_text_body: Annotated[ExecutionUserMessagesAddTextBody, Field(description="The message content to send")],
1054
+ execution_id: Annotated[StrictStr, Field(description="The unique identifier of the execution")],
1055
+ agents_execution_user_messages_add_text_body: Annotated[AgentsExecutionUserMessagesAddTextBody, Field(description="The message content to send")],
749
1056
  _request_timeout: Union[
750
1057
  None,
751
1058
  Annotated[StrictFloat, Field(gt=0)],
@@ -765,8 +1072,8 @@ class ExecutionApi:
765
1072
 
766
1073
  :param execution_id: The unique identifier of the execution (required)
767
1074
  :type execution_id: str
768
- :param execution_user_messages_add_text_body: The message content to send (required)
769
- :type execution_user_messages_add_text_body: ExecutionUserMessagesAddTextBody
1075
+ :param agents_execution_user_messages_add_text_body: The message content to send (required)
1076
+ :type agents_execution_user_messages_add_text_body: AgentsExecutionUserMessagesAddTextBody
770
1077
  :param _request_timeout: timeout setting for this request. If one
771
1078
  number provided, it will be total request
772
1079
  timeout. It can also be a pair (tuple) of
@@ -789,9 +1096,9 @@ class ExecutionApi:
789
1096
  :return: Returns the result object.
790
1097
  """ # noqa: E501
791
1098
 
792
- _param = self._user_messages_add_serialize(
1099
+ _param = self._execution_user_messages_add_serialize(
793
1100
  execution_id=execution_id,
794
- execution_user_messages_add_text_body=execution_user_messages_add_text_body,
1101
+ agents_execution_user_messages_add_text_body=agents_execution_user_messages_add_text_body,
795
1102
  _request_auth=_request_auth,
796
1103
  _content_type=_content_type,
797
1104
  _headers=_headers,
@@ -800,6 +1107,9 @@ class ExecutionApi:
800
1107
 
801
1108
  _response_types_map: Dict[str, Optional[str]] = {
802
1109
  '201': "str",
1110
+ '400': "CommonBadRequestErrorBody",
1111
+ '403': "CommonForbiddenErrorBody",
1112
+ '404': "CommonNotFoundErrorBody",
803
1113
  }
804
1114
  response_data = self.api_client.call_api(
805
1115
  *_param,
@@ -808,10 +1118,10 @@ class ExecutionApi:
808
1118
  return response_data.response
809
1119
 
810
1120
 
811
- def _user_messages_add_serialize(
1121
+ def _execution_user_messages_add_serialize(
812
1122
  self,
813
1123
  execution_id,
814
- execution_user_messages_add_text_body,
1124
+ agents_execution_user_messages_add_text_body,
815
1125
  _request_auth,
816
1126
  _content_type,
817
1127
  _headers,
@@ -839,8 +1149,8 @@ class ExecutionApi:
839
1149
  # process the header parameters
840
1150
  # process the form parameters
841
1151
  # process the body parameter
842
- if execution_user_messages_add_text_body is not None:
843
- _body_params = execution_user_messages_add_text_body
1152
+ if agents_execution_user_messages_add_text_body is not None:
1153
+ _body_params = agents_execution_user_messages_add_text_body
844
1154
 
845
1155
 
846
1156
  # set the HTTP header `Accept`