adaptive-sdk 0.1.2__py3-none-any.whl → 0.1.4__py3-none-any.whl

adaptive_sdk/graphql_client/custom_fields.py

@@ -1,6 +1,6 @@
 from typing import Any, Dict, Optional, Union
 from .base_operation import GraphQLField
-from .custom_typing_fields import AbcampaignGraphQLField, AbReportGraphQLField, AbVariantReportComparisonGraphQLField, AbVariantReportGraphQLField, ActivityGraphQLField, ActivityOutputGraphQLField, AdaptBuiltinRecipeConfigOutputGraphQLField, AdaptCustomRecipeConfigOutputGraphQLField, AdaptRequestConfigOutputUnion, ApiKeyGraphQLField, ArtifactByproductsUnion, AuthProviderGraphQLField, BaseTrainingParamsOutputGraphQLField, BatchInferenceJobStageOutputGraphQLField, ChatMessageGraphQLField, ComparisonFeedbackGraphQLField, CompletionConnectionGraphQLField, CompletionEdgeGraphQLField, CompletionFeedbackFilterOutputGraphQLField, CompletionGraphQLField, CompletionGroupDataConnectionGraphQLField, CompletionGroupDataEdgeGraphQLField, CompletionGroupDataGraphQLField, CompletionGroupFeedbackStatsGraphQLField, CompletionHistoryEntryOuputGraphQLField, CompletionLabelFilterOutputGraphQLField, CompletionLabelGraphQLField, CompletionMetadataGraphQLField, ComputePoolGraphQLField, CustomConfigOutputGraphQLField, CustomRecipeGraphQLField, CustomRecipeJobDetailsGraphQLField, DatasetGraphQLField, DatasetMetricUsageGraphQLField, DatasetValidationOutputGraphQLField, DeleteConfirmGraphQLField, DirectFeedbackGraphQLField, DpotrainingParamsOutputGraphQLField, EmojiGraphQLField, EvalJobStageOutputGraphQLField, EvaluationAnswerRelevancyRecipeGraphQLField, EvaluationByproductsGraphQLField, EvaluationContextRelevancyRecipeGraphQLField, EvaluationCustomRecipeGraphQLField, EvaluationFaithfulnessRecipeGraphQLField, EvaluationJobGraphQLField, EvaluationRecipeUnion, EvaluationResultGraphQLField, GpuAllocationGraphQLField, GraderConfigUnion, GraderGraphQLField, GrpotrainingParamsOutputGraphQLField, GuidelineGraphQLField, GuidelinesTrainingParamsOutputGraphQLField, InteractionOutputGraphQLField, IntervalGraphQLField, JobArtifactGraphQLField, JobConnectionGraphQLField, JobEdgeGraphQLField, JobGraphQLField, JobStageInfoOutputUnion, JobStageOutputGraphQLField, JudgeConfigOutputGraphQLField, JudgeExampleGraphQLField, JudgeGraphQLField, JudgeTrainingParamsOutputGraphQLField, LabelGraphQLField, LabelKeyUsageGraphQLField, LabelUsageGraphQLField, LabelValueUsageGraphQLField, ListCompletionsFilterOutputGraphQLField, MetaObjectGraphQLField, MetricActivityGraphQLField, MetricGraphQLField, MetricTrainingParamsMetadataOutputUnion, MetricTrainingParamsOutputGraphQLField, MetricWithContextGraphQLField, ModelComputeConfigOutputGraphQLField, ModelGraphQLField, ModelPlacementOutputGraphQLField, ModelServiceGraphQLField, PageInfoGraphQLField, PartitionGraphQLField, PpotrainingParamsOutputGraphQLField, PrebuiltConfigDefinitionGraphQLField, PrebuiltConfigOutputGraphQLField, PrebuiltCriteriaGraphQLField, ProviderListGraphQLField, RemoteConfigOutputGraphQLField, RemoteEnvGraphQLField, RemoteEnvTestOfflineGraphQLField, RemoteEnvTestOnlineGraphQLField, RewardServerTrainingParamsOutputGraphQLField, RoleGraphQLField, SampleConfigOutputGraphQLField, SampleDatasourceCompletionsOutputGraphQLField, SampleDatasourceDatasetOutputGraphQLField, SampleDatasourceOutputUnion, ScalarMetricConfigOutputGraphQLField, SessionGraphQLField, SettingsGraphQLField, SfttrainingParamsOutputGraphQLField, ShareGraphQLField, SystemPromptTemplateGraphQLField, TeamGraphQLField, TeamMemberGraphQLField, TeamWithroleGraphQLField, TimeRangeOutputGraphQLField, TimeseriesGraphQLField, ToolProviderGraphQLField, TrainingConfigOutputGraphQLField, TrainingJobGraphQLField, TrainingJobStageOutputGraphQLField, TrainingMetadataOutputGraphQLField, TrainingMetadataOutputParametersUnion, TrainingObjectiveOutputUnion, TrendResultGraphQLField, UnitConfigGraphQLField, UsageAggregateItemGraphQLField, UsageAggregatePerUseCaseItemGraphQLField, UsageGraphQLField, UseCaseGraphQLField, UseCaseItemGraphQLField, UseCaseMetadataGraphQLField, UserGraphQLField, WidgetGraphQLField
+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
 from .input_types import AbCampaignFilter, CursorPageInput, FeedbackFilterInput, ListCompletionsFilterInput, MetricTrendInput, ModelServiceFilter, OrderPair, TimeRange, TimeseriesInput, UseCaseFilter
 
 class AbReportFields(GraphQLField):
@@ -141,41 +141,6 @@ class ActivityOutputFields(GraphQLField):
         self._alias = alias
         return self
 
-class AdaptBuiltinRecipeConfigOutputFields(GraphQLField):
-    """@private"""
-    output_name: 'AdaptBuiltinRecipeConfigOutputGraphQLField' = AdaptBuiltinRecipeConfigOutputGraphQLField('outputName')
-
-    @classmethod
-    def sample_config(cls) -> 'SampleConfigOutputFields':
-        return SampleConfigOutputFields('sample_config')
-
-    @classmethod
-    def training_config(cls) -> 'TrainingConfigOutputFields':
-        return TrainingConfigOutputFields('training_config')
-
-    def fields(self, *subfields: Union[AdaptBuiltinRecipeConfigOutputGraphQLField, 'SampleConfigOutputFields', 'TrainingConfigOutputFields']) -> 'AdaptBuiltinRecipeConfigOutputFields':
-        """Subfields should come from the AdaptBuiltinRecipeConfigOutputFields class"""
-        self._subfields.extend(subfields)
-        return self
-
-    def alias(self, alias: str) -> 'AdaptBuiltinRecipeConfigOutputFields':
-        self._alias = alias
-        return self
-
-class AdaptCustomRecipeConfigOutputFields(GraphQLField):
-    """@private"""
-    output_name: 'AdaptCustomRecipeConfigOutputGraphQLField' = AdaptCustomRecipeConfigOutputGraphQLField('outputName')
-    args: 'AdaptCustomRecipeConfigOutputGraphQLField' = AdaptCustomRecipeConfigOutputGraphQLField('args')
-
-    def fields(self, *subfields: AdaptCustomRecipeConfigOutputGraphQLField) -> 'AdaptCustomRecipeConfigOutputFields':
-        """Subfields should come from the AdaptCustomRecipeConfigOutputFields class"""
-        self._subfields.extend(subfields)
-        return self
-
-    def alias(self, alias: str) -> 'AdaptCustomRecipeConfigOutputFields':
-        self._alias = alias
-        return self
-
 class ApiKeyFields(GraphQLField):
     """@private"""
     key: 'ApiKeyGraphQLField' = ApiKeyGraphQLField('key')
@@ -206,34 +171,41 @@ class AuthProviderFields(GraphQLField):
         self._alias = alias
         return self
 
-class BaseTrainingParamsOutputFields(GraphQLField):
+class BatchInferenceJobStageOutputFields(GraphQLField):
     """@private"""
-    learning_rate: 'BaseTrainingParamsOutputGraphQLField' = BaseTrainingParamsOutputGraphQLField('learningRate')
-    num_epochs: 'BaseTrainingParamsOutputGraphQLField' = BaseTrainingParamsOutputGraphQLField('numEpochs')
-    batch_size: 'BaseTrainingParamsOutputGraphQLField' = BaseTrainingParamsOutputGraphQLField('batchSize')
-    num_validations: 'BaseTrainingParamsOutputGraphQLField' = BaseTrainingParamsOutputGraphQLField('numValidations')
+    total_num_samples: 'BatchInferenceJobStageOutputGraphQLField' = BatchInferenceJobStageOutputGraphQLField('totalNumSamples')
+    processed_num_samples: 'BatchInferenceJobStageOutputGraphQLField' = BatchInferenceJobStageOutputGraphQLField('processedNumSamples')
+    monitoring_link: 'BatchInferenceJobStageOutputGraphQLField' = BatchInferenceJobStageOutputGraphQLField('monitoringLink')
 
-    def fields(self, *subfields: BaseTrainingParamsOutputGraphQLField) -> 'BaseTrainingParamsOutputFields':
-        """Subfields should come from the BaseTrainingParamsOutputFields class"""
+    def fields(self, *subfields: BatchInferenceJobStageOutputGraphQLField) -> 'BatchInferenceJobStageOutputFields':
+        """Subfields should come from the BatchInferenceJobStageOutputFields class"""
         self._subfields.extend(subfields)
         return self
 
-    def alias(self, alias: str) -> 'BaseTrainingParamsOutputFields':
+    def alias(self, alias: str) -> 'BatchInferenceJobStageOutputFields':
         self._alias = alias
         return self
 
-class BatchInferenceJobStageOutputFields(GraphQLField):
+class BillingUsageFields(GraphQLField):
     """@private"""
-    total_num_samples: 'BatchInferenceJobStageOutputGraphQLField' = BatchInferenceJobStageOutputGraphQLField('totalNumSamples')
-    processed_num_samples: 'BatchInferenceJobStageOutputGraphQLField' = BatchInferenceJobStageOutputGraphQLField('processedNumSamples')
-    monitoring_link: 'BatchInferenceJobStageOutputGraphQLField' = BatchInferenceJobStageOutputGraphQLField('monitoringLink')
+    now: 'BillingUsageGraphQLField' = BillingUsageGraphQLField('now')
+    start: 'BillingUsageGraphQLField' = BillingUsageGraphQLField('start')
+    end: 'BillingUsageGraphQLField' = BillingUsageGraphQLField('end')
 
-    def fields(self, *subfields: BatchInferenceJobStageOutputGraphQLField) -> 'BatchInferenceJobStageOutputFields':
-        """Subfields should come from the BatchInferenceJobStageOutputFields class"""
+    @classmethod
+    def usage(cls) -> 'UsageStatsFields':
+        return UsageStatsFields('usage')
+
+    @classmethod
+    def projection(cls) -> 'UsageStatsFields':
+        return UsageStatsFields('projection')
+
+    def fields(self, *subfields: Union[BillingUsageGraphQLField, 'UsageStatsFields']) -> 'BillingUsageFields':
+        """Subfields should come from the BillingUsageFields class"""
         self._subfields.extend(subfields)
         return self
 
-    def alias(self, alias: str) -> 'BatchInferenceJobStageOutputFields':
+    def alias(self, alias: str) -> 'BillingUsageFields':
         self._alias = alias
         return self
 
@@ -335,10 +307,13 @@ class CompletionFields(GraphQLField):
         arguments: Dict[str, Dict[str, Any]] = {'filter': {'type': 'ListCompletionsFilterInput!', 'value': filter}}
         cleared_arguments = {key: value for (key, value) in arguments.items() if value['value'] is not None}
         return CompletionGraphQLField('siblings_count', arguments=cleared_arguments)
+    has_user_metadata: 'CompletionGraphQLField' = CompletionGraphQLField('hasUserMetadata')
+    user_metadata: 'CompletionGraphQLField' = CompletionGraphQLField('userMetadata')
 
     @classmethod
     def metadata(cls) -> 'CompletionMetadataFields':
         return CompletionMetadataFields('metadata')
+    can_edit: 'CompletionGraphQLField' = CompletionGraphQLField('canEdit')
 
     def fields(self, *subfields: Union[CompletionGraphQLField, 'ChatMessageFields', 'ComparisonFeedbackFields', 'CompletionHistoryEntryOuputFields', 'CompletionLabelFields', 'CompletionMetadataFields', 'DirectFeedbackFields', 'ModelFields', 'ModelServiceFields', 'SessionFields']) -> 'CompletionFields':
         """Subfields should come from the CompletionFields class"""
@@ -391,27 +366,6 @@ class CompletionEdgeFields(GraphQLField):
         self._alias = alias
         return self
 
-class CompletionFeedbackFilterOutputFields(GraphQLField):
-    """@private"""
-    metric: 'CompletionFeedbackFilterOutputGraphQLField' = CompletionFeedbackFilterOutputGraphQLField('metric')
-    gt: 'CompletionFeedbackFilterOutputGraphQLField' = CompletionFeedbackFilterOutputGraphQLField('gt')
-    gte: 'CompletionFeedbackFilterOutputGraphQLField' = CompletionFeedbackFilterOutputGraphQLField('gte')
-    eq: 'CompletionFeedbackFilterOutputGraphQLField' = CompletionFeedbackFilterOutputGraphQLField('eq')
-    neq: 'CompletionFeedbackFilterOutputGraphQLField' = CompletionFeedbackFilterOutputGraphQLField('neq')
-    lt: 'CompletionFeedbackFilterOutputGraphQLField' = CompletionFeedbackFilterOutputGraphQLField('lt')
-    lte: 'CompletionFeedbackFilterOutputGraphQLField' = CompletionFeedbackFilterOutputGraphQLField('lte')
-    reasons: 'CompletionFeedbackFilterOutputGraphQLField' = CompletionFeedbackFilterOutputGraphQLField('reasons')
-    user: 'CompletionFeedbackFilterOutputGraphQLField' = CompletionFeedbackFilterOutputGraphQLField('user')
-
-    def fields(self, *subfields: CompletionFeedbackFilterOutputGraphQLField) -> 'CompletionFeedbackFilterOutputFields':
-        """Subfields should come from the CompletionFeedbackFilterOutputFields class"""
-        self._subfields.extend(subfields)
-        return self
-
-    def alias(self, alias: str) -> 'CompletionFeedbackFilterOutputFields':
-        self._alias = alias
-        return self
-
 class CompletionGroupDataFields(GraphQLField):
     """@private"""
     key: 'CompletionGroupDataGraphQLField' = CompletionGroupDataGraphQLField('key')
@@ -532,20 +486,6 @@ class CompletionLabelFields(GraphQLField):
         self._alias = alias
         return self
 
-class CompletionLabelFilterOutputFields(GraphQLField):
-    """@private"""
-    key: 'CompletionLabelFilterOutputGraphQLField' = CompletionLabelFilterOutputGraphQLField('key')
-    value: 'CompletionLabelFilterOutputGraphQLField' = CompletionLabelFilterOutputGraphQLField('value')
-
-    def fields(self, *subfields: CompletionLabelFilterOutputGraphQLField) -> 'CompletionLabelFilterOutputFields':
-        """Subfields should come from the CompletionLabelFilterOutputFields class"""
-        self._subfields.extend(subfields)
-        return self
-
-    def alias(self, alias: str) -> 'CompletionLabelFilterOutputFields':
-        self._alias = alias
-        return self
-
 class CompletionMetadataFields(GraphQLField):
     """@private"""
     parameters: 'CompletionMetadataGraphQLField' = CompletionMetadataGraphQLField('parameters')
@@ -573,15 +513,15 @@ class ComputePoolFields(GraphQLField):
     created_at: 'ComputePoolGraphQLField' = ComputePoolGraphQLField('createdAt')
 
     @classmethod
-    def all_partitions(cls) -> 'PartitionFields':
-        return PartitionFields('all_partitions')
+    def all_harmony_groups(cls) -> 'HarmonyGroupFields':
+        return HarmonyGroupFields('all_harmony_groups')
 
     @classmethod
-    def partitions(cls) -> 'PartitionFields':
-        return PartitionFields('partitions')
+    def harmony_groups(cls) -> 'HarmonyGroupFields':
+        return HarmonyGroupFields('harmony_groups')
     capabilities: 'ComputePoolGraphQLField' = ComputePoolGraphQLField('capabilities')
 
-    def fields(self, *subfields: Union[ComputePoolGraphQLField, 'PartitionFields']) -> 'ComputePoolFields':
+    def fields(self, *subfields: Union[ComputePoolGraphQLField, 'HarmonyGroupFields']) -> 'ComputePoolFields':
         """Subfields should come from the ComputePoolFields class"""
         self._subfields.extend(subfields)
         return self
@@ -590,6 +530,28 @@ class ComputePoolFields(GraphQLField):
         self._alias = alias
         return self
 
+class ContractFields(GraphQLField):
+    """@private"""
+    quota: 'ContractGraphQLField' = ContractGraphQLField('quota')
+    start_date: 'ContractGraphQLField' = ContractGraphQLField('startDate')
+    end_date: 'ContractGraphQLField' = ContractGraphQLField('endDate')
+    cycle: 'ContractGraphQLField' = ContractGraphQLField('cycle')
+
+    @classmethod
+    def usage(cls, *, now: Optional[int | str]=None) -> 'BillingUsageFields':
+        arguments: Dict[str, Dict[str, Any]] = {'now': {'type': 'InputDatetime', 'value': now}}
+        cleared_arguments = {key: value for (key, value) in arguments.items() if value['value'] is not None}
+        return BillingUsageFields('usage', arguments=cleared_arguments)
+
+    def fields(self, *subfields: Union[ContractGraphQLField, 'BillingUsageFields']) -> 'ContractFields':
+        """Subfields should come from the ContractFields class"""
+        self._subfields.extend(subfields)
+        return self
+
+    def alias(self, alias: str) -> 'ContractFields':
+        self._alias = alias
+        return self
+
 class CustomConfigOutputFields(GraphQLField):
     """@private"""
     description: 'CustomConfigOutputGraphQLField' = CustomConfigOutputGraphQLField('description')
@@ -609,6 +571,7 @@ class CustomRecipeFields(GraphQLField):
     key: 'CustomRecipeGraphQLField' = CustomRecipeGraphQLField('key')
     name: 'CustomRecipeGraphQLField' = CustomRecipeGraphQLField('name')
     editable: 'CustomRecipeGraphQLField' = CustomRecipeGraphQLField('editable')
+    hidden: 'CustomRecipeGraphQLField' = CustomRecipeGraphQLField('hidden')
     builtin: 'CustomRecipeGraphQLField' = CustomRecipeGraphQLField('builtin')
     global_: 'CustomRecipeGraphQLField' = CustomRecipeGraphQLField('global')
     created_at: 'CustomRecipeGraphQLField' = CustomRecipeGraphQLField('createdAt')
@@ -680,6 +643,22 @@ class DatasetFields(GraphQLField):
         self._alias = alias
         return self
 
+class DatasetByproductsFields(GraphQLField):
+    """@private"""
+
+    @classmethod
+    def dataset(cls) -> 'DatasetFields':
+        return DatasetFields('dataset')
+
+    def fields(self, *subfields: Union[DatasetByproductsGraphQLField, 'DatasetFields']) -> 'DatasetByproductsFields':
+        """Subfields should come from the DatasetByproductsFields class"""
+        self._subfields.extend(subfields)
+        return self
+
+    def alias(self, alias: str) -> 'DatasetByproductsFields':
+        self._alias = alias
+        return self
+
 class DatasetMetricUsageFields(GraphQLField):
     """@private"""
 
@@ -748,19 +727,6 @@ class DirectFeedbackFields(GraphQLField):
         self._alias = alias
         return self
 
-class DpotrainingParamsOutputFields(GraphQLField):
-    """@private"""
-    kl_div_coeff: 'DpotrainingParamsOutputGraphQLField' = DpotrainingParamsOutputGraphQLField('klDivCoeff')
-
-    def fields(self, *subfields: DpotrainingParamsOutputGraphQLField) -> 'DpotrainingParamsOutputFields':
-        """Subfields should come from the DpotrainingParamsOutputFields class"""
-        self._subfields.extend(subfields)
-        return self
-
-    def alias(self, alias: str) -> 'DpotrainingParamsOutputFields':
-        self._alias = alias
-        return self
-
 class EmojiFields(GraphQLField):
     """@private"""
     native: 'EmojiGraphQLField' = EmojiGraphQLField('native')
@@ -789,22 +755,6 @@ class EvalJobStageOutputFields(GraphQLField):
         self._alias = alias
         return self
 
-class EvaluationAnswerRelevancyRecipeFields(GraphQLField):
-    """@private"""
-
-    @classmethod
-    def metric(cls) -> 'MetricFields':
-        return MetricFields('metric')
-
-    def fields(self, *subfields: Union[EvaluationAnswerRelevancyRecipeGraphQLField, 'MetricFields']) -> 'EvaluationAnswerRelevancyRecipeFields':
-        """Subfields should come from the EvaluationAnswerRelevancyRecipeFields class"""
-        self._subfields.extend(subfields)
-        return self
-
-    def alias(self, alias: str) -> 'EvaluationAnswerRelevancyRecipeFields':
-        self._alias = alias
-        return self
-
 class EvaluationByproductsFields(GraphQLField):
     """@private"""
 
@@ -821,125 +771,6 @@ class EvaluationByproductsFields(GraphQLField):
         self._alias = alias
         return self
 
-class EvaluationContextRelevancyRecipeFields(GraphQLField):
-    """@private"""
-
-    @classmethod
-    def metric(cls) -> 'MetricFields':
-        return MetricFields('metric')
-
-    def fields(self, *subfields: Union[EvaluationContextRelevancyRecipeGraphQLField, 'MetricFields']) -> 'EvaluationContextRelevancyRecipeFields':
-        """Subfields should come from the EvaluationContextRelevancyRecipeFields class"""
-        self._subfields.extend(subfields)
-        return self
-
-    def alias(self, alias: str) -> 'EvaluationContextRelevancyRecipeFields':
-        self._alias = alias
-        return self
-
-class EvaluationCustomRecipeFields(GraphQLField):
-    """@private"""
-
-    @classmethod
-    def guidelines(cls) -> 'GuidelineFields':
-        return GuidelineFields('guidelines')
-    criteria: 'EvaluationCustomRecipeGraphQLField' = EvaluationCustomRecipeGraphQLField('criteria')
-
-    @classmethod
-    def metric(cls) -> 'MetricFields':
-        return MetricFields('metric')
-
-    def fields(self, *subfields: Union[EvaluationCustomRecipeGraphQLField, 'GuidelineFields', 'MetricFields']) -> 'EvaluationCustomRecipeFields':
-        """Subfields should come from the EvaluationCustomRecipeFields class"""
-        self._subfields.extend(subfields)
-        return self
-
-    def alias(self, alias: str) -> 'EvaluationCustomRecipeFields':
-        self._alias = alias
-        return self
-
-class EvaluationFaithfulnessRecipeFields(GraphQLField):
-    """@private"""
-
-    @classmethod
-    def metric(cls) -> 'MetricFields':
-        return MetricFields('metric')
-
-    def fields(self, *subfields: Union[EvaluationFaithfulnessRecipeGraphQLField, 'MetricFields']) -> 'EvaluationFaithfulnessRecipeFields':
-        """Subfields should come from the EvaluationFaithfulnessRecipeFields class"""
-        self._subfields.extend(subfields)
-        return self
-
-    def alias(self, alias: str) -> 'EvaluationFaithfulnessRecipeFields':
-        self._alias = alias
-        return self
-
-class EvaluationJobFields(GraphQLField):
-    """@private"""
-    id: 'EvaluationJobGraphQLField' = EvaluationJobGraphQLField('id')
-    name: 'EvaluationJobGraphQLField' = EvaluationJobGraphQLField('name')
-    created_at: 'EvaluationJobGraphQLField' = EvaluationJobGraphQLField('createdAt')
-    status: 'EvaluationJobGraphQLField' = EvaluationJobGraphQLField('status')
-    started_at: 'EvaluationJobGraphQLField' = EvaluationJobGraphQLField('startedAt')
-    ended_at: 'EvaluationJobGraphQLField' = EvaluationJobGraphQLField('endedAt')
-    duration_ms: 'EvaluationJobGraphQLField' = EvaluationJobGraphQLField('durationMs')
-
-    @classmethod
-    def model_services(cls) -> 'ModelServiceFields':
-        return ModelServiceFields('model_services')
-
-    @classmethod
-    def graders(cls) -> 'GraderFields':
-        return GraderFields('graders')
-
-    @classmethod
-    def judge_models(cls) -> 'ModelFields':
-        return ModelFields('judge_models')
-
-    @classmethod
-    def judges(cls) -> 'JudgeFields':
-        return JudgeFields('judges')
-
-    @classmethod
-    def use_case(cls) -> 'UseCaseFields':
-        return UseCaseFields('use_case')
-
-    @classmethod
-    def stages(cls) -> 'JobStageOutputFields':
-        return JobStageOutputFields('stages')
-
-    @classmethod
-    def reports(cls) -> 'EvaluationResultFields':
-        return EvaluationResultFields('reports')
-
-    @classmethod
-    def created_by(cls) -> 'UserFields':
-        return UserFields('created_by')
-
-    @classmethod
-    def dataset(cls) -> 'DatasetFields':
-        return DatasetFields('dataset')
-
-    @classmethod
-    def sample_config(cls) -> 'SampleConfigOutputFields':
-        return SampleConfigOutputFields('sample_config')
-
-    @classmethod
-    def metrics(cls) -> 'MetricFields':
-        return MetricFields('metrics')
-    eval_type: 'EvaluationJobGraphQLField' = EvaluationJobGraphQLField('evalType')
-    recipe: 'EvaluationRecipeUnion' = EvaluationRecipeUnion('recipe')
-    error: 'EvaluationJobGraphQLField' = EvaluationJobGraphQLField('error')
-
-    def fields(self, *subfields: Union[EvaluationJobGraphQLField, 'DatasetFields', 'EvaluationRecipeUnion', 'EvaluationResultFields', 'GraderFields', 'JobStageOutputFields', 'JudgeFields', 'MetricFields', 'ModelFields', 'ModelServiceFields', 'SampleConfigOutputFields', 'UseCaseFields', 'UserFields']) -> 'EvaluationJobFields':
-        """Subfields should come from the EvaluationJobFields class"""
-        self._subfields.extend(subfields)
-        return self
-
-    def alias(self, alias: str) -> 'EvaluationJobFields':
-        self._alias = alias
-        return self
-
 class EvaluationResultFields(GraphQLField):
     """@private"""
 
@@ -982,6 +813,35 @@ class EvaluationResultFields(GraphQLField):
         self._alias = alias
         return self
 
+class GlobalUsageFields(GraphQLField):
+    """@private"""
+
+    @classmethod
+    def total(cls) -> 'UsageStatsFields':
+        return UsageStatsFields('total')
+
+    @classmethod
+    def adaptive_models(cls) -> 'UsageStatsFields':
+        return UsageStatsFields('adaptive_models')
+
+    @classmethod
+    def external_models(cls) -> 'UsageStatsFields':
+        return UsageStatsFields('external_models')
+
+    @classmethod
+    def by_model(cls) -> 'UsageStatsByModelFields':
+        return UsageStatsByModelFields('by_model')
+    signature: 'GlobalUsageGraphQLField' = GlobalUsageGraphQLField('signature')
+
+    def fields(self, *subfields: Union[GlobalUsageGraphQLField, 'UsageStatsByModelFields', 'UsageStatsFields']) -> 'GlobalUsageFields':
+        """Subfields should come from the GlobalUsageFields class"""
+        self._subfields.extend(subfields)
+        return self
+
+    def alias(self, alias: str) -> 'GlobalUsageFields':
+        self._alias = alias
+        return self
+
 class GpuAllocationFields(GraphQLField):
     """@private"""
     name: 'GpuAllocationGraphQLField' = GpuAllocationGraphQLField('name')
@@ -1028,48 +888,36 @@ class GraderFields(GraphQLField):
         self._alias = alias
         return self
 
-class GrpotrainingParamsOutputFields(GraphQLField):
+class HarmonyGroupFields(GraphQLField):
     """@private"""
-    kl_div_coeff: 'GrpotrainingParamsOutputGraphQLField' = GrpotrainingParamsOutputGraphQLField('klDivCoeff')
-    steps: 'GrpotrainingParamsOutputGraphQLField' = GrpotrainingParamsOutputGraphQLField('steps')
+    id: 'HarmonyGroupGraphQLField' = HarmonyGroupGraphQLField('id')
+    key: 'HarmonyGroupGraphQLField' = HarmonyGroupGraphQLField('key')
 
-    def fields(self, *subfields: GrpotrainingParamsOutputGraphQLField) -> 'GrpotrainingParamsOutputFields':
-        """Subfields should come from the GrpotrainingParamsOutputFields class"""
-        self._subfields.extend(subfields)
-        return self
-
-    def alias(self, alias: str) -> 'GrpotrainingParamsOutputFields':
-        self._alias = alias
-        return self
-
-class GuidelineFields(GraphQLField):
-    """@private"""
-    name: 'GuidelineGraphQLField' = GuidelineGraphQLField('name')
-    description: 'GuidelineGraphQLField' = GuidelineGraphQLField('description')
-
-    def fields(self, *subfields: GuidelineGraphQLField) -> 'GuidelineFields':
-        """Subfields should come from the GuidelineFields class"""
-        self._subfields.extend(subfields)
-        return self
-
-    def alias(self, alias: str) -> 'GuidelineFields':
-        self._alias = alias
-        return self
+    @classmethod
+    def compute_pool(cls) -> 'ComputePoolFields':
+        return ComputePoolFields('compute_pool')
+    status: 'HarmonyGroupGraphQLField' = HarmonyGroupGraphQLField('status')
+    url: 'HarmonyGroupGraphQLField' = HarmonyGroupGraphQLField('url')
+    world_size: 'HarmonyGroupGraphQLField' = HarmonyGroupGraphQLField('worldSize')
+    gpu_total: 'HarmonyGroupGraphQLField' = HarmonyGroupGraphQLField('gpuTotal')
+    gpu_allocated: 'HarmonyGroupGraphQLField' = HarmonyGroupGraphQLField('gpuAllocated')
 
-class GuidelinesTrainingParamsOutputFields(GraphQLField):
-    """@private"""
-    judge_model: 'GuidelinesTrainingParamsOutputGraphQLField' = GuidelinesTrainingParamsOutputGraphQLField('judgeModel')
+    @classmethod
+    def gpu_allocations(cls) -> 'GpuAllocationFields':
+        return GpuAllocationFields('gpu_allocations')
+    gpu_types: 'HarmonyGroupGraphQLField' = HarmonyGroupGraphQLField('gpuTypes')
+    created_at: 'HarmonyGroupGraphQLField' = HarmonyGroupGraphQLField('createdAt')
 
     @classmethod
-    def judge_model_prompt(cls) -> 'GuidelineFields':
-        return GuidelineFields('judge_model_prompt')
+    def online_models(cls) -> 'ModelFields':
+        return ModelFields('online_models')
 
-    def fields(self, *subfields: Union[GuidelinesTrainingParamsOutputGraphQLField, 'GuidelineFields']) -> 'GuidelinesTrainingParamsOutputFields':
-        """Subfields should come from the GuidelinesTrainingParamsOutputFields class"""
+    def fields(self, *subfields: Union[HarmonyGroupGraphQLField, 'ComputePoolFields', 'GpuAllocationFields', 'ModelFields']) -> 'HarmonyGroupFields':
+        """Subfields should come from the HarmonyGroupFields class"""
         self._subfields.extend(subfields)
         return self
 
-    def alias(self, alias: str) -> 'GuidelinesTrainingParamsOutputFields':
+    def alias(self, alias: str) -> 'HarmonyGroupFields':
         self._alias = alias
         return self
 
@@ -1306,22 +1154,6 @@ class JudgeExampleFields(GraphQLField):
         self._alias = alias
         return self
 
-class JudgeTrainingParamsOutputFields(GraphQLField):
-    """@private"""
-
-    @classmethod
-    def judges(cls) -> 'JudgeFields':
-        return JudgeFields('judges')
-
-    def fields(self, *subfields: Union[JudgeTrainingParamsOutputGraphQLField, 'JudgeFields']) -> 'JudgeTrainingParamsOutputFields':
-        """Subfields should come from the JudgeTrainingParamsOutputFields class"""
-        self._subfields.extend(subfields)
-        return self
-
-    def alias(self, alias: str) -> 'JudgeTrainingParamsOutputFields':
-        self._alias = alias
-        return self
-
 class LabelFields(GraphQLField):
     """@private"""
     key: 'LabelGraphQLField' = LabelGraphQLField('key')
@@ -1386,38 +1218,6 @@ class LabelValueUsageFields(GraphQLField):
         self._alias = alias
         return self
 
-class ListCompletionsFilterOutputFields(GraphQLField):
-    """@private"""
-    use_case: 'ListCompletionsFilterOutputGraphQLField' = ListCompletionsFilterOutputGraphQLField('useCase')
-    models: 'ListCompletionsFilterOutputGraphQLField' = ListCompletionsFilterOutputGraphQLField('models')
-
-    @classmethod
-    def timerange(cls) -> 'TimeRangeOutputFields':
-        return TimeRangeOutputFields('timerange')
-    session_id: 'ListCompletionsFilterOutputGraphQLField' = ListCompletionsFilterOutputGraphQLField('sessionId')
-    user_id: 'ListCompletionsFilterOutputGraphQLField' = ListCompletionsFilterOutputGraphQLField('userId')
-
-    @classmethod
-    def feedbacks(cls) -> 'CompletionFeedbackFilterOutputFields':
-        return CompletionFeedbackFilterOutputFields('feedbacks')
-
-    @classmethod
-    def labels(cls) -> 'CompletionLabelFilterOutputFields':
-        return CompletionLabelFilterOutputFields('labels')
-    prompt_hash: 'ListCompletionsFilterOutputGraphQLField' = ListCompletionsFilterOutputGraphQLField('promptHash')
-    completion_id: 'ListCompletionsFilterOutputGraphQLField' = ListCompletionsFilterOutputGraphQLField('completionId')
-    tags: 'ListCompletionsFilterOutputGraphQLField' = ListCompletionsFilterOutputGraphQLField('tags')
-    source: 'ListCompletionsFilterOutputGraphQLField' = ListCompletionsFilterOutputGraphQLField('source')
-
-    def fields(self, *subfields: Union[ListCompletionsFilterOutputGraphQLField, 'CompletionFeedbackFilterOutputFields', 'CompletionLabelFilterOutputFields', 'TimeRangeOutputFields']) -> 'ListCompletionsFilterOutputFields':
-        """Subfields should come from the ListCompletionsFilterOutputFields class"""
-        self._subfields.extend(subfields)
-        return self
-
-    def alias(self, alias: str) -> 'ListCompletionsFilterOutputFields':
-        self._alias = alias
-        return self
-
 class MetaObjectFields(GraphQLField):
     """@private"""
 
@@ -1443,6 +1243,7 @@ class MetricFields(GraphQLField):
     kind: 'MetricGraphQLField' = MetricGraphQLField('kind')
     description: 'MetricGraphQLField' = MetricGraphQLField('description')
     scoring_type: 'MetricGraphQLField' = MetricGraphQLField('scoringType')
+    unit: 'MetricGraphQLField' = MetricGraphQLField('unit')
 
     @classmethod
     def use_cases(cls, filter: UseCaseFilter) -> 'UseCaseFields':
@@ -1483,20 +1284,6 @@ class MetricActivityFields(GraphQLField):
         self._alias = alias
         return self
 
-class MetricTrainingParamsOutputFields(GraphQLField):
-    """@private"""
-    metric_key: 'MetricTrainingParamsOutputGraphQLField' = MetricTrainingParamsOutputGraphQLField('metricKey')
-    metric_metadata: 'MetricTrainingParamsMetadataOutputUnion' = MetricTrainingParamsMetadataOutputUnion('metricMetadata')
-
-    def fields(self, *subfields: Union[MetricTrainingParamsOutputGraphQLField, 'MetricTrainingParamsMetadataOutputUnion']) -> 'MetricTrainingParamsOutputFields':
-        """Subfields should come from the MetricTrainingParamsOutputFields class"""
-        self._subfields.extend(subfields)
-        return self
-
-    def alias(self, alias: str) -> 'MetricTrainingParamsOutputFields':
-        self._alias = alias
-        return self
-
 class MetricWithContextFields(GraphQLField):
     """@private"""
     id: 'MetricWithContextGraphQLField' = MetricWithContextGraphQLField('id')
@@ -1506,6 +1293,7 @@ class MetricWithContextFields(GraphQLField):
     scoring_type: 'MetricWithContextGraphQLField' = MetricWithContextGraphQLField('scoringType')
     description: 'MetricWithContextGraphQLField' = MetricWithContextGraphQLField('description')
     created_at: 'MetricWithContextGraphQLField' = MetricWithContextGraphQLField('createdAt')
+    unit: 'MetricWithContextGraphQLField' = MetricWithContextGraphQLField('unit')
 
     @classmethod
     def feedback_count(cls, *, timerange: Optional[TimeRange]=None) -> 'MetricWithContextGraphQLField':
@@ -1582,10 +1370,6 @@ class ModelFields(GraphQLField):
     def backbone(cls) -> 'ModelFields':
         return ModelFields('backbone')
     is_training: 'ModelGraphQLField' = ModelGraphQLField('isTraining')
-
-    @classmethod
-    def training_job(cls) -> 'TrainingJobFields':
-        return TrainingJobFields('training_job')
     kind: 'ModelGraphQLField' = ModelGraphQLField('kind')
     size: 'ModelGraphQLField' = ModelGraphQLField('size')
 
@@ -1593,7 +1377,7 @@ class ModelFields(GraphQLField):
     def compute_config(cls) -> 'ModelComputeConfigOutputFields':
         return ModelComputeConfigOutputFields('compute_config')
 
-    def fields(self, *subfields: Union[ModelGraphQLField, 'ActivityFields', 'MetricWithContextFields', 'ModelComputeConfigOutputFields', 'ModelFields', 'ModelServiceFields', 'TrainingJobFields', 'UseCaseFields']) -> 'ModelFields':
+    def fields(self, *subfields: Union[ModelGraphQLField, 'ActivityFields', 'MetricWithContextFields', 'ModelComputeConfigOutputFields', 'ModelFields', 'ModelServiceFields', 'UseCaseFields']) -> 'ModelFields':
         """Subfields should come from the ModelFields class"""
         self._subfields.extend(subfields)
         return self
@@ -1701,53 +1485,6 @@ class PageInfoFields(GraphQLField):
         self._alias = alias
         return self
 
-class PartitionFields(GraphQLField):
-    """@private"""
-    id: 'PartitionGraphQLField' = PartitionGraphQLField('id')
-    key: 'PartitionGraphQLField' = PartitionGraphQLField('key')
-
-    @classmethod
-    def compute_pool(cls) -> 'ComputePoolFields':
-        return ComputePoolFields('compute_pool')
-    status: 'PartitionGraphQLField' = PartitionGraphQLField('status')
-    url: 'PartitionGraphQLField' = PartitionGraphQLField('url')
-    world_size: 'PartitionGraphQLField' = PartitionGraphQLField('worldSize')
-    gpu_total: 'PartitionGraphQLField' = PartitionGraphQLField('gpuTotal')
-    gpu_allocated: 'PartitionGraphQLField' = PartitionGraphQLField('gpuAllocated')
-
-    @classmethod
-    def gpu_allocations(cls) -> 'GpuAllocationFields':
-        return GpuAllocationFields('gpu_allocations')
-    gpu_types: 'PartitionGraphQLField' = PartitionGraphQLField('gpuTypes')
-    created_at: 'PartitionGraphQLField' = PartitionGraphQLField('createdAt')
-
-    @classmethod
-    def online_models(cls) -> 'ModelFields':
-        return ModelFields('online_models')
-
-    def fields(self, *subfields: Union[PartitionGraphQLField, 'ComputePoolFields', 'GpuAllocationFields', 'ModelFields']) -> 'PartitionFields':
-        """Subfields should come from the PartitionFields class"""
-        self._subfields.extend(subfields)
-        return self
-
-    def alias(self, alias: str) -> 'PartitionFields':
-        self._alias = alias
-        return self
-
-class PpotrainingParamsOutputFields(GraphQLField):
-    """@private"""
-    kl_div_coeff: 'PpotrainingParamsOutputGraphQLField' = PpotrainingParamsOutputGraphQLField('klDivCoeff')
-    steps: 'PpotrainingParamsOutputGraphQLField' = PpotrainingParamsOutputGraphQLField('steps')
-
-    def fields(self, *subfields: PpotrainingParamsOutputGraphQLField) -> 'PpotrainingParamsOutputFields':
-        """Subfields should come from the PpotrainingParamsOutputFields class"""
-        self._subfields.extend(subfields)
-        return self
-
-    def alias(self, alias: str) -> 'PpotrainingParamsOutputFields':
-        self._alias = alias
-        return self
-
 class PrebuiltConfigDefinitionFields(GraphQLField):
     """@private"""
     key: 'PrebuiltConfigDefinitionGraphQLField' = PrebuiltConfigDefinitionGraphQLField('key')
@@ -1883,19 +1620,6 @@ class RemoteEnvTestOnlineFields(GraphQLField):
         self._alias = alias
         return self
 
-class RewardServerTrainingParamsOutputFields(GraphQLField):
-    """@private"""
-    remote_env_id: 'RewardServerTrainingParamsOutputGraphQLField' = RewardServerTrainingParamsOutputGraphQLField('remoteEnvId')
-
-    def fields(self, *subfields: RewardServerTrainingParamsOutputGraphQLField) -> 'RewardServerTrainingParamsOutputFields':
-        """Subfields should come from the RewardServerTrainingParamsOutputFields class"""
-        self._subfields.extend(subfields)
-        return self
-
-    def alias(self, alias: str) -> 'RewardServerTrainingParamsOutputFields':
-        self._alias = alias
-        return self
-
 class RoleFields(GraphQLField):
     """@private"""
     id: 'RoleGraphQLField' = RoleGraphQLField('id')
@@ -1913,64 +1637,6 @@ class RoleFields(GraphQLField):
         self._alias = alias
         return self
 
-class SampleConfigOutputFields(GraphQLField):
-    """@private"""
-    feedback_type: 'SampleConfigOutputGraphQLField' = SampleConfigOutputGraphQLField('feedbackType')
-    datasource: 'SampleDatasourceOutputUnion' = SampleDatasourceOutputUnion('datasource')
-
-    def fields(self, *subfields: Union[SampleConfigOutputGraphQLField, 'SampleDatasourceOutputUnion']) -> 'SampleConfigOutputFields':
-        """Subfields should come from the SampleConfigOutputFields class"""
-        self._subfields.extend(subfields)
-        return self
-
-    def alias(self, alias: str) -> 'SampleConfigOutputFields':
-        self._alias = alias
-        return self
-
-class SampleDatasourceCompletionsOutputFields(GraphQLField):
-    """@private"""
-    selection_type: 'SampleDatasourceCompletionsOutputGraphQLField' = SampleDatasourceCompletionsOutputGraphQLField('selectionType')
-    max_samples: 'SampleDatasourceCompletionsOutputGraphQLField' = SampleDatasourceCompletionsOutputGraphQLField('maxSamples')
-
-    @classmethod
-    def filter(cls) -> 'ListCompletionsFilterOutputFields':
-        return ListCompletionsFilterOutputFields('filter')
-
-    def fields(self, *subfields: Union[SampleDatasourceCompletionsOutputGraphQLField, 'ListCompletionsFilterOutputFields']) -> 'SampleDatasourceCompletionsOutputFields':
-        """Subfields should come from the SampleDatasourceCompletionsOutputFields class"""
-        self._subfields.extend(subfields)
-        return self
-
-    def alias(self, alias: str) -> 'SampleDatasourceCompletionsOutputFields':
-        self._alias = alias
-        return self
-
-class SampleDatasourceDatasetOutputFields(GraphQLField):
-    """@private"""
-    dataset_key: 'SampleDatasourceDatasetOutputGraphQLField' = SampleDatasourceDatasetOutputGraphQLField('datasetKey')
-
-    def fields(self, *subfields: SampleDatasourceDatasetOutputGraphQLField) -> 'SampleDatasourceDatasetOutputFields':
-        """Subfields should come from the SampleDatasourceDatasetOutputFields class"""
-        self._subfields.extend(subfields)
-        return self
-
-    def alias(self, alias: str) -> 'SampleDatasourceDatasetOutputFields':
-        self._alias = alias
-        return self
-
-class ScalarMetricConfigOutputFields(GraphQLField):
-    """@private"""
-    threshold: 'ScalarMetricConfigOutputGraphQLField' = ScalarMetricConfigOutputGraphQLField('threshold')
-
-    def fields(self, *subfields: ScalarMetricConfigOutputGraphQLField) -> 'ScalarMetricConfigOutputFields':
-        """Subfields should come from the ScalarMetricConfigOutputFields class"""
-        self._subfields.extend(subfields)
-        return self
-
-    def alias(self, alias: str) -> 'ScalarMetricConfigOutputFields':
-        self._alias = alias
-        return self
-
 class SessionFields(GraphQLField):
     """@private"""
     id: 'SessionGraphQLField' = SessionGraphQLField('id')
@@ -2004,19 +1670,6 @@ class SettingsFields(GraphQLField):
         self._alias = alias
         return self
 
-class SfttrainingParamsOutputFields(GraphQLField):
-    """@private"""
-    empty: 'SfttrainingParamsOutputGraphQLField' = SfttrainingParamsOutputGraphQLField('empty')
-
-    def fields(self, *subfields: SfttrainingParamsOutputGraphQLField) -> 'SfttrainingParamsOutputFields':
-        """Subfields should come from the SfttrainingParamsOutputFields class"""
-        self._subfields.extend(subfields)
-        return self
-
-    def alias(self, alias: str) -> 'SfttrainingParamsOutputFields':
-        self._alias = alias
-        return self
-
 class ShareFields(GraphQLField):
     """@private"""
 
@@ -2116,20 +1769,6 @@ class TeamWithroleFields(GraphQLField):
         self._alias = alias
         return self
 
-class TimeRangeOutputFields(GraphQLField):
-    """@private"""
-    from_: 'TimeRangeOutputGraphQLField' = TimeRangeOutputGraphQLField('from')
-    to: 'TimeRangeOutputGraphQLField' = TimeRangeOutputGraphQLField('to')
-
-    def fields(self, *subfields: TimeRangeOutputGraphQLField) -> 'TimeRangeOutputFields':
-        """Subfields should come from the TimeRangeOutputFields class"""
-        self._subfields.extend(subfields)
-        return self
-
-    def alias(self, alias: str) -> 'TimeRangeOutputFields':
-        self._alias = alias
-        return self
-
 class TimeseriesFields(GraphQLField):
     """@private"""
 
@@ -2168,72 +1807,6 @@ class ToolProviderFields(GraphQLField):
         self._alias = alias
         return self
 
-class TrainingConfigOutputFields(GraphQLField):
-    """@private"""
-
-    @classmethod
-    def base_training_params(cls) -> 'BaseTrainingParamsOutputFields':
-        return BaseTrainingParamsOutputFields('base_training_params')
-
-    @classmethod
-    def training_metadata(cls) -> 'TrainingMetadataOutputFields':
-        return TrainingMetadataOutputFields('training_metadata')
-    training_objective: 'TrainingObjectiveOutputUnion' = TrainingObjectiveOutputUnion('trainingObjective')
-
-    def fields(self, *subfields: Union[TrainingConfigOutputGraphQLField, 'BaseTrainingParamsOutputFields', 'TrainingMetadataOutputFields', 'TrainingObjectiveOutputUnion']) -> 'TrainingConfigOutputFields':
-        """Subfields should come from the TrainingConfigOutputFields class"""
-        self._subfields.extend(subfields)
-        return self
-
-    def alias(self, alias: str) -> 'TrainingConfigOutputFields':
-        self._alias = alias
-        return self
-
-class TrainingJobFields(GraphQLField):
-    """@private"""
-    id: 'TrainingJobGraphQLField' = TrainingJobGraphQLField('id')
-    name: 'TrainingJobGraphQLField' = TrainingJobGraphQLField('name')
-    status: 'TrainingJobGraphQLField' = TrainingJobGraphQLField('status')
-    created_at: 'TrainingJobGraphQLField' = TrainingJobGraphQLField('createdAt')
-    started_at: 'TrainingJobGraphQLField' = TrainingJobGraphQLField('startedAt')
-    ended_at: 'TrainingJobGraphQLField' = TrainingJobGraphQLField('endedAt')
-    duration_ms: 'TrainingJobGraphQLField' = TrainingJobGraphQLField('durationMs')
-
-    @classmethod
-    def stages(cls) -> 'JobStageOutputFields':
-        return JobStageOutputFields('stages')
-
-    @classmethod
-    def child_model(cls) -> 'ModelFields':
-        return ModelFields('child_model')
-
-    @classmethod
-    def parent_model(cls) -> 'ModelFields':
-        return ModelFields('parent_model')
-
-    @classmethod
-    def checkpoints(cls) -> 'ModelFields':
-        return ModelFields('checkpoints')
-
-    @classmethod
-    def use_case(cls) -> 'UseCaseFields':
-        return UseCaseFields('use_case')
-    config: 'AdaptRequestConfigOutputUnion' = AdaptRequestConfigOutputUnion('config')
-    error: 'TrainingJobGraphQLField' = TrainingJobGraphQLField('error')
-
-    @classmethod
-    def created_by(cls) -> 'UserFields':
-        return UserFields('created_by')
-
-    def fields(self, *subfields: Union[TrainingJobGraphQLField, 'AdaptRequestConfigOutputUnion', 'JobStageOutputFields', 'ModelFields', 'UseCaseFields', 'UserFields']) -> 'TrainingJobFields':
-        """Subfields should come from the TrainingJobFields class"""
-        self._subfields.extend(subfields)
-        return self
-
-    def alias(self, alias: str) -> 'TrainingJobFields':
-        self._alias = alias
-        return self
-
 class TrainingJobStageOutputFields(GraphQLField):
     """@private"""
     monitoring_link: 'TrainingJobStageOutputGraphQLField' = TrainingJobStageOutputGraphQLField('monitoringLink')
@@ -2250,21 +1823,6 @@ class TrainingJobStageOutputFields(GraphQLField):
         self._alias = alias
         return self
 
-class TrainingMetadataOutputFields(GraphQLField):
-    """@private"""
-    training_type: 'TrainingMetadataOutputGraphQLField' = TrainingMetadataOutputGraphQLField('trainingType')
-    alignment_method: 'TrainingMetadataOutputGraphQLField' = TrainingMetadataOutputGraphQLField('alignmentMethod')
-    parameters: 'TrainingMetadataOutputParametersUnion' = TrainingMetadataOutputParametersUnion('parameters')
-
-    def fields(self, *subfields: Union[TrainingMetadataOutputGraphQLField, 'TrainingMetadataOutputParametersUnion']) -> 'TrainingMetadataOutputFields':
-        """Subfields should come from the TrainingMetadataOutputFields class"""
-        self._subfields.extend(subfields)
-        return self
-
-    def alias(self, alias: str) -> 'TrainingMetadataOutputFields':
-        self._alias = alias
-        return self
-
 class TrendResultFields(GraphQLField):
     """@private"""
     trend: 'TrendResultGraphQLField' = TrendResultGraphQLField('trend')
@@ -2350,6 +1908,46 @@ class UsageAggregatePerUseCaseItemFields(GraphQLField):
         self._alias = alias
         return self
 
+class UsageStatsFields(GraphQLField):
+    """@private"""
+    total_tokens: 'UsageStatsGraphQLField' = UsageStatsGraphQLField('totalTokens')
+    interactions: 'UsageStatsGraphQLField' = UsageStatsGraphQLField('interactions')
+    prompt_tokens: 'UsageStatsGraphQLField' = UsageStatsGraphQLField('promptTokens')
+    completion_tokens: 'UsageStatsGraphQLField' = UsageStatsGraphQLField('completionTokens')
+
+    def fields(self, *subfields: UsageStatsGraphQLField) -> 'UsageStatsFields':
+        """Subfields should come from the UsageStatsFields class"""
+        self._subfields.extend(subfields)
+        return self
+
+    def alias(self, alias: str) -> 'UsageStatsFields':
+        self._alias = alias
+        return self
+
+class UsageStatsByModelFields(GraphQLField):
+    """@private"""
+
+    @classmethod
+    def model(cls) -> 'ModelFields':
+        return ModelFields('model')
+    total_tokens: 'UsageStatsByModelGraphQLField' = UsageStatsByModelGraphQLField('totalTokens')
+    interactions: 'UsageStatsByModelGraphQLField' = UsageStatsByModelGraphQLField('interactions')
+    prompt_tokens: 'UsageStatsByModelGraphQLField' = UsageStatsByModelGraphQLField('promptTokens')
+    completion_tokens: 'UsageStatsByModelGraphQLField' = UsageStatsByModelGraphQLField('completionTokens')
+
+    @classmethod
+    def timeseries(cls) -> 'UsageAggregateItemFields':
+        return UsageAggregateItemFields('timeseries')
+
+    def fields(self, *subfields: Union[UsageStatsByModelGraphQLField, 'ModelFields', 'UsageAggregateItemFields']) -> 'UsageStatsByModelFields':
+        """Subfields should come from the UsageStatsByModelFields class"""
+        self._subfields.extend(subfields)
+        return self
+
+    def alias(self, alias: str) -> 'UsageStatsByModelFields':
+        self._alias = alias
+        return self
+
 class UseCaseFields(GraphQLField):
     """@private"""
     id: 'UseCaseGraphQLField' = UseCaseGraphQLField('id')
@@ -2397,14 +1995,6 @@ class UseCaseFields(GraphQLField):
         cleared_arguments = {key: value for (key, value) in arguments.items() if value['value'] is not None}
         return AbcampaignFields('ab_campaigns', arguments=cleared_arguments)
 
-    @classmethod
-    def auto_evals(cls) -> 'EvaluationJobFields':
-        return EvaluationJobFields('auto_evals')
-
-    @classmethod
-    def training_jobs(cls) -> 'TrainingJobFields':
-        return TrainingJobFields('training_jobs')
-
     @classmethod
     def widgets(cls) -> 'WidgetFields':
         return WidgetFields('widgets')
@@ -2430,7 +2020,7 @@ class UseCaseFields(GraphQLField):
     def tool_providers(cls) -> 'ToolProviderFields':
         return ToolProviderFields('tool_providers')
 
-    def fields(self, *subfields: Union[UseCaseGraphQLField, 'AbcampaignFields', 'ActivityFields', 'EvaluationJobFields', 'LabelUsageFields', 'MetricWithContextFields', 'ModelServiceFields', 'SettingsFields', 'ShareFields', 'ToolProviderFields', 'TrainingJobFields', 'UseCaseMetadataFields', 'WidgetFields']) -> 'UseCaseFields':
+    def fields(self, *subfields: Union[UseCaseGraphQLField, 'AbcampaignFields', 'ActivityFields', 'LabelUsageFields', 'MetricWithContextFields', 'ModelServiceFields', 'SettingsFields', 'ShareFields', 'ToolProviderFields', 'UseCaseMetadataFields', 'WidgetFields']) -> 'UseCaseFields':
         """Subfields should come from the UseCaseFields class"""
         self._subfields.extend(subfields)
         return self