adaptive-sdk 0.1.6__py3-none-any.whl → 0.1.8__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- adaptive_sdk/base_client.py +16 -4
- adaptive_sdk/client.py +8 -2
- adaptive_sdk/external/reward_client.py +22 -2
- adaptive_sdk/graphql_client/__init__.py +4 -4
- adaptive_sdk/graphql_client/async_base_client_open_telemetry.py +6 -6
- adaptive_sdk/graphql_client/async_client.py +34 -34
- adaptive_sdk/graphql_client/base_client_open_telemetry.py +6 -6
- adaptive_sdk/graphql_client/base_operation.py +3 -3
- adaptive_sdk/graphql_client/client.py +34 -34
- adaptive_sdk/graphql_client/custom_fields.py +89 -26
- adaptive_sdk/graphql_client/custom_mutations.py +64 -58
- adaptive_sdk/graphql_client/custom_queries.py +40 -34
- adaptive_sdk/graphql_client/custom_typing_fields.py +60 -0
- adaptive_sdk/graphql_client/enums.py +18 -2
- adaptive_sdk/graphql_client/fragments.py +39 -3
- adaptive_sdk/graphql_client/input_types.py +22 -4
- adaptive_sdk/output_types/__init__.py +3 -0
- adaptive_sdk/output_types/job_wrapper.py +106 -0
- adaptive_sdk/resources/datasets.py +16 -13
- adaptive_sdk/resources/jobs.py +34 -16
- {adaptive_sdk-0.1.6.dist-info → adaptive_sdk-0.1.8.dist-info}/METADATA +1 -1
- {adaptive_sdk-0.1.6.dist-info → adaptive_sdk-0.1.8.dist-info}/RECORD +23 -21
- {adaptive_sdk-0.1.6.dist-info → adaptive_sdk-0.1.8.dist-info}/WHEEL +0 -0
|
@@ -1,6 +1,6 @@
|
|
|
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, 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, ModelComputeConfigOutputGraphQLField, ModelGraphQLField, ModelPlacementOutputGraphQLField, ModelServiceGraphQLField, PageInfoGraphQLField, PrebuiltConfigDefinitionGraphQLField, PrebuiltConfigOutputGraphQLField, PrebuiltCriteriaGraphQLField, ProviderListGraphQLField, RemoteConfigOutputGraphQLField, RemoteEnvGraphQLField, RemoteEnvTestOfflineGraphQLField, RemoteEnvTestOnlineGraphQLField, RoleGraphQLField, SessionGraphQLField, SettingsGraphQLField, ShareGraphQLField, SystemPromptTemplateGraphQLField, TeamGraphQLField, TeamMemberGraphQLField, TeamWithroleGraphQLField, TimeseriesGraphQLField, ToolProviderGraphQLField, TrainingJobStageOutputGraphQLField, TrendResultGraphQLField, UnitConfigGraphQLField, UsageAggregateItemGraphQLField, UsageAggregatePerUseCaseItemGraphQLField, UsageGraphQLField, UsageStatsByModelGraphQLField, UsageStatsGraphQLField, UseCaseGraphQLField, UseCaseItemGraphQLField, UseCaseMetadataGraphQLField, UserGraphQLField, WidgetGraphQLField
|
|
3
|
+
from .custom_typing_fields import AbcampaignGraphQLField, AbReportGraphQLField, AbVariantReportComparisonGraphQLField, AbVariantReportGraphQLField, ActivityGraphQLField, ActivityOutputGraphQLField, ApiKeyGraphQLField, ArtifactByproductsUnion, AuthProviderGraphQLField, BatchInferenceJobStageOutputGraphQLField, BillingUsageGraphQLField, ChatMessageGraphQLField, ComparisonFeedbackGraphQLField, CompletionConnectionGraphQLField, CompletionEdgeGraphQLField, CompletionGraphQLField, CompletionGroupDataConnectionGraphQLField, CompletionGroupDataEdgeGraphQLField, CompletionGroupDataGraphQLField, CompletionGroupFeedbackStatsGraphQLField, CompletionHistoryEntryOuputGraphQLField, CompletionLabelGraphQLField, CompletionMetadataGraphQLField, ComputePoolGraphQLField, ContractGraphQLField, CustomConfigOutputGraphQLField, CustomRecipeGraphQLField, CustomRecipeJobDetailsGraphQLField, DatasetByproductsGraphQLField, DatasetGraphQLField, DatasetMetricUsageGraphQLField, DatasetProgressGraphQLField, DatasetUploadProcessingStatusGraphQLField, DatasetValidationOutputGraphQLField, DeleteConfirmGraphQLField, DirectFeedbackGraphQLField, EmojiGraphQLField, EvalJobStageOutputGraphQLField, EvaluationByproductsGraphQLField, EvaluationResultGraphQLField, GlobalUsageGraphQLField, GpuAllocationGraphQLField, GraderConfigUnion, GraderGraphQLField, HarmonyGroupGraphQLField, InteractionOutputGraphQLField, IntervalGraphQLField, JobArtifactGraphQLField, JobConnectionGraphQLField, JobEdgeGraphQLField, JobGraphQLField, JobStageInfoOutputUnion, JobStageOutputGraphQLField, JudgeConfigOutputGraphQLField, JudgeExampleGraphQLField, JudgeGraphQLField, LabelGraphQLField, LabelKeyUsageGraphQLField, LabelUsageGraphQLField, LabelValueUsageGraphQLField, MetaObjectGraphQLField, MetricActivityGraphQLField, MetricGraphQLField, MetricWithContextGraphQLField, ModelByproductsGraphQLField, ModelComputeConfigOutputGraphQLField, ModelGraphQLField, ModelPlacementOutputGraphQLField, ModelServiceGraphQLField, PageInfoGraphQLField, PrebuiltConfigDefinitionGraphQLField, PrebuiltConfigOutputGraphQLField, PrebuiltCriteriaGraphQLField, ProviderListGraphQLField, RemoteConfigOutputGraphQLField, RemoteEnvGraphQLField, RemoteEnvTestOfflineGraphQLField, RemoteEnvTestOnlineGraphQLField, RoleGraphQLField, SessionGraphQLField, SettingsGraphQLField, ShareGraphQLField, SystemPromptTemplateGraphQLField, TeamGraphQLField, TeamMemberGraphQLField, TeamWithroleGraphQLField, TimeseriesGraphQLField, ToolProviderGraphQLField, TrainingJobStageOutputGraphQLField, TrendResultGraphQLField, UnitConfigGraphQLField, UsageAggregateItemGraphQLField, UsageAggregatePerUseCaseItemGraphQLField, UsageGraphQLField, UsageStatsByModelGraphQLField, UsageStatsGraphQLField, UseCaseGraphQLField, UseCaseItemGraphQLField, UseCaseMetadataGraphQLField, UserGraphQLField, WidgetGraphQLField
|
|
4
4
|
from .input_types import AbCampaignFilter, CursorPageInput, FeedbackFilterInput, ListCompletionsFilterInput, MetricTrendInput, ModelServiceFilter, OrderPair, TimeRange, TimeseriesInput, UseCaseFilter
|
|
5
5
|
|
|
6
6
|
class AbReportFields(GraphQLField):
|
|
@@ -268,7 +268,7 @@ class CompletionFields(GraphQLField):
|
|
|
268
268
|
@classmethod
|
|
269
269
|
def completion(cls, *, max_length: Optional[int]=None) -> 'CompletionGraphQLField':
|
|
270
270
|
arguments: Dict[str, Dict[str, Any]] = {'maxLength': {'type': 'Int', 'value': max_length}}
|
|
271
|
-
cleared_arguments = {key: value for
|
|
271
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
|
|
272
272
|
return CompletionGraphQLField('completion', arguments=cleared_arguments)
|
|
273
273
|
source: 'CompletionGraphQLField' = CompletionGraphQLField('source')
|
|
274
274
|
|
|
@@ -283,7 +283,7 @@ class CompletionFields(GraphQLField):
|
|
|
283
283
|
@classmethod
|
|
284
284
|
def direct_feedbacks(cls, *, filter: Optional[FeedbackFilterInput]=None) -> 'DirectFeedbackFields':
|
|
285
285
|
arguments: Dict[str, Dict[str, Any]] = {'filter': {'type': 'FeedbackFilterInput', 'value': filter}}
|
|
286
|
-
cleared_arguments = {key: value for
|
|
286
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
|
|
287
287
|
return DirectFeedbackFields('directFeedbacks', arguments=cleared_arguments)
|
|
288
288
|
|
|
289
289
|
@classmethod
|
|
@@ -301,14 +301,14 @@ class CompletionFields(GraphQLField):
|
|
|
301
301
|
@classmethod
|
|
302
302
|
def labels(cls, with_protected: bool) -> 'CompletionLabelFields':
|
|
303
303
|
arguments: Dict[str, Dict[str, Any]] = {'withProtected': {'type': 'Boolean!', 'value': with_protected}}
|
|
304
|
-
cleared_arguments = {key: value for
|
|
304
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
|
|
305
305
|
return CompletionLabelFields('labels', arguments=cleared_arguments)
|
|
306
306
|
created_at: 'CompletionGraphQLField' = CompletionGraphQLField('createdAt')
|
|
307
307
|
|
|
308
308
|
@classmethod
|
|
309
309
|
def siblings_count(cls, filter: ListCompletionsFilterInput) -> 'CompletionGraphQLField':
|
|
310
310
|
arguments: Dict[str, Dict[str, Any]] = {'filter': {'type': 'ListCompletionsFilterInput!', 'value': filter}}
|
|
311
|
-
cleared_arguments = {key: value for
|
|
311
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
|
|
312
312
|
return CompletionGraphQLField('siblingsCount', arguments=cleared_arguments)
|
|
313
313
|
has_user_metadata: 'CompletionGraphQLField' = CompletionGraphQLField('hasUserMetadata')
|
|
314
314
|
user_metadata: 'CompletionGraphQLField' = CompletionGraphQLField('userMetadata')
|
|
@@ -387,7 +387,7 @@ class CompletionGroupDataFields(GraphQLField):
|
|
|
387
387
|
@classmethod
|
|
388
388
|
def completions(cls, page: CursorPageInput, order: List[OrderPair]) -> 'CompletionConnectionFields':
|
|
389
389
|
arguments: Dict[str, Dict[str, Any]] = {'page': {'type': 'CursorPageInput!', 'value': page}, 'order': {'type': '[OrderPair!]!', 'value': order}}
|
|
390
|
-
cleared_arguments = {key: value for
|
|
390
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
|
|
391
391
|
return CompletionConnectionFields('completions', arguments=cleared_arguments)
|
|
392
392
|
|
|
393
393
|
def fields(self, *subfields: Union[CompletionGroupDataGraphQLField, 'CompletionConnectionFields', 'CompletionGroupFeedbackStatsFields']) -> 'CompletionGroupDataFields':
|
|
@@ -557,7 +557,7 @@ class ContractFields(GraphQLField):
|
|
|
557
557
|
"""Get token usage on Adaptive models for the current billing cycle
|
|
558
558
|
returns an error if 'now' is before the start date of the contract"""
|
|
559
559
|
arguments: Dict[str, Dict[str, Any]] = {'now': {'type': 'InputDatetime', 'value': now}}
|
|
560
|
-
cleared_arguments = {key: value for
|
|
560
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
|
|
561
561
|
return BillingUsageFields('usage', arguments=cleared_arguments)
|
|
562
562
|
|
|
563
563
|
def fields(self, *subfields: Union[ContractGraphQLField, 'BillingUsageFields']) -> 'ContractFields':
|
|
@@ -601,7 +601,7 @@ class CustomRecipeFields(GraphQLField):
|
|
|
601
601
|
@classmethod
|
|
602
602
|
def labels(cls, with_protected: bool) -> 'LabelFields':
|
|
603
603
|
arguments: Dict[str, Dict[str, Any]] = {'withProtected': {'type': 'Boolean!', 'value': with_protected}}
|
|
604
|
-
cleared_arguments = {key: value for
|
|
604
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
|
|
605
605
|
return LabelFields('labels', arguments=cleared_arguments)
|
|
606
606
|
updated_at: 'CustomRecipeGraphQLField' = CustomRecipeGraphQLField('updatedAt')
|
|
607
607
|
|
|
@@ -649,9 +649,14 @@ class DatasetFields(GraphQLField):
|
|
|
649
649
|
def metrics_usage(cls) -> 'DatasetMetricUsageFields':
|
|
650
650
|
return DatasetMetricUsageFields('metricsUsage')
|
|
651
651
|
source: 'DatasetGraphQLField' = DatasetGraphQLField('source')
|
|
652
|
+
status: 'DatasetGraphQLField' = DatasetGraphQLField('status')
|
|
652
653
|
deleted: 'DatasetGraphQLField' = DatasetGraphQLField('deleted')
|
|
653
654
|
|
|
654
|
-
|
|
655
|
+
@classmethod
|
|
656
|
+
def progress(cls) -> 'DatasetProgressFields':
|
|
657
|
+
return DatasetProgressFields('progress')
|
|
658
|
+
|
|
659
|
+
def fields(self, *subfields: Union[DatasetGraphQLField, 'DatasetMetricUsageFields', 'DatasetProgressFields']) -> 'DatasetFields':
|
|
655
660
|
"""Subfields should come from the DatasetFields class"""
|
|
656
661
|
self._subfields.extend(subfields)
|
|
657
662
|
return self
|
|
@@ -694,6 +699,41 @@ class DatasetMetricUsageFields(GraphQLField):
|
|
|
694
699
|
self._alias = alias
|
|
695
700
|
return self
|
|
696
701
|
|
|
702
|
+
class DatasetProgressFields(GraphQLField):
|
|
703
|
+
"""@private"""
|
|
704
|
+
processed_parts: 'DatasetProgressGraphQLField' = DatasetProgressGraphQLField('processedParts')
|
|
705
|
+
total_parts: 'DatasetProgressGraphQLField' = DatasetProgressGraphQLField('totalParts')
|
|
706
|
+
progress: 'DatasetProgressGraphQLField' = DatasetProgressGraphQLField('progress')
|
|
707
|
+
processed_lines: 'DatasetProgressGraphQLField' = DatasetProgressGraphQLField('processedLines')
|
|
708
|
+
total_lines: 'DatasetProgressGraphQLField' = DatasetProgressGraphQLField('totalLines')
|
|
709
|
+
|
|
710
|
+
def fields(self, *subfields: DatasetProgressGraphQLField) -> 'DatasetProgressFields':
|
|
711
|
+
"""Subfields should come from the DatasetProgressFields class"""
|
|
712
|
+
self._subfields.extend(subfields)
|
|
713
|
+
return self
|
|
714
|
+
|
|
715
|
+
def alias(self, alias: str) -> 'DatasetProgressFields':
|
|
716
|
+
self._alias = alias
|
|
717
|
+
return self
|
|
718
|
+
|
|
719
|
+
class DatasetUploadProcessingStatusFields(GraphQLField):
|
|
720
|
+
"""@private"""
|
|
721
|
+
dataset_id: 'DatasetUploadProcessingStatusGraphQLField' = DatasetUploadProcessingStatusGraphQLField('datasetId')
|
|
722
|
+
status: 'DatasetUploadProcessingStatusGraphQLField' = DatasetUploadProcessingStatusGraphQLField('status')
|
|
723
|
+
total_parts: 'DatasetUploadProcessingStatusGraphQLField' = DatasetUploadProcessingStatusGraphQLField('totalParts')
|
|
724
|
+
processed_parts: 'DatasetUploadProcessingStatusGraphQLField' = DatasetUploadProcessingStatusGraphQLField('processedParts')
|
|
725
|
+
progress: 'DatasetUploadProcessingStatusGraphQLField' = DatasetUploadProcessingStatusGraphQLField('progress')
|
|
726
|
+
error: 'DatasetUploadProcessingStatusGraphQLField' = DatasetUploadProcessingStatusGraphQLField('error')
|
|
727
|
+
|
|
728
|
+
def fields(self, *subfields: DatasetUploadProcessingStatusGraphQLField) -> 'DatasetUploadProcessingStatusFields':
|
|
729
|
+
"""Subfields should come from the DatasetUploadProcessingStatusFields class"""
|
|
730
|
+
self._subfields.extend(subfields)
|
|
731
|
+
return self
|
|
732
|
+
|
|
733
|
+
def alias(self, alias: str) -> 'DatasetUploadProcessingStatusFields':
|
|
734
|
+
self._alias = alias
|
|
735
|
+
return self
|
|
736
|
+
|
|
697
737
|
class DatasetValidationOutputFields(GraphQLField):
|
|
698
738
|
"""@private"""
|
|
699
739
|
valid: 'DatasetValidationOutputGraphQLField' = DatasetValidationOutputGraphQLField('valid')
|
|
@@ -1274,13 +1314,13 @@ class MetricFields(GraphQLField):
|
|
|
1274
1314
|
def use_cases(cls, filter: UseCaseFilter) -> 'UseCaseFields':
|
|
1275
1315
|
"""Return the list of UseCase which use this metric"""
|
|
1276
1316
|
arguments: Dict[str, Dict[str, Any]] = {'filter': {'type': 'UseCaseFilter!', 'value': filter}}
|
|
1277
|
-
cleared_arguments = {key: value for
|
|
1317
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
|
|
1278
1318
|
return UseCaseFields('useCases', arguments=cleared_arguments)
|
|
1279
1319
|
|
|
1280
1320
|
@classmethod
|
|
1281
1321
|
def activity(cls, *, timerange: Optional[TimeRange]=None) -> 'MetricActivityFields':
|
|
1282
1322
|
arguments: Dict[str, Dict[str, Any]] = {'timerange': {'type': 'TimeRange', 'value': timerange}}
|
|
1283
|
-
cleared_arguments = {key: value for
|
|
1323
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
|
|
1284
1324
|
return MetricActivityFields('activity', arguments=cleared_arguments)
|
|
1285
1325
|
has_direct_feedbacks: 'MetricGraphQLField' = MetricGraphQLField('hasDirectFeedbacks')
|
|
1286
1326
|
has_comparison_feedbacks: 'MetricGraphQLField' = MetricGraphQLField('hasComparisonFeedbacks')
|
|
@@ -1324,25 +1364,25 @@ class MetricWithContextFields(GraphQLField):
|
|
|
1324
1364
|
@classmethod
|
|
1325
1365
|
def feedback_count(cls, *, timerange: Optional[TimeRange]=None) -> 'MetricWithContextGraphQLField':
|
|
1326
1366
|
arguments: Dict[str, Dict[str, Any]] = {'timerange': {'type': 'TimeRange', 'value': timerange}}
|
|
1327
|
-
cleared_arguments = {key: value for
|
|
1367
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
|
|
1328
1368
|
return MetricWithContextGraphQLField('feedbackCount', arguments=cleared_arguments)
|
|
1329
1369
|
|
|
1330
1370
|
@classmethod
|
|
1331
1371
|
def comparison_count(cls, *, timerange: Optional[TimeRange]=None) -> 'MetricWithContextGraphQLField':
|
|
1332
1372
|
arguments: Dict[str, Dict[str, Any]] = {'timerange': {'type': 'TimeRange', 'value': timerange}}
|
|
1333
|
-
cleared_arguments = {key: value for
|
|
1373
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
|
|
1334
1374
|
return MetricWithContextGraphQLField('comparisonCount', arguments=cleared_arguments)
|
|
1335
1375
|
|
|
1336
1376
|
@classmethod
|
|
1337
1377
|
def trend(cls, input: MetricTrendInput) -> 'TrendResultFields':
|
|
1338
1378
|
arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'MetricTrendInput!', 'value': input}}
|
|
1339
|
-
cleared_arguments = {key: value for
|
|
1379
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
|
|
1340
1380
|
return TrendResultFields('trend', arguments=cleared_arguments)
|
|
1341
1381
|
|
|
1342
1382
|
@classmethod
|
|
1343
1383
|
def timeseries(cls, input: TimeseriesInput) -> 'TimeseriesFields':
|
|
1344
1384
|
arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'TimeseriesInput!', 'value': input}}
|
|
1345
|
-
cleared_arguments = {key: value for
|
|
1385
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
|
|
1346
1386
|
return TimeseriesFields('timeseries', arguments=cleared_arguments)
|
|
1347
1387
|
has_comparison_feedbacks: 'MetricWithContextGraphQLField' = MetricWithContextGraphQLField('hasComparisonFeedbacks')
|
|
1348
1388
|
has_direct_feedbacks: 'MetricWithContextGraphQLField' = MetricWithContextGraphQLField('hasDirectFeedbacks')
|
|
@@ -1369,7 +1409,7 @@ class ModelFields(GraphQLField):
|
|
|
1369
1409
|
@classmethod
|
|
1370
1410
|
def activity(cls, *, timerange: Optional[TimeRange]=None) -> 'ActivityFields':
|
|
1371
1411
|
arguments: Dict[str, Dict[str, Any]] = {'timerange': {'type': 'TimeRange', 'value': timerange}}
|
|
1372
|
-
cleared_arguments = {key: value for
|
|
1412
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
|
|
1373
1413
|
return ActivityFields('activity', arguments=cleared_arguments)
|
|
1374
1414
|
|
|
1375
1415
|
@classmethod
|
|
@@ -1380,13 +1420,13 @@ class ModelFields(GraphQLField):
|
|
|
1380
1420
|
def use_cases(cls, filter: UseCaseFilter, attached: bool) -> 'UseCaseFields':
|
|
1381
1421
|
"""Return the list of UseCase which use this model"""
|
|
1382
1422
|
arguments: Dict[str, Dict[str, Any]] = {'filter': {'type': 'UseCaseFilter!', 'value': filter}, 'attached': {'type': 'Boolean!', 'value': attached}}
|
|
1383
|
-
cleared_arguments = {key: value for
|
|
1423
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
|
|
1384
1424
|
return UseCaseFields('useCases', arguments=cleared_arguments)
|
|
1385
1425
|
|
|
1386
1426
|
@classmethod
|
|
1387
1427
|
def model_services(cls, filter: UseCaseFilter, attached: bool) -> 'ModelServiceFields':
|
|
1388
1428
|
arguments: Dict[str, Dict[str, Any]] = {'filter': {'type': 'UseCaseFilter!', 'value': filter}, 'attached': {'type': 'Boolean!', 'value': attached}}
|
|
1389
|
-
cleared_arguments = {key: value for
|
|
1429
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
|
|
1390
1430
|
return ModelServiceFields('modelServices', arguments=cleared_arguments)
|
|
1391
1431
|
provider_name: 'ModelGraphQLField' = ModelGraphQLField('providerName')
|
|
1392
1432
|
is_external: 'ModelGraphQLField' = ModelGraphQLField('isExternal')
|
|
@@ -1397,9 +1437,16 @@ class ModelFields(GraphQLField):
|
|
|
1397
1437
|
@classmethod
|
|
1398
1438
|
def backbone(cls) -> 'ModelFields':
|
|
1399
1439
|
return ModelFields('backbone')
|
|
1440
|
+
|
|
1441
|
+
@classmethod
|
|
1442
|
+
def parent(cls) -> 'ModelFields':
|
|
1443
|
+
return ModelFields('parent')
|
|
1400
1444
|
is_training: 'ModelGraphQLField' = ModelGraphQLField('isTraining')
|
|
1401
1445
|
'indicates if a training is pending or running for this model'
|
|
1402
|
-
|
|
1446
|
+
capabilities: 'ModelGraphQLField' = ModelGraphQLField('capabilities')
|
|
1447
|
+
supported_tp: 'ModelGraphQLField' = ModelGraphQLField('supportedTp')
|
|
1448
|
+
family: 'ModelGraphQLField' = ModelGraphQLField('family')
|
|
1449
|
+
publisher: 'ModelGraphQLField' = ModelGraphQLField('publisher')
|
|
1403
1450
|
size: 'ModelGraphQLField' = ModelGraphQLField('size')
|
|
1404
1451
|
|
|
1405
1452
|
@classmethod
|
|
@@ -1415,6 +1462,22 @@ class ModelFields(GraphQLField):
|
|
|
1415
1462
|
self._alias = alias
|
|
1416
1463
|
return self
|
|
1417
1464
|
|
|
1465
|
+
class ModelByproductsFields(GraphQLField):
|
|
1466
|
+
"""@private"""
|
|
1467
|
+
|
|
1468
|
+
@classmethod
|
|
1469
|
+
def model(cls) -> 'ModelFields':
|
|
1470
|
+
return ModelFields('model')
|
|
1471
|
+
|
|
1472
|
+
def fields(self, *subfields: Union[ModelByproductsGraphQLField, 'ModelFields']) -> 'ModelByproductsFields':
|
|
1473
|
+
"""Subfields should come from the ModelByproductsFields class"""
|
|
1474
|
+
self._subfields.extend(subfields)
|
|
1475
|
+
return self
|
|
1476
|
+
|
|
1477
|
+
def alias(self, alias: str) -> 'ModelByproductsFields':
|
|
1478
|
+
self._alias = alias
|
|
1479
|
+
return self
|
|
1480
|
+
|
|
1418
1481
|
class ModelComputeConfigOutputFields(GraphQLField):
|
|
1419
1482
|
"""@private"""
|
|
1420
1483
|
tp: 'ModelComputeConfigOutputGraphQLField' = ModelComputeConfigOutputGraphQLField('tp')
|
|
@@ -1464,7 +1527,7 @@ class ModelServiceFields(GraphQLField):
|
|
|
1464
1527
|
@classmethod
|
|
1465
1528
|
def activity(cls, *, timerange: Optional[TimeRange]=None) -> 'ActivityFields':
|
|
1466
1529
|
arguments: Dict[str, Dict[str, Any]] = {'timerange': {'type': 'TimeRange', 'value': timerange}}
|
|
1467
|
-
cleared_arguments = {key: value for
|
|
1530
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
|
|
1468
1531
|
return ActivityFields('activity', arguments=cleared_arguments)
|
|
1469
1532
|
|
|
1470
1533
|
@classmethod
|
|
@@ -1478,7 +1541,7 @@ class ModelServiceFields(GraphQLField):
|
|
|
1478
1541
|
@classmethod
|
|
1479
1542
|
def ab_campaigns(cls, filter: AbCampaignFilter) -> 'AbcampaignFields':
|
|
1480
1543
|
arguments: Dict[str, Dict[str, Any]] = {'filter': {'type': 'AbCampaignFilter!', 'value': filter}}
|
|
1481
|
-
cleared_arguments = {key: value for
|
|
1544
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
|
|
1482
1545
|
return AbcampaignFields('abCampaigns', arguments=cleared_arguments)
|
|
1483
1546
|
|
|
1484
1547
|
@classmethod
|
|
@@ -1994,13 +2057,13 @@ class UseCaseFields(GraphQLField):
|
|
|
1994
2057
|
@classmethod
|
|
1995
2058
|
def model_services(cls, *, filter: Optional[ModelServiceFilter]=None) -> 'ModelServiceFields':
|
|
1996
2059
|
arguments: Dict[str, Dict[str, Any]] = {'filter': {'type': 'ModelServiceFilter', 'value': filter}}
|
|
1997
|
-
cleared_arguments = {key: value for
|
|
2060
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
|
|
1998
2061
|
return ModelServiceFields('modelServices', arguments=cleared_arguments)
|
|
1999
2062
|
|
|
2000
2063
|
@classmethod
|
|
2001
2064
|
def model_service(cls, id_or_key: str) -> 'ModelServiceFields':
|
|
2002
2065
|
arguments: Dict[str, Dict[str, Any]] = {'idOrKey': {'type': 'IdOrKey!', 'value': id_or_key}}
|
|
2003
|
-
cleared_arguments = {key: value for
|
|
2066
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
|
|
2004
2067
|
return ModelServiceFields('modelService', arguments=cleared_arguments)
|
|
2005
2068
|
|
|
2006
2069
|
@classmethod
|
|
@@ -2010,7 +2073,7 @@ class UseCaseFields(GraphQLField):
|
|
|
2010
2073
|
@classmethod
|
|
2011
2074
|
def activity(cls, *, timerange: Optional[TimeRange]=None) -> 'ActivityFields':
|
|
2012
2075
|
arguments: Dict[str, Dict[str, Any]] = {'timerange': {'type': 'TimeRange', 'value': timerange}}
|
|
2013
|
-
cleared_arguments = {key: value for
|
|
2076
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
|
|
2014
2077
|
return ActivityFields('activity', arguments=cleared_arguments)
|
|
2015
2078
|
|
|
2016
2079
|
@classmethod
|
|
@@ -2020,13 +2083,13 @@ class UseCaseFields(GraphQLField):
|
|
|
2020
2083
|
@classmethod
|
|
2021
2084
|
def metric(cls, metric: str) -> 'MetricWithContextFields':
|
|
2022
2085
|
arguments: Dict[str, Dict[str, Any]] = {'metric': {'type': 'IdOrKey!', 'value': metric}}
|
|
2023
|
-
cleared_arguments = {key: value for
|
|
2086
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
|
|
2024
2087
|
return MetricWithContextFields('metric', arguments=cleared_arguments)
|
|
2025
2088
|
|
|
2026
2089
|
@classmethod
|
|
2027
2090
|
def ab_campaigns(cls, filter: AbCampaignFilter) -> 'AbcampaignFields':
|
|
2028
2091
|
arguments: Dict[str, Dict[str, Any]] = {'filter': {'type': 'AbCampaignFilter!', 'value': filter}}
|
|
2029
|
-
cleared_arguments = {key: value for
|
|
2092
|
+
cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
|
|
2030
2093
|
return AbcampaignFields('abCampaigns', arguments=cleared_arguments)
|
|
2031
2094
|
|
|
2032
2095
|
@classmethod
|