adaptive-sdk 0.1.1__py3-none-any.whl → 0.1.3__py3-none-any.whl

adaptive_sdk/client.py

@@ -25,21 +25,18 @@ class Adaptive(BaseSyncClient, UseCaseClient):
 
         self.ab_tests: resources.ABTests = resources.ABTests(self)
         self.chat: resources.Chat = resources.Chat(self)
-        self.completions: resources.Completions = resources.Completions(self)
         self.compute_pools: resources.ComputePools = resources.ComputePools(self)
-        self.custom_recipes: resources.CustomRecipes = resources.CustomRecipes(self)
+        self.recipes: resources.Recipes = resources.Recipes(self)
         self.datasets: resources.Datasets = resources.Datasets(self)
         self.embeddings: resources.Embeddings = resources.Embeddings(self)
-        self.evaluation: resources.Evaluation = resources.Evaluation(self)
-        self.judges: resources.Judges = resources.Judges(self)
+        self.graders: resources.Graders = resources.Graders(self)
+        self.jobs: resources.Jobs = resources.Jobs(self)
         self.feedback: resources.Feedback = resources.Feedback(self)
         self.interactions: resources.Interactions = resources.Interactions(self)
         self.models: resources.Models = resources.Models(self)
         self.permissions: resources.Permissions = resources.Permissions(self)
-        self.reward_servers: resources.RewardServers = resources.RewardServers(self)
         self.roles: resources.Roles = resources.Roles(self)
         self.teams: resources.Teams = resources.Teams(self)
-        self.training: resources.Training = resources.Training(self)
         self.use_cases: resources.UseCase = resources.UseCase(self)
         self.users: resources.Users = resources.Users(self)
 
@@ -81,21 +78,18 @@ class AsyncAdaptive(BaseAsyncClient, UseCaseClient):
 
         self.ab_tests: resources.AsyncABTests = resources.AsyncABTests(self)
         self.chat: resources.AsyncChat = resources.AsyncChat(self)
-        self.completions: resources.AsyncCompletions = resources.AsyncCompletions(self)
         self.compute_pools: resources.AsyncComputePools = resources.AsyncComputePools(self)
-        self.custom_recipes: resources.AsyncCustomRecipes = resources.AsyncCustomRecipes(self)
+        self.recipes: resources.AsyncRecipes = resources.AsyncRecipes(self)
         self.datasets: resources.AsyncDatasets = resources.AsyncDatasets(self)
         self.embeddings: resources.AsyncEmbeddings = resources.AsyncEmbeddings(self)
-        self.evaluation: resources.AsyncEvaluation = resources.AsyncEvaluation(self)
-        self.judges: resources.AsyncJudges = resources.AsyncJudges(self)
+        self.graders: resources.AsyncGraders = resources.AsyncGraders(self)
+        self.jobs: resources.AsyncJobs = resources.AsyncJobs(self)
         self.feedback: resources.AsyncFeedback = resources.AsyncFeedback(self)
         self.interactions: resources.AsyncInteractions = resources.AsyncInteractions(self)
         self.models: resources.AsyncModels = resources.AsyncModels(self)
         self.permissions: resources.AsyncPermissions = resources.AsyncPermissions(self)
-        self.reward_servers: resources.AsyncRewardServers = resources.AsyncRewardServers(self)
         self.roles: resources.AsyncRoles = resources.AsyncRoles(self)
         self.teams: resources.AsyncTeams = resources.AsyncTeams(self)
-        self.training: resources.AsyncTraining = resources.AsyncTraining(self)
         self.use_cases: resources.AsyncUseCase = resources.AsyncUseCase(self)
         self.users: resources.AsyncUsers = resources.AsyncUsers(self)
 

adaptive_sdk/external/reward_server.py

@@ -211,4 +211,4 @@ class MyNoMetadataRewardServer(RewardServer[EmptyMetadata]):
 
 
 if __name__ == "__main__":
-    server = MyNoMetadataRewardServer(port=50056)
+    server = MyRewardServer(port=50056)

adaptive_sdk/graphql_client/__init__.py

@@ -1,74 +1,81 @@
 from .async_base_client_open_telemetry import AsyncBaseClientOpenTelemetry
 from .async_client import AsyncGQLClient
 from .add_external_model import AddExternalModel, AddExternalModelAddExternalModel
-from .add_hf_model import AddHFModel
+from .add_hf_model import AddHFModel, AddHFModelImportHfModel
 from .add_model import AddModel, AddModelAddModel, AddModelAddModelBackbone
 from .add_remote_env import AddRemoteEnv, AddRemoteEnvAddRemoteEnv
 from .attach_model_to_use_case import AttachModelToUseCase, AttachModelToUseCaseAttachModel
 from .base_client_open_telemetry import BaseClientOpenTelemetry
 from .base_model import BaseModel, Upload
 from .cancel_ab_campaign import CancelABCampaign
-from .cancel_evaluation_job import CancelEvaluationJob
-from .cancel_training_job import CancelTrainingJob
+from .cancel_job import CancelJob, CancelJobCancelJob
 from .client import GQLClient
 from .create_ab_campaign import CreateAbCampaign, CreateAbCampaignCreateAbCampaign
-from .create_custom_recipe_training_job import CreateCustomRecipeTrainingJob, CreateCustomRecipeTrainingJobCreateCustomRecipeTrainingJob
-from .create_custom_script import CreateCustomScript, CreateCustomScriptCreateCustomScript
-from .create_evaluation_job import CreateEvaluationJob, CreateEvaluationJobCreateEvaluationJob
+from .create_custom_recipe import CreateCustomRecipe, CreateCustomRecipeCreateCustomRecipe
+from .create_grader import CreateGrader, CreateGraderCreateGrader
+from .create_job import CreateJob, CreateJobCreateJob
 from .create_judge import CreateJudge, CreateJudgeCreateJudge
 from .create_metric import CreateMetric, CreateMetricCreateMetric
 from .create_prebuilt_judge import CreatePrebuiltJudge, CreatePrebuiltJudgeCreatePrebuiltJudge
 from .create_role import CreateRole, CreateRoleCreateRole
 from .create_team import CreateTeam, CreateTeamCreateTeam
-from .create_training_job import CreateTrainingJob, CreateTrainingJobCreateTrainingJob
 from .create_use_case import CreateUseCase, CreateUseCaseCreateUseCase
+from .create_user import CreateUser, CreateUserCreateUser
+from .delete_custom_recipe import DeleteCustomRecipe
+from .delete_dataset import DeleteDataset
+from .delete_grader import DeleteGrader, DeleteGraderDeleteGrader
 from .delete_judge import DeleteJudge, DeleteJudgeDeleteJudge
+from .delete_user import DeleteUser, DeleteUserDeleteUser
 from .deploy_model import DeployModel
 from .describe_ab_campaign import DescribeAbCampaign, DescribeAbCampaignAbCampaign, DescribeAbCampaignAbCampaignReport
 from .describe_dataset import DescribeDataset, DescribeDatasetDataset
-from .describe_evaluation_job import DescribeEvaluationJob, DescribeEvaluationJobEvaluationJob
 from .describe_interaction import DescribeInteraction, DescribeInteractionCompletion
+from .describe_job import DescribeJob, DescribeJobJob
 from .describe_metric import DescribeMetric, DescribeMetricMetric
 from .describe_metric_admin import DescribeMetricAdmin, DescribeMetricAdminMetric
 from .describe_model import DescribeModel, DescribeModelModel, DescribeModelModelBackbone
 from .describe_model_admin import DescribeModelAdmin, DescribeModelAdminModel, DescribeModelAdminModelBackbone
-from .describe_training_job import DescribeTrainingJob, DescribeTrainingJobTrainingJob
 from .describe_use_case import DescribeUseCase, DescribeUseCaseUseCase
-from .enums import AbcampaignStatus, AuthProviderKind, CompletionGroupBy, CompletionSource, CompletionSourceOutput, ComputePoolCapability, DatasetKind, DatasetSource, DateBucketUnit, EvaluationJobStatus, EvaluationType, ExternalModelProviderName, FeedbackType, FeedbackTypeInput, FeedbackTypeOutput, GraderTypeEnum, JobKind, JobStatus, JobStatusOutput, JudgeCapability, MetricAggregation, MetricKind, MetricScoringType, ModelKindFilter, ModelOnline, ModelserviceStatus, OpenAIModel, PartitionStatus, PrebuiltCriteriaKey, ProviderName, RemoteEnvStatus, ScriptKind, SelectionTypeInput, SelectionTypeOutput, SortDirection, TimeseriesInterval, TrainingJobStatus, TrainingMetadataInputAlignmentMethod, TrainingMetadataInputTrainingType, TrainingMetadataOutputAlignmentMethod, TrainingMetadataOutputTrainingType, UnitPosition
+from .enums import AbcampaignStatus, AuthProviderKind, BillingCycle, CompletionGroupBy, CompletionSource, CompletionSourceOutput, ComputePoolCapability, DatasetKind, DatasetSource, DateBucketUnit, EvaluationJobStatus, EvaluationType, ExternalModelProviderName, FeedbackType, FeedbackTypeOutput, GraderTypeEnum, HarmonyStatus, JobArtifactKind, JobKind, JobStatus, JobStatusOutput, JudgeCapability, MetricAggregation, MetricKind, MetricScoringType, ModelKindFilter, ModelOnline, ModelserviceStatus, OpenAIModel, PrebuiltCriteriaKey, Protocol, ProviderName, RemoteEnvStatus, SelectionTypeOutput, SortDirection, TimeseriesInterval, TrainingJobStatus, TrainingMetadataOutputAlignmentMethod, TrainingMetadataOutputTrainingType, UnitPosition
 from .exceptions import GraphQLClientError, GraphQLClientGraphQLError, GraphQLClientGraphQLMultiError, GraphQLClientHttpError, GraphQLClientInvalidResponseError
-from .fragments import AbCampaignCreateData, AbCampaignDetailData, AbCampaignDetailDataMetric, AbCampaignDetailDataModels, AbCampaignDetailDataUseCase, AbCampaignReportData, AbCampaignReportDataVariants, AbCampaignReportDataVariantsComparisons, AbCampaignReportDataVariantsComparisonsVariant, AbCampaignReportDataVariantsInterval, AbCampaignReportDataVariantsVariant, CompletionComparisonFeedbackData, CompletionComparisonFeedbackDataModel, CompletionData, CompletionDataChatMessages, CompletionDataComparisonFeedbacks, CompletionDataComparisonFeedbacksMetric, CompletionDataComparisonFeedbacksOtherCompletion, CompletionDataComparisonFeedbacksPreferedCompletion, CompletionDataComparisonFeedbacksUsecase, CompletionDataDirectFeedbacks, CompletionDataDirectFeedbacksMetric, CompletionDataLabels, CompletionDataMetadata, CompletionDataMetadataUsage, CompletionDataModel, CustomScriptData, DatasetData, DatasetDataMetricsUsage, DatasetDataMetricsUsageMetric, EvaluationJobData, EvaluationJobDataCreatedBy, EvaluationJobDataDataset, EvaluationJobDataJudgeModels, EvaluationJobDataJudges, EvaluationJobDataModelServices, EvaluationJobDataReports, EvaluationJobDataStages, EvaluationJobDataUseCase, EvaluationReportData, EvaluationReportDataDataset, EvaluationReportDataMetric, EvaluationReportDataModelService, JobStageOutputData, JobStageOutputDataInfoBatchInferenceJobStageOutput, JobStageOutputDataInfoEvalJobStageOutput, JobStageOutputDataInfoTrainingJobStageOutput, JudgeData, JudgeDataExamples, JudgeDataExamplesInput, JudgeDataMetric, JudgeDataModel, ListCompletionsFilterOutputData, ListCompletionsFilterOutputDataFeedbacks, ListCompletionsFilterOutputDataLabels, ListCompletionsFilterOutputDataTimerange, MetricData, MetricDataAdmin, MetricDataAdminUseCases, MetricWithContextData, ModelData, ModelDataAdmin, ModelDataAdminUseCases, ModelDataComputeConfig, ModelServiceData, ModelServiceDataModel, ModelServiceDataModelBackbone, PartitionData, PartitionDataComputePool, PartitionDataOnlineModels, RemoteEnvData, TrainingConfigOutputData, TrainingConfigOutputDataBaseTrainingParams, TrainingConfigOutputDataTrainingMetadata, TrainingConfigOutputDataTrainingMetadataParametersDpotrainingParamsOutput, TrainingConfigOutputDataTrainingMetadataParametersGrpotrainingParamsOutput, TrainingConfigOutputDataTrainingMetadataParametersPpotrainingParamsOutput, TrainingConfigOutputDataTrainingObjectiveGuidelinesTrainingParamsOutput, TrainingConfigOutputDataTrainingObjectiveGuidelinesTrainingParamsOutputJudgeModelPrompt, TrainingConfigOutputDataTrainingObjectiveJudgeTrainingParamsOutput, TrainingConfigOutputDataTrainingObjectiveMetricTrainingParamsOutput, TrainingConfigOutputDataTrainingObjectiveMetricTrainingParamsOutputMetricMetadataScalarMetricConfigOutput, TrainingConfigOutputDataTrainingObjectiveRewardServerTrainingParamsOutput, TrainingConfigOutputDataTrainingObjectiveSfttrainingParamsOutput, TrainingJobData, TrainingJobDataChildModel, TrainingJobDataChildModelBackbone, TrainingJobDataConfigAdaptBuiltinRecipeConfigOutput, TrainingJobDataConfigAdaptBuiltinRecipeConfigOutputSampleConfig, TrainingJobDataConfigAdaptBuiltinRecipeConfigOutputSampleConfigDatasourceSampleDatasourceCompletionsOutput, TrainingJobDataConfigAdaptBuiltinRecipeConfigOutputSampleConfigDatasourceSampleDatasourceCompletionsOutputFilter, TrainingJobDataConfigAdaptBuiltinRecipeConfigOutputSampleConfigDatasourceSampleDatasourceDatasetOutput, TrainingJobDataConfigAdaptBuiltinRecipeConfigOutputTrainingConfig, TrainingJobDataConfigAdaptCustomRecipeConfigOutput, TrainingJobDataCreatedBy, TrainingJobDataParentModel, TrainingJobDataParentModelBackbone, TrainingJobDataStages, TrainingJobDataUseCase, UseCaseData, UseCaseDataMetrics, UseCaseDataModelServices, UserData, UserDataTeams, UserDataTeamsRole, UserDataTeamsTeam
-from .generate_dataset import GenerateDataset, GenerateDatasetGenerateDataset
+from .fragments import AbCampaignCreateData, AbCampaignDetailData, AbCampaignDetailDataMetric, AbCampaignDetailDataModels, AbCampaignDetailDataUseCase, AbCampaignReportData, AbCampaignReportDataVariants, AbCampaignReportDataVariantsComparisons, AbCampaignReportDataVariantsComparisonsVariant, AbCampaignReportDataVariantsInterval, AbCampaignReportDataVariantsVariant, CompletionComparisonFeedbackData, CompletionComparisonFeedbackDataModel, CompletionData, CompletionDataChatMessages, CompletionDataComparisonFeedbacks, CompletionDataComparisonFeedbacksMetric, CompletionDataComparisonFeedbacksOtherCompletion, CompletionDataComparisonFeedbacksPreferedCompletion, CompletionDataComparisonFeedbacksUsecase, CompletionDataDirectFeedbacks, CompletionDataDirectFeedbacksMetric, CompletionDataLabels, CompletionDataMetadata, CompletionDataMetadataUsage, CompletionDataModel, CustomRecipeData, CustomRecipeDataCreatedBy, CustomRecipeDataLabels, DatasetData, DatasetDataMetricsUsage, DatasetDataMetricsUsageMetric, GraderData, GraderDataGraderConfigCustomConfigOutput, GraderDataGraderConfigJudgeConfigOutput, GraderDataGraderConfigJudgeConfigOutputExamples, GraderDataGraderConfigJudgeConfigOutputExamplesInput, GraderDataGraderConfigJudgeConfigOutputModel, GraderDataGraderConfigPrebuiltConfigOutput, GraderDataGraderConfigPrebuiltConfigOutputModel, GraderDataGraderConfigPrebuiltConfigOutputPrebuiltCriteria, GraderDataGraderConfigRemoteConfigOutput, GraderDataMetric, GraderDataUseCase, HarmonyGroupData, HarmonyGroupDataComputePool, HarmonyGroupDataOnlineModels, JobData, JobDataCreatedBy, JobDataDetails, JobDataDetailsArtifacts, JobDataRecipe, JobDataStages, JobDataStagesInfoBatchInferenceJobStageOutput, JobDataStagesInfoEvalJobStageOutput, JobDataStagesInfoTrainingJobStageOutput, JobDataUseCase, JobStageOutputData, JobStageOutputDataInfoBatchInferenceJobStageOutput, JobStageOutputDataInfoEvalJobStageOutput, JobStageOutputDataInfoTrainingJobStageOutput, JudgeData, JudgeDataExamples, JudgeDataExamplesInput, JudgeDataMetric, JudgeDataModel, ListCompletionsFilterOutputData, ListCompletionsFilterOutputDataFeedbacks, ListCompletionsFilterOutputDataLabels, ListCompletionsFilterOutputDataTimerange, MetricData, MetricDataAdmin, MetricDataAdminUseCases, MetricWithContextData, ModelData, ModelDataAdmin, ModelDataAdminUseCases, ModelDataComputeConfig, ModelServiceData, ModelServiceDataModel, ModelServiceDataModelBackbone, RemoteEnvData, UseCaseData, UseCaseDataMetrics, UseCaseDataModelServices, UseCaseDataShares, UseCaseDataSharesRole, UseCaseDataSharesTeam, UserData, UserDataTeams, UserDataTeamsRole, UserDataTeamsTeam
+from .get_custom_recipe import GetCustomRecipe, GetCustomRecipeCustomRecipe
+from .get_grader import GetGrader, GetGraderGrader
 from .get_judge import GetJudge, GetJudgeJudge
-from .input_types import AbcampaignCreate, AbCampaignFilter, AdaptRequestConfigInput, AddExternalModelInput, AddHFModelInput, AddModelInput, AijudgeEvaluation, AnswerRelevancyRecipe, AnthropicProviderDataInput, ApiKeyCreate, AttachModel, AzureProviderDataInput, BaseTrainingParamsInput, CompletionComparisonFilterInput, CompletionFeedbackFilterInput, CompletionLabelValue, ContextRelevancyRecipe, CreateRecipeInput, CursorPageInput, CustomRecipe, CustomRecipeConfigInput, CustomRecipeTrainingJobInput, CustomScriptFilterInput, DatasetCreate, DatasetGenerate, DatasetGenerationConfig, DpotrainingParamsInput, EmojiInput, EvaluationCreate, EvaluationKind, EvaluationRecipeInput, EvaluationV2CreateInput, EvaluatorsInput, FaithfulnessRecipe, FeedbackAddInput, FeedbackFilterInput, FeedbackUpdateInput, GenerateParameters, GoogleProviderDataInput, GraderConfigInput, GraderCreateInput, GraderUpdateInput, GrpotrainingParamsInput, GuidelineInput, JudgeConfigInput, JudgeCreate, JudgeExampleInput, JudgeExampleInputTurnEntry, JudgeTrainingInput, JudgeTrainingParamsInput, JudgeUpdate, LabelFilter, ListCompletionsFilterInput, MetricCreate, MetricGetOrCreate, MetricLink, MetricTrainingParamsInput, MetricTrainingParamsMetadata, MetricTrendInput, MetricUnlink, ModelComputeConfigInput, ModelFilter, ModelPlacementInput, ModelProviderDataInput, ModelServiceDisconnect, ModelServiceFilter, ModelServiceWithParams, NvidiaProviderDataInput, OpenAIProviderDataInput, OrderPair, PpotrainingParamsInput, PrebuiltConfigInput, PrebuiltJudgeCreate, RagdataGenerationConfig, RemoteEnvCreate, RemoteEnvEvaluation, RewardServerTrainingParamsInput, RoleCreate, SampleConfigInput, SampleDatasourceCompletions, SampleDatasourceDataset, SampleDatasourceInput, ScalarMetricConfigInput, SftparamsInput, SystemPromptTemplateCreate, SystemPromptTemplateUpdate, TeamCreate, TeamMemberRemove, TeamMemberSet, TimeRange, TimeseriesInput, TrainingConfigInput, TrainingJobInput, TrainingMetadataInput, TrainingMetadataInputParameters, TrainingObjectiveInput, UnitConfigInput, UpdateCompletion, UpdateModelService, UpdateRecipeInput, UsageFilterInput, UsagePerUseCaseFilterInput, UseCaseCreate, UseCaseFilter, UseCaseMetadataInput, UseCaseSettingsInput, UseCaseShareInput, UseCaseShares, UseCaseUpdate, WidgetInput
+from .input_types import AbcampaignCreate, AbCampaignFilter, AddExternalModelInput, AddHFModelInput, AddModelInput, AnthropicProviderDataInput, ApiKeyCreate, ArtifactFilter, AttachModel, AzureProviderDataInput, CompletionComparisonFilterInput, CompletionFeedbackFilterInput, CompletionLabelValue, CreateRecipeInput, CreateToolProviderInput, CursorPageInput, CustomConfigInput, CustomRecipeFilterInput, DatasetCreate, DatasetGenerate, DatasetGenerationConfig, EmojiInput, FeedbackAddInput, FeedbackFilterInput, FeedbackUpdateInput, GlobalUsageFilterInput, GoogleProviderDataInput, GraderConfigInput, GraderCreateInput, GraderUpdateInput, JobArtifactFilter, JobInput, JudgeConfigInput, JudgeCreate, JudgeExampleInput, JudgeExampleInputTurnEntry, JudgeUpdate, LabelFilter, LabelInput, ListCompletionsFilterInput, ListJobsFilterInput, MetricCreate, MetricGetOrCreate, MetricLink, MetricTrendInput, MetricUnlink, ModelComputeConfigInput, ModelFilter, ModelPlacementInput, ModelProviderDataInput, ModelServiceDisconnect, ModelServiceFilter, NvidiaProviderDataInput, OpenAIProviderDataInput, OrderPair, PrebuiltConfigInput, PrebuiltJudgeCreate, RagdataGenerationConfig, RemoteConfigInput, RemoteEnvCreate, RoleCreate, SystemPromptTemplateCreate, SystemPromptTemplateUpdate, TeamCreate, TeamMemberRemove, TeamMemberSet, TimeRange, TimeseriesInput, UnitConfigInput, UpdateCompletion, UpdateModelService, UpdateRecipeInput, UpdateToolProviderInput, UsageFilterInput, UsagePerUseCaseFilterInput, UseCaseCreate, UseCaseFilter, UseCaseMetadataInput, UseCaseSettingsInput, UseCaseShareInput, UseCaseShares, UseCaseUpdate, UserCreate, UserCreateTeamWithRole, WidgetInput
 from .link_metric import LinkMetric, LinkMetricLinkMetric
 from .list_ab_campaigns import ListAbCampaigns, ListAbCampaignsAbCampaigns
-from .list_compute_pools import ListComputePools, ListComputePoolsComputePools, ListComputePoolsComputePoolsPartitions
-from .list_custom_scripts import ListCustomScripts, ListCustomScriptsCustomScripts
+from .list_compute_pools import ListComputePools, ListComputePoolsComputePools, ListComputePoolsComputePoolsHarmonyGroups
+from .list_custom_recipes import ListCustomRecipes, ListCustomRecipesCustomRecipes
 from .list_datasets import ListDatasets, ListDatasetsDatasets
-from .list_evaluation_jobs import ListEvaluationJobs, ListEvaluationJobsEvaluationJobs
+from .list_graders import ListGraders, ListGradersGraders
 from .list_grouped_interactions import ListGroupedInteractions, ListGroupedInteractionsCompletionsGrouped, ListGroupedInteractionsCompletionsGroupedNodes, ListGroupedInteractionsCompletionsGroupedNodesCompletions, ListGroupedInteractionsCompletionsGroupedNodesCompletionsNodes, ListGroupedInteractionsCompletionsGroupedNodesDirectFeedbacksStats, ListGroupedInteractionsCompletionsGroupedNodesDirectFeedbacksStatsMetric, ListGroupedInteractionsCompletionsGroupedPageInfo
+from .list_harmony_groups import ListHarmonyGroups, ListHarmonyGroupsHarmonyGroups
 from .list_interactions import ListInteractions, ListInteractionsCompletions, ListInteractionsCompletionsNodes, ListInteractionsCompletionsPageInfo
+from .list_jobs import ListJobs, ListJobsJobs, ListJobsJobsNodes, ListJobsJobsPageInfo
 from .list_judge_versions import ListJudgeVersions, ListJudgeVersionsJudgeVersions
 from .list_judges import ListJudges, ListJudgesJudges
 from .list_metrics import ListMetrics, ListMetricsMetrics
 from .list_models import ListModels, ListModelsModels, ListModelsModelsBackbone
-from .list_partitions import ListPartitions, ListPartitionsPartitions
 from .list_permissions import ListPermissions
 from .list_remote_envs import ListRemoteEnvs, ListRemoteEnvsRemoteEnvs
 from .list_roles import ListRoles, ListRolesRoles
 from .list_teams import ListTeams, ListTeamsTeams
-from .list_training_jobs import ListTrainingJobs, ListTrainingJobsTrainingJobs
 from .list_use_cases import ListUseCases, ListUseCasesUseCases
 from .list_users import ListUsers, ListUsersUsers
 from .load_dataset import LoadDataset, LoadDatasetCreateDataset
+from .lock_grader import LockGrader, LockGraderLockGrader
 from .me import Me, MeMe, MeMeApiKeys
 from .remove_remote_env import RemoveRemoteEnv
 from .remove_team_member import RemoveTeamMember, RemoveTeamMemberRemoveTeamMember
+from .share_use_case import ShareUseCase, ShareUseCaseShareUseCase
 from .terminate_model import TerminateModel
 from .test_remote_env import TestRemoteEnv, TestRemoteEnvTestRemoteEnvRemoteEnvTestOffline, TestRemoteEnvTestRemoteEnvRemoteEnvTestOnline
 from .unlink_metric import UnlinkMetric
+from .update_custom_recipe import UpdateCustomRecipe, UpdateCustomRecipeUpdateCustomRecipe
+from .update_grader import UpdateGrader, UpdateGraderUpdateGrader
 from .update_judge import UpdateJudge, UpdateJudgeUpdateJudge
 from .update_model import UpdateModel, UpdateModelUpdateModelService
 from .update_model_compute_config import UpdateModelComputeConfig, UpdateModelComputeConfigUpdateModelComputeConfig
 from .update_user import UpdateUser, UpdateUserSetTeamMember, UpdateUserSetTeamMemberRole, UpdateUserSetTeamMemberTeam, UpdateUserSetTeamMemberUser
-__all__ = ['AbCampaignCreateData', 'AbCampaignDetailData', 'AbCampaignDetailDataMetric', 'AbCampaignDetailDataModels', 'AbCampaignDetailDataUseCase', 'AbCampaignFilter', 'AbCampaignReportData', 'AbCampaignReportDataVariants', 'AbCampaignReportDataVariantsComparisons', 'AbCampaignReportDataVariantsComparisonsVariant', 'AbCampaignReportDataVariantsInterval', 'AbCampaignReportDataVariantsVariant', 'AbcampaignCreate', 'AbcampaignStatus', 'AdaptRequestConfigInput', 'AddExternalModel', 'AddExternalModelAddExternalModel', 'AddExternalModelInput', 'AddHFModel', 'AddHFModelInput', 'AddModel', 'AddModelAddModel', 'AddModelAddModelBackbone', 'AddModelInput', 'AddRemoteEnv', 'AddRemoteEnvAddRemoteEnv', 'AijudgeEvaluation', 'AnswerRelevancyRecipe', 'AnthropicProviderDataInput', 'ApiKeyCreate', 'AsyncBaseClientOpenTelemetry', 'AsyncGQLClient', 'AttachModel', 'AttachModelToUseCase', 'AttachModelToUseCaseAttachModel', 'AuthProviderKind', 'AzureProviderDataInput', 'BaseClientOpenTelemetry', 'BaseModel', 'BaseTrainingParamsInput', 'CancelABCampaign', 'CancelEvaluationJob', 'CancelTrainingJob', 'CompletionComparisonFeedbackData', 'CompletionComparisonFeedbackDataModel', 'CompletionComparisonFilterInput', 'CompletionData', 'CompletionDataChatMessages', 'CompletionDataComparisonFeedbacks', 'CompletionDataComparisonFeedbacksMetric', 'CompletionDataComparisonFeedbacksOtherCompletion', 'CompletionDataComparisonFeedbacksPreferedCompletion', 'CompletionDataComparisonFeedbacksUsecase', 'CompletionDataDirectFeedbacks', 'CompletionDataDirectFeedbacksMetric', 'CompletionDataLabels', 'CompletionDataMetadata', 'CompletionDataMetadataUsage', 'CompletionDataModel', 'CompletionFeedbackFilterInput', 'CompletionGroupBy', 'CompletionLabelValue', 'CompletionSource', 'CompletionSourceOutput', 'ComputePoolCapability', 'ContextRelevancyRecipe', 'CreateAbCampaign', 'CreateAbCampaignCreateAbCampaign', 'CreateCustomRecipeTrainingJob', 'CreateCustomRecipeTrainingJobCreateCustomRecipeTrainingJob', 'CreateCustomScript', 'CreateCustomScriptCreateCustomScript', 'CreateEvaluationJob', 'CreateEvaluationJobCreateEvaluationJob', 'CreateJudge', 'CreateJudgeCreateJudge', 'CreateMetric', 'CreateMetricCreateMetric', 'CreatePrebuiltJudge', 'CreatePrebuiltJudgeCreatePrebuiltJudge', 'CreateRecipeInput', 'CreateRole', 'CreateRoleCreateRole', 'CreateTeam', 'CreateTeamCreateTeam', 'CreateTrainingJob', 'CreateTrainingJobCreateTrainingJob', 'CreateUseCase', 'CreateUseCaseCreateUseCase', 'CursorPageInput', 'CustomRecipe', 'CustomRecipeConfigInput', 'CustomRecipeTrainingJobInput', 'CustomScriptData', 'CustomScriptFilterInput', 'DatasetCreate', 'DatasetData', 'DatasetDataMetricsUsage', 'DatasetDataMetricsUsageMetric', 'DatasetGenerate', 'DatasetGenerationConfig', 'DatasetKind', 'DatasetSource', 'DateBucketUnit', 'DeleteJudge', 'DeleteJudgeDeleteJudge', 'DeployModel', 'DescribeAbCampaign', 'DescribeAbCampaignAbCampaign', 'DescribeAbCampaignAbCampaignReport', 'DescribeDataset', 'DescribeDatasetDataset', 'DescribeEvaluationJob', 'DescribeEvaluationJobEvaluationJob', 'DescribeInteraction', 'DescribeInteractionCompletion', 'DescribeMetric', 'DescribeMetricAdmin', 'DescribeMetricAdminMetric', 'DescribeMetricMetric', 'DescribeModel', 'DescribeModelAdmin', 'DescribeModelAdminModel', 'DescribeModelAdminModelBackbone', 'DescribeModelModel', 'DescribeModelModelBackbone', 'DescribeTrainingJob', 'DescribeTrainingJobTrainingJob', 'DescribeUseCase', 'DescribeUseCaseUseCase', 'DpotrainingParamsInput', 'EmojiInput', 'EvaluationCreate', 'EvaluationJobData', 'EvaluationJobDataCreatedBy', 'EvaluationJobDataDataset', 'EvaluationJobDataJudgeModels', 'EvaluationJobDataJudges', 'EvaluationJobDataModelServices', 'EvaluationJobDataReports', 'EvaluationJobDataStages', 'EvaluationJobDataUseCase', 'EvaluationJobStatus', 'EvaluationKind', 'EvaluationRecipeInput', 'EvaluationReportData', 'EvaluationReportDataDataset', 'EvaluationReportDataMetric', 'EvaluationReportDataModelService', 'EvaluationType', 'EvaluationV2CreateInput', 'EvaluatorsInput', 'ExternalModelProviderName', 'FaithfulnessRecipe', 'FeedbackAddInput', 'FeedbackFilterInput', 'FeedbackType', 'FeedbackTypeInput', 'FeedbackTypeOutput', 'FeedbackUpdateInput', 'GQLClient', 'GenerateDataset', 'GenerateDatasetGenerateDataset', 'GenerateParameters', 'GetJudge', 'GetJudgeJudge', 'GoogleProviderDataInput', 'GraderConfigInput', 'GraderCreateInput', 'GraderTypeEnum', 'GraderUpdateInput', 'GraphQLClientError', 'GraphQLClientGraphQLError', 'GraphQLClientGraphQLMultiError', 'GraphQLClientHttpError', 'GraphQLClientInvalidResponseError', 'GrpotrainingParamsInput', 'GuidelineInput', 'JobKind', 'JobStageOutputData', 'JobStageOutputDataInfoBatchInferenceJobStageOutput', 'JobStageOutputDataInfoEvalJobStageOutput', 'JobStageOutputDataInfoTrainingJobStageOutput', 'JobStatus', 'JobStatusOutput', 'JudgeCapability', 'JudgeConfigInput', 'JudgeCreate', 'JudgeData', 'JudgeDataExamples', 'JudgeDataExamplesInput', 'JudgeDataMetric', 'JudgeDataModel', 'JudgeExampleInput', 'JudgeExampleInputTurnEntry', 'JudgeTrainingInput', 'JudgeTrainingParamsInput', 'JudgeUpdate', 'LabelFilter', 'LinkMetric', 'LinkMetricLinkMetric', 'ListAbCampaigns', 'ListAbCampaignsAbCampaigns', 'ListCompletionsFilterInput', 'ListCompletionsFilterOutputData', 'ListCompletionsFilterOutputDataFeedbacks', 'ListCompletionsFilterOutputDataLabels', 'ListCompletionsFilterOutputDataTimerange', 'ListComputePools', 'ListComputePoolsComputePools', 'ListComputePoolsComputePoolsPartitions', 'ListCustomScripts', 'ListCustomScriptsCustomScripts', 'ListDatasets', 'ListDatasetsDatasets', 'ListEvaluationJobs', 'ListEvaluationJobsEvaluationJobs', 'ListGroupedInteractions', 'ListGroupedInteractionsCompletionsGrouped', 'ListGroupedInteractionsCompletionsGroupedNodes', 'ListGroupedInteractionsCompletionsGroupedNodesCompletions', 'ListGroupedInteractionsCompletionsGroupedNodesCompletionsNodes', 'ListGroupedInteractionsCompletionsGroupedNodesDirectFeedbacksStats', 'ListGroupedInteractionsCompletionsGroupedNodesDirectFeedbacksStatsMetric', 'ListGroupedInteractionsCompletionsGroupedPageInfo', 'ListInteractions', 'ListInteractionsCompletions', 'ListInteractionsCompletionsNodes', 'ListInteractionsCompletionsPageInfo', 'ListJudgeVersions', 'ListJudgeVersionsJudgeVersions', 'ListJudges', 'ListJudgesJudges', 'ListMetrics', 'ListMetricsMetrics', 'ListModels', 'ListModelsModels', 'ListModelsModelsBackbone', 'ListPartitions', 'ListPartitionsPartitions', 'ListPermissions', 'ListRemoteEnvs', 'ListRemoteEnvsRemoteEnvs', 'ListRoles', 'ListRolesRoles', 'ListTeams', 'ListTeamsTeams', 'ListTrainingJobs', 'ListTrainingJobsTrainingJobs', 'ListUseCases', 'ListUseCasesUseCases', 'ListUsers', 'ListUsersUsers', 'LoadDataset', 'LoadDatasetCreateDataset', 'Me', 'MeMe', 'MeMeApiKeys', 'MetricAggregation', 'MetricCreate', 'MetricData', 'MetricDataAdmin', 'MetricDataAdminUseCases', 'MetricGetOrCreate', 'MetricKind', 'MetricLink', 'MetricScoringType', 'MetricTrainingParamsInput', 'MetricTrainingParamsMetadata', 'MetricTrendInput', 'MetricUnlink', 'MetricWithContextData', 'ModelComputeConfigInput', 'ModelData', 'ModelDataAdmin', 'ModelDataAdminUseCases', 'ModelDataComputeConfig', 'ModelFilter', 'ModelKindFilter', 'ModelOnline', 'ModelPlacementInput', 'ModelProviderDataInput', 'ModelServiceData', 'ModelServiceDataModel', 'ModelServiceDataModelBackbone', 'ModelServiceDisconnect', 'ModelServiceFilter', 'ModelServiceWithParams', 'ModelserviceStatus', 'NvidiaProviderDataInput', 'OpenAIModel', 'OpenAIProviderDataInput', 'OrderPair', 'PartitionData', 'PartitionDataComputePool', 'PartitionDataOnlineModels', 'PartitionStatus', 'PpotrainingParamsInput', 'PrebuiltConfigInput', 'PrebuiltCriteriaKey', 'PrebuiltJudgeCreate', 'ProviderName', 'RagdataGenerationConfig', 'RemoteEnvCreate', 'RemoteEnvData', 'RemoteEnvEvaluation', 'RemoteEnvStatus', 'RemoveRemoteEnv', 'RemoveTeamMember', 'RemoveTeamMemberRemoveTeamMember', 'RewardServerTrainingParamsInput', 'RoleCreate', 'SampleConfigInput', 'SampleDatasourceCompletions', 'SampleDatasourceDataset', 'SampleDatasourceInput', 'ScalarMetricConfigInput', 'ScriptKind', 'SelectionTypeInput', 'SelectionTypeOutput', 'SftparamsInput', 'SortDirection', 'SystemPromptTemplateCreate', 'SystemPromptTemplateUpdate', 'TeamCreate', 'TeamMemberRemove', 'TeamMemberSet', 'TerminateModel', 'TestRemoteEnv', 'TestRemoteEnvTestRemoteEnvRemoteEnvTestOffline', 'TestRemoteEnvTestRemoteEnvRemoteEnvTestOnline', 'TimeRange', 'TimeseriesInput', 'TimeseriesInterval', 'TrainingConfigInput', 'TrainingConfigOutputData', 'TrainingConfigOutputDataBaseTrainingParams', 'TrainingConfigOutputDataTrainingMetadata', 'TrainingConfigOutputDataTrainingMetadataParametersDpotrainingParamsOutput', 'TrainingConfigOutputDataTrainingMetadataParametersGrpotrainingParamsOutput', 'TrainingConfigOutputDataTrainingMetadataParametersPpotrainingParamsOutput', 'TrainingConfigOutputDataTrainingObjectiveGuidelinesTrainingParamsOutput', 'TrainingConfigOutputDataTrainingObjectiveGuidelinesTrainingParamsOutputJudgeModelPrompt', 'TrainingConfigOutputDataTrainingObjectiveJudgeTrainingParamsOutput', 'TrainingConfigOutputDataTrainingObjectiveMetricTrainingParamsOutput', 'TrainingConfigOutputDataTrainingObjectiveMetricTrainingParamsOutputMetricMetadataScalarMetricConfigOutput', 'TrainingConfigOutputDataTrainingObjectiveRewardServerTrainingParamsOutput', 'TrainingConfigOutputDataTrainingObjectiveSfttrainingParamsOutput', 'TrainingJobData', 'TrainingJobDataChildModel', 'TrainingJobDataChildModelBackbone', 'TrainingJobDataConfigAdaptBuiltinRecipeConfigOutput', 'TrainingJobDataConfigAdaptBuiltinRecipeConfigOutputSampleConfig', 'TrainingJobDataConfigAdaptBuiltinRecipeConfigOutputSampleConfigDatasourceSampleDatasourceCompletionsOutput', 'TrainingJobDataConfigAdaptBuiltinRecipeConfigOutputSampleConfigDatasourceSampleDatasourceCompletionsOutputFilter', 'TrainingJobDataConfigAdaptBuiltinRecipeConfigOutputSampleConfigDatasourceSampleDatasourceDatasetOutput', 'TrainingJobDataConfigAdaptBuiltinRecipeConfigOutputTrainingConfig', 'TrainingJobDataConfigAdaptCustomRecipeConfigOutput', 'TrainingJobDataCreatedBy', 'TrainingJobDataParentModel', 'TrainingJobDataParentModelBackbone', 'TrainingJobDataStages', 'TrainingJobDataUseCase', 'TrainingJobInput', 'TrainingJobStatus', 'TrainingMetadataInput', 'TrainingMetadataInputAlignmentMethod', 'TrainingMetadataInputParameters', 'TrainingMetadataInputTrainingType', 'TrainingMetadataOutputAlignmentMethod', 'TrainingMetadataOutputTrainingType', 'TrainingObjectiveInput', 'UnitConfigInput', 'UnitPosition', 'UnlinkMetric', 'UpdateCompletion', 'UpdateJudge', 'UpdateJudgeUpdateJudge', 'UpdateModel', 'UpdateModelComputeConfig', 'UpdateModelComputeConfigUpdateModelComputeConfig', 'UpdateModelService', 'UpdateModelUpdateModelService', 'UpdateRecipeInput', 'UpdateUser', 'UpdateUserSetTeamMember', 'UpdateUserSetTeamMemberRole', 'UpdateUserSetTeamMemberTeam', 'UpdateUserSetTeamMemberUser', 'Upload', 'UsageFilterInput', 'UsagePerUseCaseFilterInput', 'UseCaseCreate', 'UseCaseData', 'UseCaseDataMetrics', 'UseCaseDataModelServices', 'UseCaseFilter', 'UseCaseMetadataInput', 'UseCaseSettingsInput', 'UseCaseShareInput', 'UseCaseShares', 'UseCaseUpdate', 'UserData', 'UserDataTeams', 'UserDataTeamsRole', 'UserDataTeamsTeam', 'WidgetInput']
+__all__ = ['AbCampaignCreateData', 'AbCampaignDetailData', 'AbCampaignDetailDataMetric', 'AbCampaignDetailDataModels', 'AbCampaignDetailDataUseCase', 'AbCampaignFilter', 'AbCampaignReportData', 'AbCampaignReportDataVariants', 'AbCampaignReportDataVariantsComparisons', 'AbCampaignReportDataVariantsComparisonsVariant', 'AbCampaignReportDataVariantsInterval', 'AbCampaignReportDataVariantsVariant', 'AbcampaignCreate', 'AbcampaignStatus', 'AddExternalModel', 'AddExternalModelAddExternalModel', 'AddExternalModelInput', 'AddHFModel', 'AddHFModelImportHfModel', 'AddHFModelInput', 'AddModel', 'AddModelAddModel', 'AddModelAddModelBackbone', 'AddModelInput', 'AddRemoteEnv', 'AddRemoteEnvAddRemoteEnv', 'AnthropicProviderDataInput', 'ApiKeyCreate', 'ArtifactFilter', 'AsyncBaseClientOpenTelemetry', 'AsyncGQLClient', 'AttachModel', 'AttachModelToUseCase', 'AttachModelToUseCaseAttachModel', 'AuthProviderKind', 'AzureProviderDataInput', 'BaseClientOpenTelemetry', 'BaseModel', 'BillingCycle', 'CancelABCampaign', 'CancelJob', 'CancelJobCancelJob', 'CompletionComparisonFeedbackData', 'CompletionComparisonFeedbackDataModel', 'CompletionComparisonFilterInput', 'CompletionData', 'CompletionDataChatMessages', 'CompletionDataComparisonFeedbacks', 'CompletionDataComparisonFeedbacksMetric', 'CompletionDataComparisonFeedbacksOtherCompletion', 'CompletionDataComparisonFeedbacksPreferedCompletion', 'CompletionDataComparisonFeedbacksUsecase', 'CompletionDataDirectFeedbacks', 'CompletionDataDirectFeedbacksMetric', 'CompletionDataLabels', 'CompletionDataMetadata', 'CompletionDataMetadataUsage', 'CompletionDataModel', 'CompletionFeedbackFilterInput', 'CompletionGroupBy', 'CompletionLabelValue', 'CompletionSource', 'CompletionSourceOutput', 'ComputePoolCapability', 'CreateAbCampaign', 'CreateAbCampaignCreateAbCampaign', 'CreateCustomRecipe', 'CreateCustomRecipeCreateCustomRecipe', 'CreateGrader', 'CreateGraderCreateGrader', 'CreateJob', 'CreateJobCreateJob', 'CreateJudge', 'CreateJudgeCreateJudge', 'CreateMetric', 'CreateMetricCreateMetric', 'CreatePrebuiltJudge', 'CreatePrebuiltJudgeCreatePrebuiltJudge', 'CreateRecipeInput', 'CreateRole', 'CreateRoleCreateRole', 'CreateTeam', 'CreateTeamCreateTeam', 'CreateToolProviderInput', 'CreateUseCase', 'CreateUseCaseCreateUseCase', 'CreateUser', 'CreateUserCreateUser', 'CursorPageInput', 'CustomConfigInput', 'CustomRecipeData', 'CustomRecipeDataCreatedBy', 'CustomRecipeDataLabels', 'CustomRecipeFilterInput', 'DatasetCreate', 'DatasetData', 'DatasetDataMetricsUsage', 'DatasetDataMetricsUsageMetric', 'DatasetGenerate', 'DatasetGenerationConfig', 'DatasetKind', 'DatasetSource', 'DateBucketUnit', 'DeleteCustomRecipe', 'DeleteDataset', 'DeleteGrader', 'DeleteGraderDeleteGrader', 'DeleteJudge', 'DeleteJudgeDeleteJudge', 'DeleteUser', 'DeleteUserDeleteUser', 'DeployModel', 'DescribeAbCampaign', 'DescribeAbCampaignAbCampaign', 'DescribeAbCampaignAbCampaignReport', 'DescribeDataset', 'DescribeDatasetDataset', 'DescribeInteraction', 'DescribeInteractionCompletion', 'DescribeJob', 'DescribeJobJob', 'DescribeMetric', 'DescribeMetricAdmin', 'DescribeMetricAdminMetric', 'DescribeMetricMetric', 'DescribeModel', 'DescribeModelAdmin', 'DescribeModelAdminModel', 'DescribeModelAdminModelBackbone', 'DescribeModelModel', 'DescribeModelModelBackbone', 'DescribeUseCase', 'DescribeUseCaseUseCase', 'EmojiInput', 'EvaluationJobStatus', 'EvaluationType', 'ExternalModelProviderName', 'FeedbackAddInput', 'FeedbackFilterInput', 'FeedbackType', 'FeedbackTypeOutput', 'FeedbackUpdateInput', 'GQLClient', 'GetCustomRecipe', 'GetCustomRecipeCustomRecipe', 'GetGrader', 'GetGraderGrader', 'GetJudge', 'GetJudgeJudge', 'GlobalUsageFilterInput', 'GoogleProviderDataInput', 'GraderConfigInput', 'GraderCreateInput', 'GraderData', 'GraderDataGraderConfigCustomConfigOutput', 'GraderDataGraderConfigJudgeConfigOutput', 'GraderDataGraderConfigJudgeConfigOutputExamples', 'GraderDataGraderConfigJudgeConfigOutputExamplesInput', 'GraderDataGraderConfigJudgeConfigOutputModel', 'GraderDataGraderConfigPrebuiltConfigOutput', 'GraderDataGraderConfigPrebuiltConfigOutputModel', 'GraderDataGraderConfigPrebuiltConfigOutputPrebuiltCriteria', 'GraderDataGraderConfigRemoteConfigOutput', 'GraderDataMetric', 'GraderDataUseCase', 'GraderTypeEnum', 'GraderUpdateInput', 'GraphQLClientError', 'GraphQLClientGraphQLError', 'GraphQLClientGraphQLMultiError', 'GraphQLClientHttpError', 'GraphQLClientInvalidResponseError', 'HarmonyGroupData', 'HarmonyGroupDataComputePool', 'HarmonyGroupDataOnlineModels', 'HarmonyStatus', 'JobArtifactFilter', 'JobArtifactKind', 'JobData', 'JobDataCreatedBy', 'JobDataDetails', 'JobDataDetailsArtifacts', 'JobDataRecipe', 'JobDataStages', 'JobDataStagesInfoBatchInferenceJobStageOutput', 'JobDataStagesInfoEvalJobStageOutput', 'JobDataStagesInfoTrainingJobStageOutput', 'JobDataUseCase', 'JobInput', 'JobKind', 'JobStageOutputData', 'JobStageOutputDataInfoBatchInferenceJobStageOutput', 'JobStageOutputDataInfoEvalJobStageOutput', 'JobStageOutputDataInfoTrainingJobStageOutput', 'JobStatus', 'JobStatusOutput', 'JudgeCapability', 'JudgeConfigInput', 'JudgeCreate', 'JudgeData', 'JudgeDataExamples', 'JudgeDataExamplesInput', 'JudgeDataMetric', 'JudgeDataModel', 'JudgeExampleInput', 'JudgeExampleInputTurnEntry', 'JudgeUpdate', 'LabelFilter', 'LabelInput', 'LinkMetric', 'LinkMetricLinkMetric', 'ListAbCampaigns', 'ListAbCampaignsAbCampaigns', 'ListCompletionsFilterInput', 'ListCompletionsFilterOutputData', 'ListCompletionsFilterOutputDataFeedbacks', 'ListCompletionsFilterOutputDataLabels', 'ListCompletionsFilterOutputDataTimerange', 'ListComputePools', 'ListComputePoolsComputePools', 'ListComputePoolsComputePoolsHarmonyGroups', 'ListCustomRecipes', 'ListCustomRecipesCustomRecipes', 'ListDatasets', 'ListDatasetsDatasets', 'ListGraders', 'ListGradersGraders', 'ListGroupedInteractions', 'ListGroupedInteractionsCompletionsGrouped', 'ListGroupedInteractionsCompletionsGroupedNodes', 'ListGroupedInteractionsCompletionsGroupedNodesCompletions', 'ListGroupedInteractionsCompletionsGroupedNodesCompletionsNodes', 'ListGroupedInteractionsCompletionsGroupedNodesDirectFeedbacksStats', 'ListGroupedInteractionsCompletionsGroupedNodesDirectFeedbacksStatsMetric', 'ListGroupedInteractionsCompletionsGroupedPageInfo', 'ListHarmonyGroups', 'ListHarmonyGroupsHarmonyGroups', 'ListInteractions', 'ListInteractionsCompletions', 'ListInteractionsCompletionsNodes', 'ListInteractionsCompletionsPageInfo', 'ListJobs', 'ListJobsFilterInput', 'ListJobsJobs', 'ListJobsJobsNodes', 'ListJobsJobsPageInfo', 'ListJudgeVersions', 'ListJudgeVersionsJudgeVersions', 'ListJudges', 'ListJudgesJudges', 'ListMetrics', 'ListMetricsMetrics', 'ListModels', 'ListModelsModels', 'ListModelsModelsBackbone', 'ListPermissions', 'ListRemoteEnvs', 'ListRemoteEnvsRemoteEnvs', 'ListRoles', 'ListRolesRoles', 'ListTeams', 'ListTeamsTeams', 'ListUseCases', 'ListUseCasesUseCases', 'ListUsers', 'ListUsersUsers', 'LoadDataset', 'LoadDatasetCreateDataset', 'LockGrader', 'LockGraderLockGrader', 'Me', 'MeMe', 'MeMeApiKeys', 'MetricAggregation', 'MetricCreate', 'MetricData', 'MetricDataAdmin', 'MetricDataAdminUseCases', 'MetricGetOrCreate', 'MetricKind', 'MetricLink', 'MetricScoringType', 'MetricTrendInput', 'MetricUnlink', 'MetricWithContextData', 'ModelComputeConfigInput', 'ModelData', 'ModelDataAdmin', 'ModelDataAdminUseCases', 'ModelDataComputeConfig', 'ModelFilter', 'ModelKindFilter', 'ModelOnline', 'ModelPlacementInput', 'ModelProviderDataInput', 'ModelServiceData', 'ModelServiceDataModel', 'ModelServiceDataModelBackbone', 'ModelServiceDisconnect', 'ModelServiceFilter', 'ModelserviceStatus', 'NvidiaProviderDataInput', 'OpenAIModel', 'OpenAIProviderDataInput', 'OrderPair', 'PrebuiltConfigInput', 'PrebuiltCriteriaKey', 'PrebuiltJudgeCreate', 'Protocol', 'ProviderName', 'RagdataGenerationConfig', 'RemoteConfigInput', 'RemoteEnvCreate', 'RemoteEnvData', 'RemoteEnvStatus', 'RemoveRemoteEnv', 'RemoveTeamMember', 'RemoveTeamMemberRemoveTeamMember', 'RoleCreate', 'SelectionTypeOutput', 'ShareUseCase', 'ShareUseCaseShareUseCase', 'SortDirection', 'SystemPromptTemplateCreate', 'SystemPromptTemplateUpdate', 'TeamCreate', 'TeamMemberRemove', 'TeamMemberSet', 'TerminateModel', 'TestRemoteEnv', 'TestRemoteEnvTestRemoteEnvRemoteEnvTestOffline', 'TestRemoteEnvTestRemoteEnvRemoteEnvTestOnline', 'TimeRange', 'TimeseriesInput', 'TimeseriesInterval', 'TrainingJobStatus', 'TrainingMetadataOutputAlignmentMethod', 'TrainingMetadataOutputTrainingType', 'UnitConfigInput', 'UnitPosition', 'UnlinkMetric', 'UpdateCompletion', 'UpdateCustomRecipe', 'UpdateCustomRecipeUpdateCustomRecipe', 'UpdateGrader', 'UpdateGraderUpdateGrader', 'UpdateJudge', 'UpdateJudgeUpdateJudge', 'UpdateModel', 'UpdateModelComputeConfig', 'UpdateModelComputeConfigUpdateModelComputeConfig', 'UpdateModelService', 'UpdateModelUpdateModelService', 'UpdateRecipeInput', 'UpdateToolProviderInput', 'UpdateUser', 'UpdateUserSetTeamMember', 'UpdateUserSetTeamMemberRole', 'UpdateUserSetTeamMemberTeam', 'UpdateUserSetTeamMemberUser', 'Upload', 'UsageFilterInput', 'UsagePerUseCaseFilterInput', 'UseCaseCreate', 'UseCaseData', 'UseCaseDataMetrics', 'UseCaseDataModelServices', 'UseCaseDataShares', 'UseCaseDataSharesRole', 'UseCaseDataSharesTeam', 'UseCaseFilter', 'UseCaseMetadataInput', 'UseCaseSettingsInput', 'UseCaseShareInput', 'UseCaseShares', 'UseCaseUpdate', 'UserCreate', 'UserCreateTeamWithRole', 'UserData', 'UserDataTeams', 'UserDataTeamsRole', 'UserDataTeamsTeam', 'WidgetInput']

adaptive_sdk/graphql_client/add_hf_model.py

@@ -1,6 +1,12 @@
 from pydantic import Field
 from .base_model import BaseModel
+from .fragments import JobData
 
 class AddHFModel(BaseModel):
     """@public"""
-    import_hf_model: str = Field(alias='importHfModel')
+    import_hf_model: 'AddHFModelImportHfModel' = Field(alias='importHfModel')
+
+class AddHFModelImportHfModel(JobData):
+    """@public"""
+    pass
+AddHFModel.model_rebuild()