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.
- adaptive_sdk/graphql_client/__init__.py +704 -12
- adaptive_sdk/graphql_client/custom_fields.py +1640 -960
- adaptive_sdk/graphql_client/custom_mutations.py +332 -177
- adaptive_sdk/graphql_client/custom_queries.py +278 -133
- adaptive_sdk/graphql_client/custom_typing_fields.py +74 -0
- adaptive_sdk/graphql_client/enums.py +38 -0
- adaptive_sdk/graphql_client/input_types.py +586 -213
- adaptive_sdk/resources/abtests.py +2 -0
- adaptive_sdk/resources/base_resource.py +22 -0
- adaptive_sdk/resources/chat.py +16 -7
- adaptive_sdk/resources/compute_pools.py +16 -4
- adaptive_sdk/resources/embeddings.py +11 -7
- adaptive_sdk/resources/feedback.py +22 -33
- adaptive_sdk/resources/interactions.py +6 -3
- adaptive_sdk/resources/jobs.py +90 -6
- adaptive_sdk/resources/models.py +20 -3
- adaptive_sdk/resources/permissions.py +12 -1
- adaptive_sdk/resources/recipes.py +177 -31
- adaptive_sdk/resources/roles.py +15 -8
- adaptive_sdk/resources/teams.py +30 -1
- adaptive_sdk/resources/use_cases.py +9 -6
- adaptive_sdk/resources/users.py +37 -22
- {adaptive_sdk-0.1.14.dist-info → adaptive_sdk-0.12.1.dist-info}/METADATA +1 -1
- {adaptive_sdk-0.1.14.dist-info → adaptive_sdk-0.12.1.dist-info}/RECORD +25 -25
- {adaptive_sdk-0.1.14.dist-info → adaptive_sdk-0.12.1.dist-info}/WHEEL +0 -0
|
@@ -1,355 +1,510 @@
|
|
|
1
1
|
from typing import Any, Dict, Optional
|
|
2
2
|
from .base_model import Upload
|
|
3
|
-
from .custom_fields import
|
|
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
|
|
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]] = {
|
|
13
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
14
|
-
return AbcampaignFields(field_name=
|
|
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]] = {
|
|
19
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
20
|
-
return GraphQLField(field_name=
|
|
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]] = {
|
|
25
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
26
|
-
return GraphQLField(field_name=
|
|
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]] = {
|
|
31
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
32
|
-
return GraphQLField(field_name=
|
|
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]] = {
|
|
37
|
-
|
|
38
|
-
|
|
39
|
-
|
|
40
|
-
|
|
41
|
-
|
|
42
|
-
|
|
43
|
-
|
|
44
|
-
|
|
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]] = {
|
|
49
|
-
|
|
50
|
-
|
|
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(
|
|
54
|
-
|
|
55
|
-
|
|
56
|
-
|
|
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]] = {
|
|
61
|
-
|
|
62
|
-
|
|
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]] = {
|
|
67
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
68
|
-
return DatasetFields(field_name=
|
|
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]] = {
|
|
73
|
-
|
|
74
|
-
|
|
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]] = {
|
|
79
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
80
|
-
return CompletionFields(field_name=
|
|
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]] = {
|
|
85
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
86
|
-
return SystemPromptTemplateFields(field_name=
|
|
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]] = {
|
|
91
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
92
|
-
return SystemPromptTemplateFields(field_name=
|
|
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]] = {
|
|
97
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
98
|
-
return JobFields(field_name=
|
|
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]] = {
|
|
103
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
104
|
-
return JobFields(field_name=
|
|
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]] = {
|
|
109
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
110
|
-
return MetricFields(field_name=
|
|
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]] = {
|
|
116
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
117
|
-
return MetricWithContextFields(field_name=
|
|
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]] = {
|
|
122
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
123
|
-
return GraphQLField(field_name=
|
|
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]] = {
|
|
128
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
129
|
-
return ModelServiceFields(field_name=
|
|
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]] = {
|
|
134
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
135
|
-
return ModelFields(field_name=
|
|
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]] = {
|
|
140
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
141
|
-
return ModelServiceFields(field_name=
|
|
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]] = {
|
|
147
|
-
|
|
148
|
-
|
|
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]] = {
|
|
153
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
154
|
-
return ModelFields(field_name=
|
|
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]] = {
|
|
159
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
160
|
-
return ModelFields(field_name=
|
|
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]] = {
|
|
165
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
166
|
-
return JobFields(field_name=
|
|
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]] = {
|
|
171
|
-
|
|
172
|
-
|
|
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]] = {
|
|
177
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
178
|
-
return GraphQLField(field_name=
|
|
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]] = {
|
|
184
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
185
|
-
return GraphQLField(field_name=
|
|
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]] = {
|
|
192
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
193
|
-
return GraphQLField(field_name=
|
|
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]] = {
|
|
198
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
199
|
-
return RemoteEnvFields(field_name=
|
|
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]] = {
|
|
204
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
205
|
-
return GraphQLField(field_name=
|
|
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]] = {
|
|
210
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
211
|
-
return RemoteEnvTestUnion(field_name=
|
|
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]] = {
|
|
216
|
-
|
|
217
|
-
|
|
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]] = {
|
|
222
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
223
|
-
return UseCaseFields(field_name=
|
|
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]] = {
|
|
228
|
-
|
|
229
|
-
|
|
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]] = {
|
|
234
|
-
|
|
235
|
-
|
|
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]] = {
|
|
240
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
241
|
-
return ApiKeyFields(field_name=
|
|
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]] = {
|
|
246
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
247
|
-
return TeamMemberFields(field_name=
|
|
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]] = {
|
|
252
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
253
|
-
return UserFields(field_name=
|
|
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]] = {
|
|
258
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
259
|
-
return UserFields(field_name=
|
|
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]] = {
|
|
264
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
265
|
-
return UserFields(field_name=
|
|
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]] = {
|
|
270
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
271
|
-
return RoleFields(field_name=
|
|
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]] = {
|
|
276
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
277
|
-
return TeamFields(field_name=
|
|
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]] = {
|
|
282
|
-
|
|
283
|
-
|
|
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]] = {
|
|
288
|
-
|
|
289
|
-
|
|
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]] = {
|
|
294
|
-
|
|
295
|
-
|
|
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]] = {
|
|
300
|
-
|
|
301
|
-
|
|
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]] = {
|
|
306
|
-
|
|
307
|
-
|
|
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]] = {
|
|
312
|
-
|
|
313
|
-
|
|
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]] = {
|
|
318
|
-
|
|
319
|
-
|
|
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]] = {
|
|
324
|
-
|
|
325
|
-
|
|
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]] = {
|
|
330
|
-
|
|
331
|
-
|
|
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]] = {
|
|
336
|
-
|
|
337
|
-
|
|
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]] = {
|
|
342
|
-
|
|
343
|
-
|
|
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]] = {
|
|
348
|
-
|
|
349
|
-
|
|
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]] = {
|
|
354
|
-
|
|
355
|
-
|
|
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)
|