adaptive-sdk 0.1.8__py3-none-any.whl → 0.1.10__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.
- adaptive_sdk/graphql_client/__init__.py +6 -3
- adaptive_sdk/graphql_client/async_client.py +50 -26
- adaptive_sdk/graphql_client/client.py +50 -26
- adaptive_sdk/graphql_client/create_dataset_from_multipart_upload.py +18 -0
- adaptive_sdk/graphql_client/custom_fields.py +24 -2
- adaptive_sdk/graphql_client/custom_queries.py +14 -2
- adaptive_sdk/graphql_client/custom_typing_fields.py +7 -0
- adaptive_sdk/graphql_client/dataset_upload_processing_status.py +18 -0
- adaptive_sdk/graphql_client/enums.py +1 -4
- adaptive_sdk/graphql_client/fragments.py +12 -0
- adaptive_sdk/graphql_client/input_types.py +4 -13
- adaptive_sdk/graphql_client/resize_inference_partition.py +6 -0
- adaptive_sdk/resources/compute_pools.py +63 -2
- adaptive_sdk/resources/datasets.py +82 -122
- adaptive_sdk/resources/models.py +0 -8
- {adaptive_sdk-0.1.8.dist-info → adaptive_sdk-0.1.10.dist-info}/METADATA +2 -2
- {adaptive_sdk-0.1.8.dist-info → adaptive_sdk-0.1.10.dist-info}/RECORD +18 -15
- {adaptive_sdk-0.1.8.dist-info → adaptive_sdk-0.1.10.dist-info}/WHEEL +0 -0
|
@@ -12,6 +12,7 @@ from .cancel_ab_campaign import CancelABCampaign
|
|
|
12
12
|
from .cancel_job import CancelJob
|
|
13
13
|
from .create_ab_campaign import CreateAbCampaign
|
|
14
14
|
from .create_custom_recipe import CreateCustomRecipe
|
|
15
|
+
from .create_dataset_from_multipart_upload import CreateDatasetFromMultipartUpload
|
|
15
16
|
from .create_grader import CreateGrader
|
|
16
17
|
from .create_job import CreateJob
|
|
17
18
|
from .create_judge import CreateJudge
|
|
@@ -21,6 +22,7 @@ from .create_role import CreateRole
|
|
|
21
22
|
from .create_team import CreateTeam
|
|
22
23
|
from .create_use_case import CreateUseCase
|
|
23
24
|
from .create_user import CreateUser
|
|
25
|
+
from .dataset_upload_processing_status import DatasetUploadProcessingStatus
|
|
24
26
|
from .delete_custom_recipe import DeleteCustomRecipe
|
|
25
27
|
from .delete_dataset import DeleteDataset
|
|
26
28
|
from .delete_grader import DeleteGrader
|
|
@@ -40,7 +42,7 @@ from .enums import CompletionGroupBy
|
|
|
40
42
|
from .get_custom_recipe import GetCustomRecipe
|
|
41
43
|
from .get_grader import GetGrader
|
|
42
44
|
from .get_judge import GetJudge
|
|
43
|
-
from .input_types import AbcampaignCreate, AbCampaignFilter, AddExternalModelInput, AddHFModelInput, AddModelInput, AttachModel, CreateRecipeInput, CursorPageInput, CustomRecipeFilterInput, DatasetCreate, GraderCreateInput, GraderUpdateInput, JobInput, JudgeCreate, JudgeUpdate, ListCompletionsFilterInput, ListJobsFilterInput, MetricCreate, MetricLink, MetricUnlink, ModelComputeConfigInput, ModelFilter, ModelPlacementInput, OrderPair, PrebuiltJudgeCreate, RemoteEnvCreate, RoleCreate, TeamCreate, TeamMemberRemove, TeamMemberSet, UpdateModelService, UpdateRecipeInput, UseCaseCreate, UseCaseShares, UserCreate
|
|
45
|
+
from .input_types import AbcampaignCreate, AbCampaignFilter, AddExternalModelInput, AddHFModelInput, AddModelInput, AttachModel, CreateRecipeInput, CursorPageInput, CustomRecipeFilterInput, DatasetCreate, DatasetCreateFromMultipartUpload, DatasetUploadProcessingStatusInput, GraderCreateInput, GraderUpdateInput, JobInput, JudgeCreate, JudgeUpdate, ListCompletionsFilterInput, ListJobsFilterInput, MetricCreate, MetricLink, MetricUnlink, ModelComputeConfigInput, ModelFilter, ModelPlacementInput, OrderPair, PrebuiltJudgeCreate, RemoteEnvCreate, ResizePartitionInput, RoleCreate, TeamCreate, TeamMemberRemove, TeamMemberSet, UpdateModelService, UpdateRecipeInput, UseCaseCreate, UseCaseShares, UserCreate
|
|
44
46
|
from .link_metric import LinkMetric
|
|
45
47
|
from .list_ab_campaigns import ListAbCampaigns
|
|
46
48
|
from .list_compute_pools import ListComputePools
|
|
@@ -66,6 +68,7 @@ from .lock_grader import LockGrader
|
|
|
66
68
|
from .me import Me
|
|
67
69
|
from .remove_remote_env import RemoveRemoteEnv
|
|
68
70
|
from .remove_team_member import RemoveTeamMember
|
|
71
|
+
from .resize_inference_partition import ResizeInferencePartition
|
|
69
72
|
from .share_use_case import ShareUseCase
|
|
70
73
|
from .terminate_model import TerminateModel
|
|
71
74
|
from .test_remote_env import TestRemoteEnv
|
|
@@ -105,28 +108,28 @@ class GQLClient(BaseClientOpenTelemetry):
|
|
|
105
108
|
return UnlinkMetric.model_validate(data)
|
|
106
109
|
|
|
107
110
|
def attach_model_to_use_case(self, input: AttachModel, **kwargs: Any) -> AttachModelToUseCase:
|
|
108
|
-
query = gql('\n mutation AttachModelToUseCase($input: AttachModel!) {\n attachModel(input: $input) {\n ...ModelServiceData\n }\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n\n fragment ModelServiceData on ModelService {\n id\n key\n name\n model {\n ...ModelData\n backbone {\n ...ModelData\n }\n }\n attached\n isDefault\n desiredOnline\n createdAt\n }\n ')
|
|
111
|
+
query = gql('\n mutation AttachModelToUseCase($input: AttachModel!) {\n attachModel(input: $input) {\n ...ModelServiceData\n }\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n error\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n\n fragment ModelServiceData on ModelService {\n id\n key\n name\n model {\n ...ModelData\n backbone {\n ...ModelData\n }\n }\n attached\n isDefault\n desiredOnline\n createdAt\n }\n ')
|
|
109
112
|
variables: Dict[str, object] = {'input': input}
|
|
110
113
|
response = self.execute(query=query, operation_name='AttachModelToUseCase', variables=variables, **kwargs)
|
|
111
114
|
data = self.get_data(response)
|
|
112
115
|
return AttachModelToUseCase.model_validate(data)
|
|
113
116
|
|
|
114
117
|
def add_external_model(self, input: AddExternalModelInput, **kwargs: Any) -> AddExternalModel:
|
|
115
|
-
query = gql('\n mutation AddExternalModel($input: AddExternalModelInput!) {\n addExternalModel(input: $input) {\n ...ModelData\n }\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n ')
|
|
118
|
+
query = gql('\n mutation AddExternalModel($input: AddExternalModelInput!) {\n addExternalModel(input: $input) {\n ...ModelData\n }\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n error\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n ')
|
|
116
119
|
variables: Dict[str, object] = {'input': input}
|
|
117
120
|
response = self.execute(query=query, operation_name='AddExternalModel', variables=variables, **kwargs)
|
|
118
121
|
data = self.get_data(response)
|
|
119
122
|
return AddExternalModel.model_validate(data)
|
|
120
123
|
|
|
121
124
|
def add_model(self, input: AddModelInput, **kwargs: Any) -> AddModel:
|
|
122
|
-
query = gql('\n mutation AddModel($input: AddModelInput!) {\n addModel(input: $input) {\n ...ModelData\n backbone {\n ...ModelData\n }\n }\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n ')
|
|
125
|
+
query = gql('\n mutation AddModel($input: AddModelInput!) {\n addModel(input: $input) {\n ...ModelData\n backbone {\n ...ModelData\n }\n }\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n error\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n ')
|
|
123
126
|
variables: Dict[str, object] = {'input': input}
|
|
124
127
|
response = self.execute(query=query, operation_name='AddModel', variables=variables, **kwargs)
|
|
125
128
|
data = self.get_data(response)
|
|
126
129
|
return AddModel.model_validate(data)
|
|
127
130
|
|
|
128
131
|
def update_model(self, input: UpdateModelService, **kwargs: Any) -> UpdateModel:
|
|
129
|
-
query = gql('\n mutation UpdateModel($input: UpdateModelService!) {\n updateModelService(input: $input) {\n ...ModelServiceData\n }\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n\n fragment ModelServiceData on ModelService {\n id\n key\n name\n model {\n ...ModelData\n backbone {\n ...ModelData\n }\n }\n attached\n isDefault\n desiredOnline\n createdAt\n }\n ')
|
|
132
|
+
query = gql('\n mutation UpdateModel($input: UpdateModelService!) {\n updateModelService(input: $input) {\n ...ModelServiceData\n }\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n error\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n\n fragment ModelServiceData on ModelService {\n id\n key\n name\n model {\n ...ModelData\n backbone {\n ...ModelData\n }\n }\n attached\n isDefault\n desiredOnline\n createdAt\n }\n ')
|
|
130
133
|
variables: Dict[str, object] = {'input': input}
|
|
131
134
|
response = self.execute(query=query, operation_name='UpdateModel', variables=variables, **kwargs)
|
|
132
135
|
data = self.get_data(response)
|
|
@@ -147,14 +150,14 @@ class GQLClient(BaseClientOpenTelemetry):
|
|
|
147
150
|
return TerminateModel.model_validate(data)
|
|
148
151
|
|
|
149
152
|
def create_use_case(self, input: UseCaseCreate, **kwargs: Any) -> CreateUseCase:
|
|
150
|
-
query = gql('\n mutation CreateUseCase($input: UseCaseCreate!) {\n createUseCase(input: $input) {\n ...UseCaseData\n }\n }\n\n fragment MetricWithContextData on MetricWithContext {\n id\n key\n name\n kind\n description\n scoringType\n createdAt\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n\n fragment ModelServiceData on ModelService {\n id\n key\n name\n model {\n ...ModelData\n backbone {\n ...ModelData\n }\n }\n attached\n isDefault\n desiredOnline\n createdAt\n }\n\n fragment UseCaseData on UseCase {\n id\n key\n name\n description\n createdAt\n metrics {\n ...MetricWithContextData\n }\n modelServices {\n ...ModelServiceData\n }\n permissions\n shares {\n team {\n id\n key\n name\n createdAt\n }\n role {\n id\n key\n name\n createdAt\n permissions\n }\n isOwner\n }\n }\n ')
|
|
153
|
+
query = gql('\n mutation CreateUseCase($input: UseCaseCreate!) {\n createUseCase(input: $input) {\n ...UseCaseData\n }\n }\n\n fragment MetricWithContextData on MetricWithContext {\n id\n key\n name\n kind\n description\n scoringType\n createdAt\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n error\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n\n fragment ModelServiceData on ModelService {\n id\n key\n name\n model {\n ...ModelData\n backbone {\n ...ModelData\n }\n }\n attached\n isDefault\n desiredOnline\n createdAt\n }\n\n fragment UseCaseData on UseCase {\n id\n key\n name\n description\n createdAt\n metrics {\n ...MetricWithContextData\n }\n modelServices {\n ...ModelServiceData\n }\n permissions\n shares {\n team {\n id\n key\n name\n createdAt\n }\n role {\n id\n key\n name\n createdAt\n permissions\n }\n isOwner\n }\n }\n ')
|
|
151
154
|
variables: Dict[str, object] = {'input': input}
|
|
152
155
|
response = self.execute(query=query, operation_name='CreateUseCase', variables=variables, **kwargs)
|
|
153
156
|
data = self.get_data(response)
|
|
154
157
|
return CreateUseCase.model_validate(data)
|
|
155
158
|
|
|
156
159
|
def share_use_case(self, id_or_key: str, input: UseCaseShares, **kwargs: Any) -> ShareUseCase:
|
|
157
|
-
query = gql('\n mutation ShareUseCase($idOrKey: IdOrKey!, $input: UseCaseShares!) {\n shareUseCase(idOrKey: $idOrKey, input: $input) {\n ...UseCaseData\n }\n }\n\n fragment MetricWithContextData on MetricWithContext {\n id\n key\n name\n kind\n description\n scoringType\n createdAt\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n\n fragment ModelServiceData on ModelService {\n id\n key\n name\n model {\n ...ModelData\n backbone {\n ...ModelData\n }\n }\n attached\n isDefault\n desiredOnline\n createdAt\n }\n\n fragment UseCaseData on UseCase {\n id\n key\n name\n description\n createdAt\n metrics {\n ...MetricWithContextData\n }\n modelServices {\n ...ModelServiceData\n }\n permissions\n shares {\n team {\n id\n key\n name\n createdAt\n }\n role {\n id\n key\n name\n createdAt\n permissions\n }\n isOwner\n }\n }\n ')
|
|
160
|
+
query = gql('\n mutation ShareUseCase($idOrKey: IdOrKey!, $input: UseCaseShares!) {\n shareUseCase(idOrKey: $idOrKey, input: $input) {\n ...UseCaseData\n }\n }\n\n fragment MetricWithContextData on MetricWithContext {\n id\n key\n name\n kind\n description\n scoringType\n createdAt\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n error\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n\n fragment ModelServiceData on ModelService {\n id\n key\n name\n model {\n ...ModelData\n backbone {\n ...ModelData\n }\n }\n attached\n isDefault\n desiredOnline\n createdAt\n }\n\n fragment UseCaseData on UseCase {\n id\n key\n name\n description\n createdAt\n metrics {\n ...MetricWithContextData\n }\n modelServices {\n ...ModelServiceData\n }\n permissions\n shares {\n team {\n id\n key\n name\n createdAt\n }\n role {\n id\n key\n name\n createdAt\n permissions\n }\n isOwner\n }\n }\n ')
|
|
158
161
|
variables: Dict[str, object] = {'idOrKey': id_or_key, 'input': input}
|
|
159
162
|
response = self.execute(query=query, operation_name='ShareUseCase', variables=variables, **kwargs)
|
|
160
163
|
data = self.get_data(response)
|
|
@@ -231,7 +234,7 @@ class GQLClient(BaseClientOpenTelemetry):
|
|
|
231
234
|
return CreateTeam.model_validate(data)
|
|
232
235
|
|
|
233
236
|
def update_model_compute_config(self, id_or_key: str, input: ModelComputeConfigInput, **kwargs: Any) -> UpdateModelComputeConfig:
|
|
234
|
-
query = gql('\n mutation UpdateModelComputeConfig($idOrKey: IdOrKey!, $input: ModelComputeConfigInput!) {\n updateModelComputeConfig(idOrKey: $idOrKey, input: $input) {\n ...ModelData\n }\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n ')
|
|
237
|
+
query = gql('\n mutation UpdateModelComputeConfig($idOrKey: IdOrKey!, $input: ModelComputeConfigInput!) {\n updateModelComputeConfig(idOrKey: $idOrKey, input: $input) {\n ...ModelData\n }\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n error\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n ')
|
|
235
238
|
variables: Dict[str, object] = {'idOrKey': id_or_key, 'input': input}
|
|
236
239
|
response = self.execute(query=query, operation_name='UpdateModelComputeConfig', variables=variables, **kwargs)
|
|
237
240
|
data = self.get_data(response)
|
|
@@ -280,21 +283,21 @@ class GQLClient(BaseClientOpenTelemetry):
|
|
|
280
283
|
return DeleteCustomRecipe.model_validate(data)
|
|
281
284
|
|
|
282
285
|
def create_judge(self, use_case: str, input: JudgeCreate, **kwargs: Any) -> CreateJudge:
|
|
283
|
-
query = gql('\n mutation CreateJudge($useCase: IdOrKey!, $input: JudgeCreate!) {\n createJudge(useCase: $useCase, input: $input) {\n ...JudgeData\n }\n }\n\n fragment JudgeData on Judge {\n id\n key\n version\n name\n criteria\n prebuilt\n examples {\n input {\n role\n content\n }\n output\n pass\n reasoning\n }\n capabilities\n model {\n ...ModelData\n }\n useCaseId\n metric {\n ...MetricData\n }\n createdAt\n updatedAt\n }\n\n fragment MetricData on Metric {\n id\n key\n name\n kind\n description\n scoringType\n createdAt\n hasDirectFeedbacks\n hasComparisonFeedbacks\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n ')
|
|
286
|
+
query = gql('\n mutation CreateJudge($useCase: IdOrKey!, $input: JudgeCreate!) {\n createJudge(useCase: $useCase, input: $input) {\n ...JudgeData\n }\n }\n\n fragment JudgeData on Judge {\n id\n key\n version\n name\n criteria\n prebuilt\n examples {\n input {\n role\n content\n }\n output\n pass\n reasoning\n }\n capabilities\n model {\n ...ModelData\n }\n useCaseId\n metric {\n ...MetricData\n }\n createdAt\n updatedAt\n }\n\n fragment MetricData on Metric {\n id\n key\n name\n kind\n description\n scoringType\n createdAt\n hasDirectFeedbacks\n hasComparisonFeedbacks\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n error\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n ')
|
|
284
287
|
variables: Dict[str, object] = {'useCase': use_case, 'input': input}
|
|
285
288
|
response = self.execute(query=query, operation_name='CreateJudge', variables=variables, **kwargs)
|
|
286
289
|
data = self.get_data(response)
|
|
287
290
|
return CreateJudge.model_validate(data)
|
|
288
291
|
|
|
289
292
|
def create_prebuilt_judge(self, use_case: str, input: PrebuiltJudgeCreate, **kwargs: Any) -> CreatePrebuiltJudge:
|
|
290
|
-
query = gql('\n mutation CreatePrebuiltJudge($useCase: IdOrKey!, $input: PrebuiltJudgeCreate!) {\n createPrebuiltJudge(useCase: $useCase, input: $input) {\n ...JudgeData\n }\n }\n\n fragment JudgeData on Judge {\n id\n key\n version\n name\n criteria\n prebuilt\n examples {\n input {\n role\n content\n }\n output\n pass\n reasoning\n }\n capabilities\n model {\n ...ModelData\n }\n useCaseId\n metric {\n ...MetricData\n }\n createdAt\n updatedAt\n }\n\n fragment MetricData on Metric {\n id\n key\n name\n kind\n description\n scoringType\n createdAt\n hasDirectFeedbacks\n hasComparisonFeedbacks\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n ')
|
|
293
|
+
query = gql('\n mutation CreatePrebuiltJudge($useCase: IdOrKey!, $input: PrebuiltJudgeCreate!) {\n createPrebuiltJudge(useCase: $useCase, input: $input) {\n ...JudgeData\n }\n }\n\n fragment JudgeData on Judge {\n id\n key\n version\n name\n criteria\n prebuilt\n examples {\n input {\n role\n content\n }\n output\n pass\n reasoning\n }\n capabilities\n model {\n ...ModelData\n }\n useCaseId\n metric {\n ...MetricData\n }\n createdAt\n updatedAt\n }\n\n fragment MetricData on Metric {\n id\n key\n name\n kind\n description\n scoringType\n createdAt\n hasDirectFeedbacks\n hasComparisonFeedbacks\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n error\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n ')
|
|
291
294
|
variables: Dict[str, object] = {'useCase': use_case, 'input': input}
|
|
292
295
|
response = self.execute(query=query, operation_name='CreatePrebuiltJudge', variables=variables, **kwargs)
|
|
293
296
|
data = self.get_data(response)
|
|
294
297
|
return CreatePrebuiltJudge.model_validate(data)
|
|
295
298
|
|
|
296
299
|
def update_judge(self, use_case: str, key: str, input: JudgeUpdate, **kwargs: Any) -> UpdateJudge:
|
|
297
|
-
query = gql('\n mutation UpdateJudge($useCase: IdOrKey!, $key: String!, $input: JudgeUpdate!) {\n updateJudge(useCase: $useCase, key: $key, input: $input) {\n ...JudgeData\n }\n }\n\n fragment JudgeData on Judge {\n id\n key\n version\n name\n criteria\n prebuilt\n examples {\n input {\n role\n content\n }\n output\n pass\n reasoning\n }\n capabilities\n model {\n ...ModelData\n }\n useCaseId\n metric {\n ...MetricData\n }\n createdAt\n updatedAt\n }\n\n fragment MetricData on Metric {\n id\n key\n name\n kind\n description\n scoringType\n createdAt\n hasDirectFeedbacks\n hasComparisonFeedbacks\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n ')
|
|
300
|
+
query = gql('\n mutation UpdateJudge($useCase: IdOrKey!, $key: String!, $input: JudgeUpdate!) {\n updateJudge(useCase: $useCase, key: $key, input: $input) {\n ...JudgeData\n }\n }\n\n fragment JudgeData on Judge {\n id\n key\n version\n name\n criteria\n prebuilt\n examples {\n input {\n role\n content\n }\n output\n pass\n reasoning\n }\n capabilities\n model {\n ...ModelData\n }\n useCaseId\n metric {\n ...MetricData\n }\n createdAt\n updatedAt\n }\n\n fragment MetricData on Metric {\n id\n key\n name\n kind\n description\n scoringType\n createdAt\n hasDirectFeedbacks\n hasComparisonFeedbacks\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n error\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n ')
|
|
298
301
|
variables: Dict[str, object] = {'useCase': use_case, 'key': key, 'input': input}
|
|
299
302
|
response = self.execute(query=query, operation_name='UpdateJudge', variables=variables, **kwargs)
|
|
300
303
|
data = self.get_data(response)
|
|
@@ -308,14 +311,14 @@ class GQLClient(BaseClientOpenTelemetry):
|
|
|
308
311
|
return DeleteJudge.model_validate(data)
|
|
309
312
|
|
|
310
313
|
def create_grader(self, use_case: str, input: GraderCreateInput, **kwargs: Any) -> CreateGrader:
|
|
311
|
-
query = gql('\n mutation CreateGrader($useCase: IdOrKey!, $input: GraderCreateInput!) {\n createGrader(useCase: $useCase, input: $input) {\n ...GraderData\n }\n }\n\n fragment GraderData on Grader {\n id\n name\n key\n locked\n graderType\n graderConfig {\n __typename\n ... on JudgeConfigOutput {\n judgeCriteria: criteria\n examples {\n input {\n role\n content\n }\n output\n pass\n reasoning\n }\n model {\n ...ModelData\n }\n }\n ... on PrebuiltConfigOutput {\n prebuiltCriteria: criteria {\n key\n name\n feedbackKey\n description\n }\n model {\n ...ModelData\n }\n }\n ... on RemoteConfigOutput {\n url\n version\n description\n }\n ... on CustomConfigOutput {\n graderDescription: description\n }\n }\n useCase {\n id\n key\n name\n }\n metric {\n ...MetricData\n }\n createdAt\n updatedAt\n }\n\n fragment MetricData on Metric {\n id\n key\n name\n kind\n description\n scoringType\n createdAt\n hasDirectFeedbacks\n hasComparisonFeedbacks\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n ')
|
|
314
|
+
query = gql('\n mutation CreateGrader($useCase: IdOrKey!, $input: GraderCreateInput!) {\n createGrader(useCase: $useCase, input: $input) {\n ...GraderData\n }\n }\n\n fragment GraderData on Grader {\n id\n name\n key\n locked\n graderType\n graderConfig {\n __typename\n ... on JudgeConfigOutput {\n judgeCriteria: criteria\n examples {\n input {\n role\n content\n }\n output\n pass\n reasoning\n }\n model {\n ...ModelData\n }\n }\n ... on PrebuiltConfigOutput {\n prebuiltCriteria: criteria {\n key\n name\n feedbackKey\n description\n }\n model {\n ...ModelData\n }\n }\n ... on RemoteConfigOutput {\n url\n version\n description\n }\n ... on CustomConfigOutput {\n graderDescription: description\n }\n }\n useCase {\n id\n key\n name\n }\n metric {\n ...MetricData\n }\n createdAt\n updatedAt\n }\n\n fragment MetricData on Metric {\n id\n key\n name\n kind\n description\n scoringType\n createdAt\n hasDirectFeedbacks\n hasComparisonFeedbacks\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n error\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n ')
|
|
312
315
|
variables: Dict[str, object] = {'useCase': use_case, 'input': input}
|
|
313
316
|
response = self.execute(query=query, operation_name='CreateGrader', variables=variables, **kwargs)
|
|
314
317
|
data = self.get_data(response)
|
|
315
318
|
return CreateGrader.model_validate(data)
|
|
316
319
|
|
|
317
320
|
def update_grader(self, use_case: str, id: str, input: GraderUpdateInput, **kwargs: Any) -> UpdateGrader:
|
|
318
|
-
query = gql('\n mutation UpdateGrader($useCase: IdOrKey!, $id: IdOrKey!, $input: GraderUpdateInput!) {\n updateGrader(useCase: $useCase, id: $id, input: $input) {\n ...GraderData\n }\n }\n\n fragment GraderData on Grader {\n id\n name\n key\n locked\n graderType\n graderConfig {\n __typename\n ... on JudgeConfigOutput {\n judgeCriteria: criteria\n examples {\n input {\n role\n content\n }\n output\n pass\n reasoning\n }\n model {\n ...ModelData\n }\n }\n ... on PrebuiltConfigOutput {\n prebuiltCriteria: criteria {\n key\n name\n feedbackKey\n description\n }\n model {\n ...ModelData\n }\n }\n ... on RemoteConfigOutput {\n url\n version\n description\n }\n ... on CustomConfigOutput {\n graderDescription: description\n }\n }\n useCase {\n id\n key\n name\n }\n metric {\n ...MetricData\n }\n createdAt\n updatedAt\n }\n\n fragment MetricData on Metric {\n id\n key\n name\n kind\n description\n scoringType\n createdAt\n hasDirectFeedbacks\n hasComparisonFeedbacks\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n ')
|
|
321
|
+
query = gql('\n mutation UpdateGrader($useCase: IdOrKey!, $id: IdOrKey!, $input: GraderUpdateInput!) {\n updateGrader(useCase: $useCase, id: $id, input: $input) {\n ...GraderData\n }\n }\n\n fragment GraderData on Grader {\n id\n name\n key\n locked\n graderType\n graderConfig {\n __typename\n ... on JudgeConfigOutput {\n judgeCriteria: criteria\n examples {\n input {\n role\n content\n }\n output\n pass\n reasoning\n }\n model {\n ...ModelData\n }\n }\n ... on PrebuiltConfigOutput {\n prebuiltCriteria: criteria {\n key\n name\n feedbackKey\n description\n }\n model {\n ...ModelData\n }\n }\n ... on RemoteConfigOutput {\n url\n version\n description\n }\n ... on CustomConfigOutput {\n graderDescription: description\n }\n }\n useCase {\n id\n key\n name\n }\n metric {\n ...MetricData\n }\n createdAt\n updatedAt\n }\n\n fragment MetricData on Metric {\n id\n key\n name\n kind\n description\n scoringType\n createdAt\n hasDirectFeedbacks\n hasComparisonFeedbacks\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n error\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n ')
|
|
319
322
|
variables: Dict[str, object] = {'useCase': use_case, 'id': id, 'input': input}
|
|
320
323
|
response = self.execute(query=query, operation_name='UpdateGrader', variables=variables, **kwargs)
|
|
321
324
|
data = self.get_data(response)
|
|
@@ -329,7 +332,7 @@ class GQLClient(BaseClientOpenTelemetry):
|
|
|
329
332
|
return DeleteGrader.model_validate(data)
|
|
330
333
|
|
|
331
334
|
def lock_grader(self, use_case: str, id: str, locked: bool, **kwargs: Any) -> LockGrader:
|
|
332
|
-
query = gql('\n mutation LockGrader($useCase: IdOrKey!, $id: IdOrKey!, $locked: Boolean!) {\n lockGrader(useCase: $useCase, id: $id, locked: $locked) {\n ...GraderData\n }\n }\n\n fragment GraderData on Grader {\n id\n name\n key\n locked\n graderType\n graderConfig {\n __typename\n ... on JudgeConfigOutput {\n judgeCriteria: criteria\n examples {\n input {\n role\n content\n }\n output\n pass\n reasoning\n }\n model {\n ...ModelData\n }\n }\n ... on PrebuiltConfigOutput {\n prebuiltCriteria: criteria {\n key\n name\n feedbackKey\n description\n }\n model {\n ...ModelData\n }\n }\n ... on RemoteConfigOutput {\n url\n version\n description\n }\n ... on CustomConfigOutput {\n graderDescription: description\n }\n }\n useCase {\n id\n key\n name\n }\n metric {\n ...MetricData\n }\n createdAt\n updatedAt\n }\n\n fragment MetricData on Metric {\n id\n key\n name\n kind\n description\n scoringType\n createdAt\n hasDirectFeedbacks\n hasComparisonFeedbacks\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n ')
|
|
335
|
+
query = gql('\n mutation LockGrader($useCase: IdOrKey!, $id: IdOrKey!, $locked: Boolean!) {\n lockGrader(useCase: $useCase, id: $id, locked: $locked) {\n ...GraderData\n }\n }\n\n fragment GraderData on Grader {\n id\n name\n key\n locked\n graderType\n graderConfig {\n __typename\n ... on JudgeConfigOutput {\n judgeCriteria: criteria\n examples {\n input {\n role\n content\n }\n output\n pass\n reasoning\n }\n model {\n ...ModelData\n }\n }\n ... on PrebuiltConfigOutput {\n prebuiltCriteria: criteria {\n key\n name\n feedbackKey\n description\n }\n model {\n ...ModelData\n }\n }\n ... on RemoteConfigOutput {\n url\n version\n description\n }\n ... on CustomConfigOutput {\n graderDescription: description\n }\n }\n useCase {\n id\n key\n name\n }\n metric {\n ...MetricData\n }\n createdAt\n updatedAt\n }\n\n fragment MetricData on Metric {\n id\n key\n name\n kind\n description\n scoringType\n createdAt\n hasDirectFeedbacks\n hasComparisonFeedbacks\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n error\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n ')
|
|
333
336
|
variables: Dict[str, object] = {'useCase': use_case, 'id': id, 'locked': locked}
|
|
334
337
|
response = self.execute(query=query, operation_name='LockGrader', variables=variables, **kwargs)
|
|
335
338
|
data = self.get_data(response)
|
|
@@ -356,6 +359,20 @@ class GQLClient(BaseClientOpenTelemetry):
|
|
|
356
359
|
data = self.get_data(response)
|
|
357
360
|
return CancelJob.model_validate(data)
|
|
358
361
|
|
|
362
|
+
def create_dataset_from_multipart_upload(self, input: DatasetCreateFromMultipartUpload, **kwargs: Any) -> CreateDatasetFromMultipartUpload:
|
|
363
|
+
query = gql('\n mutation CreateDatasetFromMultipartUpload($input: DatasetCreateFromMultipartUpload!) {\n createDatasetFromMultipartUpload(input: $input) {\n datasetId\n status\n totalParts\n processedParts\n progress\n error\n }\n }\n ')
|
|
364
|
+
variables: Dict[str, object] = {'input': input}
|
|
365
|
+
response = self.execute(query=query, operation_name='CreateDatasetFromMultipartUpload', variables=variables, **kwargs)
|
|
366
|
+
data = self.get_data(response)
|
|
367
|
+
return CreateDatasetFromMultipartUpload.model_validate(data)
|
|
368
|
+
|
|
369
|
+
def resize_inference_partition(self, input: ResizePartitionInput, **kwargs: Any) -> ResizeInferencePartition:
|
|
370
|
+
query = gql('\n mutation ResizeInferencePartition($input: ResizePartitionInput!) {\n resizeInferencePartition(input: $input)\n }\n ')
|
|
371
|
+
variables: Dict[str, object] = {'input': input}
|
|
372
|
+
response = self.execute(query=query, operation_name='ResizeInferencePartition', variables=variables, **kwargs)
|
|
373
|
+
data = self.get_data(response)
|
|
374
|
+
return ResizeInferencePartition.model_validate(data)
|
|
375
|
+
|
|
359
376
|
def list_datasets(self, input: str, **kwargs: Any) -> ListDatasets:
|
|
360
377
|
query = gql('\n query ListDatasets($input: IdOrKey!) {\n datasets(useCase: $input) {\n ...DatasetData\n }\n }\n\n fragment DatasetData on Dataset {\n id\n key\n name\n createdAt\n kind\n records\n metricsUsage {\n feedbackCount\n comparisonCount\n metric {\n ...MetricData\n }\n }\n }\n\n fragment MetricData on Metric {\n id\n key\n name\n kind\n description\n scoringType\n createdAt\n hasDirectFeedbacks\n hasComparisonFeedbacks\n }\n ')
|
|
361
378
|
variables: Dict[str, object] = {'input': input}
|
|
@@ -371,35 +388,35 @@ class GQLClient(BaseClientOpenTelemetry):
|
|
|
371
388
|
return DescribeDataset.model_validate(data)
|
|
372
389
|
|
|
373
390
|
def describe_use_case(self, input: str, **kwargs: Any) -> DescribeUseCase:
|
|
374
|
-
query = gql('\n query DescribeUseCase($input: IdOrKey!) {\n useCase(idOrKey: $input) {\n ...UseCaseData\n }\n }\n\n fragment MetricWithContextData on MetricWithContext {\n id\n key\n name\n kind\n description\n scoringType\n createdAt\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n\n fragment ModelServiceData on ModelService {\n id\n key\n name\n model {\n ...ModelData\n backbone {\n ...ModelData\n }\n }\n attached\n isDefault\n desiredOnline\n createdAt\n }\n\n fragment UseCaseData on UseCase {\n id\n key\n name\n description\n createdAt\n metrics {\n ...MetricWithContextData\n }\n modelServices {\n ...ModelServiceData\n }\n permissions\n shares {\n team {\n id\n key\n name\n createdAt\n }\n role {\n id\n key\n name\n createdAt\n permissions\n }\n isOwner\n }\n }\n ')
|
|
391
|
+
query = gql('\n query DescribeUseCase($input: IdOrKey!) {\n useCase(idOrKey: $input) {\n ...UseCaseData\n }\n }\n\n fragment MetricWithContextData on MetricWithContext {\n id\n key\n name\n kind\n description\n scoringType\n createdAt\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n error\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n\n fragment ModelServiceData on ModelService {\n id\n key\n name\n model {\n ...ModelData\n backbone {\n ...ModelData\n }\n }\n attached\n isDefault\n desiredOnline\n createdAt\n }\n\n fragment UseCaseData on UseCase {\n id\n key\n name\n description\n createdAt\n metrics {\n ...MetricWithContextData\n }\n modelServices {\n ...ModelServiceData\n }\n permissions\n shares {\n team {\n id\n key\n name\n createdAt\n }\n role {\n id\n key\n name\n createdAt\n permissions\n }\n isOwner\n }\n }\n ')
|
|
375
392
|
variables: Dict[str, object] = {'input': input}
|
|
376
393
|
response = self.execute(query=query, operation_name='DescribeUseCase', variables=variables, **kwargs)
|
|
377
394
|
data = self.get_data(response)
|
|
378
395
|
return DescribeUseCase.model_validate(data)
|
|
379
396
|
|
|
380
397
|
def list_use_cases(self, **kwargs: Any) -> ListUseCases:
|
|
381
|
-
query = gql('\n query ListUseCases {\n useCases {\n ...UseCaseData\n }\n }\n\n fragment MetricWithContextData on MetricWithContext {\n id\n key\n name\n kind\n description\n scoringType\n createdAt\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n\n fragment ModelServiceData on ModelService {\n id\n key\n name\n model {\n ...ModelData\n backbone {\n ...ModelData\n }\n }\n attached\n isDefault\n desiredOnline\n createdAt\n }\n\n fragment UseCaseData on UseCase {\n id\n key\n name\n description\n createdAt\n metrics {\n ...MetricWithContextData\n }\n modelServices {\n ...ModelServiceData\n }\n permissions\n shares {\n team {\n id\n key\n name\n createdAt\n }\n role {\n id\n key\n name\n createdAt\n permissions\n }\n isOwner\n }\n }\n ')
|
|
398
|
+
query = gql('\n query ListUseCases {\n useCases {\n ...UseCaseData\n }\n }\n\n fragment MetricWithContextData on MetricWithContext {\n id\n key\n name\n kind\n description\n scoringType\n createdAt\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n error\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n\n fragment ModelServiceData on ModelService {\n id\n key\n name\n model {\n ...ModelData\n backbone {\n ...ModelData\n }\n }\n attached\n isDefault\n desiredOnline\n createdAt\n }\n\n fragment UseCaseData on UseCase {\n id\n key\n name\n description\n createdAt\n metrics {\n ...MetricWithContextData\n }\n modelServices {\n ...ModelServiceData\n }\n permissions\n shares {\n team {\n id\n key\n name\n createdAt\n }\n role {\n id\n key\n name\n createdAt\n permissions\n }\n isOwner\n }\n }\n ')
|
|
382
399
|
variables: Dict[str, object] = {}
|
|
383
400
|
response = self.execute(query=query, operation_name='ListUseCases', variables=variables, **kwargs)
|
|
384
401
|
data = self.get_data(response)
|
|
385
402
|
return ListUseCases.model_validate(data)
|
|
386
403
|
|
|
387
404
|
def list_models(self, filter: ModelFilter, **kwargs: Any) -> ListModels:
|
|
388
|
-
query = gql('\n query ListModels($filter: ModelFilter! = {inStorage: null, available: null, trainable: null, capabilities: {any: [TextGeneration, Embedding, Reasoning, ImageUnderstanding]}, viewAll: false, online: null}) {\n models(filter: $filter) {\n ...ModelDataAdmin\n backbone {\n ...ModelDataAdmin\n }\n }\n }\n\n fragment ModelDataAdmin on Model {\n id\n key\n name\n online\n useCases {\n id\n key\n name\n }\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n }\n ')
|
|
405
|
+
query = gql('\n query ListModels($filter: ModelFilter! = {inStorage: null, available: null, trainable: null, capabilities: {any: [TextGeneration, Embedding, Reasoning, ImageUnderstanding]}, viewAll: false, online: null}) {\n models(filter: $filter) {\n ...ModelDataAdmin\n backbone {\n ...ModelDataAdmin\n }\n }\n }\n\n fragment ModelDataAdmin on Model {\n id\n key\n name\n online\n error\n useCases {\n id\n key\n name\n }\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n }\n ')
|
|
389
406
|
variables: Dict[str, object] = {'filter': filter}
|
|
390
407
|
response = self.execute(query=query, operation_name='ListModels', variables=variables, **kwargs)
|
|
391
408
|
data = self.get_data(response)
|
|
392
409
|
return ListModels.model_validate(data)
|
|
393
410
|
|
|
394
411
|
def describe_model_admin(self, input: str, **kwargs: Any) -> DescribeModelAdmin:
|
|
395
|
-
query = gql('\n query DescribeModelAdmin($input: IdOrKey!) {\n model(idOrKey: $input) {\n ...ModelDataAdmin\n backbone {\n ...ModelDataAdmin\n }\n }\n }\n\n fragment ModelDataAdmin on Model {\n id\n key\n name\n online\n useCases {\n id\n key\n name\n }\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n }\n ')
|
|
412
|
+
query = gql('\n query DescribeModelAdmin($input: IdOrKey!) {\n model(idOrKey: $input) {\n ...ModelDataAdmin\n backbone {\n ...ModelDataAdmin\n }\n }\n }\n\n fragment ModelDataAdmin on Model {\n id\n key\n name\n online\n error\n useCases {\n id\n key\n name\n }\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n }\n ')
|
|
396
413
|
variables: Dict[str, object] = {'input': input}
|
|
397
414
|
response = self.execute(query=query, operation_name='DescribeModelAdmin', variables=variables, **kwargs)
|
|
398
415
|
data = self.get_data(response)
|
|
399
416
|
return DescribeModelAdmin.model_validate(data)
|
|
400
417
|
|
|
401
418
|
def describe_model(self, input: str, **kwargs: Any) -> DescribeModel:
|
|
402
|
-
query = gql('\n query DescribeModel($input: IdOrKey!) {\n model(idOrKey: $input) {\n ...ModelData\n backbone {\n ...ModelData\n }\n }\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n ')
|
|
419
|
+
query = gql('\n query DescribeModel($input: IdOrKey!) {\n model(idOrKey: $input) {\n ...ModelData\n backbone {\n ...ModelData\n }\n }\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n error\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n ')
|
|
403
420
|
variables: Dict[str, object] = {'input': input}
|
|
404
421
|
response = self.execute(query=query, operation_name='DescribeModel', variables=variables, **kwargs)
|
|
405
422
|
data = self.get_data(response)
|
|
@@ -497,14 +514,14 @@ class GQLClient(BaseClientOpenTelemetry):
|
|
|
497
514
|
return ListPermissions.model_validate(data)
|
|
498
515
|
|
|
499
516
|
def list_harmony_groups(self, **kwargs: Any) -> ListHarmonyGroups:
|
|
500
|
-
query = gql('\n query ListHarmonyGroups {\n harmonyGroups {\n ...HarmonyGroupData\n }\n }\n\n fragment HarmonyGroupData on HarmonyGroup {\n id\n key\n computePool {\n key\n name\n }\n status\n url\n worldSize\n gpuTypes\n createdAt\n onlineModels {\n ...ModelData\n }\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n ')
|
|
517
|
+
query = gql('\n query ListHarmonyGroups {\n harmonyGroups {\n ...HarmonyGroupData\n }\n }\n\n fragment HarmonyGroupData on HarmonyGroup {\n id\n key\n computePool {\n key\n name\n }\n status\n url\n worldSize\n gpuTypes\n createdAt\n onlineModels {\n ...ModelData\n }\n gpuAllocations {\n name\n numGpus\n ranks\n createdAt\n userName\n jobId\n }\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n error\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n ')
|
|
501
518
|
variables: Dict[str, object] = {}
|
|
502
519
|
response = self.execute(query=query, operation_name='ListHarmonyGroups', variables=variables, **kwargs)
|
|
503
520
|
data = self.get_data(response)
|
|
504
521
|
return ListHarmonyGroups.model_validate(data)
|
|
505
522
|
|
|
506
523
|
def list_compute_pools(self, **kwargs: Any) -> ListComputePools:
|
|
507
|
-
query = gql('\n query ListComputePools {\n computePools {\n id\n key\n name\n createdAt\n capabilities\n harmonyGroups {\n ...HarmonyGroupData\n }\n }\n }\n\n fragment HarmonyGroupData on HarmonyGroup {\n id\n key\n computePool {\n key\n name\n }\n status\n url\n worldSize\n gpuTypes\n createdAt\n onlineModels {\n ...ModelData\n }\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n ')
|
|
524
|
+
query = gql('\n query ListComputePools {\n computePools {\n id\n key\n name\n createdAt\n capabilities\n harmonyGroups {\n ...HarmonyGroupData\n }\n }\n }\n\n fragment HarmonyGroupData on HarmonyGroup {\n id\n key\n computePool {\n key\n name\n }\n status\n url\n worldSize\n gpuTypes\n createdAt\n onlineModels {\n ...ModelData\n }\n gpuAllocations {\n name\n numGpus\n ranks\n createdAt\n userName\n jobId\n }\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n error\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n ')
|
|
508
525
|
variables: Dict[str, object] = {}
|
|
509
526
|
response = self.execute(query=query, operation_name='ListComputePools', variables=variables, **kwargs)
|
|
510
527
|
data = self.get_data(response)
|
|
@@ -532,35 +549,35 @@ class GQLClient(BaseClientOpenTelemetry):
|
|
|
532
549
|
return GetCustomRecipe.model_validate(data)
|
|
533
550
|
|
|
534
551
|
def get_judge(self, id: str, use_case: str, version: Union[Optional[int], UnsetType]=UNSET, **kwargs: Any) -> GetJudge:
|
|
535
|
-
query = gql('\n query GetJudge($id: IdOrKey!, $useCase: IdOrKey!, $version: Int) {\n judge(id: $id, useCase: $useCase, version: $version) {\n ...JudgeData\n }\n }\n\n fragment JudgeData on Judge {\n id\n key\n version\n name\n criteria\n prebuilt\n examples {\n input {\n role\n content\n }\n output\n pass\n reasoning\n }\n capabilities\n model {\n ...ModelData\n }\n useCaseId\n metric {\n ...MetricData\n }\n createdAt\n updatedAt\n }\n\n fragment MetricData on Metric {\n id\n key\n name\n kind\n description\n scoringType\n createdAt\n hasDirectFeedbacks\n hasComparisonFeedbacks\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n ')
|
|
552
|
+
query = gql('\n query GetJudge($id: IdOrKey!, $useCase: IdOrKey!, $version: Int) {\n judge(id: $id, useCase: $useCase, version: $version) {\n ...JudgeData\n }\n }\n\n fragment JudgeData on Judge {\n id\n key\n version\n name\n criteria\n prebuilt\n examples {\n input {\n role\n content\n }\n output\n pass\n reasoning\n }\n capabilities\n model {\n ...ModelData\n }\n useCaseId\n metric {\n ...MetricData\n }\n createdAt\n updatedAt\n }\n\n fragment MetricData on Metric {\n id\n key\n name\n kind\n description\n scoringType\n createdAt\n hasDirectFeedbacks\n hasComparisonFeedbacks\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n error\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n ')
|
|
536
553
|
variables: Dict[str, object] = {'id': id, 'useCase': use_case, 'version': version}
|
|
537
554
|
response = self.execute(query=query, operation_name='GetJudge', variables=variables, **kwargs)
|
|
538
555
|
data = self.get_data(response)
|
|
539
556
|
return GetJudge.model_validate(data)
|
|
540
557
|
|
|
541
558
|
def list_judges(self, use_case: str, **kwargs: Any) -> ListJudges:
|
|
542
|
-
query = gql('\n query ListJudges($useCase: IdOrKey!) {\n judges(useCase: $useCase) {\n ...JudgeData\n }\n }\n\n fragment JudgeData on Judge {\n id\n key\n version\n name\n criteria\n prebuilt\n examples {\n input {\n role\n content\n }\n output\n pass\n reasoning\n }\n capabilities\n model {\n ...ModelData\n }\n useCaseId\n metric {\n ...MetricData\n }\n createdAt\n updatedAt\n }\n\n fragment MetricData on Metric {\n id\n key\n name\n kind\n description\n scoringType\n createdAt\n hasDirectFeedbacks\n hasComparisonFeedbacks\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n ')
|
|
559
|
+
query = gql('\n query ListJudges($useCase: IdOrKey!) {\n judges(useCase: $useCase) {\n ...JudgeData\n }\n }\n\n fragment JudgeData on Judge {\n id\n key\n version\n name\n criteria\n prebuilt\n examples {\n input {\n role\n content\n }\n output\n pass\n reasoning\n }\n capabilities\n model {\n ...ModelData\n }\n useCaseId\n metric {\n ...MetricData\n }\n createdAt\n updatedAt\n }\n\n fragment MetricData on Metric {\n id\n key\n name\n kind\n description\n scoringType\n createdAt\n hasDirectFeedbacks\n hasComparisonFeedbacks\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n error\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n ')
|
|
543
560
|
variables: Dict[str, object] = {'useCase': use_case}
|
|
544
561
|
response = self.execute(query=query, operation_name='ListJudges', variables=variables, **kwargs)
|
|
545
562
|
data = self.get_data(response)
|
|
546
563
|
return ListJudges.model_validate(data)
|
|
547
564
|
|
|
548
565
|
def list_judge_versions(self, use_case: str, key: str, **kwargs: Any) -> ListJudgeVersions:
|
|
549
|
-
query = gql('\n query ListJudgeVersions($useCase: IdOrKey!, $key: String!) {\n judgeVersions(useCase: $useCase, key: $key) {\n ...JudgeData\n }\n }\n\n fragment JudgeData on Judge {\n id\n key\n version\n name\n criteria\n prebuilt\n examples {\n input {\n role\n content\n }\n output\n pass\n reasoning\n }\n capabilities\n model {\n ...ModelData\n }\n useCaseId\n metric {\n ...MetricData\n }\n createdAt\n updatedAt\n }\n\n fragment MetricData on Metric {\n id\n key\n name\n kind\n description\n scoringType\n createdAt\n hasDirectFeedbacks\n hasComparisonFeedbacks\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n ')
|
|
566
|
+
query = gql('\n query ListJudgeVersions($useCase: IdOrKey!, $key: String!) {\n judgeVersions(useCase: $useCase, key: $key) {\n ...JudgeData\n }\n }\n\n fragment JudgeData on Judge {\n id\n key\n version\n name\n criteria\n prebuilt\n examples {\n input {\n role\n content\n }\n output\n pass\n reasoning\n }\n capabilities\n model {\n ...ModelData\n }\n useCaseId\n metric {\n ...MetricData\n }\n createdAt\n updatedAt\n }\n\n fragment MetricData on Metric {\n id\n key\n name\n kind\n description\n scoringType\n createdAt\n hasDirectFeedbacks\n hasComparisonFeedbacks\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n error\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n ')
|
|
550
567
|
variables: Dict[str, object] = {'useCase': use_case, 'key': key}
|
|
551
568
|
response = self.execute(query=query, operation_name='ListJudgeVersions', variables=variables, **kwargs)
|
|
552
569
|
data = self.get_data(response)
|
|
553
570
|
return ListJudgeVersions.model_validate(data)
|
|
554
571
|
|
|
555
572
|
def get_grader(self, id: str, use_case: str, **kwargs: Any) -> GetGrader:
|
|
556
|
-
query = gql('\n query GetGrader($id: IdOrKey!, $useCase: IdOrKey!) {\n grader(id: $id, useCase: $useCase) {\n ...GraderData\n }\n }\n\n fragment GraderData on Grader {\n id\n name\n key\n locked\n graderType\n graderConfig {\n __typename\n ... on JudgeConfigOutput {\n judgeCriteria: criteria\n examples {\n input {\n role\n content\n }\n output\n pass\n reasoning\n }\n model {\n ...ModelData\n }\n }\n ... on PrebuiltConfigOutput {\n prebuiltCriteria: criteria {\n key\n name\n feedbackKey\n description\n }\n model {\n ...ModelData\n }\n }\n ... on RemoteConfigOutput {\n url\n version\n description\n }\n ... on CustomConfigOutput {\n graderDescription: description\n }\n }\n useCase {\n id\n key\n name\n }\n metric {\n ...MetricData\n }\n createdAt\n updatedAt\n }\n\n fragment MetricData on Metric {\n id\n key\n name\n kind\n description\n scoringType\n createdAt\n hasDirectFeedbacks\n hasComparisonFeedbacks\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n ')
|
|
573
|
+
query = gql('\n query GetGrader($id: IdOrKey!, $useCase: IdOrKey!) {\n grader(id: $id, useCase: $useCase) {\n ...GraderData\n }\n }\n\n fragment GraderData on Grader {\n id\n name\n key\n locked\n graderType\n graderConfig {\n __typename\n ... on JudgeConfigOutput {\n judgeCriteria: criteria\n examples {\n input {\n role\n content\n }\n output\n pass\n reasoning\n }\n model {\n ...ModelData\n }\n }\n ... on PrebuiltConfigOutput {\n prebuiltCriteria: criteria {\n key\n name\n feedbackKey\n description\n }\n model {\n ...ModelData\n }\n }\n ... on RemoteConfigOutput {\n url\n version\n description\n }\n ... on CustomConfigOutput {\n graderDescription: description\n }\n }\n useCase {\n id\n key\n name\n }\n metric {\n ...MetricData\n }\n createdAt\n updatedAt\n }\n\n fragment MetricData on Metric {\n id\n key\n name\n kind\n description\n scoringType\n createdAt\n hasDirectFeedbacks\n hasComparisonFeedbacks\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n error\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n ')
|
|
557
574
|
variables: Dict[str, object] = {'id': id, 'useCase': use_case}
|
|
558
575
|
response = self.execute(query=query, operation_name='GetGrader', variables=variables, **kwargs)
|
|
559
576
|
data = self.get_data(response)
|
|
560
577
|
return GetGrader.model_validate(data)
|
|
561
578
|
|
|
562
579
|
def list_graders(self, use_case: str, **kwargs: Any) -> ListGraders:
|
|
563
|
-
query = gql('\n query ListGraders($useCase: IdOrKey!) {\n graders(useCase: $useCase) {\n ...GraderData\n }\n }\n\n fragment GraderData on Grader {\n id\n name\n key\n locked\n graderType\n graderConfig {\n __typename\n ... on JudgeConfigOutput {\n judgeCriteria: criteria\n examples {\n input {\n role\n content\n }\n output\n pass\n reasoning\n }\n model {\n ...ModelData\n }\n }\n ... on PrebuiltConfigOutput {\n prebuiltCriteria: criteria {\n key\n name\n feedbackKey\n description\n }\n model {\n ...ModelData\n }\n }\n ... on RemoteConfigOutput {\n url\n version\n description\n }\n ... on CustomConfigOutput {\n graderDescription: description\n }\n }\n useCase {\n id\n key\n name\n }\n metric {\n ...MetricData\n }\n createdAt\n updatedAt\n }\n\n fragment MetricData on Metric {\n id\n key\n name\n kind\n description\n scoringType\n createdAt\n hasDirectFeedbacks\n hasComparisonFeedbacks\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n ')
|
|
580
|
+
query = gql('\n query ListGraders($useCase: IdOrKey!) {\n graders(useCase: $useCase) {\n ...GraderData\n }\n }\n\n fragment GraderData on Grader {\n id\n name\n key\n locked\n graderType\n graderConfig {\n __typename\n ... on JudgeConfigOutput {\n judgeCriteria: criteria\n examples {\n input {\n role\n content\n }\n output\n pass\n reasoning\n }\n model {\n ...ModelData\n }\n }\n ... on PrebuiltConfigOutput {\n prebuiltCriteria: criteria {\n key\n name\n feedbackKey\n description\n }\n model {\n ...ModelData\n }\n }\n ... on RemoteConfigOutput {\n url\n version\n description\n }\n ... on CustomConfigOutput {\n graderDescription: description\n }\n }\n useCase {\n id\n key\n name\n }\n metric {\n ...MetricData\n }\n createdAt\n updatedAt\n }\n\n fragment MetricData on Metric {\n id\n key\n name\n kind\n description\n scoringType\n createdAt\n hasDirectFeedbacks\n hasComparisonFeedbacks\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n error\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n ')
|
|
564
581
|
variables: Dict[str, object] = {'useCase': use_case}
|
|
565
582
|
response = self.execute(query=query, operation_name='ListGraders', variables=variables, **kwargs)
|
|
566
583
|
data = self.get_data(response)
|
|
@@ -580,6 +597,13 @@ class GQLClient(BaseClientOpenTelemetry):
|
|
|
580
597
|
data = self.get_data(response)
|
|
581
598
|
return DescribeJob.model_validate(data)
|
|
582
599
|
|
|
600
|
+
def dataset_upload_processing_status(self, input: DatasetUploadProcessingStatusInput, **kwargs: Any) -> DatasetUploadProcessingStatus:
|
|
601
|
+
query = gql('\n query DatasetUploadProcessingStatus($input: DatasetUploadProcessingStatusInput!) {\n datasetUploadProcessingStatus(input: $input) {\n datasetId\n status\n totalParts\n processedParts\n progress\n error\n }\n }\n ')
|
|
602
|
+
variables: Dict[str, object] = {'input': input}
|
|
603
|
+
response = self.execute(query=query, operation_name='DatasetUploadProcessingStatus', variables=variables, **kwargs)
|
|
604
|
+
data = self.get_data(response)
|
|
605
|
+
return DatasetUploadProcessingStatus.model_validate(data)
|
|
606
|
+
|
|
583
607
|
def execute_custom_operation(self, *fields: GraphQLField, operation_type: OperationType, operation_name: str) -> Dict[str, Any]:
|
|
584
608
|
selections = self._build_selection_set(fields)
|
|
585
609
|
combined_variables = self._combine_variables(fields)
|
|
@@ -0,0 +1,18 @@
|
|
|
1
|
+
from typing import Any, Optional
|
|
2
|
+
from pydantic import Field
|
|
3
|
+
from .base_model import BaseModel
|
|
4
|
+
from .enums import SessionStatus
|
|
5
|
+
|
|
6
|
+
class CreateDatasetFromMultipartUpload(BaseModel):
|
|
7
|
+
"""@public"""
|
|
8
|
+
create_dataset_from_multipart_upload: 'CreateDatasetFromMultipartUploadCreateDatasetFromMultipartUpload' = Field(alias='createDatasetFromMultipartUpload')
|
|
9
|
+
|
|
10
|
+
class CreateDatasetFromMultipartUploadCreateDatasetFromMultipartUpload(BaseModel):
|
|
11
|
+
"""@public"""
|
|
12
|
+
dataset_id: Any = Field(alias='datasetId')
|
|
13
|
+
status: SessionStatus
|
|
14
|
+
total_parts: int = Field(alias='totalParts')
|
|
15
|
+
processed_parts: int = Field(alias='processedParts')
|
|
16
|
+
progress: float
|
|
17
|
+
error: Optional[str]
|
|
18
|
+
CreateDatasetFromMultipartUpload.model_rebuild()
|
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
from typing import Any, Dict, List, Optional, Union
|
|
2
2
|
from .base_operation import GraphQLField
|
|
3
|
-
from .custom_typing_fields import AbcampaignGraphQLField, AbReportGraphQLField, AbVariantReportComparisonGraphQLField, AbVariantReportGraphQLField, ActivityGraphQLField, ActivityOutputGraphQLField, ApiKeyGraphQLField, ArtifactByproductsUnion, AuthProviderGraphQLField, BatchInferenceJobStageOutputGraphQLField, BillingUsageGraphQLField, ChatMessageGraphQLField, ComparisonFeedbackGraphQLField, CompletionConnectionGraphQLField, CompletionEdgeGraphQLField, CompletionGraphQLField, CompletionGroupDataConnectionGraphQLField, CompletionGroupDataEdgeGraphQLField, CompletionGroupDataGraphQLField, CompletionGroupFeedbackStatsGraphQLField, CompletionHistoryEntryOuputGraphQLField, CompletionLabelGraphQLField, CompletionMetadataGraphQLField, ComputePoolGraphQLField, ContractGraphQLField, CustomConfigOutputGraphQLField, CustomRecipeGraphQLField, CustomRecipeJobDetailsGraphQLField, DatasetByproductsGraphQLField, DatasetGraphQLField, DatasetMetricUsageGraphQLField, DatasetProgressGraphQLField, DatasetUploadProcessingStatusGraphQLField, DatasetValidationOutputGraphQLField, DeleteConfirmGraphQLField, DirectFeedbackGraphQLField, EmojiGraphQLField, EvalJobStageOutputGraphQLField, EvaluationByproductsGraphQLField, EvaluationResultGraphQLField, GlobalUsageGraphQLField, GpuAllocationGraphQLField, GraderConfigUnion, GraderGraphQLField, HarmonyGroupGraphQLField, InteractionOutputGraphQLField, IntervalGraphQLField, JobArtifactGraphQLField, JobConnectionGraphQLField, JobEdgeGraphQLField, JobGraphQLField, JobStageInfoOutputUnion, JobStageOutputGraphQLField, JudgeConfigOutputGraphQLField, JudgeExampleGraphQLField, JudgeGraphQLField, LabelGraphQLField, LabelKeyUsageGraphQLField, LabelUsageGraphQLField, LabelValueUsageGraphQLField, MetaObjectGraphQLField, MetricActivityGraphQLField, MetricGraphQLField, MetricWithContextGraphQLField, ModelByproductsGraphQLField, ModelComputeConfigOutputGraphQLField, ModelGraphQLField, ModelPlacementOutputGraphQLField, ModelServiceGraphQLField, PageInfoGraphQLField, PrebuiltConfigDefinitionGraphQLField, PrebuiltConfigOutputGraphQLField, PrebuiltCriteriaGraphQLField, ProviderListGraphQLField, RemoteConfigOutputGraphQLField, RemoteEnvGraphQLField, RemoteEnvTestOfflineGraphQLField, RemoteEnvTestOnlineGraphQLField, RoleGraphQLField, SessionGraphQLField, SettingsGraphQLField, ShareGraphQLField, SystemPromptTemplateGraphQLField, TeamGraphQLField, TeamMemberGraphQLField, TeamWithroleGraphQLField, TimeseriesGraphQLField, ToolProviderGraphQLField, TrainingJobStageOutputGraphQLField, TrendResultGraphQLField, UnitConfigGraphQLField, UsageAggregateItemGraphQLField, UsageAggregatePerUseCaseItemGraphQLField, UsageGraphQLField, UsageStatsByModelGraphQLField, UsageStatsGraphQLField, UseCaseGraphQLField, UseCaseItemGraphQLField, UseCaseMetadataGraphQLField, UserGraphQLField, WidgetGraphQLField
|
|
4
|
-
from .input_types import AbCampaignFilter, CursorPageInput, FeedbackFilterInput, ListCompletionsFilterInput, MetricTrendInput, ModelServiceFilter, OrderPair, TimeRange, TimeseriesInput, UseCaseFilter
|
|
3
|
+
from .custom_typing_fields import AbcampaignGraphQLField, AbReportGraphQLField, AbVariantReportComparisonGraphQLField, AbVariantReportGraphQLField, ActivityGraphQLField, ActivityOutputGraphQLField, ApiKeyGraphQLField, ArtifactByproductsUnion, AuthProviderGraphQLField, BatchInferenceJobStageOutputGraphQLField, BillingUsageGraphQLField, ChatMessageGraphQLField, ComparisonFeedbackGraphQLField, CompletionConnectionGraphQLField, CompletionEdgeGraphQLField, CompletionGraphQLField, CompletionGroupDataConnectionGraphQLField, CompletionGroupDataEdgeGraphQLField, CompletionGroupDataGraphQLField, CompletionGroupFeedbackStatsGraphQLField, CompletionHistoryEntryOuputGraphQLField, CompletionLabelGraphQLField, CompletionMetadataGraphQLField, ComputePoolGraphQLField, ContractGraphQLField, CustomConfigOutputGraphQLField, CustomRecipeGraphQLField, CustomRecipeJobDetailsGraphQLField, DatasetByproductsGraphQLField, DatasetGraphQLField, DatasetMetricUsageGraphQLField, DatasetProgressGraphQLField, DatasetUploadProcessingStatusGraphQLField, DatasetValidationOutputGraphQLField, DeleteConfirmGraphQLField, DirectFeedbackGraphQLField, EmojiGraphQLField, EvalJobStageOutputGraphQLField, EvaluationByproductsGraphQLField, EvaluationResultGraphQLField, GlobalUsageGraphQLField, GpuAllocationGraphQLField, GraderConfigUnion, GraderGraphQLField, HarmonyGroupGraphQLField, InteractionOutputGraphQLField, IntervalGraphQLField, JobArtifactGraphQLField, JobConnectionGraphQLField, JobEdgeGraphQLField, JobGraphQLField, JobStageInfoOutputUnion, JobStageOutputGraphQLField, JudgeConfigOutputGraphQLField, JudgeExampleGraphQLField, JudgeGraphQLField, LabelGraphQLField, LabelKeyUsageGraphQLField, LabelUsageGraphQLField, LabelValueUsageGraphQLField, MetaObjectGraphQLField, MetricActivityGraphQLField, MetricGraphQLField, MetricWithContextGraphQLField, ModelByproductsGraphQLField, ModelComputeConfigOutputGraphQLField, ModelGraphQLField, ModelPlacementOutputGraphQLField, ModelServiceGraphQLField, PageInfoGraphQLField, PrebuiltConfigDefinitionGraphQLField, PrebuiltConfigOutputGraphQLField, PrebuiltCriteriaGraphQLField, ProviderListGraphQLField, RemoteConfigOutputGraphQLField, RemoteEnvGraphQLField, RemoteEnvTestOfflineGraphQLField, RemoteEnvTestOnlineGraphQLField, RoleGraphQLField, SearchResultGraphQLField, SessionGraphQLField, SettingsGraphQLField, ShareGraphQLField, SystemPromptTemplateGraphQLField, TeamGraphQLField, TeamMemberGraphQLField, TeamWithroleGraphQLField, TimeseriesGraphQLField, ToolProviderGraphQLField, TrainingJobStageOutputGraphQLField, TrendResultGraphQLField, UnitConfigGraphQLField, UsageAggregateItemGraphQLField, UsageAggregatePerUseCaseItemGraphQLField, UsageGraphQLField, UsageStatsByModelGraphQLField, UsageStatsGraphQLField, UseCaseGraphQLField, UseCaseItemGraphQLField, UseCaseMetadataGraphQLField, UserGraphQLField, WidgetGraphQLField
|
|
4
|
+
from .input_types import AbCampaignFilter, ArtifactFilter, CursorPageInput, FeedbackFilterInput, ListCompletionsFilterInput, MetricTrendInput, ModelServiceFilter, OrderPair, TimeRange, TimeseriesInput, UseCaseFilter
|
|
5
5
|
|
|
6
6
|
class AbReportFields(GraphQLField):
|
|
7
7
|
"""@private"""
|
|
@@ -1734,6 +1734,28 @@ class RoleFields(GraphQLField):
|
|
|
1734
1734
|
self._alias = alias
|
|
1735
1735
|
return self
|
|
1736
1736
|
|
|
1737
|
+
class SearchResultFields(GraphQLField):
|
|
1738
|
+
"""@private"""
|
|
1739
|
+
|
|
1740
|
+
@classmethod
|
|
1741
|
+
def jobs(cls) -> 'JobFields':
|
|
1742
|
+
return JobFields('jobs')
|
|
1743
|
+
|
|
1744
|
+
@classmethod
|
|
1745
|
+
def artifacts(cls, *, filter: Optional[ArtifactFilter]=None) -> 'JobArtifactFields':
|
|
1746
|
+
arguments: Dict[str, Dict[str, Any]] = {'filter': {'type': 'ArtifactFilter', 'value': filter}}
|
|
1747
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
|
|
1748
|
+
return JobArtifactFields('artifacts', arguments=cleared_arguments)
|
|
1749
|
+
|
|
1750
|
+
def fields(self, *subfields: Union[SearchResultGraphQLField, 'JobArtifactFields', 'JobFields']) -> 'SearchResultFields':
|
|
1751
|
+
"""Subfields should come from the SearchResultFields class"""
|
|
1752
|
+
self._subfields.extend(subfields)
|
|
1753
|
+
return self
|
|
1754
|
+
|
|
1755
|
+
def alias(self, alias: str) -> 'SearchResultFields':
|
|
1756
|
+
self._alias = alias
|
|
1757
|
+
return self
|
|
1758
|
+
|
|
1737
1759
|
class SessionFields(GraphQLField):
|
|
1738
1760
|
"""@private"""
|
|
1739
1761
|
id: 'SessionGraphQLField' = SessionGraphQLField('id')
|
|
@@ -1,8 +1,8 @@
|
|
|
1
1
|
from typing import Any, Dict, List, Optional
|
|
2
2
|
from . import CompletionGroupBy
|
|
3
|
-
from .custom_fields import AbcampaignFields, CompletionConnectionFields, CompletionFields, CompletionGroupDataConnectionFields, ComputePoolFields, ContractFields, CustomRecipeFields, DatasetFields, DatasetUploadProcessingStatusFields, GlobalUsageFields, GraderFields, HarmonyGroupFields, JobArtifactFields, JobConnectionFields, JobFields, JudgeFields, MetaObjectFields, MetricFields, ModelFields, PrebuiltConfigDefinitionFields, PrebuiltCriteriaFields, RemoteConfigOutputFields, RemoteEnvFields, RoleFields, SystemPromptTemplateFields, TeamFields, ToolProviderFields, UsageAggregateItemFields, UsageAggregatePerUseCaseItemFields, UseCaseFields, UserFields
|
|
3
|
+
from .custom_fields import AbcampaignFields, CompletionConnectionFields, CompletionFields, CompletionGroupDataConnectionFields, ComputePoolFields, ContractFields, CustomRecipeFields, DatasetFields, DatasetUploadProcessingStatusFields, GlobalUsageFields, GraderFields, HarmonyGroupFields, JobArtifactFields, JobConnectionFields, JobFields, JudgeFields, MetaObjectFields, MetricFields, ModelFields, PrebuiltConfigDefinitionFields, PrebuiltCriteriaFields, RemoteConfigOutputFields, RemoteEnvFields, RoleFields, SearchResultFields, SystemPromptTemplateFields, TeamFields, ToolProviderFields, UsageAggregateItemFields, UsageAggregatePerUseCaseItemFields, UseCaseFields, UserFields
|
|
4
4
|
from .custom_typing_fields import GraphQLField
|
|
5
|
-
from .input_types import AbCampaignFilter, ArtifactFilter, CursorPageInput, CustomRecipeFilterInput, DatasetCreateFromFilters, DatasetUploadProcessingStatusInput, FeedbackFilterInput, GlobalUsageFilterInput, ListCompletionsFilterInput, ListJobsFilterInput, ModelFilter, OrderPair, UsageFilterInput, UsagePerUseCaseFilterInput, UseCaseFilter
|
|
5
|
+
from .input_types import AbCampaignFilter, ArtifactFilter, CursorPageInput, CustomRecipeFilterInput, DatasetCreateFromFilters, DatasetUploadProcessingStatusInput, FeedbackFilterInput, GlobalUsageFilterInput, ListCompletionsFilterInput, ListJobsFilterInput, ModelFilter, OrderPair, SearchInput, UsageFilterInput, UsagePerUseCaseFilterInput, UseCaseFilter
|
|
6
6
|
|
|
7
7
|
class Query:
|
|
8
8
|
"""@private"""
|
|
@@ -35,6 +35,12 @@ class Query:
|
|
|
35
35
|
cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
|
|
36
36
|
return CustomRecipeFields(field_name='customRecipe', arguments=cleared_arguments)
|
|
37
37
|
|
|
38
|
+
@classmethod
|
|
39
|
+
def parse_recipe_schema(cls, recipe_content: str) -> GraphQLField:
|
|
40
|
+
arguments: Dict[str, Dict[str, Any]] = {'recipeContent': {'type': 'String!', 'value': recipe_content}}
|
|
41
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
|
|
42
|
+
return GraphQLField(field_name='parseRecipeSchema', arguments=cleared_arguments)
|
|
43
|
+
|
|
38
44
|
@classmethod
|
|
39
45
|
def datasets(cls, use_case: str) -> DatasetFields:
|
|
40
46
|
arguments: Dict[str, Dict[str, Any]] = {'useCase': {'type': 'IdOrKey!', 'value': use_case}}
|
|
@@ -263,6 +269,12 @@ class Query:
|
|
|
263
269
|
cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
|
|
264
270
|
return JobArtifactFields(field_name='artifact', arguments=cleared_arguments)
|
|
265
271
|
|
|
272
|
+
@classmethod
|
|
273
|
+
def search_use_case(cls, input: SearchInput, use_case: str) -> SearchResultFields:
|
|
274
|
+
arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'SearchInput!', 'value': input}, 'useCase': {'type': 'IdOrKey!', 'value': use_case}}
|
|
275
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
|
|
276
|
+
return SearchResultFields(field_name='searchUseCase', arguments=cleared_arguments)
|
|
277
|
+
|
|
266
278
|
@classmethod
|
|
267
279
|
def meta(cls) -> MetaObjectFields:
|
|
268
280
|
return MetaObjectFields(field_name='meta')
|
|
@@ -622,6 +622,13 @@ class RoleGraphQLField(GraphQLField):
|
|
|
622
622
|
self._alias = alias
|
|
623
623
|
return self
|
|
624
624
|
|
|
625
|
+
class SearchResultGraphQLField(GraphQLField):
|
|
626
|
+
"""@private"""
|
|
627
|
+
|
|
628
|
+
def alias(self, alias: str) -> 'SearchResultGraphQLField':
|
|
629
|
+
self._alias = alias
|
|
630
|
+
return self
|
|
631
|
+
|
|
625
632
|
class SessionGraphQLField(GraphQLField):
|
|
626
633
|
"""@private"""
|
|
627
634
|
|
|
@@ -0,0 +1,18 @@
|
|
|
1
|
+
from typing import Any, Optional
|
|
2
|
+
from pydantic import Field
|
|
3
|
+
from .base_model import BaseModel
|
|
4
|
+
from .enums import SessionStatus
|
|
5
|
+
|
|
6
|
+
class DatasetUploadProcessingStatus(BaseModel):
|
|
7
|
+
"""@public"""
|
|
8
|
+
dataset_upload_processing_status: 'DatasetUploadProcessingStatusDatasetUploadProcessingStatus' = Field(alias='datasetUploadProcessingStatus')
|
|
9
|
+
|
|
10
|
+
class DatasetUploadProcessingStatusDatasetUploadProcessingStatus(BaseModel):
|
|
11
|
+
"""@public"""
|
|
12
|
+
dataset_id: Any = Field(alias='datasetId')
|
|
13
|
+
status: SessionStatus
|
|
14
|
+
total_parts: int = Field(alias='totalParts')
|
|
15
|
+
processed_parts: int = Field(alias='processedParts')
|
|
16
|
+
progress: float
|
|
17
|
+
error: Optional[str]
|
|
18
|
+
DatasetUploadProcessingStatus.model_rebuild()
|
|
@@ -49,6 +49,7 @@ class DatasetSource(str, Enum):
|
|
|
49
49
|
|
|
50
50
|
class DatasetStatus(str, Enum):
|
|
51
51
|
"""@public"""
|
|
52
|
+
PENDING = 'PENDING'
|
|
52
53
|
PROCESSING = 'PROCESSING'
|
|
53
54
|
READY = 'READY'
|
|
54
55
|
FAILED = 'FAILED'
|
|
@@ -65,11 +66,9 @@ class DateBucketUnit(str, Enum):
|
|
|
65
66
|
|
|
66
67
|
class ExternalModelProviderName(str, Enum):
|
|
67
68
|
"""@public"""
|
|
68
|
-
AZURE = 'AZURE'
|
|
69
69
|
OPEN_AI = 'OPEN_AI'
|
|
70
70
|
GOOGLE = 'GOOGLE'
|
|
71
71
|
ANTHROPIC = 'ANTHROPIC'
|
|
72
|
-
NVIDIA = 'NVIDIA'
|
|
73
72
|
|
|
74
73
|
class FeedbackType(str, Enum):
|
|
75
74
|
"""@public"""
|
|
@@ -194,12 +193,10 @@ class Protocol(str, Enum):
|
|
|
194
193
|
|
|
195
194
|
class ProviderName(str, Enum):
|
|
196
195
|
"""@public"""
|
|
197
|
-
AZURE = 'AZURE'
|
|
198
196
|
OPEN_AI = 'OPEN_AI'
|
|
199
197
|
HARMONY = 'HARMONY'
|
|
200
198
|
GOOGLE = 'GOOGLE'
|
|
201
199
|
ANTHROPIC = 'ANTHROPIC'
|
|
202
|
-
NVIDIA = 'NVIDIA'
|
|
203
200
|
|
|
204
201
|
class RemoteEnvStatus(str, Enum):
|
|
205
202
|
"""@public"""
|