adaptive-sdk 0.1.14__py3-none-any.whl → 0.12.1__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.
@@ -1,355 +1,510 @@
1
1
  from typing import Any, Dict, Optional
2
2
  from .base_model import Upload
3
- from .custom_fields import AbcampaignFields, ApiKeyFields, CompletionFields, CustomRecipeFields, DatasetFields, DatasetUploadProcessingStatusFields, DatasetValidationOutputFields, DeleteConfirmFields, DirectFeedbackFields, GraderFields, JobFields, JudgeFields, MetricFields, MetricWithContextFields, ModelFields, ModelServiceFields, RemoteEnvFields, RoleFields, SystemPromptTemplateFields, TeamFields, TeamMemberFields, ToolProviderFields, UseCaseFields, UserFields
3
+ from .custom_fields import (
4
+ AbcampaignFields,
5
+ ApiKeyFields,
6
+ CompletionFields,
7
+ CustomRecipeFields,
8
+ DatasetFields,
9
+ DatasetUploadProcessingStatusFields,
10
+ DatasetValidationOutputFields,
11
+ DeleteConfirmFields,
12
+ DirectFeedbackFields,
13
+ GraderFields,
14
+ JobFields,
15
+ JudgeFields,
16
+ MetricFields,
17
+ MetricWithContextFields,
18
+ ModelFields,
19
+ ModelServiceFields,
20
+ RemoteEnvFields,
21
+ RoleFields,
22
+ SystemPromptTemplateFields,
23
+ TeamFields,
24
+ TeamMemberFields,
25
+ ToolProviderFields,
26
+ UseCaseFields,
27
+ UserFields,
28
+ )
4
29
  from .custom_typing_fields import GraphQLField, RemoteEnvTestUnion
5
- from .input_types import AbcampaignCreate, AddExternalModelInput, AddHFModelInput, AddModelInput, AddModelToUseCaseInput, ApiKeyCreate, CancelAllocationInput, CreateRecipeInput, CreateToolProviderInput, DatasetCreate, DatasetCreateFromFilters, DatasetCreateFromMultipartUpload, DeleteModelInput, DeployModelInput, FeedbackAddInput, FeedbackUpdateInput, GraderCreateInput, GraderUpdateInput, JobInput, JudgeCreate, JudgeUpdate, MetricCreate, MetricLink, MetricUnlink, ModelComputeConfigInput, PrebuiltJudgeCreate, RemoteEnvCreate, RemoveModelFromUseCaseInput, ResizePartitionInput, RoleCreate, SystemPromptTemplateCreate, SystemPromptTemplateUpdate, TeamCreate, TeamMemberRemove, TeamMemberSet, UpdateCompletion, UpdateModelInput, UpdateModelService, UpdateRecipeInput, UpdateToolProviderInput, UseCaseCreate, UseCaseShares, UseCaseUpdate, UserCreate
30
+ from .input_types import (
31
+ AbcampaignCreate,
32
+ AddExternalModelInput,
33
+ AddHFModelInput,
34
+ AddModelInput,
35
+ AddModelToUseCaseInput,
36
+ ApiKeyCreate,
37
+ CancelAllocationInput,
38
+ CreateRecipeInput,
39
+ CreateToolProviderInput,
40
+ DatasetCreate,
41
+ DatasetCreateFromFilters,
42
+ DatasetCreateFromMultipartUpload,
43
+ DeleteModelInput,
44
+ DeployModelInput,
45
+ FeedbackAddInput,
46
+ FeedbackUpdateInput,
47
+ GraderCreateInput,
48
+ GraderUpdateInput,
49
+ JobInput,
50
+ JudgeCreate,
51
+ JudgeUpdate,
52
+ MetricCreate,
53
+ MetricLink,
54
+ MetricUnlink,
55
+ ModelComputeConfigInput,
56
+ PrebuiltJudgeCreate,
57
+ RemoteEnvCreate,
58
+ RemoveModelFromUseCaseInput,
59
+ ResizePartitionInput,
60
+ RoleCreate,
61
+ SystemPromptTemplateCreate,
62
+ SystemPromptTemplateUpdate,
63
+ TeamCreate,
64
+ TeamMemberRemove,
65
+ TeamMemberSet,
66
+ UpdateCompletion,
67
+ UpdateModelInput,
68
+ UpdateModelService,
69
+ UpdateRecipeInput,
70
+ UpdateToolProviderInput,
71
+ UseCaseCreate,
72
+ UseCaseShares,
73
+ UseCaseUpdate,
74
+ UserCreate,
75
+ )
76
+
6
77
 
7
78
  class Mutation:
8
79
  """@private"""
9
80
 
10
81
  @classmethod
11
82
  def create_ab_campaign(cls, input: AbcampaignCreate) -> AbcampaignFields:
12
- arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'AbcampaignCreate!', 'value': input}}
13
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
14
- return AbcampaignFields(field_name='createAbCampaign', arguments=cleared_arguments)
83
+ arguments: Dict[str, Dict[str, Any]] = {"input": {"type": "AbcampaignCreate!", "value": input}}
84
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
85
+ return AbcampaignFields(field_name="createAbCampaign", arguments=cleared_arguments)
15
86
 
16
87
  @classmethod
17
88
  def cancel_ab_campaign(cls, input: str) -> GraphQLField:
18
- arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'IdOrKey!', 'value': input}}
19
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
20
- return GraphQLField(field_name='cancelAbCampaign', arguments=cleared_arguments)
89
+ arguments: Dict[str, Dict[str, Any]] = {"input": {"type": "IdOrKey!", "value": input}}
90
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
91
+ return GraphQLField(field_name="cancelAbCampaign", arguments=cleared_arguments)
21
92
 
22
93
  @classmethod
23
94
  def resize_inference_partition(cls, input: ResizePartitionInput) -> GraphQLField:
24
- arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'ResizePartitionInput!', 'value': input}}
25
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
26
- return GraphQLField(field_name='resizeInferencePartition', arguments=cleared_arguments)
95
+ arguments: Dict[str, Dict[str, Any]] = {"input": {"type": "ResizePartitionInput!", "value": input}}
96
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
97
+ return GraphQLField(field_name="resizeInferencePartition", arguments=cleared_arguments)
27
98
 
28
99
  @classmethod
29
100
  def cancel_allocation(cls, input: CancelAllocationInput) -> GraphQLField:
30
- arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'CancelAllocationInput!', 'value': input}}
31
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
32
- return GraphQLField(field_name='cancelAllocation', arguments=cleared_arguments)
101
+ arguments: Dict[str, Dict[str, Any]] = {"input": {"type": "CancelAllocationInput!", "value": input}}
102
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
103
+ return GraphQLField(field_name="cancelAllocation", arguments=cleared_arguments)
33
104
 
34
105
  @classmethod
35
106
  def create_custom_recipe(cls, use_case: str, input: CreateRecipeInput, file: Upload) -> CustomRecipeFields:
36
- arguments: Dict[str, Dict[str, Any]] = {'useCase': {'type': 'IdOrKey!', 'value': use_case}, 'input': {'type': 'CreateRecipeInput!', 'value': input}, 'file': {'type': 'Upload!', 'value': file}}
37
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
38
- return CustomRecipeFields(field_name='createCustomRecipe', arguments=cleared_arguments)
39
-
40
- @classmethod
41
- def update_custom_recipe(cls, use_case: str, id: str, input: UpdateRecipeInput, *, file: Optional[Upload]=None) -> CustomRecipeFields:
42
- arguments: Dict[str, Dict[str, Any]] = {'useCase': {'type': 'IdOrKey!', 'value': use_case}, 'id': {'type': 'IdOrKey!', 'value': id}, 'input': {'type': 'UpdateRecipeInput!', 'value': input}, 'file': {'type': 'Upload', 'value': file}}
43
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
44
- return CustomRecipeFields(field_name='updateCustomRecipe', arguments=cleared_arguments)
107
+ arguments: Dict[str, Dict[str, Any]] = {
108
+ "useCase": {"type": "IdOrKey!", "value": use_case},
109
+ "input": {"type": "CreateRecipeInput!", "value": input},
110
+ "file": {"type": "Upload!", "value": file},
111
+ }
112
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
113
+ return CustomRecipeFields(field_name="createCustomRecipe", arguments=cleared_arguments)
114
+
115
+ @classmethod
116
+ def update_custom_recipe(
117
+ cls, use_case: str, id: str, input: UpdateRecipeInput, *, file: Optional[Upload] = None
118
+ ) -> CustomRecipeFields:
119
+ arguments: Dict[str, Dict[str, Any]] = {
120
+ "useCase": {"type": "IdOrKey!", "value": use_case},
121
+ "id": {"type": "IdOrKey!", "value": id},
122
+ "input": {"type": "UpdateRecipeInput!", "value": input},
123
+ "file": {"type": "Upload", "value": file},
124
+ }
125
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
126
+ return CustomRecipeFields(field_name="updateCustomRecipe", arguments=cleared_arguments)
45
127
 
46
128
  @classmethod
47
129
  def delete_custom_recipe(cls, use_case: str, id: str) -> GraphQLField:
48
- arguments: Dict[str, Dict[str, Any]] = {'useCase': {'type': 'IdOrKey!', 'value': use_case}, 'id': {'type': 'IdOrKey!', 'value': id}}
49
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
50
- return GraphQLField(field_name='deleteCustomRecipe', arguments=cleared_arguments)
130
+ arguments: Dict[str, Dict[str, Any]] = {
131
+ "useCase": {"type": "IdOrKey!", "value": use_case},
132
+ "id": {"type": "IdOrKey!", "value": id},
133
+ }
134
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
135
+ return GraphQLField(field_name="deleteCustomRecipe", arguments=cleared_arguments)
51
136
 
52
137
  @classmethod
53
- def create_dataset_from_multipart_upload(cls, input: DatasetCreateFromMultipartUpload) -> DatasetUploadProcessingStatusFields:
54
- arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'DatasetCreateFromMultipartUpload!', 'value': input}}
55
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
56
- return DatasetUploadProcessingStatusFields(field_name='createDatasetFromMultipartUpload', arguments=cleared_arguments)
138
+ def create_dataset_from_multipart_upload(
139
+ cls, input: DatasetCreateFromMultipartUpload
140
+ ) -> DatasetUploadProcessingStatusFields:
141
+ arguments: Dict[str, Dict[str, Any]] = {"input": {"type": "DatasetCreateFromMultipartUpload!", "value": input}}
142
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
143
+ return DatasetUploadProcessingStatusFields(
144
+ field_name="createDatasetFromMultipartUpload", arguments=cleared_arguments
145
+ )
57
146
 
58
147
  @classmethod
59
148
  def create_dataset(cls, input: DatasetCreate, file: Upload) -> DatasetFields:
60
- arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'DatasetCreate!', 'value': input}, 'file': {'type': 'Upload!', 'value': file}}
61
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
62
- return DatasetFields(field_name='createDataset', arguments=cleared_arguments)
149
+ arguments: Dict[str, Dict[str, Any]] = {
150
+ "input": {"type": "DatasetCreate!", "value": input},
151
+ "file": {"type": "Upload!", "value": file},
152
+ }
153
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
154
+ return DatasetFields(field_name="createDataset", arguments=cleared_arguments)
63
155
 
64
156
  @classmethod
65
157
  def create_dataset_from_filters(cls, input: DatasetCreateFromFilters) -> DatasetFields:
66
- arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'DatasetCreateFromFilters!', 'value': input}}
67
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
68
- return DatasetFields(field_name='createDatasetFromFilters', arguments=cleared_arguments)
158
+ arguments: Dict[str, Dict[str, Any]] = {"input": {"type": "DatasetCreateFromFilters!", "value": input}}
159
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
160
+ return DatasetFields(field_name="createDatasetFromFilters", arguments=cleared_arguments)
69
161
 
70
162
  @classmethod
71
163
  def delete_dataset(cls, id_or_key: str, use_case: str) -> GraphQLField:
72
- arguments: Dict[str, Dict[str, Any]] = {'idOrKey': {'type': 'IdOrKey!', 'value': id_or_key}, 'useCase': {'type': 'IdOrKey!', 'value': use_case}}
73
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
74
- return GraphQLField(field_name='deleteDataset', arguments=cleared_arguments)
164
+ arguments: Dict[str, Dict[str, Any]] = {
165
+ "idOrKey": {"type": "IdOrKey!", "value": id_or_key},
166
+ "useCase": {"type": "IdOrKey!", "value": use_case},
167
+ }
168
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
169
+ return GraphQLField(field_name="deleteDataset", arguments=cleared_arguments)
75
170
 
76
171
  @classmethod
77
172
  def update_completion(cls, input: UpdateCompletion) -> CompletionFields:
78
- arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'UpdateCompletion!', 'value': input}}
79
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
80
- return CompletionFields(field_name='updateCompletion', arguments=cleared_arguments)
173
+ arguments: Dict[str, Dict[str, Any]] = {"input": {"type": "UpdateCompletion!", "value": input}}
174
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
175
+ return CompletionFields(field_name="updateCompletion", arguments=cleared_arguments)
81
176
 
82
177
  @classmethod
83
178
  def create_system_prompt_template(cls, input: SystemPromptTemplateCreate) -> SystemPromptTemplateFields:
84
- arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'SystemPromptTemplateCreate!', 'value': input}}
85
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
86
- return SystemPromptTemplateFields(field_name='createSystemPromptTemplate', arguments=cleared_arguments)
179
+ arguments: Dict[str, Dict[str, Any]] = {"input": {"type": "SystemPromptTemplateCreate!", "value": input}}
180
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
181
+ return SystemPromptTemplateFields(field_name="createSystemPromptTemplate", arguments=cleared_arguments)
87
182
 
88
183
  @classmethod
89
184
  def derive_system_prompt_template(cls, input: SystemPromptTemplateUpdate) -> SystemPromptTemplateFields:
90
- arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'SystemPromptTemplateUpdate!', 'value': input}}
91
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
92
- return SystemPromptTemplateFields(field_name='deriveSystemPromptTemplate', arguments=cleared_arguments)
185
+ arguments: Dict[str, Dict[str, Any]] = {"input": {"type": "SystemPromptTemplateUpdate!", "value": input}}
186
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
187
+ return SystemPromptTemplateFields(field_name="deriveSystemPromptTemplate", arguments=cleared_arguments)
93
188
 
94
189
  @classmethod
95
190
  def create_job(cls, input: JobInput) -> JobFields:
96
- arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'JobInput!', 'value': input}}
97
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
98
- return JobFields(field_name='createJob', arguments=cleared_arguments)
191
+ arguments: Dict[str, Dict[str, Any]] = {"input": {"type": "JobInput!", "value": input}}
192
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
193
+ return JobFields(field_name="createJob", arguments=cleared_arguments)
99
194
 
100
195
  @classmethod
101
196
  def cancel_job(cls, id: Any) -> JobFields:
102
- arguments: Dict[str, Dict[str, Any]] = {'id': {'type': 'UUID!', 'value': id}}
103
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
104
- return JobFields(field_name='cancelJob', arguments=cleared_arguments)
197
+ arguments: Dict[str, Dict[str, Any]] = {"id": {"type": "UUID!", "value": id}}
198
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
199
+ return JobFields(field_name="cancelJob", arguments=cleared_arguments)
105
200
 
106
201
  @classmethod
107
202
  def create_metric(cls, input: MetricCreate) -> MetricFields:
108
- arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'MetricCreate!', 'value': input}}
109
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
110
- return MetricFields(field_name='createMetric', arguments=cleared_arguments)
203
+ arguments: Dict[str, Dict[str, Any]] = {"input": {"type": "MetricCreate!", "value": input}}
204
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
205
+ return MetricFields(field_name="createMetric", arguments=cleared_arguments)
111
206
 
112
207
  @classmethod
113
208
  def link_metric(cls, input: MetricLink) -> MetricWithContextFields:
114
209
  """Link a metric and a use case. Can also be used to update the `isPinned`"""
115
- arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'MetricLink!', 'value': input}}
116
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
117
- return MetricWithContextFields(field_name='linkMetric', arguments=cleared_arguments)
210
+ arguments: Dict[str, Dict[str, Any]] = {"input": {"type": "MetricLink!", "value": input}}
211
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
212
+ return MetricWithContextFields(field_name="linkMetric", arguments=cleared_arguments)
118
213
 
119
214
  @classmethod
120
215
  def unlink_metric(cls, input: MetricUnlink) -> GraphQLField:
121
- arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'MetricUnlink!', 'value': input}}
122
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
123
- return GraphQLField(field_name='unlinkMetric', arguments=cleared_arguments)
216
+ arguments: Dict[str, Dict[str, Any]] = {"input": {"type": "MetricUnlink!", "value": input}}
217
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
218
+ return GraphQLField(field_name="unlinkMetric", arguments=cleared_arguments)
124
219
 
125
220
  @classmethod
126
221
  def deploy_model(cls, input: DeployModelInput) -> ModelServiceFields:
127
- arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'DeployModelInput!', 'value': input}}
128
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
129
- return ModelServiceFields(field_name='deployModel', arguments=cleared_arguments)
222
+ arguments: Dict[str, Dict[str, Any]] = {"input": {"type": "DeployModelInput!", "value": input}}
223
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
224
+ return ModelServiceFields(field_name="deployModel", arguments=cleared_arguments)
130
225
 
131
226
  @classmethod
132
227
  def update_model(cls, input: UpdateModelInput) -> ModelFields:
133
- arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'UpdateModelInput!', 'value': input}}
134
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
135
- return ModelFields(field_name='updateModel', arguments=cleared_arguments)
228
+ arguments: Dict[str, Dict[str, Any]] = {"input": {"type": "UpdateModelInput!", "value": input}}
229
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
230
+ return ModelFields(field_name="updateModel", arguments=cleared_arguments)
136
231
 
137
232
  @classmethod
138
233
  def update_model_service(cls, input: UpdateModelService) -> ModelServiceFields:
139
- arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'UpdateModelService!', 'value': input}}
140
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
141
- return ModelServiceFields(field_name='updateModelService', arguments=cleared_arguments)
234
+ arguments: Dict[str, Dict[str, Any]] = {"input": {"type": "UpdateModelService!", "value": input}}
235
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
236
+ return ModelServiceFields(field_name="updateModelService", arguments=cleared_arguments)
142
237
 
143
238
  @classmethod
144
239
  def terminate_model(cls, id_or_key: str, force: bool) -> GraphQLField:
145
240
  """If a model is used by several use cases with `desiredOnline = true`, you need to specify 'force = true' to be able to deactivate the model"""
146
- arguments: Dict[str, Dict[str, Any]] = {'idOrKey': {'type': 'IdOrKey!', 'value': id_or_key}, 'force': {'type': 'Boolean!', 'value': force}}
147
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
148
- return GraphQLField(field_name='terminateModel', arguments=cleared_arguments)
241
+ arguments: Dict[str, Dict[str, Any]] = {
242
+ "idOrKey": {"type": "IdOrKey!", "value": id_or_key},
243
+ "force": {"type": "Boolean!", "value": force},
244
+ }
245
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
246
+ return GraphQLField(field_name="terminateModel", arguments=cleared_arguments)
149
247
 
150
248
  @classmethod
151
249
  def add_external_model(cls, input: AddExternalModelInput) -> ModelFields:
152
- arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'AddExternalModelInput!', 'value': input}}
153
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
154
- return ModelFields(field_name='addExternalModel', arguments=cleared_arguments)
250
+ arguments: Dict[str, Dict[str, Any]] = {"input": {"type": "AddExternalModelInput!", "value": input}}
251
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
252
+ return ModelFields(field_name="addExternalModel", arguments=cleared_arguments)
155
253
 
156
254
  @classmethod
157
255
  def add_model(cls, input: AddModelInput) -> ModelFields:
158
- arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'AddModelInput!', 'value': input}}
159
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
160
- return ModelFields(field_name='addModel', arguments=cleared_arguments)
256
+ arguments: Dict[str, Dict[str, Any]] = {"input": {"type": "AddModelInput!", "value": input}}
257
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
258
+ return ModelFields(field_name="addModel", arguments=cleared_arguments)
161
259
 
162
260
  @classmethod
163
261
  def import_hf_model(cls, input: AddHFModelInput) -> JobFields:
164
- arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'AddHFModelInput!', 'value': input}}
165
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
166
- return JobFields(field_name='importHfModel', arguments=cleared_arguments)
262
+ arguments: Dict[str, Dict[str, Any]] = {"input": {"type": "AddHFModelInput!", "value": input}}
263
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
264
+ return JobFields(field_name="importHfModel", arguments=cleared_arguments)
167
265
 
168
266
  @classmethod
169
267
  def update_model_compute_config(cls, id_or_key: str, input: ModelComputeConfigInput) -> ModelFields:
170
- arguments: Dict[str, Dict[str, Any]] = {'idOrKey': {'type': 'IdOrKey!', 'value': id_or_key}, 'input': {'type': 'ModelComputeConfigInput!', 'value': input}}
171
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
172
- return ModelFields(field_name='updateModelComputeConfig', arguments=cleared_arguments)
268
+ arguments: Dict[str, Dict[str, Any]] = {
269
+ "idOrKey": {"type": "IdOrKey!", "value": id_or_key},
270
+ "input": {"type": "ModelComputeConfigInput!", "value": input},
271
+ }
272
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
273
+ return ModelFields(field_name="updateModelComputeConfig", arguments=cleared_arguments)
173
274
 
174
275
  @classmethod
175
276
  def add_model_to_use_case(cls, input: AddModelToUseCaseInput) -> GraphQLField:
176
- arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'AddModelToUseCaseInput!', 'value': input}}
177
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
178
- return GraphQLField(field_name='addModelToUseCase', arguments=cleared_arguments)
277
+ arguments: Dict[str, Dict[str, Any]] = {"input": {"type": "AddModelToUseCaseInput!", "value": input}}
278
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
279
+ return GraphQLField(field_name="addModelToUseCase", arguments=cleared_arguments)
179
280
 
180
281
  @classmethod
181
282
  def remove_model_from_use_case(cls, input: RemoveModelFromUseCaseInput) -> GraphQLField:
182
283
  """Removes a model from a use case. If the model is not bound to any other use case or published organisation wide, it is deleted from storage."""
183
- arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'RemoveModelFromUseCaseInput!', 'value': input}}
184
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
185
- return GraphQLField(field_name='removeModelFromUseCase', arguments=cleared_arguments)
284
+ arguments: Dict[str, Dict[str, Any]] = {"input": {"type": "RemoveModelFromUseCaseInput!", "value": input}}
285
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
286
+ return GraphQLField(field_name="removeModelFromUseCase", arguments=cleared_arguments)
186
287
 
187
288
  @classmethod
188
289
  def delete_model(cls, input: DeleteModelInput) -> GraphQLField:
189
290
  """Deletes a model: removes from all use cases, unpublishes from org registry,
190
291
  and deletes from storage."""
191
- arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'DeleteModelInput!', 'value': input}}
192
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
193
- return GraphQLField(field_name='deleteModel', arguments=cleared_arguments)
292
+ arguments: Dict[str, Dict[str, Any]] = {"input": {"type": "DeleteModelInput!", "value": input}}
293
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
294
+ return GraphQLField(field_name="deleteModel", arguments=cleared_arguments)
194
295
 
195
296
  @classmethod
196
297
  def add_remote_env(cls, input: RemoteEnvCreate) -> RemoteEnvFields:
197
- arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'RemoteEnvCreate!', 'value': input}}
198
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
199
- return RemoteEnvFields(field_name='addRemoteEnv', arguments=cleared_arguments)
298
+ arguments: Dict[str, Dict[str, Any]] = {"input": {"type": "RemoteEnvCreate!", "value": input}}
299
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
300
+ return RemoteEnvFields(field_name="addRemoteEnv", arguments=cleared_arguments)
200
301
 
201
302
  @classmethod
202
303
  def remove_remote_env(cls, id_or_key: str) -> GraphQLField:
203
- arguments: Dict[str, Dict[str, Any]] = {'idOrKey': {'type': 'IdOrKey!', 'value': id_or_key}}
204
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
205
- return GraphQLField(field_name='removeRemoteEnv', arguments=cleared_arguments)
304
+ arguments: Dict[str, Dict[str, Any]] = {"idOrKey": {"type": "IdOrKey!", "value": id_or_key}}
305
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
306
+ return GraphQLField(field_name="removeRemoteEnv", arguments=cleared_arguments)
206
307
 
207
308
  @classmethod
208
309
  def test_remote_env(cls, input: RemoteEnvCreate) -> RemoteEnvTestUnion:
209
- arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'RemoteEnvCreate!', 'value': input}}
210
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
211
- return RemoteEnvTestUnion(field_name='testRemoteEnv', arguments=cleared_arguments)
310
+ arguments: Dict[str, Dict[str, Any]] = {"input": {"type": "RemoteEnvCreate!", "value": input}}
311
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
312
+ return RemoteEnvTestUnion(field_name="testRemoteEnv", arguments=cleared_arguments)
212
313
 
213
314
  @classmethod
214
315
  def validate_dataset_schema(cls, remote_env: str, dataset: str, usecase: str) -> DatasetValidationOutputFields:
215
- arguments: Dict[str, Dict[str, Any]] = {'remoteEnv': {'type': 'IdOrKey!', 'value': remote_env}, 'dataset': {'type': 'IdOrKey!', 'value': dataset}, 'usecase': {'type': 'IdOrKey!', 'value': usecase}}
216
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
217
- return DatasetValidationOutputFields(field_name='validateDatasetSchema', arguments=cleared_arguments)
316
+ arguments: Dict[str, Dict[str, Any]] = {
317
+ "remoteEnv": {"type": "IdOrKey!", "value": remote_env},
318
+ "dataset": {"type": "IdOrKey!", "value": dataset},
319
+ "usecase": {"type": "IdOrKey!", "value": usecase},
320
+ }
321
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
322
+ return DatasetValidationOutputFields(field_name="validateDatasetSchema", arguments=cleared_arguments)
218
323
 
219
324
  @classmethod
220
325
  def create_use_case(cls, input: UseCaseCreate) -> UseCaseFields:
221
- arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'UseCaseCreate!', 'value': input}}
222
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
223
- return UseCaseFields(field_name='createUseCase', arguments=cleared_arguments)
326
+ arguments: Dict[str, Dict[str, Any]] = {"input": {"type": "UseCaseCreate!", "value": input}}
327
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
328
+ return UseCaseFields(field_name="createUseCase", arguments=cleared_arguments)
224
329
 
225
330
  @classmethod
226
331
  def update_use_case(cls, id_or_key: str, input: UseCaseUpdate) -> UseCaseFields:
227
- arguments: Dict[str, Dict[str, Any]] = {'idOrKey': {'type': 'IdOrKey!', 'value': id_or_key}, 'input': {'type': 'UseCaseUpdate!', 'value': input}}
228
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
229
- return UseCaseFields(field_name='updateUseCase', arguments=cleared_arguments)
332
+ arguments: Dict[str, Dict[str, Any]] = {
333
+ "idOrKey": {"type": "IdOrKey!", "value": id_or_key},
334
+ "input": {"type": "UseCaseUpdate!", "value": input},
335
+ }
336
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
337
+ return UseCaseFields(field_name="updateUseCase", arguments=cleared_arguments)
230
338
 
231
339
  @classmethod
232
340
  def share_use_case(cls, id_or_key: str, input: UseCaseShares) -> UseCaseFields:
233
- arguments: Dict[str, Dict[str, Any]] = {'idOrKey': {'type': 'IdOrKey!', 'value': id_or_key}, 'input': {'type': 'UseCaseShares!', 'value': input}}
234
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
235
- return UseCaseFields(field_name='shareUseCase', arguments=cleared_arguments)
341
+ arguments: Dict[str, Dict[str, Any]] = {
342
+ "idOrKey": {"type": "IdOrKey!", "value": id_or_key},
343
+ "input": {"type": "UseCaseShares!", "value": input},
344
+ }
345
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
346
+ return UseCaseFields(field_name="shareUseCase", arguments=cleared_arguments)
236
347
 
237
348
  @classmethod
238
349
  def create_api_key(cls, input: ApiKeyCreate) -> ApiKeyFields:
239
- arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'ApiKeyCreate!', 'value': input}}
240
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
241
- return ApiKeyFields(field_name='createApiKey', arguments=cleared_arguments)
350
+ arguments: Dict[str, Dict[str, Any]] = {"input": {"type": "ApiKeyCreate!", "value": input}}
351
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
352
+ return ApiKeyFields(field_name="createApiKey", arguments=cleared_arguments)
242
353
 
243
354
  @classmethod
244
355
  def set_team_member(cls, input: TeamMemberSet) -> TeamMemberFields:
245
- arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'TeamMemberSet!', 'value': input}}
246
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
247
- return TeamMemberFields(field_name='setTeamMember', arguments=cleared_arguments)
356
+ arguments: Dict[str, Dict[str, Any]] = {"input": {"type": "TeamMemberSet!", "value": input}}
357
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
358
+ return TeamMemberFields(field_name="setTeamMember", arguments=cleared_arguments)
248
359
 
249
360
  @classmethod
250
361
  def remove_team_member(cls, input: TeamMemberRemove) -> UserFields:
251
- arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'TeamMemberRemove!', 'value': input}}
252
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
253
- return UserFields(field_name='removeTeamMember', arguments=cleared_arguments)
362
+ arguments: Dict[str, Dict[str, Any]] = {"input": {"type": "TeamMemberRemove!", "value": input}}
363
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
364
+ return UserFields(field_name="removeTeamMember", arguments=cleared_arguments)
254
365
 
255
366
  @classmethod
256
367
  def create_user(cls, input: UserCreate) -> UserFields:
257
- arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'UserCreate!', 'value': input}}
258
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
259
- return UserFields(field_name='createUser', arguments=cleared_arguments)
368
+ arguments: Dict[str, Dict[str, Any]] = {"input": {"type": "UserCreate!", "value": input}}
369
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
370
+ return UserFields(field_name="createUser", arguments=cleared_arguments)
260
371
 
261
372
  @classmethod
262
373
  def delete_user(cls, user: str) -> UserFields:
263
- arguments: Dict[str, Dict[str, Any]] = {'user': {'type': 'IdOrKey!', 'value': user}}
264
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
265
- return UserFields(field_name='deleteUser', arguments=cleared_arguments)
374
+ arguments: Dict[str, Dict[str, Any]] = {"user": {"type": "IdOrKey!", "value": user}}
375
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
376
+ return UserFields(field_name="deleteUser", arguments=cleared_arguments)
266
377
 
267
378
  @classmethod
268
379
  def create_role(cls, input: RoleCreate) -> RoleFields:
269
- arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'RoleCreate!', 'value': input}}
270
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
271
- return RoleFields(field_name='createRole', arguments=cleared_arguments)
380
+ arguments: Dict[str, Dict[str, Any]] = {"input": {"type": "RoleCreate!", "value": input}}
381
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
382
+ return RoleFields(field_name="createRole", arguments=cleared_arguments)
272
383
 
273
384
  @classmethod
274
385
  def create_team(cls, input: TeamCreate) -> TeamFields:
275
- arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'TeamCreate!', 'value': input}}
276
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
277
- return TeamFields(field_name='createTeam', arguments=cleared_arguments)
386
+ arguments: Dict[str, Dict[str, Any]] = {"input": {"type": "TeamCreate!", "value": input}}
387
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
388
+ return TeamFields(field_name="createTeam", arguments=cleared_arguments)
278
389
 
279
390
  @classmethod
280
391
  def update_feedback(cls, id: Any, input: FeedbackUpdateInput) -> DirectFeedbackFields:
281
- arguments: Dict[str, Dict[str, Any]] = {'id': {'type': 'UUID!', 'value': id}, 'input': {'type': 'FeedbackUpdateInput!', 'value': input}}
282
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
283
- return DirectFeedbackFields(field_name='updateFeedback', arguments=cleared_arguments)
392
+ arguments: Dict[str, Dict[str, Any]] = {
393
+ "id": {"type": "UUID!", "value": id},
394
+ "input": {"type": "FeedbackUpdateInput!", "value": input},
395
+ }
396
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
397
+ return DirectFeedbackFields(field_name="updateFeedback", arguments=cleared_arguments)
284
398
 
285
399
  @classmethod
286
400
  def add_direct_feedback(cls, completion_id: Any, metric_id: str, input: FeedbackAddInput) -> DirectFeedbackFields:
287
- arguments: Dict[str, Dict[str, Any]] = {'completionId': {'type': 'UUID!', 'value': completion_id}, 'metricId': {'type': 'IdOrKey!', 'value': metric_id}, 'input': {'type': 'FeedbackAddInput!', 'value': input}}
288
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
289
- return DirectFeedbackFields(field_name='addDirectFeedback', arguments=cleared_arguments)
401
+ arguments: Dict[str, Dict[str, Any]] = {
402
+ "completionId": {"type": "UUID!", "value": completion_id},
403
+ "metricId": {"type": "IdOrKey!", "value": metric_id},
404
+ "input": {"type": "FeedbackAddInput!", "value": input},
405
+ }
406
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
407
+ return DirectFeedbackFields(field_name="addDirectFeedback", arguments=cleared_arguments)
290
408
 
291
409
  @classmethod
292
410
  def create_judge(cls, use_case: str, input: JudgeCreate) -> JudgeFields:
293
- arguments: Dict[str, Dict[str, Any]] = {'useCase': {'type': 'IdOrKey!', 'value': use_case}, 'input': {'type': 'JudgeCreate!', 'value': input}}
294
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
295
- return JudgeFields(field_name='createJudge', arguments=cleared_arguments)
411
+ arguments: Dict[str, Dict[str, Any]] = {
412
+ "useCase": {"type": "IdOrKey!", "value": use_case},
413
+ "input": {"type": "JudgeCreate!", "value": input},
414
+ }
415
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
416
+ return JudgeFields(field_name="createJudge", arguments=cleared_arguments)
296
417
 
297
418
  @classmethod
298
419
  def create_prebuilt_judge(cls, use_case: str, input: PrebuiltJudgeCreate) -> JudgeFields:
299
- arguments: Dict[str, Dict[str, Any]] = {'useCase': {'type': 'IdOrKey!', 'value': use_case}, 'input': {'type': 'PrebuiltJudgeCreate!', 'value': input}}
300
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
301
- return JudgeFields(field_name='createPrebuiltJudge', arguments=cleared_arguments)
420
+ arguments: Dict[str, Dict[str, Any]] = {
421
+ "useCase": {"type": "IdOrKey!", "value": use_case},
422
+ "input": {"type": "PrebuiltJudgeCreate!", "value": input},
423
+ }
424
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
425
+ return JudgeFields(field_name="createPrebuiltJudge", arguments=cleared_arguments)
302
426
 
303
427
  @classmethod
304
428
  def update_judge(cls, use_case: str, key: str, input: JudgeUpdate) -> JudgeFields:
305
- arguments: Dict[str, Dict[str, Any]] = {'useCase': {'type': 'IdOrKey!', 'value': use_case}, 'key': {'type': 'String!', 'value': key}, 'input': {'type': 'JudgeUpdate!', 'value': input}}
306
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
307
- return JudgeFields(field_name='updateJudge', arguments=cleared_arguments)
429
+ arguments: Dict[str, Dict[str, Any]] = {
430
+ "useCase": {"type": "IdOrKey!", "value": use_case},
431
+ "key": {"type": "String!", "value": key},
432
+ "input": {"type": "JudgeUpdate!", "value": input},
433
+ }
434
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
435
+ return JudgeFields(field_name="updateJudge", arguments=cleared_arguments)
308
436
 
309
437
  @classmethod
310
438
  def delete_judge(cls, use_case: str, key: str) -> DeleteConfirmFields:
311
- arguments: Dict[str, Dict[str, Any]] = {'useCase': {'type': 'IdOrKey!', 'value': use_case}, 'key': {'type': 'String!', 'value': key}}
312
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
313
- return DeleteConfirmFields(field_name='deleteJudge', arguments=cleared_arguments)
439
+ arguments: Dict[str, Dict[str, Any]] = {
440
+ "useCase": {"type": "IdOrKey!", "value": use_case},
441
+ "key": {"type": "String!", "value": key},
442
+ }
443
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
444
+ return DeleteConfirmFields(field_name="deleteJudge", arguments=cleared_arguments)
314
445
 
315
446
  @classmethod
316
447
  def create_grader(cls, use_case: str, input: GraderCreateInput) -> GraderFields:
317
- arguments: Dict[str, Dict[str, Any]] = {'useCase': {'type': 'IdOrKey!', 'value': use_case}, 'input': {'type': 'GraderCreateInput!', 'value': input}}
318
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
319
- return GraderFields(field_name='createGrader', arguments=cleared_arguments)
448
+ arguments: Dict[str, Dict[str, Any]] = {
449
+ "useCase": {"type": "IdOrKey!", "value": use_case},
450
+ "input": {"type": "GraderCreateInput!", "value": input},
451
+ }
452
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
453
+ return GraderFields(field_name="createGrader", arguments=cleared_arguments)
320
454
 
321
455
  @classmethod
322
456
  def update_grader(cls, use_case: str, id: str, input: GraderUpdateInput) -> GraderFields:
323
- arguments: Dict[str, Dict[str, Any]] = {'useCase': {'type': 'IdOrKey!', 'value': use_case}, 'id': {'type': 'IdOrKey!', 'value': id}, 'input': {'type': 'GraderUpdateInput!', 'value': input}}
324
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
325
- return GraderFields(field_name='updateGrader', arguments=cleared_arguments)
457
+ arguments: Dict[str, Dict[str, Any]] = {
458
+ "useCase": {"type": "IdOrKey!", "value": use_case},
459
+ "id": {"type": "IdOrKey!", "value": id},
460
+ "input": {"type": "GraderUpdateInput!", "value": input},
461
+ }
462
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
463
+ return GraderFields(field_name="updateGrader", arguments=cleared_arguments)
326
464
 
327
465
  @classmethod
328
466
  def delete_grader(cls, use_case: str, id: str) -> DeleteConfirmFields:
329
- arguments: Dict[str, Dict[str, Any]] = {'useCase': {'type': 'IdOrKey!', 'value': use_case}, 'id': {'type': 'IdOrKey!', 'value': id}}
330
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
331
- return DeleteConfirmFields(field_name='deleteGrader', arguments=cleared_arguments)
467
+ arguments: Dict[str, Dict[str, Any]] = {
468
+ "useCase": {"type": "IdOrKey!", "value": use_case},
469
+ "id": {"type": "IdOrKey!", "value": id},
470
+ }
471
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
472
+ return DeleteConfirmFields(field_name="deleteGrader", arguments=cleared_arguments)
332
473
 
333
474
  @classmethod
334
475
  def lock_grader(cls, use_case: str, id: str, locked: bool) -> GraderFields:
335
- arguments: Dict[str, Dict[str, Any]] = {'useCase': {'type': 'IdOrKey!', 'value': use_case}, 'id': {'type': 'IdOrKey!', 'value': id}, 'locked': {'type': 'Boolean!', 'value': locked}}
336
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
337
- return GraderFields(field_name='lockGrader', arguments=cleared_arguments)
476
+ arguments: Dict[str, Dict[str, Any]] = {
477
+ "useCase": {"type": "IdOrKey!", "value": use_case},
478
+ "id": {"type": "IdOrKey!", "value": id},
479
+ "locked": {"type": "Boolean!", "value": locked},
480
+ }
481
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
482
+ return GraderFields(field_name="lockGrader", arguments=cleared_arguments)
338
483
 
339
484
  @classmethod
340
485
  def create_tool_provider(cls, use_case: str, input: CreateToolProviderInput) -> ToolProviderFields:
341
- arguments: Dict[str, Dict[str, Any]] = {'useCase': {'type': 'IdOrKey!', 'value': use_case}, 'input': {'type': 'CreateToolProviderInput!', 'value': input}}
342
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
343
- return ToolProviderFields(field_name='createToolProvider', arguments=cleared_arguments)
486
+ arguments: Dict[str, Dict[str, Any]] = {
487
+ "useCase": {"type": "IdOrKey!", "value": use_case},
488
+ "input": {"type": "CreateToolProviderInput!", "value": input},
489
+ }
490
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
491
+ return ToolProviderFields(field_name="createToolProvider", arguments=cleared_arguments)
344
492
 
345
493
  @classmethod
346
494
  def update_tool_provider(cls, id_or_key: str, use_case: str, input: UpdateToolProviderInput) -> ToolProviderFields:
347
- arguments: Dict[str, Dict[str, Any]] = {'idOrKey': {'type': 'IdOrKey!', 'value': id_or_key}, 'useCase': {'type': 'IdOrKey!', 'value': use_case}, 'input': {'type': 'UpdateToolProviderInput!', 'value': input}}
348
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
349
- return ToolProviderFields(field_name='updateToolProvider', arguments=cleared_arguments)
495
+ arguments: Dict[str, Dict[str, Any]] = {
496
+ "idOrKey": {"type": "IdOrKey!", "value": id_or_key},
497
+ "useCase": {"type": "IdOrKey!", "value": use_case},
498
+ "input": {"type": "UpdateToolProviderInput!", "value": input},
499
+ }
500
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
501
+ return ToolProviderFields(field_name="updateToolProvider", arguments=cleared_arguments)
350
502
 
351
503
  @classmethod
352
504
  def delete_tool_provider(cls, id_or_key: str, use_case: str) -> GraphQLField:
353
- arguments: Dict[str, Dict[str, Any]] = {'idOrKey': {'type': 'IdOrKey!', 'value': id_or_key}, 'useCase': {'type': 'IdOrKey!', 'value': use_case}}
354
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
355
- return GraphQLField(field_name='deleteToolProvider', arguments=cleared_arguments)
505
+ arguments: Dict[str, Dict[str, Any]] = {
506
+ "idOrKey": {"type": "IdOrKey!", "value": id_or_key},
507
+ "useCase": {"type": "IdOrKey!", "value": use_case},
508
+ }
509
+ cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
510
+ return GraphQLField(field_name="deleteToolProvider", arguments=cleared_arguments)