c63a5cfe-b235-4fbe-8bbb-82a9e02a482a-python 0.1.0a4__py3-none-any.whl → 0.1.0a6__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 (143) hide show
  1. {c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a4.dist-info → c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a6.dist-info}/METADATA +41 -16
  2. c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a6.dist-info/RECORD +230 -0
  3. gradientai/_client.py +73 -71
  4. gradientai/_version.py +1 -1
  5. gradientai/resources/__init__.py +14 -28
  6. gradientai/resources/agents/__init__.py +69 -13
  7. gradientai/resources/agents/agents.py +148 -20
  8. gradientai/resources/{regions → agents}/evaluation_datasets.py +3 -3
  9. gradientai/resources/agents/evaluation_metrics/__init__.py +33 -0
  10. gradientai/resources/agents/evaluation_metrics/evaluation_metrics.py +177 -0
  11. gradientai/resources/agents/evaluation_metrics/workspaces/__init__.py +33 -0
  12. gradientai/resources/agents/evaluation_metrics/workspaces/agents.py +324 -0
  13. gradientai/resources/agents/evaluation_metrics/workspaces/workspaces.py +654 -0
  14. gradientai/resources/{regions/evaluation_runs → agents}/evaluation_runs.py +197 -47
  15. gradientai/resources/{regions → agents}/evaluation_test_cases.py +30 -9
  16. gradientai/resources/agents/{child_agents.py → routes.py} +64 -64
  17. gradientai/resources/chat/completions.py +20 -2
  18. gradientai/resources/inference/__init__.py +0 -14
  19. gradientai/resources/inference/inference.py +0 -32
  20. gradientai/resources/knowledge_bases/__init__.py +14 -0
  21. gradientai/resources/{indexing_jobs.py → knowledge_bases/indexing_jobs.py} +16 -12
  22. gradientai/resources/knowledge_bases/knowledge_bases.py +32 -0
  23. gradientai/resources/{providers → model_providers}/__init__.py +13 -13
  24. gradientai/resources/{providers → model_providers}/anthropic/keys.py +12 -7
  25. gradientai/resources/{providers/providers.py → model_providers/model_providers.py} +31 -31
  26. gradientai/resources/{providers → model_providers}/openai/keys.py +12 -7
  27. gradientai/resources/models.py +77 -105
  28. gradientai/resources/regions.py +195 -0
  29. gradientai/types/__init__.py +2 -17
  30. gradientai/types/agent_list_response.py +2 -2
  31. gradientai/types/agents/__init__.py +44 -8
  32. gradientai/types/{api_evaluation_metric.py → agents/api_evaluation_metric.py} +1 -1
  33. gradientai/types/{regions/evaluation_runs → agents}/api_evaluation_metric_result.py +1 -1
  34. gradientai/types/{regions/evaluation_runs/api_prompt.py → agents/api_evaluation_prompt.py} +3 -3
  35. gradientai/types/{regions/evaluation_runs → agents}/api_evaluation_run.py +13 -1
  36. gradientai/types/{regions → agents}/api_evaluation_test_case.py +1 -1
  37. gradientai/types/agents/api_key_list_response.py +2 -2
  38. gradientai/types/{region_list_evaluation_metrics_response.py → agents/evaluation_metric_list_response.py} +3 -3
  39. gradientai/types/agents/evaluation_metrics/__init__.py +14 -0
  40. gradientai/types/agents/evaluation_metrics/workspace_create_params.py +16 -0
  41. gradientai/types/agents/evaluation_metrics/workspace_create_response.py +16 -0
  42. gradientai/types/agents/evaluation_metrics/workspace_delete_response.py +11 -0
  43. gradientai/types/agents/evaluation_metrics/workspace_list_evaluation_test_cases_response.py +12 -0
  44. gradientai/types/agents/evaluation_metrics/workspace_list_response.py +16 -0
  45. gradientai/types/agents/evaluation_metrics/workspace_retrieve_response.py +16 -0
  46. gradientai/types/agents/evaluation_metrics/workspace_update_params.py +18 -0
  47. gradientai/types/agents/evaluation_metrics/workspace_update_response.py +16 -0
  48. gradientai/types/agents/evaluation_metrics/workspaces/__init__.py +8 -0
  49. gradientai/types/agents/evaluation_metrics/workspaces/agent_list_params.py +26 -0
  50. gradientai/types/agents/evaluation_metrics/workspaces/agent_list_response.py +22 -0
  51. gradientai/types/agents/evaluation_metrics/workspaces/agent_move_params.py +16 -0
  52. gradientai/types/agents/evaluation_metrics/workspaces/agent_move_response.py +16 -0
  53. gradientai/types/{regions → agents}/evaluation_run_create_params.py +3 -2
  54. gradientai/types/{regions → agents}/evaluation_run_create_response.py +2 -2
  55. gradientai/types/{regions/evaluation_runs/result_retrieve_response.py → agents/evaluation_run_list_results_response.py} +5 -5
  56. gradientai/types/{regions → agents}/evaluation_run_retrieve_response.py +1 -1
  57. gradientai/types/agents/evaluation_run_retrieve_results_response.py +12 -0
  58. gradientai/types/{regions → agents}/evaluation_test_case_list_evaluation_runs_response.py +1 -1
  59. gradientai/types/agents/evaluation_test_case_retrieve_params.py +12 -0
  60. gradientai/types/agents/{child_agent_add_params.py → route_add_params.py} +2 -2
  61. gradientai/types/agents/{child_agent_add_response.py → route_add_response.py} +2 -2
  62. gradientai/types/agents/{child_agent_delete_response.py → route_delete_response.py} +2 -2
  63. gradientai/types/agents/{child_agent_update_params.py → route_update_params.py} +2 -2
  64. gradientai/types/agents/{child_agent_update_response.py → route_update_response.py} +2 -2
  65. gradientai/types/agents/{child_agent_view_response.py → route_view_response.py} +2 -2
  66. gradientai/types/agents/version_list_response.py +2 -2
  67. gradientai/types/api_knowledge_base.py +1 -1
  68. gradientai/types/api_workspace.py +1 -1
  69. gradientai/types/inference/__init__.py +0 -2
  70. gradientai/types/inference/api_key_list_response.py +2 -2
  71. gradientai/types/knowledge_base_list_response.py +2 -2
  72. gradientai/types/knowledge_bases/__init__.py +12 -0
  73. gradientai/types/{indexing_job_retrieve_data_sources_response.py → knowledge_bases/api_indexed_data_source.py} +4 -8
  74. gradientai/types/{api_indexing_job.py → knowledge_bases/api_indexing_job.py} +13 -1
  75. gradientai/types/knowledge_bases/api_knowledge_base_data_source.py +4 -1
  76. gradientai/types/knowledge_bases/data_source_list_response.py +2 -2
  77. gradientai/types/{indexing_job_create_response.py → knowledge_bases/indexing_job_create_response.py} +1 -1
  78. gradientai/types/{indexing_job_list_response.py → knowledge_bases/indexing_job_list_response.py} +3 -3
  79. gradientai/types/knowledge_bases/indexing_job_retrieve_data_sources_response.py +12 -0
  80. gradientai/types/{indexing_job_retrieve_response.py → knowledge_bases/indexing_job_retrieve_response.py} +1 -1
  81. gradientai/types/{indexing_job_update_cancel_params.py → knowledge_bases/indexing_job_update_cancel_params.py} +1 -1
  82. gradientai/types/{indexing_job_update_cancel_response.py → knowledge_bases/indexing_job_update_cancel_response.py} +1 -1
  83. gradientai/types/{inference/model.py → model.py} +1 -1
  84. gradientai/types/model_list_response.py +5 -8
  85. gradientai/types/{providers → model_providers}/anthropic/key_list_agents_response.py +2 -2
  86. gradientai/types/{providers → model_providers}/anthropic/key_list_response.py +2 -2
  87. gradientai/types/{providers → model_providers}/openai/key_list_response.py +2 -2
  88. gradientai/types/{providers → model_providers}/openai/key_retrieve_agents_response.py +2 -2
  89. gradientai/types/shared/__init__.py +4 -0
  90. c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a4.dist-info/RECORD +0 -217
  91. gradientai/resources/inference/models.py +0 -226
  92. gradientai/resources/regions/__init__.py +0 -61
  93. gradientai/resources/regions/evaluation_runs/__init__.py +0 -33
  94. gradientai/resources/regions/evaluation_runs/results.py +0 -264
  95. gradientai/resources/regions/regions.py +0 -352
  96. gradientai/types/api_model.py +0 -32
  97. gradientai/types/inference/model_list_response.py +0 -15
  98. gradientai/types/model_list_params.py +0 -42
  99. gradientai/types/regions/__init__.py +0 -32
  100. gradientai/types/regions/evaluation_runs/__init__.py +0 -9
  101. gradientai/types/regions/evaluation_runs/result_retrieve_prompt_response.py +0 -12
  102. {c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a4.dist-info → c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a6.dist-info}/WHEEL +0 -0
  103. {c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a4.dist-info → c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a6.dist-info}/licenses/LICENSE +0 -0
  104. /gradientai/resources/{providers → model_providers}/anthropic/__init__.py +0 -0
  105. /gradientai/resources/{providers → model_providers}/anthropic/anthropic.py +0 -0
  106. /gradientai/resources/{providers → model_providers}/openai/__init__.py +0 -0
  107. /gradientai/resources/{providers → model_providers}/openai/openai.py +0 -0
  108. /gradientai/types/{regions → agents}/api_star_metric.py +0 -0
  109. /gradientai/types/{regions → agents}/api_star_metric_param.py +0 -0
  110. /gradientai/types/{regions → agents}/evaluation_dataset_create_file_upload_presigned_urls_params.py +0 -0
  111. /gradientai/types/{regions → agents}/evaluation_dataset_create_file_upload_presigned_urls_response.py +0 -0
  112. /gradientai/types/{regions → agents}/evaluation_dataset_create_params.py +0 -0
  113. /gradientai/types/{regions → agents}/evaluation_dataset_create_response.py +0 -0
  114. /gradientai/types/{regions → agents}/evaluation_test_case_create_params.py +0 -0
  115. /gradientai/types/{regions → agents}/evaluation_test_case_create_response.py +0 -0
  116. /gradientai/types/{regions → agents}/evaluation_test_case_list_evaluation_runs_params.py +0 -0
  117. /gradientai/types/{regions → agents}/evaluation_test_case_list_response.py +0 -0
  118. /gradientai/types/{regions → agents}/evaluation_test_case_retrieve_response.py +0 -0
  119. /gradientai/types/{regions → agents}/evaluation_test_case_update_params.py +0 -0
  120. /gradientai/types/{regions → agents}/evaluation_test_case_update_response.py +0 -0
  121. /gradientai/types/{indexing_job_create_params.py → knowledge_bases/indexing_job_create_params.py} +0 -0
  122. /gradientai/types/{indexing_job_list_params.py → knowledge_bases/indexing_job_list_params.py} +0 -0
  123. /gradientai/types/{providers → model_providers}/__init__.py +0 -0
  124. /gradientai/types/{providers → model_providers}/anthropic/__init__.py +0 -0
  125. /gradientai/types/{providers → model_providers}/anthropic/key_create_params.py +0 -0
  126. /gradientai/types/{providers → model_providers}/anthropic/key_create_response.py +0 -0
  127. /gradientai/types/{providers → model_providers}/anthropic/key_delete_response.py +0 -0
  128. /gradientai/types/{providers → model_providers}/anthropic/key_list_agents_params.py +0 -0
  129. /gradientai/types/{providers → model_providers}/anthropic/key_list_params.py +0 -0
  130. /gradientai/types/{providers → model_providers}/anthropic/key_retrieve_response.py +0 -0
  131. /gradientai/types/{providers → model_providers}/anthropic/key_update_params.py +0 -0
  132. /gradientai/types/{providers → model_providers}/anthropic/key_update_response.py +0 -0
  133. /gradientai/types/{providers → model_providers}/openai/__init__.py +0 -0
  134. /gradientai/types/{providers → model_providers}/openai/key_create_params.py +0 -0
  135. /gradientai/types/{providers → model_providers}/openai/key_create_response.py +0 -0
  136. /gradientai/types/{providers → model_providers}/openai/key_delete_response.py +0 -0
  137. /gradientai/types/{providers → model_providers}/openai/key_list_params.py +0 -0
  138. /gradientai/types/{providers → model_providers}/openai/key_retrieve_agents_params.py +0 -0
  139. /gradientai/types/{providers → model_providers}/openai/key_retrieve_response.py +0 -0
  140. /gradientai/types/{providers → model_providers}/openai/key_update_params.py +0 -0
  141. /gradientai/types/{providers → model_providers}/openai/key_update_response.py +0 -0
  142. /gradientai/types/{agents → shared}/api_links.py +0 -0
  143. /gradientai/types/{agents → shared}/api_meta.py +0 -0
@@ -2,15 +2,14 @@
2
2
 
3
3
  from __future__ import annotations
4
4
 
5
+ from .model import Model as Model
6
+ from .shared import APIMeta as APIMeta, APILinks as APILinks
5
7
  from .api_agent import APIAgent as APIAgent
6
- from .api_model import APIModel as APIModel
7
8
  from .api_agreement import APIAgreement as APIAgreement
8
9
  from .api_workspace import APIWorkspace as APIWorkspace
9
10
  from .api_agent_model import APIAgentModel as APIAgentModel
10
- from .api_indexing_job import APIIndexingJob as APIIndexingJob
11
11
  from .agent_list_params import AgentListParams as AgentListParams
12
12
  from .api_model_version import APIModelVersion as APIModelVersion
13
- from .model_list_params import ModelListParams as ModelListParams
14
13
  from .api_knowledge_base import APIKnowledgeBase as APIKnowledgeBase
15
14
  from .region_list_params import RegionListParams as RegionListParams
16
15
  from .agent_create_params import AgentCreateParams as AgentCreateParams
@@ -22,32 +21,18 @@ from .region_list_response import RegionListResponse as RegionListResponse
22
21
  from .agent_create_response import AgentCreateResponse as AgentCreateResponse
23
22
  from .agent_delete_response import AgentDeleteResponse as AgentDeleteResponse
24
23
  from .agent_update_response import AgentUpdateResponse as AgentUpdateResponse
25
- from .api_evaluation_metric import APIEvaluationMetric as APIEvaluationMetric
26
24
  from .api_agent_api_key_info import APIAgentAPIKeyInfo as APIAgentAPIKeyInfo
27
25
  from .agent_retrieve_response import AgentRetrieveResponse as AgentRetrieveResponse
28
26
  from .api_openai_api_key_info import APIOpenAIAPIKeyInfo as APIOpenAIAPIKeyInfo
29
- from .indexing_job_list_params import IndexingJobListParams as IndexingJobListParams
30
27
  from .api_deployment_visibility import APIDeploymentVisibility as APIDeploymentVisibility
31
28
  from .agent_update_status_params import AgentUpdateStatusParams as AgentUpdateStatusParams
32
29
  from .api_anthropic_api_key_info import APIAnthropicAPIKeyInfo as APIAnthropicAPIKeyInfo
33
- from .indexing_job_create_params import IndexingJobCreateParams as IndexingJobCreateParams
34
- from .indexing_job_list_response import IndexingJobListResponse as IndexingJobListResponse
35
30
  from .knowledge_base_list_params import KnowledgeBaseListParams as KnowledgeBaseListParams
36
31
  from .agent_update_status_response import AgentUpdateStatusResponse as AgentUpdateStatusResponse
37
- from .indexing_job_create_response import IndexingJobCreateResponse as IndexingJobCreateResponse
38
32
  from .knowledge_base_create_params import KnowledgeBaseCreateParams as KnowledgeBaseCreateParams
39
33
  from .knowledge_base_list_response import KnowledgeBaseListResponse as KnowledgeBaseListResponse
40
34
  from .knowledge_base_update_params import KnowledgeBaseUpdateParams as KnowledgeBaseUpdateParams
41
- from .indexing_job_retrieve_response import IndexingJobRetrieveResponse as IndexingJobRetrieveResponse
42
35
  from .knowledge_base_create_response import KnowledgeBaseCreateResponse as KnowledgeBaseCreateResponse
43
36
  from .knowledge_base_delete_response import KnowledgeBaseDeleteResponse as KnowledgeBaseDeleteResponse
44
37
  from .knowledge_base_update_response import KnowledgeBaseUpdateResponse as KnowledgeBaseUpdateResponse
45
38
  from .knowledge_base_retrieve_response import KnowledgeBaseRetrieveResponse as KnowledgeBaseRetrieveResponse
46
- from .indexing_job_update_cancel_params import IndexingJobUpdateCancelParams as IndexingJobUpdateCancelParams
47
- from .indexing_job_update_cancel_response import IndexingJobUpdateCancelResponse as IndexingJobUpdateCancelResponse
48
- from .region_list_evaluation_metrics_response import (
49
- RegionListEvaluationMetricsResponse as RegionListEvaluationMetricsResponse,
50
- )
51
- from .indexing_job_retrieve_data_sources_response import (
52
- IndexingJobRetrieveDataSourcesResponse as IndexingJobRetrieveDataSourcesResponse,
53
- )
@@ -5,9 +5,9 @@ from datetime import datetime
5
5
  from typing_extensions import Literal
6
6
 
7
7
  from .._models import BaseModel
8
- from .agents.api_meta import APIMeta
9
8
  from .api_agent_model import APIAgentModel
10
- from .agents.api_links import APILinks
9
+ from .shared.api_meta import APIMeta
10
+ from .shared.api_links import APILinks
11
11
  from .api_knowledge_base import APIKnowledgeBase
12
12
  from .api_retrieval_method import APIRetrievalMethod
13
13
  from .api_deployment_visibility import APIDeploymentVisibility
@@ -2,30 +2,66 @@
2
2
 
3
3
  from __future__ import annotations
4
4
 
5
- from .api_meta import APIMeta as APIMeta
6
- from .api_links import APILinks as APILinks
5
+ from .api_star_metric import APIStarMetric as APIStarMetric
6
+ from .route_add_params import RouteAddParams as RouteAddParams
7
+ from .api_evaluation_run import APIEvaluationRun as APIEvaluationRun
8
+ from .route_add_response import RouteAddResponse as RouteAddResponse
7
9
  from .api_key_list_params import APIKeyListParams as APIKeyListParams
10
+ from .route_update_params import RouteUpdateParams as RouteUpdateParams
11
+ from .route_view_response import RouteViewResponse as RouteViewResponse
8
12
  from .version_list_params import VersionListParams as VersionListParams
13
+ from .api_evaluation_metric import APIEvaluationMetric as APIEvaluationMetric
14
+ from .api_evaluation_prompt import APIEvaluationPrompt as APIEvaluationPrompt
9
15
  from .api_key_create_params import APIKeyCreateParams as APIKeyCreateParams
10
16
  from .api_key_list_response import APIKeyListResponse as APIKeyListResponse
11
17
  from .api_key_update_params import APIKeyUpdateParams as APIKeyUpdateParams
18
+ from .api_star_metric_param import APIStarMetricParam as APIStarMetricParam
19
+ from .route_delete_response import RouteDeleteResponse as RouteDeleteResponse
20
+ from .route_update_response import RouteUpdateResponse as RouteUpdateResponse
12
21
  from .version_list_response import VersionListResponse as VersionListResponse
13
22
  from .version_update_params import VersionUpdateParams as VersionUpdateParams
14
- from .child_agent_add_params import ChildAgentAddParams as ChildAgentAddParams
15
23
  from .function_create_params import FunctionCreateParams as FunctionCreateParams
16
24
  from .function_update_params import FunctionUpdateParams as FunctionUpdateParams
17
25
  from .api_key_create_response import APIKeyCreateResponse as APIKeyCreateResponse
18
26
  from .api_key_delete_response import APIKeyDeleteResponse as APIKeyDeleteResponse
19
27
  from .api_key_update_response import APIKeyUpdateResponse as APIKeyUpdateResponse
20
28
  from .version_update_response import VersionUpdateResponse as VersionUpdateResponse
21
- from .child_agent_add_response import ChildAgentAddResponse as ChildAgentAddResponse
29
+ from .api_evaluation_test_case import APIEvaluationTestCase as APIEvaluationTestCase
22
30
  from .function_create_response import FunctionCreateResponse as FunctionCreateResponse
23
31
  from .function_delete_response import FunctionDeleteResponse as FunctionDeleteResponse
24
32
  from .function_update_response import FunctionUpdateResponse as FunctionUpdateResponse
25
- from .child_agent_update_params import ChildAgentUpdateParams as ChildAgentUpdateParams
26
- from .child_agent_view_response import ChildAgentViewResponse as ChildAgentViewResponse
27
33
  from .api_key_regenerate_response import APIKeyRegenerateResponse as APIKeyRegenerateResponse
28
- from .child_agent_delete_response import ChildAgentDeleteResponse as ChildAgentDeleteResponse
29
- from .child_agent_update_response import ChildAgentUpdateResponse as ChildAgentUpdateResponse
34
+ from .api_evaluation_metric_result import APIEvaluationMetricResult as APIEvaluationMetricResult
35
+ from .evaluation_run_create_params import EvaluationRunCreateParams as EvaluationRunCreateParams
30
36
  from .api_link_knowledge_base_output import APILinkKnowledgeBaseOutput as APILinkKnowledgeBaseOutput
37
+ from .evaluation_run_create_response import EvaluationRunCreateResponse as EvaluationRunCreateResponse
31
38
  from .knowledge_base_detach_response import KnowledgeBaseDetachResponse as KnowledgeBaseDetachResponse
39
+ from .evaluation_metric_list_response import EvaluationMetricListResponse as EvaluationMetricListResponse
40
+ from .evaluation_dataset_create_params import EvaluationDatasetCreateParams as EvaluationDatasetCreateParams
41
+ from .evaluation_run_retrieve_response import EvaluationRunRetrieveResponse as EvaluationRunRetrieveResponse
42
+ from .evaluation_dataset_create_response import EvaluationDatasetCreateResponse as EvaluationDatasetCreateResponse
43
+ from .evaluation_test_case_create_params import EvaluationTestCaseCreateParams as EvaluationTestCaseCreateParams
44
+ from .evaluation_test_case_list_response import EvaluationTestCaseListResponse as EvaluationTestCaseListResponse
45
+ from .evaluation_test_case_update_params import EvaluationTestCaseUpdateParams as EvaluationTestCaseUpdateParams
46
+ from .evaluation_run_list_results_response import EvaluationRunListResultsResponse as EvaluationRunListResultsResponse
47
+ from .evaluation_test_case_create_response import EvaluationTestCaseCreateResponse as EvaluationTestCaseCreateResponse
48
+ from .evaluation_test_case_retrieve_params import EvaluationTestCaseRetrieveParams as EvaluationTestCaseRetrieveParams
49
+ from .evaluation_test_case_update_response import EvaluationTestCaseUpdateResponse as EvaluationTestCaseUpdateResponse
50
+ from .evaluation_test_case_retrieve_response import (
51
+ EvaluationTestCaseRetrieveResponse as EvaluationTestCaseRetrieveResponse,
52
+ )
53
+ from .evaluation_run_retrieve_results_response import (
54
+ EvaluationRunRetrieveResultsResponse as EvaluationRunRetrieveResultsResponse,
55
+ )
56
+ from .evaluation_test_case_list_evaluation_runs_params import (
57
+ EvaluationTestCaseListEvaluationRunsParams as EvaluationTestCaseListEvaluationRunsParams,
58
+ )
59
+ from .evaluation_test_case_list_evaluation_runs_response import (
60
+ EvaluationTestCaseListEvaluationRunsResponse as EvaluationTestCaseListEvaluationRunsResponse,
61
+ )
62
+ from .evaluation_dataset_create_file_upload_presigned_urls_params import (
63
+ EvaluationDatasetCreateFileUploadPresignedURLsParams as EvaluationDatasetCreateFileUploadPresignedURLsParams,
64
+ )
65
+ from .evaluation_dataset_create_file_upload_presigned_urls_response import (
66
+ EvaluationDatasetCreateFileUploadPresignedURLsResponse as EvaluationDatasetCreateFileUploadPresignedURLsResponse,
67
+ )
@@ -3,7 +3,7 @@
3
3
  from typing import Optional
4
4
  from typing_extensions import Literal
5
5
 
6
- from .._models import BaseModel
6
+ from ..._models import BaseModel
7
7
 
8
8
  __all__ = ["APIEvaluationMetric"]
9
9
 
@@ -2,7 +2,7 @@
2
2
 
3
3
  from typing import Optional
4
4
 
5
- from ...._models import BaseModel
5
+ from ..._models import BaseModel
6
6
 
7
7
  __all__ = ["APIEvaluationMetricResult"]
8
8
 
@@ -2,10 +2,10 @@
2
2
 
3
3
  from typing import List, Optional
4
4
 
5
- from ...._models import BaseModel
5
+ from ..._models import BaseModel
6
6
  from .api_evaluation_metric_result import APIEvaluationMetricResult
7
7
 
8
- __all__ = ["APIPrompt", "PromptChunk"]
8
+ __all__ = ["APIEvaluationPrompt", "PromptChunk"]
9
9
 
10
10
 
11
11
  class PromptChunk(BaseModel):
@@ -25,7 +25,7 @@ class PromptChunk(BaseModel):
25
25
  """Text content of the chunk."""
26
26
 
27
27
 
28
- class APIPrompt(BaseModel):
28
+ class APIEvaluationPrompt(BaseModel):
29
29
  ground_truth: Optional[str] = None
30
30
  """The ground truth for the prompt."""
31
31
 
@@ -4,18 +4,30 @@ from typing import List, Optional
4
4
  from datetime import datetime
5
5
  from typing_extensions import Literal
6
6
 
7
- from ...._models import BaseModel
7
+ from ..._models import BaseModel
8
8
  from .api_evaluation_metric_result import APIEvaluationMetricResult
9
9
 
10
10
  __all__ = ["APIEvaluationRun"]
11
11
 
12
12
 
13
13
  class APIEvaluationRun(BaseModel):
14
+ agent_deleted: Optional[bool] = None
15
+
16
+ agent_name: Optional[str] = None
17
+
14
18
  agent_uuid: Optional[str] = None
15
19
  """Agent UUID."""
16
20
 
17
21
  agent_version_hash: Optional[str] = None
18
22
 
23
+ agent_workspace_uuid: Optional[str] = None
24
+
25
+ created_by_user_email: Optional[str] = None
26
+
27
+ created_by_user_id: Optional[str] = None
28
+
29
+ error_description: Optional[str] = None
30
+
19
31
  evaluation_run_uuid: Optional[str] = None
20
32
  """Evaluation run UUID."""
21
33
 
@@ -5,7 +5,7 @@ from datetime import datetime
5
5
 
6
6
  from ..._models import BaseModel
7
7
  from .api_star_metric import APIStarMetric
8
- from ..api_evaluation_metric import APIEvaluationMetric
8
+ from .api_evaluation_metric import APIEvaluationMetric
9
9
 
10
10
  __all__ = ["APIEvaluationTestCase"]
11
11
 
@@ -2,9 +2,9 @@
2
2
 
3
3
  from typing import List, Optional
4
4
 
5
- from .api_meta import APIMeta
6
5
  from ..._models import BaseModel
7
- from .api_links import APILinks
6
+ from ..shared.api_meta import APIMeta
7
+ from ..shared.api_links import APILinks
8
8
  from ..api_agent_api_key_info import APIAgentAPIKeyInfo
9
9
 
10
10
  __all__ = ["APIKeyListResponse"]
@@ -2,11 +2,11 @@
2
2
 
3
3
  from typing import List, Optional
4
4
 
5
- from .._models import BaseModel
5
+ from ..._models import BaseModel
6
6
  from .api_evaluation_metric import APIEvaluationMetric
7
7
 
8
- __all__ = ["RegionListEvaluationMetricsResponse"]
8
+ __all__ = ["EvaluationMetricListResponse"]
9
9
 
10
10
 
11
- class RegionListEvaluationMetricsResponse(BaseModel):
11
+ class EvaluationMetricListResponse(BaseModel):
12
12
  metrics: Optional[List[APIEvaluationMetric]] = None
@@ -0,0 +1,14 @@
1
+ # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2
+
3
+ from __future__ import annotations
4
+
5
+ from .workspace_create_params import WorkspaceCreateParams as WorkspaceCreateParams
6
+ from .workspace_list_response import WorkspaceListResponse as WorkspaceListResponse
7
+ from .workspace_update_params import WorkspaceUpdateParams as WorkspaceUpdateParams
8
+ from .workspace_create_response import WorkspaceCreateResponse as WorkspaceCreateResponse
9
+ from .workspace_delete_response import WorkspaceDeleteResponse as WorkspaceDeleteResponse
10
+ from .workspace_update_response import WorkspaceUpdateResponse as WorkspaceUpdateResponse
11
+ from .workspace_retrieve_response import WorkspaceRetrieveResponse as WorkspaceRetrieveResponse
12
+ from .workspace_list_evaluation_test_cases_response import (
13
+ WorkspaceListEvaluationTestCasesResponse as WorkspaceListEvaluationTestCasesResponse,
14
+ )
@@ -0,0 +1,16 @@
1
+ # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2
+
3
+ from __future__ import annotations
4
+
5
+ from typing import List
6
+ from typing_extensions import TypedDict
7
+
8
+ __all__ = ["WorkspaceCreateParams"]
9
+
10
+
11
+ class WorkspaceCreateParams(TypedDict, total=False):
12
+ agent_uuids: List[str]
13
+
14
+ description: str
15
+
16
+ name: str
@@ -0,0 +1,16 @@
1
+ # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2
+
3
+ from __future__ import annotations
4
+
5
+ from typing import Optional
6
+
7
+ from ...._models import BaseModel
8
+
9
+ __all__ = ["WorkspaceCreateResponse"]
10
+
11
+
12
+ class WorkspaceCreateResponse(BaseModel):
13
+ workspace: Optional["APIWorkspace"] = None
14
+
15
+
16
+ from ...api_workspace import APIWorkspace
@@ -0,0 +1,11 @@
1
+ # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2
+
3
+ from typing import Optional
4
+
5
+ from ...._models import BaseModel
6
+
7
+ __all__ = ["WorkspaceDeleteResponse"]
8
+
9
+
10
+ class WorkspaceDeleteResponse(BaseModel):
11
+ workspace_uuid: Optional[str] = None
@@ -0,0 +1,12 @@
1
+ # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2
+
3
+ from typing import List, Optional
4
+
5
+ from ...._models import BaseModel
6
+ from ..api_evaluation_test_case import APIEvaluationTestCase
7
+
8
+ __all__ = ["WorkspaceListEvaluationTestCasesResponse"]
9
+
10
+
11
+ class WorkspaceListEvaluationTestCasesResponse(BaseModel):
12
+ evaluation_test_cases: Optional[List[APIEvaluationTestCase]] = None
@@ -0,0 +1,16 @@
1
+ # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2
+
3
+ from __future__ import annotations
4
+
5
+ from typing import List, Optional
6
+
7
+ from ...._models import BaseModel
8
+
9
+ __all__ = ["WorkspaceListResponse"]
10
+
11
+
12
+ class WorkspaceListResponse(BaseModel):
13
+ workspaces: Optional[List["APIWorkspace"]] = None
14
+
15
+
16
+ from ...api_workspace import APIWorkspace
@@ -0,0 +1,16 @@
1
+ # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2
+
3
+ from __future__ import annotations
4
+
5
+ from typing import Optional
6
+
7
+ from ...._models import BaseModel
8
+
9
+ __all__ = ["WorkspaceRetrieveResponse"]
10
+
11
+
12
+ class WorkspaceRetrieveResponse(BaseModel):
13
+ workspace: Optional["APIWorkspace"] = None
14
+
15
+
16
+ from ...api_workspace import APIWorkspace
@@ -0,0 +1,18 @@
1
+ # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2
+
3
+ from __future__ import annotations
4
+
5
+ from typing_extensions import Annotated, TypedDict
6
+
7
+ from ...._utils import PropertyInfo
8
+
9
+ __all__ = ["WorkspaceUpdateParams"]
10
+
11
+
12
+ class WorkspaceUpdateParams(TypedDict, total=False):
13
+ description: str
14
+
15
+ name: str
16
+
17
+ body_workspace_uuid: Annotated[str, PropertyInfo(alias="workspace_uuid")]
18
+ """Workspace UUID."""
@@ -0,0 +1,16 @@
1
+ # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2
+
3
+ from __future__ import annotations
4
+
5
+ from typing import Optional
6
+
7
+ from ...._models import BaseModel
8
+
9
+ __all__ = ["WorkspaceUpdateResponse"]
10
+
11
+
12
+ class WorkspaceUpdateResponse(BaseModel):
13
+ workspace: Optional["APIWorkspace"] = None
14
+
15
+
16
+ from ...api_workspace import APIWorkspace
@@ -0,0 +1,8 @@
1
+ # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2
+
3
+ from __future__ import annotations
4
+
5
+ from .agent_list_params import AgentListParams as AgentListParams
6
+ from .agent_move_params import AgentMoveParams as AgentMoveParams
7
+ from .agent_list_response import AgentListResponse as AgentListResponse
8
+ from .agent_move_response import AgentMoveResponse as AgentMoveResponse
@@ -0,0 +1,26 @@
1
+ # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2
+
3
+ from __future__ import annotations
4
+
5
+ from typing import List
6
+ from typing_extensions import TypedDict
7
+
8
+ __all__ = ["AgentListParams", "FieldMask"]
9
+
10
+
11
+ class AgentListParams(TypedDict, total=False):
12
+ field_mask: FieldMask
13
+
14
+ only_deployed: bool
15
+ """Only list agents that are deployed."""
16
+
17
+ page: int
18
+ """page number."""
19
+
20
+ per_page: int
21
+ """items per page."""
22
+
23
+
24
+ class FieldMask(TypedDict, total=False):
25
+ paths: List[str]
26
+ """The set of field mask paths."""
@@ -0,0 +1,22 @@
1
+ # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2
+
3
+ from __future__ import annotations
4
+
5
+ from typing import List, Optional
6
+
7
+ from ....._models import BaseModel
8
+ from ....shared.api_meta import APIMeta
9
+ from ....shared.api_links import APILinks
10
+
11
+ __all__ = ["AgentListResponse"]
12
+
13
+
14
+ class AgentListResponse(BaseModel):
15
+ agents: Optional[List["APIAgent"]] = None
16
+
17
+ links: Optional[APILinks] = None
18
+
19
+ meta: Optional[APIMeta] = None
20
+
21
+
22
+ from ....api_agent import APIAgent
@@ -0,0 +1,16 @@
1
+ # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2
+
3
+ from __future__ import annotations
4
+
5
+ from typing import List
6
+ from typing_extensions import Annotated, TypedDict
7
+
8
+ from ....._utils import PropertyInfo
9
+
10
+ __all__ = ["AgentMoveParams"]
11
+
12
+
13
+ class AgentMoveParams(TypedDict, total=False):
14
+ agent_uuids: List[str]
15
+
16
+ body_workspace_uuid: Annotated[str, PropertyInfo(alias="workspace_uuid")]
@@ -0,0 +1,16 @@
1
+ # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2
+
3
+ from __future__ import annotations
4
+
5
+ from typing import Optional
6
+
7
+ from ....._models import BaseModel
8
+
9
+ __all__ = ["AgentMoveResponse"]
10
+
11
+
12
+ class AgentMoveResponse(BaseModel):
13
+ workspace: Optional["APIWorkspace"] = None
14
+
15
+
16
+ from ....api_workspace import APIWorkspace
@@ -2,14 +2,15 @@
2
2
 
3
3
  from __future__ import annotations
4
4
 
5
+ from typing import List
5
6
  from typing_extensions import TypedDict
6
7
 
7
8
  __all__ = ["EvaluationRunCreateParams"]
8
9
 
9
10
 
10
11
  class EvaluationRunCreateParams(TypedDict, total=False):
11
- agent_uuid: str
12
- """Agent UUID to run the test case against."""
12
+ agent_uuids: List[str]
13
+ """Agent UUIDs to run the test case against."""
13
14
 
14
15
  run_name: str
15
16
  """The name of the run."""
@@ -1,6 +1,6 @@
1
1
  # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2
2
 
3
- from typing import Optional
3
+ from typing import List, Optional
4
4
 
5
5
  from ..._models import BaseModel
6
6
 
@@ -8,4 +8,4 @@ __all__ = ["EvaluationRunCreateResponse"]
8
8
 
9
9
 
10
10
  class EvaluationRunCreateResponse(BaseModel):
11
- evaluation_run_uuid: Optional[str] = None
11
+ evaluation_run_uuids: Optional[List[str]] = None
@@ -2,15 +2,15 @@
2
2
 
3
3
  from typing import List, Optional
4
4
 
5
- from ...._models import BaseModel
6
- from .api_prompt import APIPrompt
5
+ from ..._models import BaseModel
7
6
  from .api_evaluation_run import APIEvaluationRun
7
+ from .api_evaluation_prompt import APIEvaluationPrompt
8
8
 
9
- __all__ = ["ResultRetrieveResponse"]
9
+ __all__ = ["EvaluationRunListResultsResponse"]
10
10
 
11
11
 
12
- class ResultRetrieveResponse(BaseModel):
12
+ class EvaluationRunListResultsResponse(BaseModel):
13
13
  evaluation_run: Optional[APIEvaluationRun] = None
14
14
 
15
- prompts: Optional[List[APIPrompt]] = None
15
+ prompts: Optional[List[APIEvaluationPrompt]] = None
16
16
  """The prompt level results."""
@@ -3,7 +3,7 @@
3
3
  from typing import Optional
4
4
 
5
5
  from ..._models import BaseModel
6
- from .evaluation_runs.api_evaluation_run import APIEvaluationRun
6
+ from .api_evaluation_run import APIEvaluationRun
7
7
 
8
8
  __all__ = ["EvaluationRunRetrieveResponse"]
9
9
 
@@ -0,0 +1,12 @@
1
+ # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2
+
3
+ from typing import Optional
4
+
5
+ from ..._models import BaseModel
6
+ from .api_evaluation_prompt import APIEvaluationPrompt
7
+
8
+ __all__ = ["EvaluationRunRetrieveResultsResponse"]
9
+
10
+
11
+ class EvaluationRunRetrieveResultsResponse(BaseModel):
12
+ prompt: Optional[APIEvaluationPrompt] = None
@@ -3,7 +3,7 @@
3
3
  from typing import List, Optional
4
4
 
5
5
  from ..._models import BaseModel
6
- from .evaluation_runs.api_evaluation_run import APIEvaluationRun
6
+ from .api_evaluation_run import APIEvaluationRun
7
7
 
8
8
  __all__ = ["EvaluationTestCaseListEvaluationRunsResponse"]
9
9
 
@@ -0,0 +1,12 @@
1
+ # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2
+
3
+ from __future__ import annotations
4
+
5
+ from typing_extensions import TypedDict
6
+
7
+ __all__ = ["EvaluationTestCaseRetrieveParams"]
8
+
9
+
10
+ class EvaluationTestCaseRetrieveParams(TypedDict, total=False):
11
+ evaluation_test_case_version: int
12
+ """Version of the test case."""
@@ -6,10 +6,10 @@ from typing_extensions import Required, Annotated, TypedDict
6
6
 
7
7
  from ..._utils import PropertyInfo
8
8
 
9
- __all__ = ["ChildAgentAddParams"]
9
+ __all__ = ["RouteAddParams"]
10
10
 
11
11
 
12
- class ChildAgentAddParams(TypedDict, total=False):
12
+ class RouteAddParams(TypedDict, total=False):
13
13
  path_parent_agent_uuid: Required[Annotated[str, PropertyInfo(alias="parent_agent_uuid")]]
14
14
 
15
15
  body_child_agent_uuid: Annotated[str, PropertyInfo(alias="child_agent_uuid")]
@@ -4,10 +4,10 @@ from typing import Optional
4
4
 
5
5
  from ..._models import BaseModel
6
6
 
7
- __all__ = ["ChildAgentAddResponse"]
7
+ __all__ = ["RouteAddResponse"]
8
8
 
9
9
 
10
- class ChildAgentAddResponse(BaseModel):
10
+ class RouteAddResponse(BaseModel):
11
11
  child_agent_uuid: Optional[str] = None
12
12
 
13
13
  parent_agent_uuid: Optional[str] = None
@@ -4,10 +4,10 @@ from typing import Optional
4
4
 
5
5
  from ..._models import BaseModel
6
6
 
7
- __all__ = ["ChildAgentDeleteResponse"]
7
+ __all__ = ["RouteDeleteResponse"]
8
8
 
9
9
 
10
- class ChildAgentDeleteResponse(BaseModel):
10
+ class RouteDeleteResponse(BaseModel):
11
11
  child_agent_uuid: Optional[str] = None
12
12
 
13
13
  parent_agent_uuid: Optional[str] = None