polyaxon 2.0.0rc49__py3-none-any.whl → 2.4.0rc1__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 (177) hide show
  1. polyaxon/_auxiliaries/cleaner.py +8 -3
  2. polyaxon/_auxiliaries/init.py +7 -2
  3. polyaxon/_auxiliaries/notifier.py +8 -2
  4. polyaxon/_auxiliaries/sidecar.py +30 -2
  5. polyaxon/_cli/artifacts.py +96 -11
  6. polyaxon/_cli/components.py +96 -11
  7. polyaxon/_cli/config.py +118 -22
  8. polyaxon/_cli/dashboard.py +15 -2
  9. polyaxon/_cli/init.py +1 -1
  10. polyaxon/_cli/models.py +96 -11
  11. polyaxon/_cli/operations.py +267 -90
  12. polyaxon/_cli/project_versions.py +139 -6
  13. polyaxon/_cli/projects.py +23 -9
  14. polyaxon/_cli/run.py +37 -9
  15. polyaxon/_cli/services/agent.py +2 -2
  16. polyaxon/_cli/services/clean_artifacts.py +1 -1
  17. polyaxon/_cli/services/sidecar.py +8 -1
  18. polyaxon/_client/client.py +17 -0
  19. polyaxon/_client/mixin.py +39 -0
  20. polyaxon/_client/project.py +218 -23
  21. polyaxon/_client/run.py +131 -33
  22. polyaxon/_compiler/contexts/contexts.py +2 -2
  23. polyaxon/_compiler/contexts/ray_job.py +4 -2
  24. polyaxon/_compiler/resolver/agent.py +12 -2
  25. polyaxon/_compiler/resolver/runtime.py +2 -2
  26. polyaxon/_contexts/paths.py +4 -7
  27. polyaxon/_deploy/operators/compose.py +1 -27
  28. polyaxon/_deploy/schemas/deployment.py +4 -1
  29. polyaxon/_deploy/schemas/intervals.py +0 -7
  30. polyaxon/_deploy/schemas/proxy.py +1 -0
  31. polyaxon/_deploy/schemas/service.py +11 -1
  32. polyaxon/_docker/converter/base/base.py +8 -0
  33. polyaxon/_docker/executor.py +10 -4
  34. polyaxon/_env_vars/getters/owner_entity.py +4 -2
  35. polyaxon/_env_vars/getters/project.py +4 -2
  36. polyaxon/_env_vars/getters/run.py +5 -2
  37. polyaxon/_env_vars/keys.py +7 -1
  38. polyaxon/_flow/__init__.py +2 -0
  39. polyaxon/_flow/builds/__init__.py +19 -6
  40. polyaxon/_flow/component/base.py +1 -0
  41. polyaxon/_flow/component/component.py +14 -0
  42. polyaxon/_flow/environment/__init__.py +8 -8
  43. polyaxon/_flow/hooks/__init__.py +19 -6
  44. polyaxon/_flow/init/__init__.py +6 -6
  45. polyaxon/_flow/matrix/iterative.py +0 -1
  46. polyaxon/_flow/matrix/tuner.py +18 -6
  47. polyaxon/_flow/operations/operation.py +44 -17
  48. polyaxon/_flow/plugins/__init__.py +6 -0
  49. polyaxon/_flow/run/__init__.py +2 -2
  50. polyaxon/_flow/run/dag.py +2 -2
  51. polyaxon/_flow/run/dask/dask.py +0 -1
  52. polyaxon/_flow/run/dask/replica.py +3 -3
  53. polyaxon/_flow/run/enums.py +5 -0
  54. polyaxon/_flow/run/job.py +4 -4
  55. polyaxon/_flow/run/kubeflow/mpi_job.py +1 -2
  56. polyaxon/_flow/run/kubeflow/mx_job.py +1 -2
  57. polyaxon/_flow/run/kubeflow/paddle_job.py +35 -4
  58. polyaxon/_flow/run/kubeflow/pytorch_job.py +51 -5
  59. polyaxon/_flow/run/kubeflow/replica.py +4 -4
  60. polyaxon/_flow/run/kubeflow/scheduling_policy.py +12 -0
  61. polyaxon/_flow/run/kubeflow/tf_job.py +3 -3
  62. polyaxon/_flow/run/kubeflow/xgboost_job.py +1 -2
  63. polyaxon/_flow/run/ray/ray.py +2 -3
  64. polyaxon/_flow/run/ray/replica.py +3 -3
  65. polyaxon/_flow/run/service.py +4 -4
  66. polyaxon/_fs/fs.py +7 -2
  67. polyaxon/_fs/utils.py +3 -2
  68. polyaxon/_k8s/converter/base/base.py +2 -1
  69. polyaxon/_k8s/converter/base/main.py +1 -0
  70. polyaxon/_k8s/converter/base/sidecar.py +16 -1
  71. polyaxon/_k8s/converter/common/accelerators.py +7 -4
  72. polyaxon/_k8s/converter/converters/job.py +1 -1
  73. polyaxon/_k8s/converter/converters/kubeflow/paddle_job.py +1 -0
  74. polyaxon/_k8s/converter/converters/kubeflow/pytroch_job.py +2 -0
  75. polyaxon/_k8s/converter/converters/kubeflow/tf_job.py +1 -0
  76. polyaxon/_k8s/converter/converters/ray_job.py +4 -2
  77. polyaxon/_k8s/custom_resources/dask_job.py +3 -0
  78. polyaxon/_k8s/custom_resources/kubeflow/common.py +4 -1
  79. polyaxon/_k8s/custom_resources/kubeflow/paddle_job.py +10 -1
  80. polyaxon/_k8s/custom_resources/kubeflow/pytorch_job.py +14 -1
  81. polyaxon/_k8s/custom_resources/kubeflow/tf_job.py +4 -0
  82. polyaxon/_k8s/custom_resources/ray_job.py +3 -0
  83. polyaxon/_k8s/custom_resources/setter.py +1 -1
  84. polyaxon/_k8s/executor/async_executor.py +2 -0
  85. polyaxon/_k8s/executor/base.py +23 -6
  86. polyaxon/_k8s/logging/async_monitor.py +150 -5
  87. polyaxon/_k8s/manager/async_manager.py +96 -23
  88. polyaxon/_k8s/manager/base.py +4 -0
  89. polyaxon/_k8s/manager/manager.py +282 -134
  90. polyaxon/_local_process/__init__.py +0 -0
  91. polyaxon/_local_process/agent.py +6 -0
  92. polyaxon/_local_process/converter/__init__.py +1 -0
  93. polyaxon/_local_process/converter/base/__init__.py +1 -0
  94. polyaxon/_local_process/converter/base/base.py +140 -0
  95. polyaxon/_local_process/converter/base/containers.py +69 -0
  96. polyaxon/_local_process/converter/base/env_vars.py +253 -0
  97. polyaxon/_local_process/converter/base/init.py +414 -0
  98. polyaxon/_local_process/converter/base/main.py +74 -0
  99. polyaxon/_local_process/converter/base/mounts.py +82 -0
  100. polyaxon/_local_process/converter/converters/__init__.py +8 -0
  101. polyaxon/_local_process/converter/converters/job.py +40 -0
  102. polyaxon/_local_process/converter/converters/service.py +41 -0
  103. polyaxon/_local_process/converter/mixins.py +38 -0
  104. polyaxon/_local_process/executor.py +132 -0
  105. polyaxon/_local_process/process_types.py +39 -0
  106. polyaxon/_managers/agent.py +2 -0
  107. polyaxon/_managers/home.py +2 -1
  108. polyaxon/_operations/tuner.py +1 -0
  109. polyaxon/_polyaxonfile/check.py +2 -0
  110. polyaxon/_polyaxonfile/manager/operations.py +3 -0
  111. polyaxon/_polyaxonfile/manager/workflows.py +2 -0
  112. polyaxon/_polyaxonfile/specs/compiled_operation.py +1 -0
  113. polyaxon/_polyaxonfile/specs/operation.py +1 -0
  114. polyaxon/_polyaxonfile/specs/sections.py +3 -0
  115. polyaxon/_pql/manager.py +1 -1
  116. polyaxon/_runner/agent/async_agent.py +97 -21
  117. polyaxon/_runner/agent/base_agent.py +27 -9
  118. polyaxon/_runner/agent/client.py +15 -1
  119. polyaxon/_runner/agent/sync_agent.py +85 -20
  120. polyaxon/_runner/converter/converter.py +6 -2
  121. polyaxon/_runner/executor.py +13 -7
  122. polyaxon/_schemas/agent.py +27 -1
  123. polyaxon/_schemas/client.py +30 -3
  124. polyaxon/_schemas/installation.py +4 -3
  125. polyaxon/_schemas/lifecycle.py +10 -5
  126. polyaxon/_schemas/log_handler.py +2 -3
  127. polyaxon/_schemas/types/artifacts.py +3 -3
  128. polyaxon/_schemas/types/dockerfile.py +3 -3
  129. polyaxon/_schemas/types/file.py +3 -3
  130. polyaxon/_schemas/types/git.py +3 -3
  131. polyaxon/_schemas/types/tensorboard.py +3 -3
  132. polyaxon/_sdk/api/agents_v1_api.py +1076 -73
  133. polyaxon/_sdk/api/organizations_v1_api.py +371 -10
  134. polyaxon/_sdk/api/project_dashboards_v1_api.py +12 -12
  135. polyaxon/_sdk/api/project_searches_v1_api.py +12 -12
  136. polyaxon/_sdk/api/projects_v1_api.py +221 -44
  137. polyaxon/_sdk/api/runs_v1_api.py +917 -445
  138. polyaxon/_sdk/api/service_accounts_v1_api.py +16 -16
  139. polyaxon/_sdk/api/teams_v1_api.py +2827 -375
  140. polyaxon/_sdk/api/users_v1_api.py +231 -55
  141. polyaxon/_sdk/async_client/api_client.py +4 -0
  142. polyaxon/_sdk/schemas/__init__.py +10 -2
  143. polyaxon/_sdk/schemas/v1_agent.py +2 -1
  144. polyaxon/_sdk/schemas/v1_agent_reconcile_body_request.py +14 -0
  145. polyaxon/_sdk/schemas/v1_artifact_tree.py +1 -1
  146. polyaxon/_sdk/schemas/v1_dashboard_spec.py +4 -0
  147. polyaxon/_sdk/schemas/v1_events_response.py +4 -0
  148. polyaxon/_sdk/schemas/v1_organization.py +1 -0
  149. polyaxon/_sdk/schemas/v1_preset.py +8 -0
  150. polyaxon/_sdk/schemas/v1_project.py +1 -0
  151. polyaxon/_sdk/schemas/v1_project_settings.py +4 -2
  152. polyaxon/_sdk/schemas/v1_run.py +2 -2
  153. polyaxon/_sdk/schemas/v1_run_edge_lineage.py +14 -0
  154. polyaxon/_sdk/schemas/v1_run_edges_graph.py +9 -0
  155. polyaxon/_sdk/schemas/v1_section_spec.py +7 -2
  156. polyaxon/_sdk/schemas/v1_settings_catalog.py +1 -0
  157. polyaxon/_sdk/schemas/v1_team.py +3 -0
  158. polyaxon/_sdk/schemas/v1_user.py +1 -2
  159. polyaxon/_sdk/schemas/v1_user_access.py +17 -0
  160. polyaxon/_services/values.py +1 -0
  161. polyaxon/_sidecar/container/__init__.py +39 -18
  162. polyaxon/_sidecar/container/monitors/__init__.py +1 -0
  163. polyaxon/_sidecar/container/monitors/logs.py +10 -13
  164. polyaxon/_sidecar/container/monitors/spec.py +24 -0
  165. polyaxon/_sidecar/ignore.py +0 -1
  166. polyaxon/_utils/fqn_utils.py +25 -2
  167. polyaxon/client.py +1 -1
  168. polyaxon/pkg.py +1 -1
  169. polyaxon/schemas.py +8 -1
  170. polyaxon/settings.py +6 -0
  171. {polyaxon-2.0.0rc49.dist-info → polyaxon-2.4.0rc1.dist-info}/METADATA +43 -43
  172. {polyaxon-2.0.0rc49.dist-info → polyaxon-2.4.0rc1.dist-info}/RECORD +176 -155
  173. {polyaxon-2.0.0rc49.dist-info → polyaxon-2.4.0rc1.dist-info}/WHEEL +1 -1
  174. polyaxon/_sdk/schemas/v1_project_user_access.py +0 -10
  175. {polyaxon-2.0.0rc49.dist-info → polyaxon-2.4.0rc1.dist-info}/LICENSE +0 -0
  176. {polyaxon-2.0.0rc49.dist-info → polyaxon-2.4.0rc1.dist-info}/entry_points.txt +0 -0
  177. {polyaxon-2.0.0rc49.dist-info → polyaxon-2.4.0rc1.dist-info}/top_level.txt +0 -0
@@ -1,4 +1,4 @@
1
- from typing import Optional
1
+ from typing import Dict, Optional
2
2
  from typing_extensions import Annotated
3
3
 
4
4
  from clipped.compact.pydantic import Field, StrictInt, StrictStr, validate_arguments
@@ -6,14 +6,189 @@ from clipped.compact.pydantic import Field, StrictInt, StrictStr, validate_argum
6
6
  from polyaxon._schemas.lifecycle import V1Status
7
7
  from polyaxon._sdk.base_api import BaseApi
8
8
  from polyaxon._sdk.schemas.v1_agent import V1Agent
9
+ from polyaxon._sdk.schemas.v1_agent_reconcile_body_request import (
10
+ V1AgentReconcileBodyRequest,
11
+ )
9
12
  from polyaxon._sdk.schemas.v1_agent_state_response import V1AgentStateResponse
10
13
  from polyaxon._sdk.schemas.v1_agent_status_body_request import V1AgentStatusBodyRequest
11
14
  from polyaxon._sdk.schemas.v1_list_agents_response import V1ListAgentsResponse
12
15
  from polyaxon._sdk.schemas.v1_token import V1Token
13
16
  from polyaxon.exceptions import ApiTypeError
17
+ from traceml.logging import V1Logs
14
18
 
15
19
 
16
20
  class AgentsV1Api(BaseApi):
21
+ @validate_arguments
22
+ def collect_agent_data(
23
+ self,
24
+ namespace: Annotated[StrictStr, Field(..., description="namespace")],
25
+ owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
26
+ uuid: Annotated[
27
+ StrictStr, Field(..., description="Uuid identifier of the entity")
28
+ ],
29
+ **kwargs
30
+ ) -> object: # noqa: E501
31
+ """collect agent # noqa: E501
32
+
33
+ This method makes a synchronous HTTP request by default. To make an
34
+ asynchronous HTTP request, please pass async_req=True
35
+
36
+ >>> thread = api.collect_agent_data(namespace, owner, uuid, async_req=True)
37
+ >>> result = thread.get()
38
+
39
+ :param namespace: namespace (required)
40
+ :type namespace: str
41
+ :param owner: Owner of the namespace (required)
42
+ :type owner: str
43
+ :param uuid: Uuid identifier of the entity (required)
44
+ :type uuid: str
45
+ :param async_req: Whether to execute the request asynchronously.
46
+ :type async_req: bool, optional
47
+ :param _preload_content: if False, the urllib3.HTTPResponse object will
48
+ be returned without reading/decoding response
49
+ data. Default is True.
50
+ :type _preload_content: bool, optional
51
+ :param _request_timeout: timeout setting for this request. If one
52
+ number provided, it will be total request
53
+ timeout. It can also be a pair (tuple) of
54
+ (connection, read) timeouts.
55
+ :return: Returns the result object.
56
+ If the method is called asynchronously,
57
+ returns the request thread.
58
+ :rtype: object
59
+ """
60
+ kwargs["_return_http_data_only"] = True
61
+ return self.collect_agent_data_with_http_info(
62
+ namespace, owner, uuid, **kwargs
63
+ ) # noqa: E501
64
+
65
+ @validate_arguments
66
+ def collect_agent_data_with_http_info(
67
+ self,
68
+ namespace: Annotated[StrictStr, Field(..., description="namespace")],
69
+ owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
70
+ uuid: Annotated[
71
+ StrictStr, Field(..., description="Uuid identifier of the entity")
72
+ ],
73
+ **kwargs
74
+ ): # noqa: E501
75
+ """collect agent # noqa: E501
76
+
77
+ This method makes a synchronous HTTP request by default. To make an
78
+ asynchronous HTTP request, please pass async_req=True
79
+
80
+ >>> thread = api.collect_agent_data_with_http_info(namespace, owner, uuid, async_req=True)
81
+ >>> result = thread.get()
82
+
83
+ :param namespace: namespace (required)
84
+ :type namespace: str
85
+ :param owner: Owner of the namespace (required)
86
+ :type owner: str
87
+ :param uuid: Uuid identifier of the entity (required)
88
+ :type uuid: str
89
+ :param async_req: Whether to execute the request asynchronously.
90
+ :type async_req: bool, optional
91
+ :param _return_http_data_only: response data without head status code
92
+ and headers
93
+ :type _return_http_data_only: bool, optional
94
+ :param _preload_content: if False, the urllib3.HTTPResponse object will
95
+ be returned without reading/decoding response
96
+ data. Default is True.
97
+ :type _preload_content: bool, optional
98
+ :param _request_timeout: timeout setting for this request. If one
99
+ number provided, it will be total request
100
+ timeout. It can also be a pair (tuple) of
101
+ (connection, read) timeouts.
102
+ :param _request_auth: set to override the auth_settings for an a single
103
+ request; this effectively ignores the authentication
104
+ in the spec for a single request.
105
+ :type _request_auth: dict, optional
106
+ :type _content_type: string, optional: force content-type for the request
107
+ :return: Returns the result object.
108
+ If the method is called asynchronously,
109
+ returns the request thread.
110
+ :rtype: tuple(object, status_code(int), headers(HTTPHeaderDict))
111
+ """
112
+
113
+ _params = locals()
114
+
115
+ _all_params = ["namespace", "owner", "uuid"]
116
+ _all_params.extend(
117
+ [
118
+ "async_req",
119
+ "_return_http_data_only",
120
+ "_preload_content",
121
+ "_request_timeout",
122
+ "_request_auth",
123
+ "_content_type",
124
+ "_headers",
125
+ ]
126
+ )
127
+
128
+ # validate the arguments
129
+ for _key, _val in _params["kwargs"].items():
130
+ if _key not in _all_params:
131
+ raise ApiTypeError(
132
+ "Got an unexpected keyword argument '%s'"
133
+ " to method collect_agent_data" % _key
134
+ )
135
+ _params[_key] = _val
136
+ del _params["kwargs"]
137
+
138
+ _collection_formats = {}
139
+
140
+ # process the path parameters
141
+ _path_params = {}
142
+ if _params["namespace"]:
143
+ _path_params["namespace"] = _params["namespace"]
144
+ if _params["owner"]:
145
+ _path_params["owner"] = _params["owner"]
146
+ if _params["uuid"]:
147
+ _path_params["uuid"] = _params["uuid"]
148
+
149
+ # process the query parameters
150
+ _query_params = []
151
+ # process the header parameters
152
+ _header_params = dict(_params.get("_headers", {}))
153
+ # process the form parameters
154
+ _form_params = []
155
+ _files = {}
156
+ # process the body parameter
157
+ _body_params = None
158
+ # set the HTTP header `Accept`
159
+ _header_params["Accept"] = self.api_client.select_header_accept(
160
+ ["application/json"]
161
+ ) # noqa: E501
162
+
163
+ # authentication setting
164
+ _auth_settings = ["ApiKey"] # noqa: E501
165
+
166
+ _response_types_map = {
167
+ "200": "object",
168
+ "204": "object",
169
+ "403": "object",
170
+ "404": "object",
171
+ }
172
+
173
+ return self.api_client.call_api(
174
+ "/internal/v1/{namespace}/{owner}/agents/{uuid}/collect",
175
+ "POST",
176
+ _path_params,
177
+ _query_params,
178
+ _header_params,
179
+ body=_body_params,
180
+ post_params=_form_params,
181
+ files=_files,
182
+ response_types_map=_response_types_map,
183
+ auth_settings=_auth_settings,
184
+ async_req=_params.get("async_req"),
185
+ _return_http_data_only=_params.get("_return_http_data_only"), # noqa: E501
186
+ _preload_content=_params.get("_preload_content", True),
187
+ _request_timeout=_params.get("_request_timeout"),
188
+ collection_formats=_collection_formats,
189
+ _request_auth=_params.get("_request_auth"),
190
+ )
191
+
17
192
  @validate_arguments
18
193
  def create_agent(
19
194
  self,
@@ -905,26 +1080,47 @@ class AgentsV1Api(BaseApi):
905
1080
  )
906
1081
 
907
1082
  @validate_arguments
908
- def get_agent_state(
1083
+ def get_agent_logs(
909
1084
  self,
1085
+ namespace: Annotated[StrictStr, Field(..., description="namespace")],
910
1086
  owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
911
1087
  uuid: Annotated[
912
1088
  StrictStr, Field(..., description="Uuid identifier of the entity")
913
1089
  ],
1090
+ service: Annotated[Optional[StrictStr], Field(description="Service.")] = None,
1091
+ last_file: Annotated[
1092
+ Optional[StrictStr], Field(description="last_file.")
1093
+ ] = None,
1094
+ force: Annotated[
1095
+ Optional[bool], Field(description="Force query param.")
1096
+ ] = None,
1097
+ connection: Annotated[
1098
+ Optional[StrictStr], Field(description="Connection to use.")
1099
+ ] = None,
914
1100
  **kwargs
915
- ) -> V1AgentStateResponse: # noqa: E501
916
- """Get State (queues/runs) # noqa: E501
1101
+ ) -> V1Logs: # noqa: E501
1102
+ """Get run logs # noqa: E501
917
1103
 
918
1104
  This method makes a synchronous HTTP request by default. To make an
919
1105
  asynchronous HTTP request, please pass async_req=True
920
1106
 
921
- >>> thread = api.get_agent_state(owner, uuid, async_req=True)
1107
+ >>> thread = api.get_agent_logs(namespace, owner, uuid, service, last_file, force, connection, async_req=True)
922
1108
  >>> result = thread.get()
923
1109
 
1110
+ :param namespace: namespace (required)
1111
+ :type namespace: str
924
1112
  :param owner: Owner of the namespace (required)
925
1113
  :type owner: str
926
1114
  :param uuid: Uuid identifier of the entity (required)
927
1115
  :type uuid: str
1116
+ :param service: Service.
1117
+ :type service: str
1118
+ :param last_file: last_file.
1119
+ :type last_file: str
1120
+ :param force: Force query param.
1121
+ :type force: bool
1122
+ :param connection: Connection to use.
1123
+ :type connection: str
928
1124
  :param async_req: Whether to execute the request asynchronously.
929
1125
  :type async_req: bool, optional
930
1126
  :param _preload_content: if False, the urllib3.HTTPResponse object will
@@ -938,32 +1134,55 @@ class AgentsV1Api(BaseApi):
938
1134
  :return: Returns the result object.
939
1135
  If the method is called asynchronously,
940
1136
  returns the request thread.
941
- :rtype: V1AgentStateResponse
1137
+ :rtype: V1Logs
942
1138
  """
943
1139
  kwargs["_return_http_data_only"] = True
944
- return self.get_agent_state_with_http_info(owner, uuid, **kwargs) # noqa: E501
1140
+ return self.get_agent_logs_with_http_info(
1141
+ namespace, owner, uuid, service, last_file, force, connection, **kwargs
1142
+ ) # noqa: E501
945
1143
 
946
1144
  @validate_arguments
947
- def get_agent_state_with_http_info(
1145
+ def get_agent_logs_with_http_info(
948
1146
  self,
1147
+ namespace: Annotated[StrictStr, Field(..., description="namespace")],
949
1148
  owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
950
1149
  uuid: Annotated[
951
1150
  StrictStr, Field(..., description="Uuid identifier of the entity")
952
1151
  ],
1152
+ service: Annotated[Optional[StrictStr], Field(description="Service.")] = None,
1153
+ last_file: Annotated[
1154
+ Optional[StrictStr], Field(description="last_file.")
1155
+ ] = None,
1156
+ force: Annotated[
1157
+ Optional[bool], Field(description="Force query param.")
1158
+ ] = None,
1159
+ connection: Annotated[
1160
+ Optional[StrictStr], Field(description="Connection to use.")
1161
+ ] = None,
953
1162
  **kwargs
954
1163
  ): # noqa: E501
955
- """Get State (queues/runs) # noqa: E501
1164
+ """Get run logs # noqa: E501
956
1165
 
957
1166
  This method makes a synchronous HTTP request by default. To make an
958
1167
  asynchronous HTTP request, please pass async_req=True
959
1168
 
960
- >>> thread = api.get_agent_state_with_http_info(owner, uuid, async_req=True)
1169
+ >>> thread = api.get_agent_logs_with_http_info(namespace, owner, uuid, service, last_file, force, connection, async_req=True)
961
1170
  >>> result = thread.get()
962
1171
 
1172
+ :param namespace: namespace (required)
1173
+ :type namespace: str
963
1174
  :param owner: Owner of the namespace (required)
964
1175
  :type owner: str
965
1176
  :param uuid: Uuid identifier of the entity (required)
966
1177
  :type uuid: str
1178
+ :param service: Service.
1179
+ :type service: str
1180
+ :param last_file: last_file.
1181
+ :type last_file: str
1182
+ :param force: Force query param.
1183
+ :type force: bool
1184
+ :param connection: Connection to use.
1185
+ :type connection: str
967
1186
  :param async_req: Whether to execute the request asynchronously.
968
1187
  :type async_req: bool, optional
969
1188
  :param _return_http_data_only: response data without head status code
@@ -985,12 +1204,20 @@ class AgentsV1Api(BaseApi):
985
1204
  :return: Returns the result object.
986
1205
  If the method is called asynchronously,
987
1206
  returns the request thread.
988
- :rtype: tuple(V1AgentStateResponse, status_code(int), headers(HTTPHeaderDict))
1207
+ :rtype: tuple(V1Logs, status_code(int), headers(HTTPHeaderDict))
989
1208
  """
990
1209
 
991
1210
  _params = locals()
992
1211
 
993
- _all_params = ["owner", "uuid"]
1212
+ _all_params = [
1213
+ "namespace",
1214
+ "owner",
1215
+ "uuid",
1216
+ "service",
1217
+ "last_file",
1218
+ "force",
1219
+ "connection",
1220
+ ]
994
1221
  _all_params.extend(
995
1222
  [
996
1223
  "async_req",
@@ -1008,7 +1235,7 @@ class AgentsV1Api(BaseApi):
1008
1235
  if _key not in _all_params:
1009
1236
  raise ApiTypeError(
1010
1237
  "Got an unexpected keyword argument '%s'"
1011
- " to method get_agent_state" % _key
1238
+ " to method get_agent_logs" % _key
1012
1239
  )
1013
1240
  _params[_key] = _val
1014
1241
  del _params["kwargs"]
@@ -1017,6 +1244,8 @@ class AgentsV1Api(BaseApi):
1017
1244
 
1018
1245
  # process the path parameters
1019
1246
  _path_params = {}
1247
+ if _params["namespace"]:
1248
+ _path_params["namespace"] = _params["namespace"]
1020
1249
  if _params["owner"]:
1021
1250
  _path_params["owner"] = _params["owner"]
1022
1251
  if _params["uuid"]:
@@ -1024,17 +1253,22 @@ class AgentsV1Api(BaseApi):
1024
1253
 
1025
1254
  # process the query parameters
1026
1255
  _query_params = []
1256
+ if _params.get("service") is not None: # noqa: E501
1257
+ _query_params.append(("service", _params["service"]))
1258
+ if _params.get("last_file") is not None: # noqa: E501
1259
+ _query_params.append(("last_file", _params["last_file"]))
1260
+ if _params.get("force") is not None: # noqa: E501
1261
+ _query_params.append(("force", _params["force"]))
1262
+ if _params.get("connection") is not None: # noqa: E501
1263
+ _query_params.append(("connection", _params["connection"]))
1027
1264
 
1028
1265
  # process the header parameters
1029
1266
  _header_params = dict(_params.get("_headers", {}))
1030
-
1031
1267
  # process the form parameters
1032
1268
  _form_params = []
1033
1269
  _files = {}
1034
-
1035
1270
  # process the body parameter
1036
1271
  _body_params = None
1037
-
1038
1272
  # set the HTTP header `Accept`
1039
1273
  _header_params["Accept"] = self.api_client.select_header_accept(
1040
1274
  ["application/json"]
@@ -1044,14 +1278,14 @@ class AgentsV1Api(BaseApi):
1044
1278
  _auth_settings = ["ApiKey"] # noqa: E501
1045
1279
 
1046
1280
  _response_types_map = {
1047
- "200": "V1AgentStateResponse",
1281
+ "200": "V1Logs",
1048
1282
  "204": "object",
1049
1283
  "403": "object",
1050
1284
  "404": "object",
1051
1285
  }
1052
1286
 
1053
1287
  return self.api_client.call_api(
1054
- "/api/v1/orgs/{owner}/agents/{uuid}/state",
1288
+ "/streams/v1/{namespace}/{owner}/agents/{uuid}/logs",
1055
1289
  "GET",
1056
1290
  _path_params,
1057
1291
  _query_params,
@@ -1070,20 +1304,20 @@ class AgentsV1Api(BaseApi):
1070
1304
  )
1071
1305
 
1072
1306
  @validate_arguments
1073
- def get_agent_token(
1307
+ def get_agent_state(
1074
1308
  self,
1075
1309
  owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
1076
1310
  uuid: Annotated[
1077
1311
  StrictStr, Field(..., description="Uuid identifier of the entity")
1078
1312
  ],
1079
1313
  **kwargs
1080
- ) -> V1Token: # noqa: E501
1081
- """Get agent token # noqa: E501
1314
+ ) -> V1AgentStateResponse: # noqa: E501
1315
+ """Get State (queues/runs) # noqa: E501
1082
1316
 
1083
1317
  This method makes a synchronous HTTP request by default. To make an
1084
1318
  asynchronous HTTP request, please pass async_req=True
1085
1319
 
1086
- >>> thread = api.get_agent_token(owner, uuid, async_req=True)
1320
+ >>> thread = api.get_agent_state(owner, uuid, async_req=True)
1087
1321
  >>> result = thread.get()
1088
1322
 
1089
1323
  :param owner: Owner of the namespace (required)
@@ -1103,32 +1337,531 @@ class AgentsV1Api(BaseApi):
1103
1337
  :return: Returns the result object.
1104
1338
  If the method is called asynchronously,
1105
1339
  returns the request thread.
1106
- :rtype: V1Token
1340
+ :rtype: V1AgentStateResponse
1341
+ """
1342
+ kwargs["_return_http_data_only"] = True
1343
+ return self.get_agent_state_with_http_info(owner, uuid, **kwargs) # noqa: E501
1344
+
1345
+ @validate_arguments
1346
+ def get_agent_state_with_http_info(
1347
+ self,
1348
+ owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
1349
+ uuid: Annotated[
1350
+ StrictStr, Field(..., description="Uuid identifier of the entity")
1351
+ ],
1352
+ **kwargs
1353
+ ): # noqa: E501
1354
+ """Get State (queues/runs) # noqa: E501
1355
+
1356
+ This method makes a synchronous HTTP request by default. To make an
1357
+ asynchronous HTTP request, please pass async_req=True
1358
+
1359
+ >>> thread = api.get_agent_state_with_http_info(owner, uuid, async_req=True)
1360
+ >>> result = thread.get()
1361
+
1362
+ :param owner: Owner of the namespace (required)
1363
+ :type owner: str
1364
+ :param uuid: Uuid identifier of the entity (required)
1365
+ :type uuid: str
1366
+ :param async_req: Whether to execute the request asynchronously.
1367
+ :type async_req: bool, optional
1368
+ :param _return_http_data_only: response data without head status code
1369
+ and headers
1370
+ :type _return_http_data_only: bool, optional
1371
+ :param _preload_content: if False, the urllib3.HTTPResponse object will
1372
+ be returned without reading/decoding response
1373
+ data. Default is True.
1374
+ :type _preload_content: bool, optional
1375
+ :param _request_timeout: timeout setting for this request. If one
1376
+ number provided, it will be total request
1377
+ timeout. It can also be a pair (tuple) of
1378
+ (connection, read) timeouts.
1379
+ :param _request_auth: set to override the auth_settings for an a single
1380
+ request; this effectively ignores the authentication
1381
+ in the spec for a single request.
1382
+ :type _request_auth: dict, optional
1383
+ :type _content_type: string, optional: force content-type for the request
1384
+ :return: Returns the result object.
1385
+ If the method is called asynchronously,
1386
+ returns the request thread.
1387
+ :rtype: tuple(V1AgentStateResponse, status_code(int), headers(HTTPHeaderDict))
1388
+ """
1389
+
1390
+ _params = locals()
1391
+
1392
+ _all_params = ["owner", "uuid"]
1393
+ _all_params.extend(
1394
+ [
1395
+ "async_req",
1396
+ "_return_http_data_only",
1397
+ "_preload_content",
1398
+ "_request_timeout",
1399
+ "_request_auth",
1400
+ "_content_type",
1401
+ "_headers",
1402
+ ]
1403
+ )
1404
+
1405
+ # validate the arguments
1406
+ for _key, _val in _params["kwargs"].items():
1407
+ if _key not in _all_params:
1408
+ raise ApiTypeError(
1409
+ "Got an unexpected keyword argument '%s'"
1410
+ " to method get_agent_state" % _key
1411
+ )
1412
+ _params[_key] = _val
1413
+ del _params["kwargs"]
1414
+
1415
+ _collection_formats = {}
1416
+
1417
+ # process the path parameters
1418
+ _path_params = {}
1419
+ if _params["owner"]:
1420
+ _path_params["owner"] = _params["owner"]
1421
+ if _params["uuid"]:
1422
+ _path_params["uuid"] = _params["uuid"]
1423
+
1424
+ # process the query parameters
1425
+ _query_params = []
1426
+
1427
+ # process the header parameters
1428
+ _header_params = dict(_params.get("_headers", {}))
1429
+
1430
+ # process the form parameters
1431
+ _form_params = []
1432
+ _files = {}
1433
+
1434
+ # process the body parameter
1435
+ _body_params = None
1436
+
1437
+ # set the HTTP header `Accept`
1438
+ _header_params["Accept"] = self.api_client.select_header_accept(
1439
+ ["application/json"]
1440
+ ) # noqa: E501
1441
+
1442
+ # authentication setting
1443
+ _auth_settings = ["ApiKey"] # noqa: E501
1444
+
1445
+ _response_types_map = {
1446
+ "200": "V1AgentStateResponse",
1447
+ "204": "object",
1448
+ "403": "object",
1449
+ "404": "object",
1450
+ }
1451
+
1452
+ return self.api_client.call_api(
1453
+ "/api/v1/orgs/{owner}/agents/{uuid}/state",
1454
+ "GET",
1455
+ _path_params,
1456
+ _query_params,
1457
+ _header_params,
1458
+ body=_body_params,
1459
+ post_params=_form_params,
1460
+ files=_files,
1461
+ response_types_map=_response_types_map,
1462
+ auth_settings=_auth_settings,
1463
+ async_req=_params.get("async_req"),
1464
+ _return_http_data_only=_params.get("_return_http_data_only"), # noqa: E501
1465
+ _preload_content=_params.get("_preload_content", True),
1466
+ _request_timeout=_params.get("_request_timeout"),
1467
+ collection_formats=_collection_formats,
1468
+ _request_auth=_params.get("_request_auth"),
1469
+ )
1470
+
1471
+ @validate_arguments
1472
+ def get_agent_statuses(
1473
+ self,
1474
+ owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
1475
+ uuid: Annotated[
1476
+ StrictStr, Field(..., description="Uuid identifier of the sub-entity")
1477
+ ],
1478
+ entity: Annotated[
1479
+ Optional[StrictStr],
1480
+ Field(description="Entity: project name, hub name, registry name, ..."),
1481
+ ] = None,
1482
+ **kwargs
1483
+ ) -> V1Status: # noqa: E501
1484
+ """Get Agent statuses # noqa: E501
1485
+
1486
+ This method makes a synchronous HTTP request by default. To make an
1487
+ asynchronous HTTP request, please pass async_req=True
1488
+
1489
+ >>> thread = api.get_agent_statuses(owner, uuid, entity, async_req=True)
1490
+ >>> result = thread.get()
1491
+
1492
+ :param owner: Owner of the namespace (required)
1493
+ :type owner: str
1494
+ :param uuid: Uuid identifier of the sub-entity (required)
1495
+ :type uuid: str
1496
+ :param entity: Entity: project name, hub name, registry name, ...
1497
+ :type entity: str
1498
+ :param async_req: Whether to execute the request asynchronously.
1499
+ :type async_req: bool, optional
1500
+ :param _preload_content: if False, the urllib3.HTTPResponse object will
1501
+ be returned without reading/decoding response
1502
+ data. Default is True.
1503
+ :type _preload_content: bool, optional
1504
+ :param _request_timeout: timeout setting for this request. If one
1505
+ number provided, it will be total request
1506
+ timeout. It can also be a pair (tuple) of
1507
+ (connection, read) timeouts.
1508
+ :return: Returns the result object.
1509
+ If the method is called asynchronously,
1510
+ returns the request thread.
1511
+ :rtype: V1Status
1512
+ """
1513
+ kwargs["_return_http_data_only"] = True
1514
+ return self.get_agent_statuses_with_http_info(
1515
+ owner, uuid, entity, **kwargs
1516
+ ) # noqa: E501
1517
+
1518
+ @validate_arguments
1519
+ def get_agent_statuses_with_http_info(
1520
+ self,
1521
+ owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
1522
+ uuid: Annotated[
1523
+ StrictStr, Field(..., description="Uuid identifier of the sub-entity")
1524
+ ],
1525
+ entity: Annotated[
1526
+ Optional[StrictStr],
1527
+ Field(description="Entity: project name, hub name, registry name, ..."),
1528
+ ] = None,
1529
+ **kwargs
1530
+ ): # noqa: E501
1531
+ """Get Agent statuses # noqa: E501
1532
+
1533
+ This method makes a synchronous HTTP request by default. To make an
1534
+ asynchronous HTTP request, please pass async_req=True
1535
+
1536
+ >>> thread = api.get_agent_statuses_with_http_info(owner, uuid, entity, async_req=True)
1537
+ >>> result = thread.get()
1538
+
1539
+ :param owner: Owner of the namespace (required)
1540
+ :type owner: str
1541
+ :param uuid: Uuid identifier of the sub-entity (required)
1542
+ :type uuid: str
1543
+ :param entity: Entity: project name, hub name, registry name, ...
1544
+ :type entity: str
1545
+ :param async_req: Whether to execute the request asynchronously.
1546
+ :type async_req: bool, optional
1547
+ :param _return_http_data_only: response data without head status code
1548
+ and headers
1549
+ :type _return_http_data_only: bool, optional
1550
+ :param _preload_content: if False, the urllib3.HTTPResponse object will
1551
+ be returned without reading/decoding response
1552
+ data. Default is True.
1553
+ :type _preload_content: bool, optional
1554
+ :param _request_timeout: timeout setting for this request. If one
1555
+ number provided, it will be total request
1556
+ timeout. It can also be a pair (tuple) of
1557
+ (connection, read) timeouts.
1558
+ :param _request_auth: set to override the auth_settings for an a single
1559
+ request; this effectively ignores the authentication
1560
+ in the spec for a single request.
1561
+ :type _request_auth: dict, optional
1562
+ :type _content_type: string, optional: force content-type for the request
1563
+ :return: Returns the result object.
1564
+ If the method is called asynchronously,
1565
+ returns the request thread.
1566
+ :rtype: tuple(V1Status, status_code(int), headers(HTTPHeaderDict))
1567
+ """
1568
+
1569
+ _params = locals()
1570
+
1571
+ _all_params = ["owner", "uuid", "entity"]
1572
+ _all_params.extend(
1573
+ [
1574
+ "async_req",
1575
+ "_return_http_data_only",
1576
+ "_preload_content",
1577
+ "_request_timeout",
1578
+ "_request_auth",
1579
+ "_content_type",
1580
+ "_headers",
1581
+ ]
1582
+ )
1583
+
1584
+ # validate the arguments
1585
+ for _key, _val in _params["kwargs"].items():
1586
+ if _key not in _all_params:
1587
+ raise ApiTypeError(
1588
+ "Got an unexpected keyword argument '%s'"
1589
+ " to method get_agent_statuses" % _key
1590
+ )
1591
+ _params[_key] = _val
1592
+ del _params["kwargs"]
1593
+
1594
+ _collection_formats = {}
1595
+
1596
+ # process the path parameters
1597
+ _path_params = {}
1598
+ if _params["owner"]:
1599
+ _path_params["owner"] = _params["owner"]
1600
+
1601
+ if _params["uuid"]:
1602
+ _path_params["uuid"] = _params["uuid"]
1603
+
1604
+ # process the query parameters
1605
+ _query_params = []
1606
+ if _params.get("entity") is not None: # noqa: E501
1607
+ _query_params.append(("entity", _params["entity"]))
1608
+
1609
+ # process the header parameters
1610
+ _header_params = dict(_params.get("_headers", {}))
1611
+ # process the form parameters
1612
+ _form_params = []
1613
+ _files = {}
1614
+ # process the body parameter
1615
+ _body_params = None
1616
+ # set the HTTP header `Accept`
1617
+ _header_params["Accept"] = self.api_client.select_header_accept(
1618
+ ["application/json"]
1619
+ ) # noqa: E501
1620
+
1621
+ # authentication setting
1622
+ _auth_settings = ["ApiKey"] # noqa: E501
1623
+
1624
+ _response_types_map = {
1625
+ "200": "V1Status",
1626
+ "204": "object",
1627
+ "403": "object",
1628
+ "404": "object",
1629
+ }
1630
+
1631
+ return self.api_client.call_api(
1632
+ "/api/v1/orgs/{owner}/agents/{uuid}/statuses",
1633
+ "GET",
1634
+ _path_params,
1635
+ _query_params,
1636
+ _header_params,
1637
+ body=_body_params,
1638
+ post_params=_form_params,
1639
+ files=_files,
1640
+ response_types_map=_response_types_map,
1641
+ auth_settings=_auth_settings,
1642
+ async_req=_params.get("async_req"),
1643
+ _return_http_data_only=_params.get("_return_http_data_only"), # noqa: E501
1644
+ _preload_content=_params.get("_preload_content", True),
1645
+ _request_timeout=_params.get("_request_timeout"),
1646
+ collection_formats=_collection_formats,
1647
+ _request_auth=_params.get("_request_auth"),
1648
+ )
1649
+
1650
+ @validate_arguments
1651
+ def get_agent_token(
1652
+ self,
1653
+ owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
1654
+ uuid: Annotated[
1655
+ StrictStr, Field(..., description="Uuid identifier of the entity")
1656
+ ],
1657
+ **kwargs
1658
+ ) -> V1Token: # noqa: E501
1659
+ """Get agent token # noqa: E501
1660
+
1661
+ This method makes a synchronous HTTP request by default. To make an
1662
+ asynchronous HTTP request, please pass async_req=True
1663
+
1664
+ >>> thread = api.get_agent_token(owner, uuid, async_req=True)
1665
+ >>> result = thread.get()
1666
+
1667
+ :param owner: Owner of the namespace (required)
1668
+ :type owner: str
1669
+ :param uuid: Uuid identifier of the entity (required)
1670
+ :type uuid: str
1671
+ :param async_req: Whether to execute the request asynchronously.
1672
+ :type async_req: bool, optional
1673
+ :param _preload_content: if False, the urllib3.HTTPResponse object will
1674
+ be returned without reading/decoding response
1675
+ data. Default is True.
1676
+ :type _preload_content: bool, optional
1677
+ :param _request_timeout: timeout setting for this request. If one
1678
+ number provided, it will be total request
1679
+ timeout. It can also be a pair (tuple) of
1680
+ (connection, read) timeouts.
1681
+ :return: Returns the result object.
1682
+ If the method is called asynchronously,
1683
+ returns the request thread.
1684
+ :rtype: V1Token
1685
+ """
1686
+ kwargs["_return_http_data_only"] = True
1687
+ return self.get_agent_token_with_http_info(owner, uuid, **kwargs) # noqa: E501
1688
+
1689
+ @validate_arguments
1690
+ def get_agent_token_with_http_info(
1691
+ self,
1692
+ owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
1693
+ uuid: Annotated[
1694
+ StrictStr, Field(..., description="Uuid identifier of the entity")
1695
+ ],
1696
+ **kwargs
1697
+ ): # noqa: E501
1698
+ """Get agent token # noqa: E501
1699
+
1700
+ This method makes a synchronous HTTP request by default. To make an
1701
+ asynchronous HTTP request, please pass async_req=True
1702
+
1703
+ >>> thread = api.get_agent_token_with_http_info(owner, uuid, async_req=True)
1704
+ >>> result = thread.get()
1705
+
1706
+ :param owner: Owner of the namespace (required)
1707
+ :type owner: str
1708
+ :param uuid: Uuid identifier of the entity (required)
1709
+ :type uuid: str
1710
+ :param async_req: Whether to execute the request asynchronously.
1711
+ :type async_req: bool, optional
1712
+ :param _return_http_data_only: response data without head status code
1713
+ and headers
1714
+ :type _return_http_data_only: bool, optional
1715
+ :param _preload_content: if False, the urllib3.HTTPResponse object will
1716
+ be returned without reading/decoding response
1717
+ data. Default is True.
1718
+ :type _preload_content: bool, optional
1719
+ :param _request_timeout: timeout setting for this request. If one
1720
+ number provided, it will be total request
1721
+ timeout. It can also be a pair (tuple) of
1722
+ (connection, read) timeouts.
1723
+ :param _request_auth: set to override the auth_settings for an a single
1724
+ request; this effectively ignores the authentication
1725
+ in the spec for a single request.
1726
+ :type _request_auth: dict, optional
1727
+ :type _content_type: string, optional: force content-type for the request
1728
+ :return: Returns the result object.
1729
+ If the method is called asynchronously,
1730
+ returns the request thread.
1731
+ :rtype: tuple(V1Token, status_code(int), headers(HTTPHeaderDict))
1732
+ """
1733
+
1734
+ _params = locals()
1735
+
1736
+ _all_params = ["owner", "uuid"]
1737
+ _all_params.extend(
1738
+ [
1739
+ "async_req",
1740
+ "_return_http_data_only",
1741
+ "_preload_content",
1742
+ "_request_timeout",
1743
+ "_request_auth",
1744
+ "_content_type",
1745
+ "_headers",
1746
+ ]
1747
+ )
1748
+
1749
+ # validate the arguments
1750
+ for _key, _val in _params["kwargs"].items():
1751
+ if _key not in _all_params:
1752
+ raise ApiTypeError(
1753
+ "Got an unexpected keyword argument '%s'"
1754
+ " to method get_agent_token" % _key
1755
+ )
1756
+ _params[_key] = _val
1757
+ del _params["kwargs"]
1758
+
1759
+ _collection_formats = {}
1760
+
1761
+ # process the path parameters
1762
+ _path_params = {}
1763
+ if _params["owner"]:
1764
+ _path_params["owner"] = _params["owner"]
1765
+ if _params["uuid"]:
1766
+ _path_params["uuid"] = _params["uuid"]
1767
+
1768
+ # process the query parameters
1769
+ _query_params = []
1770
+
1771
+ # process the header parameters
1772
+ _header_params = dict(_params.get("_headers", {}))
1773
+
1774
+ # process the form parameters
1775
+ _form_params = []
1776
+ _files = {}
1777
+
1778
+ # process the body parameter
1779
+ _body_params = None
1780
+
1781
+ # set the HTTP header `Accept`
1782
+ _header_params["Accept"] = self.api_client.select_header_accept(
1783
+ ["application/json"]
1784
+ ) # noqa: E501
1785
+
1786
+ # authentication setting
1787
+ _auth_settings = ["ApiKey"] # noqa: E501
1788
+
1789
+ _response_types_map = {
1790
+ "200": "V1Token",
1791
+ "204": "object",
1792
+ "403": "object",
1793
+ "404": "object",
1794
+ }
1795
+
1796
+ return self.api_client.call_api(
1797
+ "/api/v1/orgs/{owner}/agents/{uuid}/token",
1798
+ "GET",
1799
+ _path_params,
1800
+ _query_params,
1801
+ _header_params,
1802
+ body=_body_params,
1803
+ post_params=_form_params,
1804
+ files=_files,
1805
+ response_types_map=_response_types_map,
1806
+ auth_settings=_auth_settings,
1807
+ async_req=_params.get("async_req"),
1808
+ _return_http_data_only=_params.get("_return_http_data_only"), # noqa: E501
1809
+ _preload_content=_params.get("_preload_content", True),
1810
+ _request_timeout=_params.get("_request_timeout"),
1811
+ collection_formats=_collection_formats,
1812
+ _request_auth=_params.get("_request_auth"),
1813
+ )
1814
+
1815
+ @validate_arguments
1816
+ def get_global_state(
1817
+ self,
1818
+ owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
1819
+ **kwargs
1820
+ ) -> V1AgentStateResponse: # noqa: E501
1821
+ """Get Global State (queues/runs) # noqa: E501
1822
+
1823
+ This method makes a synchronous HTTP request by default. To make an
1824
+ asynchronous HTTP request, please pass async_req=True
1825
+
1826
+ >>> thread = api.get_global_state(owner, async_req=True)
1827
+ >>> result = thread.get()
1828
+
1829
+ :param owner: Owner of the namespace (required)
1830
+ :type owner: str
1831
+ :param async_req: Whether to execute the request asynchronously.
1832
+ :type async_req: bool, optional
1833
+ :param _preload_content: if False, the urllib3.HTTPResponse object will
1834
+ be returned without reading/decoding response
1835
+ data. Default is True.
1836
+ :type _preload_content: bool, optional
1837
+ :param _request_timeout: timeout setting for this request. If one
1838
+ number provided, it will be total request
1839
+ timeout. It can also be a pair (tuple) of
1840
+ (connection, read) timeouts.
1841
+ :return: Returns the result object.
1842
+ If the method is called asynchronously,
1843
+ returns the request thread.
1844
+ :rtype: V1AgentStateResponse
1107
1845
  """
1108
1846
  kwargs["_return_http_data_only"] = True
1109
- return self.get_agent_token_with_http_info(owner, uuid, **kwargs) # noqa: E501
1847
+ return self.get_global_state_with_http_info(owner, **kwargs) # noqa: E501
1110
1848
 
1111
1849
  @validate_arguments
1112
- def get_agent_token_with_http_info(
1850
+ def get_global_state_with_http_info(
1113
1851
  self,
1114
1852
  owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
1115
- uuid: Annotated[
1116
- StrictStr, Field(..., description="Uuid identifier of the entity")
1117
- ],
1118
1853
  **kwargs
1119
1854
  ): # noqa: E501
1120
- """Get agent token # noqa: E501
1855
+ """Get Global State (queues/runs) # noqa: E501
1121
1856
 
1122
1857
  This method makes a synchronous HTTP request by default. To make an
1123
1858
  asynchronous HTTP request, please pass async_req=True
1124
1859
 
1125
- >>> thread = api.get_agent_token_with_http_info(owner, uuid, async_req=True)
1860
+ >>> thread = api.get_global_state_with_http_info(owner, async_req=True)
1126
1861
  >>> result = thread.get()
1127
1862
 
1128
1863
  :param owner: Owner of the namespace (required)
1129
1864
  :type owner: str
1130
- :param uuid: Uuid identifier of the entity (required)
1131
- :type uuid: str
1132
1865
  :param async_req: Whether to execute the request asynchronously.
1133
1866
  :type async_req: bool, optional
1134
1867
  :param _return_http_data_only: response data without head status code
@@ -1150,12 +1883,12 @@ class AgentsV1Api(BaseApi):
1150
1883
  :return: Returns the result object.
1151
1884
  If the method is called asynchronously,
1152
1885
  returns the request thread.
1153
- :rtype: tuple(V1Token, status_code(int), headers(HTTPHeaderDict))
1886
+ :rtype: tuple(V1AgentStateResponse, status_code(int), headers(HTTPHeaderDict))
1154
1887
  """
1155
1888
 
1156
1889
  _params = locals()
1157
1890
 
1158
- _all_params = ["owner", "uuid"]
1891
+ _all_params = ["owner"]
1159
1892
  _all_params.extend(
1160
1893
  [
1161
1894
  "async_req",
@@ -1173,7 +1906,7 @@ class AgentsV1Api(BaseApi):
1173
1906
  if _key not in _all_params:
1174
1907
  raise ApiTypeError(
1175
1908
  "Got an unexpected keyword argument '%s'"
1176
- " to method get_agent_token" % _key
1909
+ " to method get_global_state" % _key
1177
1910
  )
1178
1911
  _params[_key] = _val
1179
1912
  del _params["kwargs"]
@@ -1184,22 +1917,16 @@ class AgentsV1Api(BaseApi):
1184
1917
  _path_params = {}
1185
1918
  if _params["owner"]:
1186
1919
  _path_params["owner"] = _params["owner"]
1187
- if _params["uuid"]:
1188
- _path_params["uuid"] = _params["uuid"]
1189
1920
 
1190
1921
  # process the query parameters
1191
1922
  _query_params = []
1192
-
1193
1923
  # process the header parameters
1194
1924
  _header_params = dict(_params.get("_headers", {}))
1195
-
1196
1925
  # process the form parameters
1197
1926
  _form_params = []
1198
1927
  _files = {}
1199
-
1200
1928
  # process the body parameter
1201
1929
  _body_params = None
1202
-
1203
1930
  # set the HTTP header `Accept`
1204
1931
  _header_params["Accept"] = self.api_client.select_header_accept(
1205
1932
  ["application/json"]
@@ -1209,14 +1936,14 @@ class AgentsV1Api(BaseApi):
1209
1936
  _auth_settings = ["ApiKey"] # noqa: E501
1210
1937
 
1211
1938
  _response_types_map = {
1212
- "200": "V1Token",
1939
+ "200": "V1AgentStateResponse",
1213
1940
  "204": "object",
1214
1941
  "403": "object",
1215
1942
  "404": "object",
1216
1943
  }
1217
1944
 
1218
1945
  return self.api_client.call_api(
1219
- "/api/v1/orgs/{owner}/agents/{uuid}/token",
1946
+ "/api/v1/orgs/{owner}/agents/state",
1220
1947
  "GET",
1221
1948
  _path_params,
1222
1949
  _query_params,
@@ -1235,21 +1962,47 @@ class AgentsV1Api(BaseApi):
1235
1962
  )
1236
1963
 
1237
1964
  @validate_arguments
1238
- def get_global_state(
1965
+ def inspect_agent(
1239
1966
  self,
1967
+ namespace: Annotated[StrictStr, Field(..., description="namespace")],
1240
1968
  owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
1969
+ uuid: Annotated[
1970
+ StrictStr, Field(..., description="Uuid identifier of the entity")
1971
+ ],
1972
+ service: Annotated[Optional[StrictStr], Field(description="Service.")] = None,
1973
+ last_file: Annotated[
1974
+ Optional[StrictStr], Field(description="last_file.")
1975
+ ] = None,
1976
+ force: Annotated[
1977
+ Optional[bool], Field(description="Force query param.")
1978
+ ] = None,
1979
+ connection: Annotated[
1980
+ Optional[StrictStr], Field(description="Connection to use.")
1981
+ ] = None,
1241
1982
  **kwargs
1242
- ) -> V1AgentStateResponse: # noqa: E501
1243
- """Get Global State (queues/runs) # noqa: E501
1983
+ ) -> object: # noqa: E501
1984
+ """Inspect an agent's service full conditions # noqa: E501
1244
1985
 
1245
1986
  This method makes a synchronous HTTP request by default. To make an
1246
1987
  asynchronous HTTP request, please pass async_req=True
1247
1988
 
1248
- >>> thread = api.get_global_state(owner, async_req=True)
1989
+ >>> thread = api.inspect_agent(namespace, owner, uuid, service, last_file, force, connection, async_req=True)
1249
1990
  >>> result = thread.get()
1250
1991
 
1992
+ :param namespace: namespace (required)
1993
+ :type namespace: str
1251
1994
  :param owner: Owner of the namespace (required)
1252
1995
  :type owner: str
1996
+ :param uuid: Uuid identifier of the entity (required)
1997
+ :type uuid: str
1998
+ :param service: Service.
1999
+ :type service: str
2000
+ :param last_file: last_file.
2001
+ :type last_file: str
2002
+ :param force: Force query param.
2003
+ :type force: bool
2004
+ :param connection: Connection to use.
2005
+ :type connection: str
1253
2006
  :param async_req: Whether to execute the request asynchronously.
1254
2007
  :type async_req: bool, optional
1255
2008
  :param _preload_content: if False, the urllib3.HTTPResponse object will
@@ -1263,27 +2016,55 @@ class AgentsV1Api(BaseApi):
1263
2016
  :return: Returns the result object.
1264
2017
  If the method is called asynchronously,
1265
2018
  returns the request thread.
1266
- :rtype: V1AgentStateResponse
2019
+ :rtype: object
1267
2020
  """
1268
2021
  kwargs["_return_http_data_only"] = True
1269
- return self.get_global_state_with_http_info(owner, **kwargs) # noqa: E501
2022
+ return self.inspect_agent_with_http_info(
2023
+ namespace, owner, uuid, service, last_file, force, connection, **kwargs
2024
+ ) # noqa: E501
1270
2025
 
1271
2026
  @validate_arguments
1272
- def get_global_state_with_http_info(
2027
+ def inspect_agent_with_http_info(
1273
2028
  self,
2029
+ namespace: Annotated[StrictStr, Field(..., description="namespace")],
1274
2030
  owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
2031
+ uuid: Annotated[
2032
+ StrictStr, Field(..., description="Uuid identifier of the entity")
2033
+ ],
2034
+ service: Annotated[Optional[StrictStr], Field(description="Service.")] = None,
2035
+ last_file: Annotated[
2036
+ Optional[StrictStr], Field(description="last_file.")
2037
+ ] = None,
2038
+ force: Annotated[
2039
+ Optional[bool], Field(description="Force query param.")
2040
+ ] = None,
2041
+ connection: Annotated[
2042
+ Optional[StrictStr], Field(description="Connection to use.")
2043
+ ] = None,
1275
2044
  **kwargs
1276
2045
  ): # noqa: E501
1277
- """Get Global State (queues/runs) # noqa: E501
2046
+ """Inspect an agent's service full conditions # noqa: E501
1278
2047
 
1279
2048
  This method makes a synchronous HTTP request by default. To make an
1280
2049
  asynchronous HTTP request, please pass async_req=True
1281
2050
 
1282
- >>> thread = api.get_global_state_with_http_info(owner, async_req=True)
2051
+ >>> thread = api.inspect_agent_with_http_info(namespace, owner, uuid, service, last_file, force, connection, async_req=True)
1283
2052
  >>> result = thread.get()
1284
2053
 
2054
+ :param namespace: namespace (required)
2055
+ :type namespace: str
1285
2056
  :param owner: Owner of the namespace (required)
1286
2057
  :type owner: str
2058
+ :param uuid: Uuid identifier of the entity (required)
2059
+ :type uuid: str
2060
+ :param service: Service.
2061
+ :type service: str
2062
+ :param last_file: last_file.
2063
+ :type last_file: str
2064
+ :param force: Force query param.
2065
+ :type force: bool
2066
+ :param connection: Connection to use.
2067
+ :type connection: str
1287
2068
  :param async_req: Whether to execute the request asynchronously.
1288
2069
  :type async_req: bool, optional
1289
2070
  :param _return_http_data_only: response data without head status code
@@ -1305,12 +2086,20 @@ class AgentsV1Api(BaseApi):
1305
2086
  :return: Returns the result object.
1306
2087
  If the method is called asynchronously,
1307
2088
  returns the request thread.
1308
- :rtype: tuple(V1AgentStateResponse, status_code(int), headers(HTTPHeaderDict))
2089
+ :rtype: tuple(object, status_code(int), headers(HTTPHeaderDict))
1309
2090
  """
1310
2091
 
1311
2092
  _params = locals()
1312
2093
 
1313
- _all_params = ["owner"]
2094
+ _all_params = [
2095
+ "namespace",
2096
+ "owner",
2097
+ "uuid",
2098
+ "service",
2099
+ "last_file",
2100
+ "force",
2101
+ "connection",
2102
+ ]
1314
2103
  _all_params.extend(
1315
2104
  [
1316
2105
  "async_req",
@@ -1328,7 +2117,7 @@ class AgentsV1Api(BaseApi):
1328
2117
  if _key not in _all_params:
1329
2118
  raise ApiTypeError(
1330
2119
  "Got an unexpected keyword argument '%s'"
1331
- " to method get_global_state" % _key
2120
+ " to method inspect_agent" % _key
1332
2121
  )
1333
2122
  _params[_key] = _val
1334
2123
  del _params["kwargs"]
@@ -1337,11 +2126,24 @@ class AgentsV1Api(BaseApi):
1337
2126
 
1338
2127
  # process the path parameters
1339
2128
  _path_params = {}
2129
+ if _params["namespace"]:
2130
+ _path_params["namespace"] = _params["namespace"]
1340
2131
  if _params["owner"]:
1341
2132
  _path_params["owner"] = _params["owner"]
2133
+ if _params["uuid"]:
2134
+ _path_params["uuid"] = _params["uuid"]
1342
2135
 
1343
2136
  # process the query parameters
1344
2137
  _query_params = []
2138
+ if _params.get("service") is not None: # noqa: E501
2139
+ _query_params.append(("service", _params["service"]))
2140
+ if _params.get("last_file") is not None: # noqa: E501
2141
+ _query_params.append(("last_file", _params["last_file"]))
2142
+ if _params.get("force") is not None: # noqa: E501
2143
+ _query_params.append(("force", _params["force"]))
2144
+ if _params.get("connection") is not None: # noqa: E501
2145
+ _query_params.append(("connection", _params["connection"]))
2146
+
1345
2147
  # process the header parameters
1346
2148
  _header_params = dict(_params.get("_headers", {}))
1347
2149
  # process the form parameters
@@ -1358,14 +2160,14 @@ class AgentsV1Api(BaseApi):
1358
2160
  _auth_settings = ["ApiKey"] # noqa: E501
1359
2161
 
1360
2162
  _response_types_map = {
1361
- "200": "V1AgentStateResponse",
2163
+ "200": "object",
1362
2164
  "204": "object",
1363
2165
  "403": "object",
1364
2166
  "404": "object",
1365
2167
  }
1366
2168
 
1367
2169
  return self.api_client.call_api(
1368
- "/api/v1/orgs/{owner}/agents/state",
2170
+ "/streams/v1/{namespace}/{owner}/agents/{uuid}/k8s_inspect",
1369
2171
  "GET",
1370
2172
  _path_params,
1371
2173
  _query_params,
@@ -2046,7 +2848,7 @@ class AgentsV1Api(BaseApi):
2046
2848
  def patch_agent_token(
2047
2849
  self,
2048
2850
  owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
2049
- entity: Annotated[StrictStr, Field(..., description="Rntity")],
2851
+ entity: Annotated[StrictStr, Field(..., description="Entity")],
2050
2852
  body: Annotated[V1Token, Field(..., description="Token body")],
2051
2853
  **kwargs
2052
2854
  ) -> V1Token: # noqa: E501
@@ -2060,7 +2862,7 @@ class AgentsV1Api(BaseApi):
2060
2862
 
2061
2863
  :param owner: Owner of the namespace (required)
2062
2864
  :type owner: str
2063
- :param entity: Rntity (required)
2865
+ :param entity: Entity (required)
2064
2866
  :type entity: str
2065
2867
  :param body: Token body (required)
2066
2868
  :type body: V1Token
@@ -2088,7 +2890,7 @@ class AgentsV1Api(BaseApi):
2088
2890
  def patch_agent_token_with_http_info(
2089
2891
  self,
2090
2892
  owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
2091
- entity: Annotated[StrictStr, Field(..., description="Rntity")],
2893
+ entity: Annotated[StrictStr, Field(..., description="Entity")],
2092
2894
  body: Annotated[V1Token, Field(..., description="Token body")],
2093
2895
  **kwargs
2094
2896
  ): # noqa: E501
@@ -2102,7 +2904,7 @@ class AgentsV1Api(BaseApi):
2102
2904
 
2103
2905
  :param owner: Owner of the namespace (required)
2104
2906
  :type owner: str
2105
- :param entity: Rntity (required)
2907
+ :param entity: Entity (required)
2106
2908
  :type entity: str
2107
2909
  :param body: Token body (required)
2108
2910
  :type body: V1Token
@@ -2221,6 +3023,186 @@ class AgentsV1Api(BaseApi):
2221
3023
  _request_auth=_params.get("_request_auth"),
2222
3024
  )
2223
3025
 
3026
+ @validate_arguments
3027
+ def reconcile_agent(
3028
+ self,
3029
+ owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
3030
+ uuid: Annotated[
3031
+ StrictStr, Field(..., description="Uuid identifier of the entity")
3032
+ ],
3033
+ body: V1AgentReconcileBodyRequest,
3034
+ **kwargs
3035
+ ) -> object: # noqa: E501
3036
+ """Reconcile agent # noqa: E501
3037
+
3038
+ This method makes a synchronous HTTP request by default. To make an
3039
+ asynchronous HTTP request, please pass async_req=True
3040
+
3041
+ >>> thread = api.reconcile_agent(owner, uuid, body, async_req=True)
3042
+ >>> result = thread.get()
3043
+
3044
+ :param owner: Owner of the namespace (required)
3045
+ :type owner: str
3046
+ :param uuid: Uuid identifier of the entity (required)
3047
+ :type uuid: str
3048
+ :param body: (required)
3049
+ :type body: V1AgentReconcileBodyRequest
3050
+ :param async_req: Whether to execute the request asynchronously.
3051
+ :type async_req: bool, optional
3052
+ :param _preload_content: if False, the urllib3.HTTPResponse object will
3053
+ be returned without reading/decoding response
3054
+ data. Default is True.
3055
+ :type _preload_content: bool, optional
3056
+ :param _request_timeout: timeout setting for this request. If one
3057
+ number provided, it will be total request
3058
+ timeout. It can also be a pair (tuple) of
3059
+ (connection, read) timeouts.
3060
+ :return: Returns the result object.
3061
+ If the method is called asynchronously,
3062
+ returns the request thread.
3063
+ :rtype: object
3064
+ """
3065
+ kwargs["_return_http_data_only"] = True
3066
+ return self.reconcile_agent_with_http_info(
3067
+ owner, uuid, body, **kwargs
3068
+ ) # noqa: E501
3069
+
3070
+ @validate_arguments
3071
+ def reconcile_agent_with_http_info(
3072
+ self,
3073
+ owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
3074
+ uuid: Annotated[
3075
+ StrictStr, Field(..., description="Uuid identifier of the entity")
3076
+ ],
3077
+ body: V1AgentReconcileBodyRequest,
3078
+ **kwargs
3079
+ ): # noqa: E501
3080
+ """Reconcile agent # noqa: E501
3081
+
3082
+ This method makes a synchronous HTTP request by default. To make an
3083
+ asynchronous HTTP request, please pass async_req=True
3084
+
3085
+ >>> thread = api.reconcile_agent_with_http_info(owner, uuid, body, async_req=True)
3086
+ >>> result = thread.get()
3087
+
3088
+ :param owner: Owner of the namespace (required)
3089
+ :type owner: str
3090
+ :param uuid: Uuid identifier of the entity (required)
3091
+ :type uuid: str
3092
+ :param body: (required)
3093
+ :type body: V1AgentReconcileBodyRequest
3094
+ :param async_req: Whether to execute the request asynchronously.
3095
+ :type async_req: bool, optional
3096
+ :param _return_http_data_only: response data without head status code
3097
+ and headers
3098
+ :type _return_http_data_only: bool, optional
3099
+ :param _preload_content: if False, the urllib3.HTTPResponse object will
3100
+ be returned without reading/decoding response
3101
+ data. Default is True.
3102
+ :type _preload_content: bool, optional
3103
+ :param _request_timeout: timeout setting for this request. If one
3104
+ number provided, it will be total request
3105
+ timeout. It can also be a pair (tuple) of
3106
+ (connection, read) timeouts.
3107
+ :param _request_auth: set to override the auth_settings for an a single
3108
+ request; this effectively ignores the authentication
3109
+ in the spec for a single request.
3110
+ :type _request_auth: dict, optional
3111
+ :type _content_type: string, optional: force content-type for the request
3112
+ :return: Returns the result object.
3113
+ If the method is called asynchronously,
3114
+ returns the request thread.
3115
+ :rtype: None
3116
+ """
3117
+
3118
+ _params = locals()
3119
+
3120
+ _all_params = ["owner", "uuid", "body"]
3121
+ _all_params.extend(
3122
+ [
3123
+ "async_req",
3124
+ "_return_http_data_only",
3125
+ "_preload_content",
3126
+ "_request_timeout",
3127
+ "_request_auth",
3128
+ "_content_type",
3129
+ "_headers",
3130
+ ]
3131
+ )
3132
+
3133
+ # validate the arguments
3134
+ for _key, _val in _params["kwargs"].items():
3135
+ if _key not in _all_params:
3136
+ raise ApiTypeError(
3137
+ "Got an unexpected keyword argument '%s'"
3138
+ " to method reconcile_agent" % _key
3139
+ )
3140
+ _params[_key] = _val
3141
+ del _params["kwargs"]
3142
+
3143
+ _collection_formats = {}
3144
+
3145
+ # process the path parameters
3146
+ _path_params = {}
3147
+ if _params["owner"]:
3148
+ _path_params["owner"] = _params["owner"]
3149
+ if _params["uuid"]:
3150
+ _path_params["uuid"] = _params["uuid"]
3151
+
3152
+ # process the query parameters
3153
+ _query_params = []
3154
+ # process the header parameters
3155
+ _header_params = dict(_params.get("_headers", {}))
3156
+ # process the form parameters
3157
+ _form_params = []
3158
+ _files = {}
3159
+ # process the body parameter
3160
+ _body_params = None
3161
+ if _params["body"]:
3162
+ _body_params = _params["body"]
3163
+
3164
+ # set the HTTP header `Accept`
3165
+ _header_params["Accept"] = self.api_client.select_header_accept(
3166
+ ["application/json"]
3167
+ ) # noqa: E501
3168
+
3169
+ # set the HTTP header `Content-Type`
3170
+ _content_types_list = _params.get(
3171
+ "_content_type",
3172
+ self.api_client.select_header_content_type(["application/json"]),
3173
+ )
3174
+ if _content_types_list:
3175
+ _header_params["Content-Type"] = _content_types_list
3176
+
3177
+ # authentication setting
3178
+ _auth_settings = ["ApiKey"] # noqa: E501
3179
+
3180
+ _response_types_map = {
3181
+ "200": "V1Agent",
3182
+ "204": "object",
3183
+ "403": "object",
3184
+ "404": "object",
3185
+ }
3186
+
3187
+ return self.api_client.call_api(
3188
+ "/api/v1/orgs/{owner}/agents/{uuid}/reconcile",
3189
+ "PATCH",
3190
+ _path_params,
3191
+ _query_params,
3192
+ _header_params,
3193
+ body=_body_params,
3194
+ post_params=_form_params,
3195
+ files=_files,
3196
+ response_types_map=_response_types_map,
3197
+ auth_settings=_auth_settings,
3198
+ async_req=_params.get("async_req"),
3199
+ _return_http_data_only=_params.get("_return_http_data_only"), # noqa: E501
3200
+ _preload_content=_params.get("_preload_content", True),
3201
+ _request_timeout=_params.get("_request_timeout"),
3202
+ collection_formats=_collection_formats,
3203
+ _request_auth=_params.get("_request_auth"),
3204
+ )
3205
+
2224
3206
  @validate_arguments
2225
3207
  def sync_agent(
2226
3208
  self,
@@ -2399,18 +3381,21 @@ class AgentsV1Api(BaseApi):
2399
3381
  def cron_agent(
2400
3382
  self,
2401
3383
  owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
3384
+ body: Annotated[Optional[Dict], Field(..., description="Cron body")] = None,
2402
3385
  **kwargs
2403
- ) -> None: # noqa: E501
3386
+ ) -> V1AgentStateResponse: # noqa: E501
2404
3387
  """Global Cron agent # noqa: E501
2405
3388
 
2406
3389
  This method makes a synchronous HTTP request by default. To make an
2407
3390
  asynchronous HTTP request, please pass async_req=True
2408
3391
 
2409
- >>> thread = api.cron_agent(owner, async_req=True)
3392
+ >>> thread = api.cron_agent(owner, body, async_req=True)
2410
3393
  >>> result = thread.get()
2411
3394
 
2412
3395
  :param owner: Owner of the namespace (required)
2413
3396
  :type owner: str
3397
+ :param body: Cron body (required)
3398
+ :type body: object
2414
3399
  :param async_req: Whether to execute the request asynchronously.
2415
3400
  :type async_req: bool, optional
2416
3401
  :param _preload_content: if False, the urllib3.HTTPResponse object will
@@ -2424,15 +3409,16 @@ class AgentsV1Api(BaseApi):
2424
3409
  :return: Returns the result object.
2425
3410
  If the method is called asynchronously,
2426
3411
  returns the request thread.
2427
- :rtype: None
3412
+ :rtype: V1AgentStateResponse
2428
3413
  """
2429
3414
  kwargs["_return_http_data_only"] = True
2430
- return self.cron_agent_with_http_info(owner, **kwargs) # noqa: E501
3415
+ return self.cron_agent_with_http_info(owner, body, **kwargs) # noqa: E501
2431
3416
 
2432
3417
  @validate_arguments
2433
3418
  def cron_agent_with_http_info(
2434
3419
  self,
2435
3420
  owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
3421
+ body: Annotated[Optional[Dict], Field(..., description="Cron body")] = None,
2436
3422
  **kwargs
2437
3423
  ): # noqa: E501
2438
3424
  """Cron agent # noqa: E501
@@ -2440,11 +3426,13 @@ class AgentsV1Api(BaseApi):
2440
3426
  This method makes a synchronous HTTP request by default. To make an
2441
3427
  asynchronous HTTP request, please pass async_req=True
2442
3428
 
2443
- >>> thread = api.cron_agent_with_http_info(owner, async_req=True)
3429
+ >>> thread = api.cron_agent_with_http_info(owner, body, async_req=True)
2444
3430
  >>> result = thread.get()
2445
3431
 
2446
3432
  :param owner: Owner of the namespace (required)
2447
3433
  :type owner: str
3434
+ :param body: Cron body (required)
3435
+ :type body: object
2448
3436
  :param async_req: Whether to execute the request asynchronously.
2449
3437
  :type async_req: bool, optional
2450
3438
  :param _return_http_data_only: response data without head status code
@@ -2466,12 +3454,12 @@ class AgentsV1Api(BaseApi):
2466
3454
  :return: Returns the result object.
2467
3455
  If the method is called asynchronously,
2468
3456
  returns the request thread.
2469
- :rtype: None
3457
+ :rtype: tuple(V1AgentStateResponse, status_code(int), headers(HTTPHeaderDict))
2470
3458
  """
2471
3459
 
2472
3460
  _params = locals()
2473
3461
 
2474
- _all_params = ["owner"]
3462
+ _all_params = ["owner", "body"]
2475
3463
  _all_params.extend(
2476
3464
  [
2477
3465
  "async_req",
@@ -2510,15 +3498,30 @@ class AgentsV1Api(BaseApi):
2510
3498
  _files = {}
2511
3499
  # process the body parameter
2512
3500
  _body_params = None
3501
+ if _params["body"]:
3502
+ _body_params = _params["body"]
2513
3503
  # set the HTTP header `Accept`
2514
3504
  _header_params["Accept"] = self.api_client.select_header_accept(
2515
3505
  ["application/json"]
2516
3506
  ) # noqa: E501
2517
3507
 
3508
+ # set the HTTP header `Content-Type`
3509
+ _content_types_list = _params.get(
3510
+ "_content_type",
3511
+ self.api_client.select_header_content_type(["application/json"]),
3512
+ )
3513
+ if _content_types_list:
3514
+ _header_params["Content-Type"] = _content_types_list
3515
+
2518
3516
  # authentication setting
2519
3517
  _auth_settings = ["ApiKey"] # noqa: E501
2520
3518
 
2521
- _response_types_map = {}
3519
+ _response_types_map = {
3520
+ "200": "V1AgentStateResponse",
3521
+ "204": "object",
3522
+ "403": "object",
3523
+ "404": "object",
3524
+ }
2522
3525
 
2523
3526
  return self.api_client.call_api(
2524
3527
  "/api/v1/orgs/{owner}/agents/cron",
@@ -2901,7 +3904,7 @@ class AgentsV1Api(BaseApi):
2901
3904
  def update_agent_token(
2902
3905
  self,
2903
3906
  owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
2904
- entity: Annotated[StrictStr, Field(..., description="Rntity")],
3907
+ entity: Annotated[StrictStr, Field(..., description="Entity")],
2905
3908
  body: Annotated[V1Token, Field(..., description="Token body")],
2906
3909
  **kwargs
2907
3910
  ) -> V1Token: # noqa: E501
@@ -2915,7 +3918,7 @@ class AgentsV1Api(BaseApi):
2915
3918
 
2916
3919
  :param owner: Owner of the namespace (required)
2917
3920
  :type owner: str
2918
- :param entity: Rntity (required)
3921
+ :param entity: Entity (required)
2919
3922
  :type entity: str
2920
3923
  :param body: Token body (required)
2921
3924
  :type body: V1Token
@@ -2943,7 +3946,7 @@ class AgentsV1Api(BaseApi):
2943
3946
  def update_agent_token_with_http_info(
2944
3947
  self,
2945
3948
  owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
2946
- entity: Annotated[StrictStr, Field(..., description="Rntity")],
3949
+ entity: Annotated[StrictStr, Field(..., description="Entity")],
2947
3950
  body: Annotated[V1Token, Field(..., description="Token body")],
2948
3951
  **kwargs
2949
3952
  ): # noqa: E501
@@ -2957,7 +3960,7 @@ class AgentsV1Api(BaseApi):
2957
3960
 
2958
3961
  :param owner: Owner of the namespace (required)
2959
3962
  :type owner: str
2960
- :param entity: Rntity (required)
3963
+ :param entity: Entity (required)
2961
3964
  :type entity: str
2962
3965
  :param body: Token body (required)
2963
3966
  :type body: V1Token