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

adaptive_sdk/graphql_client/custom_mutations.py

@@ -2,7 +2,7 @@ from typing import Any, Dict, Optional
 from .base_model import Upload
 from .custom_fields import AbcampaignFields, ApiKeyFields, CompletionFields, CustomRecipeFields, DatasetFields, DatasetValidationOutputFields, DeleteConfirmFields, DirectFeedbackFields, GraderFields, JobFields, JudgeFields, MetricFields, MetricWithContextFields, ModelFields, ModelServiceFields, RemoteEnvFields, RoleFields, SystemPromptTemplateFields, TeamFields, TeamMemberFields, ToolProviderFields, UseCaseFields, UserFields
 from .custom_typing_fields import GraphQLField, RemoteEnvTestUnion
-from .input_types import AbcampaignCreate, AddExternalModelInput, AddHFModelInput, AddModelInput, ApiKeyCreate, AttachModel, CreateRecipeInput, CreateToolProviderInput, DatasetCreate, DatasetGenerate, FeedbackAddInput, FeedbackUpdateInput, GraderCreateInput, GraderUpdateInput, JobInput, JudgeCreate, JudgeUpdate, MetricCreate, MetricLink, MetricUnlink, ModelComputeConfigInput, ModelPlacementInput, ModelServiceDisconnect, PrebuiltJudgeCreate, RemoteEnvCreate, RoleCreate, SystemPromptTemplateCreate, SystemPromptTemplateUpdate, TeamCreate, TeamMemberRemove, TeamMemberSet, UpdateCompletion, UpdateModelService, UpdateRecipeInput, UpdateToolProviderInput, UseCaseCreate, UseCaseShares, UseCaseUpdate, UserCreate
+from .input_types import AbcampaignCreate, AddExternalModelInput, AddHFModelInput, AddModelInput, ApiKeyCreate, AttachModel, CreateRecipeInput, CreateToolProviderInput, DatasetCreate, DatasetCreateFromFilters, FeedbackAddInput, FeedbackUpdateInput, GraderCreateInput, GraderUpdateInput, JobInput, JudgeCreate, JudgeUpdate, MetricCreate, MetricLink, MetricUnlink, ModelComputeConfigInput, ModelPlacementInput, ModelServiceDisconnect, PrebuiltJudgeCreate, RemoteEnvCreate, ResizePartitionInput, RoleCreate, SystemPromptTemplateCreate, SystemPromptTemplateUpdate, TeamCreate, TeamMemberRemove, TeamMemberSet, UpdateCompletion, UpdateModelService, UpdateRecipeInput, UpdateToolProviderInput, UseCaseCreate, UseCaseShares, UseCaseUpdate, UserCreate
 
 class Mutation:
     """@private"""
@@ -19,6 +19,12 @@ class Mutation:
         cleared_arguments = {key: value for (key, value) in arguments.items() if value['value'] is not None}
         return GraphQLField(field_name='cancelAbCampaign', arguments=cleared_arguments)
 
+    @classmethod
+    def resize_inference_partition(cls, input: ResizePartitionInput) -> GraphQLField:
+        arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'ResizePartitionInput!', 'value': input}}
+        cleared_arguments = {key: value for (key, value) in arguments.items() if value['value'] is not None}
+        return GraphQLField(field_name='resizeInferencePartition', arguments=cleared_arguments)
+
     @classmethod
     def create_custom_recipe(cls, use_case: str, input: CreateRecipeInput, file: Upload) -> CustomRecipeFields:
         arguments: Dict[str, Dict[str, Any]] = {'useCase': {'type': 'IdOrKey!', 'value': use_case}, 'input': {'type': 'CreateRecipeInput!', 'value': input}, 'file': {'type': 'Upload!', 'value': file}}
@@ -44,10 +50,10 @@ class Mutation:
         return DatasetFields(field_name='createDataset', arguments=cleared_arguments)
 
     @classmethod
-    def generate_dataset(cls, input: DatasetGenerate, file: Upload) -> DatasetFields:
-        arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'DatasetGenerate!', 'value': input}, 'file': {'type': 'Upload!', 'value': file}}
+    def create_dataset_from_filters(cls, input: DatasetCreateFromFilters) -> DatasetFields:
+        arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'DatasetCreateFromFilters!', 'value': input}}
         cleared_arguments = {key: value for (key, value) in arguments.items() if value['value'] is not None}
-        return DatasetFields(field_name='generateDataset', arguments=cleared_arguments)
+        return DatasetFields(field_name='createDatasetFromFilters', arguments=cleared_arguments)
 
     @classmethod
     def delete_dataset(cls, id_or_key: str, use_case: str) -> GraphQLField:
@@ -146,10 +152,10 @@ class Mutation:
         return ModelFields(field_name='addModel', arguments=cleared_arguments)
 
     @classmethod
-    def import_hf_model(cls, input: AddHFModelInput) -> GraphQLField:
+    def import_hf_model(cls, input: AddHFModelInput) -> JobFields:
         arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'AddHFModelInput!', 'value': input}}
         cleared_arguments = {key: value for (key, value) in arguments.items() if value['value'] is not None}
-        return GraphQLField(field_name='importHfModel', arguments=cleared_arguments)
+        return JobFields(field_name='importHfModel', arguments=cleared_arguments)
 
     @classmethod
     def update_model_compute_config(cls, id_or_key: str, input: ModelComputeConfigInput) -> ModelFields:

adaptive_sdk/graphql_client/custom_queries.py

@@ -1,8 +1,8 @@
 from typing import Any, Dict, Optional
 from . import CompletionGroupBy
-from .custom_fields import AbcampaignFields, CompletionConnectionFields, CompletionFields, CompletionGroupDataConnectionFields, ComputePoolFields, CustomRecipeFields, DatasetFields, GraderFields, JobArtifactFields, JobConnectionFields, JobFields, JudgeFields, MetaObjectFields, MetricFields, ModelFields, PartitionFields, PrebuiltConfigDefinitionFields, PrebuiltCriteriaFields, RemoteConfigOutputFields, RemoteEnvFields, RoleFields, SystemPromptTemplateFields, TeamFields, ToolProviderFields, UsageAggregateItemFields, UsageAggregatePerUseCaseItemFields, UseCaseFields, UserFields
+from .custom_fields import AbcampaignFields, CompletionConnectionFields, CompletionFields, CompletionGroupDataConnectionFields, ComputePoolFields, ContractFields, CustomRecipeFields, DatasetFields, GlobalUsageFields, GraderFields, HarmonyGroupFields, JobArtifactFields, JobConnectionFields, JobFields, JudgeFields, MetaObjectFields, MetricFields, ModelFields, PrebuiltConfigDefinitionFields, PrebuiltCriteriaFields, RemoteConfigOutputFields, RemoteEnvFields, RoleFields, SystemPromptTemplateFields, TeamFields, ToolProviderFields, UsageAggregateItemFields, UsageAggregatePerUseCaseItemFields, UseCaseFields, UserFields
 from .custom_typing_fields import GraphQLField
-from .input_types import AbCampaignFilter, ArtifactFilter, CursorPageInput, CustomRecipeFilterInput, FeedbackFilterInput, ListCompletionsFilterInput, ListJobsFilterInput, ModelFilter, OrderPair, UsageFilterInput, UsagePerUseCaseFilterInput, UseCaseFilter
+from .input_types import AbCampaignFilter, ArtifactFilter, CursorPageInput, CustomRecipeFilterInput, DatasetCreateFromFilters, FeedbackFilterInput, GlobalUsageFilterInput, ListCompletionsFilterInput, ListJobsFilterInput, ModelFilter, OrderPair, UsageFilterInput, UsagePerUseCaseFilterInput, UseCaseFilter
 
 class Query:
     """@private"""
@@ -19,6 +19,10 @@ class Query:
         cleared_arguments = {key: value for (key, value) in arguments.items() if value['value'] is not None}
         return AbcampaignFields(field_name='abCampaign', arguments=cleared_arguments)
 
+    @classmethod
+    def contract(cls) -> ContractFields:
+        return ContractFields(field_name='contract')
+
     @classmethod
     def custom_recipes(cls, use_case: str, filter: CustomRecipeFilterInput) -> CustomRecipeFields:
         arguments: Dict[str, Dict[str, Any]] = {'useCase': {'type': 'IdOrKey!', 'value': use_case}, 'filter': {'type': 'CustomRecipeFilterInput!', 'value': filter}}
@@ -43,6 +47,12 @@ class Query:
         cleared_arguments = {key: value for (key, value) in arguments.items() if value['value'] is not None}
         return DatasetFields(field_name='dataset', arguments=cleared_arguments)
 
+    @classmethod
+    def preview_dataset_from_filters(cls, input: DatasetCreateFromFilters) -> GraphQLField:
+        arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'DatasetCreateFromFilters!', 'value': input}}
+        cleared_arguments = {key: value for (key, value) in arguments.items() if value['value'] is not None}
+        return GraphQLField(field_name='previewDatasetFromFilters', arguments=cleared_arguments)
+
     @classmethod
     def completions(cls, filter: ListCompletionsFilterInput, page: CursorPageInput, order: OrderPair) -> CompletionConnectionFields:
         arguments: Dict[str, Dict[str, Any]] = {'filter': {'type': 'ListCompletionsFilterInput!', 'value': filter}, 'page': {'type': 'CursorPageInput!', 'value': page}, 'order': {'type': 'OrderPair!', 'value': order}}
@@ -79,6 +89,12 @@ class Query:
         cleared_arguments = {key: value for (key, value) in arguments.items() if value['value'] is not None}
         return UsageAggregatePerUseCaseItemFields(field_name='modelUsageByUseCase', arguments=cleared_arguments)
 
+    @classmethod
+    def global_usage(cls, filter: GlobalUsageFilterInput) -> GlobalUsageFields:
+        arguments: Dict[str, Dict[str, Any]] = {'filter': {'type': 'GlobalUsageFilterInput!', 'value': filter}}
+        cleared_arguments = {key: value for (key, value) in arguments.items() if value['value'] is not None}
+        return GlobalUsageFields(field_name='globalUsage', arguments=cleared_arguments)
+
     @classmethod
     def system_prompt_templates(cls) -> SystemPromptTemplateFields:
         return SystemPromptTemplateFields(field_name='systemPromptTemplates')
@@ -118,12 +134,12 @@ class Query:
         return ModelFields(field_name='model', arguments=cleared_arguments)
 
     @classmethod
-    def all_partitions(cls) -> PartitionFields:
-        return PartitionFields(field_name='allPartitions')
+    def all_harmony_groups(cls) -> HarmonyGroupFields:
+        return HarmonyGroupFields(field_name='allHarmonyGroups')
 
     @classmethod
-    def partitions(cls) -> PartitionFields:
-        return PartitionFields(field_name='partitions')
+    def harmony_groups(cls) -> HarmonyGroupFields:
+        return HarmonyGroupFields(field_name='harmonyGroups')
 
     @classmethod
     def compute_pools(cls) -> ComputePoolFields:

adaptive_sdk/graphql_client/custom_typing_fields.py

@@ -42,31 +42,6 @@ class ActivityOutputGraphQLField(GraphQLField):
         self._alias = alias
         return self
 
-class AdaptBuiltinRecipeConfigOutputGraphQLField(GraphQLField):
-    """@private"""
-
-    def alias(self, alias: str) -> 'AdaptBuiltinRecipeConfigOutputGraphQLField':
-        self._alias = alias
-        return self
-
-class AdaptCustomRecipeConfigOutputGraphQLField(GraphQLField):
-    """@private"""
-
-    def alias(self, alias: str) -> 'AdaptCustomRecipeConfigOutputGraphQLField':
-        self._alias = alias
-        return self
-
-class AdaptRequestConfigOutputUnion(GraphQLField):
-    """@private"""
-
-    def on(self, type_name: str, *subfields: GraphQLField) -> 'AdaptRequestConfigOutputUnion':
-        self._inline_fragments[type_name] = subfields
-        return self
-
-    def alias(self, alias: str) -> 'AdaptRequestConfigOutputUnion':
-        self._alias = alias
-        return self
-
 class ApiKeyGraphQLField(GraphQLField):
     """@private"""
 
@@ -92,17 +67,17 @@ class AuthProviderGraphQLField(GraphQLField):
         self._alias = alias
         return self
 
-class BaseTrainingParamsOutputGraphQLField(GraphQLField):
+class BatchInferenceJobStageOutputGraphQLField(GraphQLField):
     """@private"""
 
-    def alias(self, alias: str) -> 'BaseTrainingParamsOutputGraphQLField':
+    def alias(self, alias: str) -> 'BatchInferenceJobStageOutputGraphQLField':
         self._alias = alias
         return self
 
-class BatchInferenceJobStageOutputGraphQLField(GraphQLField):
+class BillingUsageGraphQLField(GraphQLField):
     """@private"""
 
-    def alias(self, alias: str) -> 'BatchInferenceJobStageOutputGraphQLField':
+    def alias(self, alias: str) -> 'BillingUsageGraphQLField':
         self._alias = alias
         return self
 
@@ -141,13 +116,6 @@ class CompletionEdgeGraphQLField(GraphQLField):
         self._alias = alias
         return self
 
-class CompletionFeedbackFilterOutputGraphQLField(GraphQLField):
-    """@private"""
-
-    def alias(self, alias: str) -> 'CompletionFeedbackFilterOutputGraphQLField':
-        self._alias = alias
-        return self
-
 class CompletionGroupDataGraphQLField(GraphQLField):
     """@private"""
 
@@ -190,24 +158,24 @@ class CompletionLabelGraphQLField(GraphQLField):
         self._alias = alias
         return self
 
-class CompletionLabelFilterOutputGraphQLField(GraphQLField):
+class CompletionMetadataGraphQLField(GraphQLField):
     """@private"""
 
-    def alias(self, alias: str) -> 'CompletionLabelFilterOutputGraphQLField':
+    def alias(self, alias: str) -> 'CompletionMetadataGraphQLField':
         self._alias = alias
         return self
 
-class CompletionMetadataGraphQLField(GraphQLField):
+class ComputePoolGraphQLField(GraphQLField):
     """@private"""
 
-    def alias(self, alias: str) -> 'CompletionMetadataGraphQLField':
+    def alias(self, alias: str) -> 'ComputePoolGraphQLField':
         self._alias = alias
         return self
 
-class ComputePoolGraphQLField(GraphQLField):
+class ContractGraphQLField(GraphQLField):
     """@private"""
 
-    def alias(self, alias: str) -> 'ComputePoolGraphQLField':
+    def alias(self, alias: str) -> 'ContractGraphQLField':
         self._alias = alias
         return self
 
@@ -239,6 +207,13 @@ class DatasetGraphQLField(GraphQLField):
         self._alias = alias
         return self
 
+class DatasetByproductsGraphQLField(GraphQLField):
+    """@private"""
+
+    def alias(self, alias: str) -> 'DatasetByproductsGraphQLField':
+        self._alias = alias
+        return self
+
 class DatasetMetricUsageGraphQLField(GraphQLField):
     """@private"""
 
@@ -267,13 +242,6 @@ class DirectFeedbackGraphQLField(GraphQLField):
         self._alias = alias
         return self
 
-class DpotrainingParamsOutputGraphQLField(GraphQLField):
-    """@private"""
-
-    def alias(self, alias: str) -> 'DpotrainingParamsOutputGraphQLField':
-        self._alias = alias
-        return self
-
 class EmojiGraphQLField(GraphQLField):
     """@private"""
 
@@ -288,13 +256,6 @@ class EvalJobStageOutputGraphQLField(GraphQLField):
         self._alias = alias
         return self
 
-class EvaluationAnswerRelevancyRecipeGraphQLField(GraphQLField):
-    """@private"""
-
-    def alias(self, alias: str) -> 'EvaluationAnswerRelevancyRecipeGraphQLField':
-        self._alias = alias
-        return self
-
 class EvaluationByproductsGraphQLField(GraphQLField):
     """@private"""
 
@@ -302,49 +263,17 @@ class EvaluationByproductsGraphQLField(GraphQLField):
         self._alias = alias
         return self
 
-class EvaluationContextRelevancyRecipeGraphQLField(GraphQLField):
-    """@private"""
-
-    def alias(self, alias: str) -> 'EvaluationContextRelevancyRecipeGraphQLField':
-        self._alias = alias
-        return self
-
-class EvaluationCustomRecipeGraphQLField(GraphQLField):
-    """@private"""
-
-    def alias(self, alias: str) -> 'EvaluationCustomRecipeGraphQLField':
-        self._alias = alias
-        return self
-
-class EvaluationFaithfulnessRecipeGraphQLField(GraphQLField):
-    """@private"""
-
-    def alias(self, alias: str) -> 'EvaluationFaithfulnessRecipeGraphQLField':
-        self._alias = alias
-        return self
-
-class EvaluationJobGraphQLField(GraphQLField):
-    """@private"""
-
-    def alias(self, alias: str) -> 'EvaluationJobGraphQLField':
-        self._alias = alias
-        return self
-
-class EvaluationRecipeUnion(GraphQLField):
+class EvaluationResultGraphQLField(GraphQLField):
     """@private"""
 
-    def on(self, type_name: str, *subfields: GraphQLField) -> 'EvaluationRecipeUnion':
-        self._inline_fragments[type_name] = subfields
-        return self
-
-    def alias(self, alias: str) -> 'EvaluationRecipeUnion':
+    def alias(self, alias: str) -> 'EvaluationResultGraphQLField':
         self._alias = alias
         return self
 
-class EvaluationResultGraphQLField(GraphQLField):
+class GlobalUsageGraphQLField(GraphQLField):
     """@private"""
 
-    def alias(self, alias: str) -> 'EvaluationResultGraphQLField':
+    def alias(self, alias: str) -> 'GlobalUsageGraphQLField':
         self._alias = alias
         return self
 
@@ -373,24 +302,10 @@ class GraderConfigUnion(GraphQLField):
         self._alias = alias
         return self
 
-class GrpotrainingParamsOutputGraphQLField(GraphQLField):
+class HarmonyGroupGraphQLField(GraphQLField):
     """@private"""
 
-    def alias(self, alias: str) -> 'GrpotrainingParamsOutputGraphQLField':
-        self._alias = alias
-        return self
-
-class GuidelineGraphQLField(GraphQLField):
-    """@private"""
-
-    def alias(self, alias: str) -> 'GuidelineGraphQLField':
-        self._alias = alias
-        return self
-
-class GuidelinesTrainingParamsOutputGraphQLField(GraphQLField):
-    """@private"""
-
-    def alias(self, alias: str) -> 'GuidelinesTrainingParamsOutputGraphQLField':
+    def alias(self, alias: str) -> 'HarmonyGroupGraphQLField':
         self._alias = alias
         return self
 
@@ -475,13 +390,6 @@ class JudgeExampleGraphQLField(GraphQLField):
         self._alias = alias
         return self
 
-class JudgeTrainingParamsOutputGraphQLField(GraphQLField):
-    """@private"""
-
-    def alias(self, alias: str) -> 'JudgeTrainingParamsOutputGraphQLField':
-        self._alias = alias
-        return self
-
 class LabelGraphQLField(GraphQLField):
     """@private"""
 
@@ -510,13 +418,6 @@ class LabelValueUsageGraphQLField(GraphQLField):
         self._alias = alias
         return self
 
-class ListCompletionsFilterOutputGraphQLField(GraphQLField):
-    """@private"""
-
-    def alias(self, alias: str) -> 'ListCompletionsFilterOutputGraphQLField':
-        self._alias = alias
-        return self
-
 class MetaObjectGraphQLField(GraphQLField):
     """@private"""
 
@@ -538,24 +439,6 @@ class MetricActivityGraphQLField(GraphQLField):
         self._alias = alias
         return self
 
-class MetricTrainingParamsMetadataOutputUnion(GraphQLField):
-    """@private"""
-
-    def on(self, type_name: str, *subfields: GraphQLField) -> 'MetricTrainingParamsMetadataOutputUnion':
-        self._inline_fragments[type_name] = subfields
-        return self
-
-    def alias(self, alias: str) -> 'MetricTrainingParamsMetadataOutputUnion':
-        self._alias = alias
-        return self
-
-class MetricTrainingParamsOutputGraphQLField(GraphQLField):
-    """@private"""
-
-    def alias(self, alias: str) -> 'MetricTrainingParamsOutputGraphQLField':
-        self._alias = alias
-        return self
-
 class MetricWithContextGraphQLField(GraphQLField):
     """@private"""
 
@@ -605,20 +488,6 @@ class PageInfoGraphQLField(GraphQLField):
         self._alias = alias
         return self
 
-class PartitionGraphQLField(GraphQLField):
-    """@private"""
-
-    def alias(self, alias: str) -> 'PartitionGraphQLField':
-        self._alias = alias
-        return self
-
-class PpotrainingParamsOutputGraphQLField(GraphQLField):
-    """@private"""
-
-    def alias(self, alias: str) -> 'PpotrainingParamsOutputGraphQLField':
-        self._alias = alias
-        return self
-
 class PrebuiltConfigDefinitionGraphQLField(GraphQLField):
     """@private"""
 
@@ -693,13 +562,6 @@ class RemoteEnvTestOnlineGraphQLField(GraphQLField):
         self._alias = alias
         return self
 
-class RewardServerTrainingParamsOutputGraphQLField(GraphQLField):
-    """@private"""
-
-    def alias(self, alias: str) -> 'RewardServerTrainingParamsOutputGraphQLField':
-        self._alias = alias
-        return self
-
 class RoleGraphQLField(GraphQLField):
     """@private"""
 
@@ -707,45 +569,6 @@ class RoleGraphQLField(GraphQLField):
         self._alias = alias
         return self
 
-class SampleConfigOutputGraphQLField(GraphQLField):
-    """@private"""
-
-    def alias(self, alias: str) -> 'SampleConfigOutputGraphQLField':
-        self._alias = alias
-        return self
-
-class SampleDatasourceCompletionsOutputGraphQLField(GraphQLField):
-    """@private"""
-
-    def alias(self, alias: str) -> 'SampleDatasourceCompletionsOutputGraphQLField':
-        self._alias = alias
-        return self
-
-class SampleDatasourceDatasetOutputGraphQLField(GraphQLField):
-    """@private"""
-
-    def alias(self, alias: str) -> 'SampleDatasourceDatasetOutputGraphQLField':
-        self._alias = alias
-        return self
-
-class SampleDatasourceOutputUnion(GraphQLField):
-    """@private"""
-
-    def on(self, type_name: str, *subfields: GraphQLField) -> 'SampleDatasourceOutputUnion':
-        self._inline_fragments[type_name] = subfields
-        return self
-
-    def alias(self, alias: str) -> 'SampleDatasourceOutputUnion':
-        self._alias = alias
-        return self
-
-class ScalarMetricConfigOutputGraphQLField(GraphQLField):
-    """@private"""
-
-    def alias(self, alias: str) -> 'ScalarMetricConfigOutputGraphQLField':
-        self._alias = alias
-        return self
-
 class SessionGraphQLField(GraphQLField):
     """@private"""
 
@@ -760,13 +583,6 @@ class SettingsGraphQLField(GraphQLField):
         self._alias = alias
         return self
 
-class SfttrainingParamsOutputGraphQLField(GraphQLField):
-    """@private"""
-
-    def alias(self, alias: str) -> 'SfttrainingParamsOutputGraphQLField':
-        self._alias = alias
-        return self
-
 class ShareGraphQLField(GraphQLField):
     """@private"""
 
@@ -802,13 +618,6 @@ class TeamWithroleGraphQLField(GraphQLField):
         self._alias = alias
         return self
 
-class TimeRangeOutputGraphQLField(GraphQLField):
-    """@private"""
-
-    def alias(self, alias: str) -> 'TimeRangeOutputGraphQLField':
-        self._alias = alias
-        return self
-
 class TimeseriesGraphQLField(GraphQLField):
     """@private"""
 
@@ -823,20 +632,6 @@ class ToolProviderGraphQLField(GraphQLField):
         self._alias = alias
         return self
 
-class TrainingConfigOutputGraphQLField(GraphQLField):
-    """@private"""
-
-    def alias(self, alias: str) -> 'TrainingConfigOutputGraphQLField':
-        self._alias = alias
-        return self
-
-class TrainingJobGraphQLField(GraphQLField):
-    """@private"""
-
-    def alias(self, alias: str) -> 'TrainingJobGraphQLField':
-        self._alias = alias
-        return self
-
 class TrainingJobStageOutputGraphQLField(GraphQLField):
     """@private"""
 
@@ -844,35 +639,6 @@ class TrainingJobStageOutputGraphQLField(GraphQLField):
         self._alias = alias
         return self
 
-class TrainingMetadataOutputGraphQLField(GraphQLField):
-    """@private"""
-
-    def alias(self, alias: str) -> 'TrainingMetadataOutputGraphQLField':
-        self._alias = alias
-        return self
-
-class TrainingMetadataOutputParametersUnion(GraphQLField):
-    """@private"""
-
-    def on(self, type_name: str, *subfields: GraphQLField) -> 'TrainingMetadataOutputParametersUnion':
-        self._inline_fragments[type_name] = subfields
-        return self
-
-    def alias(self, alias: str) -> 'TrainingMetadataOutputParametersUnion':
-        self._alias = alias
-        return self
-
-class TrainingObjectiveOutputUnion(GraphQLField):
-    """@private"""
-
-    def on(self, type_name: str, *subfields: GraphQLField) -> 'TrainingObjectiveOutputUnion':
-        self._inline_fragments[type_name] = subfields
-        return self
-
-    def alias(self, alias: str) -> 'TrainingObjectiveOutputUnion':
-        self._alias = alias
-        return self
-
 class TrendResultGraphQLField(GraphQLField):
     """@private"""
 
@@ -908,6 +674,20 @@ class UsageAggregatePerUseCaseItemGraphQLField(GraphQLField):
         self._alias = alias
         return self
 
+class UsageStatsGraphQLField(GraphQLField):
+    """@private"""
+
+    def alias(self, alias: str) -> 'UsageStatsGraphQLField':
+        self._alias = alias
+        return self
+
+class UsageStatsByModelGraphQLField(GraphQLField):
+    """@private"""
+
+    def alias(self, alias: str) -> 'UsageStatsByModelGraphQLField':
+        self._alias = alias
+        return self
+
 class UseCaseGraphQLField(GraphQLField):
     """@private"""
 

adaptive_sdk/graphql_client/describe_job.py

@@ -0,0 +1,12 @@
+from typing import Optional
+from .base_model import BaseModel
+from .fragments import JobData
+
+class DescribeJob(BaseModel):
+    """@public"""
+    job: Optional['DescribeJobJob']
+
+class DescribeJobJob(JobData):
+    """@public"""
+    pass
+DescribeJob.model_rebuild()