adaptive-sdk 0.1.14__py3-none-any.whl → 0.12.1__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -1,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 AbcampaignGraphQLField, AbReportGraphQLField, AbVariantReportComparisonGraphQLField, AbVariantReportGraphQLField, ActivityGraphQLField, ActivityOutputGraphQLField, ApiKeyGraphQLField, ArtifactByproductsUnion, AuthProviderGraphQLField, BatchInferenceJobStageOutputGraphQLField, BillingUsageGraphQLField, ChatMessageGraphQLField, ComparisonFeedbackGraphQLField, CompletionConnectionGraphQLField, CompletionEdgeGraphQLField, CompletionGraphQLField, CompletionGroupDataConnectionGraphQLField, CompletionGroupDataEdgeGraphQLField, CompletionGroupDataGraphQLField, CompletionGroupFeedbackStatsGraphQLField, CompletionHistoryEntryOuputGraphQLField, CompletionLabelGraphQLField, CompletionMetadataGraphQLField, ComputePoolGraphQLField, ContractGraphQLField, CustomConfigOutputGraphQLField, CustomRecipeGraphQLField, CustomRecipeJobDetailsGraphQLField, DatasetByproductsGraphQLField, DatasetGraphQLField, DatasetMetricUsageGraphQLField, DatasetProgressGraphQLField, DatasetUploadProcessingStatusGraphQLField, DatasetValidationOutputGraphQLField, DeleteConfirmGraphQLField, DirectFeedbackGraphQLField, EmojiGraphQLField, EvalJobStageOutputGraphQLField, EvaluationByproductsGraphQLField, EvaluationResultGraphQLField, GlobalUsageGraphQLField, GpuAllocationGraphQLField, GraderConfigUnion, GraderGraphQLField, HarmonyGroupGraphQLField, InteractionOutputGraphQLField, IntervalGraphQLField, JobArtifactGraphQLField, JobConnectionGraphQLField, JobEdgeGraphQLField, JobGraphQLField, JobStageInfoOutputUnion, JobStageOutputGraphQLField, JudgeConfigOutputGraphQLField, JudgeExampleGraphQLField, JudgeGraphQLField, LabelGraphQLField, LabelKeyUsageGraphQLField, LabelUsageGraphQLField, LabelValueUsageGraphQLField, MetaObjectGraphQLField, MetricActivityGraphQLField, MetricGraphQLField, MetricWithContextGraphQLField, ModelByproductsGraphQLField, ModelComputeConfigOutputGraphQLField, ModelGraphQLField, ModelPlacementOutputGraphQLField, ModelServiceGraphQLField, PageInfoGraphQLField, PrebuiltConfigDefinitionGraphQLField, PrebuiltConfigOutputGraphQLField, PrebuiltCriteriaGraphQLField, ProviderListGraphQLField, RemoteConfigOutputGraphQLField, RemoteEnvGraphQLField, RemoteEnvTestOfflineGraphQLField, RemoteEnvTestOnlineGraphQLField, RoleGraphQLField, SearchResultGraphQLField, SessionGraphQLField, SettingsGraphQLField, ShareGraphQLField, SystemPromptTemplateGraphQLField, TeamGraphQLField, TeamMemberGraphQLField, TeamWithroleGraphQLField, TimeseriesGraphQLField, ToolProviderGraphQLField, TrainingJobStageOutputGraphQLField, TrendResultGraphQLField, UnitConfigGraphQLField, UsageAggregateItemGraphQLField, UsageAggregatePerUseCaseItemGraphQLField, UsageGraphQLField, UsageStatsByModelGraphQLField, UsageStatsGraphQLField, UseCaseGraphQLField, UseCaseItemGraphQLField, UseCaseMetadataGraphQLField, UserGraphQLField, WidgetGraphQLField
4
- from .input_types import AbCampaignFilter, ArtifactFilter, CursorPageInput, FeedbackFilterInput, ListCompletionsFilterInput, MetricTrendInput, ModelFilter, ModelServiceFilter, OrderPair, TimeRange, TimeseriesInput, UseCaseFilter
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
- p_value: 'AbReportGraphQLField' = AbReportGraphQLField('pValue')
125
+
126
+ p_value: "AbReportGraphQLField" = AbReportGraphQLField("pValue")
9
127
 
10
128
  @classmethod
11
- def variants(cls) -> 'AbVariantReportFields':
12
- return AbVariantReportFields('variants')
129
+ def variants(cls) -> "AbVariantReportFields":
130
+ return AbVariantReportFields("variants")
13
131
 
14
- def fields(self, *subfields: Union[AbReportGraphQLField, 'AbVariantReportFields']) -> 'AbReportFields':
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) -> 'AbReportFields':
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) -> 'ModelServiceFields':
28
- return ModelServiceFields('variant')
146
+ def variant(cls) -> "ModelServiceFields":
147
+ return ModelServiceFields("variant")
29
148
 
30
149
  @classmethod
31
- def interval(cls) -> 'IntervalFields':
32
- return IntervalFields('interval')
33
- mean: 'AbVariantReportGraphQLField' = AbVariantReportGraphQLField('mean')
34
- feedbacks: 'AbVariantReportGraphQLField' = AbVariantReportGraphQLField('feedbacks')
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) -> 'AbVariantReportComparisonFields':
38
- return AbVariantReportComparisonFields('comparisons')
157
+ def comparisons(cls) -> "AbVariantReportComparisonFields":
158
+ return AbVariantReportComparisonFields("comparisons")
39
159
 
40
- def fields(self, *subfields: Union[AbVariantReportGraphQLField, 'AbVariantReportComparisonFields', 'IntervalFields', 'ModelServiceFields']) -> 'AbVariantReportFields':
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) -> 'AbVariantReportFields':
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) -> 'ModelServiceFields':
54
- return ModelServiceFields('variant')
55
- feedbacks: 'AbVariantReportComparisonGraphQLField' = AbVariantReportComparisonGraphQLField('feedbacks')
56
- wins: 'AbVariantReportComparisonGraphQLField' = AbVariantReportComparisonGraphQLField('wins')
57
- losses: 'AbVariantReportComparisonGraphQLField' = AbVariantReportComparisonGraphQLField('losses')
58
- ties_good: 'AbVariantReportComparisonGraphQLField' = AbVariantReportComparisonGraphQLField('tiesGood')
59
- ties_bad: 'AbVariantReportComparisonGraphQLField' = AbVariantReportComparisonGraphQLField('tiesBad')
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(self, *subfields: Union[AbVariantReportComparisonGraphQLField, 'ModelServiceFields']) -> 'AbVariantReportComparisonFields':
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) -> 'AbVariantReportComparisonFields':
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
- id: 'AbcampaignGraphQLField' = AbcampaignGraphQLField('id')
73
- key: 'AbcampaignGraphQLField' = AbcampaignGraphQLField('key')
74
- name: 'AbcampaignGraphQLField' = AbcampaignGraphQLField('name')
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) -> 'MetricFields':
78
- return MetricFields('metric')
208
+ def metric(cls) -> "MetricFields":
209
+ return MetricFields("metric")
79
210
 
80
211
  @classmethod
81
- def use_case(cls) -> 'UseCaseFields':
82
- return UseCaseFields('useCase')
83
- auto_deploy: 'AbcampaignGraphQLField' = AbcampaignGraphQLField('autoDeploy')
84
- status: 'AbcampaignGraphQLField' = AbcampaignGraphQLField('status')
85
- feedback_type: 'AbcampaignGraphQLField' = AbcampaignGraphQLField('feedbackType')
86
- traffic_split: 'AbcampaignGraphQLField' = AbcampaignGraphQLField('trafficSplit')
87
- begin_date: 'AbcampaignGraphQLField' = AbcampaignGraphQLField('beginDate')
88
- end_date: 'AbcampaignGraphQLField' = AbcampaignGraphQLField('endDate')
89
- created_at: 'AbcampaignGraphQLField' = AbcampaignGraphQLField('createdAt')
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) -> 'AbReportFields':
93
- return AbReportFields('report')
224
+ def report(cls) -> "AbReportFields":
225
+ return AbReportFields("report")
94
226
 
95
227
  @classmethod
96
- def models(cls) -> 'ModelServiceFields':
97
- return ModelServiceFields('models')
98
- feedbacks: 'AbcampaignGraphQLField' = AbcampaignGraphQLField('feedbacks')
99
- has_enough_feedbacks: 'AbcampaignGraphQLField' = AbcampaignGraphQLField('hasEnoughFeedbacks')
228
+ def models(cls) -> "ModelServiceFields":
229
+ return ModelServiceFields("models")
100
230
 
101
- def fields(self, *subfields: Union[AbcampaignGraphQLField, 'AbReportFields', 'MetricFields', 'ModelServiceFields', 'UseCaseFields']) -> 'AbcampaignFields':
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) -> 'AbcampaignFields':
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) -> 'InteractionOutputFields':
115
- return InteractionOutputFields('interactions')
253
+ def interactions(cls) -> "InteractionOutputFields":
254
+ return InteractionOutputFields("interactions")
116
255
 
117
256
  @classmethod
118
- def feedbacks(cls) -> 'ActivityOutputFields':
119
- return ActivityOutputFields('feedbacks')
257
+ def feedbacks(cls) -> "ActivityOutputFields":
258
+ return ActivityOutputFields("feedbacks")
120
259
 
121
- def fields(self, *subfields: Union[ActivityGraphQLField, 'ActivityOutputFields', 'InteractionOutputFields']) -> 'ActivityFields':
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) -> 'ActivityFields':
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
- def fields(self, *subfields: ActivityOutputGraphQLField) -> 'ActivityOutputFields':
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) -> 'ActivityOutputFields':
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
- def fields(self, *subfields: ApiKeyGraphQLField) -> 'ApiKeyFields':
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) -> 'ApiKeyFields':
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
- def fields(self, *subfields: AuthProviderGraphQLField) -> 'AuthProviderFields':
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) -> 'AuthProviderFields':
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
- def fields(self, *subfields: BatchInferenceJobStageOutputGraphQLField) -> 'BatchInferenceJobStageOutputFields':
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) -> 'BatchInferenceJobStageOutputFields':
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
- now: 'BillingUsageGraphQLField' = BillingUsageGraphQLField('now')
194
- start: 'BillingUsageGraphQLField' = BillingUsageGraphQLField('start')
195
- end: 'BillingUsageGraphQLField' = BillingUsageGraphQLField('end')
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) -> 'UsageStatsFields':
199
- return UsageStatsFields('usage')
355
+ def usage(cls) -> "UsageStatsFields":
356
+ return UsageStatsFields("usage")
200
357
 
201
358
  @classmethod
202
- def projection(cls) -> 'UsageStatsFields':
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('projection')
361
+ return UsageStatsFields("projection")
205
362
 
206
- def fields(self, *subfields: Union[BillingUsageGraphQLField, 'UsageStatsFields']) -> 'BillingUsageFields':
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) -> 'BillingUsageFields':
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
- def fields(self, *subfields: ChatMessageGraphQLField) -> 'ChatMessageFields':
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) -> 'ChatMessageFields':
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
- id: 'ComparisonFeedbackGraphQLField' = ComparisonFeedbackGraphQLField('id')
232
- created_at: 'ComparisonFeedbackGraphQLField' = ComparisonFeedbackGraphQLField('createdAt')
391
+
392
+ id: "ComparisonFeedbackGraphQLField" = ComparisonFeedbackGraphQLField("id")
393
+ created_at: "ComparisonFeedbackGraphQLField" = ComparisonFeedbackGraphQLField("createdAt")
233
394
 
234
395
  @classmethod
235
- def usecase(cls) -> 'UseCaseFields':
236
- return UseCaseFields('usecase')
396
+ def usecase(cls) -> "UseCaseFields":
397
+ return UseCaseFields("usecase")
237
398
 
238
399
  @classmethod
239
- def metric(cls) -> 'MetricFields':
240
- return MetricFields('metric')
400
+ def metric(cls) -> "MetricFields":
401
+ return MetricFields("metric")
241
402
 
242
403
  @classmethod
243
- def prefered_completion(cls) -> 'CompletionFields':
244
- return CompletionFields('preferedCompletion')
404
+ def prefered_completion(cls) -> "CompletionFields":
405
+ return CompletionFields("preferedCompletion")
245
406
 
246
407
  @classmethod
247
- def other_completion(cls) -> 'CompletionFields':
248
- return CompletionFields('otherCompletion')
408
+ def other_completion(cls) -> "CompletionFields":
409
+ return CompletionFields("otherCompletion")
249
410
 
250
- def fields(self, *subfields: Union[ComparisonFeedbackGraphQLField, 'CompletionFields', 'MetricFields', 'UseCaseFields']) -> 'ComparisonFeedbackFields':
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) -> 'ComparisonFeedbackFields':
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
- id: 'CompletionGraphQLField' = CompletionGraphQLField('id')
262
- prompt_hash: 'CompletionGraphQLField' = CompletionGraphQLField('promptHash')
425
+
426
+ id: "CompletionGraphQLField" = CompletionGraphQLField("id")
427
+ prompt_hash: "CompletionGraphQLField" = CompletionGraphQLField("promptHash")
263
428
 
264
429
  @classmethod
265
- def chat_messages(cls) -> 'ChatMessageFields':
266
- return ChatMessageFields('chatMessages')
430
+ def chat_messages(cls) -> "ChatMessageFields":
431
+ return ChatMessageFields("chatMessages")
267
432
 
268
433
  @classmethod
269
- def completion(cls, *, max_length: Optional[int]=None) -> 'CompletionGraphQLField':
270
- arguments: Dict[str, Dict[str, Any]] = {'maxLength': {'type': 'Int', 'value': max_length}}
271
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
272
- return CompletionGraphQLField('completion', arguments=cleared_arguments)
273
- source: 'CompletionGraphQLField' = CompletionGraphQLField('source')
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) -> 'ModelServiceFields':
277
- return ModelServiceFields('modelService')
442
+ def model_service(cls) -> "ModelServiceFields":
443
+ return ModelServiceFields("modelService")
278
444
 
279
445
  @classmethod
280
- def model(cls) -> 'ModelFields':
281
- return ModelFields('model')
446
+ def model(cls) -> "ModelFields":
447
+ return ModelFields("model")
282
448
 
283
449
  @classmethod
284
- def direct_feedbacks(cls, *, filter: Optional[FeedbackFilterInput]=None) -> 'DirectFeedbackFields':
285
- arguments: Dict[str, Dict[str, Any]] = {'filter': {'type': 'FeedbackFilterInput', 'value': filter}}
286
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
287
- return DirectFeedbackFields('directFeedbacks', arguments=cleared_arguments)
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) -> 'ComparisonFeedbackFields':
291
- return ComparisonFeedbackFields('comparisonFeedbacks')
456
+ def comparison_feedbacks(cls) -> "ComparisonFeedbackFields":
457
+ return ComparisonFeedbackFields("comparisonFeedbacks")
292
458
 
293
459
  @classmethod
294
- def session(cls) -> 'SessionFields':
295
- return SessionFields('session')
460
+ def session(cls) -> "SessionFields":
461
+ return SessionFields("session")
296
462
 
297
463
  @classmethod
298
- def history(cls) -> 'CompletionHistoryEntryOuputFields':
299
- return CompletionHistoryEntryOuputFields('history')
464
+ def history(cls) -> "CompletionHistoryEntryOuputFields":
465
+ return CompletionHistoryEntryOuputFields("history")
300
466
 
301
467
  @classmethod
302
- def labels(cls, with_protected: bool) -> 'CompletionLabelFields':
303
- arguments: Dict[str, Dict[str, Any]] = {'withProtected': {'type': 'Boolean!', 'value': with_protected}}
304
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
305
- return CompletionLabelFields('labels', arguments=cleared_arguments)
306
- created_at: 'CompletionGraphQLField' = CompletionGraphQLField('createdAt')
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) -> 'CompletionGraphQLField':
310
- arguments: Dict[str, Dict[str, Any]] = {'filter': {'type': 'ListCompletionsFilterInput!', 'value': filter}}
311
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
312
- return CompletionGraphQLField('siblingsCount', arguments=cleared_arguments)
313
- has_user_metadata: 'CompletionGraphQLField' = CompletionGraphQLField('hasUserMetadata')
314
- user_metadata: 'CompletionGraphQLField' = CompletionGraphQLField('userMetadata')
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) -> 'CompletionMetadataFields':
318
- return CompletionMetadataFields('metadata')
319
- can_edit: 'CompletionGraphQLField' = CompletionGraphQLField('canEdit')
485
+ def metadata(cls) -> "CompletionMetadataFields":
486
+ return CompletionMetadataFields("metadata")
320
487
 
321
- def fields(self, *subfields: Union[CompletionGraphQLField, 'ChatMessageFields', 'ComparisonFeedbackFields', 'CompletionHistoryEntryOuputFields', 'CompletionLabelFields', 'CompletionMetadataFields', 'DirectFeedbackFields', 'ModelFields', 'ModelServiceFields', 'SessionFields']) -> 'CompletionFields':
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) -> 'CompletionFields':
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) -> 'PageInfoFields':
518
+ def page_info(cls) -> "PageInfoFields":
335
519
  """Information to aid in pagination."""
336
- return PageInfoFields('pageInfo')
520
+ return PageInfoFields("pageInfo")
337
521
 
338
522
  @classmethod
339
- def edges(cls) -> 'CompletionEdgeFields':
523
+ def edges(cls) -> "CompletionEdgeFields":
340
524
  """A list of edges."""
341
- return CompletionEdgeFields('edges')
525
+ return CompletionEdgeFields("edges")
342
526
 
343
527
  @classmethod
344
- def nodes(cls) -> 'CompletionFields':
528
+ def nodes(cls) -> "CompletionFields":
345
529
  """A list of nodes."""
346
- return CompletionFields('nodes')
347
- total_count: 'CompletionConnectionGraphQLField' = CompletionConnectionGraphQLField('totalCount')
530
+ return CompletionFields("nodes")
348
531
 
349
- def fields(self, *subfields: Union[CompletionConnectionGraphQLField, 'CompletionEdgeFields', 'CompletionFields', 'PageInfoFields']) -> 'CompletionConnectionFields':
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) -> 'CompletionConnectionFields':
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) -> 'CompletionFields':
554
+ def node(cls) -> "CompletionFields":
364
555
  """The item at the end of the edge"""
365
- return CompletionFields('node')
366
- cursor: 'CompletionEdgeGraphQLField' = CompletionEdgeGraphQLField('cursor')
367
- 'A cursor for use in pagination'
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, 'CompletionFields']) -> 'CompletionEdgeFields':
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) -> 'CompletionEdgeFields':
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
- key: 'CompletionGroupDataGraphQLField' = CompletionGroupDataGraphQLField('key')
381
- count: 'CompletionGroupDataGraphQLField' = CompletionGroupDataGraphQLField('count')
573
+
574
+ key: "CompletionGroupDataGraphQLField" = CompletionGroupDataGraphQLField("key")
575
+ count: "CompletionGroupDataGraphQLField" = CompletionGroupDataGraphQLField("count")
382
576
 
383
577
  @classmethod
384
- def direct_feedbacks_stats(cls) -> 'CompletionGroupFeedbackStatsFields':
385
- return CompletionGroupFeedbackStatsFields('directFeedbacksStats')
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]) -> 'CompletionConnectionFields':
389
- arguments: Dict[str, Dict[str, Any]] = {'page': {'type': 'CursorPageInput!', 'value': page}, 'order': {'type': '[OrderPair!]!', 'value': order}}
390
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
391
- return CompletionConnectionFields('completions', arguments=cleared_arguments)
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(self, *subfields: Union[CompletionGroupDataGraphQLField, 'CompletionConnectionFields', 'CompletionGroupFeedbackStatsFields']) -> 'CompletionGroupDataFields':
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) -> 'CompletionGroupDataFields':
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) -> 'PageInfoFields':
609
+ def page_info(cls) -> "PageInfoFields":
407
610
  """Information to aid in pagination."""
408
- return PageInfoFields('pageInfo')
611
+ return PageInfoFields("pageInfo")
409
612
 
410
613
  @classmethod
411
- def edges(cls) -> 'CompletionGroupDataEdgeFields':
614
+ def edges(cls) -> "CompletionGroupDataEdgeFields":
412
615
  """A list of edges."""
413
- return CompletionGroupDataEdgeFields('edges')
616
+ return CompletionGroupDataEdgeFields("edges")
414
617
 
415
618
  @classmethod
416
- def nodes(cls) -> 'CompletionGroupDataFields':
619
+ def nodes(cls) -> "CompletionGroupDataFields":
417
620
  """A list of nodes."""
418
- return CompletionGroupDataFields('nodes')
419
- group_by: 'CompletionGroupDataConnectionGraphQLField' = CompletionGroupDataConnectionGraphQLField('groupBy')
420
- total_count: 'CompletionGroupDataConnectionGraphQLField' = CompletionGroupDataConnectionGraphQLField('totalCount')
421
-
422
- def fields(self, *subfields: Union[CompletionGroupDataConnectionGraphQLField, 'CompletionGroupDataEdgeFields', 'CompletionGroupDataFields', 'PageInfoFields']) -> 'CompletionGroupDataConnectionFields':
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) -> 'CompletionGroupDataConnectionFields':
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) -> 'CompletionGroupDataFields':
649
+ def node(cls) -> "CompletionGroupDataFields":
437
650
  """The item at the end of the edge"""
438
- return CompletionGroupDataFields('node')
439
- cursor: 'CompletionGroupDataEdgeGraphQLField' = CompletionGroupDataEdgeGraphQLField('cursor')
440
- 'A cursor for use in pagination'
651
+ return CompletionGroupDataFields("node")
441
652
 
442
- def fields(self, *subfields: Union[CompletionGroupDataEdgeGraphQLField, 'CompletionGroupDataFields']) -> 'CompletionGroupDataEdgeFields':
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) -> 'CompletionGroupDataEdgeFields':
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) -> 'MetricFields':
456
- return MetricFields('metric')
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
- def fields(self, *subfields: Union[CompletionGroupFeedbackStatsGraphQLField, 'MetricFields']) -> 'CompletionGroupFeedbackStatsFields':
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) -> 'CompletionGroupFeedbackStatsFields':
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
- def fields(self, *subfields: CompletionHistoryEntryOuputGraphQLField) -> 'CompletionHistoryEntryOuputFields':
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) -> 'CompletionHistoryEntryOuputFields':
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
- def fields(self, *subfields: CompletionLabelGraphQLField) -> 'CompletionLabelFields':
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) -> 'CompletionLabelFields':
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
- parameters: 'CompletionMetadataGraphQLField' = CompletionMetadataGraphQLField('parameters')
507
- timings: 'CompletionMetadataGraphQLField' = CompletionMetadataGraphQLField('timings')
508
- system: 'CompletionMetadataGraphQLField' = CompletionMetadataGraphQLField('system')
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) -> 'UsageFields':
512
- return UsageFields('usage')
737
+ def usage(cls) -> "UsageFields":
738
+ return UsageFields("usage")
513
739
 
514
- def fields(self, *subfields: Union[CompletionMetadataGraphQLField, 'UsageFields']) -> 'CompletionMetadataFields':
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) -> 'CompletionMetadataFields':
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
- id: 'ComputePoolGraphQLField' = ComputePoolGraphQLField('id')
526
- key: 'ComputePoolGraphQLField' = ComputePoolGraphQLField('key')
527
- name: 'ComputePoolGraphQLField' = ComputePoolGraphQLField('name')
528
- created_at: 'ComputePoolGraphQLField' = ComputePoolGraphQLField('createdAt')
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) -> 'HarmonyGroupFields':
532
- return HarmonyGroupFields('allHarmonyGroups')
759
+ def all_harmony_groups(cls) -> "HarmonyGroupFields":
760
+ return HarmonyGroupFields("allHarmonyGroups")
533
761
 
534
762
  @classmethod
535
- def harmony_groups(cls) -> 'HarmonyGroupFields':
536
- return HarmonyGroupFields('harmonyGroups')
537
- capabilities: 'ComputePoolGraphQLField' = ComputePoolGraphQLField('capabilities')
763
+ def harmony_groups(cls) -> "HarmonyGroupFields":
764
+ return HarmonyGroupFields("harmonyGroups")
538
765
 
539
- def fields(self, *subfields: Union[ComputePoolGraphQLField, 'HarmonyGroupFields']) -> 'ComputePoolFields':
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) -> 'ComputePoolFields':
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
- quota: 'ContractGraphQLField' = ContractGraphQLField('quota')
551
- start_date: 'ContractGraphQLField' = ContractGraphQLField('startDate')
552
- end_date: 'ContractGraphQLField' = ContractGraphQLField('endDate')
553
- cycle: 'ContractGraphQLField' = ContractGraphQLField('cycle')
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) -> 'BillingUsageFields':
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]] = {'now': {'type': 'InputDatetime', 'value': now}}
560
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
561
- return BillingUsageFields('usage', arguments=cleared_arguments)
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, 'BillingUsageFields']) -> 'ContractFields':
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) -> 'ContractFields':
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
- def fields(self, *subfields: CustomConfigOutputGraphQLField) -> 'CustomConfigOutputFields':
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) -> 'CustomConfigOutputFields':
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
- id: 'CustomRecipeGraphQLField' = CustomRecipeGraphQLField('id')
588
- key: 'CustomRecipeGraphQLField' = CustomRecipeGraphQLField('key')
589
- name: 'CustomRecipeGraphQLField' = CustomRecipeGraphQLField('name')
590
- is_multifile: 'CustomRecipeGraphQLField' = CustomRecipeGraphQLField('isMultifile')
591
- editable: 'CustomRecipeGraphQLField' = CustomRecipeGraphQLField('editable')
592
- hidden: 'CustomRecipeGraphQLField' = CustomRecipeGraphQLField('hidden')
593
- builtin: 'CustomRecipeGraphQLField' = CustomRecipeGraphQLField('builtin')
594
- global_: 'CustomRecipeGraphQLField' = CustomRecipeGraphQLField('global')
595
- created_at: 'CustomRecipeGraphQLField' = CustomRecipeGraphQLField('createdAt')
596
- content: 'CustomRecipeGraphQLField' = CustomRecipeGraphQLField('content')
597
- input_schema: 'CustomRecipeGraphQLField' = CustomRecipeGraphQLField('inputSchema')
598
- json_schema: 'CustomRecipeGraphQLField' = CustomRecipeGraphQLField('jsonSchema')
599
- content_hash: 'CustomRecipeGraphQLField' = CustomRecipeGraphQLField('contentHash')
600
- description: 'CustomRecipeGraphQLField' = CustomRecipeGraphQLField('description')
601
-
602
- @classmethod
603
- def labels(cls, with_protected: bool) -> 'LabelFields':
604
- arguments: Dict[str, Dict[str, Any]] = {'withProtected': {'type': 'Boolean!', 'value': with_protected}}
605
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
606
- return LabelFields('labels', arguments=cleared_arguments)
607
- updated_at: 'CustomRecipeGraphQLField' = CustomRecipeGraphQLField('updatedAt')
608
-
609
- @classmethod
610
- def created_by(cls) -> 'UserFields':
611
- return UserFields('createdBy')
612
-
613
- def fields(self, *subfields: Union[CustomRecipeGraphQLField, 'LabelFields', 'UserFields']) -> 'CustomRecipeFields':
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) -> 'CustomRecipeFields':
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
- args: 'CustomRecipeJobDetailsGraphQLField' = CustomRecipeJobDetailsGraphQLField('args')
625
- recipe_hash: 'CustomRecipeJobDetailsGraphQLField' = CustomRecipeJobDetailsGraphQLField('recipeHash')
861
+
862
+ args: "CustomRecipeJobDetailsGraphQLField" = CustomRecipeJobDetailsGraphQLField("args")
863
+ recipe_hash: "CustomRecipeJobDetailsGraphQLField" = CustomRecipeJobDetailsGraphQLField("recipeHash")
626
864
 
627
865
  @classmethod
628
- def artifacts(cls) -> 'JobArtifactFields':
629
- return JobArtifactFields('artifacts')
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
- def fields(self, *subfields: Union[CustomRecipeJobDetailsGraphQLField, 'JobArtifactFields']) -> 'CustomRecipeJobDetailsFields':
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) -> 'CustomRecipeJobDetailsFields':
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
- id: 'DatasetGraphQLField' = DatasetGraphQLField('id')
646
- key: 'DatasetGraphQLField' = DatasetGraphQLField('key')
647
- name: 'DatasetGraphQLField' = DatasetGraphQLField('name')
648
- created_at: 'DatasetGraphQLField' = DatasetGraphQLField('createdAt')
649
- kind: 'DatasetGraphQLField' = DatasetGraphQLField('kind')
650
- records: 'DatasetGraphQLField' = DatasetGraphQLField('records')
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) -> 'DatasetMetricUsageFields':
654
- return DatasetMetricUsageFields('metricsUsage')
655
- source: 'DatasetGraphQLField' = DatasetGraphQLField('source')
656
- status: 'DatasetGraphQLField' = DatasetGraphQLField('status')
657
- deleted: 'DatasetGraphQLField' = DatasetGraphQLField('deleted')
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) -> 'DatasetProgressFields':
661
- return DatasetProgressFields('progress')
662
- download_url: 'DatasetGraphQLField' = DatasetGraphQLField('downloadUrl')
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) -> 'UseCaseFields':
666
- return UseCaseFields('useCase')
910
+ def use_case(cls) -> "UseCaseFields":
911
+ return UseCaseFields("useCase")
667
912
 
668
- def fields(self, *subfields: Union[DatasetGraphQLField, 'DatasetMetricUsageFields', 'DatasetProgressFields', 'UseCaseFields']) -> 'DatasetFields':
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) -> 'DatasetFields':
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) -> 'DatasetFields':
682
- return DatasetFields('dataset')
930
+ def dataset(cls) -> "DatasetFields":
931
+ return DatasetFields("dataset")
683
932
 
684
- def fields(self, *subfields: Union[DatasetByproductsGraphQLField, 'DatasetFields']) -> 'DatasetByproductsFields':
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) -> 'DatasetByproductsFields':
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) -> 'MetricFields':
698
- return MetricFields('metric')
699
- feedback_count: 'DatasetMetricUsageGraphQLField' = DatasetMetricUsageGraphQLField('feedbackCount')
700
- comparison_count: 'DatasetMetricUsageGraphQLField' = DatasetMetricUsageGraphQLField('comparisonCount')
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, 'MetricFields']) -> 'DatasetMetricUsageFields':
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) -> 'DatasetMetricUsageFields':
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
- def fields(self, *subfields: DatasetProgressGraphQLField) -> 'DatasetProgressFields':
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) -> 'DatasetProgressFields':
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
- def fields(self, *subfields: DatasetUploadProcessingStatusGraphQLField) -> 'DatasetUploadProcessingStatusFields':
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) -> 'DatasetUploadProcessingStatusFields':
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
- def fields(self, *subfields: DatasetValidationOutputGraphQLField) -> 'DatasetValidationOutputFields':
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) -> 'DatasetValidationOutputFields':
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
- def fields(self, *subfields: DeleteConfirmGraphQLField) -> 'DeleteConfirmFields':
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) -> 'DeleteConfirmFields':
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
- id: 'DirectFeedbackGraphQLField' = DirectFeedbackGraphQLField('id')
777
- value: 'DirectFeedbackGraphQLField' = DirectFeedbackGraphQLField('value')
778
- user_id: 'DirectFeedbackGraphQLField' = DirectFeedbackGraphQLField('userId')
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) -> 'MetricFields':
782
- return MetricFields('metric')
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
- def fields(self, *subfields: Union[DirectFeedbackGraphQLField, 'MetricFields']) -> 'DirectFeedbackFields':
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) -> 'DirectFeedbackFields':
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
- def fields(self, *subfields: EmojiGraphQLField) -> 'EmojiFields':
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) -> 'EmojiFields':
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
- def fields(self, *subfields: EvalJobStageOutputGraphQLField) -> 'EvalJobStageOutputFields':
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) -> 'EvalJobStageOutputFields':
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) -> 'EvaluationResultFields':
829
- return EvaluationResultFields('evalResults')
1097
+ def eval_results(cls) -> "EvaluationResultFields":
1098
+ return EvaluationResultFields("evalResults")
830
1099
 
831
- def fields(self, *subfields: Union[EvaluationByproductsGraphQLField, 'EvaluationResultFields']) -> 'EvaluationByproductsFields':
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) -> 'EvaluationByproductsFields':
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) -> 'ModelServiceFields':
845
- return ModelServiceFields('modelService')
1116
+ def model_service(cls) -> "ModelServiceFields":
1117
+ return ModelServiceFields("modelService")
846
1118
 
847
1119
  @classmethod
848
- def dataset(cls) -> 'DatasetFields':
849
- return DatasetFields('dataset')
1120
+ def dataset(cls) -> "DatasetFields":
1121
+ return DatasetFields("dataset")
850
1122
 
851
1123
  @classmethod
852
- def metric(cls) -> 'MetricFields':
853
- return MetricFields('metric')
854
- mean: 'EvaluationResultGraphQLField' = EvaluationResultGraphQLField('mean')
855
- min: 'EvaluationResultGraphQLField' = EvaluationResultGraphQLField('min')
856
- max: 'EvaluationResultGraphQLField' = EvaluationResultGraphQLField('max')
857
- stddev: 'EvaluationResultGraphQLField' = EvaluationResultGraphQLField('stddev')
858
- sum: 'EvaluationResultGraphQLField' = EvaluationResultGraphQLField('sum')
859
- count: 'EvaluationResultGraphQLField' = EvaluationResultGraphQLField('count')
860
- sum_squared: 'EvaluationResultGraphQLField' = EvaluationResultGraphQLField('sumSquared')
861
- job_id: 'EvaluationResultGraphQLField' = EvaluationResultGraphQLField('jobId')
862
- artifact_id: 'EvaluationResultGraphQLField' = EvaluationResultGraphQLField('artifactId')
863
- feedback_count: 'EvaluationResultGraphQLField' = EvaluationResultGraphQLField('feedbackCount')
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) -> 'JudgeFields':
867
- return JudgeFields('judge')
1139
+ def judge(cls) -> "JudgeFields":
1140
+ return JudgeFields("judge")
868
1141
 
869
1142
  @classmethod
870
- def grader(cls) -> 'GraderFields':
871
- return GraderFields('grader')
1143
+ def grader(cls) -> "GraderFields":
1144
+ return GraderFields("grader")
872
1145
 
873
- def fields(self, *subfields: Union[EvaluationResultGraphQLField, 'DatasetFields', 'GraderFields', 'JudgeFields', 'MetricFields', 'ModelServiceFields']) -> 'EvaluationResultFields':
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) -> 'EvaluationResultFields':
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) -> 'UsageStatsFields':
887
- return UsageStatsFields('total')
1187
+ def total(cls) -> "UsageStatsFields":
1188
+ return UsageStatsFields("total")
888
1189
 
889
1190
  @classmethod
890
- def adaptive_models(cls) -> 'UsageStatsFields':
891
- return UsageStatsFields('adaptiveModels')
1191
+ def adaptive_models(cls) -> "UsageStatsFields":
1192
+ return UsageStatsFields("adaptiveModels")
892
1193
 
893
1194
  @classmethod
894
- def external_models(cls) -> 'UsageStatsFields':
895
- return UsageStatsFields('externalModels')
1195
+ def external_models(cls) -> "UsageStatsFields":
1196
+ return UsageStatsFields("externalModels")
896
1197
 
897
1198
  @classmethod
898
- def by_model(cls) -> 'UsageStatsByModelFields':
899
- return UsageStatsByModelFields('byModel')
900
- signature: 'GlobalUsageGraphQLField' = GlobalUsageGraphQLField('signature')
1199
+ def by_model(cls) -> "UsageStatsByModelFields":
1200
+ return UsageStatsByModelFields("byModel")
901
1201
 
902
- def fields(self, *subfields: Union[GlobalUsageGraphQLField, 'UsageStatsByModelFields', 'UsageStatsFields']) -> 'GlobalUsageFields':
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) -> 'GlobalUsageFields':
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
- def fields(self, *subfields: GpuAllocationGraphQLField) -> 'GpuAllocationFields':
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) -> 'GpuAllocationFields':
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
- id: 'GraderGraphQLField' = GraderGraphQLField('id')
932
- name: 'GraderGraphQLField' = GraderGraphQLField('name')
933
- key: 'GraderGraphQLField' = GraderGraphQLField('key')
934
- locked: 'GraderGraphQLField' = GraderGraphQLField('locked')
935
- grader_type: 'GraderGraphQLField' = GraderGraphQLField('graderType')
936
- grader_config: 'GraderConfigUnion' = GraderConfigUnion('graderConfig')
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) -> 'UseCaseFields':
940
- return UseCaseFields('useCase')
1247
+ def use_case(cls) -> "UseCaseFields":
1248
+ return UseCaseFields("useCase")
941
1249
 
942
1250
  @classmethod
943
- def metric(cls) -> 'MetricFields':
944
- return MetricFields('metric')
945
- created_at: 'GraderGraphQLField' = GraderGraphQLField('createdAt')
946
- updated_at: 'GraderGraphQLField' = GraderGraphQLField('updatedAt')
1251
+ def metric(cls) -> "MetricFields":
1252
+ return MetricFields("metric")
947
1253
 
948
- def fields(self, *subfields: Union[GraderGraphQLField, 'GraderConfigUnion', 'MetricFields', 'UseCaseFields']) -> 'GraderFields':
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) -> 'GraderFields':
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
- id: 'HarmonyGroupGraphQLField' = HarmonyGroupGraphQLField('id')
960
- key: 'HarmonyGroupGraphQLField' = HarmonyGroupGraphQLField('key')
1271
+
1272
+ id: "HarmonyGroupGraphQLField" = HarmonyGroupGraphQLField("id")
1273
+ key: "HarmonyGroupGraphQLField" = HarmonyGroupGraphQLField("key")
961
1274
 
962
1275
  @classmethod
963
- def compute_pool(cls) -> 'ComputePoolFields':
964
- return ComputePoolFields('computePool')
965
- status: 'HarmonyGroupGraphQLField' = HarmonyGroupGraphQLField('status')
966
- url: 'HarmonyGroupGraphQLField' = HarmonyGroupGraphQLField('url')
967
- world_size: 'HarmonyGroupGraphQLField' = HarmonyGroupGraphQLField('worldSize')
968
- gpu_total: 'HarmonyGroupGraphQLField' = HarmonyGroupGraphQLField('gpuTotal')
969
- gpu_allocated: 'HarmonyGroupGraphQLField' = HarmonyGroupGraphQLField('gpuAllocated')
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) -> 'GpuAllocationFields':
973
- return GpuAllocationFields('gpuAllocations')
974
- gpu_types: 'HarmonyGroupGraphQLField' = HarmonyGroupGraphQLField('gpuTypes')
975
- created_at: 'HarmonyGroupGraphQLField' = HarmonyGroupGraphQLField('createdAt')
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) -> 'ModelFields':
979
- return ModelFields('onlineModels')
1293
+ def online_models(cls) -> "ModelFields":
1294
+ return ModelFields("onlineModels")
980
1295
 
981
- def fields(self, *subfields: Union[HarmonyGroupGraphQLField, 'ComputePoolFields', 'GpuAllocationFields', 'ModelFields']) -> 'HarmonyGroupFields':
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) -> 'HarmonyGroupFields':
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
- def fields(self, *subfields: InteractionOutputGraphQLField) -> 'InteractionOutputFields':
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) -> 'InteractionOutputFields':
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
- def fields(self, *subfields: IntervalGraphQLField) -> 'IntervalFields':
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) -> 'IntervalFields':
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
- id: 'JobGraphQLField' = JobGraphQLField('id')
1393
+
1394
+ id: "JobGraphQLField" = JobGraphQLField("id")
1023
1395
 
1024
1396
  @classmethod
1025
- def use_case(cls) -> 'UseCaseFields':
1026
- return UseCaseFields('useCase')
1027
- name: 'JobGraphQLField' = JobGraphQLField('name')
1028
- status: 'JobGraphQLField' = JobGraphQLField('status')
1029
- created_at: 'JobGraphQLField' = JobGraphQLField('createdAt')
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) -> 'UserFields':
1033
- return UserFields('createdBy')
1034
- started_at: 'JobGraphQLField' = JobGraphQLField('startedAt')
1035
- ended_at: 'JobGraphQLField' = JobGraphQLField('endedAt')
1036
- duration_ms: 'JobGraphQLField' = JobGraphQLField('durationMs')
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) -> 'JobStageOutputFields':
1040
- return JobStageOutputFields('stages')
1041
- progress: 'JobGraphQLField' = JobGraphQLField('progress')
1042
- error: 'JobGraphQLField' = JobGraphQLField('error')
1043
- kind: 'JobGraphQLField' = JobGraphQLField('kind')
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) -> 'CustomRecipeFields':
1047
- return CustomRecipeFields('recipe')
1421
+ def recipe(cls) -> "CustomRecipeFields":
1422
+ return CustomRecipeFields("recipe")
1048
1423
 
1049
1424
  @classmethod
1050
- def details(cls) -> 'CustomRecipeJobDetailsFields':
1051
- return CustomRecipeJobDetailsFields('details')
1425
+ def details(cls) -> "CustomRecipeJobDetailsFields":
1426
+ return CustomRecipeJobDetailsFields("details")
1052
1427
 
1053
- def fields(self, *subfields: Union[JobGraphQLField, 'CustomRecipeFields', 'CustomRecipeJobDetailsFields', 'JobStageOutputFields', 'UseCaseFields', 'UserFields']) -> 'JobFields':
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) -> 'JobFields':
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
- id: 'JobArtifactGraphQLField' = JobArtifactGraphQLField('id')
1065
- job_id: 'JobArtifactGraphQLField' = JobArtifactGraphQLField('jobId')
1066
- name: 'JobArtifactGraphQLField' = JobArtifactGraphQLField('name')
1067
- kind: 'JobArtifactGraphQLField' = JobArtifactGraphQLField('kind')
1068
- status: 'JobArtifactGraphQLField' = JobArtifactGraphQLField('status')
1069
- uri: 'JobArtifactGraphQLField' = JobArtifactGraphQLField('uri')
1070
- metadata: 'JobArtifactGraphQLField' = JobArtifactGraphQLField('metadata')
1071
- created_at: 'JobArtifactGraphQLField' = JobArtifactGraphQLField('createdAt')
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) -> 'JobFields':
1075
- return JobFields('job')
1076
- download_url: 'JobArtifactGraphQLField' = JobArtifactGraphQLField('downloadUrl')
1077
- byproducts: 'ArtifactByproductsUnion' = ArtifactByproductsUnion('byproducts')
1461
+ def job(cls) -> "JobFields":
1462
+ return JobFields("job")
1078
1463
 
1079
- def fields(self, *subfields: Union[JobArtifactGraphQLField, 'ArtifactByproductsUnion', 'JobFields']) -> 'JobArtifactFields':
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) -> 'JobArtifactFields':
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) -> 'PageInfoFields':
1483
+ def page_info(cls) -> "PageInfoFields":
1093
1484
  """Information to aid in pagination."""
1094
- return PageInfoFields('pageInfo')
1485
+ return PageInfoFields("pageInfo")
1095
1486
 
1096
1487
  @classmethod
1097
- def edges(cls) -> 'JobEdgeFields':
1488
+ def edges(cls) -> "JobEdgeFields":
1098
1489
  """A list of edges."""
1099
- return JobEdgeFields('edges')
1490
+ return JobEdgeFields("edges")
1100
1491
 
1101
1492
  @classmethod
1102
- def nodes(cls) -> 'JobFields':
1493
+ def nodes(cls) -> "JobFields":
1103
1494
  """A list of nodes."""
1104
- return JobFields('nodes')
1105
- total_count: 'JobConnectionGraphQLField' = JobConnectionGraphQLField('totalCount')
1495
+ return JobFields("nodes")
1496
+
1497
+ total_count: "JobConnectionGraphQLField" = JobConnectionGraphQLField("totalCount")
1106
1498
 
1107
- def fields(self, *subfields: Union[JobConnectionGraphQLField, 'JobEdgeFields', 'JobFields', 'PageInfoFields']) -> 'JobConnectionFields':
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) -> 'JobConnectionFields':
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) -> 'JobFields':
1516
+ def node(cls) -> "JobFields":
1122
1517
  """The item at the end of the edge"""
1123
- return JobFields('node')
1124
- cursor: 'JobEdgeGraphQLField' = JobEdgeGraphQLField('cursor')
1125
- 'A cursor for use in pagination'
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, 'JobFields']) -> 'JobEdgeFields':
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) -> 'JobEdgeFields':
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
- def fields(self, *subfields: Union[JobStageOutputGraphQLField, 'JobStageInfoOutputUnion']) -> 'JobStageOutputFields':
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) -> 'JobStageOutputFields':
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
- id: 'JudgeGraphQLField' = JudgeGraphQLField('id')
1159
- key: 'JudgeGraphQLField' = JudgeGraphQLField('key')
1160
- version: 'JudgeGraphQLField' = JudgeGraphQLField('version')
1161
- name: 'JudgeGraphQLField' = JudgeGraphQLField('name')
1162
- criteria: 'JudgeGraphQLField' = JudgeGraphQLField('criteria')
1163
- prebuilt: 'JudgeGraphQLField' = JudgeGraphQLField('prebuilt')
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) -> 'JudgeExampleFields':
1167
- return JudgeExampleFields('examples')
1168
- capabilities: 'JudgeGraphQLField' = JudgeGraphQLField('capabilities')
1568
+ def examples(cls) -> "JudgeExampleFields":
1569
+ return JudgeExampleFields("examples")
1570
+
1571
+ capabilities: "JudgeGraphQLField" = JudgeGraphQLField("capabilities")
1169
1572
 
1170
1573
  @classmethod
1171
- def model(cls) -> 'ModelFields':
1172
- return ModelFields('model')
1173
- use_case_id: 'JudgeGraphQLField' = JudgeGraphQLField('useCaseId')
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) -> 'MetricFields':
1177
- return MetricFields('metric')
1178
- created_at: 'JudgeGraphQLField' = JudgeGraphQLField('createdAt')
1179
- updated_at: 'JudgeGraphQLField' = JudgeGraphQLField('updatedAt')
1580
+ def metric(cls) -> "MetricFields":
1581
+ return MetricFields("metric")
1180
1582
 
1181
- def fields(self, *subfields: Union[JudgeGraphQLField, 'JudgeExampleFields', 'MetricFields', 'ModelFields']) -> 'JudgeFields':
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) -> 'JudgeFields':
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
- criteria: 'JudgeConfigOutputGraphQLField' = JudgeConfigOutputGraphQLField('criteria')
1600
+
1601
+ criteria: "JudgeConfigOutputGraphQLField" = JudgeConfigOutputGraphQLField("criteria")
1193
1602
 
1194
1603
  @classmethod
1195
- def examples(cls) -> 'JudgeExampleFields':
1196
- return JudgeExampleFields('examples')
1197
- system_template: 'JudgeConfigOutputGraphQLField' = JudgeConfigOutputGraphQLField('systemTemplate')
1198
- user_template: 'JudgeConfigOutputGraphQLField' = JudgeConfigOutputGraphQLField('userTemplate')
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) -> 'ModelFields':
1202
- return ModelFields('model')
1611
+ def model(cls) -> "ModelFields":
1612
+ return ModelFields("model")
1203
1613
 
1204
- def fields(self, *subfields: Union[JudgeConfigOutputGraphQLField, 'JudgeExampleFields', 'ModelFields']) -> 'JudgeConfigOutputFields':
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) -> 'JudgeConfigOutputFields':
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) -> 'ChatMessageFields':
1218
- return ChatMessageFields('input')
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
- def fields(self, *subfields: Union[JudgeExampleGraphQLField, 'ChatMessageFields']) -> 'JudgeExampleFields':
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) -> 'JudgeExampleFields':
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
- def fields(self, *subfields: LabelGraphQLField) -> 'LabelFields':
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) -> 'LabelFields':
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
- key: 'LabelKeyUsageGraphQLField' = LabelKeyUsageGraphQLField('key')
1249
- count: 'LabelKeyUsageGraphQLField' = LabelKeyUsageGraphQLField('count')
1665
+
1666
+ key: "LabelKeyUsageGraphQLField" = LabelKeyUsageGraphQLField("key")
1667
+ count: "LabelKeyUsageGraphQLField" = LabelKeyUsageGraphQLField("count")
1250
1668
 
1251
1669
  @classmethod
1252
- def values(cls) -> 'LabelValueUsageFields':
1253
- return LabelValueUsageFields('values')
1254
- last_used: 'LabelKeyUsageGraphQLField' = LabelKeyUsageGraphQLField('lastUsed')
1670
+ def values(cls) -> "LabelValueUsageFields":
1671
+ return LabelValueUsageFields("values")
1255
1672
 
1256
- def fields(self, *subfields: Union[LabelKeyUsageGraphQLField, 'LabelValueUsageFields']) -> 'LabelKeyUsageFields':
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) -> 'LabelKeyUsageFields':
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) -> 'LabelKeyUsageFields':
1270
- return LabelKeyUsageFields('keys')
1689
+ def keys(cls) -> "LabelKeyUsageFields":
1690
+ return LabelKeyUsageFields("keys")
1271
1691
 
1272
- def fields(self, *subfields: Union[LabelUsageGraphQLField, 'LabelKeyUsageFields']) -> 'LabelUsageFields':
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) -> 'LabelUsageFields':
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
- def fields(self, *subfields: LabelValueUsageGraphQLField) -> 'LabelValueUsageFields':
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) -> 'LabelValueUsageFields':
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) -> 'ProviderListFields':
1301
- return ProviderListFields('authProviders')
1723
+ def auth_providers(cls) -> "ProviderListFields":
1724
+ return ProviderListFields("authProviders")
1302
1725
 
1303
- def fields(self, *subfields: Union[MetaObjectGraphQLField, 'ProviderListFields']) -> 'MetaObjectFields':
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) -> 'MetaObjectFields':
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
- id: 'MetricGraphQLField' = MetricGraphQLField('id')
1315
- key: 'MetricGraphQLField' = MetricGraphQLField('key')
1316
- name: 'MetricGraphQLField' = MetricGraphQLField('name')
1317
- created_at: 'MetricGraphQLField' = MetricGraphQLField('createdAt')
1318
- kind: 'MetricGraphQLField' = MetricGraphQLField('kind')
1319
- description: 'MetricGraphQLField' = MetricGraphQLField('description')
1320
- scoring_type: 'MetricGraphQLField' = MetricGraphQLField('scoringType')
1321
- unit: 'MetricGraphQLField' = MetricGraphQLField('unit')
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) -> 'UseCaseFields':
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]] = {'filter': {'type': 'UseCaseFilter!', 'value': filter}}
1327
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
1328
- return UseCaseFields('useCases', arguments=cleared_arguments)
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) -> 'MetricActivityFields':
1332
- arguments: Dict[str, Dict[str, Any]] = {'timerange': {'type': 'TimeRange', 'value': timerange}}
1333
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
1334
- return MetricActivityFields('activity', arguments=cleared_arguments)
1335
- has_direct_feedbacks: 'MetricGraphQLField' = MetricGraphQLField('hasDirectFeedbacks')
1336
- has_comparison_feedbacks: 'MetricGraphQLField' = MetricGraphQLField('hasComparisonFeedbacks')
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, 'MetricActivityFields', 'UseCaseFields']) -> 'MetricFields':
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) -> 'MetricFields':
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) -> 'ActivityOutputFields':
1352
- return ActivityOutputFields('feedbacks')
1778
+ def feedbacks(cls) -> "ActivityOutputFields":
1779
+ return ActivityOutputFields("feedbacks")
1353
1780
 
1354
- def fields(self, *subfields: Union[MetricActivityGraphQLField, 'ActivityOutputFields']) -> 'MetricActivityFields':
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) -> 'MetricActivityFields':
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
- id: 'MetricWithContextGraphQLField' = MetricWithContextGraphQLField('id')
1366
- key: 'MetricWithContextGraphQLField' = MetricWithContextGraphQLField('key')
1367
- name: 'MetricWithContextGraphQLField' = MetricWithContextGraphQLField('name')
1368
- kind: 'MetricWithContextGraphQLField' = MetricWithContextGraphQLField('kind')
1369
- scoring_type: 'MetricWithContextGraphQLField' = MetricWithContextGraphQLField('scoringType')
1370
- description: 'MetricWithContextGraphQLField' = MetricWithContextGraphQLField('description')
1371
- created_at: 'MetricWithContextGraphQLField' = MetricWithContextGraphQLField('createdAt')
1372
- unit: 'MetricWithContextGraphQLField' = MetricWithContextGraphQLField('unit')
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) -> 'MetricWithContextGraphQLField':
1376
- arguments: Dict[str, Dict[str, Any]] = {'timerange': {'type': 'TimeRange', 'value': timerange}}
1377
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
1378
- return MetricWithContextGraphQLField('feedbackCount', arguments=cleared_arguments)
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) -> 'MetricWithContextGraphQLField':
1382
- arguments: Dict[str, Dict[str, Any]] = {'timerange': {'type': 'TimeRange', 'value': timerange}}
1383
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
1384
- return MetricWithContextGraphQLField('comparisonCount', arguments=cleared_arguments)
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) -> 'TrendResultFields':
1388
- arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'MetricTrendInput!', 'value': input}}
1389
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
1390
- return TrendResultFields('trend', arguments=cleared_arguments)
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) -> 'TimeseriesFields':
1394
- arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'TimeseriesInput!', 'value': input}}
1395
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
1396
- return TimeseriesFields('timeseries', arguments=cleared_arguments)
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
- def fields(self, *subfields: Union[MetricWithContextGraphQLField, 'TimeseriesFields', 'TrendResultFields']) -> 'MetricWithContextFields':
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) -> 'MetricWithContextFields':
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
- @classmethod
1437
- def model_services(cls, filter: UseCaseFilter) -> 'ModelServiceFields':
1438
- arguments: Dict[str, Dict[str, Any]] = {'filter': {'type': 'UseCaseFilter!', 'value': filter}}
1439
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
1440
- return ModelServiceFields('modelServices', arguments=cleared_arguments)
1441
- provider_name: 'ModelGraphQLField' = ModelGraphQLField('providerName')
1442
- is_external: 'ModelGraphQLField' = ModelGraphQLField('isExternal')
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 backbone(cls) -> 'ModelFields':
1449
- return ModelFields('backbone')
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 parent(cls) -> 'ModelFields':
1453
- return ModelFields('parent')
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 compute_config(cls) -> 'ModelComputeConfigOutputFields':
1466
- return ModelComputeConfigOutputFields('computeConfig')
1467
-
1468
- def fields(self, *subfields: Union[ModelGraphQLField, 'ActivityFields', 'MetricWithContextFields', 'ModelComputeConfigOutputFields', 'ModelFields', 'ModelServiceFields', 'UseCaseFields']) -> 'ModelFields':
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) -> 'ModelFields':
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) -> 'ModelFields':
1482
- return ModelFields('model')
1929
+ def model(cls) -> "ModelFields":
1930
+ return ModelFields("model")
1483
1931
 
1484
- def fields(self, *subfields: Union[ModelByproductsGraphQLField, 'ModelFields']) -> 'ModelByproductsFields':
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) -> 'ModelByproductsFields':
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
- def fields(self, *subfields: ModelComputeConfigOutputGraphQLField) -> 'ModelComputeConfigOutputFields':
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) -> 'ModelComputeConfigOutputFields':
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
- def fields(self, *subfields: ModelPlacementOutputGraphQLField) -> 'ModelPlacementOutputFields':
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) -> 'ModelPlacementOutputFields':
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
- status: 'ModelServiceGraphQLField' = ModelServiceGraphQLField('status')
1525
- error: 'ModelServiceGraphQLField' = ModelServiceGraphQLField('error')
1526
- id: 'ModelServiceGraphQLField' = ModelServiceGraphQLField('id')
1527
- use_case_id: 'ModelServiceGraphQLField' = ModelServiceGraphQLField('useCaseId')
1528
- key: 'ModelServiceGraphQLField' = ModelServiceGraphQLField('key')
1529
- name: 'ModelServiceGraphQLField' = ModelServiceGraphQLField('name')
1530
- created_at: 'ModelServiceGraphQLField' = ModelServiceGraphQLField('createdAt')
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) -> 'ModelFields':
1534
- return ModelFields('model')
1535
- is_default: 'ModelServiceGraphQLField' = ModelServiceGraphQLField('isDefault')
1536
- desired_online: 'ModelServiceGraphQLField' = ModelServiceGraphQLField('desiredOnline')
1537
- deleted: 'ModelServiceGraphQLField' = ModelServiceGraphQLField('deleted')
1538
- 'Whether or not this model service has been deleted.'
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) -> 'ActivityFields':
1542
- arguments: Dict[str, Dict[str, Any]] = {'timerange': {'type': 'TimeRange', 'value': timerange}}
1543
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
1544
- return ActivityFields('activity', arguments=cleared_arguments)
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) -> 'SystemPromptTemplateFields':
1548
- return SystemPromptTemplateFields('systemPromptTemplate')
2002
+ def system_prompt_template(cls) -> "SystemPromptTemplateFields":
2003
+ return SystemPromptTemplateFields("systemPromptTemplate")
1549
2004
 
1550
2005
  @classmethod
1551
- def metrics(cls) -> 'MetricWithContextFields':
1552
- return MetricWithContextFields('metrics')
2006
+ def metrics(cls) -> "MetricWithContextFields":
2007
+ return MetricWithContextFields("metrics")
1553
2008
 
1554
2009
  @classmethod
1555
- def ab_campaigns(cls, filter: AbCampaignFilter) -> 'AbcampaignFields':
1556
- arguments: Dict[str, Dict[str, Any]] = {'filter': {'type': 'AbCampaignFilter!', 'value': filter}}
1557
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
1558
- return AbcampaignFields('abCampaigns', arguments=cleared_arguments)
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) -> 'ModelPlacementOutputFields':
1562
- return ModelPlacementOutputFields('placement')
2016
+ def placement(cls) -> "ModelPlacementOutputFields":
2017
+ return ModelPlacementOutputFields("placement")
1563
2018
 
1564
2019
  @classmethod
1565
- def tool_providers(cls) -> 'ToolProviderFields':
1566
- return ToolProviderFields('toolProviders')
2020
+ def tool_providers(cls) -> "ToolProviderFields":
2021
+ return ToolProviderFields("toolProviders")
1567
2022
 
1568
- def fields(self, *subfields: Union[ModelServiceGraphQLField, 'AbcampaignFields', 'ActivityFields', 'MetricWithContextFields', 'ModelFields', 'ModelPlacementOutputFields', 'SystemPromptTemplateFields', 'ToolProviderFields']) -> 'ModelServiceFields':
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) -> 'ModelServiceFields':
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
- has_previous_page: 'PageInfoGraphQLField' = PageInfoGraphQLField('hasPreviousPage')
1581
- 'When paginating backwards, are there more items?'
1582
- has_next_page: 'PageInfoGraphQLField' = PageInfoGraphQLField('hasNextPage')
1583
- 'When paginating forwards, are there more items?'
1584
- start_cursor: 'PageInfoGraphQLField' = PageInfoGraphQLField('startCursor')
1585
- 'When paginating backwards, the cursor to continue.'
1586
- end_cursor: 'PageInfoGraphQLField' = PageInfoGraphQLField('endCursor')
1587
- 'When paginating forwards, the cursor to continue.'
1588
-
1589
- def fields(self, *subfields: PageInfoGraphQLField) -> 'PageInfoFields':
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) -> 'PageInfoFields':
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
- def fields(self, *subfields: PrebuiltConfigDefinitionGraphQLField) -> 'PrebuiltConfigDefinitionFields':
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) -> 'PrebuiltConfigDefinitionFields':
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) -> 'PrebuiltConfigDefinitionFields':
1619
- return PrebuiltConfigDefinitionFields('criteria')
2109
+ def criteria(cls) -> "PrebuiltConfigDefinitionFields":
2110
+ return PrebuiltConfigDefinitionFields("criteria")
1620
2111
 
1621
2112
  @classmethod
1622
- def model(cls) -> 'ModelFields':
1623
- return ModelFields('model')
2113
+ def model(cls) -> "ModelFields":
2114
+ return ModelFields("model")
1624
2115
 
1625
- def fields(self, *subfields: Union[PrebuiltConfigOutputGraphQLField, 'ModelFields', 'PrebuiltConfigDefinitionFields']) -> 'PrebuiltConfigOutputFields':
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) -> 'PrebuiltConfigOutputFields':
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
- key: 'PrebuiltCriteriaGraphQLField' = PrebuiltCriteriaGraphQLField('key')
1637
- name: 'PrebuiltCriteriaGraphQLField' = PrebuiltCriteriaGraphQLField('name')
2130
+
2131
+ key: "PrebuiltCriteriaGraphQLField" = PrebuiltCriteriaGraphQLField("key")
2132
+ name: "PrebuiltCriteriaGraphQLField" = PrebuiltCriteriaGraphQLField("name")
1638
2133
 
1639
2134
  @classmethod
1640
- def feedback(cls) -> 'MetricFields':
1641
- return MetricFields('feedback')
1642
- description: 'PrebuiltCriteriaGraphQLField' = PrebuiltCriteriaGraphQLField('description')
2135
+ def feedback(cls) -> "MetricFields":
2136
+ return MetricFields("feedback")
1643
2137
 
1644
- def fields(self, *subfields: Union[PrebuiltCriteriaGraphQLField, 'MetricFields']) -> 'PrebuiltCriteriaFields':
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) -> 'PrebuiltCriteriaFields':
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) -> 'AuthProviderFields':
1658
- return AuthProviderFields('providers')
2172
+ def providers(cls) -> "AuthProviderFields":
2173
+ return AuthProviderFields("providers")
1659
2174
 
1660
- def fields(self, *subfields: Union[ProviderListGraphQLField, 'AuthProviderFields']) -> 'ProviderListFields':
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) -> 'ProviderListFields':
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
- def fields(self, *subfields: RemoteConfigOutputGraphQLField) -> 'RemoteConfigOutputFields':
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) -> 'RemoteConfigOutputFields':
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
- id: 'RemoteEnvGraphQLField' = RemoteEnvGraphQLField('id')
1687
- key: 'RemoteEnvGraphQLField' = RemoteEnvGraphQLField('key')
1688
- name: 'RemoteEnvGraphQLField' = RemoteEnvGraphQLField('name')
1689
- url: 'RemoteEnvGraphQLField' = RemoteEnvGraphQLField('url')
1690
- description: 'RemoteEnvGraphQLField' = RemoteEnvGraphQLField('description')
1691
- created_at: 'RemoteEnvGraphQLField' = RemoteEnvGraphQLField('createdAt')
1692
- version: 'RemoteEnvGraphQLField' = RemoteEnvGraphQLField('version')
1693
- status: 'RemoteEnvGraphQLField' = RemoteEnvGraphQLField('status')
1694
- metadata_schema: 'RemoteEnvGraphQLField' = RemoteEnvGraphQLField('metadataSchema')
1695
-
1696
- def fields(self, *subfields: RemoteEnvGraphQLField) -> 'RemoteEnvFields':
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) -> 'RemoteEnvFields':
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
- def fields(self, *subfields: RemoteEnvTestOfflineGraphQLField) -> 'RemoteEnvTestOfflineFields':
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) -> 'RemoteEnvTestOfflineFields':
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
- def fields(self, *subfields: RemoteEnvTestOnlineGraphQLField) -> 'RemoteEnvTestOnlineFields':
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) -> 'RemoteEnvTestOnlineFields':
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
- def fields(self, *subfields: RoleGraphQLField) -> 'RoleFields':
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) -> 'RoleFields':
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) -> 'JobFields':
1755
- return JobFields('jobs')
2280
+ def jobs(cls) -> "JobFields":
2281
+ return JobFields("jobs")
1756
2282
 
1757
2283
  @classmethod
1758
- def artifacts(cls, *, filter: Optional[ArtifactFilter]=None) -> 'JobArtifactFields':
1759
- arguments: Dict[str, Dict[str, Any]] = {'filter': {'type': 'ArtifactFilter', 'value': filter}}
1760
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
1761
- return JobArtifactFields('artifacts', arguments=cleared_arguments)
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(self, *subfields: Union[SearchResultGraphQLField, 'JobArtifactFields', 'JobFields']) -> 'SearchResultFields':
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) -> 'SearchResultFields':
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
- id: 'SessionGraphQLField' = SessionGraphQLField('id')
2303
+
2304
+ id: "SessionGraphQLField" = SessionGraphQLField("id")
1775
2305
 
1776
2306
  @classmethod
1777
- def turns(cls) -> 'CompletionFields':
1778
- return CompletionFields('turns')
2307
+ def turns(cls) -> "CompletionFields":
2308
+ return CompletionFields("turns")
1779
2309
 
1780
- def fields(self, *subfields: Union[SessionGraphQLField, 'CompletionFields']) -> 'SessionFields':
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) -> 'SessionFields':
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) -> 'MetricWithContextFields':
1794
- return MetricWithContextFields('defaultMetric')
2324
+ def default_metric(cls) -> "MetricWithContextFields":
2325
+ return MetricWithContextFields("defaultMetric")
1795
2326
 
1796
- def fields(self, *subfields: Union[SettingsGraphQLField, 'MetricWithContextFields']) -> 'SettingsFields':
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) -> 'SettingsFields':
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) -> 'TeamFields':
1810
- return TeamFields('team')
2341
+ def team(cls) -> "TeamFields":
2342
+ return TeamFields("team")
1811
2343
 
1812
2344
  @classmethod
1813
- def role(cls) -> 'RoleFields':
1814
- return RoleFields('role')
1815
- is_owner: 'ShareGraphQLField' = ShareGraphQLField('isOwner')
2345
+ def role(cls) -> "RoleFields":
2346
+ return RoleFields("role")
1816
2347
 
1817
- def fields(self, *subfields: Union[ShareGraphQLField, 'RoleFields', 'TeamFields']) -> 'ShareFields':
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) -> 'ShareFields':
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
- def fields(self, *subfields: SystemPromptTemplateGraphQLField) -> 'SystemPromptTemplateFields':
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) -> 'SystemPromptTemplateFields':
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
- def fields(self, *subfields: TeamGraphQLField) -> 'TeamFields':
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) -> 'TeamFields':
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) -> 'UserFields':
1865
- return UserFields('user')
2454
+ def user(cls) -> "UserFields":
2455
+ return UserFields("user")
1866
2456
 
1867
2457
  @classmethod
1868
- def team(cls) -> 'TeamFields':
1869
- return TeamFields('team')
2458
+ def team(cls) -> "TeamFields":
2459
+ return TeamFields("team")
1870
2460
 
1871
2461
  @classmethod
1872
- def role(cls) -> 'RoleFields':
1873
- return RoleFields('role')
2462
+ def role(cls) -> "RoleFields":
2463
+ return RoleFields("role")
1874
2464
 
1875
- def fields(self, *subfields: Union[TeamMemberGraphQLField, 'RoleFields', 'TeamFields', 'UserFields']) -> 'TeamMemberFields':
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) -> 'TeamMemberFields':
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) -> 'TeamFields':
1889
- return TeamFields('team')
2481
+ def team(cls) -> "TeamFields":
2482
+ return TeamFields("team")
1890
2483
 
1891
2484
  @classmethod
1892
- def role(cls) -> 'RoleFields':
1893
- return RoleFields('role')
2485
+ def role(cls) -> "RoleFields":
2486
+ return RoleFields("role")
1894
2487
 
1895
- def fields(self, *subfields: Union[TeamWithroleGraphQLField, 'RoleFields', 'TeamFields']) -> 'TeamWithroleFields':
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) -> 'TeamWithroleFields':
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) -> 'ModelFields':
1909
- return ModelFields('model')
1910
- time_buckets: 'TimeseriesGraphQLField' = TimeseriesGraphQLField('timeBuckets')
1911
- count: 'TimeseriesGraphQLField' = TimeseriesGraphQLField('count')
1912
- values: 'TimeseriesGraphQLField' = TimeseriesGraphQLField('values')
1913
- aggregation: 'TimeseriesGraphQLField' = TimeseriesGraphQLField('aggregation')
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, 'ModelFields']) -> 'TimeseriesFields':
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) -> 'TimeseriesFields':
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
- def fields(self, *subfields: ToolProviderGraphQLField) -> 'ToolProviderFields':
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) -> 'ToolProviderFields':
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
- def fields(self, *subfields: TrainingJobStageOutputGraphQLField) -> 'TrainingJobStageOutputFields':
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) -> 'TrainingJobStageOutputFields':
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
- def fields(self, *subfields: TrendResultGraphQLField) -> 'TrendResultFields':
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) -> 'TrendResultFields':
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
- def fields(self, *subfields: UnitConfigGraphQLField) -> 'UnitConfigFields':
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) -> 'UnitConfigFields':
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
- def fields(self, *subfields: UsageGraphQLField) -> 'UsageFields':
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) -> 'UsageFields':
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
- def fields(self, *subfields: UsageAggregateItemGraphQLField) -> 'UsageAggregateItemFields':
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) -> 'UsageAggregateItemFields':
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) -> 'UseCaseItemFields':
2024
- return UseCaseItemFields('useCase')
2649
+ def use_case(cls) -> "UseCaseItemFields":
2650
+ return UseCaseItemFields("useCase")
2025
2651
 
2026
2652
  @classmethod
2027
- def model_service(cls) -> 'ModelServiceFields':
2028
- return ModelServiceFields('modelService')
2029
- prompt_tokens: 'UsageAggregatePerUseCaseItemGraphQLField' = UsageAggregatePerUseCaseItemGraphQLField('promptTokens')
2030
- completion_tokens: 'UsageAggregatePerUseCaseItemGraphQLField' = UsageAggregatePerUseCaseItemGraphQLField('completionTokens')
2031
- total_tokens: 'UsageAggregatePerUseCaseItemGraphQLField' = UsageAggregatePerUseCaseItemGraphQLField('totalTokens')
2032
- interactions: 'UsageAggregatePerUseCaseItemGraphQLField' = UsageAggregatePerUseCaseItemGraphQLField('interactions')
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(self, *subfields: Union[UsageAggregatePerUseCaseItemGraphQLField, 'ModelServiceFields', 'UseCaseItemFields']) -> 'UsageAggregatePerUseCaseItemFields':
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) -> 'UsageAggregatePerUseCaseItemFields':
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
- def fields(self, *subfields: UsageStatsGraphQLField) -> 'UsageStatsFields':
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) -> 'UsageStatsFields':
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) -> 'ModelFields':
2064
- return ModelFields('model')
2065
- total_tokens: 'UsageStatsByModelGraphQLField' = UsageStatsByModelGraphQLField('totalTokens')
2066
- interactions: 'UsageStatsByModelGraphQLField' = UsageStatsByModelGraphQLField('interactions')
2067
- prompt_tokens: 'UsageStatsByModelGraphQLField' = UsageStatsByModelGraphQLField('promptTokens')
2068
- completion_tokens: 'UsageStatsByModelGraphQLField' = UsageStatsByModelGraphQLField('completionTokens')
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) -> 'UsageAggregateItemFields':
2072
- return UsageAggregateItemFields('timeseries')
2706
+ def timeseries(cls) -> "UsageAggregateItemFields":
2707
+ return UsageAggregateItemFields("timeseries")
2073
2708
 
2074
- def fields(self, *subfields: Union[UsageStatsByModelGraphQLField, 'ModelFields', 'UsageAggregateItemFields']) -> 'UsageStatsByModelFields':
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) -> 'UsageStatsByModelFields':
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
- id: 'UseCaseGraphQLField' = UseCaseGraphQLField('id')
2086
- name: 'UseCaseGraphQLField' = UseCaseGraphQLField('name')
2087
- key: 'UseCaseGraphQLField' = UseCaseGraphQLField('key')
2088
- description: 'UseCaseGraphQLField' = UseCaseGraphQLField('description')
2089
- created_at: 'UseCaseGraphQLField' = UseCaseGraphQLField('createdAt')
2090
- is_archived: 'UseCaseGraphQLField' = UseCaseGraphQLField('isArchived')
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) -> 'ModelServiceFields':
2094
- arguments: Dict[str, Dict[str, Any]] = {'filter': {'type': 'ModelServiceFilter', 'value': filter}}
2095
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
2096
- return ModelServiceFields('modelServices', arguments=cleared_arguments)
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) -> 'ModelServiceFields':
2100
- arguments: Dict[str, Dict[str, Any]] = {'idOrKey': {'type': 'IdOrKey!', 'value': id_or_key}}
2101
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
2102
- return ModelServiceFields('modelService', arguments=cleared_arguments)
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) -> 'ModelFields':
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]] = {'filter': {'type': 'ModelFilter', 'value': filter}}
2108
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
2109
- return ModelFields('models', arguments=cleared_arguments)
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) -> 'ModelServiceFields':
2113
- return ModelServiceFields('defaultModelService')
2751
+ def default_model_service(cls) -> "ModelServiceFields":
2752
+ return ModelServiceFields("defaultModelService")
2114
2753
 
2115
2754
  @classmethod
2116
- def activity(cls, *, timerange: Optional[TimeRange]=None) -> 'ActivityFields':
2117
- arguments: Dict[str, Dict[str, Any]] = {'timerange': {'type': 'TimeRange', 'value': timerange}}
2118
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
2119
- return ActivityFields('activity', arguments=cleared_arguments)
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) -> 'MetricWithContextFields':
2123
- return MetricWithContextFields('metrics')
2761
+ def metrics(cls) -> "MetricWithContextFields":
2762
+ return MetricWithContextFields("metrics")
2124
2763
 
2125
2764
  @classmethod
2126
- def metric(cls, metric: str) -> 'MetricWithContextFields':
2127
- arguments: Dict[str, Dict[str, Any]] = {'metric': {'type': 'IdOrKey!', 'value': metric}}
2128
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
2129
- return MetricWithContextFields('metric', arguments=cleared_arguments)
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) -> 'AbcampaignFields':
2133
- arguments: Dict[str, Dict[str, Any]] = {'filter': {'type': 'AbCampaignFilter!', 'value': filter}}
2134
- cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
2135
- return AbcampaignFields('abCampaigns', arguments=cleared_arguments)
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) -> 'WidgetFields':
2139
- return WidgetFields('widgets')
2777
+ def widgets(cls) -> "WidgetFields":
2778
+ return WidgetFields("widgets")
2140
2779
 
2141
2780
  @classmethod
2142
- def metadata(cls) -> 'UseCaseMetadataFields':
2143
- return UseCaseMetadataFields('metadata')
2144
- permissions: 'UseCaseGraphQLField' = UseCaseGraphQLField('permissions')
2781
+ def metadata(cls) -> "UseCaseMetadataFields":
2782
+ return UseCaseMetadataFields("metadata")
2783
+
2784
+ permissions: "UseCaseGraphQLField" = UseCaseGraphQLField("permissions")
2145
2785
 
2146
2786
  @classmethod
2147
- def shares(cls) -> 'ShareFields':
2148
- return ShareFields('shares')
2787
+ def shares(cls) -> "ShareFields":
2788
+ return ShareFields("shares")
2149
2789
 
2150
2790
  @classmethod
2151
- def settings(cls) -> 'SettingsFields':
2152
- return SettingsFields('settings')
2791
+ def settings(cls) -> "SettingsFields":
2792
+ return SettingsFields("settings")
2153
2793
 
2154
2794
  @classmethod
2155
- def label_usage(cls) -> 'LabelUsageFields':
2156
- return LabelUsageFields('labelUsage')
2795
+ def label_usage(cls) -> "LabelUsageFields":
2796
+ return LabelUsageFields("labelUsage")
2157
2797
 
2158
2798
  @classmethod
2159
- def tool_providers(cls) -> 'ToolProviderFields':
2160
- return ToolProviderFields('toolProviders')
2799
+ def tool_providers(cls) -> "ToolProviderFields":
2800
+ return ToolProviderFields("toolProviders")
2161
2801
 
2162
- def fields(self, *subfields: Union[UseCaseGraphQLField, 'AbcampaignFields', 'ActivityFields', 'LabelUsageFields', 'MetricWithContextFields', 'ModelFields', 'ModelServiceFields', 'SettingsFields', 'ShareFields', 'ToolProviderFields', 'UseCaseMetadataFields', 'WidgetFields']) -> 'UseCaseFields':
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) -> 'UseCaseFields':
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
- def fields(self, *subfields: UseCaseItemGraphQLField) -> 'UseCaseItemFields':
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) -> 'UseCaseItemFields':
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) -> 'EmojiFields':
2192
- return EmojiFields('emoji')
2850
+ def emoji(cls) -> "EmojiFields":
2851
+ return EmojiFields("emoji")
2193
2852
 
2194
- def fields(self, *subfields: Union[UseCaseMetadataGraphQLField, 'EmojiFields']) -> 'UseCaseMetadataFields':
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) -> 'UseCaseMetadataFields':
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
- id: 'UserGraphQLField' = UserGraphQLField('id')
2206
- email: 'UserGraphQLField' = UserGraphQLField('email')
2207
- name: 'UserGraphQLField' = UserGraphQLField('name')
2208
- created_at: 'UserGraphQLField' = UserGraphQLField('createdAt')
2209
- deleted: 'UserGraphQLField' = UserGraphQLField('deleted')
2210
- deleted_at: 'UserGraphQLField' = UserGraphQLField('deletedAt')
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) -> 'TeamWithroleFields':
2214
- return TeamWithroleFields('teams')
2874
+ def teams(cls) -> "TeamWithroleFields":
2875
+ return TeamWithroleFields("teams")
2215
2876
 
2216
- def fields(self, *subfields: Union[UserGraphQLField, 'TeamWithroleFields']) -> 'UserFields':
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) -> 'UserFields':
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
- title: 'WidgetGraphQLField' = WidgetGraphQLField('title')
2228
- metric: 'WidgetGraphQLField' = WidgetGraphQLField('metric')
2229
- aggregation: 'WidgetGraphQLField' = WidgetGraphQLField('aggregation')
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) -> 'UnitConfigFields':
2233
- return UnitConfigFields('unit')
2912
+ def unit(cls) -> "UnitConfigFields":
2913
+ return UnitConfigFields("unit")
2234
2914
 
2235
- def fields(self, *subfields: Union[WidgetGraphQLField, 'UnitConfigFields']) -> 'WidgetFields':
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) -> 'WidgetFields':
2920
+ def alias(self, alias: str) -> "WidgetFields":
2241
2921
  self._alias = alias
2242
- return self
2922
+ return self