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

adaptive_sdk/graphql_client/client.py

@@ -3,8 +3,8 @@ from graphql import DocumentNode, NamedTypeNode, NameNode, OperationDefinitionNo
 from .add_external_model import AddExternalModel
 from .add_hf_model import AddHFModel
 from .add_model import AddModel
+from .add_model_to_use_case import AddModelToUseCase
 from .add_remote_env import AddRemoteEnv
-from .attach_model_to_use_case import AttachModelToUseCase
 from .base_client_open_telemetry import BaseClientOpenTelemetry
 from .base_model import UNSET, UnsetType, Upload
 from .base_operation import GraphQLField
@@ -42,7 +42,7 @@ from .enums import CompletionGroupBy
 from .get_custom_recipe import GetCustomRecipe
 from .get_grader import GetGrader
 from .get_judge import GetJudge
-from .input_types import AbcampaignCreate, AbCampaignFilter, AddExternalModelInput, AddHFModelInput, AddModelInput, AttachModel, CreateRecipeInput, CursorPageInput, CustomRecipeFilterInput, DatasetCreate, DatasetCreateFromMultipartUpload, DatasetUploadProcessingStatusInput, GraderCreateInput, GraderUpdateInput, JobInput, JudgeCreate, JudgeUpdate, ListCompletionsFilterInput, ListJobsFilterInput, MetricCreate, MetricLink, MetricUnlink, ModelComputeConfigInput, ModelFilter, ModelPlacementInput, OrderPair, PrebuiltJudgeCreate, RemoteEnvCreate, ResizePartitionInput, RoleCreate, TeamCreate, TeamMemberRemove, TeamMemberSet, UpdateModelService, UpdateRecipeInput, UseCaseCreate, UseCaseShares, UserCreate
+from .input_types import AbcampaignCreate, AbCampaignFilter, AddExternalModelInput, AddHFModelInput, AddModelInput, AddModelToUseCaseInput, CreateRecipeInput, CursorPageInput, CustomRecipeFilterInput, DatasetCreate, DatasetCreateFromMultipartUpload, DatasetUploadProcessingStatusInput, DeployModelInput, GraderCreateInput, GraderUpdateInput, JobInput, JudgeCreate, JudgeUpdate, ListCompletionsFilterInput, ListJobsFilterInput, MetricCreate, MetricLink, MetricUnlink, ModelComputeConfigInput, ModelFilter, OrderPair, PrebuiltJudgeCreate, RemoteEnvCreate, RemoveModelFromUseCaseInput, ResizePartitionInput, RoleCreate, TeamCreate, TeamMemberRemove, TeamMemberSet, UpdateModelService, UpdateRecipeInput, UseCaseCreate, UseCaseShares, UserCreate
 from .link_metric import LinkMetric
 from .list_ab_campaigns import ListAbCampaigns
 from .list_compute_pools import ListComputePools
@@ -66,6 +66,7 @@ from .list_users import ListUsers
 from .load_dataset import LoadDataset
 from .lock_grader import LockGrader
 from .me import Me
+from .remove_model_from_use_case import RemoveModelFromUseCase
 from .remove_remote_env import RemoveRemoteEnv
 from .remove_team_member import RemoveTeamMember
 from .resize_inference_partition import ResizeInferencePartition
@@ -107,13 +108,6 @@ class GQLClient(BaseClientOpenTelemetry):
         data = self.get_data(response)
         return UnlinkMetric.model_validate(data)
 
-    def attach_model_to_use_case(self, input: AttachModel, **kwargs: Any) -> AttachModelToUseCase:
-        query = gql('\n            mutation AttachModelToUseCase($input: AttachModel!) {\n              attachModel(input: $input) {\n                ...ModelServiceData\n              }\n            }\n\n            fragment ModelData on Model {\n              id\n              key\n              name\n              online\n              error\n              isExternal\n              providerName\n              isAdapter\n              isTraining\n              createdAt\n              size\n              computeConfig {\n                tp\n                kvCacheLen\n                maxSeqLen\n              }\n            }\n\n            fragment ModelServiceData on ModelService {\n              id\n              key\n              name\n              model {\n                ...ModelData\n                backbone {\n                  ...ModelData\n                }\n              }\n              attached\n              isDefault\n              desiredOnline\n              createdAt\n            }\n            ')
-        variables: Dict[str, object] = {'input': input}
-        response = self.execute(query=query, operation_name='AttachModelToUseCase', variables=variables, **kwargs)
-        data = self.get_data(response)
-        return AttachModelToUseCase.model_validate(data)
-
     def add_external_model(self, input: AddExternalModelInput, **kwargs: Any) -> AddExternalModel:
         query = gql('\n            mutation AddExternalModel($input: AddExternalModelInput!) {\n              addExternalModel(input: $input) {\n                ...ModelData\n              }\n            }\n\n            fragment ModelData on Model {\n              id\n              key\n              name\n              online\n              error\n              isExternal\n              providerName\n              isAdapter\n              isTraining\n              createdAt\n              size\n              computeConfig {\n                tp\n                kvCacheLen\n                maxSeqLen\n              }\n            }\n            ')
         variables: Dict[str, object] = {'input': input}
@@ -129,19 +123,12 @@ class GQLClient(BaseClientOpenTelemetry):
         return AddModel.model_validate(data)
 
     def update_model(self, input: UpdateModelService, **kwargs: Any) -> UpdateModel:
-        query = gql('\n            mutation UpdateModel($input: UpdateModelService!) {\n              updateModelService(input: $input) {\n                ...ModelServiceData\n              }\n            }\n\n            fragment ModelData on Model {\n              id\n              key\n              name\n              online\n              error\n              isExternal\n              providerName\n              isAdapter\n              isTraining\n              createdAt\n              size\n              computeConfig {\n                tp\n                kvCacheLen\n                maxSeqLen\n              }\n            }\n\n            fragment ModelServiceData on ModelService {\n              id\n              key\n              name\n              model {\n                ...ModelData\n                backbone {\n                  ...ModelData\n                }\n              }\n              attached\n              isDefault\n              desiredOnline\n              createdAt\n            }\n            ')
+        query = gql('\n            mutation UpdateModel($input: UpdateModelService!) {\n              updateModelService(input: $input) {\n                ...ModelServiceData\n              }\n            }\n\n            fragment ModelData on Model {\n              id\n              key\n              name\n              online\n              error\n              isExternal\n              providerName\n              isAdapter\n              isTraining\n              createdAt\n              size\n              computeConfig {\n                tp\n                kvCacheLen\n                maxSeqLen\n              }\n            }\n\n            fragment ModelServiceData on ModelService {\n              id\n              key\n              name\n              model {\n                ...ModelData\n                backbone {\n                  ...ModelData\n                }\n              }\n              isDefault\n              desiredOnline\n              createdAt\n            }\n            ')
         variables: Dict[str, object] = {'input': input}
         response = self.execute(query=query, operation_name='UpdateModel', variables=variables, **kwargs)
         data = self.get_data(response)
         return UpdateModel.model_validate(data)
 
-    def deploy_model(self, id_or_key: str, wait: bool, placement: Union[Optional[ModelPlacementInput], UnsetType]=UNSET, **kwargs: Any) -> DeployModel:
-        query = gql('\n            mutation DeployModel($idOrKey: IdOrKey!, $wait: Boolean!, $placement: ModelPlacementInput) {\n              deployModel(idOrKey: $idOrKey, wait: $wait, placement: $placement)\n            }\n            ')
-        variables: Dict[str, object] = {'idOrKey': id_or_key, 'wait': wait, 'placement': placement}
-        response = self.execute(query=query, operation_name='DeployModel', variables=variables, **kwargs)
-        data = self.get_data(response)
-        return DeployModel.model_validate(data)
-
     def terminate_model(self, id_or_key: str, force: bool, **kwargs: Any) -> TerminateModel:
         query = gql('\n            mutation TerminateModel($idOrKey: IdOrKey!, $force: Boolean!) {\n              terminateModel(idOrKey: $idOrKey, force: $force)\n            }\n            ')
         variables: Dict[str, object] = {'idOrKey': id_or_key, 'force': force}
@@ -150,14 +137,14 @@ class GQLClient(BaseClientOpenTelemetry):
         return TerminateModel.model_validate(data)
 
     def create_use_case(self, input: UseCaseCreate, **kwargs: Any) -> CreateUseCase:
-        query = gql('\n            mutation CreateUseCase($input: UseCaseCreate!) {\n              createUseCase(input: $input) {\n                ...UseCaseData\n              }\n            }\n\n            fragment MetricWithContextData on MetricWithContext {\n              id\n              key\n              name\n              kind\n              description\n              scoringType\n              createdAt\n            }\n\n            fragment ModelData on Model {\n              id\n              key\n              name\n              online\n              error\n              isExternal\n              providerName\n              isAdapter\n              isTraining\n              createdAt\n              size\n              computeConfig {\n                tp\n                kvCacheLen\n                maxSeqLen\n              }\n            }\n\n            fragment ModelServiceData on ModelService {\n              id\n              key\n              name\n              model {\n                ...ModelData\n                backbone {\n                  ...ModelData\n                }\n              }\n              attached\n              isDefault\n              desiredOnline\n              createdAt\n            }\n\n            fragment UseCaseData on UseCase {\n              id\n              key\n              name\n              description\n              createdAt\n              metrics {\n                ...MetricWithContextData\n              }\n              modelServices {\n                ...ModelServiceData\n              }\n              permissions\n              shares {\n                team {\n                  id\n                  key\n                  name\n                  createdAt\n                }\n                role {\n                  id\n                  key\n                  name\n                  createdAt\n                  permissions\n                }\n                isOwner\n              }\n            }\n            ')
+        query = gql('\n            mutation CreateUseCase($input: UseCaseCreate!) {\n              createUseCase(input: $input) {\n                ...UseCaseData\n              }\n            }\n\n            fragment MetricWithContextData on MetricWithContext {\n              id\n              key\n              name\n              kind\n              description\n              scoringType\n              createdAt\n            }\n\n            fragment ModelData on Model {\n              id\n              key\n              name\n              online\n              error\n              isExternal\n              providerName\n              isAdapter\n              isTraining\n              createdAt\n              size\n              computeConfig {\n                tp\n                kvCacheLen\n                maxSeqLen\n              }\n            }\n\n            fragment ModelServiceData on ModelService {\n              id\n              key\n              name\n              model {\n                ...ModelData\n                backbone {\n                  ...ModelData\n                }\n              }\n              isDefault\n              desiredOnline\n              createdAt\n            }\n\n            fragment UseCaseData on UseCase {\n              id\n              key\n              name\n              description\n              createdAt\n              metrics {\n                ...MetricWithContextData\n              }\n              modelServices {\n                ...ModelServiceData\n              }\n              permissions\n              shares {\n                team {\n                  id\n                  key\n                  name\n                  createdAt\n                }\n                role {\n                  id\n                  key\n                  name\n                  createdAt\n                  permissions\n                }\n                isOwner\n              }\n            }\n            ')
         variables: Dict[str, object] = {'input': input}
         response = self.execute(query=query, operation_name='CreateUseCase', variables=variables, **kwargs)
         data = self.get_data(response)
         return CreateUseCase.model_validate(data)
 
     def share_use_case(self, id_or_key: str, input: UseCaseShares, **kwargs: Any) -> ShareUseCase:
-        query = gql('\n            mutation ShareUseCase($idOrKey: IdOrKey!, $input: UseCaseShares!) {\n              shareUseCase(idOrKey: $idOrKey, input: $input) {\n                ...UseCaseData\n              }\n            }\n\n            fragment MetricWithContextData on MetricWithContext {\n              id\n              key\n              name\n              kind\n              description\n              scoringType\n              createdAt\n            }\n\n            fragment ModelData on Model {\n              id\n              key\n              name\n              online\n              error\n              isExternal\n              providerName\n              isAdapter\n              isTraining\n              createdAt\n              size\n              computeConfig {\n                tp\n                kvCacheLen\n                maxSeqLen\n              }\n            }\n\n            fragment ModelServiceData on ModelService {\n              id\n              key\n              name\n              model {\n                ...ModelData\n                backbone {\n                  ...ModelData\n                }\n              }\n              attached\n              isDefault\n              desiredOnline\n              createdAt\n            }\n\n            fragment UseCaseData on UseCase {\n              id\n              key\n              name\n              description\n              createdAt\n              metrics {\n                ...MetricWithContextData\n              }\n              modelServices {\n                ...ModelServiceData\n              }\n              permissions\n              shares {\n                team {\n                  id\n                  key\n                  name\n                  createdAt\n                }\n                role {\n                  id\n                  key\n                  name\n                  createdAt\n                  permissions\n                }\n                isOwner\n              }\n            }\n            ')
+        query = gql('\n            mutation ShareUseCase($idOrKey: IdOrKey!, $input: UseCaseShares!) {\n              shareUseCase(idOrKey: $idOrKey, input: $input) {\n                ...UseCaseData\n              }\n            }\n\n            fragment MetricWithContextData on MetricWithContext {\n              id\n              key\n              name\n              kind\n              description\n              scoringType\n              createdAt\n            }\n\n            fragment ModelData on Model {\n              id\n              key\n              name\n              online\n              error\n              isExternal\n              providerName\n              isAdapter\n              isTraining\n              createdAt\n              size\n              computeConfig {\n                tp\n                kvCacheLen\n                maxSeqLen\n              }\n            }\n\n            fragment ModelServiceData on ModelService {\n              id\n              key\n              name\n              model {\n                ...ModelData\n                backbone {\n                  ...ModelData\n                }\n              }\n              isDefault\n              desiredOnline\n              createdAt\n            }\n\n            fragment UseCaseData on UseCase {\n              id\n              key\n              name\n              description\n              createdAt\n              metrics {\n                ...MetricWithContextData\n              }\n              modelServices {\n                ...ModelServiceData\n              }\n              permissions\n              shares {\n                team {\n                  id\n                  key\n                  name\n                  createdAt\n                }\n                role {\n                  id\n                  key\n                  name\n                  createdAt\n                  permissions\n                }\n                isOwner\n              }\n            }\n            ')
         variables: Dict[str, object] = {'idOrKey': id_or_key, 'input': input}
         response = self.execute(query=query, operation_name='ShareUseCase', variables=variables, **kwargs)
         data = self.get_data(response)
@@ -192,7 +179,7 @@ class GQLClient(BaseClientOpenTelemetry):
         return DeleteDataset.model_validate(data)
 
     def add_hf_model(self, input: AddHFModelInput, **kwargs: Any) -> AddHFModel:
-        query = gql('\n            mutation AddHFModel($input: AddHFModelInput!) {\n              importHfModel(input: $input) {\n                ...JobData\n              }\n            }\n\n            fragment CustomRecipeData on CustomRecipe {\n              id\n              key\n              name\n              content\n              contentHash\n              editable\n              global\n              builtin\n              inputSchema\n              jsonSchema\n              description\n              labels {\n                key\n                value\n              }\n              createdAt\n              updatedAt\n              createdBy {\n                id\n                name\n                email\n              }\n            }\n\n            fragment JobData on Job {\n              id\n              name\n              status\n              createdAt\n              createdBy {\n                id\n                name\n              }\n              startedAt\n              endedAt\n              durationMs\n              progress\n              error\n              kind\n              stages {\n                name\n                status\n                info {\n                  __typename\n                  ... on TrainingJobStageOutput {\n                    monitoringLink\n                    totalNumSamples\n                    processedNumSamples\n                    checkpoints\n                  }\n                  ... on EvalJobStageOutput {\n                    totalNumSamples\n                    processedNumSamples\n                  }\n                  ... on BatchInferenceJobStageOutput {\n                    totalNumSamples\n                    processedNumSamples\n                  }\n                }\n              }\n              useCase {\n                id\n                key\n                name\n              }\n              recipe {\n                ...CustomRecipeData\n              }\n              details {\n                args\n                recipeHash\n                artifacts {\n                  id\n                  name\n                  kind\n                  uri\n                  metadata\n                  createdAt\n                  byproducts {\n                    __typename\n                    ... on EvaluationByproducts {\n                      evalResults {\n                        mean\n                        min\n                        max\n                        stddev\n                        count\n                        sum\n                        feedbackCount\n                        jobId\n                        artifactId\n                        modelService {\n                          key\n                          name\n                        }\n                        metric {\n                          key\n                          name\n                        }\n                      }\n                    }\n                  }\n                }\n              }\n            }\n            ')
+        query = gql('\n            mutation AddHFModel($input: AddHFModelInput!) {\n              importHfModel(input: $input) {\n                ...JobData\n              }\n            }\n\n            fragment CustomRecipeData on CustomRecipe {\n              id\n              key\n              name\n              content\n              contentHash\n              editable\n              global\n              builtin\n              inputSchema\n              jsonSchema\n              description\n              labels {\n                key\n                value\n              }\n              createdAt\n              updatedAt\n              createdBy {\n                id\n                name\n                email\n              }\n            }\n\n            fragment JobData on Job {\n              id\n              name\n              status\n              createdAt\n              createdBy {\n                id\n                name\n              }\n              startedAt\n              endedAt\n              durationMs\n              progress\n              error\n              kind\n              stages {\n                name\n                status\n                info {\n                  __typename\n                  ... on TrainingJobStageOutput {\n                    monitoringLink\n                    totalNumSamples\n                    processedNumSamples\n                    checkpoints\n                  }\n                  ... on EvalJobStageOutput {\n                    totalNumSamples\n                    processedNumSamples\n                  }\n                  ... on BatchInferenceJobStageOutput {\n                    totalNumSamples\n                    processedNumSamples\n                  }\n                }\n              }\n              useCase {\n                id\n                key\n                name\n              }\n              recipe {\n                ...CustomRecipeData\n              }\n              details {\n                args\n                recipeHash\n                artifacts {\n                  id\n                  name\n                  kind\n                  status\n                  uri\n                  metadata\n                  createdAt\n                  byproducts {\n                    __typename\n                    ... on EvaluationByproducts {\n                      evalResults {\n                        mean\n                        min\n                        max\n                        stddev\n                        count\n                        sum\n                        feedbackCount\n                        jobId\n                        artifactId\n                        modelService {\n                          key\n                          name\n                        }\n                        metric {\n                          key\n                          name\n                        }\n                      }\n                    }\n                  }\n                }\n              }\n            }\n            ')
         variables: Dict[str, object] = {'input': input}
         response = self.execute(query=query, operation_name='AddHFModel', variables=variables, **kwargs)
         data = self.get_data(response)
@@ -346,14 +333,14 @@ class GQLClient(BaseClientOpenTelemetry):
         return RemoveTeamMember.model_validate(data)
 
     def create_job(self, input: JobInput, **kwargs: Any) -> CreateJob:
-        query = gql('\n            mutation CreateJob($input: JobInput!) {\n              createJob(input: $input) {\n                ...JobData\n              }\n            }\n\n            fragment CustomRecipeData on CustomRecipe {\n              id\n              key\n              name\n              content\n              contentHash\n              editable\n              global\n              builtin\n              inputSchema\n              jsonSchema\n              description\n              labels {\n                key\n                value\n              }\n              createdAt\n              updatedAt\n              createdBy {\n                id\n                name\n                email\n              }\n            }\n\n            fragment JobData on Job {\n              id\n              name\n              status\n              createdAt\n              createdBy {\n                id\n                name\n              }\n              startedAt\n              endedAt\n              durationMs\n              progress\n              error\n              kind\n              stages {\n                name\n                status\n                info {\n                  __typename\n                  ... on TrainingJobStageOutput {\n                    monitoringLink\n                    totalNumSamples\n                    processedNumSamples\n                    checkpoints\n                  }\n                  ... on EvalJobStageOutput {\n                    totalNumSamples\n                    processedNumSamples\n                  }\n                  ... on BatchInferenceJobStageOutput {\n                    totalNumSamples\n                    processedNumSamples\n                  }\n                }\n              }\n              useCase {\n                id\n                key\n                name\n              }\n              recipe {\n                ...CustomRecipeData\n              }\n              details {\n                args\n                recipeHash\n                artifacts {\n                  id\n                  name\n                  kind\n                  uri\n                  metadata\n                  createdAt\n                  byproducts {\n                    __typename\n                    ... on EvaluationByproducts {\n                      evalResults {\n                        mean\n                        min\n                        max\n                        stddev\n                        count\n                        sum\n                        feedbackCount\n                        jobId\n                        artifactId\n                        modelService {\n                          key\n                          name\n                        }\n                        metric {\n                          key\n                          name\n                        }\n                      }\n                    }\n                  }\n                }\n              }\n            }\n            ')
+        query = gql('\n            mutation CreateJob($input: JobInput!) {\n              createJob(input: $input) {\n                ...JobData\n              }\n            }\n\n            fragment CustomRecipeData on CustomRecipe {\n              id\n              key\n              name\n              content\n              contentHash\n              editable\n              global\n              builtin\n              inputSchema\n              jsonSchema\n              description\n              labels {\n                key\n                value\n              }\n              createdAt\n              updatedAt\n              createdBy {\n                id\n                name\n                email\n              }\n            }\n\n            fragment JobData on Job {\n              id\n              name\n              status\n              createdAt\n              createdBy {\n                id\n                name\n              }\n              startedAt\n              endedAt\n              durationMs\n              progress\n              error\n              kind\n              stages {\n                name\n                status\n                info {\n                  __typename\n                  ... on TrainingJobStageOutput {\n                    monitoringLink\n                    totalNumSamples\n                    processedNumSamples\n                    checkpoints\n                  }\n                  ... on EvalJobStageOutput {\n                    totalNumSamples\n                    processedNumSamples\n                  }\n                  ... on BatchInferenceJobStageOutput {\n                    totalNumSamples\n                    processedNumSamples\n                  }\n                }\n              }\n              useCase {\n                id\n                key\n                name\n              }\n              recipe {\n                ...CustomRecipeData\n              }\n              details {\n                args\n                recipeHash\n                artifacts {\n                  id\n                  name\n                  kind\n                  status\n                  uri\n                  metadata\n                  createdAt\n                  byproducts {\n                    __typename\n                    ... on EvaluationByproducts {\n                      evalResults {\n                        mean\n                        min\n                        max\n                        stddev\n                        count\n                        sum\n                        feedbackCount\n                        jobId\n                        artifactId\n                        modelService {\n                          key\n                          name\n                        }\n                        metric {\n                          key\n                          name\n                        }\n                      }\n                    }\n                  }\n                }\n              }\n            }\n            ')
         variables: Dict[str, object] = {'input': input}
         response = self.execute(query=query, operation_name='CreateJob', variables=variables, **kwargs)
         data = self.get_data(response)
         return CreateJob.model_validate(data)
 
     def cancel_job(self, job_id: Any, **kwargs: Any) -> CancelJob:
-        query = gql('\n            mutation CancelJob($jobId: UUID!) {\n              cancelJob(id: $jobId) {\n                ...JobData\n              }\n            }\n\n            fragment CustomRecipeData on CustomRecipe {\n              id\n              key\n              name\n              content\n              contentHash\n              editable\n              global\n              builtin\n              inputSchema\n              jsonSchema\n              description\n              labels {\n                key\n                value\n              }\n              createdAt\n              updatedAt\n              createdBy {\n                id\n                name\n                email\n              }\n            }\n\n            fragment JobData on Job {\n              id\n              name\n              status\n              createdAt\n              createdBy {\n                id\n                name\n              }\n              startedAt\n              endedAt\n              durationMs\n              progress\n              error\n              kind\n              stages {\n                name\n                status\n                info {\n                  __typename\n                  ... on TrainingJobStageOutput {\n                    monitoringLink\n                    totalNumSamples\n                    processedNumSamples\n                    checkpoints\n                  }\n                  ... on EvalJobStageOutput {\n                    totalNumSamples\n                    processedNumSamples\n                  }\n                  ... on BatchInferenceJobStageOutput {\n                    totalNumSamples\n                    processedNumSamples\n                  }\n                }\n              }\n              useCase {\n                id\n                key\n                name\n              }\n              recipe {\n                ...CustomRecipeData\n              }\n              details {\n                args\n                recipeHash\n                artifacts {\n                  id\n                  name\n                  kind\n                  uri\n                  metadata\n                  createdAt\n                  byproducts {\n                    __typename\n                    ... on EvaluationByproducts {\n                      evalResults {\n                        mean\n                        min\n                        max\n                        stddev\n                        count\n                        sum\n                        feedbackCount\n                        jobId\n                        artifactId\n                        modelService {\n                          key\n                          name\n                        }\n                        metric {\n                          key\n                          name\n                        }\n                      }\n                    }\n                  }\n                }\n              }\n            }\n            ')
+        query = gql('\n            mutation CancelJob($jobId: UUID!) {\n              cancelJob(id: $jobId) {\n                ...JobData\n              }\n            }\n\n            fragment CustomRecipeData on CustomRecipe {\n              id\n              key\n              name\n              content\n              contentHash\n              editable\n              global\n              builtin\n              inputSchema\n              jsonSchema\n              description\n              labels {\n                key\n                value\n              }\n              createdAt\n              updatedAt\n              createdBy {\n                id\n                name\n                email\n              }\n            }\n\n            fragment JobData on Job {\n              id\n              name\n              status\n              createdAt\n              createdBy {\n                id\n                name\n              }\n              startedAt\n              endedAt\n              durationMs\n              progress\n              error\n              kind\n              stages {\n                name\n                status\n                info {\n                  __typename\n                  ... on TrainingJobStageOutput {\n                    monitoringLink\n                    totalNumSamples\n                    processedNumSamples\n                    checkpoints\n                  }\n                  ... on EvalJobStageOutput {\n                    totalNumSamples\n                    processedNumSamples\n                  }\n                  ... on BatchInferenceJobStageOutput {\n                    totalNumSamples\n                    processedNumSamples\n                  }\n                }\n              }\n              useCase {\n                id\n                key\n                name\n              }\n              recipe {\n                ...CustomRecipeData\n              }\n              details {\n                args\n                recipeHash\n                artifacts {\n                  id\n                  name\n                  kind\n                  status\n                  uri\n                  metadata\n                  createdAt\n                  byproducts {\n                    __typename\n                    ... on EvaluationByproducts {\n                      evalResults {\n                        mean\n                        min\n                        max\n                        stddev\n                        count\n                        sum\n                        feedbackCount\n                        jobId\n                        artifactId\n                        modelService {\n                          key\n                          name\n                        }\n                        metric {\n                          key\n                          name\n                        }\n                      }\n                    }\n                  }\n                }\n              }\n            }\n            ')
         variables: Dict[str, object] = {'jobId': job_id}
         response = self.execute(query=query, operation_name='CancelJob', variables=variables, **kwargs)
         data = self.get_data(response)
@@ -373,6 +360,27 @@ class GQLClient(BaseClientOpenTelemetry):
         data = self.get_data(response)
         return ResizeInferencePartition.model_validate(data)
 
+    def add_model_to_use_case(self, input: AddModelToUseCaseInput, **kwargs: Any) -> AddModelToUseCase:
+        query = gql('\n            mutation AddModelToUseCase($input: AddModelToUseCaseInput!) {\n              addModelToUseCase(input: $input)\n            }\n            ')
+        variables: Dict[str, object] = {'input': input}
+        response = self.execute(query=query, operation_name='AddModelToUseCase', variables=variables, **kwargs)
+        data = self.get_data(response)
+        return AddModelToUseCase.model_validate(data)
+
+    def remove_model_from_use_case(self, input: RemoveModelFromUseCaseInput, **kwargs: Any) -> RemoveModelFromUseCase:
+        query = gql('\n            mutation RemoveModelFromUseCase($input: RemoveModelFromUseCaseInput!) {\n              removeModelFromUseCase(input: $input)\n            }\n            ')
+        variables: Dict[str, object] = {'input': input}
+        response = self.execute(query=query, operation_name='RemoveModelFromUseCase', variables=variables, **kwargs)
+        data = self.get_data(response)
+        return RemoveModelFromUseCase.model_validate(data)
+
+    def deploy_model(self, input: DeployModelInput, **kwargs: Any) -> DeployModel:
+        query = gql('\n            mutation DeployModel($input: DeployModelInput!) {\n              deployModel(input: $input) {\n                ...ModelServiceData\n              }\n            }\n\n            fragment ModelData on Model {\n              id\n              key\n              name\n              online\n              error\n              isExternal\n              providerName\n              isAdapter\n              isTraining\n              createdAt\n              size\n              computeConfig {\n                tp\n                kvCacheLen\n                maxSeqLen\n              }\n            }\n\n            fragment ModelServiceData on ModelService {\n              id\n              key\n              name\n              model {\n                ...ModelData\n                backbone {\n                  ...ModelData\n                }\n              }\n              isDefault\n              desiredOnline\n              createdAt\n            }\n            ')
+        variables: Dict[str, object] = {'input': input}
+        response = self.execute(query=query, operation_name='DeployModel', variables=variables, **kwargs)
+        data = self.get_data(response)
+        return DeployModel.model_validate(data)
+
     def list_datasets(self, input: str, **kwargs: Any) -> ListDatasets:
         query = gql('\n            query ListDatasets($input: IdOrKey!) {\n              datasets(useCase: $input) {\n                ...DatasetData\n              }\n            }\n\n            fragment DatasetData on Dataset {\n              id\n              key\n              name\n              createdAt\n              kind\n              records\n              metricsUsage {\n                feedbackCount\n                comparisonCount\n                metric {\n                  ...MetricData\n                }\n              }\n            }\n\n            fragment MetricData on Metric {\n              id\n              key\n              name\n              kind\n              description\n              scoringType\n              createdAt\n              hasDirectFeedbacks\n              hasComparisonFeedbacks\n            }\n            ')
         variables: Dict[str, object] = {'input': input}
@@ -388,14 +396,14 @@ class GQLClient(BaseClientOpenTelemetry):
         return DescribeDataset.model_validate(data)
 
     def describe_use_case(self, input: str, **kwargs: Any) -> DescribeUseCase:
-        query = gql('\n            query DescribeUseCase($input: IdOrKey!) {\n              useCase(idOrKey: $input) {\n                ...UseCaseData\n              }\n            }\n\n            fragment MetricWithContextData on MetricWithContext {\n              id\n              key\n              name\n              kind\n              description\n              scoringType\n              createdAt\n            }\n\n            fragment ModelData on Model {\n              id\n              key\n              name\n              online\n              error\n              isExternal\n              providerName\n              isAdapter\n              isTraining\n              createdAt\n              size\n              computeConfig {\n                tp\n                kvCacheLen\n                maxSeqLen\n              }\n            }\n\n            fragment ModelServiceData on ModelService {\n              id\n              key\n              name\n              model {\n                ...ModelData\n                backbone {\n                  ...ModelData\n                }\n              }\n              attached\n              isDefault\n              desiredOnline\n              createdAt\n            }\n\n            fragment UseCaseData on UseCase {\n              id\n              key\n              name\n              description\n              createdAt\n              metrics {\n                ...MetricWithContextData\n              }\n              modelServices {\n                ...ModelServiceData\n              }\n              permissions\n              shares {\n                team {\n                  id\n                  key\n                  name\n                  createdAt\n                }\n                role {\n                  id\n                  key\n                  name\n                  createdAt\n                  permissions\n                }\n                isOwner\n              }\n            }\n            ')
+        query = gql('\n            query DescribeUseCase($input: IdOrKey!) {\n              useCase(idOrKey: $input) {\n                ...UseCaseData\n              }\n            }\n\n            fragment MetricWithContextData on MetricWithContext {\n              id\n              key\n              name\n              kind\n              description\n              scoringType\n              createdAt\n            }\n\n            fragment ModelData on Model {\n              id\n              key\n              name\n              online\n              error\n              isExternal\n              providerName\n              isAdapter\n              isTraining\n              createdAt\n              size\n              computeConfig {\n                tp\n                kvCacheLen\n                maxSeqLen\n              }\n            }\n\n            fragment ModelServiceData on ModelService {\n              id\n              key\n              name\n              model {\n                ...ModelData\n                backbone {\n                  ...ModelData\n                }\n              }\n              isDefault\n              desiredOnline\n              createdAt\n            }\n\n            fragment UseCaseData on UseCase {\n              id\n              key\n              name\n              description\n              createdAt\n              metrics {\n                ...MetricWithContextData\n              }\n              modelServices {\n                ...ModelServiceData\n              }\n              permissions\n              shares {\n                team {\n                  id\n                  key\n                  name\n                  createdAt\n                }\n                role {\n                  id\n                  key\n                  name\n                  createdAt\n                  permissions\n                }\n                isOwner\n              }\n            }\n            ')
         variables: Dict[str, object] = {'input': input}
         response = self.execute(query=query, operation_name='DescribeUseCase', variables=variables, **kwargs)
         data = self.get_data(response)
         return DescribeUseCase.model_validate(data)
 
     def list_use_cases(self, **kwargs: Any) -> ListUseCases:
-        query = gql('\n            query ListUseCases {\n              useCases {\n                ...UseCaseData\n              }\n            }\n\n            fragment MetricWithContextData on MetricWithContext {\n              id\n              key\n              name\n              kind\n              description\n              scoringType\n              createdAt\n            }\n\n            fragment ModelData on Model {\n              id\n              key\n              name\n              online\n              error\n              isExternal\n              providerName\n              isAdapter\n              isTraining\n              createdAt\n              size\n              computeConfig {\n                tp\n                kvCacheLen\n                maxSeqLen\n              }\n            }\n\n            fragment ModelServiceData on ModelService {\n              id\n              key\n              name\n              model {\n                ...ModelData\n                backbone {\n                  ...ModelData\n                }\n              }\n              attached\n              isDefault\n              desiredOnline\n              createdAt\n            }\n\n            fragment UseCaseData on UseCase {\n              id\n              key\n              name\n              description\n              createdAt\n              metrics {\n                ...MetricWithContextData\n              }\n              modelServices {\n                ...ModelServiceData\n              }\n              permissions\n              shares {\n                team {\n                  id\n                  key\n                  name\n                  createdAt\n                }\n                role {\n                  id\n                  key\n                  name\n                  createdAt\n                  permissions\n                }\n                isOwner\n              }\n            }\n            ')
+        query = gql('\n            query ListUseCases {\n              useCases {\n                ...UseCaseData\n              }\n            }\n\n            fragment MetricWithContextData on MetricWithContext {\n              id\n              key\n              name\n              kind\n              description\n              scoringType\n              createdAt\n            }\n\n            fragment ModelData on Model {\n              id\n              key\n              name\n              online\n              error\n              isExternal\n              providerName\n              isAdapter\n              isTraining\n              createdAt\n              size\n              computeConfig {\n                tp\n                kvCacheLen\n                maxSeqLen\n              }\n            }\n\n            fragment ModelServiceData on ModelService {\n              id\n              key\n              name\n              model {\n                ...ModelData\n                backbone {\n                  ...ModelData\n                }\n              }\n              isDefault\n              desiredOnline\n              createdAt\n            }\n\n            fragment UseCaseData on UseCase {\n              id\n              key\n              name\n              description\n              createdAt\n              metrics {\n                ...MetricWithContextData\n              }\n              modelServices {\n                ...ModelServiceData\n              }\n              permissions\n              shares {\n                team {\n                  id\n                  key\n                  name\n                  createdAt\n                }\n                role {\n                  id\n                  key\n                  name\n                  createdAt\n                  permissions\n                }\n                isOwner\n              }\n            }\n            ')
         variables: Dict[str, object] = {}
         response = self.execute(query=query, operation_name='ListUseCases', variables=variables, **kwargs)
         data = self.get_data(response)
@@ -584,14 +592,14 @@ class GQLClient(BaseClientOpenTelemetry):
         return ListGraders.model_validate(data)
 
     def list_jobs(self, page: CursorPageInput, filter: Union[Optional[ListJobsFilterInput], UnsetType]=UNSET, order: Union[Optional[List[OrderPair]], UnsetType]=UNSET, **kwargs: Any) -> ListJobs:
-        query = gql('\n            query ListJobs($page: CursorPageInput!, $filter: ListJobsFilterInput, $order: [OrderPair!]) {\n              jobs(page: $page, filter: $filter, order: $order) {\n                totalCount\n                pageInfo {\n                  hasNextPage\n                  hasPreviousPage\n                  startCursor\n                  endCursor\n                }\n                nodes {\n                  ...JobData\n                }\n              }\n            }\n\n            fragment CustomRecipeData on CustomRecipe {\n              id\n              key\n              name\n              content\n              contentHash\n              editable\n              global\n              builtin\n              inputSchema\n              jsonSchema\n              description\n              labels {\n                key\n                value\n              }\n              createdAt\n              updatedAt\n              createdBy {\n                id\n                name\n                email\n              }\n            }\n\n            fragment JobData on Job {\n              id\n              name\n              status\n              createdAt\n              createdBy {\n                id\n                name\n              }\n              startedAt\n              endedAt\n              durationMs\n              progress\n              error\n              kind\n              stages {\n                name\n                status\n                info {\n                  __typename\n                  ... on TrainingJobStageOutput {\n                    monitoringLink\n                    totalNumSamples\n                    processedNumSamples\n                    checkpoints\n                  }\n                  ... on EvalJobStageOutput {\n                    totalNumSamples\n                    processedNumSamples\n                  }\n                  ... on BatchInferenceJobStageOutput {\n                    totalNumSamples\n                    processedNumSamples\n                  }\n                }\n              }\n              useCase {\n                id\n                key\n                name\n              }\n              recipe {\n                ...CustomRecipeData\n              }\n              details {\n                args\n                recipeHash\n                artifacts {\n                  id\n                  name\n                  kind\n                  uri\n                  metadata\n                  createdAt\n                  byproducts {\n                    __typename\n                    ... on EvaluationByproducts {\n                      evalResults {\n                        mean\n                        min\n                        max\n                        stddev\n                        count\n                        sum\n                        feedbackCount\n                        jobId\n                        artifactId\n                        modelService {\n                          key\n                          name\n                        }\n                        metric {\n                          key\n                          name\n                        }\n                      }\n                    }\n                  }\n                }\n              }\n            }\n            ')
+        query = gql('\n            query ListJobs($page: CursorPageInput!, $filter: ListJobsFilterInput, $order: [OrderPair!]) {\n              jobs(page: $page, filter: $filter, order: $order) {\n                totalCount\n                pageInfo {\n                  hasNextPage\n                  hasPreviousPage\n                  startCursor\n                  endCursor\n                }\n                nodes {\n                  ...JobData\n                }\n              }\n            }\n\n            fragment CustomRecipeData on CustomRecipe {\n              id\n              key\n              name\n              content\n              contentHash\n              editable\n              global\n              builtin\n              inputSchema\n              jsonSchema\n              description\n              labels {\n                key\n                value\n              }\n              createdAt\n              updatedAt\n              createdBy {\n                id\n                name\n                email\n              }\n            }\n\n            fragment JobData on Job {\n              id\n              name\n              status\n              createdAt\n              createdBy {\n                id\n                name\n              }\n              startedAt\n              endedAt\n              durationMs\n              progress\n              error\n              kind\n              stages {\n                name\n                status\n                info {\n                  __typename\n                  ... on TrainingJobStageOutput {\n                    monitoringLink\n                    totalNumSamples\n                    processedNumSamples\n                    checkpoints\n                  }\n                  ... on EvalJobStageOutput {\n                    totalNumSamples\n                    processedNumSamples\n                  }\n                  ... on BatchInferenceJobStageOutput {\n                    totalNumSamples\n                    processedNumSamples\n                  }\n                }\n              }\n              useCase {\n                id\n                key\n                name\n              }\n              recipe {\n                ...CustomRecipeData\n              }\n              details {\n                args\n                recipeHash\n                artifacts {\n                  id\n                  name\n                  kind\n                  status\n                  uri\n                  metadata\n                  createdAt\n                  byproducts {\n                    __typename\n                    ... on EvaluationByproducts {\n                      evalResults {\n                        mean\n                        min\n                        max\n                        stddev\n                        count\n                        sum\n                        feedbackCount\n                        jobId\n                        artifactId\n                        modelService {\n                          key\n                          name\n                        }\n                        metric {\n                          key\n                          name\n                        }\n                      }\n                    }\n                  }\n                }\n              }\n            }\n            ')
         variables: Dict[str, object] = {'page': page, 'filter': filter, 'order': order}
         response = self.execute(query=query, operation_name='ListJobs', variables=variables, **kwargs)
         data = self.get_data(response)
         return ListJobs.model_validate(data)
 
     def describe_job(self, id: Any, **kwargs: Any) -> DescribeJob:
-        query = gql('\n            query DescribeJob($id: UUID!) {\n              job(id: $id) {\n                ...JobData\n              }\n            }\n\n            fragment CustomRecipeData on CustomRecipe {\n              id\n              key\n              name\n              content\n              contentHash\n              editable\n              global\n              builtin\n              inputSchema\n              jsonSchema\n              description\n              labels {\n                key\n                value\n              }\n              createdAt\n              updatedAt\n              createdBy {\n                id\n                name\n                email\n              }\n            }\n\n            fragment JobData on Job {\n              id\n              name\n              status\n              createdAt\n              createdBy {\n                id\n                name\n              }\n              startedAt\n              endedAt\n              durationMs\n              progress\n              error\n              kind\n              stages {\n                name\n                status\n                info {\n                  __typename\n                  ... on TrainingJobStageOutput {\n                    monitoringLink\n                    totalNumSamples\n                    processedNumSamples\n                    checkpoints\n                  }\n                  ... on EvalJobStageOutput {\n                    totalNumSamples\n                    processedNumSamples\n                  }\n                  ... on BatchInferenceJobStageOutput {\n                    totalNumSamples\n                    processedNumSamples\n                  }\n                }\n              }\n              useCase {\n                id\n                key\n                name\n              }\n              recipe {\n                ...CustomRecipeData\n              }\n              details {\n                args\n                recipeHash\n                artifacts {\n                  id\n                  name\n                  kind\n                  uri\n                  metadata\n                  createdAt\n                  byproducts {\n                    __typename\n                    ... on EvaluationByproducts {\n                      evalResults {\n                        mean\n                        min\n                        max\n                        stddev\n                        count\n                        sum\n                        feedbackCount\n                        jobId\n                        artifactId\n                        modelService {\n                          key\n                          name\n                        }\n                        metric {\n                          key\n                          name\n                        }\n                      }\n                    }\n                  }\n                }\n              }\n            }\n            ')
+        query = gql('\n            query DescribeJob($id: UUID!) {\n              job(id: $id) {\n                ...JobData\n              }\n            }\n\n            fragment CustomRecipeData on CustomRecipe {\n              id\n              key\n              name\n              content\n              contentHash\n              editable\n              global\n              builtin\n              inputSchema\n              jsonSchema\n              description\n              labels {\n                key\n                value\n              }\n              createdAt\n              updatedAt\n              createdBy {\n                id\n                name\n                email\n              }\n            }\n\n            fragment JobData on Job {\n              id\n              name\n              status\n              createdAt\n              createdBy {\n                id\n                name\n              }\n              startedAt\n              endedAt\n              durationMs\n              progress\n              error\n              kind\n              stages {\n                name\n                status\n                info {\n                  __typename\n                  ... on TrainingJobStageOutput {\n                    monitoringLink\n                    totalNumSamples\n                    processedNumSamples\n                    checkpoints\n                  }\n                  ... on EvalJobStageOutput {\n                    totalNumSamples\n                    processedNumSamples\n                  }\n                  ... on BatchInferenceJobStageOutput {\n                    totalNumSamples\n                    processedNumSamples\n                  }\n                }\n              }\n              useCase {\n                id\n                key\n                name\n              }\n              recipe {\n                ...CustomRecipeData\n              }\n              details {\n                args\n                recipeHash\n                artifacts {\n                  id\n                  name\n                  kind\n                  status\n                  uri\n                  metadata\n                  createdAt\n                  byproducts {\n                    __typename\n                    ... on EvaluationByproducts {\n                      evalResults {\n                        mean\n                        min\n                        max\n                        stddev\n                        count\n                        sum\n                        feedbackCount\n                        jobId\n                        artifactId\n                        modelService {\n                          key\n                          name\n                        }\n                        metric {\n                          key\n                          name\n                        }\n                      }\n                    }\n                  }\n                }\n              }\n            }\n            ')
         variables: Dict[str, object] = {'id': id}
         response = self.execute(query=query, operation_name='DescribeJob', variables=variables, **kwargs)
         data = self.get_data(response)

adaptive_sdk/graphql_client/custom_fields.py

@@ -1,7 +1,7 @@
 from typing import Any, Dict, List, Optional, Union
 from .base_operation import GraphQLField
 from .custom_typing_fields import AbcampaignGraphQLField, AbReportGraphQLField, AbVariantReportComparisonGraphQLField, AbVariantReportGraphQLField, ActivityGraphQLField, ActivityOutputGraphQLField, ApiKeyGraphQLField, ArtifactByproductsUnion, AuthProviderGraphQLField, BatchInferenceJobStageOutputGraphQLField, BillingUsageGraphQLField, ChatMessageGraphQLField, ComparisonFeedbackGraphQLField, CompletionConnectionGraphQLField, CompletionEdgeGraphQLField, CompletionGraphQLField, CompletionGroupDataConnectionGraphQLField, CompletionGroupDataEdgeGraphQLField, CompletionGroupDataGraphQLField, CompletionGroupFeedbackStatsGraphQLField, CompletionHistoryEntryOuputGraphQLField, CompletionLabelGraphQLField, CompletionMetadataGraphQLField, ComputePoolGraphQLField, ContractGraphQLField, CustomConfigOutputGraphQLField, CustomRecipeGraphQLField, CustomRecipeJobDetailsGraphQLField, DatasetByproductsGraphQLField, DatasetGraphQLField, DatasetMetricUsageGraphQLField, DatasetProgressGraphQLField, DatasetUploadProcessingStatusGraphQLField, DatasetValidationOutputGraphQLField, DeleteConfirmGraphQLField, DirectFeedbackGraphQLField, EmojiGraphQLField, EvalJobStageOutputGraphQLField, EvaluationByproductsGraphQLField, EvaluationResultGraphQLField, GlobalUsageGraphQLField, GpuAllocationGraphQLField, GraderConfigUnion, GraderGraphQLField, HarmonyGroupGraphQLField, InteractionOutputGraphQLField, IntervalGraphQLField, JobArtifactGraphQLField, JobConnectionGraphQLField, JobEdgeGraphQLField, JobGraphQLField, JobStageInfoOutputUnion, JobStageOutputGraphQLField, JudgeConfigOutputGraphQLField, JudgeExampleGraphQLField, JudgeGraphQLField, LabelGraphQLField, LabelKeyUsageGraphQLField, LabelUsageGraphQLField, LabelValueUsageGraphQLField, MetaObjectGraphQLField, MetricActivityGraphQLField, MetricGraphQLField, MetricWithContextGraphQLField, ModelByproductsGraphQLField, ModelComputeConfigOutputGraphQLField, ModelGraphQLField, ModelPlacementOutputGraphQLField, ModelServiceGraphQLField, PageInfoGraphQLField, PrebuiltConfigDefinitionGraphQLField, PrebuiltConfigOutputGraphQLField, PrebuiltCriteriaGraphQLField, ProviderListGraphQLField, RemoteConfigOutputGraphQLField, RemoteEnvGraphQLField, RemoteEnvTestOfflineGraphQLField, RemoteEnvTestOnlineGraphQLField, RoleGraphQLField, SearchResultGraphQLField, SessionGraphQLField, SettingsGraphQLField, ShareGraphQLField, SystemPromptTemplateGraphQLField, TeamGraphQLField, TeamMemberGraphQLField, TeamWithroleGraphQLField, TimeseriesGraphQLField, ToolProviderGraphQLField, TrainingJobStageOutputGraphQLField, TrendResultGraphQLField, UnitConfigGraphQLField, UsageAggregateItemGraphQLField, UsageAggregatePerUseCaseItemGraphQLField, UsageGraphQLField, UsageStatsByModelGraphQLField, UsageStatsGraphQLField, UseCaseGraphQLField, UseCaseItemGraphQLField, UseCaseMetadataGraphQLField, UserGraphQLField, WidgetGraphQLField
-from .input_types import AbCampaignFilter, ArtifactFilter, CursorPageInput, FeedbackFilterInput, ListCompletionsFilterInput, MetricTrendInput, ModelServiceFilter, OrderPair, TimeRange, TimeseriesInput, UseCaseFilter
+from .input_types import AbCampaignFilter, ArtifactFilter, CursorPageInput, FeedbackFilterInput, ListCompletionsFilterInput, MetricTrendInput, ModelFilter, ModelServiceFilter, OrderPair, TimeRange, TimeseriesInput, UseCaseFilter
 
 class AbReportFields(GraphQLField):
     """@private"""
@@ -629,6 +629,7 @@ class CustomRecipeJobDetailsFields(GraphQLField):
         return JobArtifactFields('artifacts')
     num_gpus: 'CustomRecipeJobDetailsGraphQLField' = CustomRecipeJobDetailsGraphQLField('numGpus')
     gpu_duration_ms: 'CustomRecipeJobDetailsGraphQLField' = CustomRecipeJobDetailsGraphQLField('gpuDurationMs')
+    compute_pool_id: 'CustomRecipeJobDetailsGraphQLField' = CustomRecipeJobDetailsGraphQLField('computePoolId')
 
     def fields(self, *subfields: Union[CustomRecipeJobDetailsGraphQLField, 'JobArtifactFields']) -> 'CustomRecipeJobDetailsFields':
         """Subfields should come from the CustomRecipeJobDetailsFields class"""
@@ -658,8 +659,13 @@ class DatasetFields(GraphQLField):
     @classmethod
     def progress(cls) -> 'DatasetProgressFields':
         return DatasetProgressFields('progress')
+    download_url: 'DatasetGraphQLField' = DatasetGraphQLField('downloadUrl')
 
-    def fields(self, *subfields: Union[DatasetGraphQLField, 'DatasetMetricUsageFields', 'DatasetProgressFields']) -> 'DatasetFields':
+    @classmethod
+    def use_case(cls) -> 'UseCaseFields':
+        return UseCaseFields('useCase')
+
+    def fields(self, *subfields: Union[DatasetGraphQLField, 'DatasetMetricUsageFields', 'DatasetProgressFields', 'UseCaseFields']) -> 'DatasetFields':
         """Subfields should come from the DatasetFields class"""
         self._subfields.extend(subfields)
         return self
@@ -1447,6 +1453,8 @@ class ModelFields(GraphQLField):
         return ModelFields('parent')
     is_training: 'ModelGraphQLField' = ModelGraphQLField('isTraining')
     'indicates if a training is pending or running for this model'
+    is_published: 'ModelGraphQLField' = ModelGraphQLField('isPublished')
+    is_stable: 'ModelGraphQLField' = ModelGraphQLField('isStable')
     capabilities: 'ModelGraphQLField' = ModelGraphQLField('capabilities')
     supported_tp: 'ModelGraphQLField' = ModelGraphQLField('supportedTp')
     family: 'ModelGraphQLField' = ModelGraphQLField('family')
@@ -1524,9 +1532,10 @@ class ModelServiceFields(GraphQLField):
     @classmethod
     def model(cls) -> 'ModelFields':
         return ModelFields('model')
-    attached: 'ModelServiceGraphQLField' = ModelServiceGraphQLField('attached')
     is_default: 'ModelServiceGraphQLField' = ModelServiceGraphQLField('isDefault')
     desired_online: 'ModelServiceGraphQLField' = ModelServiceGraphQLField('desiredOnline')
+    deleted: 'ModelServiceGraphQLField' = ModelServiceGraphQLField('deleted')
+    'Whether or not this model service has been deleted.'
 
     @classmethod
     def activity(cls, *, timerange: Optional[TimeRange]=None) -> 'ActivityFields':
@@ -2092,6 +2101,13 @@ class UseCaseFields(GraphQLField):
         cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
         return ModelServiceFields('modelService', arguments=cleared_arguments)
 
+    @classmethod
+    def models(cls, *, filter: Optional[ModelFilter]=None) -> 'ModelFields':
+        """Returns models associated with this use case."""
+        arguments: Dict[str, Dict[str, Any]] = {'filter': {'type': 'ModelFilter', 'value': filter}}
+        cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
+        return ModelFields('models', arguments=cleared_arguments)
+
     @classmethod
     def default_model_service(cls) -> 'ModelServiceFields':
         return ModelServiceFields('defaultModelService')
@@ -2143,7 +2159,7 @@ class UseCaseFields(GraphQLField):
     def tool_providers(cls) -> 'ToolProviderFields':
         return ToolProviderFields('toolProviders')
 
-    def fields(self, *subfields: Union[UseCaseGraphQLField, 'AbcampaignFields', 'ActivityFields', 'LabelUsageFields', 'MetricWithContextFields', 'ModelServiceFields', 'SettingsFields', 'ShareFields', 'ToolProviderFields', 'UseCaseMetadataFields', 'WidgetFields']) -> 'UseCaseFields':
+    def fields(self, *subfields: Union[UseCaseGraphQLField, 'AbcampaignFields', 'ActivityFields', 'LabelUsageFields', 'MetricWithContextFields', 'ModelFields', 'ModelServiceFields', 'SettingsFields', 'ShareFields', 'ToolProviderFields', 'UseCaseMetadataFields', 'WidgetFields']) -> 'UseCaseFields':
         """Subfields should come from the UseCaseFields class"""
         self._subfields.extend(subfields)
         return self

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, DatasetUploadProcessingStatusFields, 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, CancelAllocationInput, CreateRecipeInput, CreateToolProviderInput, DatasetCreate, DatasetCreateFromFilters, DatasetCreateFromMultipartUpload, 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
+from .input_types import AbcampaignCreate, AddExternalModelInput, AddHFModelInput, AddModelInput, AddModelToUseCaseInput, ApiKeyCreate, CancelAllocationInput, CreateRecipeInput, CreateToolProviderInput, DatasetCreate, DatasetCreateFromFilters, DatasetCreateFromMultipartUpload, DeleteModelInput, DeployModelInput, FeedbackAddInput, FeedbackUpdateInput, GraderCreateInput, GraderUpdateInput, JobInput, JudgeCreate, JudgeUpdate, MetricCreate, MetricLink, MetricUnlink, ModelComputeConfigInput, PrebuiltJudgeCreate, RemoteEnvCreate, RemoveModelFromUseCaseInput, ResizePartitionInput, RoleCreate, SystemPromptTemplateCreate, SystemPromptTemplateUpdate, TeamCreate, TeamMemberRemove, TeamMemberSet, UpdateCompletion, UpdateModelInput, UpdateModelService, UpdateRecipeInput, UpdateToolProviderInput, UseCaseCreate, UseCaseShares, UseCaseUpdate, UserCreate
 
 class Mutation:
     """@private"""
@@ -123,16 +123,16 @@ class Mutation:
         return GraphQLField(field_name='unlinkMetric', arguments=cleared_arguments)
 
     @classmethod
-    def attach_model(cls, input: AttachModel) -> ModelServiceFields:
-        arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'AttachModel!', 'value': input}}
+    def deploy_model(cls, input: DeployModelInput) -> ModelServiceFields:
+        arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'DeployModelInput!', 'value': input}}
         cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
-        return ModelServiceFields(field_name='attachModel', arguments=cleared_arguments)
+        return ModelServiceFields(field_name='deployModel', arguments=cleared_arguments)
 
     @classmethod
-    def disconnect_model(cls, input: ModelServiceDisconnect) -> GraphQLField:
-        arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'ModelServiceDisconnect!', 'value': input}}
+    def update_model(cls, input: UpdateModelInput) -> ModelFields:
+        arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'UpdateModelInput!', 'value': input}}
         cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
-        return GraphQLField(field_name='disconnectModel', arguments=cleared_arguments)
+        return ModelFields(field_name='updateModel', arguments=cleared_arguments)
 
     @classmethod
     def update_model_service(cls, input: UpdateModelService) -> ModelServiceFields:
@@ -140,15 +140,9 @@ class Mutation:
         cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
         return ModelServiceFields(field_name='updateModelService', arguments=cleared_arguments)
 
-    @classmethod
-    def deploy_model(cls, id_or_key: str, wait: bool, *, placement: Optional[ModelPlacementInput]=None) -> GraphQLField:
-        arguments: Dict[str, Dict[str, Any]] = {'idOrKey': {'type': 'IdOrKey!', 'value': id_or_key}, 'wait': {'type': 'Boolean!', 'value': wait}, 'placement': {'type': 'ModelPlacementInput', 'value': placement}}
-        cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
-        return GraphQLField(field_name='deployModel', arguments=cleared_arguments)
-
     @classmethod
     def terminate_model(cls, id_or_key: str, force: bool) -> GraphQLField:
-        """If a model is used by several use cases with `attached = true`, you need to specify 'force = true' to be able to deactivate the model"""
+        """If a model is used by several use cases with `desiredOnline = true`, you need to specify 'force = true' to be able to deactivate the model"""
         arguments: Dict[str, Dict[str, Any]] = {'idOrKey': {'type': 'IdOrKey!', 'value': id_or_key}, 'force': {'type': 'Boolean!', 'value': force}}
         cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
         return GraphQLField(field_name='terminateModel', arguments=cleared_arguments)
@@ -177,6 +171,27 @@ class Mutation:
         cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
         return ModelFields(field_name='updateModelComputeConfig', arguments=cleared_arguments)
 
+    @classmethod
+    def add_model_to_use_case(cls, input: AddModelToUseCaseInput) -> GraphQLField:
+        arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'AddModelToUseCaseInput!', 'value': input}}
+        cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
+        return GraphQLField(field_name='addModelToUseCase', arguments=cleared_arguments)
+
+    @classmethod
+    def remove_model_from_use_case(cls, input: RemoveModelFromUseCaseInput) -> GraphQLField:
+        """Removes a model from a use case. If the model is not bound to any other use case or published organisation wide, it is deleted from storage."""
+        arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'RemoveModelFromUseCaseInput!', 'value': input}}
+        cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
+        return GraphQLField(field_name='removeModelFromUseCase', arguments=cleared_arguments)
+
+    @classmethod
+    def delete_model(cls, input: DeleteModelInput) -> GraphQLField:
+        """Deletes a model: removes from all use cases, unpublishes from org registry,
+        and deletes from storage."""
+        arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'DeleteModelInput!', 'value': input}}
+        cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
+        return GraphQLField(field_name='deleteModel', arguments=cleared_arguments)
+
     @classmethod
     def add_remote_env(cls, input: RemoteEnvCreate) -> RemoteEnvFields:
         arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'RemoteEnvCreate!', 'value': input}}

adaptive_sdk/graphql_client/deploy_model.py

@@ -1,6 +1,12 @@
 from pydantic import Field
 from .base_model import BaseModel
+from .fragments import ModelServiceData
 
 class DeployModel(BaseModel):
     """@public"""
-    deploy_model: str = Field(alias='deployModel')
+    deploy_model: 'DeployModelDeployModel' = Field(alias='deployModel')
+
+class DeployModelDeployModel(ModelServiceData):
+    """@public"""
+    pass
+DeployModel.model_rebuild()

adaptive_sdk/graphql_client/enums.py

@@ -67,6 +67,7 @@ class DateBucketUnit(str, Enum):
 class ExternalModelProviderName(str, Enum):
     """@public"""
     OPEN_AI = 'OPEN_AI'
+    LEGACY_OPEN_AI = 'LEGACY_OPEN_AI'
     GOOGLE = 'GOOGLE'
     ANTHROPIC = 'ANTHROPIC'
 
@@ -161,33 +162,14 @@ class ModelOnline(str, Enum):
     OFFLINE = 'OFFLINE'
     ERROR = 'ERROR'
 
-class ModelserviceStatus(str, Enum):
+class ModelServiceStatus(str, Enum):
     """@public"""
     PENDING = 'PENDING'
     ONLINE = 'ONLINE'
     OFFLINE = 'OFFLINE'
-    DETACHED = 'DETACHED'
     TURNED_OFF = 'TURNED_OFF'
     ERROR = 'ERROR'
 
-class OpenAIModel(str, Enum):
-    """@public"""
-    GPT41 = 'GPT41'
-    GPT4O = 'GPT4O'
-    GPT4O_MINI = 'GPT4O_MINI'
-    O1 = 'O1'
-    O1_MINI = 'O1_MINI'
-    O3_MINI = 'O3_MINI'
-    O4_MINI = 'O4_MINI'
-    O3 = 'O3'
-    GPT4 = 'GPT4'
-    GPT4_TURBO = 'GPT4_TURBO'
-    GPT3_5_TURBO = 'GPT3_5_TURBO'
-    GPT5 = 'GPT5'
-    GPT5_MINI = 'GPT5_MINI'
-    GPT5_NANO = 'GPT5_NANO'
-    GPT41_MINI = 'GPT41_MINI'
-
 class PrebuiltCriteriaKey(str, Enum):
     """@public"""
     FAITHFULNESS = 'FAITHFULNESS'
@@ -201,6 +183,7 @@ class Protocol(str, Enum):
 class ProviderName(str, Enum):
     """@public"""
     OPEN_AI = 'OPEN_AI'
+    LEGACY_OPEN_AI = 'LEGACY_OPEN_AI'
     HARMONY = 'HARMONY'
     GOOGLE = 'GOOGLE'
     ANTHROPIC = 'ANTHROPIC'

adaptive_sdk/graphql_client/fragments.py

@@ -1,7 +1,7 @@
 from typing import Annotated, Any, List, Literal, Optional, Union
 from pydantic import Field
 from .base_model import BaseModel
-from .enums import AbcampaignStatus, CompletionSource, DatasetKind, FeedbackType, GraderTypeEnum, HarmonyStatus, JobArtifactKind, JobKind, JobStatus, JobStatusOutput, JudgeCapability, MetricKind, MetricScoringType, ModelOnline, PrebuiltCriteriaKey, ProviderName, RemoteEnvStatus
+from .enums import AbcampaignStatus, CompletionSource, DatasetKind, FeedbackType, GraderTypeEnum, HarmonyStatus, JobArtifactKind, JobArtifactStatus, JobKind, JobStatus, JobStatusOutput, JudgeCapability, MetricKind, MetricScoringType, ModelOnline, PrebuiltCriteriaKey, ProviderName, RemoteEnvStatus
 
 class AbCampaignCreateData(BaseModel):
     """@public"""
@@ -246,7 +246,7 @@ class ModelData(BaseModel):
     is_adapter: bool = Field(alias='isAdapter')
     is_training: bool = Field(alias='isTraining')
     created_at: int = Field(alias='createdAt')
-    size: Optional[str]
+    size: Optional[int]
     compute_config: Optional['ModelDataComputeConfig'] = Field(alias='computeConfig')
 
 class ModelDataComputeConfig(BaseModel):
@@ -431,6 +431,7 @@ class JobDataDetailsArtifacts(BaseModel):
     id: Any
     name: str
     kind: JobArtifactKind
+    status: JobArtifactStatus
     uri: Optional[str]
     metadata: Any
     created_at: int = Field(alias='createdAt')
@@ -584,7 +585,7 @@ class ModelDataAdmin(BaseModel):
     is_adapter: bool = Field(alias='isAdapter')
     is_training: bool = Field(alias='isTraining')
     created_at: int = Field(alias='createdAt')
-    size: Optional[str]
+    size: Optional[int]
 
 class ModelDataAdminUseCases(BaseModel):
     """@public"""
@@ -598,7 +599,6 @@ class ModelServiceData(BaseModel):
     key: str
     name: str
     model: 'ModelServiceDataModel'
-    attached: bool
     is_default: bool = Field(alias='isDefault')
     desired_online: bool = Field(alias='desiredOnline')
     created_at: int = Field(alias='createdAt')