adaptive-sdk 0.12.0__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.12.0.dist-info → adaptive_sdk-0.12.1.dist-info}/METADATA +1 -1
- {adaptive_sdk-0.12.0.dist-info → adaptive_sdk-0.12.1.dist-info}/RECORD +25 -25
- {adaptive_sdk-0.12.0.dist-info → adaptive_sdk-0.12.1.dist-info}/WHEEL +0 -0
|
@@ -1,2242 +1,2922 @@
|
|
|
1
1
|
from typing import Any, Dict, List, Optional, Union
|
|
2
2
|
from .base_operation import GraphQLField
|
|
3
|
-
from .custom_typing_fields import
|
|
4
|
-
|
|
3
|
+
from .custom_typing_fields import (
|
|
4
|
+
AbcampaignGraphQLField,
|
|
5
|
+
AbReportGraphQLField,
|
|
6
|
+
AbVariantReportComparisonGraphQLField,
|
|
7
|
+
AbVariantReportGraphQLField,
|
|
8
|
+
ActivityGraphQLField,
|
|
9
|
+
ActivityOutputGraphQLField,
|
|
10
|
+
ApiKeyGraphQLField,
|
|
11
|
+
ArtifactByproductsUnion,
|
|
12
|
+
AuthProviderGraphQLField,
|
|
13
|
+
BatchInferenceJobStageOutputGraphQLField,
|
|
14
|
+
BillingUsageGraphQLField,
|
|
15
|
+
ChatMessageGraphQLField,
|
|
16
|
+
ComparisonFeedbackGraphQLField,
|
|
17
|
+
CompletionConnectionGraphQLField,
|
|
18
|
+
CompletionEdgeGraphQLField,
|
|
19
|
+
CompletionGraphQLField,
|
|
20
|
+
CompletionGroupDataConnectionGraphQLField,
|
|
21
|
+
CompletionGroupDataEdgeGraphQLField,
|
|
22
|
+
CompletionGroupDataGraphQLField,
|
|
23
|
+
CompletionGroupFeedbackStatsGraphQLField,
|
|
24
|
+
CompletionHistoryEntryOuputGraphQLField,
|
|
25
|
+
CompletionLabelGraphQLField,
|
|
26
|
+
CompletionMetadataGraphQLField,
|
|
27
|
+
ComputePoolGraphQLField,
|
|
28
|
+
ContractGraphQLField,
|
|
29
|
+
CustomConfigOutputGraphQLField,
|
|
30
|
+
CustomRecipeGraphQLField,
|
|
31
|
+
CustomRecipeJobDetailsGraphQLField,
|
|
32
|
+
DatasetByproductsGraphQLField,
|
|
33
|
+
DatasetGraphQLField,
|
|
34
|
+
DatasetMetricUsageGraphQLField,
|
|
35
|
+
DatasetProgressGraphQLField,
|
|
36
|
+
DatasetUploadProcessingStatusGraphQLField,
|
|
37
|
+
DatasetValidationOutputGraphQLField,
|
|
38
|
+
DeleteConfirmGraphQLField,
|
|
39
|
+
DirectFeedbackGraphQLField,
|
|
40
|
+
EmojiGraphQLField,
|
|
41
|
+
EvalJobStageOutputGraphQLField,
|
|
42
|
+
EvaluationByproductsGraphQLField,
|
|
43
|
+
EvaluationResultGraphQLField,
|
|
44
|
+
GlobalUsageGraphQLField,
|
|
45
|
+
GpuAllocationGraphQLField,
|
|
46
|
+
GraderConfigUnion,
|
|
47
|
+
GraderGraphQLField,
|
|
48
|
+
HarmonyGroupGraphQLField,
|
|
49
|
+
InteractionOutputGraphQLField,
|
|
50
|
+
IntervalGraphQLField,
|
|
51
|
+
JobArtifactGraphQLField,
|
|
52
|
+
JobConnectionGraphQLField,
|
|
53
|
+
JobEdgeGraphQLField,
|
|
54
|
+
JobGraphQLField,
|
|
55
|
+
JobStageInfoOutputUnion,
|
|
56
|
+
JobStageOutputGraphQLField,
|
|
57
|
+
JudgeConfigOutputGraphQLField,
|
|
58
|
+
JudgeExampleGraphQLField,
|
|
59
|
+
JudgeGraphQLField,
|
|
60
|
+
LabelGraphQLField,
|
|
61
|
+
LabelKeyUsageGraphQLField,
|
|
62
|
+
LabelUsageGraphQLField,
|
|
63
|
+
LabelValueUsageGraphQLField,
|
|
64
|
+
MetaObjectGraphQLField,
|
|
65
|
+
MetricActivityGraphQLField,
|
|
66
|
+
MetricGraphQLField,
|
|
67
|
+
MetricWithContextGraphQLField,
|
|
68
|
+
ModelByproductsGraphQLField,
|
|
69
|
+
ModelComputeConfigOutputGraphQLField,
|
|
70
|
+
ModelGraphQLField,
|
|
71
|
+
ModelPlacementOutputGraphQLField,
|
|
72
|
+
ModelServiceGraphQLField,
|
|
73
|
+
PageInfoGraphQLField,
|
|
74
|
+
PrebuiltConfigDefinitionGraphQLField,
|
|
75
|
+
PrebuiltConfigOutputGraphQLField,
|
|
76
|
+
PrebuiltCriteriaGraphQLField,
|
|
77
|
+
ProviderListGraphQLField,
|
|
78
|
+
RemoteConfigOutputGraphQLField,
|
|
79
|
+
RemoteEnvGraphQLField,
|
|
80
|
+
RemoteEnvTestOfflineGraphQLField,
|
|
81
|
+
RemoteEnvTestOnlineGraphQLField,
|
|
82
|
+
RoleGraphQLField,
|
|
83
|
+
SearchResultGraphQLField,
|
|
84
|
+
SessionGraphQLField,
|
|
85
|
+
SettingsGraphQLField,
|
|
86
|
+
ShareGraphQLField,
|
|
87
|
+
SystemPromptTemplateGraphQLField,
|
|
88
|
+
TeamGraphQLField,
|
|
89
|
+
TeamMemberGraphQLField,
|
|
90
|
+
TeamWithroleGraphQLField,
|
|
91
|
+
TimeseriesGraphQLField,
|
|
92
|
+
ToolProviderGraphQLField,
|
|
93
|
+
TrainingJobStageOutputGraphQLField,
|
|
94
|
+
TrendResultGraphQLField,
|
|
95
|
+
UnitConfigGraphQLField,
|
|
96
|
+
UsageAggregateItemGraphQLField,
|
|
97
|
+
UsageAggregatePerUseCaseItemGraphQLField,
|
|
98
|
+
UsageGraphQLField,
|
|
99
|
+
UsageStatsByModelGraphQLField,
|
|
100
|
+
UsageStatsGraphQLField,
|
|
101
|
+
UseCaseGraphQLField,
|
|
102
|
+
UseCaseItemGraphQLField,
|
|
103
|
+
UseCaseMetadataGraphQLField,
|
|
104
|
+
UserGraphQLField,
|
|
105
|
+
WidgetGraphQLField,
|
|
106
|
+
)
|
|
107
|
+
from .input_types import (
|
|
108
|
+
AbCampaignFilter,
|
|
109
|
+
ArtifactFilter,
|
|
110
|
+
CursorPageInput,
|
|
111
|
+
FeedbackFilterInput,
|
|
112
|
+
ListCompletionsFilterInput,
|
|
113
|
+
MetricTrendInput,
|
|
114
|
+
ModelFilter,
|
|
115
|
+
ModelServiceFilter,
|
|
116
|
+
OrderPair,
|
|
117
|
+
TimeRange,
|
|
118
|
+
TimeseriesInput,
|
|
119
|
+
UseCaseFilter,
|
|
120
|
+
)
|
|
121
|
+
|
|
5
122
|
|
|
6
123
|
class AbReportFields(GraphQLField):
|
|
7
124
|
"""@private"""
|
|
8
|
-
|
|
125
|
+
|
|
126
|
+
p_value: "AbReportGraphQLField" = AbReportGraphQLField("pValue")
|
|
9
127
|
|
|
10
128
|
@classmethod
|
|
11
|
-
def variants(cls) ->
|
|
12
|
-
return AbVariantReportFields(
|
|
129
|
+
def variants(cls) -> "AbVariantReportFields":
|
|
130
|
+
return AbVariantReportFields("variants")
|
|
13
131
|
|
|
14
|
-
def fields(self, *subfields: Union[AbReportGraphQLField,
|
|
132
|
+
def fields(self, *subfields: Union[AbReportGraphQLField, "AbVariantReportFields"]) -> "AbReportFields":
|
|
15
133
|
"""Subfields should come from the AbReportFields class"""
|
|
16
134
|
self._subfields.extend(subfields)
|
|
17
135
|
return self
|
|
18
136
|
|
|
19
|
-
def alias(self, alias: str) ->
|
|
137
|
+
def alias(self, alias: str) -> "AbReportFields":
|
|
20
138
|
self._alias = alias
|
|
21
139
|
return self
|
|
22
140
|
|
|
141
|
+
|
|
23
142
|
class AbVariantReportFields(GraphQLField):
|
|
24
143
|
"""@private"""
|
|
25
144
|
|
|
26
145
|
@classmethod
|
|
27
|
-
def variant(cls) ->
|
|
28
|
-
return ModelServiceFields(
|
|
146
|
+
def variant(cls) -> "ModelServiceFields":
|
|
147
|
+
return ModelServiceFields("variant")
|
|
29
148
|
|
|
30
149
|
@classmethod
|
|
31
|
-
def interval(cls) ->
|
|
32
|
-
return IntervalFields(
|
|
33
|
-
|
|
34
|
-
|
|
150
|
+
def interval(cls) -> "IntervalFields":
|
|
151
|
+
return IntervalFields("interval")
|
|
152
|
+
|
|
153
|
+
mean: "AbVariantReportGraphQLField" = AbVariantReportGraphQLField("mean")
|
|
154
|
+
feedbacks: "AbVariantReportGraphQLField" = AbVariantReportGraphQLField("feedbacks")
|
|
35
155
|
|
|
36
156
|
@classmethod
|
|
37
|
-
def comparisons(cls) ->
|
|
38
|
-
return AbVariantReportComparisonFields(
|
|
157
|
+
def comparisons(cls) -> "AbVariantReportComparisonFields":
|
|
158
|
+
return AbVariantReportComparisonFields("comparisons")
|
|
39
159
|
|
|
40
|
-
def fields(
|
|
160
|
+
def fields(
|
|
161
|
+
self,
|
|
162
|
+
*subfields: Union[
|
|
163
|
+
AbVariantReportGraphQLField, "AbVariantReportComparisonFields", "IntervalFields", "ModelServiceFields"
|
|
164
|
+
],
|
|
165
|
+
) -> "AbVariantReportFields":
|
|
41
166
|
"""Subfields should come from the AbVariantReportFields class"""
|
|
42
167
|
self._subfields.extend(subfields)
|
|
43
168
|
return self
|
|
44
169
|
|
|
45
|
-
def alias(self, alias: str) ->
|
|
170
|
+
def alias(self, alias: str) -> "AbVariantReportFields":
|
|
46
171
|
self._alias = alias
|
|
47
172
|
return self
|
|
48
173
|
|
|
174
|
+
|
|
49
175
|
class AbVariantReportComparisonFields(GraphQLField):
|
|
50
176
|
"""@private"""
|
|
51
177
|
|
|
52
178
|
@classmethod
|
|
53
|
-
def variant(cls) ->
|
|
54
|
-
return ModelServiceFields(
|
|
55
|
-
|
|
56
|
-
|
|
57
|
-
|
|
58
|
-
|
|
59
|
-
|
|
179
|
+
def variant(cls) -> "ModelServiceFields":
|
|
180
|
+
return ModelServiceFields("variant")
|
|
181
|
+
|
|
182
|
+
feedbacks: "AbVariantReportComparisonGraphQLField" = AbVariantReportComparisonGraphQLField("feedbacks")
|
|
183
|
+
wins: "AbVariantReportComparisonGraphQLField" = AbVariantReportComparisonGraphQLField("wins")
|
|
184
|
+
losses: "AbVariantReportComparisonGraphQLField" = AbVariantReportComparisonGraphQLField("losses")
|
|
185
|
+
ties_good: "AbVariantReportComparisonGraphQLField" = AbVariantReportComparisonGraphQLField("tiesGood")
|
|
186
|
+
ties_bad: "AbVariantReportComparisonGraphQLField" = AbVariantReportComparisonGraphQLField("tiesBad")
|
|
60
187
|
|
|
61
|
-
def fields(
|
|
188
|
+
def fields(
|
|
189
|
+
self, *subfields: Union[AbVariantReportComparisonGraphQLField, "ModelServiceFields"]
|
|
190
|
+
) -> "AbVariantReportComparisonFields":
|
|
62
191
|
"""Subfields should come from the AbVariantReportComparisonFields class"""
|
|
63
192
|
self._subfields.extend(subfields)
|
|
64
193
|
return self
|
|
65
194
|
|
|
66
|
-
def alias(self, alias: str) ->
|
|
195
|
+
def alias(self, alias: str) -> "AbVariantReportComparisonFields":
|
|
67
196
|
self._alias = alias
|
|
68
197
|
return self
|
|
69
198
|
|
|
199
|
+
|
|
70
200
|
class AbcampaignFields(GraphQLField):
|
|
71
201
|
"""@private"""
|
|
72
|
-
|
|
73
|
-
|
|
74
|
-
|
|
202
|
+
|
|
203
|
+
id: "AbcampaignGraphQLField" = AbcampaignGraphQLField("id")
|
|
204
|
+
key: "AbcampaignGraphQLField" = AbcampaignGraphQLField("key")
|
|
205
|
+
name: "AbcampaignGraphQLField" = AbcampaignGraphQLField("name")
|
|
75
206
|
|
|
76
207
|
@classmethod
|
|
77
|
-
def metric(cls) ->
|
|
78
|
-
return MetricFields(
|
|
208
|
+
def metric(cls) -> "MetricFields":
|
|
209
|
+
return MetricFields("metric")
|
|
79
210
|
|
|
80
211
|
@classmethod
|
|
81
|
-
def use_case(cls) ->
|
|
82
|
-
return UseCaseFields(
|
|
83
|
-
|
|
84
|
-
|
|
85
|
-
|
|
86
|
-
|
|
87
|
-
|
|
88
|
-
|
|
89
|
-
|
|
212
|
+
def use_case(cls) -> "UseCaseFields":
|
|
213
|
+
return UseCaseFields("useCase")
|
|
214
|
+
|
|
215
|
+
auto_deploy: "AbcampaignGraphQLField" = AbcampaignGraphQLField("autoDeploy")
|
|
216
|
+
status: "AbcampaignGraphQLField" = AbcampaignGraphQLField("status")
|
|
217
|
+
feedback_type: "AbcampaignGraphQLField" = AbcampaignGraphQLField("feedbackType")
|
|
218
|
+
traffic_split: "AbcampaignGraphQLField" = AbcampaignGraphQLField("trafficSplit")
|
|
219
|
+
begin_date: "AbcampaignGraphQLField" = AbcampaignGraphQLField("beginDate")
|
|
220
|
+
end_date: "AbcampaignGraphQLField" = AbcampaignGraphQLField("endDate")
|
|
221
|
+
created_at: "AbcampaignGraphQLField" = AbcampaignGraphQLField("createdAt")
|
|
90
222
|
|
|
91
223
|
@classmethod
|
|
92
|
-
def report(cls) ->
|
|
93
|
-
return AbReportFields(
|
|
224
|
+
def report(cls) -> "AbReportFields":
|
|
225
|
+
return AbReportFields("report")
|
|
94
226
|
|
|
95
227
|
@classmethod
|
|
96
|
-
def models(cls) ->
|
|
97
|
-
return ModelServiceFields(
|
|
98
|
-
feedbacks: 'AbcampaignGraphQLField' = AbcampaignGraphQLField('feedbacks')
|
|
99
|
-
has_enough_feedbacks: 'AbcampaignGraphQLField' = AbcampaignGraphQLField('hasEnoughFeedbacks')
|
|
228
|
+
def models(cls) -> "ModelServiceFields":
|
|
229
|
+
return ModelServiceFields("models")
|
|
100
230
|
|
|
101
|
-
|
|
231
|
+
feedbacks: "AbcampaignGraphQLField" = AbcampaignGraphQLField("feedbacks")
|
|
232
|
+
has_enough_feedbacks: "AbcampaignGraphQLField" = AbcampaignGraphQLField("hasEnoughFeedbacks")
|
|
233
|
+
|
|
234
|
+
def fields(
|
|
235
|
+
self,
|
|
236
|
+
*subfields: Union[
|
|
237
|
+
AbcampaignGraphQLField, "AbReportFields", "MetricFields", "ModelServiceFields", "UseCaseFields"
|
|
238
|
+
],
|
|
239
|
+
) -> "AbcampaignFields":
|
|
102
240
|
"""Subfields should come from the AbcampaignFields class"""
|
|
103
241
|
self._subfields.extend(subfields)
|
|
104
242
|
return self
|
|
105
243
|
|
|
106
|
-
def alias(self, alias: str) ->
|
|
244
|
+
def alias(self, alias: str) -> "AbcampaignFields":
|
|
107
245
|
self._alias = alias
|
|
108
246
|
return self
|
|
109
247
|
|
|
248
|
+
|
|
110
249
|
class ActivityFields(GraphQLField):
|
|
111
250
|
"""@private"""
|
|
112
251
|
|
|
113
252
|
@classmethod
|
|
114
|
-
def interactions(cls) ->
|
|
115
|
-
return InteractionOutputFields(
|
|
253
|
+
def interactions(cls) -> "InteractionOutputFields":
|
|
254
|
+
return InteractionOutputFields("interactions")
|
|
116
255
|
|
|
117
256
|
@classmethod
|
|
118
|
-
def feedbacks(cls) ->
|
|
119
|
-
return ActivityOutputFields(
|
|
257
|
+
def feedbacks(cls) -> "ActivityOutputFields":
|
|
258
|
+
return ActivityOutputFields("feedbacks")
|
|
120
259
|
|
|
121
|
-
def fields(
|
|
260
|
+
def fields(
|
|
261
|
+
self, *subfields: Union[ActivityGraphQLField, "ActivityOutputFields", "InteractionOutputFields"]
|
|
262
|
+
) -> "ActivityFields":
|
|
122
263
|
"""Subfields should come from the ActivityFields class"""
|
|
123
264
|
self._subfields.extend(subfields)
|
|
124
265
|
return self
|
|
125
266
|
|
|
126
|
-
def alias(self, alias: str) ->
|
|
267
|
+
def alias(self, alias: str) -> "ActivityFields":
|
|
127
268
|
self._alias = alias
|
|
128
269
|
return self
|
|
129
270
|
|
|
271
|
+
|
|
130
272
|
class ActivityOutputFields(GraphQLField):
|
|
131
273
|
"""@private"""
|
|
132
|
-
value: 'ActivityOutputGraphQLField' = ActivityOutputGraphQLField('value')
|
|
133
|
-
trend: 'ActivityOutputGraphQLField' = ActivityOutputGraphQLField('trend')
|
|
134
274
|
|
|
135
|
-
|
|
275
|
+
value: "ActivityOutputGraphQLField" = ActivityOutputGraphQLField("value")
|
|
276
|
+
trend: "ActivityOutputGraphQLField" = ActivityOutputGraphQLField("trend")
|
|
277
|
+
|
|
278
|
+
def fields(self, *subfields: ActivityOutputGraphQLField) -> "ActivityOutputFields":
|
|
136
279
|
"""Subfields should come from the ActivityOutputFields class"""
|
|
137
280
|
self._subfields.extend(subfields)
|
|
138
281
|
return self
|
|
139
282
|
|
|
140
|
-
def alias(self, alias: str) ->
|
|
283
|
+
def alias(self, alias: str) -> "ActivityOutputFields":
|
|
141
284
|
self._alias = alias
|
|
142
285
|
return self
|
|
143
286
|
|
|
287
|
+
|
|
144
288
|
class ApiKeyFields(GraphQLField):
|
|
145
289
|
"""@private"""
|
|
146
|
-
key: 'ApiKeyGraphQLField' = ApiKeyGraphQLField('key')
|
|
147
|
-
created_at: 'ApiKeyGraphQLField' = ApiKeyGraphQLField('createdAt')
|
|
148
290
|
|
|
149
|
-
|
|
291
|
+
key: "ApiKeyGraphQLField" = ApiKeyGraphQLField("key")
|
|
292
|
+
created_at: "ApiKeyGraphQLField" = ApiKeyGraphQLField("createdAt")
|
|
293
|
+
|
|
294
|
+
def fields(self, *subfields: ApiKeyGraphQLField) -> "ApiKeyFields":
|
|
150
295
|
"""Subfields should come from the ApiKeyFields class"""
|
|
151
296
|
self._subfields.extend(subfields)
|
|
152
297
|
return self
|
|
153
298
|
|
|
154
|
-
def alias(self, alias: str) ->
|
|
299
|
+
def alias(self, alias: str) -> "ApiKeyFields":
|
|
155
300
|
self._alias = alias
|
|
156
301
|
return self
|
|
157
302
|
|
|
303
|
+
|
|
158
304
|
class AuthProviderFields(GraphQLField):
|
|
159
305
|
"""@private"""
|
|
160
|
-
name: 'AuthProviderGraphQLField' = AuthProviderGraphQLField('name')
|
|
161
|
-
key: 'AuthProviderGraphQLField' = AuthProviderGraphQLField('key')
|
|
162
|
-
kind: 'AuthProviderGraphQLField' = AuthProviderGraphQLField('kind')
|
|
163
|
-
login_url: 'AuthProviderGraphQLField' = AuthProviderGraphQLField('loginUrl')
|
|
164
306
|
|
|
165
|
-
|
|
307
|
+
name: "AuthProviderGraphQLField" = AuthProviderGraphQLField("name")
|
|
308
|
+
key: "AuthProviderGraphQLField" = AuthProviderGraphQLField("key")
|
|
309
|
+
kind: "AuthProviderGraphQLField" = AuthProviderGraphQLField("kind")
|
|
310
|
+
login_url: "AuthProviderGraphQLField" = AuthProviderGraphQLField("loginUrl")
|
|
311
|
+
|
|
312
|
+
def fields(self, *subfields: AuthProviderGraphQLField) -> "AuthProviderFields":
|
|
166
313
|
"""Subfields should come from the AuthProviderFields class"""
|
|
167
314
|
self._subfields.extend(subfields)
|
|
168
315
|
return self
|
|
169
316
|
|
|
170
|
-
def alias(self, alias: str) ->
|
|
317
|
+
def alias(self, alias: str) -> "AuthProviderFields":
|
|
171
318
|
self._alias = alias
|
|
172
319
|
return self
|
|
173
320
|
|
|
321
|
+
|
|
174
322
|
class BatchInferenceJobStageOutputFields(GraphQLField):
|
|
175
323
|
"""@private"""
|
|
176
|
-
total_num_samples: 'BatchInferenceJobStageOutputGraphQLField' = BatchInferenceJobStageOutputGraphQLField('totalNumSamples')
|
|
177
|
-
processed_num_samples: 'BatchInferenceJobStageOutputGraphQLField' = BatchInferenceJobStageOutputGraphQLField('processedNumSamples')
|
|
178
|
-
monitoring_link: 'BatchInferenceJobStageOutputGraphQLField' = BatchInferenceJobStageOutputGraphQLField('monitoringLink')
|
|
179
324
|
|
|
180
|
-
|
|
325
|
+
total_num_samples: "BatchInferenceJobStageOutputGraphQLField" = BatchInferenceJobStageOutputGraphQLField(
|
|
326
|
+
"totalNumSamples"
|
|
327
|
+
)
|
|
328
|
+
processed_num_samples: "BatchInferenceJobStageOutputGraphQLField" = BatchInferenceJobStageOutputGraphQLField(
|
|
329
|
+
"processedNumSamples"
|
|
330
|
+
)
|
|
331
|
+
monitoring_link: "BatchInferenceJobStageOutputGraphQLField" = BatchInferenceJobStageOutputGraphQLField(
|
|
332
|
+
"monitoringLink"
|
|
333
|
+
)
|
|
334
|
+
|
|
335
|
+
def fields(self, *subfields: BatchInferenceJobStageOutputGraphQLField) -> "BatchInferenceJobStageOutputFields":
|
|
181
336
|
"""Subfields should come from the BatchInferenceJobStageOutputFields class"""
|
|
182
337
|
self._subfields.extend(subfields)
|
|
183
338
|
return self
|
|
184
339
|
|
|
185
|
-
def alias(self, alias: str) ->
|
|
340
|
+
def alias(self, alias: str) -> "BatchInferenceJobStageOutputFields":
|
|
186
341
|
self._alias = alias
|
|
187
342
|
return self
|
|
188
343
|
|
|
344
|
+
|
|
189
345
|
class BillingUsageFields(GraphQLField):
|
|
190
346
|
"""@private
|
|
191
|
-
Report token usage on this billing cycle: from the start of the period until now
|
|
192
|
-
Also indicate a projection usage at the end of the period"""
|
|
193
|
-
|
|
194
|
-
|
|
195
|
-
|
|
347
|
+
Report token usage on this billing cycle: from the start of the period until now
|
|
348
|
+
Also indicate a projection usage at the end of the period"""
|
|
349
|
+
|
|
350
|
+
now: "BillingUsageGraphQLField" = BillingUsageGraphQLField("now")
|
|
351
|
+
start: "BillingUsageGraphQLField" = BillingUsageGraphQLField("start")
|
|
352
|
+
end: "BillingUsageGraphQLField" = BillingUsageGraphQLField("end")
|
|
196
353
|
|
|
197
354
|
@classmethod
|
|
198
|
-
def usage(cls) ->
|
|
199
|
-
return UsageStatsFields(
|
|
355
|
+
def usage(cls) -> "UsageStatsFields":
|
|
356
|
+
return UsageStatsFields("usage")
|
|
200
357
|
|
|
201
358
|
@classmethod
|
|
202
|
-
def projection(cls) ->
|
|
359
|
+
def projection(cls) -> "UsageStatsFields":
|
|
203
360
|
"""projection of token usage at the end of the billing cycle based on the current usage"""
|
|
204
|
-
return UsageStatsFields(
|
|
361
|
+
return UsageStatsFields("projection")
|
|
205
362
|
|
|
206
|
-
def fields(self, *subfields: Union[BillingUsageGraphQLField,
|
|
363
|
+
def fields(self, *subfields: Union[BillingUsageGraphQLField, "UsageStatsFields"]) -> "BillingUsageFields":
|
|
207
364
|
"""Subfields should come from the BillingUsageFields class"""
|
|
208
365
|
self._subfields.extend(subfields)
|
|
209
366
|
return self
|
|
210
367
|
|
|
211
|
-
def alias(self, alias: str) ->
|
|
368
|
+
def alias(self, alias: str) -> "BillingUsageFields":
|
|
212
369
|
self._alias = alias
|
|
213
370
|
return self
|
|
214
371
|
|
|
372
|
+
|
|
215
373
|
class ChatMessageFields(GraphQLField):
|
|
216
374
|
"""@private"""
|
|
217
|
-
role: 'ChatMessageGraphQLField' = ChatMessageGraphQLField('role')
|
|
218
|
-
content: 'ChatMessageGraphQLField' = ChatMessageGraphQLField('content')
|
|
219
375
|
|
|
220
|
-
|
|
376
|
+
role: "ChatMessageGraphQLField" = ChatMessageGraphQLField("role")
|
|
377
|
+
content: "ChatMessageGraphQLField" = ChatMessageGraphQLField("content")
|
|
378
|
+
|
|
379
|
+
def fields(self, *subfields: ChatMessageGraphQLField) -> "ChatMessageFields":
|
|
221
380
|
"""Subfields should come from the ChatMessageFields class"""
|
|
222
381
|
self._subfields.extend(subfields)
|
|
223
382
|
return self
|
|
224
383
|
|
|
225
|
-
def alias(self, alias: str) ->
|
|
384
|
+
def alias(self, alias: str) -> "ChatMessageFields":
|
|
226
385
|
self._alias = alias
|
|
227
386
|
return self
|
|
228
387
|
|
|
388
|
+
|
|
229
389
|
class ComparisonFeedbackFields(GraphQLField):
|
|
230
390
|
"""@private"""
|
|
231
|
-
|
|
232
|
-
|
|
391
|
+
|
|
392
|
+
id: "ComparisonFeedbackGraphQLField" = ComparisonFeedbackGraphQLField("id")
|
|
393
|
+
created_at: "ComparisonFeedbackGraphQLField" = ComparisonFeedbackGraphQLField("createdAt")
|
|
233
394
|
|
|
234
395
|
@classmethod
|
|
235
|
-
def usecase(cls) ->
|
|
236
|
-
return UseCaseFields(
|
|
396
|
+
def usecase(cls) -> "UseCaseFields":
|
|
397
|
+
return UseCaseFields("usecase")
|
|
237
398
|
|
|
238
399
|
@classmethod
|
|
239
|
-
def metric(cls) ->
|
|
240
|
-
return MetricFields(
|
|
400
|
+
def metric(cls) -> "MetricFields":
|
|
401
|
+
return MetricFields("metric")
|
|
241
402
|
|
|
242
403
|
@classmethod
|
|
243
|
-
def prefered_completion(cls) ->
|
|
244
|
-
return CompletionFields(
|
|
404
|
+
def prefered_completion(cls) -> "CompletionFields":
|
|
405
|
+
return CompletionFields("preferedCompletion")
|
|
245
406
|
|
|
246
407
|
@classmethod
|
|
247
|
-
def other_completion(cls) ->
|
|
248
|
-
return CompletionFields(
|
|
408
|
+
def other_completion(cls) -> "CompletionFields":
|
|
409
|
+
return CompletionFields("otherCompletion")
|
|
249
410
|
|
|
250
|
-
def fields(
|
|
411
|
+
def fields(
|
|
412
|
+
self, *subfields: Union[ComparisonFeedbackGraphQLField, "CompletionFields", "MetricFields", "UseCaseFields"]
|
|
413
|
+
) -> "ComparisonFeedbackFields":
|
|
251
414
|
"""Subfields should come from the ComparisonFeedbackFields class"""
|
|
252
415
|
self._subfields.extend(subfields)
|
|
253
416
|
return self
|
|
254
417
|
|
|
255
|
-
def alias(self, alias: str) ->
|
|
418
|
+
def alias(self, alias: str) -> "ComparisonFeedbackFields":
|
|
256
419
|
self._alias = alias
|
|
257
420
|
return self
|
|
258
421
|
|
|
422
|
+
|
|
259
423
|
class CompletionFields(GraphQLField):
|
|
260
424
|
"""@private"""
|
|
261
|
-
|
|
262
|
-
|
|
425
|
+
|
|
426
|
+
id: "CompletionGraphQLField" = CompletionGraphQLField("id")
|
|
427
|
+
prompt_hash: "CompletionGraphQLField" = CompletionGraphQLField("promptHash")
|
|
263
428
|
|
|
264
429
|
@classmethod
|
|
265
|
-
def chat_messages(cls) ->
|
|
266
|
-
return ChatMessageFields(
|
|
430
|
+
def chat_messages(cls) -> "ChatMessageFields":
|
|
431
|
+
return ChatMessageFields("chatMessages")
|
|
267
432
|
|
|
268
433
|
@classmethod
|
|
269
|
-
def completion(cls, *, max_length: Optional[int]=None) ->
|
|
270
|
-
arguments: Dict[str, Dict[str, Any]] = {
|
|
271
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
272
|
-
return CompletionGraphQLField(
|
|
273
|
-
|
|
434
|
+
def completion(cls, *, max_length: Optional[int] = None) -> "CompletionGraphQLField":
|
|
435
|
+
arguments: Dict[str, Dict[str, Any]] = {"maxLength": {"type": "Int", "value": max_length}}
|
|
436
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
|
|
437
|
+
return CompletionGraphQLField("completion", arguments=cleared_arguments)
|
|
438
|
+
|
|
439
|
+
source: "CompletionGraphQLField" = CompletionGraphQLField("source")
|
|
274
440
|
|
|
275
441
|
@classmethod
|
|
276
|
-
def model_service(cls) ->
|
|
277
|
-
return ModelServiceFields(
|
|
442
|
+
def model_service(cls) -> "ModelServiceFields":
|
|
443
|
+
return ModelServiceFields("modelService")
|
|
278
444
|
|
|
279
445
|
@classmethod
|
|
280
|
-
def model(cls) ->
|
|
281
|
-
return ModelFields(
|
|
446
|
+
def model(cls) -> "ModelFields":
|
|
447
|
+
return ModelFields("model")
|
|
282
448
|
|
|
283
449
|
@classmethod
|
|
284
|
-
def direct_feedbacks(cls, *, filter: Optional[FeedbackFilterInput]=None) ->
|
|
285
|
-
arguments: Dict[str, Dict[str, Any]] = {
|
|
286
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
287
|
-
return DirectFeedbackFields(
|
|
450
|
+
def direct_feedbacks(cls, *, filter: Optional[FeedbackFilterInput] = None) -> "DirectFeedbackFields":
|
|
451
|
+
arguments: Dict[str, Dict[str, Any]] = {"filter": {"type": "FeedbackFilterInput", "value": filter}}
|
|
452
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
|
|
453
|
+
return DirectFeedbackFields("directFeedbacks", arguments=cleared_arguments)
|
|
288
454
|
|
|
289
455
|
@classmethod
|
|
290
|
-
def comparison_feedbacks(cls) ->
|
|
291
|
-
return ComparisonFeedbackFields(
|
|
456
|
+
def comparison_feedbacks(cls) -> "ComparisonFeedbackFields":
|
|
457
|
+
return ComparisonFeedbackFields("comparisonFeedbacks")
|
|
292
458
|
|
|
293
459
|
@classmethod
|
|
294
|
-
def session(cls) ->
|
|
295
|
-
return SessionFields(
|
|
460
|
+
def session(cls) -> "SessionFields":
|
|
461
|
+
return SessionFields("session")
|
|
296
462
|
|
|
297
463
|
@classmethod
|
|
298
|
-
def history(cls) ->
|
|
299
|
-
return CompletionHistoryEntryOuputFields(
|
|
464
|
+
def history(cls) -> "CompletionHistoryEntryOuputFields":
|
|
465
|
+
return CompletionHistoryEntryOuputFields("history")
|
|
300
466
|
|
|
301
467
|
@classmethod
|
|
302
|
-
def labels(cls, with_protected: bool) ->
|
|
303
|
-
arguments: Dict[str, Dict[str, Any]] = {
|
|
304
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
305
|
-
return CompletionLabelFields(
|
|
306
|
-
|
|
468
|
+
def labels(cls, with_protected: bool) -> "CompletionLabelFields":
|
|
469
|
+
arguments: Dict[str, Dict[str, Any]] = {"withProtected": {"type": "Boolean!", "value": with_protected}}
|
|
470
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
|
|
471
|
+
return CompletionLabelFields("labels", arguments=cleared_arguments)
|
|
472
|
+
|
|
473
|
+
created_at: "CompletionGraphQLField" = CompletionGraphQLField("createdAt")
|
|
307
474
|
|
|
308
475
|
@classmethod
|
|
309
|
-
def siblings_count(cls, filter: ListCompletionsFilterInput) ->
|
|
310
|
-
arguments: Dict[str, Dict[str, Any]] = {
|
|
311
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
312
|
-
return CompletionGraphQLField(
|
|
313
|
-
|
|
314
|
-
|
|
476
|
+
def siblings_count(cls, filter: ListCompletionsFilterInput) -> "CompletionGraphQLField":
|
|
477
|
+
arguments: Dict[str, Dict[str, Any]] = {"filter": {"type": "ListCompletionsFilterInput!", "value": filter}}
|
|
478
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
|
|
479
|
+
return CompletionGraphQLField("siblingsCount", arguments=cleared_arguments)
|
|
480
|
+
|
|
481
|
+
has_user_metadata: "CompletionGraphQLField" = CompletionGraphQLField("hasUserMetadata")
|
|
482
|
+
user_metadata: "CompletionGraphQLField" = CompletionGraphQLField("userMetadata")
|
|
315
483
|
|
|
316
484
|
@classmethod
|
|
317
|
-
def metadata(cls) ->
|
|
318
|
-
return CompletionMetadataFields(
|
|
319
|
-
can_edit: 'CompletionGraphQLField' = CompletionGraphQLField('canEdit')
|
|
485
|
+
def metadata(cls) -> "CompletionMetadataFields":
|
|
486
|
+
return CompletionMetadataFields("metadata")
|
|
320
487
|
|
|
321
|
-
|
|
488
|
+
can_edit: "CompletionGraphQLField" = CompletionGraphQLField("canEdit")
|
|
489
|
+
|
|
490
|
+
def fields(
|
|
491
|
+
self,
|
|
492
|
+
*subfields: Union[
|
|
493
|
+
CompletionGraphQLField,
|
|
494
|
+
"ChatMessageFields",
|
|
495
|
+
"ComparisonFeedbackFields",
|
|
496
|
+
"CompletionHistoryEntryOuputFields",
|
|
497
|
+
"CompletionLabelFields",
|
|
498
|
+
"CompletionMetadataFields",
|
|
499
|
+
"DirectFeedbackFields",
|
|
500
|
+
"ModelFields",
|
|
501
|
+
"ModelServiceFields",
|
|
502
|
+
"SessionFields",
|
|
503
|
+
],
|
|
504
|
+
) -> "CompletionFields":
|
|
322
505
|
"""Subfields should come from the CompletionFields class"""
|
|
323
506
|
self._subfields.extend(subfields)
|
|
324
507
|
return self
|
|
325
508
|
|
|
326
|
-
def alias(self, alias: str) ->
|
|
509
|
+
def alias(self, alias: str) -> "CompletionFields":
|
|
327
510
|
self._alias = alias
|
|
328
511
|
return self
|
|
329
512
|
|
|
513
|
+
|
|
330
514
|
class CompletionConnectionFields(GraphQLField):
|
|
331
515
|
"""@private"""
|
|
332
516
|
|
|
333
517
|
@classmethod
|
|
334
|
-
def page_info(cls) ->
|
|
518
|
+
def page_info(cls) -> "PageInfoFields":
|
|
335
519
|
"""Information to aid in pagination."""
|
|
336
|
-
return PageInfoFields(
|
|
520
|
+
return PageInfoFields("pageInfo")
|
|
337
521
|
|
|
338
522
|
@classmethod
|
|
339
|
-
def edges(cls) ->
|
|
523
|
+
def edges(cls) -> "CompletionEdgeFields":
|
|
340
524
|
"""A list of edges."""
|
|
341
|
-
return CompletionEdgeFields(
|
|
525
|
+
return CompletionEdgeFields("edges")
|
|
342
526
|
|
|
343
527
|
@classmethod
|
|
344
|
-
def nodes(cls) ->
|
|
528
|
+
def nodes(cls) -> "CompletionFields":
|
|
345
529
|
"""A list of nodes."""
|
|
346
|
-
return CompletionFields(
|
|
347
|
-
total_count: 'CompletionConnectionGraphQLField' = CompletionConnectionGraphQLField('totalCount')
|
|
530
|
+
return CompletionFields("nodes")
|
|
348
531
|
|
|
349
|
-
|
|
532
|
+
total_count: "CompletionConnectionGraphQLField" = CompletionConnectionGraphQLField("totalCount")
|
|
533
|
+
|
|
534
|
+
def fields(
|
|
535
|
+
self,
|
|
536
|
+
*subfields: Union[
|
|
537
|
+
CompletionConnectionGraphQLField, "CompletionEdgeFields", "CompletionFields", "PageInfoFields"
|
|
538
|
+
],
|
|
539
|
+
) -> "CompletionConnectionFields":
|
|
350
540
|
"""Subfields should come from the CompletionConnectionFields class"""
|
|
351
541
|
self._subfields.extend(subfields)
|
|
352
542
|
return self
|
|
353
543
|
|
|
354
|
-
def alias(self, alias: str) ->
|
|
544
|
+
def alias(self, alias: str) -> "CompletionConnectionFields":
|
|
355
545
|
self._alias = alias
|
|
356
546
|
return self
|
|
357
547
|
|
|
548
|
+
|
|
358
549
|
class CompletionEdgeFields(GraphQLField):
|
|
359
550
|
"""@private
|
|
360
|
-
An edge in a connection."""
|
|
551
|
+
An edge in a connection."""
|
|
361
552
|
|
|
362
553
|
@classmethod
|
|
363
|
-
def node(cls) ->
|
|
554
|
+
def node(cls) -> "CompletionFields":
|
|
364
555
|
"""The item at the end of the edge"""
|
|
365
|
-
return CompletionFields(
|
|
366
|
-
|
|
367
|
-
|
|
556
|
+
return CompletionFields("node")
|
|
557
|
+
|
|
558
|
+
cursor: "CompletionEdgeGraphQLField" = CompletionEdgeGraphQLField("cursor")
|
|
559
|
+
"A cursor for use in pagination"
|
|
368
560
|
|
|
369
|
-
def fields(self, *subfields: Union[CompletionEdgeGraphQLField,
|
|
561
|
+
def fields(self, *subfields: Union[CompletionEdgeGraphQLField, "CompletionFields"]) -> "CompletionEdgeFields":
|
|
370
562
|
"""Subfields should come from the CompletionEdgeFields class"""
|
|
371
563
|
self._subfields.extend(subfields)
|
|
372
564
|
return self
|
|
373
565
|
|
|
374
|
-
def alias(self, alias: str) ->
|
|
566
|
+
def alias(self, alias: str) -> "CompletionEdgeFields":
|
|
375
567
|
self._alias = alias
|
|
376
568
|
return self
|
|
377
569
|
|
|
570
|
+
|
|
378
571
|
class CompletionGroupDataFields(GraphQLField):
|
|
379
572
|
"""@private"""
|
|
380
|
-
|
|
381
|
-
|
|
573
|
+
|
|
574
|
+
key: "CompletionGroupDataGraphQLField" = CompletionGroupDataGraphQLField("key")
|
|
575
|
+
count: "CompletionGroupDataGraphQLField" = CompletionGroupDataGraphQLField("count")
|
|
382
576
|
|
|
383
577
|
@classmethod
|
|
384
|
-
def direct_feedbacks_stats(cls) ->
|
|
385
|
-
return CompletionGroupFeedbackStatsFields(
|
|
578
|
+
def direct_feedbacks_stats(cls) -> "CompletionGroupFeedbackStatsFields":
|
|
579
|
+
return CompletionGroupFeedbackStatsFields("directFeedbacksStats")
|
|
386
580
|
|
|
387
581
|
@classmethod
|
|
388
|
-
def completions(cls, page: CursorPageInput, order: List[OrderPair]) ->
|
|
389
|
-
arguments: Dict[str, Dict[str, Any]] = {
|
|
390
|
-
|
|
391
|
-
|
|
582
|
+
def completions(cls, page: CursorPageInput, order: List[OrderPair]) -> "CompletionConnectionFields":
|
|
583
|
+
arguments: Dict[str, Dict[str, Any]] = {
|
|
584
|
+
"page": {"type": "CursorPageInput!", "value": page},
|
|
585
|
+
"order": {"type": "[OrderPair!]!", "value": order},
|
|
586
|
+
}
|
|
587
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
|
|
588
|
+
return CompletionConnectionFields("completions", arguments=cleared_arguments)
|
|
392
589
|
|
|
393
|
-
def fields(
|
|
590
|
+
def fields(
|
|
591
|
+
self,
|
|
592
|
+
*subfields: Union[
|
|
593
|
+
CompletionGroupDataGraphQLField, "CompletionConnectionFields", "CompletionGroupFeedbackStatsFields"
|
|
594
|
+
],
|
|
595
|
+
) -> "CompletionGroupDataFields":
|
|
394
596
|
"""Subfields should come from the CompletionGroupDataFields class"""
|
|
395
597
|
self._subfields.extend(subfields)
|
|
396
598
|
return self
|
|
397
599
|
|
|
398
|
-
def alias(self, alias: str) ->
|
|
600
|
+
def alias(self, alias: str) -> "CompletionGroupDataFields":
|
|
399
601
|
self._alias = alias
|
|
400
602
|
return self
|
|
401
603
|
|
|
604
|
+
|
|
402
605
|
class CompletionGroupDataConnectionFields(GraphQLField):
|
|
403
606
|
"""@private"""
|
|
404
607
|
|
|
405
608
|
@classmethod
|
|
406
|
-
def page_info(cls) ->
|
|
609
|
+
def page_info(cls) -> "PageInfoFields":
|
|
407
610
|
"""Information to aid in pagination."""
|
|
408
|
-
return PageInfoFields(
|
|
611
|
+
return PageInfoFields("pageInfo")
|
|
409
612
|
|
|
410
613
|
@classmethod
|
|
411
|
-
def edges(cls) ->
|
|
614
|
+
def edges(cls) -> "CompletionGroupDataEdgeFields":
|
|
412
615
|
"""A list of edges."""
|
|
413
|
-
return CompletionGroupDataEdgeFields(
|
|
616
|
+
return CompletionGroupDataEdgeFields("edges")
|
|
414
617
|
|
|
415
618
|
@classmethod
|
|
416
|
-
def nodes(cls) ->
|
|
619
|
+
def nodes(cls) -> "CompletionGroupDataFields":
|
|
417
620
|
"""A list of nodes."""
|
|
418
|
-
return CompletionGroupDataFields(
|
|
419
|
-
|
|
420
|
-
|
|
421
|
-
|
|
422
|
-
|
|
621
|
+
return CompletionGroupDataFields("nodes")
|
|
622
|
+
|
|
623
|
+
group_by: "CompletionGroupDataConnectionGraphQLField" = CompletionGroupDataConnectionGraphQLField("groupBy")
|
|
624
|
+
total_count: "CompletionGroupDataConnectionGraphQLField" = CompletionGroupDataConnectionGraphQLField("totalCount")
|
|
625
|
+
|
|
626
|
+
def fields(
|
|
627
|
+
self,
|
|
628
|
+
*subfields: Union[
|
|
629
|
+
CompletionGroupDataConnectionGraphQLField,
|
|
630
|
+
"CompletionGroupDataEdgeFields",
|
|
631
|
+
"CompletionGroupDataFields",
|
|
632
|
+
"PageInfoFields",
|
|
633
|
+
],
|
|
634
|
+
) -> "CompletionGroupDataConnectionFields":
|
|
423
635
|
"""Subfields should come from the CompletionGroupDataConnectionFields class"""
|
|
424
636
|
self._subfields.extend(subfields)
|
|
425
637
|
return self
|
|
426
638
|
|
|
427
|
-
def alias(self, alias: str) ->
|
|
639
|
+
def alias(self, alias: str) -> "CompletionGroupDataConnectionFields":
|
|
428
640
|
self._alias = alias
|
|
429
641
|
return self
|
|
430
642
|
|
|
643
|
+
|
|
431
644
|
class CompletionGroupDataEdgeFields(GraphQLField):
|
|
432
645
|
"""@private
|
|
433
|
-
An edge in a connection."""
|
|
646
|
+
An edge in a connection."""
|
|
434
647
|
|
|
435
648
|
@classmethod
|
|
436
|
-
def node(cls) ->
|
|
649
|
+
def node(cls) -> "CompletionGroupDataFields":
|
|
437
650
|
"""The item at the end of the edge"""
|
|
438
|
-
return CompletionGroupDataFields(
|
|
439
|
-
cursor: 'CompletionGroupDataEdgeGraphQLField' = CompletionGroupDataEdgeGraphQLField('cursor')
|
|
440
|
-
'A cursor for use in pagination'
|
|
651
|
+
return CompletionGroupDataFields("node")
|
|
441
652
|
|
|
442
|
-
|
|
653
|
+
cursor: "CompletionGroupDataEdgeGraphQLField" = CompletionGroupDataEdgeGraphQLField("cursor")
|
|
654
|
+
"A cursor for use in pagination"
|
|
655
|
+
|
|
656
|
+
def fields(
|
|
657
|
+
self, *subfields: Union[CompletionGroupDataEdgeGraphQLField, "CompletionGroupDataFields"]
|
|
658
|
+
) -> "CompletionGroupDataEdgeFields":
|
|
443
659
|
"""Subfields should come from the CompletionGroupDataEdgeFields class"""
|
|
444
660
|
self._subfields.extend(subfields)
|
|
445
661
|
return self
|
|
446
662
|
|
|
447
|
-
def alias(self, alias: str) ->
|
|
663
|
+
def alias(self, alias: str) -> "CompletionGroupDataEdgeFields":
|
|
448
664
|
self._alias = alias
|
|
449
665
|
return self
|
|
450
666
|
|
|
667
|
+
|
|
451
668
|
class CompletionGroupFeedbackStatsFields(GraphQLField):
|
|
452
669
|
"""@private"""
|
|
453
670
|
|
|
454
671
|
@classmethod
|
|
455
|
-
def metric(cls) ->
|
|
456
|
-
return MetricFields(
|
|
457
|
-
feedbacks: 'CompletionGroupFeedbackStatsGraphQLField' = CompletionGroupFeedbackStatsGraphQLField('feedbacks')
|
|
458
|
-
average: 'CompletionGroupFeedbackStatsGraphQLField' = CompletionGroupFeedbackStatsGraphQLField('average')
|
|
459
|
-
max: 'CompletionGroupFeedbackStatsGraphQLField' = CompletionGroupFeedbackStatsGraphQLField('max')
|
|
460
|
-
min: 'CompletionGroupFeedbackStatsGraphQLField' = CompletionGroupFeedbackStatsGraphQLField('min')
|
|
461
|
-
stddev: 'CompletionGroupFeedbackStatsGraphQLField' = CompletionGroupFeedbackStatsGraphQLField('stddev')
|
|
462
|
-
sum: 'CompletionGroupFeedbackStatsGraphQLField' = CompletionGroupFeedbackStatsGraphQLField('sum')
|
|
672
|
+
def metric(cls) -> "MetricFields":
|
|
673
|
+
return MetricFields("metric")
|
|
463
674
|
|
|
464
|
-
|
|
675
|
+
feedbacks: "CompletionGroupFeedbackStatsGraphQLField" = CompletionGroupFeedbackStatsGraphQLField("feedbacks")
|
|
676
|
+
average: "CompletionGroupFeedbackStatsGraphQLField" = CompletionGroupFeedbackStatsGraphQLField("average")
|
|
677
|
+
max: "CompletionGroupFeedbackStatsGraphQLField" = CompletionGroupFeedbackStatsGraphQLField("max")
|
|
678
|
+
min: "CompletionGroupFeedbackStatsGraphQLField" = CompletionGroupFeedbackStatsGraphQLField("min")
|
|
679
|
+
stddev: "CompletionGroupFeedbackStatsGraphQLField" = CompletionGroupFeedbackStatsGraphQLField("stddev")
|
|
680
|
+
sum: "CompletionGroupFeedbackStatsGraphQLField" = CompletionGroupFeedbackStatsGraphQLField("sum")
|
|
681
|
+
|
|
682
|
+
def fields(
|
|
683
|
+
self, *subfields: Union[CompletionGroupFeedbackStatsGraphQLField, "MetricFields"]
|
|
684
|
+
) -> "CompletionGroupFeedbackStatsFields":
|
|
465
685
|
"""Subfields should come from the CompletionGroupFeedbackStatsFields class"""
|
|
466
686
|
self._subfields.extend(subfields)
|
|
467
687
|
return self
|
|
468
688
|
|
|
469
|
-
def alias(self, alias: str) ->
|
|
689
|
+
def alias(self, alias: str) -> "CompletionGroupFeedbackStatsFields":
|
|
470
690
|
self._alias = alias
|
|
471
691
|
return self
|
|
472
692
|
|
|
693
|
+
|
|
473
694
|
class CompletionHistoryEntryOuputFields(GraphQLField):
|
|
474
695
|
"""@private"""
|
|
475
|
-
level: 'CompletionHistoryEntryOuputGraphQLField' = CompletionHistoryEntryOuputGraphQLField('level')
|
|
476
|
-
completion_id: 'CompletionHistoryEntryOuputGraphQLField' = CompletionHistoryEntryOuputGraphQLField('completionId')
|
|
477
|
-
prompt: 'CompletionHistoryEntryOuputGraphQLField' = CompletionHistoryEntryOuputGraphQLField('prompt')
|
|
478
|
-
completion: 'CompletionHistoryEntryOuputGraphQLField' = CompletionHistoryEntryOuputGraphQLField('completion')
|
|
479
|
-
created_at: 'CompletionHistoryEntryOuputGraphQLField' = CompletionHistoryEntryOuputGraphQLField('createdAt')
|
|
480
696
|
|
|
481
|
-
|
|
697
|
+
level: "CompletionHistoryEntryOuputGraphQLField" = CompletionHistoryEntryOuputGraphQLField("level")
|
|
698
|
+
completion_id: "CompletionHistoryEntryOuputGraphQLField" = CompletionHistoryEntryOuputGraphQLField("completionId")
|
|
699
|
+
prompt: "CompletionHistoryEntryOuputGraphQLField" = CompletionHistoryEntryOuputGraphQLField("prompt")
|
|
700
|
+
completion: "CompletionHistoryEntryOuputGraphQLField" = CompletionHistoryEntryOuputGraphQLField("completion")
|
|
701
|
+
created_at: "CompletionHistoryEntryOuputGraphQLField" = CompletionHistoryEntryOuputGraphQLField("createdAt")
|
|
702
|
+
|
|
703
|
+
def fields(self, *subfields: CompletionHistoryEntryOuputGraphQLField) -> "CompletionHistoryEntryOuputFields":
|
|
482
704
|
"""Subfields should come from the CompletionHistoryEntryOuputFields class"""
|
|
483
705
|
self._subfields.extend(subfields)
|
|
484
706
|
return self
|
|
485
707
|
|
|
486
|
-
def alias(self, alias: str) ->
|
|
708
|
+
def alias(self, alias: str) -> "CompletionHistoryEntryOuputFields":
|
|
487
709
|
self._alias = alias
|
|
488
710
|
return self
|
|
489
711
|
|
|
712
|
+
|
|
490
713
|
class CompletionLabelFields(GraphQLField):
|
|
491
714
|
"""@private"""
|
|
492
|
-
key: 'CompletionLabelGraphQLField' = CompletionLabelGraphQLField('key')
|
|
493
|
-
value: 'CompletionLabelGraphQLField' = CompletionLabelGraphQLField('value')
|
|
494
715
|
|
|
495
|
-
|
|
716
|
+
key: "CompletionLabelGraphQLField" = CompletionLabelGraphQLField("key")
|
|
717
|
+
value: "CompletionLabelGraphQLField" = CompletionLabelGraphQLField("value")
|
|
718
|
+
|
|
719
|
+
def fields(self, *subfields: CompletionLabelGraphQLField) -> "CompletionLabelFields":
|
|
496
720
|
"""Subfields should come from the CompletionLabelFields class"""
|
|
497
721
|
self._subfields.extend(subfields)
|
|
498
722
|
return self
|
|
499
723
|
|
|
500
|
-
def alias(self, alias: str) ->
|
|
724
|
+
def alias(self, alias: str) -> "CompletionLabelFields":
|
|
501
725
|
self._alias = alias
|
|
502
726
|
return self
|
|
503
727
|
|
|
728
|
+
|
|
504
729
|
class CompletionMetadataFields(GraphQLField):
|
|
505
730
|
"""@private"""
|
|
506
|
-
|
|
507
|
-
|
|
508
|
-
|
|
731
|
+
|
|
732
|
+
parameters: "CompletionMetadataGraphQLField" = CompletionMetadataGraphQLField("parameters")
|
|
733
|
+
timings: "CompletionMetadataGraphQLField" = CompletionMetadataGraphQLField("timings")
|
|
734
|
+
system: "CompletionMetadataGraphQLField" = CompletionMetadataGraphQLField("system")
|
|
509
735
|
|
|
510
736
|
@classmethod
|
|
511
|
-
def usage(cls) ->
|
|
512
|
-
return UsageFields(
|
|
737
|
+
def usage(cls) -> "UsageFields":
|
|
738
|
+
return UsageFields("usage")
|
|
513
739
|
|
|
514
|
-
def fields(self, *subfields: Union[CompletionMetadataGraphQLField,
|
|
740
|
+
def fields(self, *subfields: Union[CompletionMetadataGraphQLField, "UsageFields"]) -> "CompletionMetadataFields":
|
|
515
741
|
"""Subfields should come from the CompletionMetadataFields class"""
|
|
516
742
|
self._subfields.extend(subfields)
|
|
517
743
|
return self
|
|
518
744
|
|
|
519
|
-
def alias(self, alias: str) ->
|
|
745
|
+
def alias(self, alias: str) -> "CompletionMetadataFields":
|
|
520
746
|
self._alias = alias
|
|
521
747
|
return self
|
|
522
748
|
|
|
749
|
+
|
|
523
750
|
class ComputePoolFields(GraphQLField):
|
|
524
751
|
"""@private"""
|
|
525
|
-
|
|
526
|
-
|
|
527
|
-
|
|
528
|
-
|
|
752
|
+
|
|
753
|
+
id: "ComputePoolGraphQLField" = ComputePoolGraphQLField("id")
|
|
754
|
+
key: "ComputePoolGraphQLField" = ComputePoolGraphQLField("key")
|
|
755
|
+
name: "ComputePoolGraphQLField" = ComputePoolGraphQLField("name")
|
|
756
|
+
created_at: "ComputePoolGraphQLField" = ComputePoolGraphQLField("createdAt")
|
|
529
757
|
|
|
530
758
|
@classmethod
|
|
531
|
-
def all_harmony_groups(cls) ->
|
|
532
|
-
return HarmonyGroupFields(
|
|
759
|
+
def all_harmony_groups(cls) -> "HarmonyGroupFields":
|
|
760
|
+
return HarmonyGroupFields("allHarmonyGroups")
|
|
533
761
|
|
|
534
762
|
@classmethod
|
|
535
|
-
def harmony_groups(cls) ->
|
|
536
|
-
return HarmonyGroupFields(
|
|
537
|
-
capabilities: 'ComputePoolGraphQLField' = ComputePoolGraphQLField('capabilities')
|
|
763
|
+
def harmony_groups(cls) -> "HarmonyGroupFields":
|
|
764
|
+
return HarmonyGroupFields("harmonyGroups")
|
|
538
765
|
|
|
539
|
-
|
|
766
|
+
capabilities: "ComputePoolGraphQLField" = ComputePoolGraphQLField("capabilities")
|
|
767
|
+
|
|
768
|
+
def fields(self, *subfields: Union[ComputePoolGraphQLField, "HarmonyGroupFields"]) -> "ComputePoolFields":
|
|
540
769
|
"""Subfields should come from the ComputePoolFields class"""
|
|
541
770
|
self._subfields.extend(subfields)
|
|
542
771
|
return self
|
|
543
772
|
|
|
544
|
-
def alias(self, alias: str) ->
|
|
773
|
+
def alias(self, alias: str) -> "ComputePoolFields":
|
|
545
774
|
self._alias = alias
|
|
546
775
|
return self
|
|
547
776
|
|
|
777
|
+
|
|
548
778
|
class ContractFields(GraphQLField):
|
|
549
779
|
"""@private"""
|
|
550
|
-
|
|
551
|
-
|
|
552
|
-
|
|
553
|
-
|
|
780
|
+
|
|
781
|
+
quota: "ContractGraphQLField" = ContractGraphQLField("quota")
|
|
782
|
+
start_date: "ContractGraphQLField" = ContractGraphQLField("startDate")
|
|
783
|
+
end_date: "ContractGraphQLField" = ContractGraphQLField("endDate")
|
|
784
|
+
cycle: "ContractGraphQLField" = ContractGraphQLField("cycle")
|
|
554
785
|
|
|
555
786
|
@classmethod
|
|
556
|
-
def usage(cls, *, now: Optional[int | str]=None) ->
|
|
787
|
+
def usage(cls, *, now: Optional[int | str] = None) -> "BillingUsageFields":
|
|
557
788
|
"""Get token usage on Adaptive models for the current billing cycle
|
|
558
789
|
returns an error if 'now' is before the start date of the contract"""
|
|
559
|
-
arguments: Dict[str, Dict[str, Any]] = {
|
|
560
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
561
|
-
return BillingUsageFields(
|
|
790
|
+
arguments: Dict[str, Dict[str, Any]] = {"now": {"type": "InputDatetime", "value": now}}
|
|
791
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
|
|
792
|
+
return BillingUsageFields("usage", arguments=cleared_arguments)
|
|
562
793
|
|
|
563
|
-
def fields(self, *subfields: Union[ContractGraphQLField,
|
|
794
|
+
def fields(self, *subfields: Union[ContractGraphQLField, "BillingUsageFields"]) -> "ContractFields":
|
|
564
795
|
"""Subfields should come from the ContractFields class"""
|
|
565
796
|
self._subfields.extend(subfields)
|
|
566
797
|
return self
|
|
567
798
|
|
|
568
|
-
def alias(self, alias: str) ->
|
|
799
|
+
def alias(self, alias: str) -> "ContractFields":
|
|
569
800
|
self._alias = alias
|
|
570
801
|
return self
|
|
571
802
|
|
|
803
|
+
|
|
572
804
|
class CustomConfigOutputFields(GraphQLField):
|
|
573
805
|
"""@private"""
|
|
574
|
-
description: 'CustomConfigOutputGraphQLField' = CustomConfigOutputGraphQLField('description')
|
|
575
806
|
|
|
576
|
-
|
|
807
|
+
description: "CustomConfigOutputGraphQLField" = CustomConfigOutputGraphQLField("description")
|
|
808
|
+
|
|
809
|
+
def fields(self, *subfields: CustomConfigOutputGraphQLField) -> "CustomConfigOutputFields":
|
|
577
810
|
"""Subfields should come from the CustomConfigOutputFields class"""
|
|
578
811
|
self._subfields.extend(subfields)
|
|
579
812
|
return self
|
|
580
813
|
|
|
581
|
-
def alias(self, alias: str) ->
|
|
814
|
+
def alias(self, alias: str) -> "CustomConfigOutputFields":
|
|
582
815
|
self._alias = alias
|
|
583
816
|
return self
|
|
584
817
|
|
|
818
|
+
|
|
585
819
|
class CustomRecipeFields(GraphQLField):
|
|
586
820
|
"""@private"""
|
|
587
|
-
|
|
588
|
-
|
|
589
|
-
|
|
590
|
-
|
|
591
|
-
|
|
592
|
-
|
|
593
|
-
|
|
594
|
-
|
|
595
|
-
|
|
596
|
-
|
|
597
|
-
|
|
598
|
-
|
|
599
|
-
|
|
600
|
-
|
|
601
|
-
|
|
602
|
-
|
|
603
|
-
|
|
604
|
-
|
|
605
|
-
|
|
606
|
-
|
|
607
|
-
|
|
608
|
-
|
|
609
|
-
|
|
610
|
-
|
|
611
|
-
|
|
612
|
-
|
|
613
|
-
|
|
821
|
+
|
|
822
|
+
id: "CustomRecipeGraphQLField" = CustomRecipeGraphQLField("id")
|
|
823
|
+
key: "CustomRecipeGraphQLField" = CustomRecipeGraphQLField("key")
|
|
824
|
+
name: "CustomRecipeGraphQLField" = CustomRecipeGraphQLField("name")
|
|
825
|
+
is_multifile: "CustomRecipeGraphQLField" = CustomRecipeGraphQLField("isMultifile")
|
|
826
|
+
editable: "CustomRecipeGraphQLField" = CustomRecipeGraphQLField("editable")
|
|
827
|
+
hidden: "CustomRecipeGraphQLField" = CustomRecipeGraphQLField("hidden")
|
|
828
|
+
builtin: "CustomRecipeGraphQLField" = CustomRecipeGraphQLField("builtin")
|
|
829
|
+
global_: "CustomRecipeGraphQLField" = CustomRecipeGraphQLField("global")
|
|
830
|
+
created_at: "CustomRecipeGraphQLField" = CustomRecipeGraphQLField("createdAt")
|
|
831
|
+
content: "CustomRecipeGraphQLField" = CustomRecipeGraphQLField("content")
|
|
832
|
+
input_schema: "CustomRecipeGraphQLField" = CustomRecipeGraphQLField("inputSchema")
|
|
833
|
+
json_schema: "CustomRecipeGraphQLField" = CustomRecipeGraphQLField("jsonSchema")
|
|
834
|
+
content_hash: "CustomRecipeGraphQLField" = CustomRecipeGraphQLField("contentHash")
|
|
835
|
+
description: "CustomRecipeGraphQLField" = CustomRecipeGraphQLField("description")
|
|
836
|
+
|
|
837
|
+
@classmethod
|
|
838
|
+
def labels(cls, with_protected: bool) -> "LabelFields":
|
|
839
|
+
arguments: Dict[str, Dict[str, Any]] = {"withProtected": {"type": "Boolean!", "value": with_protected}}
|
|
840
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
|
|
841
|
+
return LabelFields("labels", arguments=cleared_arguments)
|
|
842
|
+
|
|
843
|
+
updated_at: "CustomRecipeGraphQLField" = CustomRecipeGraphQLField("updatedAt")
|
|
844
|
+
|
|
845
|
+
@classmethod
|
|
846
|
+
def created_by(cls) -> "UserFields":
|
|
847
|
+
return UserFields("createdBy")
|
|
848
|
+
|
|
849
|
+
def fields(self, *subfields: Union[CustomRecipeGraphQLField, "LabelFields", "UserFields"]) -> "CustomRecipeFields":
|
|
614
850
|
"""Subfields should come from the CustomRecipeFields class"""
|
|
615
851
|
self._subfields.extend(subfields)
|
|
616
852
|
return self
|
|
617
853
|
|
|
618
|
-
def alias(self, alias: str) ->
|
|
854
|
+
def alias(self, alias: str) -> "CustomRecipeFields":
|
|
619
855
|
self._alias = alias
|
|
620
856
|
return self
|
|
621
857
|
|
|
858
|
+
|
|
622
859
|
class CustomRecipeJobDetailsFields(GraphQLField):
|
|
623
860
|
"""@private"""
|
|
624
|
-
|
|
625
|
-
|
|
861
|
+
|
|
862
|
+
args: "CustomRecipeJobDetailsGraphQLField" = CustomRecipeJobDetailsGraphQLField("args")
|
|
863
|
+
recipe_hash: "CustomRecipeJobDetailsGraphQLField" = CustomRecipeJobDetailsGraphQLField("recipeHash")
|
|
626
864
|
|
|
627
865
|
@classmethod
|
|
628
|
-
def artifacts(cls) ->
|
|
629
|
-
return JobArtifactFields(
|
|
630
|
-
num_gpus: 'CustomRecipeJobDetailsGraphQLField' = CustomRecipeJobDetailsGraphQLField('numGpus')
|
|
631
|
-
gpu_duration_ms: 'CustomRecipeJobDetailsGraphQLField' = CustomRecipeJobDetailsGraphQLField('gpuDurationMs')
|
|
632
|
-
compute_pool_id: 'CustomRecipeJobDetailsGraphQLField' = CustomRecipeJobDetailsGraphQLField('computePoolId')
|
|
866
|
+
def artifacts(cls) -> "JobArtifactFields":
|
|
867
|
+
return JobArtifactFields("artifacts")
|
|
633
868
|
|
|
634
|
-
|
|
869
|
+
num_gpus: "CustomRecipeJobDetailsGraphQLField" = CustomRecipeJobDetailsGraphQLField("numGpus")
|
|
870
|
+
gpu_duration_ms: "CustomRecipeJobDetailsGraphQLField" = CustomRecipeJobDetailsGraphQLField("gpuDurationMs")
|
|
871
|
+
compute_pool_id: "CustomRecipeJobDetailsGraphQLField" = CustomRecipeJobDetailsGraphQLField("computePoolId")
|
|
872
|
+
|
|
873
|
+
def fields(
|
|
874
|
+
self, *subfields: Union[CustomRecipeJobDetailsGraphQLField, "JobArtifactFields"]
|
|
875
|
+
) -> "CustomRecipeJobDetailsFields":
|
|
635
876
|
"""Subfields should come from the CustomRecipeJobDetailsFields class"""
|
|
636
877
|
self._subfields.extend(subfields)
|
|
637
878
|
return self
|
|
638
879
|
|
|
639
|
-
def alias(self, alias: str) ->
|
|
880
|
+
def alias(self, alias: str) -> "CustomRecipeJobDetailsFields":
|
|
640
881
|
self._alias = alias
|
|
641
882
|
return self
|
|
642
883
|
|
|
884
|
+
|
|
643
885
|
class DatasetFields(GraphQLField):
|
|
644
886
|
"""@private"""
|
|
645
|
-
|
|
646
|
-
|
|
647
|
-
|
|
648
|
-
|
|
649
|
-
|
|
650
|
-
|
|
887
|
+
|
|
888
|
+
id: "DatasetGraphQLField" = DatasetGraphQLField("id")
|
|
889
|
+
key: "DatasetGraphQLField" = DatasetGraphQLField("key")
|
|
890
|
+
name: "DatasetGraphQLField" = DatasetGraphQLField("name")
|
|
891
|
+
created_at: "DatasetGraphQLField" = DatasetGraphQLField("createdAt")
|
|
892
|
+
kind: "DatasetGraphQLField" = DatasetGraphQLField("kind")
|
|
893
|
+
records: "DatasetGraphQLField" = DatasetGraphQLField("records")
|
|
651
894
|
|
|
652
895
|
@classmethod
|
|
653
|
-
def metrics_usage(cls) ->
|
|
654
|
-
return DatasetMetricUsageFields(
|
|
655
|
-
|
|
656
|
-
|
|
657
|
-
|
|
896
|
+
def metrics_usage(cls) -> "DatasetMetricUsageFields":
|
|
897
|
+
return DatasetMetricUsageFields("metricsUsage")
|
|
898
|
+
|
|
899
|
+
source: "DatasetGraphQLField" = DatasetGraphQLField("source")
|
|
900
|
+
status: "DatasetGraphQLField" = DatasetGraphQLField("status")
|
|
901
|
+
deleted: "DatasetGraphQLField" = DatasetGraphQLField("deleted")
|
|
658
902
|
|
|
659
903
|
@classmethod
|
|
660
|
-
def progress(cls) ->
|
|
661
|
-
return DatasetProgressFields(
|
|
662
|
-
|
|
904
|
+
def progress(cls) -> "DatasetProgressFields":
|
|
905
|
+
return DatasetProgressFields("progress")
|
|
906
|
+
|
|
907
|
+
download_url: "DatasetGraphQLField" = DatasetGraphQLField("downloadUrl")
|
|
663
908
|
|
|
664
909
|
@classmethod
|
|
665
|
-
def use_case(cls) ->
|
|
666
|
-
return UseCaseFields(
|
|
910
|
+
def use_case(cls) -> "UseCaseFields":
|
|
911
|
+
return UseCaseFields("useCase")
|
|
667
912
|
|
|
668
|
-
def fields(
|
|
913
|
+
def fields(
|
|
914
|
+
self,
|
|
915
|
+
*subfields: Union[DatasetGraphQLField, "DatasetMetricUsageFields", "DatasetProgressFields", "UseCaseFields"],
|
|
916
|
+
) -> "DatasetFields":
|
|
669
917
|
"""Subfields should come from the DatasetFields class"""
|
|
670
918
|
self._subfields.extend(subfields)
|
|
671
919
|
return self
|
|
672
920
|
|
|
673
|
-
def alias(self, alias: str) ->
|
|
921
|
+
def alias(self, alias: str) -> "DatasetFields":
|
|
674
922
|
self._alias = alias
|
|
675
923
|
return self
|
|
676
924
|
|
|
925
|
+
|
|
677
926
|
class DatasetByproductsFields(GraphQLField):
|
|
678
927
|
"""@private"""
|
|
679
928
|
|
|
680
929
|
@classmethod
|
|
681
|
-
def dataset(cls) ->
|
|
682
|
-
return DatasetFields(
|
|
930
|
+
def dataset(cls) -> "DatasetFields":
|
|
931
|
+
return DatasetFields("dataset")
|
|
683
932
|
|
|
684
|
-
def fields(self, *subfields: Union[DatasetByproductsGraphQLField,
|
|
933
|
+
def fields(self, *subfields: Union[DatasetByproductsGraphQLField, "DatasetFields"]) -> "DatasetByproductsFields":
|
|
685
934
|
"""Subfields should come from the DatasetByproductsFields class"""
|
|
686
935
|
self._subfields.extend(subfields)
|
|
687
936
|
return self
|
|
688
937
|
|
|
689
|
-
def alias(self, alias: str) ->
|
|
938
|
+
def alias(self, alias: str) -> "DatasetByproductsFields":
|
|
690
939
|
self._alias = alias
|
|
691
940
|
return self
|
|
692
941
|
|
|
942
|
+
|
|
693
943
|
class DatasetMetricUsageFields(GraphQLField):
|
|
694
944
|
"""@private"""
|
|
695
945
|
|
|
696
946
|
@classmethod
|
|
697
|
-
def metric(cls) ->
|
|
698
|
-
return MetricFields(
|
|
699
|
-
|
|
700
|
-
|
|
947
|
+
def metric(cls) -> "MetricFields":
|
|
948
|
+
return MetricFields("metric")
|
|
949
|
+
|
|
950
|
+
feedback_count: "DatasetMetricUsageGraphQLField" = DatasetMetricUsageGraphQLField("feedbackCount")
|
|
951
|
+
comparison_count: "DatasetMetricUsageGraphQLField" = DatasetMetricUsageGraphQLField("comparisonCount")
|
|
701
952
|
|
|
702
|
-
def fields(self, *subfields: Union[DatasetMetricUsageGraphQLField,
|
|
953
|
+
def fields(self, *subfields: Union[DatasetMetricUsageGraphQLField, "MetricFields"]) -> "DatasetMetricUsageFields":
|
|
703
954
|
"""Subfields should come from the DatasetMetricUsageFields class"""
|
|
704
955
|
self._subfields.extend(subfields)
|
|
705
956
|
return self
|
|
706
957
|
|
|
707
|
-
def alias(self, alias: str) ->
|
|
958
|
+
def alias(self, alias: str) -> "DatasetMetricUsageFields":
|
|
708
959
|
self._alias = alias
|
|
709
960
|
return self
|
|
710
961
|
|
|
962
|
+
|
|
711
963
|
class DatasetProgressFields(GraphQLField):
|
|
712
964
|
"""@private"""
|
|
713
|
-
processed_parts: 'DatasetProgressGraphQLField' = DatasetProgressGraphQLField('processedParts')
|
|
714
|
-
total_parts: 'DatasetProgressGraphQLField' = DatasetProgressGraphQLField('totalParts')
|
|
715
|
-
progress: 'DatasetProgressGraphQLField' = DatasetProgressGraphQLField('progress')
|
|
716
|
-
processed_lines: 'DatasetProgressGraphQLField' = DatasetProgressGraphQLField('processedLines')
|
|
717
|
-
total_lines: 'DatasetProgressGraphQLField' = DatasetProgressGraphQLField('totalLines')
|
|
718
965
|
|
|
719
|
-
|
|
966
|
+
processed_parts: "DatasetProgressGraphQLField" = DatasetProgressGraphQLField("processedParts")
|
|
967
|
+
total_parts: "DatasetProgressGraphQLField" = DatasetProgressGraphQLField("totalParts")
|
|
968
|
+
progress: "DatasetProgressGraphQLField" = DatasetProgressGraphQLField("progress")
|
|
969
|
+
processed_lines: "DatasetProgressGraphQLField" = DatasetProgressGraphQLField("processedLines")
|
|
970
|
+
total_lines: "DatasetProgressGraphQLField" = DatasetProgressGraphQLField("totalLines")
|
|
971
|
+
|
|
972
|
+
def fields(self, *subfields: DatasetProgressGraphQLField) -> "DatasetProgressFields":
|
|
720
973
|
"""Subfields should come from the DatasetProgressFields class"""
|
|
721
974
|
self._subfields.extend(subfields)
|
|
722
975
|
return self
|
|
723
976
|
|
|
724
|
-
def alias(self, alias: str) ->
|
|
977
|
+
def alias(self, alias: str) -> "DatasetProgressFields":
|
|
725
978
|
self._alias = alias
|
|
726
979
|
return self
|
|
727
980
|
|
|
981
|
+
|
|
728
982
|
class DatasetUploadProcessingStatusFields(GraphQLField):
|
|
729
983
|
"""@private"""
|
|
730
|
-
dataset_id: 'DatasetUploadProcessingStatusGraphQLField' = DatasetUploadProcessingStatusGraphQLField('datasetId')
|
|
731
|
-
status: 'DatasetUploadProcessingStatusGraphQLField' = DatasetUploadProcessingStatusGraphQLField('status')
|
|
732
|
-
total_parts: 'DatasetUploadProcessingStatusGraphQLField' = DatasetUploadProcessingStatusGraphQLField('totalParts')
|
|
733
|
-
processed_parts: 'DatasetUploadProcessingStatusGraphQLField' = DatasetUploadProcessingStatusGraphQLField('processedParts')
|
|
734
|
-
progress: 'DatasetUploadProcessingStatusGraphQLField' = DatasetUploadProcessingStatusGraphQLField('progress')
|
|
735
|
-
error: 'DatasetUploadProcessingStatusGraphQLField' = DatasetUploadProcessingStatusGraphQLField('error')
|
|
736
984
|
|
|
737
|
-
|
|
985
|
+
dataset_id: "DatasetUploadProcessingStatusGraphQLField" = DatasetUploadProcessingStatusGraphQLField("datasetId")
|
|
986
|
+
status: "DatasetUploadProcessingStatusGraphQLField" = DatasetUploadProcessingStatusGraphQLField("status")
|
|
987
|
+
total_parts: "DatasetUploadProcessingStatusGraphQLField" = DatasetUploadProcessingStatusGraphQLField("totalParts")
|
|
988
|
+
processed_parts: "DatasetUploadProcessingStatusGraphQLField" = DatasetUploadProcessingStatusGraphQLField(
|
|
989
|
+
"processedParts"
|
|
990
|
+
)
|
|
991
|
+
progress: "DatasetUploadProcessingStatusGraphQLField" = DatasetUploadProcessingStatusGraphQLField("progress")
|
|
992
|
+
error: "DatasetUploadProcessingStatusGraphQLField" = DatasetUploadProcessingStatusGraphQLField("error")
|
|
993
|
+
|
|
994
|
+
def fields(self, *subfields: DatasetUploadProcessingStatusGraphQLField) -> "DatasetUploadProcessingStatusFields":
|
|
738
995
|
"""Subfields should come from the DatasetUploadProcessingStatusFields class"""
|
|
739
996
|
self._subfields.extend(subfields)
|
|
740
997
|
return self
|
|
741
998
|
|
|
742
|
-
def alias(self, alias: str) ->
|
|
999
|
+
def alias(self, alias: str) -> "DatasetUploadProcessingStatusFields":
|
|
743
1000
|
self._alias = alias
|
|
744
1001
|
return self
|
|
745
1002
|
|
|
1003
|
+
|
|
746
1004
|
class DatasetValidationOutputFields(GraphQLField):
|
|
747
1005
|
"""@private"""
|
|
748
|
-
valid: 'DatasetValidationOutputGraphQLField' = DatasetValidationOutputGraphQLField('valid')
|
|
749
|
-
message: 'DatasetValidationOutputGraphQLField' = DatasetValidationOutputGraphQLField('message')
|
|
750
1006
|
|
|
751
|
-
|
|
1007
|
+
valid: "DatasetValidationOutputGraphQLField" = DatasetValidationOutputGraphQLField("valid")
|
|
1008
|
+
message: "DatasetValidationOutputGraphQLField" = DatasetValidationOutputGraphQLField("message")
|
|
1009
|
+
|
|
1010
|
+
def fields(self, *subfields: DatasetValidationOutputGraphQLField) -> "DatasetValidationOutputFields":
|
|
752
1011
|
"""Subfields should come from the DatasetValidationOutputFields class"""
|
|
753
1012
|
self._subfields.extend(subfields)
|
|
754
1013
|
return self
|
|
755
1014
|
|
|
756
|
-
def alias(self, alias: str) ->
|
|
1015
|
+
def alias(self, alias: str) -> "DatasetValidationOutputFields":
|
|
757
1016
|
self._alias = alias
|
|
758
1017
|
return self
|
|
759
1018
|
|
|
1019
|
+
|
|
760
1020
|
class DeleteConfirmFields(GraphQLField):
|
|
761
1021
|
"""@private"""
|
|
762
|
-
success: 'DeleteConfirmGraphQLField' = DeleteConfirmGraphQLField('success')
|
|
763
|
-
details: 'DeleteConfirmGraphQLField' = DeleteConfirmGraphQLField('details')
|
|
764
1022
|
|
|
765
|
-
|
|
1023
|
+
success: "DeleteConfirmGraphQLField" = DeleteConfirmGraphQLField("success")
|
|
1024
|
+
details: "DeleteConfirmGraphQLField" = DeleteConfirmGraphQLField("details")
|
|
1025
|
+
|
|
1026
|
+
def fields(self, *subfields: DeleteConfirmGraphQLField) -> "DeleteConfirmFields":
|
|
766
1027
|
"""Subfields should come from the DeleteConfirmFields class"""
|
|
767
1028
|
self._subfields.extend(subfields)
|
|
768
1029
|
return self
|
|
769
1030
|
|
|
770
|
-
def alias(self, alias: str) ->
|
|
1031
|
+
def alias(self, alias: str) -> "DeleteConfirmFields":
|
|
771
1032
|
self._alias = alias
|
|
772
1033
|
return self
|
|
773
1034
|
|
|
1035
|
+
|
|
774
1036
|
class DirectFeedbackFields(GraphQLField):
|
|
775
1037
|
"""@private"""
|
|
776
|
-
|
|
777
|
-
|
|
778
|
-
|
|
1038
|
+
|
|
1039
|
+
id: "DirectFeedbackGraphQLField" = DirectFeedbackGraphQLField("id")
|
|
1040
|
+
value: "DirectFeedbackGraphQLField" = DirectFeedbackGraphQLField("value")
|
|
1041
|
+
user_id: "DirectFeedbackGraphQLField" = DirectFeedbackGraphQLField("userId")
|
|
779
1042
|
|
|
780
1043
|
@classmethod
|
|
781
|
-
def metric(cls) ->
|
|
782
|
-
return MetricFields(
|
|
783
|
-
reason: 'DirectFeedbackGraphQLField' = DirectFeedbackGraphQLField('reason')
|
|
784
|
-
details: 'DirectFeedbackGraphQLField' = DirectFeedbackGraphQLField('details')
|
|
785
|
-
created_at: 'DirectFeedbackGraphQLField' = DirectFeedbackGraphQLField('createdAt')
|
|
1044
|
+
def metric(cls) -> "MetricFields":
|
|
1045
|
+
return MetricFields("metric")
|
|
786
1046
|
|
|
787
|
-
|
|
1047
|
+
reason: "DirectFeedbackGraphQLField" = DirectFeedbackGraphQLField("reason")
|
|
1048
|
+
details: "DirectFeedbackGraphQLField" = DirectFeedbackGraphQLField("details")
|
|
1049
|
+
created_at: "DirectFeedbackGraphQLField" = DirectFeedbackGraphQLField("createdAt")
|
|
1050
|
+
|
|
1051
|
+
def fields(self, *subfields: Union[DirectFeedbackGraphQLField, "MetricFields"]) -> "DirectFeedbackFields":
|
|
788
1052
|
"""Subfields should come from the DirectFeedbackFields class"""
|
|
789
1053
|
self._subfields.extend(subfields)
|
|
790
1054
|
return self
|
|
791
1055
|
|
|
792
|
-
def alias(self, alias: str) ->
|
|
1056
|
+
def alias(self, alias: str) -> "DirectFeedbackFields":
|
|
793
1057
|
self._alias = alias
|
|
794
1058
|
return self
|
|
795
1059
|
|
|
1060
|
+
|
|
796
1061
|
class EmojiFields(GraphQLField):
|
|
797
1062
|
"""@private"""
|
|
798
|
-
native: 'EmojiGraphQLField' = EmojiGraphQLField('native')
|
|
799
1063
|
|
|
800
|
-
|
|
1064
|
+
native: "EmojiGraphQLField" = EmojiGraphQLField("native")
|
|
1065
|
+
|
|
1066
|
+
def fields(self, *subfields: EmojiGraphQLField) -> "EmojiFields":
|
|
801
1067
|
"""Subfields should come from the EmojiFields class"""
|
|
802
1068
|
self._subfields.extend(subfields)
|
|
803
1069
|
return self
|
|
804
1070
|
|
|
805
|
-
def alias(self, alias: str) ->
|
|
1071
|
+
def alias(self, alias: str) -> "EmojiFields":
|
|
806
1072
|
self._alias = alias
|
|
807
1073
|
return self
|
|
808
1074
|
|
|
1075
|
+
|
|
809
1076
|
class EvalJobStageOutputFields(GraphQLField):
|
|
810
1077
|
"""@private"""
|
|
811
|
-
total_num_samples: 'EvalJobStageOutputGraphQLField' = EvalJobStageOutputGraphQLField('totalNumSamples')
|
|
812
|
-
processed_num_samples: 'EvalJobStageOutputGraphQLField' = EvalJobStageOutputGraphQLField('processedNumSamples')
|
|
813
|
-
monitoring_link: 'EvalJobStageOutputGraphQLField' = EvalJobStageOutputGraphQLField('monitoringLink')
|
|
814
1078
|
|
|
815
|
-
|
|
1079
|
+
total_num_samples: "EvalJobStageOutputGraphQLField" = EvalJobStageOutputGraphQLField("totalNumSamples")
|
|
1080
|
+
processed_num_samples: "EvalJobStageOutputGraphQLField" = EvalJobStageOutputGraphQLField("processedNumSamples")
|
|
1081
|
+
monitoring_link: "EvalJobStageOutputGraphQLField" = EvalJobStageOutputGraphQLField("monitoringLink")
|
|
1082
|
+
|
|
1083
|
+
def fields(self, *subfields: EvalJobStageOutputGraphQLField) -> "EvalJobStageOutputFields":
|
|
816
1084
|
"""Subfields should come from the EvalJobStageOutputFields class"""
|
|
817
1085
|
self._subfields.extend(subfields)
|
|
818
1086
|
return self
|
|
819
1087
|
|
|
820
|
-
def alias(self, alias: str) ->
|
|
1088
|
+
def alias(self, alias: str) -> "EvalJobStageOutputFields":
|
|
821
1089
|
self._alias = alias
|
|
822
1090
|
return self
|
|
823
1091
|
|
|
1092
|
+
|
|
824
1093
|
class EvaluationByproductsFields(GraphQLField):
|
|
825
1094
|
"""@private"""
|
|
826
1095
|
|
|
827
1096
|
@classmethod
|
|
828
|
-
def eval_results(cls) ->
|
|
829
|
-
return EvaluationResultFields(
|
|
1097
|
+
def eval_results(cls) -> "EvaluationResultFields":
|
|
1098
|
+
return EvaluationResultFields("evalResults")
|
|
830
1099
|
|
|
831
|
-
def fields(
|
|
1100
|
+
def fields(
|
|
1101
|
+
self, *subfields: Union[EvaluationByproductsGraphQLField, "EvaluationResultFields"]
|
|
1102
|
+
) -> "EvaluationByproductsFields":
|
|
832
1103
|
"""Subfields should come from the EvaluationByproductsFields class"""
|
|
833
1104
|
self._subfields.extend(subfields)
|
|
834
1105
|
return self
|
|
835
1106
|
|
|
836
|
-
def alias(self, alias: str) ->
|
|
1107
|
+
def alias(self, alias: str) -> "EvaluationByproductsFields":
|
|
837
1108
|
self._alias = alias
|
|
838
1109
|
return self
|
|
839
1110
|
|
|
1111
|
+
|
|
840
1112
|
class EvaluationResultFields(GraphQLField):
|
|
841
1113
|
"""@private"""
|
|
842
1114
|
|
|
843
1115
|
@classmethod
|
|
844
|
-
def model_service(cls) ->
|
|
845
|
-
return ModelServiceFields(
|
|
1116
|
+
def model_service(cls) -> "ModelServiceFields":
|
|
1117
|
+
return ModelServiceFields("modelService")
|
|
846
1118
|
|
|
847
1119
|
@classmethod
|
|
848
|
-
def dataset(cls) ->
|
|
849
|
-
return DatasetFields(
|
|
1120
|
+
def dataset(cls) -> "DatasetFields":
|
|
1121
|
+
return DatasetFields("dataset")
|
|
850
1122
|
|
|
851
1123
|
@classmethod
|
|
852
|
-
def metric(cls) ->
|
|
853
|
-
return MetricFields(
|
|
854
|
-
|
|
855
|
-
|
|
856
|
-
|
|
857
|
-
|
|
858
|
-
|
|
859
|
-
|
|
860
|
-
|
|
861
|
-
|
|
862
|
-
|
|
863
|
-
|
|
1124
|
+
def metric(cls) -> "MetricFields":
|
|
1125
|
+
return MetricFields("metric")
|
|
1126
|
+
|
|
1127
|
+
mean: "EvaluationResultGraphQLField" = EvaluationResultGraphQLField("mean")
|
|
1128
|
+
min: "EvaluationResultGraphQLField" = EvaluationResultGraphQLField("min")
|
|
1129
|
+
max: "EvaluationResultGraphQLField" = EvaluationResultGraphQLField("max")
|
|
1130
|
+
stddev: "EvaluationResultGraphQLField" = EvaluationResultGraphQLField("stddev")
|
|
1131
|
+
sum: "EvaluationResultGraphQLField" = EvaluationResultGraphQLField("sum")
|
|
1132
|
+
count: "EvaluationResultGraphQLField" = EvaluationResultGraphQLField("count")
|
|
1133
|
+
sum_squared: "EvaluationResultGraphQLField" = EvaluationResultGraphQLField("sumSquared")
|
|
1134
|
+
job_id: "EvaluationResultGraphQLField" = EvaluationResultGraphQLField("jobId")
|
|
1135
|
+
artifact_id: "EvaluationResultGraphQLField" = EvaluationResultGraphQLField("artifactId")
|
|
1136
|
+
feedback_count: "EvaluationResultGraphQLField" = EvaluationResultGraphQLField("feedbackCount")
|
|
864
1137
|
|
|
865
1138
|
@classmethod
|
|
866
|
-
def judge(cls) ->
|
|
867
|
-
return JudgeFields(
|
|
1139
|
+
def judge(cls) -> "JudgeFields":
|
|
1140
|
+
return JudgeFields("judge")
|
|
868
1141
|
|
|
869
1142
|
@classmethod
|
|
870
|
-
def grader(cls) ->
|
|
871
|
-
return GraderFields(
|
|
1143
|
+
def grader(cls) -> "GraderFields":
|
|
1144
|
+
return GraderFields("grader")
|
|
872
1145
|
|
|
873
|
-
def fields(
|
|
1146
|
+
def fields(
|
|
1147
|
+
self,
|
|
1148
|
+
*subfields: Union[
|
|
1149
|
+
EvaluationResultGraphQLField,
|
|
1150
|
+
"DatasetFields",
|
|
1151
|
+
"GraderFields",
|
|
1152
|
+
"JudgeFields",
|
|
1153
|
+
"MetricFields",
|
|
1154
|
+
"ModelServiceFields",
|
|
1155
|
+
],
|
|
1156
|
+
) -> "EvaluationResultFields":
|
|
874
1157
|
"""Subfields should come from the EvaluationResultFields class"""
|
|
875
1158
|
self._subfields.extend(subfields)
|
|
876
1159
|
return self
|
|
877
1160
|
|
|
878
|
-
def alias(self, alias: str) ->
|
|
1161
|
+
def alias(self, alias: str) -> "EvaluationResultFields":
|
|
879
1162
|
self._alias = alias
|
|
880
1163
|
return self
|
|
881
1164
|
|
|
1165
|
+
|
|
1166
|
+
class GitHubConfigFields(GraphQLField):
|
|
1167
|
+
"""@private"""
|
|
1168
|
+
|
|
1169
|
+
api_token: "GitHubConfigGraphQLField" = GitHubConfigGraphQLField("apiToken")
|
|
1170
|
+
org: "GitHubConfigGraphQLField" = GitHubConfigGraphQLField("org")
|
|
1171
|
+
repo: "GitHubConfigGraphQLField" = GitHubConfigGraphQLField("repo")
|
|
1172
|
+
|
|
1173
|
+
def fields(self, *subfields: GitHubConfigGraphQLField) -> "GitHubConfigFields":
|
|
1174
|
+
"""Subfields should come from the GitHubConfigFields class"""
|
|
1175
|
+
self._subfields.extend(subfields)
|
|
1176
|
+
return self
|
|
1177
|
+
|
|
1178
|
+
def alias(self, alias: str) -> "GitHubConfigFields":
|
|
1179
|
+
self._alias = alias
|
|
1180
|
+
return self
|
|
1181
|
+
|
|
1182
|
+
|
|
882
1183
|
class GlobalUsageFields(GraphQLField):
|
|
883
1184
|
"""@private"""
|
|
884
1185
|
|
|
885
1186
|
@classmethod
|
|
886
|
-
def total(cls) ->
|
|
887
|
-
return UsageStatsFields(
|
|
1187
|
+
def total(cls) -> "UsageStatsFields":
|
|
1188
|
+
return UsageStatsFields("total")
|
|
888
1189
|
|
|
889
1190
|
@classmethod
|
|
890
|
-
def adaptive_models(cls) ->
|
|
891
|
-
return UsageStatsFields(
|
|
1191
|
+
def adaptive_models(cls) -> "UsageStatsFields":
|
|
1192
|
+
return UsageStatsFields("adaptiveModels")
|
|
892
1193
|
|
|
893
1194
|
@classmethod
|
|
894
|
-
def external_models(cls) ->
|
|
895
|
-
return UsageStatsFields(
|
|
1195
|
+
def external_models(cls) -> "UsageStatsFields":
|
|
1196
|
+
return UsageStatsFields("externalModels")
|
|
896
1197
|
|
|
897
1198
|
@classmethod
|
|
898
|
-
def by_model(cls) ->
|
|
899
|
-
return UsageStatsByModelFields(
|
|
900
|
-
signature: 'GlobalUsageGraphQLField' = GlobalUsageGraphQLField('signature')
|
|
1199
|
+
def by_model(cls) -> "UsageStatsByModelFields":
|
|
1200
|
+
return UsageStatsByModelFields("byModel")
|
|
901
1201
|
|
|
902
|
-
|
|
1202
|
+
signature: "GlobalUsageGraphQLField" = GlobalUsageGraphQLField("signature")
|
|
1203
|
+
|
|
1204
|
+
def fields(
|
|
1205
|
+
self, *subfields: Union[GlobalUsageGraphQLField, "UsageStatsByModelFields", "UsageStatsFields"]
|
|
1206
|
+
) -> "GlobalUsageFields":
|
|
903
1207
|
"""Subfields should come from the GlobalUsageFields class"""
|
|
904
1208
|
self._subfields.extend(subfields)
|
|
905
1209
|
return self
|
|
906
1210
|
|
|
907
|
-
def alias(self, alias: str) ->
|
|
1211
|
+
def alias(self, alias: str) -> "GlobalUsageFields":
|
|
908
1212
|
self._alias = alias
|
|
909
1213
|
return self
|
|
910
1214
|
|
|
1215
|
+
|
|
911
1216
|
class GpuAllocationFields(GraphQLField):
|
|
912
1217
|
"""@private"""
|
|
913
|
-
name: 'GpuAllocationGraphQLField' = GpuAllocationGraphQLField('name')
|
|
914
|
-
num_gpus: 'GpuAllocationGraphQLField' = GpuAllocationGraphQLField('numGpus')
|
|
915
|
-
ranks: 'GpuAllocationGraphQLField' = GpuAllocationGraphQLField('ranks')
|
|
916
|
-
created_at: 'GpuAllocationGraphQLField' = GpuAllocationGraphQLField('createdAt')
|
|
917
|
-
user_name: 'GpuAllocationGraphQLField' = GpuAllocationGraphQLField('userName')
|
|
918
|
-
job_id: 'GpuAllocationGraphQLField' = GpuAllocationGraphQLField('jobId')
|
|
919
1218
|
|
|
920
|
-
|
|
1219
|
+
name: "GpuAllocationGraphQLField" = GpuAllocationGraphQLField("name")
|
|
1220
|
+
num_gpus: "GpuAllocationGraphQLField" = GpuAllocationGraphQLField("numGpus")
|
|
1221
|
+
ranks: "GpuAllocationGraphQLField" = GpuAllocationGraphQLField("ranks")
|
|
1222
|
+
created_at: "GpuAllocationGraphQLField" = GpuAllocationGraphQLField("createdAt")
|
|
1223
|
+
user_name: "GpuAllocationGraphQLField" = GpuAllocationGraphQLField("userName")
|
|
1224
|
+
job_id: "GpuAllocationGraphQLField" = GpuAllocationGraphQLField("jobId")
|
|
1225
|
+
|
|
1226
|
+
def fields(self, *subfields: GpuAllocationGraphQLField) -> "GpuAllocationFields":
|
|
921
1227
|
"""Subfields should come from the GpuAllocationFields class"""
|
|
922
1228
|
self._subfields.extend(subfields)
|
|
923
1229
|
return self
|
|
924
1230
|
|
|
925
|
-
def alias(self, alias: str) ->
|
|
1231
|
+
def alias(self, alias: str) -> "GpuAllocationFields":
|
|
926
1232
|
self._alias = alias
|
|
927
1233
|
return self
|
|
928
1234
|
|
|
1235
|
+
|
|
929
1236
|
class GraderFields(GraphQLField):
|
|
930
1237
|
"""@private"""
|
|
931
|
-
|
|
932
|
-
|
|
933
|
-
|
|
934
|
-
|
|
935
|
-
|
|
936
|
-
|
|
1238
|
+
|
|
1239
|
+
id: "GraderGraphQLField" = GraderGraphQLField("id")
|
|
1240
|
+
name: "GraderGraphQLField" = GraderGraphQLField("name")
|
|
1241
|
+
key: "GraderGraphQLField" = GraderGraphQLField("key")
|
|
1242
|
+
locked: "GraderGraphQLField" = GraderGraphQLField("locked")
|
|
1243
|
+
grader_type: "GraderGraphQLField" = GraderGraphQLField("graderType")
|
|
1244
|
+
grader_config: "GraderConfigUnion" = GraderConfigUnion("graderConfig")
|
|
937
1245
|
|
|
938
1246
|
@classmethod
|
|
939
|
-
def use_case(cls) ->
|
|
940
|
-
return UseCaseFields(
|
|
1247
|
+
def use_case(cls) -> "UseCaseFields":
|
|
1248
|
+
return UseCaseFields("useCase")
|
|
941
1249
|
|
|
942
1250
|
@classmethod
|
|
943
|
-
def metric(cls) ->
|
|
944
|
-
return MetricFields(
|
|
945
|
-
created_at: 'GraderGraphQLField' = GraderGraphQLField('createdAt')
|
|
946
|
-
updated_at: 'GraderGraphQLField' = GraderGraphQLField('updatedAt')
|
|
1251
|
+
def metric(cls) -> "MetricFields":
|
|
1252
|
+
return MetricFields("metric")
|
|
947
1253
|
|
|
948
|
-
|
|
1254
|
+
created_at: "GraderGraphQLField" = GraderGraphQLField("createdAt")
|
|
1255
|
+
updated_at: "GraderGraphQLField" = GraderGraphQLField("updatedAt")
|
|
1256
|
+
|
|
1257
|
+
def fields(
|
|
1258
|
+
self, *subfields: Union[GraderGraphQLField, "GraderConfigUnion", "MetricFields", "UseCaseFields"]
|
|
1259
|
+
) -> "GraderFields":
|
|
949
1260
|
"""Subfields should come from the GraderFields class"""
|
|
950
1261
|
self._subfields.extend(subfields)
|
|
951
1262
|
return self
|
|
952
1263
|
|
|
953
|
-
def alias(self, alias: str) ->
|
|
1264
|
+
def alias(self, alias: str) -> "GraderFields":
|
|
954
1265
|
self._alias = alias
|
|
955
1266
|
return self
|
|
956
1267
|
|
|
1268
|
+
|
|
957
1269
|
class HarmonyGroupFields(GraphQLField):
|
|
958
1270
|
"""@private"""
|
|
959
|
-
|
|
960
|
-
|
|
1271
|
+
|
|
1272
|
+
id: "HarmonyGroupGraphQLField" = HarmonyGroupGraphQLField("id")
|
|
1273
|
+
key: "HarmonyGroupGraphQLField" = HarmonyGroupGraphQLField("key")
|
|
961
1274
|
|
|
962
1275
|
@classmethod
|
|
963
|
-
def compute_pool(cls) ->
|
|
964
|
-
return ComputePoolFields(
|
|
965
|
-
|
|
966
|
-
|
|
967
|
-
|
|
968
|
-
|
|
969
|
-
|
|
1276
|
+
def compute_pool(cls) -> "ComputePoolFields":
|
|
1277
|
+
return ComputePoolFields("computePool")
|
|
1278
|
+
|
|
1279
|
+
status: "HarmonyGroupGraphQLField" = HarmonyGroupGraphQLField("status")
|
|
1280
|
+
url: "HarmonyGroupGraphQLField" = HarmonyGroupGraphQLField("url")
|
|
1281
|
+
world_size: "HarmonyGroupGraphQLField" = HarmonyGroupGraphQLField("worldSize")
|
|
1282
|
+
gpu_total: "HarmonyGroupGraphQLField" = HarmonyGroupGraphQLField("gpuTotal")
|
|
1283
|
+
gpu_allocated: "HarmonyGroupGraphQLField" = HarmonyGroupGraphQLField("gpuAllocated")
|
|
970
1284
|
|
|
971
1285
|
@classmethod
|
|
972
|
-
def gpu_allocations(cls) ->
|
|
973
|
-
return GpuAllocationFields(
|
|
974
|
-
|
|
975
|
-
|
|
1286
|
+
def gpu_allocations(cls) -> "GpuAllocationFields":
|
|
1287
|
+
return GpuAllocationFields("gpuAllocations")
|
|
1288
|
+
|
|
1289
|
+
gpu_types: "HarmonyGroupGraphQLField" = HarmonyGroupGraphQLField("gpuTypes")
|
|
1290
|
+
created_at: "HarmonyGroupGraphQLField" = HarmonyGroupGraphQLField("createdAt")
|
|
976
1291
|
|
|
977
1292
|
@classmethod
|
|
978
|
-
def online_models(cls) ->
|
|
979
|
-
return ModelFields(
|
|
1293
|
+
def online_models(cls) -> "ModelFields":
|
|
1294
|
+
return ModelFields("onlineModels")
|
|
980
1295
|
|
|
981
|
-
def fields(
|
|
1296
|
+
def fields(
|
|
1297
|
+
self, *subfields: Union[HarmonyGroupGraphQLField, "ComputePoolFields", "GpuAllocationFields", "ModelFields"]
|
|
1298
|
+
) -> "HarmonyGroupFields":
|
|
982
1299
|
"""Subfields should come from the HarmonyGroupFields class"""
|
|
983
1300
|
self._subfields.extend(subfields)
|
|
984
1301
|
return self
|
|
985
1302
|
|
|
986
|
-
def alias(self, alias: str) ->
|
|
1303
|
+
def alias(self, alias: str) -> "HarmonyGroupFields":
|
|
1304
|
+
self._alias = alias
|
|
1305
|
+
return self
|
|
1306
|
+
|
|
1307
|
+
|
|
1308
|
+
class IntegrationFields(GraphQLField):
|
|
1309
|
+
"""@private"""
|
|
1310
|
+
|
|
1311
|
+
id: "IntegrationGraphQLField" = IntegrationGraphQLField("id")
|
|
1312
|
+
team_id: "IntegrationGraphQLField" = IntegrationGraphQLField("teamId")
|
|
1313
|
+
name: "IntegrationGraphQLField" = IntegrationGraphQLField("name")
|
|
1314
|
+
capabilities: "IntegrationGraphQLField" = IntegrationGraphQLField("capabilities")
|
|
1315
|
+
|
|
1316
|
+
@classmethod
|
|
1317
|
+
def config(cls) -> "IntegrationConfigFields":
|
|
1318
|
+
return IntegrationConfigFields("config")
|
|
1319
|
+
|
|
1320
|
+
enabled: "IntegrationGraphQLField" = IntegrationGraphQLField("enabled")
|
|
1321
|
+
created_at: "IntegrationGraphQLField" = IntegrationGraphQLField("createdAt")
|
|
1322
|
+
|
|
1323
|
+
def fields(self, *subfields: Union[IntegrationGraphQLField, "IntegrationConfigFields"]) -> "IntegrationFields":
|
|
1324
|
+
"""Subfields should come from the IntegrationFields class"""
|
|
1325
|
+
self._subfields.extend(subfields)
|
|
1326
|
+
return self
|
|
1327
|
+
|
|
1328
|
+
def alias(self, alias: str) -> "IntegrationFields":
|
|
987
1329
|
self._alias = alias
|
|
988
1330
|
return self
|
|
989
1331
|
|
|
1332
|
+
|
|
1333
|
+
class IntegrationConfigFields(GraphQLField):
|
|
1334
|
+
"""@private"""
|
|
1335
|
+
|
|
1336
|
+
provider: "IntegrationConfigGraphQLField" = IntegrationConfigGraphQLField("provider")
|
|
1337
|
+
connection: "ConnectionConfigUnion" = ConnectionConfigUnion("connection")
|
|
1338
|
+
|
|
1339
|
+
@classmethod
|
|
1340
|
+
def notification_settings(cls) -> "NotificationSettingsFields":
|
|
1341
|
+
return NotificationSettingsFields("notificationSettings")
|
|
1342
|
+
|
|
1343
|
+
delivery_policy: "IntegrationConfigGraphQLField" = IntegrationConfigGraphQLField("deliveryPolicy")
|
|
1344
|
+
|
|
1345
|
+
def fields(
|
|
1346
|
+
self, *subfields: Union[IntegrationConfigGraphQLField, "ConnectionConfigUnion", "NotificationSettingsFields"]
|
|
1347
|
+
) -> "IntegrationConfigFields":
|
|
1348
|
+
"""Subfields should come from the IntegrationConfigFields class"""
|
|
1349
|
+
self._subfields.extend(subfields)
|
|
1350
|
+
return self
|
|
1351
|
+
|
|
1352
|
+
def alias(self, alias: str) -> "IntegrationConfigFields":
|
|
1353
|
+
self._alias = alias
|
|
1354
|
+
return self
|
|
1355
|
+
|
|
1356
|
+
|
|
990
1357
|
class InteractionOutputFields(GraphQLField):
|
|
991
1358
|
"""@private"""
|
|
992
|
-
value: 'InteractionOutputGraphQLField' = InteractionOutputGraphQLField('value')
|
|
993
|
-
per_second: 'InteractionOutputGraphQLField' = InteractionOutputGraphQLField('perSecond')
|
|
994
|
-
trend: 'InteractionOutputGraphQLField' = InteractionOutputGraphQLField('trend')
|
|
995
1359
|
|
|
996
|
-
|
|
1360
|
+
value: "InteractionOutputGraphQLField" = InteractionOutputGraphQLField("value")
|
|
1361
|
+
per_second: "InteractionOutputGraphQLField" = InteractionOutputGraphQLField("perSecond")
|
|
1362
|
+
trend: "InteractionOutputGraphQLField" = InteractionOutputGraphQLField("trend")
|
|
1363
|
+
|
|
1364
|
+
def fields(self, *subfields: InteractionOutputGraphQLField) -> "InteractionOutputFields":
|
|
997
1365
|
"""Subfields should come from the InteractionOutputFields class"""
|
|
998
1366
|
self._subfields.extend(subfields)
|
|
999
1367
|
return self
|
|
1000
1368
|
|
|
1001
|
-
def alias(self, alias: str) ->
|
|
1369
|
+
def alias(self, alias: str) -> "InteractionOutputFields":
|
|
1002
1370
|
self._alias = alias
|
|
1003
1371
|
return self
|
|
1004
1372
|
|
|
1373
|
+
|
|
1005
1374
|
class IntervalFields(GraphQLField):
|
|
1006
1375
|
"""@private"""
|
|
1007
|
-
start: 'IntervalGraphQLField' = IntervalGraphQLField('start')
|
|
1008
|
-
middle: 'IntervalGraphQLField' = IntervalGraphQLField('middle')
|
|
1009
|
-
end: 'IntervalGraphQLField' = IntervalGraphQLField('end')
|
|
1010
1376
|
|
|
1011
|
-
|
|
1377
|
+
start: "IntervalGraphQLField" = IntervalGraphQLField("start")
|
|
1378
|
+
middle: "IntervalGraphQLField" = IntervalGraphQLField("middle")
|
|
1379
|
+
end: "IntervalGraphQLField" = IntervalGraphQLField("end")
|
|
1380
|
+
|
|
1381
|
+
def fields(self, *subfields: IntervalGraphQLField) -> "IntervalFields":
|
|
1012
1382
|
"""Subfields should come from the IntervalFields class"""
|
|
1013
1383
|
self._subfields.extend(subfields)
|
|
1014
1384
|
return self
|
|
1015
1385
|
|
|
1016
|
-
def alias(self, alias: str) ->
|
|
1386
|
+
def alias(self, alias: str) -> "IntervalFields":
|
|
1017
1387
|
self._alias = alias
|
|
1018
1388
|
return self
|
|
1019
1389
|
|
|
1390
|
+
|
|
1020
1391
|
class JobFields(GraphQLField):
|
|
1021
1392
|
"""@private"""
|
|
1022
|
-
|
|
1393
|
+
|
|
1394
|
+
id: "JobGraphQLField" = JobGraphQLField("id")
|
|
1023
1395
|
|
|
1024
1396
|
@classmethod
|
|
1025
|
-
def use_case(cls) ->
|
|
1026
|
-
return UseCaseFields(
|
|
1027
|
-
|
|
1028
|
-
|
|
1029
|
-
|
|
1397
|
+
def use_case(cls) -> "UseCaseFields":
|
|
1398
|
+
return UseCaseFields("useCase")
|
|
1399
|
+
|
|
1400
|
+
name: "JobGraphQLField" = JobGraphQLField("name")
|
|
1401
|
+
status: "JobGraphQLField" = JobGraphQLField("status")
|
|
1402
|
+
created_at: "JobGraphQLField" = JobGraphQLField("createdAt")
|
|
1030
1403
|
|
|
1031
1404
|
@classmethod
|
|
1032
|
-
def created_by(cls) ->
|
|
1033
|
-
return UserFields(
|
|
1034
|
-
|
|
1035
|
-
|
|
1036
|
-
|
|
1405
|
+
def created_by(cls) -> "UserFields":
|
|
1406
|
+
return UserFields("createdBy")
|
|
1407
|
+
|
|
1408
|
+
started_at: "JobGraphQLField" = JobGraphQLField("startedAt")
|
|
1409
|
+
ended_at: "JobGraphQLField" = JobGraphQLField("endedAt")
|
|
1410
|
+
duration_ms: "JobGraphQLField" = JobGraphQLField("durationMs")
|
|
1037
1411
|
|
|
1038
1412
|
@classmethod
|
|
1039
|
-
def stages(cls) ->
|
|
1040
|
-
return JobStageOutputFields(
|
|
1041
|
-
|
|
1042
|
-
|
|
1043
|
-
|
|
1413
|
+
def stages(cls) -> "JobStageOutputFields":
|
|
1414
|
+
return JobStageOutputFields("stages")
|
|
1415
|
+
|
|
1416
|
+
progress: "JobGraphQLField" = JobGraphQLField("progress")
|
|
1417
|
+
error: "JobGraphQLField" = JobGraphQLField("error")
|
|
1418
|
+
kind: "JobGraphQLField" = JobGraphQLField("kind")
|
|
1044
1419
|
|
|
1045
1420
|
@classmethod
|
|
1046
|
-
def recipe(cls) ->
|
|
1047
|
-
return CustomRecipeFields(
|
|
1421
|
+
def recipe(cls) -> "CustomRecipeFields":
|
|
1422
|
+
return CustomRecipeFields("recipe")
|
|
1048
1423
|
|
|
1049
1424
|
@classmethod
|
|
1050
|
-
def details(cls) ->
|
|
1051
|
-
return CustomRecipeJobDetailsFields(
|
|
1425
|
+
def details(cls) -> "CustomRecipeJobDetailsFields":
|
|
1426
|
+
return CustomRecipeJobDetailsFields("details")
|
|
1052
1427
|
|
|
1053
|
-
def fields(
|
|
1428
|
+
def fields(
|
|
1429
|
+
self,
|
|
1430
|
+
*subfields: Union[
|
|
1431
|
+
JobGraphQLField,
|
|
1432
|
+
"CustomRecipeFields",
|
|
1433
|
+
"CustomRecipeJobDetailsFields",
|
|
1434
|
+
"JobStageOutputFields",
|
|
1435
|
+
"UseCaseFields",
|
|
1436
|
+
"UserFields",
|
|
1437
|
+
],
|
|
1438
|
+
) -> "JobFields":
|
|
1054
1439
|
"""Subfields should come from the JobFields class"""
|
|
1055
1440
|
self._subfields.extend(subfields)
|
|
1056
1441
|
return self
|
|
1057
1442
|
|
|
1058
|
-
def alias(self, alias: str) ->
|
|
1443
|
+
def alias(self, alias: str) -> "JobFields":
|
|
1059
1444
|
self._alias = alias
|
|
1060
1445
|
return self
|
|
1061
1446
|
|
|
1447
|
+
|
|
1062
1448
|
class JobArtifactFields(GraphQLField):
|
|
1063
1449
|
"""@private"""
|
|
1064
|
-
|
|
1065
|
-
|
|
1066
|
-
|
|
1067
|
-
|
|
1068
|
-
|
|
1069
|
-
|
|
1070
|
-
|
|
1071
|
-
|
|
1450
|
+
|
|
1451
|
+
id: "JobArtifactGraphQLField" = JobArtifactGraphQLField("id")
|
|
1452
|
+
job_id: "JobArtifactGraphQLField" = JobArtifactGraphQLField("jobId")
|
|
1453
|
+
name: "JobArtifactGraphQLField" = JobArtifactGraphQLField("name")
|
|
1454
|
+
kind: "JobArtifactGraphQLField" = JobArtifactGraphQLField("kind")
|
|
1455
|
+
status: "JobArtifactGraphQLField" = JobArtifactGraphQLField("status")
|
|
1456
|
+
uri: "JobArtifactGraphQLField" = JobArtifactGraphQLField("uri")
|
|
1457
|
+
metadata: "JobArtifactGraphQLField" = JobArtifactGraphQLField("metadata")
|
|
1458
|
+
created_at: "JobArtifactGraphQLField" = JobArtifactGraphQLField("createdAt")
|
|
1072
1459
|
|
|
1073
1460
|
@classmethod
|
|
1074
|
-
def job(cls) ->
|
|
1075
|
-
return JobFields(
|
|
1076
|
-
download_url: 'JobArtifactGraphQLField' = JobArtifactGraphQLField('downloadUrl')
|
|
1077
|
-
byproducts: 'ArtifactByproductsUnion' = ArtifactByproductsUnion('byproducts')
|
|
1461
|
+
def job(cls) -> "JobFields":
|
|
1462
|
+
return JobFields("job")
|
|
1078
1463
|
|
|
1079
|
-
|
|
1464
|
+
download_url: "JobArtifactGraphQLField" = JobArtifactGraphQLField("downloadUrl")
|
|
1465
|
+
byproducts: "ArtifactByproductsUnion" = ArtifactByproductsUnion("byproducts")
|
|
1466
|
+
|
|
1467
|
+
def fields(
|
|
1468
|
+
self, *subfields: Union[JobArtifactGraphQLField, "ArtifactByproductsUnion", "JobFields"]
|
|
1469
|
+
) -> "JobArtifactFields":
|
|
1080
1470
|
"""Subfields should come from the JobArtifactFields class"""
|
|
1081
1471
|
self._subfields.extend(subfields)
|
|
1082
1472
|
return self
|
|
1083
1473
|
|
|
1084
|
-
def alias(self, alias: str) ->
|
|
1474
|
+
def alias(self, alias: str) -> "JobArtifactFields":
|
|
1085
1475
|
self._alias = alias
|
|
1086
1476
|
return self
|
|
1087
1477
|
|
|
1478
|
+
|
|
1088
1479
|
class JobConnectionFields(GraphQLField):
|
|
1089
1480
|
"""@private"""
|
|
1090
1481
|
|
|
1091
1482
|
@classmethod
|
|
1092
|
-
def page_info(cls) ->
|
|
1483
|
+
def page_info(cls) -> "PageInfoFields":
|
|
1093
1484
|
"""Information to aid in pagination."""
|
|
1094
|
-
return PageInfoFields(
|
|
1485
|
+
return PageInfoFields("pageInfo")
|
|
1095
1486
|
|
|
1096
1487
|
@classmethod
|
|
1097
|
-
def edges(cls) ->
|
|
1488
|
+
def edges(cls) -> "JobEdgeFields":
|
|
1098
1489
|
"""A list of edges."""
|
|
1099
|
-
return JobEdgeFields(
|
|
1490
|
+
return JobEdgeFields("edges")
|
|
1100
1491
|
|
|
1101
1492
|
@classmethod
|
|
1102
|
-
def nodes(cls) ->
|
|
1493
|
+
def nodes(cls) -> "JobFields":
|
|
1103
1494
|
"""A list of nodes."""
|
|
1104
|
-
return JobFields(
|
|
1105
|
-
|
|
1495
|
+
return JobFields("nodes")
|
|
1496
|
+
|
|
1497
|
+
total_count: "JobConnectionGraphQLField" = JobConnectionGraphQLField("totalCount")
|
|
1106
1498
|
|
|
1107
|
-
def fields(
|
|
1499
|
+
def fields(
|
|
1500
|
+
self, *subfields: Union[JobConnectionGraphQLField, "JobEdgeFields", "JobFields", "PageInfoFields"]
|
|
1501
|
+
) -> "JobConnectionFields":
|
|
1108
1502
|
"""Subfields should come from the JobConnectionFields class"""
|
|
1109
1503
|
self._subfields.extend(subfields)
|
|
1110
1504
|
return self
|
|
1111
1505
|
|
|
1112
|
-
def alias(self, alias: str) ->
|
|
1506
|
+
def alias(self, alias: str) -> "JobConnectionFields":
|
|
1113
1507
|
self._alias = alias
|
|
1114
1508
|
return self
|
|
1115
1509
|
|
|
1510
|
+
|
|
1116
1511
|
class JobEdgeFields(GraphQLField):
|
|
1117
1512
|
"""@private
|
|
1118
|
-
An edge in a connection."""
|
|
1513
|
+
An edge in a connection."""
|
|
1119
1514
|
|
|
1120
1515
|
@classmethod
|
|
1121
|
-
def node(cls) ->
|
|
1516
|
+
def node(cls) -> "JobFields":
|
|
1122
1517
|
"""The item at the end of the edge"""
|
|
1123
|
-
return JobFields(
|
|
1124
|
-
|
|
1125
|
-
|
|
1518
|
+
return JobFields("node")
|
|
1519
|
+
|
|
1520
|
+
cursor: "JobEdgeGraphQLField" = JobEdgeGraphQLField("cursor")
|
|
1521
|
+
"A cursor for use in pagination"
|
|
1126
1522
|
|
|
1127
|
-
def fields(self, *subfields: Union[JobEdgeGraphQLField,
|
|
1523
|
+
def fields(self, *subfields: Union[JobEdgeGraphQLField, "JobFields"]) -> "JobEdgeFields":
|
|
1128
1524
|
"""Subfields should come from the JobEdgeFields class"""
|
|
1129
1525
|
self._subfields.extend(subfields)
|
|
1130
1526
|
return self
|
|
1131
1527
|
|
|
1132
|
-
def alias(self, alias: str) ->
|
|
1528
|
+
def alias(self, alias: str) -> "JobEdgeFields":
|
|
1133
1529
|
self._alias = alias
|
|
1134
1530
|
return self
|
|
1135
1531
|
|
|
1532
|
+
|
|
1136
1533
|
class JobStageOutputFields(GraphQLField):
|
|
1137
1534
|
"""@private"""
|
|
1138
|
-
name: 'JobStageOutputGraphQLField' = JobStageOutputGraphQLField('name')
|
|
1139
|
-
status: 'JobStageOutputGraphQLField' = JobStageOutputGraphQLField('status')
|
|
1140
|
-
parent: 'JobStageOutputGraphQLField' = JobStageOutputGraphQLField('parent')
|
|
1141
|
-
stage_id: 'JobStageOutputGraphQLField' = JobStageOutputGraphQLField('stageId')
|
|
1142
|
-
info: 'JobStageInfoOutputUnion' = JobStageInfoOutputUnion('info')
|
|
1143
|
-
started_at: 'JobStageOutputGraphQLField' = JobStageOutputGraphQLField('startedAt')
|
|
1144
|
-
ended_at: 'JobStageOutputGraphQLField' = JobStageOutputGraphQLField('endedAt')
|
|
1145
|
-
duration_ms: 'JobStageOutputGraphQLField' = JobStageOutputGraphQLField('durationMs')
|
|
1146
1535
|
|
|
1147
|
-
|
|
1536
|
+
name: "JobStageOutputGraphQLField" = JobStageOutputGraphQLField("name")
|
|
1537
|
+
status: "JobStageOutputGraphQLField" = JobStageOutputGraphQLField("status")
|
|
1538
|
+
parent: "JobStageOutputGraphQLField" = JobStageOutputGraphQLField("parent")
|
|
1539
|
+
stage_id: "JobStageOutputGraphQLField" = JobStageOutputGraphQLField("stageId")
|
|
1540
|
+
info: "JobStageInfoOutputUnion" = JobStageInfoOutputUnion("info")
|
|
1541
|
+
started_at: "JobStageOutputGraphQLField" = JobStageOutputGraphQLField("startedAt")
|
|
1542
|
+
ended_at: "JobStageOutputGraphQLField" = JobStageOutputGraphQLField("endedAt")
|
|
1543
|
+
duration_ms: "JobStageOutputGraphQLField" = JobStageOutputGraphQLField("durationMs")
|
|
1544
|
+
|
|
1545
|
+
def fields(
|
|
1546
|
+
self, *subfields: Union[JobStageOutputGraphQLField, "JobStageInfoOutputUnion"]
|
|
1547
|
+
) -> "JobStageOutputFields":
|
|
1148
1548
|
"""Subfields should come from the JobStageOutputFields class"""
|
|
1149
1549
|
self._subfields.extend(subfields)
|
|
1150
1550
|
return self
|
|
1151
1551
|
|
|
1152
|
-
def alias(self, alias: str) ->
|
|
1552
|
+
def alias(self, alias: str) -> "JobStageOutputFields":
|
|
1153
1553
|
self._alias = alias
|
|
1154
1554
|
return self
|
|
1155
1555
|
|
|
1556
|
+
|
|
1156
1557
|
class JudgeFields(GraphQLField):
|
|
1157
1558
|
"""@private"""
|
|
1158
|
-
|
|
1159
|
-
|
|
1160
|
-
|
|
1161
|
-
|
|
1162
|
-
|
|
1163
|
-
|
|
1559
|
+
|
|
1560
|
+
id: "JudgeGraphQLField" = JudgeGraphQLField("id")
|
|
1561
|
+
key: "JudgeGraphQLField" = JudgeGraphQLField("key")
|
|
1562
|
+
version: "JudgeGraphQLField" = JudgeGraphQLField("version")
|
|
1563
|
+
name: "JudgeGraphQLField" = JudgeGraphQLField("name")
|
|
1564
|
+
criteria: "JudgeGraphQLField" = JudgeGraphQLField("criteria")
|
|
1565
|
+
prebuilt: "JudgeGraphQLField" = JudgeGraphQLField("prebuilt")
|
|
1164
1566
|
|
|
1165
1567
|
@classmethod
|
|
1166
|
-
def examples(cls) ->
|
|
1167
|
-
return JudgeExampleFields(
|
|
1168
|
-
|
|
1568
|
+
def examples(cls) -> "JudgeExampleFields":
|
|
1569
|
+
return JudgeExampleFields("examples")
|
|
1570
|
+
|
|
1571
|
+
capabilities: "JudgeGraphQLField" = JudgeGraphQLField("capabilities")
|
|
1169
1572
|
|
|
1170
1573
|
@classmethod
|
|
1171
|
-
def model(cls) ->
|
|
1172
|
-
return ModelFields(
|
|
1173
|
-
|
|
1574
|
+
def model(cls) -> "ModelFields":
|
|
1575
|
+
return ModelFields("model")
|
|
1576
|
+
|
|
1577
|
+
use_case_id: "JudgeGraphQLField" = JudgeGraphQLField("useCaseId")
|
|
1174
1578
|
|
|
1175
1579
|
@classmethod
|
|
1176
|
-
def metric(cls) ->
|
|
1177
|
-
return MetricFields(
|
|
1178
|
-
created_at: 'JudgeGraphQLField' = JudgeGraphQLField('createdAt')
|
|
1179
|
-
updated_at: 'JudgeGraphQLField' = JudgeGraphQLField('updatedAt')
|
|
1580
|
+
def metric(cls) -> "MetricFields":
|
|
1581
|
+
return MetricFields("metric")
|
|
1180
1582
|
|
|
1181
|
-
|
|
1583
|
+
created_at: "JudgeGraphQLField" = JudgeGraphQLField("createdAt")
|
|
1584
|
+
updated_at: "JudgeGraphQLField" = JudgeGraphQLField("updatedAt")
|
|
1585
|
+
|
|
1586
|
+
def fields(
|
|
1587
|
+
self, *subfields: Union[JudgeGraphQLField, "JudgeExampleFields", "MetricFields", "ModelFields"]
|
|
1588
|
+
) -> "JudgeFields":
|
|
1182
1589
|
"""Subfields should come from the JudgeFields class"""
|
|
1183
1590
|
self._subfields.extend(subfields)
|
|
1184
1591
|
return self
|
|
1185
1592
|
|
|
1186
|
-
def alias(self, alias: str) ->
|
|
1593
|
+
def alias(self, alias: str) -> "JudgeFields":
|
|
1187
1594
|
self._alias = alias
|
|
1188
1595
|
return self
|
|
1189
1596
|
|
|
1597
|
+
|
|
1190
1598
|
class JudgeConfigOutputFields(GraphQLField):
|
|
1191
1599
|
"""@private"""
|
|
1192
|
-
|
|
1600
|
+
|
|
1601
|
+
criteria: "JudgeConfigOutputGraphQLField" = JudgeConfigOutputGraphQLField("criteria")
|
|
1193
1602
|
|
|
1194
1603
|
@classmethod
|
|
1195
|
-
def examples(cls) ->
|
|
1196
|
-
return JudgeExampleFields(
|
|
1197
|
-
|
|
1198
|
-
|
|
1604
|
+
def examples(cls) -> "JudgeExampleFields":
|
|
1605
|
+
return JudgeExampleFields("examples")
|
|
1606
|
+
|
|
1607
|
+
system_template: "JudgeConfigOutputGraphQLField" = JudgeConfigOutputGraphQLField("systemTemplate")
|
|
1608
|
+
user_template: "JudgeConfigOutputGraphQLField" = JudgeConfigOutputGraphQLField("userTemplate")
|
|
1199
1609
|
|
|
1200
1610
|
@classmethod
|
|
1201
|
-
def model(cls) ->
|
|
1202
|
-
return ModelFields(
|
|
1611
|
+
def model(cls) -> "ModelFields":
|
|
1612
|
+
return ModelFields("model")
|
|
1203
1613
|
|
|
1204
|
-
def fields(
|
|
1614
|
+
def fields(
|
|
1615
|
+
self, *subfields: Union[JudgeConfigOutputGraphQLField, "JudgeExampleFields", "ModelFields"]
|
|
1616
|
+
) -> "JudgeConfigOutputFields":
|
|
1205
1617
|
"""Subfields should come from the JudgeConfigOutputFields class"""
|
|
1206
1618
|
self._subfields.extend(subfields)
|
|
1207
1619
|
return self
|
|
1208
1620
|
|
|
1209
|
-
def alias(self, alias: str) ->
|
|
1621
|
+
def alias(self, alias: str) -> "JudgeConfigOutputFields":
|
|
1210
1622
|
self._alias = alias
|
|
1211
1623
|
return self
|
|
1212
1624
|
|
|
1625
|
+
|
|
1213
1626
|
class JudgeExampleFields(GraphQLField):
|
|
1214
1627
|
"""@private"""
|
|
1215
1628
|
|
|
1216
1629
|
@classmethod
|
|
1217
|
-
def input(cls) ->
|
|
1218
|
-
return ChatMessageFields(
|
|
1219
|
-
output: 'JudgeExampleGraphQLField' = JudgeExampleGraphQLField('output')
|
|
1220
|
-
pass_: 'JudgeExampleGraphQLField' = JudgeExampleGraphQLField('pass')
|
|
1221
|
-
reasoning: 'JudgeExampleGraphQLField' = JudgeExampleGraphQLField('reasoning')
|
|
1630
|
+
def input(cls) -> "ChatMessageFields":
|
|
1631
|
+
return ChatMessageFields("input")
|
|
1222
1632
|
|
|
1223
|
-
|
|
1633
|
+
output: "JudgeExampleGraphQLField" = JudgeExampleGraphQLField("output")
|
|
1634
|
+
pass_: "JudgeExampleGraphQLField" = JudgeExampleGraphQLField("pass")
|
|
1635
|
+
reasoning: "JudgeExampleGraphQLField" = JudgeExampleGraphQLField("reasoning")
|
|
1636
|
+
|
|
1637
|
+
def fields(self, *subfields: Union[JudgeExampleGraphQLField, "ChatMessageFields"]) -> "JudgeExampleFields":
|
|
1224
1638
|
"""Subfields should come from the JudgeExampleFields class"""
|
|
1225
1639
|
self._subfields.extend(subfields)
|
|
1226
1640
|
return self
|
|
1227
1641
|
|
|
1228
|
-
def alias(self, alias: str) ->
|
|
1642
|
+
def alias(self, alias: str) -> "JudgeExampleFields":
|
|
1229
1643
|
self._alias = alias
|
|
1230
1644
|
return self
|
|
1231
1645
|
|
|
1646
|
+
|
|
1232
1647
|
class LabelFields(GraphQLField):
|
|
1233
1648
|
"""@private"""
|
|
1234
|
-
key: 'LabelGraphQLField' = LabelGraphQLField('key')
|
|
1235
|
-
value: 'LabelGraphQLField' = LabelGraphQLField('value')
|
|
1236
1649
|
|
|
1237
|
-
|
|
1650
|
+
key: "LabelGraphQLField" = LabelGraphQLField("key")
|
|
1651
|
+
value: "LabelGraphQLField" = LabelGraphQLField("value")
|
|
1652
|
+
|
|
1653
|
+
def fields(self, *subfields: LabelGraphQLField) -> "LabelFields":
|
|
1238
1654
|
"""Subfields should come from the LabelFields class"""
|
|
1239
1655
|
self._subfields.extend(subfields)
|
|
1240
1656
|
return self
|
|
1241
1657
|
|
|
1242
|
-
def alias(self, alias: str) ->
|
|
1658
|
+
def alias(self, alias: str) -> "LabelFields":
|
|
1243
1659
|
self._alias = alias
|
|
1244
1660
|
return self
|
|
1245
1661
|
|
|
1662
|
+
|
|
1246
1663
|
class LabelKeyUsageFields(GraphQLField):
|
|
1247
1664
|
"""@private"""
|
|
1248
|
-
|
|
1249
|
-
|
|
1665
|
+
|
|
1666
|
+
key: "LabelKeyUsageGraphQLField" = LabelKeyUsageGraphQLField("key")
|
|
1667
|
+
count: "LabelKeyUsageGraphQLField" = LabelKeyUsageGraphQLField("count")
|
|
1250
1668
|
|
|
1251
1669
|
@classmethod
|
|
1252
|
-
def values(cls) ->
|
|
1253
|
-
return LabelValueUsageFields(
|
|
1254
|
-
last_used: 'LabelKeyUsageGraphQLField' = LabelKeyUsageGraphQLField('lastUsed')
|
|
1670
|
+
def values(cls) -> "LabelValueUsageFields":
|
|
1671
|
+
return LabelValueUsageFields("values")
|
|
1255
1672
|
|
|
1256
|
-
|
|
1673
|
+
last_used: "LabelKeyUsageGraphQLField" = LabelKeyUsageGraphQLField("lastUsed")
|
|
1674
|
+
|
|
1675
|
+
def fields(self, *subfields: Union[LabelKeyUsageGraphQLField, "LabelValueUsageFields"]) -> "LabelKeyUsageFields":
|
|
1257
1676
|
"""Subfields should come from the LabelKeyUsageFields class"""
|
|
1258
1677
|
self._subfields.extend(subfields)
|
|
1259
1678
|
return self
|
|
1260
1679
|
|
|
1261
|
-
def alias(self, alias: str) ->
|
|
1680
|
+
def alias(self, alias: str) -> "LabelKeyUsageFields":
|
|
1262
1681
|
self._alias = alias
|
|
1263
1682
|
return self
|
|
1264
1683
|
|
|
1684
|
+
|
|
1265
1685
|
class LabelUsageFields(GraphQLField):
|
|
1266
1686
|
"""@private"""
|
|
1267
1687
|
|
|
1268
1688
|
@classmethod
|
|
1269
|
-
def keys(cls) ->
|
|
1270
|
-
return LabelKeyUsageFields(
|
|
1689
|
+
def keys(cls) -> "LabelKeyUsageFields":
|
|
1690
|
+
return LabelKeyUsageFields("keys")
|
|
1271
1691
|
|
|
1272
|
-
def fields(self, *subfields: Union[LabelUsageGraphQLField,
|
|
1692
|
+
def fields(self, *subfields: Union[LabelUsageGraphQLField, "LabelKeyUsageFields"]) -> "LabelUsageFields":
|
|
1273
1693
|
"""Subfields should come from the LabelUsageFields class"""
|
|
1274
1694
|
self._subfields.extend(subfields)
|
|
1275
1695
|
return self
|
|
1276
1696
|
|
|
1277
|
-
def alias(self, alias: str) ->
|
|
1697
|
+
def alias(self, alias: str) -> "LabelUsageFields":
|
|
1278
1698
|
self._alias = alias
|
|
1279
1699
|
return self
|
|
1280
1700
|
|
|
1701
|
+
|
|
1281
1702
|
class LabelValueUsageFields(GraphQLField):
|
|
1282
1703
|
"""@private"""
|
|
1283
|
-
value: 'LabelValueUsageGraphQLField' = LabelValueUsageGraphQLField('value')
|
|
1284
|
-
count: 'LabelValueUsageGraphQLField' = LabelValueUsageGraphQLField('count')
|
|
1285
|
-
last_used: 'LabelValueUsageGraphQLField' = LabelValueUsageGraphQLField('lastUsed')
|
|
1286
1704
|
|
|
1287
|
-
|
|
1705
|
+
value: "LabelValueUsageGraphQLField" = LabelValueUsageGraphQLField("value")
|
|
1706
|
+
count: "LabelValueUsageGraphQLField" = LabelValueUsageGraphQLField("count")
|
|
1707
|
+
last_used: "LabelValueUsageGraphQLField" = LabelValueUsageGraphQLField("lastUsed")
|
|
1708
|
+
|
|
1709
|
+
def fields(self, *subfields: LabelValueUsageGraphQLField) -> "LabelValueUsageFields":
|
|
1288
1710
|
"""Subfields should come from the LabelValueUsageFields class"""
|
|
1289
1711
|
self._subfields.extend(subfields)
|
|
1290
1712
|
return self
|
|
1291
1713
|
|
|
1292
|
-
def alias(self, alias: str) ->
|
|
1714
|
+
def alias(self, alias: str) -> "LabelValueUsageFields":
|
|
1293
1715
|
self._alias = alias
|
|
1294
1716
|
return self
|
|
1295
1717
|
|
|
1718
|
+
|
|
1296
1719
|
class MetaObjectFields(GraphQLField):
|
|
1297
1720
|
"""@private"""
|
|
1298
1721
|
|
|
1299
1722
|
@classmethod
|
|
1300
|
-
def auth_providers(cls) ->
|
|
1301
|
-
return ProviderListFields(
|
|
1723
|
+
def auth_providers(cls) -> "ProviderListFields":
|
|
1724
|
+
return ProviderListFields("authProviders")
|
|
1302
1725
|
|
|
1303
|
-
def fields(self, *subfields: Union[MetaObjectGraphQLField,
|
|
1726
|
+
def fields(self, *subfields: Union[MetaObjectGraphQLField, "ProviderListFields"]) -> "MetaObjectFields":
|
|
1304
1727
|
"""Subfields should come from the MetaObjectFields class"""
|
|
1305
1728
|
self._subfields.extend(subfields)
|
|
1306
1729
|
return self
|
|
1307
1730
|
|
|
1308
|
-
def alias(self, alias: str) ->
|
|
1731
|
+
def alias(self, alias: str) -> "MetaObjectFields":
|
|
1309
1732
|
self._alias = alias
|
|
1310
1733
|
return self
|
|
1311
1734
|
|
|
1735
|
+
|
|
1312
1736
|
class MetricFields(GraphQLField):
|
|
1313
1737
|
"""@private"""
|
|
1314
|
-
|
|
1315
|
-
|
|
1316
|
-
|
|
1317
|
-
|
|
1318
|
-
|
|
1319
|
-
|
|
1320
|
-
|
|
1321
|
-
|
|
1738
|
+
|
|
1739
|
+
id: "MetricGraphQLField" = MetricGraphQLField("id")
|
|
1740
|
+
key: "MetricGraphQLField" = MetricGraphQLField("key")
|
|
1741
|
+
name: "MetricGraphQLField" = MetricGraphQLField("name")
|
|
1742
|
+
created_at: "MetricGraphQLField" = MetricGraphQLField("createdAt")
|
|
1743
|
+
kind: "MetricGraphQLField" = MetricGraphQLField("kind")
|
|
1744
|
+
description: "MetricGraphQLField" = MetricGraphQLField("description")
|
|
1745
|
+
scoring_type: "MetricGraphQLField" = MetricGraphQLField("scoringType")
|
|
1746
|
+
unit: "MetricGraphQLField" = MetricGraphQLField("unit")
|
|
1322
1747
|
|
|
1323
1748
|
@classmethod
|
|
1324
|
-
def use_cases(cls, filter: UseCaseFilter) ->
|
|
1749
|
+
def use_cases(cls, filter: UseCaseFilter) -> "UseCaseFields":
|
|
1325
1750
|
"""Return the list of UseCase which use this metric"""
|
|
1326
|
-
arguments: Dict[str, Dict[str, Any]] = {
|
|
1327
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
1328
|
-
return UseCaseFields(
|
|
1751
|
+
arguments: Dict[str, Dict[str, Any]] = {"filter": {"type": "UseCaseFilter!", "value": filter}}
|
|
1752
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
|
|
1753
|
+
return UseCaseFields("useCases", arguments=cleared_arguments)
|
|
1329
1754
|
|
|
1330
1755
|
@classmethod
|
|
1331
|
-
def activity(cls, *, timerange: Optional[TimeRange]=None) ->
|
|
1332
|
-
arguments: Dict[str, Dict[str, Any]] = {
|
|
1333
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
1334
|
-
return MetricActivityFields(
|
|
1335
|
-
|
|
1336
|
-
|
|
1756
|
+
def activity(cls, *, timerange: Optional[TimeRange] = None) -> "MetricActivityFields":
|
|
1757
|
+
arguments: Dict[str, Dict[str, Any]] = {"timerange": {"type": "TimeRange", "value": timerange}}
|
|
1758
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
|
|
1759
|
+
return MetricActivityFields("activity", arguments=cleared_arguments)
|
|
1760
|
+
|
|
1761
|
+
has_direct_feedbacks: "MetricGraphQLField" = MetricGraphQLField("hasDirectFeedbacks")
|
|
1762
|
+
has_comparison_feedbacks: "MetricGraphQLField" = MetricGraphQLField("hasComparisonFeedbacks")
|
|
1337
1763
|
|
|
1338
|
-
def fields(self, *subfields: Union[MetricGraphQLField,
|
|
1764
|
+
def fields(self, *subfields: Union[MetricGraphQLField, "MetricActivityFields", "UseCaseFields"]) -> "MetricFields":
|
|
1339
1765
|
"""Subfields should come from the MetricFields class"""
|
|
1340
1766
|
self._subfields.extend(subfields)
|
|
1341
1767
|
return self
|
|
1342
1768
|
|
|
1343
|
-
def alias(self, alias: str) ->
|
|
1769
|
+
def alias(self, alias: str) -> "MetricFields":
|
|
1344
1770
|
self._alias = alias
|
|
1345
1771
|
return self
|
|
1346
1772
|
|
|
1773
|
+
|
|
1347
1774
|
class MetricActivityFields(GraphQLField):
|
|
1348
1775
|
"""@private"""
|
|
1349
1776
|
|
|
1350
1777
|
@classmethod
|
|
1351
|
-
def feedbacks(cls) ->
|
|
1352
|
-
return ActivityOutputFields(
|
|
1778
|
+
def feedbacks(cls) -> "ActivityOutputFields":
|
|
1779
|
+
return ActivityOutputFields("feedbacks")
|
|
1353
1780
|
|
|
1354
|
-
def fields(self, *subfields: Union[MetricActivityGraphQLField,
|
|
1781
|
+
def fields(self, *subfields: Union[MetricActivityGraphQLField, "ActivityOutputFields"]) -> "MetricActivityFields":
|
|
1355
1782
|
"""Subfields should come from the MetricActivityFields class"""
|
|
1356
1783
|
self._subfields.extend(subfields)
|
|
1357
1784
|
return self
|
|
1358
1785
|
|
|
1359
|
-
def alias(self, alias: str) ->
|
|
1786
|
+
def alias(self, alias: str) -> "MetricActivityFields":
|
|
1360
1787
|
self._alias = alias
|
|
1361
1788
|
return self
|
|
1362
1789
|
|
|
1790
|
+
|
|
1363
1791
|
class MetricWithContextFields(GraphQLField):
|
|
1364
1792
|
"""@private"""
|
|
1365
|
-
|
|
1366
|
-
|
|
1367
|
-
|
|
1368
|
-
|
|
1369
|
-
|
|
1370
|
-
|
|
1371
|
-
|
|
1372
|
-
|
|
1793
|
+
|
|
1794
|
+
id: "MetricWithContextGraphQLField" = MetricWithContextGraphQLField("id")
|
|
1795
|
+
key: "MetricWithContextGraphQLField" = MetricWithContextGraphQLField("key")
|
|
1796
|
+
name: "MetricWithContextGraphQLField" = MetricWithContextGraphQLField("name")
|
|
1797
|
+
kind: "MetricWithContextGraphQLField" = MetricWithContextGraphQLField("kind")
|
|
1798
|
+
scoring_type: "MetricWithContextGraphQLField" = MetricWithContextGraphQLField("scoringType")
|
|
1799
|
+
description: "MetricWithContextGraphQLField" = MetricWithContextGraphQLField("description")
|
|
1800
|
+
created_at: "MetricWithContextGraphQLField" = MetricWithContextGraphQLField("createdAt")
|
|
1801
|
+
unit: "MetricWithContextGraphQLField" = MetricWithContextGraphQLField("unit")
|
|
1373
1802
|
|
|
1374
1803
|
@classmethod
|
|
1375
|
-
def feedback_count(cls, *, timerange: Optional[TimeRange]=None) ->
|
|
1376
|
-
arguments: Dict[str, Dict[str, Any]] = {
|
|
1377
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
1378
|
-
return MetricWithContextGraphQLField(
|
|
1804
|
+
def feedback_count(cls, *, timerange: Optional[TimeRange] = None) -> "MetricWithContextGraphQLField":
|
|
1805
|
+
arguments: Dict[str, Dict[str, Any]] = {"timerange": {"type": "TimeRange", "value": timerange}}
|
|
1806
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
|
|
1807
|
+
return MetricWithContextGraphQLField("feedbackCount", arguments=cleared_arguments)
|
|
1379
1808
|
|
|
1380
1809
|
@classmethod
|
|
1381
|
-
def comparison_count(cls, *, timerange: Optional[TimeRange]=None) ->
|
|
1382
|
-
arguments: Dict[str, Dict[str, Any]] = {
|
|
1383
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
1384
|
-
return MetricWithContextGraphQLField(
|
|
1810
|
+
def comparison_count(cls, *, timerange: Optional[TimeRange] = None) -> "MetricWithContextGraphQLField":
|
|
1811
|
+
arguments: Dict[str, Dict[str, Any]] = {"timerange": {"type": "TimeRange", "value": timerange}}
|
|
1812
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
|
|
1813
|
+
return MetricWithContextGraphQLField("comparisonCount", arguments=cleared_arguments)
|
|
1385
1814
|
|
|
1386
1815
|
@classmethod
|
|
1387
|
-
def trend(cls, input: MetricTrendInput) ->
|
|
1388
|
-
arguments: Dict[str, Dict[str, Any]] = {
|
|
1389
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
1390
|
-
return TrendResultFields(
|
|
1816
|
+
def trend(cls, input: MetricTrendInput) -> "TrendResultFields":
|
|
1817
|
+
arguments: Dict[str, Dict[str, Any]] = {"input": {"type": "MetricTrendInput!", "value": input}}
|
|
1818
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
|
|
1819
|
+
return TrendResultFields("trend", arguments=cleared_arguments)
|
|
1391
1820
|
|
|
1392
1821
|
@classmethod
|
|
1393
|
-
def timeseries(cls, input: TimeseriesInput) ->
|
|
1394
|
-
arguments: Dict[str, Dict[str, Any]] = {
|
|
1395
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
1396
|
-
return TimeseriesFields(
|
|
1397
|
-
has_comparison_feedbacks: 'MetricWithContextGraphQLField' = MetricWithContextGraphQLField('hasComparisonFeedbacks')
|
|
1398
|
-
has_direct_feedbacks: 'MetricWithContextGraphQLField' = MetricWithContextGraphQLField('hasDirectFeedbacks')
|
|
1822
|
+
def timeseries(cls, input: TimeseriesInput) -> "TimeseriesFields":
|
|
1823
|
+
arguments: Dict[str, Dict[str, Any]] = {"input": {"type": "TimeseriesInput!", "value": input}}
|
|
1824
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
|
|
1825
|
+
return TimeseriesFields("timeseries", arguments=cleared_arguments)
|
|
1399
1826
|
|
|
1400
|
-
|
|
1827
|
+
has_comparison_feedbacks: "MetricWithContextGraphQLField" = MetricWithContextGraphQLField("hasComparisonFeedbacks")
|
|
1828
|
+
has_direct_feedbacks: "MetricWithContextGraphQLField" = MetricWithContextGraphQLField("hasDirectFeedbacks")
|
|
1829
|
+
|
|
1830
|
+
def fields(
|
|
1831
|
+
self, *subfields: Union[MetricWithContextGraphQLField, "TimeseriesFields", "TrendResultFields"]
|
|
1832
|
+
) -> "MetricWithContextFields":
|
|
1401
1833
|
"""Subfields should come from the MetricWithContextFields class"""
|
|
1402
1834
|
self._subfields.extend(subfields)
|
|
1403
1835
|
return self
|
|
1404
1836
|
|
|
1405
|
-
def alias(self, alias: str) ->
|
|
1837
|
+
def alias(self, alias: str) -> "MetricWithContextFields":
|
|
1406
1838
|
self._alias = alias
|
|
1407
1839
|
return self
|
|
1408
1840
|
|
|
1841
|
+
|
|
1409
1842
|
class ModelFields(GraphQLField):
|
|
1410
1843
|
"""@private"""
|
|
1411
|
-
id: 'ModelGraphQLField' = ModelGraphQLField('id')
|
|
1412
|
-
name: 'ModelGraphQLField' = ModelGraphQLField('name')
|
|
1413
|
-
key: 'ModelGraphQLField' = ModelGraphQLField('key')
|
|
1414
|
-
created_at: 'ModelGraphQLField' = ModelGraphQLField('createdAt')
|
|
1415
|
-
online: 'ModelGraphQLField' = ModelGraphQLField('online')
|
|
1416
|
-
'indicates if this model is spawned in mangrove or not'
|
|
1417
|
-
error: 'ModelGraphQLField' = ModelGraphQLField('error')
|
|
1418
|
-
|
|
1419
|
-
@classmethod
|
|
1420
|
-
def activity(cls, *, timerange: Optional[TimeRange]=None) -> 'ActivityFields':
|
|
1421
|
-
arguments: Dict[str, Dict[str, Any]] = {'timerange': {'type': 'TimeRange', 'value': timerange}}
|
|
1422
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
|
|
1423
|
-
return ActivityFields('activity', arguments=cleared_arguments)
|
|
1424
|
-
|
|
1425
|
-
@classmethod
|
|
1426
|
-
def metrics(cls) -> 'MetricWithContextFields':
|
|
1427
|
-
return MetricWithContextFields('metrics')
|
|
1428
|
-
|
|
1429
|
-
@classmethod
|
|
1430
|
-
def use_cases(cls, filter: UseCaseFilter) -> 'UseCaseFields':
|
|
1431
|
-
"""Return the list of UseCase which use this model"""
|
|
1432
|
-
arguments: Dict[str, Dict[str, Any]] = {'filter': {'type': 'UseCaseFilter!', 'value': filter}}
|
|
1433
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
|
|
1434
|
-
return UseCaseFields('useCases', arguments=cleared_arguments)
|
|
1435
1844
|
|
|
1436
|
-
|
|
1437
|
-
|
|
1438
|
-
|
|
1439
|
-
|
|
1440
|
-
|
|
1441
|
-
|
|
1442
|
-
|
|
1443
|
-
order: 'ModelGraphQLField' = ModelGraphQLField('order')
|
|
1444
|
-
in_storage: 'ModelGraphQLField' = ModelGraphQLField('inStorage')
|
|
1445
|
-
is_adapter: 'ModelGraphQLField' = ModelGraphQLField('isAdapter')
|
|
1845
|
+
id: "ModelGraphQLField" = ModelGraphQLField("id")
|
|
1846
|
+
name: "ModelGraphQLField" = ModelGraphQLField("name")
|
|
1847
|
+
key: "ModelGraphQLField" = ModelGraphQLField("key")
|
|
1848
|
+
created_at: "ModelGraphQLField" = ModelGraphQLField("createdAt")
|
|
1849
|
+
online: "ModelGraphQLField" = ModelGraphQLField("online")
|
|
1850
|
+
"indicates if this model is spawned in mangrove or not"
|
|
1851
|
+
error: "ModelGraphQLField" = ModelGraphQLField("error")
|
|
1446
1852
|
|
|
1447
1853
|
@classmethod
|
|
1448
|
-
def
|
|
1449
|
-
|
|
1854
|
+
def activity(cls, *, timerange: Optional[TimeRange] = None) -> "ActivityFields":
|
|
1855
|
+
arguments: Dict[str, Dict[str, Any]] = {"timerange": {"type": "TimeRange", "value": timerange}}
|
|
1856
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
|
|
1857
|
+
return ActivityFields("activity", arguments=cleared_arguments)
|
|
1450
1858
|
|
|
1451
1859
|
@classmethod
|
|
1452
|
-
def
|
|
1453
|
-
return
|
|
1454
|
-
is_training: 'ModelGraphQLField' = ModelGraphQLField('isTraining')
|
|
1455
|
-
'indicates if a training is pending or running for this model'
|
|
1456
|
-
is_published: 'ModelGraphQLField' = ModelGraphQLField('isPublished')
|
|
1457
|
-
is_stable: 'ModelGraphQLField' = ModelGraphQLField('isStable')
|
|
1458
|
-
capabilities: 'ModelGraphQLField' = ModelGraphQLField('capabilities')
|
|
1459
|
-
supported_tp: 'ModelGraphQLField' = ModelGraphQLField('supportedTp')
|
|
1460
|
-
family: 'ModelGraphQLField' = ModelGraphQLField('family')
|
|
1461
|
-
publisher: 'ModelGraphQLField' = ModelGraphQLField('publisher')
|
|
1462
|
-
size: 'ModelGraphQLField' = ModelGraphQLField('size')
|
|
1860
|
+
def metrics(cls) -> "MetricWithContextFields":
|
|
1861
|
+
return MetricWithContextFields("metrics")
|
|
1463
1862
|
|
|
1464
1863
|
@classmethod
|
|
1465
|
-
def
|
|
1466
|
-
|
|
1467
|
-
|
|
1468
|
-
|
|
1864
|
+
def use_cases(cls, filter: UseCaseFilter) -> "UseCaseFields":
|
|
1865
|
+
"""Return the list of UseCase which use this model"""
|
|
1866
|
+
arguments: Dict[str, Dict[str, Any]] = {"filter": {"type": "UseCaseFilter!", "value": filter}}
|
|
1867
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
|
|
1868
|
+
return UseCaseFields("useCases", arguments=cleared_arguments)
|
|
1869
|
+
|
|
1870
|
+
@classmethod
|
|
1871
|
+
def model_services(cls, filter: UseCaseFilter) -> "ModelServiceFields":
|
|
1872
|
+
arguments: Dict[str, Dict[str, Any]] = {"filter": {"type": "UseCaseFilter!", "value": filter}}
|
|
1873
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
|
|
1874
|
+
return ModelServiceFields("modelServices", arguments=cleared_arguments)
|
|
1875
|
+
|
|
1876
|
+
provider_name: "ModelGraphQLField" = ModelGraphQLField("providerName")
|
|
1877
|
+
is_external: "ModelGraphQLField" = ModelGraphQLField("isExternal")
|
|
1878
|
+
order: "ModelGraphQLField" = ModelGraphQLField("order")
|
|
1879
|
+
in_storage: "ModelGraphQLField" = ModelGraphQLField("inStorage")
|
|
1880
|
+
is_adapter: "ModelGraphQLField" = ModelGraphQLField("isAdapter")
|
|
1881
|
+
|
|
1882
|
+
@classmethod
|
|
1883
|
+
def backbone(cls) -> "ModelFields":
|
|
1884
|
+
return ModelFields("backbone")
|
|
1885
|
+
|
|
1886
|
+
@classmethod
|
|
1887
|
+
def parent(cls) -> "ModelFields":
|
|
1888
|
+
return ModelFields("parent")
|
|
1889
|
+
|
|
1890
|
+
is_training: "ModelGraphQLField" = ModelGraphQLField("isTraining")
|
|
1891
|
+
"indicates if a training is pending or running for this model"
|
|
1892
|
+
is_published: "ModelGraphQLField" = ModelGraphQLField("isPublished")
|
|
1893
|
+
is_stable: "ModelGraphQLField" = ModelGraphQLField("isStable")
|
|
1894
|
+
capabilities: "ModelGraphQLField" = ModelGraphQLField("capabilities")
|
|
1895
|
+
supported_tp: "ModelGraphQLField" = ModelGraphQLField("supportedTp")
|
|
1896
|
+
family: "ModelGraphQLField" = ModelGraphQLField("family")
|
|
1897
|
+
publisher: "ModelGraphQLField" = ModelGraphQLField("publisher")
|
|
1898
|
+
size: "ModelGraphQLField" = ModelGraphQLField("size")
|
|
1899
|
+
|
|
1900
|
+
@classmethod
|
|
1901
|
+
def compute_config(cls) -> "ModelComputeConfigOutputFields":
|
|
1902
|
+
return ModelComputeConfigOutputFields("computeConfig")
|
|
1903
|
+
|
|
1904
|
+
def fields(
|
|
1905
|
+
self,
|
|
1906
|
+
*subfields: Union[
|
|
1907
|
+
ModelGraphQLField,
|
|
1908
|
+
"ActivityFields",
|
|
1909
|
+
"MetricWithContextFields",
|
|
1910
|
+
"ModelComputeConfigOutputFields",
|
|
1911
|
+
"ModelFields",
|
|
1912
|
+
"ModelServiceFields",
|
|
1913
|
+
"UseCaseFields",
|
|
1914
|
+
],
|
|
1915
|
+
) -> "ModelFields":
|
|
1469
1916
|
"""Subfields should come from the ModelFields class"""
|
|
1470
1917
|
self._subfields.extend(subfields)
|
|
1471
1918
|
return self
|
|
1472
1919
|
|
|
1473
|
-
def alias(self, alias: str) ->
|
|
1920
|
+
def alias(self, alias: str) -> "ModelFields":
|
|
1474
1921
|
self._alias = alias
|
|
1475
1922
|
return self
|
|
1476
1923
|
|
|
1924
|
+
|
|
1477
1925
|
class ModelByproductsFields(GraphQLField):
|
|
1478
1926
|
"""@private"""
|
|
1479
1927
|
|
|
1480
1928
|
@classmethod
|
|
1481
|
-
def model(cls) ->
|
|
1482
|
-
return ModelFields(
|
|
1929
|
+
def model(cls) -> "ModelFields":
|
|
1930
|
+
return ModelFields("model")
|
|
1483
1931
|
|
|
1484
|
-
def fields(self, *subfields: Union[ModelByproductsGraphQLField,
|
|
1932
|
+
def fields(self, *subfields: Union[ModelByproductsGraphQLField, "ModelFields"]) -> "ModelByproductsFields":
|
|
1485
1933
|
"""Subfields should come from the ModelByproductsFields class"""
|
|
1486
1934
|
self._subfields.extend(subfields)
|
|
1487
1935
|
return self
|
|
1488
1936
|
|
|
1489
|
-
def alias(self, alias: str) ->
|
|
1937
|
+
def alias(self, alias: str) -> "ModelByproductsFields":
|
|
1490
1938
|
self._alias = alias
|
|
1491
1939
|
return self
|
|
1492
1940
|
|
|
1941
|
+
|
|
1493
1942
|
class ModelComputeConfigOutputFields(GraphQLField):
|
|
1494
1943
|
"""@private"""
|
|
1495
|
-
tp: 'ModelComputeConfigOutputGraphQLField' = ModelComputeConfigOutputGraphQLField('tp')
|
|
1496
|
-
kv_cache_len: 'ModelComputeConfigOutputGraphQLField' = ModelComputeConfigOutputGraphQLField('kvCacheLen')
|
|
1497
|
-
max_seq_len: 'ModelComputeConfigOutputGraphQLField' = ModelComputeConfigOutputGraphQLField('maxSeqLen')
|
|
1498
1944
|
|
|
1499
|
-
|
|
1945
|
+
tp: "ModelComputeConfigOutputGraphQLField" = ModelComputeConfigOutputGraphQLField("tp")
|
|
1946
|
+
kv_cache_len: "ModelComputeConfigOutputGraphQLField" = ModelComputeConfigOutputGraphQLField("kvCacheLen")
|
|
1947
|
+
max_seq_len: "ModelComputeConfigOutputGraphQLField" = ModelComputeConfigOutputGraphQLField("maxSeqLen")
|
|
1948
|
+
|
|
1949
|
+
def fields(self, *subfields: ModelComputeConfigOutputGraphQLField) -> "ModelComputeConfigOutputFields":
|
|
1500
1950
|
"""Subfields should come from the ModelComputeConfigOutputFields class"""
|
|
1501
1951
|
self._subfields.extend(subfields)
|
|
1502
1952
|
return self
|
|
1503
1953
|
|
|
1504
|
-
def alias(self, alias: str) ->
|
|
1954
|
+
def alias(self, alias: str) -> "ModelComputeConfigOutputFields":
|
|
1505
1955
|
self._alias = alias
|
|
1506
1956
|
return self
|
|
1507
1957
|
|
|
1958
|
+
|
|
1508
1959
|
class ModelPlacementOutputFields(GraphQLField):
|
|
1509
1960
|
"""@private"""
|
|
1510
|
-
compute_pools: 'ModelPlacementOutputGraphQLField' = ModelPlacementOutputGraphQLField('computePools')
|
|
1511
|
-
max_ttft_ms: 'ModelPlacementOutputGraphQLField' = ModelPlacementOutputGraphQLField('maxTtftMs')
|
|
1512
1961
|
|
|
1513
|
-
|
|
1962
|
+
compute_pools: "ModelPlacementOutputGraphQLField" = ModelPlacementOutputGraphQLField("computePools")
|
|
1963
|
+
max_ttft_ms: "ModelPlacementOutputGraphQLField" = ModelPlacementOutputGraphQLField("maxTtftMs")
|
|
1964
|
+
|
|
1965
|
+
def fields(self, *subfields: ModelPlacementOutputGraphQLField) -> "ModelPlacementOutputFields":
|
|
1514
1966
|
"""Subfields should come from the ModelPlacementOutputFields class"""
|
|
1515
1967
|
self._subfields.extend(subfields)
|
|
1516
1968
|
return self
|
|
1517
1969
|
|
|
1518
|
-
def alias(self, alias: str) ->
|
|
1970
|
+
def alias(self, alias: str) -> "ModelPlacementOutputFields":
|
|
1519
1971
|
self._alias = alias
|
|
1520
1972
|
return self
|
|
1521
1973
|
|
|
1974
|
+
|
|
1522
1975
|
class ModelServiceFields(GraphQLField):
|
|
1523
1976
|
"""@private"""
|
|
1524
|
-
|
|
1525
|
-
|
|
1526
|
-
|
|
1527
|
-
|
|
1528
|
-
|
|
1529
|
-
|
|
1530
|
-
|
|
1977
|
+
|
|
1978
|
+
status: "ModelServiceGraphQLField" = ModelServiceGraphQLField("status")
|
|
1979
|
+
error: "ModelServiceGraphQLField" = ModelServiceGraphQLField("error")
|
|
1980
|
+
id: "ModelServiceGraphQLField" = ModelServiceGraphQLField("id")
|
|
1981
|
+
use_case_id: "ModelServiceGraphQLField" = ModelServiceGraphQLField("useCaseId")
|
|
1982
|
+
key: "ModelServiceGraphQLField" = ModelServiceGraphQLField("key")
|
|
1983
|
+
name: "ModelServiceGraphQLField" = ModelServiceGraphQLField("name")
|
|
1984
|
+
created_at: "ModelServiceGraphQLField" = ModelServiceGraphQLField("createdAt")
|
|
1531
1985
|
|
|
1532
1986
|
@classmethod
|
|
1533
|
-
def model(cls) ->
|
|
1534
|
-
return ModelFields(
|
|
1535
|
-
|
|
1536
|
-
|
|
1537
|
-
|
|
1538
|
-
|
|
1987
|
+
def model(cls) -> "ModelFields":
|
|
1988
|
+
return ModelFields("model")
|
|
1989
|
+
|
|
1990
|
+
is_default: "ModelServiceGraphQLField" = ModelServiceGraphQLField("isDefault")
|
|
1991
|
+
desired_online: "ModelServiceGraphQLField" = ModelServiceGraphQLField("desiredOnline")
|
|
1992
|
+
deleted: "ModelServiceGraphQLField" = ModelServiceGraphQLField("deleted")
|
|
1993
|
+
"Whether or not this model service has been deleted."
|
|
1539
1994
|
|
|
1540
1995
|
@classmethod
|
|
1541
|
-
def activity(cls, *, timerange: Optional[TimeRange]=None) ->
|
|
1542
|
-
arguments: Dict[str, Dict[str, Any]] = {
|
|
1543
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
1544
|
-
return ActivityFields(
|
|
1996
|
+
def activity(cls, *, timerange: Optional[TimeRange] = None) -> "ActivityFields":
|
|
1997
|
+
arguments: Dict[str, Dict[str, Any]] = {"timerange": {"type": "TimeRange", "value": timerange}}
|
|
1998
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
|
|
1999
|
+
return ActivityFields("activity", arguments=cleared_arguments)
|
|
1545
2000
|
|
|
1546
2001
|
@classmethod
|
|
1547
|
-
def system_prompt_template(cls) ->
|
|
1548
|
-
return SystemPromptTemplateFields(
|
|
2002
|
+
def system_prompt_template(cls) -> "SystemPromptTemplateFields":
|
|
2003
|
+
return SystemPromptTemplateFields("systemPromptTemplate")
|
|
1549
2004
|
|
|
1550
2005
|
@classmethod
|
|
1551
|
-
def metrics(cls) ->
|
|
1552
|
-
return MetricWithContextFields(
|
|
2006
|
+
def metrics(cls) -> "MetricWithContextFields":
|
|
2007
|
+
return MetricWithContextFields("metrics")
|
|
1553
2008
|
|
|
1554
2009
|
@classmethod
|
|
1555
|
-
def ab_campaigns(cls, filter: AbCampaignFilter) ->
|
|
1556
|
-
arguments: Dict[str, Dict[str, Any]] = {
|
|
1557
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
1558
|
-
return AbcampaignFields(
|
|
2010
|
+
def ab_campaigns(cls, filter: AbCampaignFilter) -> "AbcampaignFields":
|
|
2011
|
+
arguments: Dict[str, Dict[str, Any]] = {"filter": {"type": "AbCampaignFilter!", "value": filter}}
|
|
2012
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
|
|
2013
|
+
return AbcampaignFields("abCampaigns", arguments=cleared_arguments)
|
|
1559
2014
|
|
|
1560
2015
|
@classmethod
|
|
1561
|
-
def placement(cls) ->
|
|
1562
|
-
return ModelPlacementOutputFields(
|
|
2016
|
+
def placement(cls) -> "ModelPlacementOutputFields":
|
|
2017
|
+
return ModelPlacementOutputFields("placement")
|
|
1563
2018
|
|
|
1564
2019
|
@classmethod
|
|
1565
|
-
def tool_providers(cls) ->
|
|
1566
|
-
return ToolProviderFields(
|
|
2020
|
+
def tool_providers(cls) -> "ToolProviderFields":
|
|
2021
|
+
return ToolProviderFields("toolProviders")
|
|
1567
2022
|
|
|
1568
|
-
def fields(
|
|
2023
|
+
def fields(
|
|
2024
|
+
self,
|
|
2025
|
+
*subfields: Union[
|
|
2026
|
+
ModelServiceGraphQLField,
|
|
2027
|
+
"AbcampaignFields",
|
|
2028
|
+
"ActivityFields",
|
|
2029
|
+
"MetricWithContextFields",
|
|
2030
|
+
"ModelFields",
|
|
2031
|
+
"ModelPlacementOutputFields",
|
|
2032
|
+
"SystemPromptTemplateFields",
|
|
2033
|
+
"ToolProviderFields",
|
|
2034
|
+
],
|
|
2035
|
+
) -> "ModelServiceFields":
|
|
1569
2036
|
"""Subfields should come from the ModelServiceFields class"""
|
|
1570
2037
|
self._subfields.extend(subfields)
|
|
1571
2038
|
return self
|
|
1572
2039
|
|
|
1573
|
-
def alias(self, alias: str) ->
|
|
2040
|
+
def alias(self, alias: str) -> "ModelServiceFields":
|
|
2041
|
+
self._alias = alias
|
|
2042
|
+
return self
|
|
2043
|
+
|
|
2044
|
+
|
|
2045
|
+
class NotificationSettingsFields(GraphQLField):
|
|
2046
|
+
"""@private"""
|
|
2047
|
+
|
|
2048
|
+
@classmethod
|
|
2049
|
+
def subscriptions(cls) -> "SubscriptionFields":
|
|
2050
|
+
return SubscriptionFields("subscriptions")
|
|
2051
|
+
|
|
2052
|
+
def fields(
|
|
2053
|
+
self, *subfields: Union[NotificationSettingsGraphQLField, "SubscriptionFields"]
|
|
2054
|
+
) -> "NotificationSettingsFields":
|
|
2055
|
+
"""Subfields should come from the NotificationSettingsFields class"""
|
|
2056
|
+
self._subfields.extend(subfields)
|
|
2057
|
+
return self
|
|
2058
|
+
|
|
2059
|
+
def alias(self, alias: str) -> "NotificationSettingsFields":
|
|
1574
2060
|
self._alias = alias
|
|
1575
2061
|
return self
|
|
1576
2062
|
|
|
2063
|
+
|
|
1577
2064
|
class PageInfoFields(GraphQLField):
|
|
1578
2065
|
"""@private
|
|
1579
|
-
Information about pagination in a connection"""
|
|
1580
|
-
|
|
1581
|
-
|
|
1582
|
-
|
|
1583
|
-
|
|
1584
|
-
|
|
1585
|
-
|
|
1586
|
-
|
|
1587
|
-
|
|
1588
|
-
|
|
1589
|
-
|
|
2066
|
+
Information about pagination in a connection"""
|
|
2067
|
+
|
|
2068
|
+
has_previous_page: "PageInfoGraphQLField" = PageInfoGraphQLField("hasPreviousPage")
|
|
2069
|
+
"When paginating backwards, are there more items?"
|
|
2070
|
+
has_next_page: "PageInfoGraphQLField" = PageInfoGraphQLField("hasNextPage")
|
|
2071
|
+
"When paginating forwards, are there more items?"
|
|
2072
|
+
start_cursor: "PageInfoGraphQLField" = PageInfoGraphQLField("startCursor")
|
|
2073
|
+
"When paginating backwards, the cursor to continue."
|
|
2074
|
+
end_cursor: "PageInfoGraphQLField" = PageInfoGraphQLField("endCursor")
|
|
2075
|
+
"When paginating forwards, the cursor to continue."
|
|
2076
|
+
|
|
2077
|
+
def fields(self, *subfields: PageInfoGraphQLField) -> "PageInfoFields":
|
|
1590
2078
|
"""Subfields should come from the PageInfoFields class"""
|
|
1591
2079
|
self._subfields.extend(subfields)
|
|
1592
2080
|
return self
|
|
1593
2081
|
|
|
1594
|
-
def alias(self, alias: str) ->
|
|
2082
|
+
def alias(self, alias: str) -> "PageInfoFields":
|
|
1595
2083
|
self._alias = alias
|
|
1596
2084
|
return self
|
|
1597
2085
|
|
|
2086
|
+
|
|
1598
2087
|
class PrebuiltConfigDefinitionFields(GraphQLField):
|
|
1599
2088
|
"""@private"""
|
|
1600
|
-
key: 'PrebuiltConfigDefinitionGraphQLField' = PrebuiltConfigDefinitionGraphQLField('key')
|
|
1601
|
-
name: 'PrebuiltConfigDefinitionGraphQLField' = PrebuiltConfigDefinitionGraphQLField('name')
|
|
1602
|
-
feedback_key: 'PrebuiltConfigDefinitionGraphQLField' = PrebuiltConfigDefinitionGraphQLField('feedbackKey')
|
|
1603
|
-
description: 'PrebuiltConfigDefinitionGraphQLField' = PrebuiltConfigDefinitionGraphQLField('description')
|
|
1604
2089
|
|
|
1605
|
-
|
|
2090
|
+
key: "PrebuiltConfigDefinitionGraphQLField" = PrebuiltConfigDefinitionGraphQLField("key")
|
|
2091
|
+
name: "PrebuiltConfigDefinitionGraphQLField" = PrebuiltConfigDefinitionGraphQLField("name")
|
|
2092
|
+
feedback_key: "PrebuiltConfigDefinitionGraphQLField" = PrebuiltConfigDefinitionGraphQLField("feedbackKey")
|
|
2093
|
+
description: "PrebuiltConfigDefinitionGraphQLField" = PrebuiltConfigDefinitionGraphQLField("description")
|
|
2094
|
+
|
|
2095
|
+
def fields(self, *subfields: PrebuiltConfigDefinitionGraphQLField) -> "PrebuiltConfigDefinitionFields":
|
|
1606
2096
|
"""Subfields should come from the PrebuiltConfigDefinitionFields class"""
|
|
1607
2097
|
self._subfields.extend(subfields)
|
|
1608
2098
|
return self
|
|
1609
2099
|
|
|
1610
|
-
def alias(self, alias: str) ->
|
|
2100
|
+
def alias(self, alias: str) -> "PrebuiltConfigDefinitionFields":
|
|
1611
2101
|
self._alias = alias
|
|
1612
2102
|
return self
|
|
1613
2103
|
|
|
2104
|
+
|
|
1614
2105
|
class PrebuiltConfigOutputFields(GraphQLField):
|
|
1615
2106
|
"""@private"""
|
|
1616
2107
|
|
|
1617
2108
|
@classmethod
|
|
1618
|
-
def criteria(cls) ->
|
|
1619
|
-
return PrebuiltConfigDefinitionFields(
|
|
2109
|
+
def criteria(cls) -> "PrebuiltConfigDefinitionFields":
|
|
2110
|
+
return PrebuiltConfigDefinitionFields("criteria")
|
|
1620
2111
|
|
|
1621
2112
|
@classmethod
|
|
1622
|
-
def model(cls) ->
|
|
1623
|
-
return ModelFields(
|
|
2113
|
+
def model(cls) -> "ModelFields":
|
|
2114
|
+
return ModelFields("model")
|
|
1624
2115
|
|
|
1625
|
-
def fields(
|
|
2116
|
+
def fields(
|
|
2117
|
+
self, *subfields: Union[PrebuiltConfigOutputGraphQLField, "ModelFields", "PrebuiltConfigDefinitionFields"]
|
|
2118
|
+
) -> "PrebuiltConfigOutputFields":
|
|
1626
2119
|
"""Subfields should come from the PrebuiltConfigOutputFields class"""
|
|
1627
2120
|
self._subfields.extend(subfields)
|
|
1628
2121
|
return self
|
|
1629
2122
|
|
|
1630
|
-
def alias(self, alias: str) ->
|
|
2123
|
+
def alias(self, alias: str) -> "PrebuiltConfigOutputFields":
|
|
1631
2124
|
self._alias = alias
|
|
1632
2125
|
return self
|
|
1633
2126
|
|
|
2127
|
+
|
|
1634
2128
|
class PrebuiltCriteriaFields(GraphQLField):
|
|
1635
2129
|
"""@private"""
|
|
1636
|
-
|
|
1637
|
-
|
|
2130
|
+
|
|
2131
|
+
key: "PrebuiltCriteriaGraphQLField" = PrebuiltCriteriaGraphQLField("key")
|
|
2132
|
+
name: "PrebuiltCriteriaGraphQLField" = PrebuiltCriteriaGraphQLField("name")
|
|
1638
2133
|
|
|
1639
2134
|
@classmethod
|
|
1640
|
-
def feedback(cls) ->
|
|
1641
|
-
return MetricFields(
|
|
1642
|
-
description: 'PrebuiltCriteriaGraphQLField' = PrebuiltCriteriaGraphQLField('description')
|
|
2135
|
+
def feedback(cls) -> "MetricFields":
|
|
2136
|
+
return MetricFields("feedback")
|
|
1643
2137
|
|
|
1644
|
-
|
|
2138
|
+
description: "PrebuiltCriteriaGraphQLField" = PrebuiltCriteriaGraphQLField("description")
|
|
2139
|
+
|
|
2140
|
+
def fields(self, *subfields: Union[PrebuiltCriteriaGraphQLField, "MetricFields"]) -> "PrebuiltCriteriaFields":
|
|
1645
2141
|
"""Subfields should come from the PrebuiltCriteriaFields class"""
|
|
1646
2142
|
self._subfields.extend(subfields)
|
|
1647
2143
|
return self
|
|
1648
2144
|
|
|
1649
|
-
def alias(self, alias: str) ->
|
|
2145
|
+
def alias(self, alias: str) -> "PrebuiltCriteriaFields":
|
|
1650
2146
|
self._alias = alias
|
|
1651
2147
|
return self
|
|
1652
2148
|
|
|
2149
|
+
|
|
2150
|
+
class ProviderFields(GraphQLField):
|
|
2151
|
+
"""@private"""
|
|
2152
|
+
|
|
2153
|
+
name: "ProviderGraphQLField" = ProviderGraphQLField("name")
|
|
2154
|
+
description: "ProviderGraphQLField" = ProviderGraphQLField("description")
|
|
2155
|
+
capabilities: "ProviderGraphQLField" = ProviderGraphQLField("capabilities")
|
|
2156
|
+
allowed_scopes: "ProviderGraphQLField" = ProviderGraphQLField("allowedScopes")
|
|
2157
|
+
|
|
2158
|
+
def fields(self, *subfields: ProviderGraphQLField) -> "ProviderFields":
|
|
2159
|
+
"""Subfields should come from the ProviderFields class"""
|
|
2160
|
+
self._subfields.extend(subfields)
|
|
2161
|
+
return self
|
|
2162
|
+
|
|
2163
|
+
def alias(self, alias: str) -> "ProviderFields":
|
|
2164
|
+
self._alias = alias
|
|
2165
|
+
return self
|
|
2166
|
+
|
|
2167
|
+
|
|
1653
2168
|
class ProviderListFields(GraphQLField):
|
|
1654
2169
|
"""@private"""
|
|
1655
2170
|
|
|
1656
2171
|
@classmethod
|
|
1657
|
-
def providers(cls) ->
|
|
1658
|
-
return AuthProviderFields(
|
|
2172
|
+
def providers(cls) -> "AuthProviderFields":
|
|
2173
|
+
return AuthProviderFields("providers")
|
|
1659
2174
|
|
|
1660
|
-
def fields(self, *subfields: Union[ProviderListGraphQLField,
|
|
2175
|
+
def fields(self, *subfields: Union[ProviderListGraphQLField, "AuthProviderFields"]) -> "ProviderListFields":
|
|
1661
2176
|
"""Subfields should come from the ProviderListFields class"""
|
|
1662
2177
|
self._subfields.extend(subfields)
|
|
1663
2178
|
return self
|
|
1664
2179
|
|
|
1665
|
-
def alias(self, alias: str) ->
|
|
2180
|
+
def alias(self, alias: str) -> "ProviderListFields":
|
|
1666
2181
|
self._alias = alias
|
|
1667
2182
|
return self
|
|
1668
2183
|
|
|
2184
|
+
|
|
1669
2185
|
class RemoteConfigOutputFields(GraphQLField):
|
|
1670
2186
|
"""@private"""
|
|
1671
|
-
url: 'RemoteConfigOutputGraphQLField' = RemoteConfigOutputGraphQLField('url')
|
|
1672
|
-
version: 'RemoteConfigOutputGraphQLField' = RemoteConfigOutputGraphQLField('version')
|
|
1673
|
-
description: 'RemoteConfigOutputGraphQLField' = RemoteConfigOutputGraphQLField('description')
|
|
1674
2187
|
|
|
1675
|
-
|
|
2188
|
+
url: "RemoteConfigOutputGraphQLField" = RemoteConfigOutputGraphQLField("url")
|
|
2189
|
+
version: "RemoteConfigOutputGraphQLField" = RemoteConfigOutputGraphQLField("version")
|
|
2190
|
+
description: "RemoteConfigOutputGraphQLField" = RemoteConfigOutputGraphQLField("description")
|
|
2191
|
+
|
|
2192
|
+
def fields(self, *subfields: RemoteConfigOutputGraphQLField) -> "RemoteConfigOutputFields":
|
|
1676
2193
|
"""Subfields should come from the RemoteConfigOutputFields class"""
|
|
1677
2194
|
self._subfields.extend(subfields)
|
|
1678
2195
|
return self
|
|
1679
2196
|
|
|
1680
|
-
def alias(self, alias: str) ->
|
|
2197
|
+
def alias(self, alias: str) -> "RemoteConfigOutputFields":
|
|
1681
2198
|
self._alias = alias
|
|
1682
2199
|
return self
|
|
1683
2200
|
|
|
2201
|
+
|
|
1684
2202
|
class RemoteEnvFields(GraphQLField):
|
|
1685
2203
|
"""@private"""
|
|
1686
|
-
|
|
1687
|
-
|
|
1688
|
-
|
|
1689
|
-
|
|
1690
|
-
|
|
1691
|
-
|
|
1692
|
-
|
|
1693
|
-
|
|
1694
|
-
|
|
1695
|
-
|
|
1696
|
-
|
|
2204
|
+
|
|
2205
|
+
id: "RemoteEnvGraphQLField" = RemoteEnvGraphQLField("id")
|
|
2206
|
+
key: "RemoteEnvGraphQLField" = RemoteEnvGraphQLField("key")
|
|
2207
|
+
name: "RemoteEnvGraphQLField" = RemoteEnvGraphQLField("name")
|
|
2208
|
+
url: "RemoteEnvGraphQLField" = RemoteEnvGraphQLField("url")
|
|
2209
|
+
description: "RemoteEnvGraphQLField" = RemoteEnvGraphQLField("description")
|
|
2210
|
+
created_at: "RemoteEnvGraphQLField" = RemoteEnvGraphQLField("createdAt")
|
|
2211
|
+
version: "RemoteEnvGraphQLField" = RemoteEnvGraphQLField("version")
|
|
2212
|
+
status: "RemoteEnvGraphQLField" = RemoteEnvGraphQLField("status")
|
|
2213
|
+
metadata_schema: "RemoteEnvGraphQLField" = RemoteEnvGraphQLField("metadataSchema")
|
|
2214
|
+
|
|
2215
|
+
def fields(self, *subfields: RemoteEnvGraphQLField) -> "RemoteEnvFields":
|
|
1697
2216
|
"""Subfields should come from the RemoteEnvFields class"""
|
|
1698
2217
|
self._subfields.extend(subfields)
|
|
1699
2218
|
return self
|
|
1700
2219
|
|
|
1701
|
-
def alias(self, alias: str) ->
|
|
2220
|
+
def alias(self, alias: str) -> "RemoteEnvFields":
|
|
1702
2221
|
self._alias = alias
|
|
1703
2222
|
return self
|
|
1704
2223
|
|
|
2224
|
+
|
|
1705
2225
|
class RemoteEnvTestOfflineFields(GraphQLField):
|
|
1706
2226
|
"""@private"""
|
|
1707
|
-
error: 'RemoteEnvTestOfflineGraphQLField' = RemoteEnvTestOfflineGraphQLField('error')
|
|
1708
2227
|
|
|
1709
|
-
|
|
2228
|
+
error: "RemoteEnvTestOfflineGraphQLField" = RemoteEnvTestOfflineGraphQLField("error")
|
|
2229
|
+
|
|
2230
|
+
def fields(self, *subfields: RemoteEnvTestOfflineGraphQLField) -> "RemoteEnvTestOfflineFields":
|
|
1710
2231
|
"""Subfields should come from the RemoteEnvTestOfflineFields class"""
|
|
1711
2232
|
self._subfields.extend(subfields)
|
|
1712
2233
|
return self
|
|
1713
2234
|
|
|
1714
|
-
def alias(self, alias: str) ->
|
|
2235
|
+
def alias(self, alias: str) -> "RemoteEnvTestOfflineFields":
|
|
1715
2236
|
self._alias = alias
|
|
1716
2237
|
return self
|
|
1717
2238
|
|
|
2239
|
+
|
|
1718
2240
|
class RemoteEnvTestOnlineFields(GraphQLField):
|
|
1719
2241
|
"""@private"""
|
|
1720
|
-
name: 'RemoteEnvTestOnlineGraphQLField' = RemoteEnvTestOnlineGraphQLField('name')
|
|
1721
|
-
version: 'RemoteEnvTestOnlineGraphQLField' = RemoteEnvTestOnlineGraphQLField('version')
|
|
1722
|
-
description: 'RemoteEnvTestOnlineGraphQLField' = RemoteEnvTestOnlineGraphQLField('description')
|
|
1723
2242
|
|
|
1724
|
-
|
|
2243
|
+
name: "RemoteEnvTestOnlineGraphQLField" = RemoteEnvTestOnlineGraphQLField("name")
|
|
2244
|
+
version: "RemoteEnvTestOnlineGraphQLField" = RemoteEnvTestOnlineGraphQLField("version")
|
|
2245
|
+
description: "RemoteEnvTestOnlineGraphQLField" = RemoteEnvTestOnlineGraphQLField("description")
|
|
2246
|
+
|
|
2247
|
+
def fields(self, *subfields: RemoteEnvTestOnlineGraphQLField) -> "RemoteEnvTestOnlineFields":
|
|
1725
2248
|
"""Subfields should come from the RemoteEnvTestOnlineFields class"""
|
|
1726
2249
|
self._subfields.extend(subfields)
|
|
1727
2250
|
return self
|
|
1728
2251
|
|
|
1729
|
-
def alias(self, alias: str) ->
|
|
2252
|
+
def alias(self, alias: str) -> "RemoteEnvTestOnlineFields":
|
|
1730
2253
|
self._alias = alias
|
|
1731
2254
|
return self
|
|
1732
2255
|
|
|
2256
|
+
|
|
1733
2257
|
class RoleFields(GraphQLField):
|
|
1734
2258
|
"""@private"""
|
|
1735
|
-
id: 'RoleGraphQLField' = RoleGraphQLField('id')
|
|
1736
|
-
key: 'RoleGraphQLField' = RoleGraphQLField('key')
|
|
1737
|
-
name: 'RoleGraphQLField' = RoleGraphQLField('name')
|
|
1738
|
-
created_at: 'RoleGraphQLField' = RoleGraphQLField('createdAt')
|
|
1739
|
-
permissions: 'RoleGraphQLField' = RoleGraphQLField('permissions')
|
|
1740
2259
|
|
|
1741
|
-
|
|
2260
|
+
id: "RoleGraphQLField" = RoleGraphQLField("id")
|
|
2261
|
+
key: "RoleGraphQLField" = RoleGraphQLField("key")
|
|
2262
|
+
name: "RoleGraphQLField" = RoleGraphQLField("name")
|
|
2263
|
+
created_at: "RoleGraphQLField" = RoleGraphQLField("createdAt")
|
|
2264
|
+
permissions: "RoleGraphQLField" = RoleGraphQLField("permissions")
|
|
2265
|
+
|
|
2266
|
+
def fields(self, *subfields: RoleGraphQLField) -> "RoleFields":
|
|
1742
2267
|
"""Subfields should come from the RoleFields class"""
|
|
1743
2268
|
self._subfields.extend(subfields)
|
|
1744
2269
|
return self
|
|
1745
2270
|
|
|
1746
|
-
def alias(self, alias: str) ->
|
|
2271
|
+
def alias(self, alias: str) -> "RoleFields":
|
|
1747
2272
|
self._alias = alias
|
|
1748
2273
|
return self
|
|
1749
2274
|
|
|
2275
|
+
|
|
1750
2276
|
class SearchResultFields(GraphQLField):
|
|
1751
2277
|
"""@private"""
|
|
1752
2278
|
|
|
1753
2279
|
@classmethod
|
|
1754
|
-
def jobs(cls) ->
|
|
1755
|
-
return JobFields(
|
|
2280
|
+
def jobs(cls) -> "JobFields":
|
|
2281
|
+
return JobFields("jobs")
|
|
1756
2282
|
|
|
1757
2283
|
@classmethod
|
|
1758
|
-
def artifacts(cls, *, filter: Optional[ArtifactFilter]=None) ->
|
|
1759
|
-
arguments: Dict[str, Dict[str, Any]] = {
|
|
1760
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
1761
|
-
return JobArtifactFields(
|
|
2284
|
+
def artifacts(cls, *, filter: Optional[ArtifactFilter] = None) -> "JobArtifactFields":
|
|
2285
|
+
arguments: Dict[str, Dict[str, Any]] = {"filter": {"type": "ArtifactFilter", "value": filter}}
|
|
2286
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
|
|
2287
|
+
return JobArtifactFields("artifacts", arguments=cleared_arguments)
|
|
1762
2288
|
|
|
1763
|
-
def fields(
|
|
2289
|
+
def fields(
|
|
2290
|
+
self, *subfields: Union[SearchResultGraphQLField, "JobArtifactFields", "JobFields"]
|
|
2291
|
+
) -> "SearchResultFields":
|
|
1764
2292
|
"""Subfields should come from the SearchResultFields class"""
|
|
1765
2293
|
self._subfields.extend(subfields)
|
|
1766
2294
|
return self
|
|
1767
2295
|
|
|
1768
|
-
def alias(self, alias: str) ->
|
|
2296
|
+
def alias(self, alias: str) -> "SearchResultFields":
|
|
1769
2297
|
self._alias = alias
|
|
1770
2298
|
return self
|
|
1771
2299
|
|
|
2300
|
+
|
|
1772
2301
|
class SessionFields(GraphQLField):
|
|
1773
2302
|
"""@private"""
|
|
1774
|
-
|
|
2303
|
+
|
|
2304
|
+
id: "SessionGraphQLField" = SessionGraphQLField("id")
|
|
1775
2305
|
|
|
1776
2306
|
@classmethod
|
|
1777
|
-
def turns(cls) ->
|
|
1778
|
-
return CompletionFields(
|
|
2307
|
+
def turns(cls) -> "CompletionFields":
|
|
2308
|
+
return CompletionFields("turns")
|
|
1779
2309
|
|
|
1780
|
-
def fields(self, *subfields: Union[SessionGraphQLField,
|
|
2310
|
+
def fields(self, *subfields: Union[SessionGraphQLField, "CompletionFields"]) -> "SessionFields":
|
|
1781
2311
|
"""Subfields should come from the SessionFields class"""
|
|
1782
2312
|
self._subfields.extend(subfields)
|
|
1783
2313
|
return self
|
|
1784
2314
|
|
|
1785
|
-
def alias(self, alias: str) ->
|
|
2315
|
+
def alias(self, alias: str) -> "SessionFields":
|
|
1786
2316
|
self._alias = alias
|
|
1787
2317
|
return self
|
|
1788
2318
|
|
|
2319
|
+
|
|
1789
2320
|
class SettingsFields(GraphQLField):
|
|
1790
2321
|
"""@private"""
|
|
1791
2322
|
|
|
1792
2323
|
@classmethod
|
|
1793
|
-
def default_metric(cls) ->
|
|
1794
|
-
return MetricWithContextFields(
|
|
2324
|
+
def default_metric(cls) -> "MetricWithContextFields":
|
|
2325
|
+
return MetricWithContextFields("defaultMetric")
|
|
1795
2326
|
|
|
1796
|
-
def fields(self, *subfields: Union[SettingsGraphQLField,
|
|
2327
|
+
def fields(self, *subfields: Union[SettingsGraphQLField, "MetricWithContextFields"]) -> "SettingsFields":
|
|
1797
2328
|
"""Subfields should come from the SettingsFields class"""
|
|
1798
2329
|
self._subfields.extend(subfields)
|
|
1799
2330
|
return self
|
|
1800
2331
|
|
|
1801
|
-
def alias(self, alias: str) ->
|
|
2332
|
+
def alias(self, alias: str) -> "SettingsFields":
|
|
1802
2333
|
self._alias = alias
|
|
1803
2334
|
return self
|
|
1804
2335
|
|
|
2336
|
+
|
|
1805
2337
|
class ShareFields(GraphQLField):
|
|
1806
2338
|
"""@private"""
|
|
1807
2339
|
|
|
1808
2340
|
@classmethod
|
|
1809
|
-
def team(cls) ->
|
|
1810
|
-
return TeamFields(
|
|
2341
|
+
def team(cls) -> "TeamFields":
|
|
2342
|
+
return TeamFields("team")
|
|
1811
2343
|
|
|
1812
2344
|
@classmethod
|
|
1813
|
-
def role(cls) ->
|
|
1814
|
-
return RoleFields(
|
|
1815
|
-
is_owner: 'ShareGraphQLField' = ShareGraphQLField('isOwner')
|
|
2345
|
+
def role(cls) -> "RoleFields":
|
|
2346
|
+
return RoleFields("role")
|
|
1816
2347
|
|
|
1817
|
-
|
|
2348
|
+
is_owner: "ShareGraphQLField" = ShareGraphQLField("isOwner")
|
|
2349
|
+
|
|
2350
|
+
def fields(self, *subfields: Union[ShareGraphQLField, "RoleFields", "TeamFields"]) -> "ShareFields":
|
|
1818
2351
|
"""Subfields should come from the ShareFields class"""
|
|
1819
2352
|
self._subfields.extend(subfields)
|
|
1820
2353
|
return self
|
|
1821
2354
|
|
|
1822
|
-
def alias(self, alias: str) ->
|
|
2355
|
+
def alias(self, alias: str) -> "ShareFields":
|
|
1823
2356
|
self._alias = alias
|
|
1824
2357
|
return self
|
|
1825
2358
|
|
|
2359
|
+
|
|
2360
|
+
class SlackConfigFields(GraphQLField):
|
|
2361
|
+
"""@private"""
|
|
2362
|
+
|
|
2363
|
+
webhook_url: "SlackConfigGraphQLField" = SlackConfigGraphQLField("webhookUrl")
|
|
2364
|
+
bot_token: "SlackConfigGraphQLField" = SlackConfigGraphQLField("botToken")
|
|
2365
|
+
|
|
2366
|
+
def fields(self, *subfields: SlackConfigGraphQLField) -> "SlackConfigFields":
|
|
2367
|
+
"""Subfields should come from the SlackConfigFields class"""
|
|
2368
|
+
self._subfields.extend(subfields)
|
|
2369
|
+
return self
|
|
2370
|
+
|
|
2371
|
+
def alias(self, alias: str) -> "SlackConfigFields":
|
|
2372
|
+
self._alias = alias
|
|
2373
|
+
return self
|
|
2374
|
+
|
|
2375
|
+
|
|
2376
|
+
class SmtpConfigFields(GraphQLField):
|
|
2377
|
+
"""@private"""
|
|
2378
|
+
|
|
2379
|
+
host: "SmtpConfigGraphQLField" = SmtpConfigGraphQLField("host")
|
|
2380
|
+
port: "SmtpConfigGraphQLField" = SmtpConfigGraphQLField("port")
|
|
2381
|
+
username: "SmtpConfigGraphQLField" = SmtpConfigGraphQLField("username")
|
|
2382
|
+
password: "SmtpConfigGraphQLField" = SmtpConfigGraphQLField("password")
|
|
2383
|
+
from_email: "SmtpConfigGraphQLField" = SmtpConfigGraphQLField("fromEmail")
|
|
2384
|
+
|
|
2385
|
+
def fields(self, *subfields: SmtpConfigGraphQLField) -> "SmtpConfigFields":
|
|
2386
|
+
"""Subfields should come from the SmtpConfigFields class"""
|
|
2387
|
+
self._subfields.extend(subfields)
|
|
2388
|
+
return self
|
|
2389
|
+
|
|
2390
|
+
def alias(self, alias: str) -> "SmtpConfigFields":
|
|
2391
|
+
self._alias = alias
|
|
2392
|
+
return self
|
|
2393
|
+
|
|
2394
|
+
|
|
2395
|
+
class SubscriptionFields(GraphQLField):
|
|
2396
|
+
"""@private"""
|
|
2397
|
+
|
|
2398
|
+
topic_pattern: "SubscriptionGraphQLField" = SubscriptionGraphQLField("topicPattern")
|
|
2399
|
+
scope: "SubscriptionGraphQLField" = SubscriptionGraphQLField("scope")
|
|
2400
|
+
target_filter: "SubscriptionGraphQLField" = SubscriptionGraphQLField("targetFilter")
|
|
2401
|
+
|
|
2402
|
+
def fields(self, *subfields: SubscriptionGraphQLField) -> "SubscriptionFields":
|
|
2403
|
+
"""Subfields should come from the SubscriptionFields class"""
|
|
2404
|
+
self._subfields.extend(subfields)
|
|
2405
|
+
return self
|
|
2406
|
+
|
|
2407
|
+
def alias(self, alias: str) -> "SubscriptionFields":
|
|
2408
|
+
self._alias = alias
|
|
2409
|
+
return self
|
|
2410
|
+
|
|
2411
|
+
|
|
1826
2412
|
class SystemPromptTemplateFields(GraphQLField):
|
|
1827
2413
|
"""@private"""
|
|
1828
|
-
id: 'SystemPromptTemplateGraphQLField' = SystemPromptTemplateGraphQLField('id')
|
|
1829
|
-
name: 'SystemPromptTemplateGraphQLField' = SystemPromptTemplateGraphQLField('name')
|
|
1830
|
-
template: 'SystemPromptTemplateGraphQLField' = SystemPromptTemplateGraphQLField('template')
|
|
1831
|
-
arguments: 'SystemPromptTemplateGraphQLField' = SystemPromptTemplateGraphQLField('arguments')
|
|
1832
|
-
created_at: 'SystemPromptTemplateGraphQLField' = SystemPromptTemplateGraphQLField('createdAt')
|
|
1833
|
-
created_by: 'SystemPromptTemplateGraphQLField' = SystemPromptTemplateGraphQLField('createdBy')
|
|
1834
2414
|
|
|
1835
|
-
|
|
2415
|
+
id: "SystemPromptTemplateGraphQLField" = SystemPromptTemplateGraphQLField("id")
|
|
2416
|
+
name: "SystemPromptTemplateGraphQLField" = SystemPromptTemplateGraphQLField("name")
|
|
2417
|
+
template: "SystemPromptTemplateGraphQLField" = SystemPromptTemplateGraphQLField("template")
|
|
2418
|
+
arguments: "SystemPromptTemplateGraphQLField" = SystemPromptTemplateGraphQLField("arguments")
|
|
2419
|
+
created_at: "SystemPromptTemplateGraphQLField" = SystemPromptTemplateGraphQLField("createdAt")
|
|
2420
|
+
created_by: "SystemPromptTemplateGraphQLField" = SystemPromptTemplateGraphQLField("createdBy")
|
|
2421
|
+
|
|
2422
|
+
def fields(self, *subfields: SystemPromptTemplateGraphQLField) -> "SystemPromptTemplateFields":
|
|
1836
2423
|
"""Subfields should come from the SystemPromptTemplateFields class"""
|
|
1837
2424
|
self._subfields.extend(subfields)
|
|
1838
2425
|
return self
|
|
1839
2426
|
|
|
1840
|
-
def alias(self, alias: str) ->
|
|
2427
|
+
def alias(self, alias: str) -> "SystemPromptTemplateFields":
|
|
1841
2428
|
self._alias = alias
|
|
1842
2429
|
return self
|
|
1843
2430
|
|
|
2431
|
+
|
|
1844
2432
|
class TeamFields(GraphQLField):
|
|
1845
2433
|
"""@private"""
|
|
1846
|
-
id: 'TeamGraphQLField' = TeamGraphQLField('id')
|
|
1847
|
-
key: 'TeamGraphQLField' = TeamGraphQLField('key')
|
|
1848
|
-
name: 'TeamGraphQLField' = TeamGraphQLField('name')
|
|
1849
|
-
created_at: 'TeamGraphQLField' = TeamGraphQLField('createdAt')
|
|
1850
2434
|
|
|
1851
|
-
|
|
2435
|
+
id: "TeamGraphQLField" = TeamGraphQLField("id")
|
|
2436
|
+
key: "TeamGraphQLField" = TeamGraphQLField("key")
|
|
2437
|
+
name: "TeamGraphQLField" = TeamGraphQLField("name")
|
|
2438
|
+
created_at: "TeamGraphQLField" = TeamGraphQLField("createdAt")
|
|
2439
|
+
|
|
2440
|
+
def fields(self, *subfields: TeamGraphQLField) -> "TeamFields":
|
|
1852
2441
|
"""Subfields should come from the TeamFields class"""
|
|
1853
2442
|
self._subfields.extend(subfields)
|
|
1854
2443
|
return self
|
|
1855
2444
|
|
|
1856
|
-
def alias(self, alias: str) ->
|
|
2445
|
+
def alias(self, alias: str) -> "TeamFields":
|
|
1857
2446
|
self._alias = alias
|
|
1858
2447
|
return self
|
|
1859
2448
|
|
|
2449
|
+
|
|
1860
2450
|
class TeamMemberFields(GraphQLField):
|
|
1861
2451
|
"""@private"""
|
|
1862
2452
|
|
|
1863
2453
|
@classmethod
|
|
1864
|
-
def user(cls) ->
|
|
1865
|
-
return UserFields(
|
|
2454
|
+
def user(cls) -> "UserFields":
|
|
2455
|
+
return UserFields("user")
|
|
1866
2456
|
|
|
1867
2457
|
@classmethod
|
|
1868
|
-
def team(cls) ->
|
|
1869
|
-
return TeamFields(
|
|
2458
|
+
def team(cls) -> "TeamFields":
|
|
2459
|
+
return TeamFields("team")
|
|
1870
2460
|
|
|
1871
2461
|
@classmethod
|
|
1872
|
-
def role(cls) ->
|
|
1873
|
-
return RoleFields(
|
|
2462
|
+
def role(cls) -> "RoleFields":
|
|
2463
|
+
return RoleFields("role")
|
|
1874
2464
|
|
|
1875
|
-
def fields(
|
|
2465
|
+
def fields(
|
|
2466
|
+
self, *subfields: Union[TeamMemberGraphQLField, "RoleFields", "TeamFields", "UserFields"]
|
|
2467
|
+
) -> "TeamMemberFields":
|
|
1876
2468
|
"""Subfields should come from the TeamMemberFields class"""
|
|
1877
2469
|
self._subfields.extend(subfields)
|
|
1878
2470
|
return self
|
|
1879
2471
|
|
|
1880
|
-
def alias(self, alias: str) ->
|
|
2472
|
+
def alias(self, alias: str) -> "TeamMemberFields":
|
|
1881
2473
|
self._alias = alias
|
|
1882
2474
|
return self
|
|
1883
2475
|
|
|
2476
|
+
|
|
1884
2477
|
class TeamWithroleFields(GraphQLField):
|
|
1885
2478
|
"""@private"""
|
|
1886
2479
|
|
|
1887
2480
|
@classmethod
|
|
1888
|
-
def team(cls) ->
|
|
1889
|
-
return TeamFields(
|
|
2481
|
+
def team(cls) -> "TeamFields":
|
|
2482
|
+
return TeamFields("team")
|
|
1890
2483
|
|
|
1891
2484
|
@classmethod
|
|
1892
|
-
def role(cls) ->
|
|
1893
|
-
return RoleFields(
|
|
2485
|
+
def role(cls) -> "RoleFields":
|
|
2486
|
+
return RoleFields("role")
|
|
1894
2487
|
|
|
1895
|
-
def fields(self, *subfields: Union[TeamWithroleGraphQLField,
|
|
2488
|
+
def fields(self, *subfields: Union[TeamWithroleGraphQLField, "RoleFields", "TeamFields"]) -> "TeamWithroleFields":
|
|
1896
2489
|
"""Subfields should come from the TeamWithroleFields class"""
|
|
1897
2490
|
self._subfields.extend(subfields)
|
|
1898
2491
|
return self
|
|
1899
2492
|
|
|
1900
|
-
def alias(self, alias: str) ->
|
|
2493
|
+
def alias(self, alias: str) -> "TeamWithroleFields":
|
|
2494
|
+
self._alias = alias
|
|
2495
|
+
return self
|
|
2496
|
+
|
|
2497
|
+
|
|
2498
|
+
class TestNotificationResultFields(GraphQLField):
|
|
2499
|
+
"""@private"""
|
|
2500
|
+
|
|
2501
|
+
event_id: "TestNotificationResultGraphQLField" = TestNotificationResultGraphQLField("eventId")
|
|
2502
|
+
message: "TestNotificationResultGraphQLField" = TestNotificationResultGraphQLField("message")
|
|
2503
|
+
|
|
2504
|
+
def fields(self, *subfields: TestNotificationResultGraphQLField) -> "TestNotificationResultFields":
|
|
2505
|
+
"""Subfields should come from the TestNotificationResultFields class"""
|
|
2506
|
+
self._subfields.extend(subfields)
|
|
2507
|
+
return self
|
|
2508
|
+
|
|
2509
|
+
def alias(self, alias: str) -> "TestNotificationResultFields":
|
|
1901
2510
|
self._alias = alias
|
|
1902
2511
|
return self
|
|
1903
2512
|
|
|
2513
|
+
|
|
1904
2514
|
class TimeseriesFields(GraphQLField):
|
|
1905
2515
|
"""@private"""
|
|
1906
2516
|
|
|
1907
2517
|
@classmethod
|
|
1908
|
-
def model(cls) ->
|
|
1909
|
-
return ModelFields(
|
|
1910
|
-
|
|
1911
|
-
|
|
1912
|
-
|
|
1913
|
-
|
|
2518
|
+
def model(cls) -> "ModelFields":
|
|
2519
|
+
return ModelFields("model")
|
|
2520
|
+
|
|
2521
|
+
time_buckets: "TimeseriesGraphQLField" = TimeseriesGraphQLField("timeBuckets")
|
|
2522
|
+
count: "TimeseriesGraphQLField" = TimeseriesGraphQLField("count")
|
|
2523
|
+
values: "TimeseriesGraphQLField" = TimeseriesGraphQLField("values")
|
|
2524
|
+
aggregation: "TimeseriesGraphQLField" = TimeseriesGraphQLField("aggregation")
|
|
1914
2525
|
|
|
1915
|
-
def fields(self, *subfields: Union[TimeseriesGraphQLField,
|
|
2526
|
+
def fields(self, *subfields: Union[TimeseriesGraphQLField, "ModelFields"]) -> "TimeseriesFields":
|
|
1916
2527
|
"""Subfields should come from the TimeseriesFields class"""
|
|
1917
2528
|
self._subfields.extend(subfields)
|
|
1918
2529
|
return self
|
|
1919
2530
|
|
|
1920
|
-
def alias(self, alias: str) ->
|
|
2531
|
+
def alias(self, alias: str) -> "TimeseriesFields":
|
|
1921
2532
|
self._alias = alias
|
|
1922
2533
|
return self
|
|
1923
2534
|
|
|
2535
|
+
|
|
1924
2536
|
class ToolProviderFields(GraphQLField):
|
|
1925
2537
|
"""@private"""
|
|
1926
|
-
id: 'ToolProviderGraphQLField' = ToolProviderGraphQLField('id')
|
|
1927
|
-
key: 'ToolProviderGraphQLField' = ToolProviderGraphQLField('key')
|
|
1928
|
-
name: 'ToolProviderGraphQLField' = ToolProviderGraphQLField('name')
|
|
1929
|
-
created_at: 'ToolProviderGraphQLField' = ToolProviderGraphQLField('createdAt')
|
|
1930
|
-
uri: 'ToolProviderGraphQLField' = ToolProviderGraphQLField('uri')
|
|
1931
|
-
protocol: 'ToolProviderGraphQLField' = ToolProviderGraphQLField('protocol')
|
|
1932
2538
|
|
|
1933
|
-
|
|
2539
|
+
id: "ToolProviderGraphQLField" = ToolProviderGraphQLField("id")
|
|
2540
|
+
key: "ToolProviderGraphQLField" = ToolProviderGraphQLField("key")
|
|
2541
|
+
name: "ToolProviderGraphQLField" = ToolProviderGraphQLField("name")
|
|
2542
|
+
created_at: "ToolProviderGraphQLField" = ToolProviderGraphQLField("createdAt")
|
|
2543
|
+
uri: "ToolProviderGraphQLField" = ToolProviderGraphQLField("uri")
|
|
2544
|
+
protocol: "ToolProviderGraphQLField" = ToolProviderGraphQLField("protocol")
|
|
2545
|
+
|
|
2546
|
+
def fields(self, *subfields: ToolProviderGraphQLField) -> "ToolProviderFields":
|
|
1934
2547
|
"""Subfields should come from the ToolProviderFields class"""
|
|
1935
2548
|
self._subfields.extend(subfields)
|
|
1936
2549
|
return self
|
|
1937
2550
|
|
|
1938
|
-
def alias(self, alias: str) ->
|
|
2551
|
+
def alias(self, alias: str) -> "ToolProviderFields":
|
|
1939
2552
|
self._alias = alias
|
|
1940
2553
|
return self
|
|
1941
2554
|
|
|
2555
|
+
|
|
1942
2556
|
class TrainingJobStageOutputFields(GraphQLField):
|
|
1943
2557
|
"""@private"""
|
|
1944
|
-
monitoring_link: 'TrainingJobStageOutputGraphQLField' = TrainingJobStageOutputGraphQLField('monitoringLink')
|
|
1945
|
-
total_num_samples: 'TrainingJobStageOutputGraphQLField' = TrainingJobStageOutputGraphQLField('totalNumSamples')
|
|
1946
|
-
processed_num_samples: 'TrainingJobStageOutputGraphQLField' = TrainingJobStageOutputGraphQLField('processedNumSamples')
|
|
1947
|
-
checkpoints: 'TrainingJobStageOutputGraphQLField' = TrainingJobStageOutputGraphQLField('checkpoints')
|
|
1948
2558
|
|
|
1949
|
-
|
|
2559
|
+
monitoring_link: "TrainingJobStageOutputGraphQLField" = TrainingJobStageOutputGraphQLField("monitoringLink")
|
|
2560
|
+
total_num_samples: "TrainingJobStageOutputGraphQLField" = TrainingJobStageOutputGraphQLField("totalNumSamples")
|
|
2561
|
+
processed_num_samples: "TrainingJobStageOutputGraphQLField" = TrainingJobStageOutputGraphQLField(
|
|
2562
|
+
"processedNumSamples"
|
|
2563
|
+
)
|
|
2564
|
+
checkpoints: "TrainingJobStageOutputGraphQLField" = TrainingJobStageOutputGraphQLField("checkpoints")
|
|
2565
|
+
|
|
2566
|
+
def fields(self, *subfields: TrainingJobStageOutputGraphQLField) -> "TrainingJobStageOutputFields":
|
|
1950
2567
|
"""Subfields should come from the TrainingJobStageOutputFields class"""
|
|
1951
2568
|
self._subfields.extend(subfields)
|
|
1952
2569
|
return self
|
|
1953
2570
|
|
|
1954
|
-
def alias(self, alias: str) ->
|
|
2571
|
+
def alias(self, alias: str) -> "TrainingJobStageOutputFields":
|
|
1955
2572
|
self._alias = alias
|
|
1956
2573
|
return self
|
|
1957
2574
|
|
|
2575
|
+
|
|
1958
2576
|
class TrendResultFields(GraphQLField):
|
|
1959
2577
|
"""@private"""
|
|
1960
|
-
trend: 'TrendResultGraphQLField' = TrendResultGraphQLField('trend')
|
|
1961
|
-
previous: 'TrendResultGraphQLField' = TrendResultGraphQLField('previous')
|
|
1962
|
-
current: 'TrendResultGraphQLField' = TrendResultGraphQLField('current')
|
|
1963
2578
|
|
|
1964
|
-
|
|
2579
|
+
trend: "TrendResultGraphQLField" = TrendResultGraphQLField("trend")
|
|
2580
|
+
previous: "TrendResultGraphQLField" = TrendResultGraphQLField("previous")
|
|
2581
|
+
current: "TrendResultGraphQLField" = TrendResultGraphQLField("current")
|
|
2582
|
+
|
|
2583
|
+
def fields(self, *subfields: TrendResultGraphQLField) -> "TrendResultFields":
|
|
1965
2584
|
"""Subfields should come from the TrendResultFields class"""
|
|
1966
2585
|
self._subfields.extend(subfields)
|
|
1967
2586
|
return self
|
|
1968
2587
|
|
|
1969
|
-
def alias(self, alias: str) ->
|
|
2588
|
+
def alias(self, alias: str) -> "TrendResultFields":
|
|
1970
2589
|
self._alias = alias
|
|
1971
2590
|
return self
|
|
1972
2591
|
|
|
2592
|
+
|
|
1973
2593
|
class UnitConfigFields(GraphQLField):
|
|
1974
2594
|
"""@private"""
|
|
1975
|
-
symbol: 'UnitConfigGraphQLField' = UnitConfigGraphQLField('symbol')
|
|
1976
|
-
position: 'UnitConfigGraphQLField' = UnitConfigGraphQLField('position')
|
|
1977
2595
|
|
|
1978
|
-
|
|
2596
|
+
symbol: "UnitConfigGraphQLField" = UnitConfigGraphQLField("symbol")
|
|
2597
|
+
position: "UnitConfigGraphQLField" = UnitConfigGraphQLField("position")
|
|
2598
|
+
|
|
2599
|
+
def fields(self, *subfields: UnitConfigGraphQLField) -> "UnitConfigFields":
|
|
1979
2600
|
"""Subfields should come from the UnitConfigFields class"""
|
|
1980
2601
|
self._subfields.extend(subfields)
|
|
1981
2602
|
return self
|
|
1982
2603
|
|
|
1983
|
-
def alias(self, alias: str) ->
|
|
2604
|
+
def alias(self, alias: str) -> "UnitConfigFields":
|
|
1984
2605
|
self._alias = alias
|
|
1985
2606
|
return self
|
|
1986
2607
|
|
|
2608
|
+
|
|
1987
2609
|
class UsageFields(GraphQLField):
|
|
1988
2610
|
"""@private"""
|
|
1989
|
-
completion_tokens: 'UsageGraphQLField' = UsageGraphQLField('completionTokens')
|
|
1990
|
-
prompt_tokens: 'UsageGraphQLField' = UsageGraphQLField('promptTokens')
|
|
1991
|
-
total_tokens: 'UsageGraphQLField' = UsageGraphQLField('totalTokens')
|
|
1992
2611
|
|
|
1993
|
-
|
|
2612
|
+
completion_tokens: "UsageGraphQLField" = UsageGraphQLField("completionTokens")
|
|
2613
|
+
prompt_tokens: "UsageGraphQLField" = UsageGraphQLField("promptTokens")
|
|
2614
|
+
total_tokens: "UsageGraphQLField" = UsageGraphQLField("totalTokens")
|
|
2615
|
+
|
|
2616
|
+
def fields(self, *subfields: UsageGraphQLField) -> "UsageFields":
|
|
1994
2617
|
"""Subfields should come from the UsageFields class"""
|
|
1995
2618
|
self._subfields.extend(subfields)
|
|
1996
2619
|
return self
|
|
1997
2620
|
|
|
1998
|
-
def alias(self, alias: str) ->
|
|
2621
|
+
def alias(self, alias: str) -> "UsageFields":
|
|
1999
2622
|
self._alias = alias
|
|
2000
2623
|
return self
|
|
2001
2624
|
|
|
2625
|
+
|
|
2002
2626
|
class UsageAggregateItemFields(GraphQLField):
|
|
2003
2627
|
"""@private"""
|
|
2004
|
-
bucket_ts: 'UsageAggregateItemGraphQLField' = UsageAggregateItemGraphQLField('bucketTs')
|
|
2005
|
-
prompt_tokens: 'UsageAggregateItemGraphQLField' = UsageAggregateItemGraphQLField('promptTokens')
|
|
2006
|
-
completion_tokens: 'UsageAggregateItemGraphQLField' = UsageAggregateItemGraphQLField('completionTokens')
|
|
2007
|
-
total_tokens: 'UsageAggregateItemGraphQLField' = UsageAggregateItemGraphQLField('totalTokens')
|
|
2008
|
-
interactions: 'UsageAggregateItemGraphQLField' = UsageAggregateItemGraphQLField('interactions')
|
|
2009
2628
|
|
|
2010
|
-
|
|
2629
|
+
bucket_ts: "UsageAggregateItemGraphQLField" = UsageAggregateItemGraphQLField("bucketTs")
|
|
2630
|
+
prompt_tokens: "UsageAggregateItemGraphQLField" = UsageAggregateItemGraphQLField("promptTokens")
|
|
2631
|
+
completion_tokens: "UsageAggregateItemGraphQLField" = UsageAggregateItemGraphQLField("completionTokens")
|
|
2632
|
+
total_tokens: "UsageAggregateItemGraphQLField" = UsageAggregateItemGraphQLField("totalTokens")
|
|
2633
|
+
interactions: "UsageAggregateItemGraphQLField" = UsageAggregateItemGraphQLField("interactions")
|
|
2634
|
+
|
|
2635
|
+
def fields(self, *subfields: UsageAggregateItemGraphQLField) -> "UsageAggregateItemFields":
|
|
2011
2636
|
"""Subfields should come from the UsageAggregateItemFields class"""
|
|
2012
2637
|
self._subfields.extend(subfields)
|
|
2013
2638
|
return self
|
|
2014
2639
|
|
|
2015
|
-
def alias(self, alias: str) ->
|
|
2640
|
+
def alias(self, alias: str) -> "UsageAggregateItemFields":
|
|
2016
2641
|
self._alias = alias
|
|
2017
2642
|
return self
|
|
2018
2643
|
|
|
2644
|
+
|
|
2019
2645
|
class UsageAggregatePerUseCaseItemFields(GraphQLField):
|
|
2020
2646
|
"""@private"""
|
|
2021
2647
|
|
|
2022
2648
|
@classmethod
|
|
2023
|
-
def use_case(cls) ->
|
|
2024
|
-
return UseCaseItemFields(
|
|
2649
|
+
def use_case(cls) -> "UseCaseItemFields":
|
|
2650
|
+
return UseCaseItemFields("useCase")
|
|
2025
2651
|
|
|
2026
2652
|
@classmethod
|
|
2027
|
-
def model_service(cls) ->
|
|
2028
|
-
return ModelServiceFields(
|
|
2029
|
-
|
|
2030
|
-
|
|
2031
|
-
|
|
2032
|
-
|
|
2653
|
+
def model_service(cls) -> "ModelServiceFields":
|
|
2654
|
+
return ModelServiceFields("modelService")
|
|
2655
|
+
|
|
2656
|
+
prompt_tokens: "UsageAggregatePerUseCaseItemGraphQLField" = UsageAggregatePerUseCaseItemGraphQLField("promptTokens")
|
|
2657
|
+
completion_tokens: "UsageAggregatePerUseCaseItemGraphQLField" = UsageAggregatePerUseCaseItemGraphQLField(
|
|
2658
|
+
"completionTokens"
|
|
2659
|
+
)
|
|
2660
|
+
total_tokens: "UsageAggregatePerUseCaseItemGraphQLField" = UsageAggregatePerUseCaseItemGraphQLField("totalTokens")
|
|
2661
|
+
interactions: "UsageAggregatePerUseCaseItemGraphQLField" = UsageAggregatePerUseCaseItemGraphQLField("interactions")
|
|
2033
2662
|
|
|
2034
|
-
def fields(
|
|
2663
|
+
def fields(
|
|
2664
|
+
self, *subfields: Union[UsageAggregatePerUseCaseItemGraphQLField, "ModelServiceFields", "UseCaseItemFields"]
|
|
2665
|
+
) -> "UsageAggregatePerUseCaseItemFields":
|
|
2035
2666
|
"""Subfields should come from the UsageAggregatePerUseCaseItemFields class"""
|
|
2036
2667
|
self._subfields.extend(subfields)
|
|
2037
2668
|
return self
|
|
2038
2669
|
|
|
2039
|
-
def alias(self, alias: str) ->
|
|
2670
|
+
def alias(self, alias: str) -> "UsageAggregatePerUseCaseItemFields":
|
|
2040
2671
|
self._alias = alias
|
|
2041
2672
|
return self
|
|
2042
2673
|
|
|
2674
|
+
|
|
2043
2675
|
class UsageStatsFields(GraphQLField):
|
|
2044
2676
|
"""@private"""
|
|
2045
|
-
total_tokens: 'UsageStatsGraphQLField' = UsageStatsGraphQLField('totalTokens')
|
|
2046
|
-
interactions: 'UsageStatsGraphQLField' = UsageStatsGraphQLField('interactions')
|
|
2047
|
-
prompt_tokens: 'UsageStatsGraphQLField' = UsageStatsGraphQLField('promptTokens')
|
|
2048
|
-
completion_tokens: 'UsageStatsGraphQLField' = UsageStatsGraphQLField('completionTokens')
|
|
2049
2677
|
|
|
2050
|
-
|
|
2678
|
+
total_tokens: "UsageStatsGraphQLField" = UsageStatsGraphQLField("totalTokens")
|
|
2679
|
+
interactions: "UsageStatsGraphQLField" = UsageStatsGraphQLField("interactions")
|
|
2680
|
+
prompt_tokens: "UsageStatsGraphQLField" = UsageStatsGraphQLField("promptTokens")
|
|
2681
|
+
completion_tokens: "UsageStatsGraphQLField" = UsageStatsGraphQLField("completionTokens")
|
|
2682
|
+
|
|
2683
|
+
def fields(self, *subfields: UsageStatsGraphQLField) -> "UsageStatsFields":
|
|
2051
2684
|
"""Subfields should come from the UsageStatsFields class"""
|
|
2052
2685
|
self._subfields.extend(subfields)
|
|
2053
2686
|
return self
|
|
2054
2687
|
|
|
2055
|
-
def alias(self, alias: str) ->
|
|
2688
|
+
def alias(self, alias: str) -> "UsageStatsFields":
|
|
2056
2689
|
self._alias = alias
|
|
2057
2690
|
return self
|
|
2058
2691
|
|
|
2692
|
+
|
|
2059
2693
|
class UsageStatsByModelFields(GraphQLField):
|
|
2060
2694
|
"""@private"""
|
|
2061
2695
|
|
|
2062
2696
|
@classmethod
|
|
2063
|
-
def model(cls) ->
|
|
2064
|
-
return ModelFields(
|
|
2065
|
-
|
|
2066
|
-
|
|
2067
|
-
|
|
2068
|
-
|
|
2697
|
+
def model(cls) -> "ModelFields":
|
|
2698
|
+
return ModelFields("model")
|
|
2699
|
+
|
|
2700
|
+
total_tokens: "UsageStatsByModelGraphQLField" = UsageStatsByModelGraphQLField("totalTokens")
|
|
2701
|
+
interactions: "UsageStatsByModelGraphQLField" = UsageStatsByModelGraphQLField("interactions")
|
|
2702
|
+
prompt_tokens: "UsageStatsByModelGraphQLField" = UsageStatsByModelGraphQLField("promptTokens")
|
|
2703
|
+
completion_tokens: "UsageStatsByModelGraphQLField" = UsageStatsByModelGraphQLField("completionTokens")
|
|
2069
2704
|
|
|
2070
2705
|
@classmethod
|
|
2071
|
-
def timeseries(cls) ->
|
|
2072
|
-
return UsageAggregateItemFields(
|
|
2706
|
+
def timeseries(cls) -> "UsageAggregateItemFields":
|
|
2707
|
+
return UsageAggregateItemFields("timeseries")
|
|
2073
2708
|
|
|
2074
|
-
def fields(
|
|
2709
|
+
def fields(
|
|
2710
|
+
self, *subfields: Union[UsageStatsByModelGraphQLField, "ModelFields", "UsageAggregateItemFields"]
|
|
2711
|
+
) -> "UsageStatsByModelFields":
|
|
2075
2712
|
"""Subfields should come from the UsageStatsByModelFields class"""
|
|
2076
2713
|
self._subfields.extend(subfields)
|
|
2077
2714
|
return self
|
|
2078
2715
|
|
|
2079
|
-
def alias(self, alias: str) ->
|
|
2716
|
+
def alias(self, alias: str) -> "UsageStatsByModelFields":
|
|
2080
2717
|
self._alias = alias
|
|
2081
2718
|
return self
|
|
2082
2719
|
|
|
2720
|
+
|
|
2083
2721
|
class UseCaseFields(GraphQLField):
|
|
2084
2722
|
"""@private"""
|
|
2085
|
-
|
|
2086
|
-
|
|
2087
|
-
|
|
2088
|
-
|
|
2089
|
-
|
|
2090
|
-
|
|
2723
|
+
|
|
2724
|
+
id: "UseCaseGraphQLField" = UseCaseGraphQLField("id")
|
|
2725
|
+
name: "UseCaseGraphQLField" = UseCaseGraphQLField("name")
|
|
2726
|
+
key: "UseCaseGraphQLField" = UseCaseGraphQLField("key")
|
|
2727
|
+
description: "UseCaseGraphQLField" = UseCaseGraphQLField("description")
|
|
2728
|
+
created_at: "UseCaseGraphQLField" = UseCaseGraphQLField("createdAt")
|
|
2729
|
+
is_archived: "UseCaseGraphQLField" = UseCaseGraphQLField("isArchived")
|
|
2091
2730
|
|
|
2092
2731
|
@classmethod
|
|
2093
|
-
def model_services(cls, *, filter: Optional[ModelServiceFilter]=None) ->
|
|
2094
|
-
arguments: Dict[str, Dict[str, Any]] = {
|
|
2095
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
2096
|
-
return ModelServiceFields(
|
|
2732
|
+
def model_services(cls, *, filter: Optional[ModelServiceFilter] = None) -> "ModelServiceFields":
|
|
2733
|
+
arguments: Dict[str, Dict[str, Any]] = {"filter": {"type": "ModelServiceFilter", "value": filter}}
|
|
2734
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
|
|
2735
|
+
return ModelServiceFields("modelServices", arguments=cleared_arguments)
|
|
2097
2736
|
|
|
2098
2737
|
@classmethod
|
|
2099
|
-
def model_service(cls, id_or_key: str) ->
|
|
2100
|
-
arguments: Dict[str, Dict[str, Any]] = {
|
|
2101
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
2102
|
-
return ModelServiceFields(
|
|
2738
|
+
def model_service(cls, id_or_key: str) -> "ModelServiceFields":
|
|
2739
|
+
arguments: Dict[str, Dict[str, Any]] = {"idOrKey": {"type": "IdOrKey!", "value": id_or_key}}
|
|
2740
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
|
|
2741
|
+
return ModelServiceFields("modelService", arguments=cleared_arguments)
|
|
2103
2742
|
|
|
2104
2743
|
@classmethod
|
|
2105
|
-
def models(cls, *, filter: Optional[ModelFilter]=None) ->
|
|
2744
|
+
def models(cls, *, filter: Optional[ModelFilter] = None) -> "ModelFields":
|
|
2106
2745
|
"""Returns models associated with this use case."""
|
|
2107
|
-
arguments: Dict[str, Dict[str, Any]] = {
|
|
2108
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
2109
|
-
return ModelFields(
|
|
2746
|
+
arguments: Dict[str, Dict[str, Any]] = {"filter": {"type": "ModelFilter", "value": filter}}
|
|
2747
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
|
|
2748
|
+
return ModelFields("models", arguments=cleared_arguments)
|
|
2110
2749
|
|
|
2111
2750
|
@classmethod
|
|
2112
|
-
def default_model_service(cls) ->
|
|
2113
|
-
return ModelServiceFields(
|
|
2751
|
+
def default_model_service(cls) -> "ModelServiceFields":
|
|
2752
|
+
return ModelServiceFields("defaultModelService")
|
|
2114
2753
|
|
|
2115
2754
|
@classmethod
|
|
2116
|
-
def activity(cls, *, timerange: Optional[TimeRange]=None) ->
|
|
2117
|
-
arguments: Dict[str, Dict[str, Any]] = {
|
|
2118
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
2119
|
-
return ActivityFields(
|
|
2755
|
+
def activity(cls, *, timerange: Optional[TimeRange] = None) -> "ActivityFields":
|
|
2756
|
+
arguments: Dict[str, Dict[str, Any]] = {"timerange": {"type": "TimeRange", "value": timerange}}
|
|
2757
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
|
|
2758
|
+
return ActivityFields("activity", arguments=cleared_arguments)
|
|
2120
2759
|
|
|
2121
2760
|
@classmethod
|
|
2122
|
-
def metrics(cls) ->
|
|
2123
|
-
return MetricWithContextFields(
|
|
2761
|
+
def metrics(cls) -> "MetricWithContextFields":
|
|
2762
|
+
return MetricWithContextFields("metrics")
|
|
2124
2763
|
|
|
2125
2764
|
@classmethod
|
|
2126
|
-
def metric(cls, metric: str) ->
|
|
2127
|
-
arguments: Dict[str, Dict[str, Any]] = {
|
|
2128
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
2129
|
-
return MetricWithContextFields(
|
|
2765
|
+
def metric(cls, metric: str) -> "MetricWithContextFields":
|
|
2766
|
+
arguments: Dict[str, Dict[str, Any]] = {"metric": {"type": "IdOrKey!", "value": metric}}
|
|
2767
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
|
|
2768
|
+
return MetricWithContextFields("metric", arguments=cleared_arguments)
|
|
2130
2769
|
|
|
2131
2770
|
@classmethod
|
|
2132
|
-
def ab_campaigns(cls, filter: AbCampaignFilter) ->
|
|
2133
|
-
arguments: Dict[str, Dict[str, Any]] = {
|
|
2134
|
-
cleared_arguments = {key: value for key, value in arguments.items() if value[
|
|
2135
|
-
return AbcampaignFields(
|
|
2771
|
+
def ab_campaigns(cls, filter: AbCampaignFilter) -> "AbcampaignFields":
|
|
2772
|
+
arguments: Dict[str, Dict[str, Any]] = {"filter": {"type": "AbCampaignFilter!", "value": filter}}
|
|
2773
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value["value"] is not None}
|
|
2774
|
+
return AbcampaignFields("abCampaigns", arguments=cleared_arguments)
|
|
2136
2775
|
|
|
2137
2776
|
@classmethod
|
|
2138
|
-
def widgets(cls) ->
|
|
2139
|
-
return WidgetFields(
|
|
2777
|
+
def widgets(cls) -> "WidgetFields":
|
|
2778
|
+
return WidgetFields("widgets")
|
|
2140
2779
|
|
|
2141
2780
|
@classmethod
|
|
2142
|
-
def metadata(cls) ->
|
|
2143
|
-
return UseCaseMetadataFields(
|
|
2144
|
-
|
|
2781
|
+
def metadata(cls) -> "UseCaseMetadataFields":
|
|
2782
|
+
return UseCaseMetadataFields("metadata")
|
|
2783
|
+
|
|
2784
|
+
permissions: "UseCaseGraphQLField" = UseCaseGraphQLField("permissions")
|
|
2145
2785
|
|
|
2146
2786
|
@classmethod
|
|
2147
|
-
def shares(cls) ->
|
|
2148
|
-
return ShareFields(
|
|
2787
|
+
def shares(cls) -> "ShareFields":
|
|
2788
|
+
return ShareFields("shares")
|
|
2149
2789
|
|
|
2150
2790
|
@classmethod
|
|
2151
|
-
def settings(cls) ->
|
|
2152
|
-
return SettingsFields(
|
|
2791
|
+
def settings(cls) -> "SettingsFields":
|
|
2792
|
+
return SettingsFields("settings")
|
|
2153
2793
|
|
|
2154
2794
|
@classmethod
|
|
2155
|
-
def label_usage(cls) ->
|
|
2156
|
-
return LabelUsageFields(
|
|
2795
|
+
def label_usage(cls) -> "LabelUsageFields":
|
|
2796
|
+
return LabelUsageFields("labelUsage")
|
|
2157
2797
|
|
|
2158
2798
|
@classmethod
|
|
2159
|
-
def tool_providers(cls) ->
|
|
2160
|
-
return ToolProviderFields(
|
|
2799
|
+
def tool_providers(cls) -> "ToolProviderFields":
|
|
2800
|
+
return ToolProviderFields("toolProviders")
|
|
2161
2801
|
|
|
2162
|
-
def fields(
|
|
2802
|
+
def fields(
|
|
2803
|
+
self,
|
|
2804
|
+
*subfields: Union[
|
|
2805
|
+
UseCaseGraphQLField,
|
|
2806
|
+
"AbcampaignFields",
|
|
2807
|
+
"ActivityFields",
|
|
2808
|
+
"LabelUsageFields",
|
|
2809
|
+
"MetricWithContextFields",
|
|
2810
|
+
"ModelFields",
|
|
2811
|
+
"ModelServiceFields",
|
|
2812
|
+
"SettingsFields",
|
|
2813
|
+
"ShareFields",
|
|
2814
|
+
"ToolProviderFields",
|
|
2815
|
+
"UseCaseMetadataFields",
|
|
2816
|
+
"WidgetFields",
|
|
2817
|
+
],
|
|
2818
|
+
) -> "UseCaseFields":
|
|
2163
2819
|
"""Subfields should come from the UseCaseFields class"""
|
|
2164
2820
|
self._subfields.extend(subfields)
|
|
2165
2821
|
return self
|
|
2166
2822
|
|
|
2167
|
-
def alias(self, alias: str) ->
|
|
2823
|
+
def alias(self, alias: str) -> "UseCaseFields":
|
|
2168
2824
|
self._alias = alias
|
|
2169
2825
|
return self
|
|
2170
2826
|
|
|
2827
|
+
|
|
2171
2828
|
class UseCaseItemFields(GraphQLField):
|
|
2172
2829
|
"""@private"""
|
|
2173
|
-
id: 'UseCaseItemGraphQLField' = UseCaseItemGraphQLField('id')
|
|
2174
|
-
key: 'UseCaseItemGraphQLField' = UseCaseItemGraphQLField('key')
|
|
2175
|
-
name: 'UseCaseItemGraphQLField' = UseCaseItemGraphQLField('name')
|
|
2176
|
-
description: 'UseCaseItemGraphQLField' = UseCaseItemGraphQLField('description')
|
|
2177
2830
|
|
|
2178
|
-
|
|
2831
|
+
id: "UseCaseItemGraphQLField" = UseCaseItemGraphQLField("id")
|
|
2832
|
+
key: "UseCaseItemGraphQLField" = UseCaseItemGraphQLField("key")
|
|
2833
|
+
name: "UseCaseItemGraphQLField" = UseCaseItemGraphQLField("name")
|
|
2834
|
+
description: "UseCaseItemGraphQLField" = UseCaseItemGraphQLField("description")
|
|
2835
|
+
|
|
2836
|
+
def fields(self, *subfields: UseCaseItemGraphQLField) -> "UseCaseItemFields":
|
|
2179
2837
|
"""Subfields should come from the UseCaseItemFields class"""
|
|
2180
2838
|
self._subfields.extend(subfields)
|
|
2181
2839
|
return self
|
|
2182
2840
|
|
|
2183
|
-
def alias(self, alias: str) ->
|
|
2841
|
+
def alias(self, alias: str) -> "UseCaseItemFields":
|
|
2184
2842
|
self._alias = alias
|
|
2185
2843
|
return self
|
|
2186
2844
|
|
|
2845
|
+
|
|
2187
2846
|
class UseCaseMetadataFields(GraphQLField):
|
|
2188
2847
|
"""@private"""
|
|
2189
2848
|
|
|
2190
2849
|
@classmethod
|
|
2191
|
-
def emoji(cls) ->
|
|
2192
|
-
return EmojiFields(
|
|
2850
|
+
def emoji(cls) -> "EmojiFields":
|
|
2851
|
+
return EmojiFields("emoji")
|
|
2193
2852
|
|
|
2194
|
-
def fields(self, *subfields: Union[UseCaseMetadataGraphQLField,
|
|
2853
|
+
def fields(self, *subfields: Union[UseCaseMetadataGraphQLField, "EmojiFields"]) -> "UseCaseMetadataFields":
|
|
2195
2854
|
"""Subfields should come from the UseCaseMetadataFields class"""
|
|
2196
2855
|
self._subfields.extend(subfields)
|
|
2197
2856
|
return self
|
|
2198
2857
|
|
|
2199
|
-
def alias(self, alias: str) ->
|
|
2858
|
+
def alias(self, alias: str) -> "UseCaseMetadataFields":
|
|
2200
2859
|
self._alias = alias
|
|
2201
2860
|
return self
|
|
2202
2861
|
|
|
2862
|
+
|
|
2203
2863
|
class UserFields(GraphQLField):
|
|
2204
2864
|
"""@private"""
|
|
2205
|
-
|
|
2206
|
-
|
|
2207
|
-
|
|
2208
|
-
|
|
2209
|
-
|
|
2210
|
-
|
|
2865
|
+
|
|
2866
|
+
id: "UserGraphQLField" = UserGraphQLField("id")
|
|
2867
|
+
email: "UserGraphQLField" = UserGraphQLField("email")
|
|
2868
|
+
name: "UserGraphQLField" = UserGraphQLField("name")
|
|
2869
|
+
created_at: "UserGraphQLField" = UserGraphQLField("createdAt")
|
|
2870
|
+
deleted: "UserGraphQLField" = UserGraphQLField("deleted")
|
|
2871
|
+
deleted_at: "UserGraphQLField" = UserGraphQLField("deletedAt")
|
|
2211
2872
|
|
|
2212
2873
|
@classmethod
|
|
2213
|
-
def teams(cls) ->
|
|
2214
|
-
return TeamWithroleFields(
|
|
2874
|
+
def teams(cls) -> "TeamWithroleFields":
|
|
2875
|
+
return TeamWithroleFields("teams")
|
|
2215
2876
|
|
|
2216
|
-
def fields(self, *subfields: Union[UserGraphQLField,
|
|
2877
|
+
def fields(self, *subfields: Union[UserGraphQLField, "TeamWithroleFields"]) -> "UserFields":
|
|
2217
2878
|
"""Subfields should come from the UserFields class"""
|
|
2218
2879
|
self._subfields.extend(subfields)
|
|
2219
2880
|
return self
|
|
2220
2881
|
|
|
2221
|
-
def alias(self, alias: str) ->
|
|
2882
|
+
def alias(self, alias: str) -> "UserFields":
|
|
2883
|
+
self._alias = alias
|
|
2884
|
+
return self
|
|
2885
|
+
|
|
2886
|
+
|
|
2887
|
+
class WebhookConfigFields(GraphQLField):
|
|
2888
|
+
"""@private"""
|
|
2889
|
+
|
|
2890
|
+
url: "WebhookConfigGraphQLField" = WebhookConfigGraphQLField("url")
|
|
2891
|
+
method: "WebhookConfigGraphQLField" = WebhookConfigGraphQLField("method")
|
|
2892
|
+
headers: "WebhookConfigGraphQLField" = WebhookConfigGraphQLField("headers")
|
|
2893
|
+
|
|
2894
|
+
def fields(self, *subfields: WebhookConfigGraphQLField) -> "WebhookConfigFields":
|
|
2895
|
+
"""Subfields should come from the WebhookConfigFields class"""
|
|
2896
|
+
self._subfields.extend(subfields)
|
|
2897
|
+
return self
|
|
2898
|
+
|
|
2899
|
+
def alias(self, alias: str) -> "WebhookConfigFields":
|
|
2222
2900
|
self._alias = alias
|
|
2223
2901
|
return self
|
|
2224
2902
|
|
|
2903
|
+
|
|
2225
2904
|
class WidgetFields(GraphQLField):
|
|
2226
2905
|
"""@private"""
|
|
2227
|
-
|
|
2228
|
-
|
|
2229
|
-
|
|
2906
|
+
|
|
2907
|
+
title: "WidgetGraphQLField" = WidgetGraphQLField("title")
|
|
2908
|
+
metric: "WidgetGraphQLField" = WidgetGraphQLField("metric")
|
|
2909
|
+
aggregation: "WidgetGraphQLField" = WidgetGraphQLField("aggregation")
|
|
2230
2910
|
|
|
2231
2911
|
@classmethod
|
|
2232
|
-
def unit(cls) ->
|
|
2233
|
-
return UnitConfigFields(
|
|
2912
|
+
def unit(cls) -> "UnitConfigFields":
|
|
2913
|
+
return UnitConfigFields("unit")
|
|
2234
2914
|
|
|
2235
|
-
def fields(self, *subfields: Union[WidgetGraphQLField,
|
|
2915
|
+
def fields(self, *subfields: Union[WidgetGraphQLField, "UnitConfigFields"]) -> "WidgetFields":
|
|
2236
2916
|
"""Subfields should come from the WidgetFields class"""
|
|
2237
2917
|
self._subfields.extend(subfields)
|
|
2238
2918
|
return self
|
|
2239
2919
|
|
|
2240
|
-
def alias(self, alias: str) ->
|
|
2920
|
+
def alias(self, alias: str) -> "WidgetFields":
|
|
2241
2921
|
self._alias = alias
|
|
2242
|
-
return self
|
|
2922
|
+
return self
|