agenta 0.48.9__py3-none-any.whl → 0.49.0__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.

Potentially problematic release.

This version of agenta might be problematic. Click here for more details.

Files changed (297) hide show
  1. agenta/__init__.py +3 -2
  2. agenta/client/Readme.md +258 -80
  3. agenta/client/__init__.py +205 -29
  4. agenta/client/backend/__init__.py +461 -0
  5. agenta/client/backend/access_control/__init__.py +2 -0
  6. agenta/client/backend/access_control/client.py +53 -73
  7. agenta/client/backend/access_control/raw_client.py +180 -0
  8. agenta/client/backend/admin/__init__.py +2 -0
  9. agenta/client/backend/admin/client.py +473 -268
  10. agenta/client/backend/admin/raw_client.py +1017 -0
  11. agenta/client/backend/api_keys/__init__.py +2 -0
  12. agenta/client/backend/api_keys/client.py +43 -276
  13. agenta/client/backend/api_keys/raw_client.py +364 -0
  14. agenta/client/backend/apps/__init__.py +2 -0
  15. agenta/client/backend/apps/client.py +132 -895
  16. agenta/client/backend/apps/raw_client.py +1516 -0
  17. agenta/client/backend/bases/__init__.py +2 -0
  18. agenta/client/backend/bases/client.py +33 -73
  19. agenta/client/backend/bases/raw_client.py +179 -0
  20. agenta/client/backend/billing/__init__.py +3 -0
  21. agenta/client/backend/billing/client.py +564 -0
  22. agenta/client/backend/billing/raw_client.py +805 -0
  23. agenta/client/backend/client.py +1268 -0
  24. agenta/client/backend/configs/__init__.py +2 -0
  25. agenta/client/backend/configs/client.py +49 -361
  26. agenta/client/backend/configs/raw_client.py +402 -0
  27. agenta/client/backend/containers/__init__.py +1 -3
  28. agenta/client/backend/containers/client.py +25 -548
  29. agenta/client/backend/containers/raw_client.py +112 -0
  30. agenta/client/backend/core/__init__.py +5 -0
  31. agenta/client/backend/core/api_error.py +12 -6
  32. agenta/client/backend/core/client_wrapper.py +4 -4
  33. agenta/client/backend/core/file.py +1 -3
  34. agenta/client/backend/core/force_multipart.py +16 -0
  35. agenta/client/backend/core/http_client.py +78 -34
  36. agenta/client/backend/core/http_response.py +55 -0
  37. agenta/client/backend/core/jsonable_encoder.py +0 -1
  38. agenta/client/backend/core/pydantic_utilities.py +88 -113
  39. agenta/client/backend/core/serialization.py +9 -3
  40. agenta/client/backend/environment.py +7 -0
  41. agenta/client/backend/environments/__init__.py +2 -0
  42. agenta/client/backend/environments/client.py +43 -79
  43. agenta/client/backend/environments/raw_client.py +193 -0
  44. agenta/client/backend/errors/__init__.py +2 -0
  45. agenta/client/backend/errors/unprocessable_entity_error.py +8 -2
  46. agenta/client/backend/evals/__init__.py +3 -0
  47. agenta/client/backend/evals/client.py +1042 -0
  48. agenta/client/backend/evals/raw_client.py +1549 -0
  49. agenta/client/backend/evaluations/__init__.py +2 -0
  50. agenta/client/backend/evaluations/client.py +106 -590
  51. agenta/client/backend/evaluations/raw_client.py +1344 -0
  52. agenta/client/backend/evaluators/__init__.py +2 -0
  53. agenta/client/backend/evaluators/client.py +99 -516
  54. agenta/client/backend/evaluators/raw_client.py +1182 -0
  55. agenta/client/backend/human_evaluations/__init__.py +2 -0
  56. agenta/client/backend/human_evaluations/client.py +120 -680
  57. agenta/client/backend/human_evaluations/raw_client.py +1577 -0
  58. agenta/client/backend/observability/__init__.py +14 -2
  59. agenta/client/backend/observability/client.py +185 -341
  60. agenta/client/backend/observability/raw_client.py +943 -0
  61. agenta/client/backend/observability/types/__init__.py +10 -2
  62. agenta/client/backend/observability/types/{format.py → fetch_trace_by_id_request_trace_id.py} +1 -1
  63. agenta/client/backend/observability/types/fetch_trace_by_id_response.py +15 -0
  64. agenta/client/backend/observability/types/query_analytics_response.py +2 -1
  65. agenta/client/backend/observability/types/query_traces_response.py +7 -3
  66. agenta/client/backend/organization/__init__.py +2 -0
  67. agenta/client/backend/organization/client.py +105 -361
  68. agenta/client/backend/organization/raw_client.py +774 -0
  69. agenta/client/backend/raw_client.py +1432 -0
  70. agenta/client/backend/scopes/__init__.py +2 -0
  71. agenta/client/backend/scopes/client.py +31 -45
  72. agenta/client/backend/scopes/raw_client.py +105 -0
  73. agenta/client/backend/testsets/__init__.py +14 -0
  74. agenta/client/backend/testsets/client.py +1098 -653
  75. agenta/client/backend/testsets/raw_client.py +2348 -0
  76. agenta/client/backend/testsets/types/__init__.py +17 -0
  77. agenta/client/backend/testsets/types/create_testset_from_file_request_file_type.py +7 -0
  78. agenta/client/backend/testsets/types/fetch_testset_to_file_request_file_type.py +7 -0
  79. agenta/client/backend/testsets/types/update_testset_from_file_request_file_type.py +7 -0
  80. agenta/client/backend/tracing/__init__.py +7 -0
  81. agenta/client/backend/tracing/client.py +782 -0
  82. agenta/client/backend/tracing/raw_client.py +1223 -0
  83. agenta/client/backend/tracing/types/__init__.py +8 -0
  84. agenta/client/backend/{types/variant_action_enum.py → tracing/types/fetch_trace_request_trace_id.py} +1 -1
  85. agenta/client/backend/tracing/types/remove_trace_request_trace_id.py +5 -0
  86. agenta/client/backend/types/__init__.py +153 -26
  87. agenta/client/backend/types/account_request.py +24 -0
  88. agenta/client/backend/types/account_response.py +5 -7
  89. agenta/client/backend/types/agenta_node_dto.py +13 -13
  90. agenta/client/backend/types/agenta_node_dto_nodes_value.py +1 -0
  91. agenta/client/backend/types/agenta_nodes_response.py +14 -8
  92. agenta/client/backend/types/agenta_root_dto.py +16 -8
  93. agenta/client/backend/types/agenta_roots_response.py +16 -8
  94. agenta/client/backend/types/agenta_tree_dto.py +16 -8
  95. agenta/client/backend/types/agenta_trees_response.py +16 -8
  96. agenta/client/backend/types/aggregated_result.py +5 -7
  97. agenta/client/backend/types/aggregated_result_evaluator_config.py +1 -0
  98. agenta/client/backend/types/analytics_response.py +4 -6
  99. agenta/client/backend/types/annotation.py +50 -0
  100. agenta/client/backend/types/annotation_create.py +39 -0
  101. agenta/client/backend/types/annotation_edit.py +31 -0
  102. agenta/client/backend/types/annotation_kind.py +5 -0
  103. agenta/client/backend/types/{uri.py → annotation_link.py} +6 -7
  104. agenta/client/backend/types/{provider_key_dto.py → annotation_link_response.py} +6 -7
  105. agenta/client/backend/types/annotation_query.py +40 -0
  106. agenta/client/backend/types/annotation_query_request.py +20 -0
  107. agenta/client/backend/types/annotation_reference.py +21 -0
  108. agenta/client/backend/types/annotation_references.py +22 -0
  109. agenta/client/backend/types/{docker_env_vars.py → annotation_response.py} +6 -7
  110. agenta/client/backend/types/annotation_source.py +5 -0
  111. agenta/client/backend/types/annotations_response.py +24 -0
  112. agenta/client/backend/types/app.py +3 -5
  113. agenta/client/backend/types/app_variant_response.py +3 -6
  114. agenta/client/backend/types/app_variant_revision.py +5 -6
  115. agenta/client/backend/types/artifact.py +44 -0
  116. agenta/client/backend/types/base_output.py +3 -5
  117. agenta/client/backend/types/body_fetch_workflow_revision.py +21 -0
  118. agenta/client/backend/types/body_import_testset.py +3 -5
  119. agenta/client/backend/types/bucket_dto.py +4 -6
  120. agenta/client/backend/types/collect_status_response.py +3 -5
  121. agenta/client/backend/types/config_db.py +3 -5
  122. agenta/client/backend/types/config_dto.py +5 -7
  123. agenta/client/backend/types/config_response_model.py +5 -7
  124. agenta/client/backend/types/correct_answer.py +3 -5
  125. agenta/client/backend/types/create_app_output.py +3 -5
  126. agenta/client/backend/types/custom_model_settings_dto.py +3 -5
  127. agenta/client/backend/types/custom_provider_dto.py +6 -9
  128. agenta/client/backend/types/custom_provider_kind.py +5 -5
  129. agenta/client/backend/types/custom_provider_settings_dto.py +3 -5
  130. agenta/client/backend/types/data.py +2 -1
  131. agenta/client/backend/types/delete_evaluation.py +3 -5
  132. agenta/client/backend/types/environment_output.py +3 -5
  133. agenta/client/backend/types/environment_output_extended.py +4 -6
  134. agenta/client/backend/types/environment_revision.py +5 -5
  135. agenta/client/backend/types/error.py +3 -5
  136. agenta/client/backend/types/evaluation.py +6 -8
  137. agenta/client/backend/types/evaluation_scenario.py +5 -7
  138. agenta/client/backend/types/evaluation_scenario_input.py +3 -5
  139. agenta/client/backend/types/evaluation_scenario_output.py +4 -6
  140. agenta/client/backend/types/evaluation_scenario_result.py +4 -6
  141. agenta/client/backend/types/evaluator.py +31 -12
  142. agenta/client/backend/types/evaluator_config.py +3 -5
  143. agenta/client/backend/types/evaluator_flags.py +21 -0
  144. agenta/client/backend/types/evaluator_mapping_output_interface.py +3 -5
  145. agenta/client/backend/types/evaluator_output_interface.py +3 -5
  146. agenta/client/backend/types/evaluator_query.py +32 -0
  147. agenta/client/backend/types/evaluator_query_request.py +30 -0
  148. agenta/client/backend/types/evaluator_request.py +20 -0
  149. agenta/client/backend/types/evaluator_response.py +21 -0
  150. agenta/client/backend/types/evaluators_response.py +21 -0
  151. agenta/client/backend/types/exception_dto.py +3 -5
  152. agenta/client/backend/types/{o_tel_spans_response.py → extended_o_tel_tracing_response.py} +5 -7
  153. agenta/client/backend/types/focus.py +5 -0
  154. agenta/client/backend/types/format.py +5 -0
  155. agenta/client/backend/types/full_json_input.py +34 -0
  156. agenta/client/backend/types/full_json_output.py +29 -0
  157. agenta/client/backend/types/get_config_response.py +3 -5
  158. agenta/client/backend/types/{header_dto.py → header.py} +4 -6
  159. agenta/client/backend/types/http_validation_error.py +4 -6
  160. agenta/client/backend/types/human_evaluation.py +3 -5
  161. agenta/client/backend/types/human_evaluation_scenario.py +4 -6
  162. agenta/client/backend/types/human_evaluation_scenario_input.py +3 -5
  163. agenta/client/backend/types/human_evaluation_scenario_output.py +3 -5
  164. agenta/client/backend/types/invite_request.py +4 -6
  165. agenta/client/backend/types/legacy_analytics_response.py +4 -6
  166. agenta/client/backend/types/legacy_data_point.py +3 -5
  167. agenta/client/backend/types/legacy_evaluator.py +26 -0
  168. agenta/client/backend/types/legacy_scope_request.py +4 -6
  169. agenta/client/backend/types/legacy_scopes_response.py +3 -5
  170. agenta/client/backend/types/legacy_subscription_request.py +19 -0
  171. agenta/client/backend/types/legacy_user_request.py +5 -7
  172. agenta/client/backend/types/legacy_user_response.py +3 -5
  173. agenta/client/backend/types/lifecycle_dto.py +3 -5
  174. agenta/client/backend/types/link_dto.py +4 -6
  175. agenta/client/backend/types/list_api_keys_response.py +3 -5
  176. agenta/client/backend/types/llm_run_rate_limit.py +3 -5
  177. agenta/client/backend/types/meta_request.py +30 -0
  178. agenta/client/backend/types/metrics_dto.py +3 -5
  179. agenta/client/backend/types/new_testset.py +3 -5
  180. agenta/client/backend/types/node_dto.py +4 -6
  181. agenta/client/backend/types/o_tel_context_dto.py +3 -5
  182. agenta/client/backend/types/o_tel_event.py +35 -0
  183. agenta/client/backend/types/o_tel_event_dto.py +3 -5
  184. agenta/client/backend/types/o_tel_extra_dto.py +4 -6
  185. agenta/client/backend/types/o_tel_flat_span.py +56 -0
  186. agenta/client/backend/types/o_tel_flat_span_input_end_time.py +6 -0
  187. agenta/client/backend/types/o_tel_flat_span_input_start_time.py +6 -0
  188. agenta/client/backend/types/o_tel_flat_span_output_end_time.py +6 -0
  189. agenta/client/backend/types/o_tel_flat_span_output_start_time.py +6 -0
  190. agenta/client/backend/types/o_tel_link.py +34 -0
  191. agenta/client/backend/types/o_tel_link_dto.py +4 -6
  192. agenta/client/backend/types/o_tel_links_response.py +22 -0
  193. agenta/client/backend/types/o_tel_span.py +58 -0
  194. agenta/client/backend/types/o_tel_span_dto.py +8 -10
  195. agenta/client/backend/types/o_tel_span_input_end_time.py +6 -0
  196. agenta/client/backend/types/o_tel_span_input_spans_value.py +7 -0
  197. agenta/client/backend/types/o_tel_span_input_start_time.py +6 -0
  198. agenta/client/backend/types/o_tel_span_output_end_time.py +6 -0
  199. agenta/client/backend/types/o_tel_span_output_spans_value.py +30 -0
  200. agenta/client/backend/types/o_tel_span_output_start_time.py +6 -0
  201. agenta/client/backend/types/o_tel_spans_tree.py +22 -0
  202. agenta/client/backend/types/o_tel_spans_tree_input_spans_value.py +7 -0
  203. agenta/client/backend/types/o_tel_spans_tree_output_spans_value.py +5 -0
  204. agenta/client/backend/types/o_tel_status_code.py +1 -1
  205. agenta/client/backend/types/o_tel_tracing_data_response.py +22 -0
  206. agenta/client/backend/types/o_tel_tracing_request.py +22 -0
  207. agenta/client/backend/types/o_tel_tracing_response.py +27 -0
  208. agenta/client/backend/types/organization.py +3 -5
  209. agenta/client/backend/types/organization_details.py +3 -5
  210. agenta/client/backend/types/organization_membership_request.py +5 -7
  211. agenta/client/backend/types/organization_output.py +3 -5
  212. agenta/client/backend/types/organization_request.py +3 -5
  213. agenta/client/backend/types/parent_dto.py +3 -5
  214. agenta/client/backend/types/permission.py +11 -0
  215. agenta/client/backend/types/plan.py +14 -0
  216. agenta/client/backend/types/project_membership_request.py +5 -7
  217. agenta/client/backend/types/project_request.py +4 -6
  218. agenta/client/backend/types/project_scope.py +5 -7
  219. agenta/client/backend/types/projects_response.py +3 -5
  220. agenta/client/backend/types/recursive_types.py +23 -0
  221. agenta/client/backend/types/reference.py +18 -5
  222. agenta/client/backend/types/reference_dto.py +4 -5
  223. agenta/client/backend/types/reference_request_model.py +4 -5
  224. agenta/client/backend/types/result.py +4 -6
  225. agenta/client/backend/types/root_dto.py +3 -5
  226. agenta/client/backend/types/scopes_response_model.py +4 -6
  227. agenta/client/backend/types/secret_dto.py +5 -7
  228. agenta/client/backend/types/secret_response_dto.py +11 -11
  229. agenta/client/backend/types/simple_evaluation_output.py +4 -6
  230. agenta/client/backend/types/span_dto.py +18 -14
  231. agenta/client/backend/types/span_dto_nodes_value.py +1 -1
  232. agenta/client/backend/types/standard_provider_dto.py +5 -7
  233. agenta/client/backend/types/standard_provider_settings_dto.py +3 -5
  234. agenta/client/backend/types/status_dto.py +4 -6
  235. agenta/client/backend/types/tags_request.py +30 -0
  236. agenta/client/backend/types/test_set_output_response.py +5 -7
  237. agenta/client/backend/types/test_set_simple_response.py +3 -5
  238. agenta/client/backend/types/testcase_response.py +33 -0
  239. agenta/client/backend/types/testset.py +46 -0
  240. agenta/client/backend/types/testset_request.py +20 -0
  241. agenta/client/backend/types/testset_response.py +21 -0
  242. agenta/client/backend/types/testsets_response.py +21 -0
  243. agenta/client/backend/types/time_dto.py +3 -5
  244. agenta/client/backend/types/timestamp.py +6 -0
  245. agenta/client/backend/types/tree_dto.py +4 -6
  246. agenta/client/backend/types/update_app_output.py +3 -5
  247. agenta/client/backend/types/user_request.py +3 -5
  248. agenta/client/backend/types/validation_error.py +4 -6
  249. agenta/client/backend/types/workflow_artifact.py +45 -0
  250. agenta/client/backend/types/workflow_data.py +20 -0
  251. agenta/client/backend/types/workflow_flags.py +21 -0
  252. agenta/client/backend/types/workflow_request.py +20 -0
  253. agenta/client/backend/types/workflow_response.py +21 -0
  254. agenta/client/backend/types/workflow_revision.py +57 -0
  255. agenta/client/backend/types/workflow_revision_request.py +20 -0
  256. agenta/client/backend/types/workflow_revision_response.py +21 -0
  257. agenta/client/backend/types/workflow_revisions_response.py +21 -0
  258. agenta/client/backend/types/workflow_variant.py +48 -0
  259. agenta/client/backend/types/workflow_variant_request.py +20 -0
  260. agenta/client/backend/types/workflow_variant_response.py +21 -0
  261. agenta/client/backend/types/workflow_variants_response.py +21 -0
  262. agenta/client/backend/types/workflows_response.py +21 -0
  263. agenta/client/backend/types/workspace.py +3 -5
  264. agenta/client/backend/types/workspace_member_response.py +4 -6
  265. agenta/client/backend/types/workspace_membership_request.py +5 -7
  266. agenta/client/backend/types/workspace_permission.py +5 -7
  267. agenta/client/backend/types/workspace_request.py +4 -6
  268. agenta/client/backend/types/workspace_response.py +4 -6
  269. agenta/client/backend/variants/__init__.py +2 -0
  270. agenta/client/backend/variants/client.py +306 -1651
  271. agenta/client/backend/variants/raw_client.py +2482 -0
  272. agenta/client/backend/variants/types/__init__.py +2 -0
  273. agenta/client/backend/variants/types/add_variant_from_base_and_config_response.py +1 -0
  274. agenta/client/backend/vault/__init__.py +2 -0
  275. agenta/client/backend/vault/client.py +69 -323
  276. agenta/client/backend/vault/raw_client.py +616 -0
  277. agenta/client/backend/workflows/__init__.py +3 -0
  278. agenta/client/backend/workflows/client.py +2398 -0
  279. agenta/client/backend/workflows/raw_client.py +3639 -0
  280. agenta/client/backend/workspace/__init__.py +2 -0
  281. agenta/client/backend/workspace/client.py +46 -147
  282. agenta/client/backend/workspace/raw_client.py +376 -0
  283. agenta/client/types.py +4 -0
  284. agenta/sdk/litellm/mocks/__init__.py +144 -0
  285. agenta/sdk/middleware/auth.py +26 -25
  286. agenta/sdk/types.py +28 -1
  287. {agenta-0.48.9.dist-info → agenta-0.49.0.dist-info}/METADATA +1 -1
  288. agenta-0.49.0.dist-info/RECORD +362 -0
  289. agenta/client/backend/containers/types/__init__.py +0 -5
  290. agenta/client/backend/containers/types/container_templates_response.py +0 -6
  291. agenta/client/backend/types/image.py +0 -25
  292. agenta/client/backend/types/provider_kind.py +0 -21
  293. agenta/client/backend/types/template.py +0 -23
  294. agenta/client/backend/types/template_image_info.py +0 -29
  295. agenta/client/backend/types/variant_action.py +0 -22
  296. agenta-0.48.9.dist-info/RECORD +0 -255
  297. {agenta-0.48.9.dist-info → agenta-0.49.0.dist-info}/WHEEL +0 -0
agenta/client/__init__.py CHANGED
@@ -1,9 +1,16 @@
1
1
  # This file was auto-generated by Fern from our API Definition.
2
2
 
3
+ # Apply Pydantic recursion patch before any models are imported
4
+ try:
5
+ # This will prevent recursion errors during schema generation
6
+ from .backend.pydantic_patch import patch_pydantic_recursion
7
+ except ImportError:
8
+ pass
9
+
3
10
  from .backend.types import (
11
+ AccountRequest,
4
12
  AccountResponse,
5
13
  AgentaNodeDto,
6
- AgentaNodeDtoNodesValue,
7
14
  AgentaNodesResponse,
8
15
  AgentaRootDto,
9
16
  AgentaRootsResponse,
@@ -12,10 +19,25 @@ from .backend.types import (
12
19
  AggregatedResult,
13
20
  AggregatedResultEvaluatorConfig,
14
21
  AnalyticsResponse,
22
+ Annotation,
23
+ AnnotationCreate,
24
+ AnnotationEdit,
25
+ AnnotationKind,
26
+ AnnotationLink,
27
+ AnnotationLinkResponse,
28
+ AnnotationQuery,
29
+ AnnotationQueryRequest,
30
+ AnnotationReference,
31
+ AnnotationReferences,
32
+ AnnotationResponse,
33
+ AnnotationSource,
34
+ AnnotationsResponse,
15
35
  App,
16
36
  AppVariantResponse,
17
37
  AppVariantRevision,
38
+ Artifact,
18
39
  BaseOutput,
40
+ BodyFetchWorkflowRevision,
19
41
  BodyImportTestset,
20
42
  BucketDto,
21
43
  CollectStatusResponse,
@@ -24,8 +46,12 @@ from .backend.types import (
24
46
  ConfigResponseModel,
25
47
  CorrectAnswer,
26
48
  CreateAppOutput,
49
+ CustomModelSettingsDto,
50
+ CustomProviderDto,
51
+ CustomProviderKind,
52
+ CustomProviderSettingsDto,
53
+ Data,
27
54
  DeleteEvaluation,
28
- DockerEnvVars,
29
55
  EnvironmentOutput,
30
56
  EnvironmentOutputExtended,
31
57
  EnvironmentRevision,
@@ -39,46 +65,80 @@ from .backend.types import (
39
65
  EvaluationType,
40
66
  Evaluator,
41
67
  EvaluatorConfig,
68
+ EvaluatorFlags,
42
69
  EvaluatorMappingOutputInterface,
43
70
  EvaluatorOutputInterface,
71
+ EvaluatorQuery,
72
+ EvaluatorQueryRequest,
73
+ EvaluatorRequest,
74
+ EvaluatorResponse,
75
+ EvaluatorsResponse,
44
76
  ExceptionDto,
77
+ Focus,
78
+ Format,
79
+ FullJsonInput,
80
+ FullJsonOutput,
45
81
  GetConfigResponse,
46
- HeaderDto,
82
+ Header,
47
83
  HttpValidationError,
48
84
  HumanEvaluation,
49
85
  HumanEvaluationScenario,
50
86
  HumanEvaluationScenarioInput,
51
87
  HumanEvaluationScenarioOutput,
52
- Image,
53
88
  InviteRequest,
54
89
  LegacyAnalyticsResponse,
55
90
  LegacyDataPoint,
91
+ LegacyEvaluator,
56
92
  LegacyScopeRequest,
57
93
  LegacyScopesResponse,
94
+ LegacySubscriptionRequest,
58
95
  LegacyUserRequest,
59
96
  LegacyUserResponse,
60
97
  LifecycleDto,
61
98
  LinkDto,
62
99
  ListApiKeysResponse,
63
100
  LlmRunRateLimit,
101
+ MetaRequest,
64
102
  MetricsDto,
65
103
  NewTestset,
66
104
  NodeDto,
67
105
  NodeType,
68
106
  OTelContextDto,
107
+ OTelEvent,
69
108
  OTelEventDto,
70
109
  OTelExtraDto,
110
+ OTelFlatSpan,
111
+ OTelFlatSpanInputEndTime,
112
+ OTelFlatSpanInputStartTime,
113
+ OTelFlatSpanOutputEndTime,
114
+ OTelFlatSpanOutputStartTime,
115
+ OTelLink,
71
116
  OTelLinkDto,
117
+ OTelLinksResponse,
118
+ OTelSpan,
72
119
  OTelSpanDto,
120
+ OTelSpanInputEndTime,
121
+ OTelSpanInputSpansValue,
122
+ OTelSpanInputStartTime,
73
123
  OTelSpanKind,
74
- OTelSpansResponse,
124
+ OTelSpanOutputEndTime,
125
+ OTelSpanOutputSpansValue,
126
+ OTelSpanOutputStartTime,
127
+ OTelSpansTree,
128
+ OTelSpansTreeInputSpansValue,
129
+ OTelSpansTreeOutputSpansValue,
75
130
  OTelStatusCode,
131
+ OTelTracingDataResponse,
132
+ OTelTracingRequest,
133
+ OTelTracingResponse,
76
134
  Organization,
135
+ OrganizationDetails,
77
136
  OrganizationMembershipRequest,
78
137
  OrganizationOutput,
79
138
  OrganizationRequest,
80
139
  ParentDto,
81
140
  Permission,
141
+ Plan,
82
142
  ProjectMembershipRequest,
83
143
  ProjectRequest,
84
144
  ProjectScope,
@@ -95,61 +155,96 @@ from .backend.types import (
95
155
  SecretKind,
96
156
  SecretResponseDto,
97
157
  SimpleEvaluationOutput,
98
- SpanDto,
99
- SpanDtoNodesValue,
158
+ StandardProviderDto,
159
+ StandardProviderKind,
160
+ StandardProviderSettingsDto,
100
161
  StatusCode,
101
162
  StatusDto,
102
- Template,
103
- TemplateImageInfo,
104
163
  TestSetOutputResponse,
105
164
  TestSetSimpleResponse,
165
+ TestcaseResponse,
166
+ Testset,
167
+ TestsetRequest,
168
+ TestsetResponse,
169
+ TestsetsResponse,
106
170
  TimeDto,
171
+ Timestamp,
107
172
  TreeDto,
108
173
  TreeType,
109
174
  UpdateAppOutput,
110
- Uri,
111
175
  UserRequest,
112
176
  ValidationError,
113
177
  ValidationErrorLocItem,
114
- VariantAction,
115
- VariantActionEnum,
178
+ WorkflowArtifact,
179
+ WorkflowData,
180
+ WorkflowFlags,
181
+ WorkflowRequest,
182
+ WorkflowResponse,
183
+ WorkflowRevision,
184
+ WorkflowRevisionRequest,
185
+ WorkflowRevisionResponse,
186
+ WorkflowRevisionsResponse,
187
+ WorkflowVariant,
188
+ WorkflowVariantRequest,
189
+ WorkflowVariantResponse,
190
+ WorkflowVariantsResponse,
191
+ WorkflowsResponse,
192
+ Workspace,
116
193
  WorkspaceMemberResponse,
117
194
  WorkspaceMembershipRequest,
118
195
  WorkspacePermission,
119
196
  WorkspaceRequest,
120
197
  WorkspaceResponse,
121
198
  WorkspaceRole,
122
- WorkspaceRoleResponse,
123
199
  )
124
200
  from .backend.errors import UnprocessableEntityError
125
201
  from .backend import (
126
202
  access_control,
127
203
  admin,
204
+ api_keys,
128
205
  apps,
129
206
  bases,
207
+ billing,
130
208
  configs,
131
209
  containers,
132
210
  environments,
211
+ evals,
133
212
  evaluations,
134
213
  evaluators,
135
214
  human_evaluations,
136
215
  observability,
216
+ organization,
137
217
  scopes,
138
218
  testsets,
219
+ tracing,
139
220
  variants,
140
221
  vault,
222
+ workflows,
223
+ workspace,
141
224
  )
142
225
  from .client import AgentaApi, AsyncAgentaApi
143
- from .backend.containers import ContainerTemplatesResponse
144
- from .backend.observability import Format, QueryAnalyticsResponse, QueryTracesResponse
226
+ from .backend.environment import AgentaApiEnvironment
227
+ from .backend.observability import (
228
+ FetchTraceByIdRequestTraceId,
229
+ FetchTraceByIdResponse,
230
+ QueryAnalyticsResponse,
231
+ QueryTracesResponse,
232
+ )
233
+ from .backend.testsets import (
234
+ CreateTestsetFromFileRequestFileType,
235
+ FetchTestsetToFileRequestFileType,
236
+ UpdateTestsetFromFileRequestFileType,
237
+ )
238
+ from .backend.tracing import FetchTraceRequestTraceId, RemoveTraceRequestTraceId
145
239
  from .backend.variants import AddVariantFromBaseAndConfigResponse
146
240
 
147
241
  __all__ = [
242
+ "AccountRequest",
148
243
  "AccountResponse",
149
244
  "AddVariantFromBaseAndConfigResponse",
150
245
  "AgentaApi",
246
+ "AgentaApiEnvironment",
151
247
  "AgentaNodeDto",
152
- "AgentaNodeDtoNodesValue",
153
248
  "AgentaNodesResponse",
154
249
  "AgentaRootDto",
155
250
  "AgentaRootsResponse",
@@ -158,22 +253,41 @@ __all__ = [
158
253
  "AggregatedResult",
159
254
  "AggregatedResultEvaluatorConfig",
160
255
  "AnalyticsResponse",
256
+ "Annotation",
257
+ "AnnotationCreate",
258
+ "AnnotationEdit",
259
+ "AnnotationKind",
260
+ "AnnotationLink",
261
+ "AnnotationLinkResponse",
262
+ "AnnotationQuery",
263
+ "AnnotationQueryRequest",
264
+ "AnnotationReference",
265
+ "AnnotationReferences",
266
+ "AnnotationResponse",
267
+ "AnnotationSource",
268
+ "AnnotationsResponse",
161
269
  "App",
162
270
  "AppVariantResponse",
163
271
  "AppVariantRevision",
272
+ "Artifact",
164
273
  "AsyncAgentaApi",
165
274
  "BaseOutput",
275
+ "BodyFetchWorkflowRevision",
166
276
  "BodyImportTestset",
167
277
  "BucketDto",
168
278
  "CollectStatusResponse",
169
279
  "ConfigDb",
170
280
  "ConfigDto",
171
281
  "ConfigResponseModel",
172
- "ContainerTemplatesResponse",
173
282
  "CorrectAnswer",
174
283
  "CreateAppOutput",
284
+ "CreateTestsetFromFileRequestFileType",
285
+ "CustomModelSettingsDto",
286
+ "CustomProviderDto",
287
+ "CustomProviderKind",
288
+ "CustomProviderSettingsDto",
289
+ "Data",
175
290
  "DeleteEvaluation",
176
- "DockerEnvVars",
177
291
  "EnvironmentOutput",
178
292
  "EnvironmentOutputExtended",
179
293
  "EnvironmentRevision",
@@ -187,47 +301,84 @@ __all__ = [
187
301
  "EvaluationType",
188
302
  "Evaluator",
189
303
  "EvaluatorConfig",
304
+ "EvaluatorFlags",
190
305
  "EvaluatorMappingOutputInterface",
191
306
  "EvaluatorOutputInterface",
307
+ "EvaluatorQuery",
308
+ "EvaluatorQueryRequest",
309
+ "EvaluatorRequest",
310
+ "EvaluatorResponse",
311
+ "EvaluatorsResponse",
192
312
  "ExceptionDto",
313
+ "FetchTestsetToFileRequestFileType",
314
+ "FetchTraceByIdRequestTraceId",
315
+ "FetchTraceByIdResponse",
316
+ "FetchTraceRequestTraceId",
317
+ "Focus",
193
318
  "Format",
319
+ "FullJsonInput",
320
+ "FullJsonOutput",
194
321
  "GetConfigResponse",
195
- "HeaderDto",
322
+ "Header",
196
323
  "HttpValidationError",
197
324
  "HumanEvaluation",
198
325
  "HumanEvaluationScenario",
199
326
  "HumanEvaluationScenarioInput",
200
327
  "HumanEvaluationScenarioOutput",
201
- "Image",
202
328
  "InviteRequest",
203
329
  "LegacyAnalyticsResponse",
204
330
  "LegacyDataPoint",
331
+ "LegacyEvaluator",
205
332
  "LegacyScopeRequest",
206
333
  "LegacyScopesResponse",
334
+ "LegacySubscriptionRequest",
207
335
  "LegacyUserRequest",
208
336
  "LegacyUserResponse",
209
337
  "LifecycleDto",
210
338
  "LinkDto",
211
339
  "ListApiKeysResponse",
212
340
  "LlmRunRateLimit",
341
+ "MetaRequest",
213
342
  "MetricsDto",
214
343
  "NewTestset",
215
344
  "NodeDto",
216
345
  "NodeType",
217
346
  "OTelContextDto",
347
+ "OTelEvent",
218
348
  "OTelEventDto",
219
349
  "OTelExtraDto",
350
+ "OTelFlatSpan",
351
+ "OTelFlatSpanInputEndTime",
352
+ "OTelFlatSpanInputStartTime",
353
+ "OTelFlatSpanOutputEndTime",
354
+ "OTelFlatSpanOutputStartTime",
355
+ "OTelLink",
220
356
  "OTelLinkDto",
357
+ "OTelLinksResponse",
358
+ "OTelSpan",
221
359
  "OTelSpanDto",
360
+ "OTelSpanInputEndTime",
361
+ "OTelSpanInputSpansValue",
362
+ "OTelSpanInputStartTime",
222
363
  "OTelSpanKind",
223
- "OTelSpansResponse",
364
+ "OTelSpanOutputEndTime",
365
+ "OTelSpanOutputSpansValue",
366
+ "OTelSpanOutputStartTime",
367
+ "OTelSpansTree",
368
+ "OTelSpansTreeInputSpansValue",
369
+ "OTelSpansTreeOutputSpansValue",
224
370
  "OTelStatusCode",
371
+ "OTelTracingDataResponse",
372
+ "OTelTracingRequest",
373
+ "OTelTracingResponse",
225
374
  "Organization",
375
+ "OrganizationDetails",
226
376
  "OrganizationMembershipRequest",
227
377
  "OrganizationOutput",
228
378
  "OrganizationRequest",
229
379
  "ParentDto",
230
380
  "Permission",
381
+ "Plan",
231
382
  "ProjectMembershipRequest",
232
383
  "ProjectRequest",
233
384
  "ProjectScope",
@@ -237,6 +388,7 @@ __all__ = [
237
388
  "Reference",
238
389
  "ReferenceDto",
239
390
  "ReferenceRequestModel",
391
+ "RemoveTraceRequestTraceId",
240
392
  "Result",
241
393
  "Role",
242
394
  "RootDto",
@@ -246,45 +398,69 @@ __all__ = [
246
398
  "SecretKind",
247
399
  "SecretResponseDto",
248
400
  "SimpleEvaluationOutput",
249
- "SpanDto",
250
- "SpanDtoNodesValue",
401
+ "StandardProviderDto",
402
+ "StandardProviderKind",
403
+ "StandardProviderSettingsDto",
251
404
  "StatusCode",
252
405
  "StatusDto",
253
- "Template",
254
- "TemplateImageInfo",
255
406
  "TestSetOutputResponse",
256
407
  "TestSetSimpleResponse",
408
+ "TestcaseResponse",
409
+ "Testset",
410
+ "TestsetRequest",
411
+ "TestsetResponse",
412
+ "TestsetsResponse",
257
413
  "TimeDto",
414
+ "Timestamp",
258
415
  "TreeDto",
259
416
  "TreeType",
260
417
  "UnprocessableEntityError",
261
418
  "UpdateAppOutput",
262
- "Uri",
419
+ "UpdateTestsetFromFileRequestFileType",
263
420
  "UserRequest",
264
421
  "ValidationError",
265
422
  "ValidationErrorLocItem",
266
- "VariantAction",
267
- "VariantActionEnum",
423
+ "WorkflowArtifact",
424
+ "WorkflowData",
425
+ "WorkflowFlags",
426
+ "WorkflowRequest",
427
+ "WorkflowResponse",
428
+ "WorkflowRevision",
429
+ "WorkflowRevisionRequest",
430
+ "WorkflowRevisionResponse",
431
+ "WorkflowRevisionsResponse",
432
+ "WorkflowVariant",
433
+ "WorkflowVariantRequest",
434
+ "WorkflowVariantResponse",
435
+ "WorkflowVariantsResponse",
436
+ "WorkflowsResponse",
437
+ "Workspace",
268
438
  "WorkspaceMemberResponse",
269
439
  "WorkspaceMembershipRequest",
270
440
  "WorkspacePermission",
271
441
  "WorkspaceRequest",
272
442
  "WorkspaceResponse",
273
443
  "WorkspaceRole",
274
- "WorkspaceRoleResponse",
275
444
  "access_control",
276
445
  "admin",
446
+ "api_keys",
277
447
  "apps",
278
448
  "bases",
449
+ "billing",
279
450
  "configs",
280
451
  "containers",
281
452
  "environments",
453
+ "evals",
282
454
  "evaluations",
283
455
  "evaluators",
284
456
  "human_evaluations",
285
457
  "observability",
458
+ "organization",
286
459
  "scopes",
287
460
  "testsets",
461
+ "tracing",
288
462
  "variants",
289
463
  "vault",
464
+ "workflows",
465
+ "workspace",
290
466
  ]