adaptive-sdk 0.1.11__tar.gz → 0.1.13__tar.gz

{adaptive_sdk-0.1.11 → adaptive_sdk-0.1.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: adaptive-sdk
-Version: 0.1.11
+Version: 0.1.13
 Summary: Python SDK for Adaptive Engine
 Author-email: Vincent Debergue <vincent@adaptive-ml.com>, Joao Moura <joao@adaptive-ml.com>, Yacine Bouraoui <yacine@adaptive-ml.com>
 Requires-Python: >=3.10
@@ -10,7 +10,7 @@ Requires-Dist: httpx-sse >= 0.4.0
 Requires-Dist: gql >= 3.5.0
 Requires-Dist: pydantic[email] >= 2.9.0
 Requires-Dist: pyhumps >= 3.8.0
-Requires-Dist: fastapi >= 0.121.1
+Requires-Dist: fastapi >= 0.121.2
 Requires-Dist: uvicorn >= 0.34.0
 Requires-Dist: jsonschema==4.24.0
 Requires-Dist: websockets==15.0.1

{adaptive_sdk-0.1.11 → adaptive_sdk-0.1.13}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "flit_core.buildapi"
 
 [project]
 name = "adaptive-sdk"
-version = "0.1.11"
+version = "0.1.13"
 description = "Python SDK for Adaptive Engine"
 authors = [
   { name = "Vincent Debergue", email = "vincent@adaptive-ml.com" },
@@ -20,7 +20,7 @@ dependencies = [
   "gql >= 3.5.0",
   "pydantic[email] >= 2.9.0",
   "pyhumps >= 3.8.0",
-  "fastapi >= 0.121.1",
+  "fastapi >= 0.121.2",
   "uvicorn >= 0.34.0",
   "jsonschema==4.24.0",
   "websockets==15.0.1",
@@ -77,4 +77,4 @@ allow_redefinition = true
 required-version = '25.1.0'
 
 [tool.uv]
-constraint-dependencies = ["graphql-core==3.2.6"]
+constraint-dependencies = ["graphql-core==3.2.6", "starlette==0.49.1"]

{adaptive_sdk-0.1.11 → adaptive_sdk-0.1.13}/src/adaptive_sdk/graphql_client/__init__.py

@@ -28,7 +28,6 @@ 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_interaction import DescribeInteraction, DescribeInteractionCompletion
@@ -38,7 +37,7 @@ from .describe_metric_admin import DescribeMetricAdmin, DescribeMetricAdminMetri
 from .describe_model import DescribeModel, DescribeModelModel, DescribeModelModelBackbone
 from .describe_model_admin import DescribeModelAdmin, DescribeModelAdminModel, DescribeModelAdminModelBackbone
 from .describe_use_case import DescribeUseCase, DescribeUseCaseUseCase
-from .enums import AbcampaignStatus, AuthProviderKind, BillingCycle, CompletionGroupBy, CompletionSource, ComputePoolCapability, DatasetKind, DatasetSource, DatasetStatus, DateBucketUnit, ExternalModelProviderName, FeedbackType, GraderTypeEnum, HarmonyStatus, JobArtifactKind, JobKind, JobStatus, JobStatusOutput, JudgeCapability, MetricAggregation, MetricKind, MetricScoringType, ModelCapabilityFilter, ModelOnline, ModelserviceStatus, OpenAIModel, PrebuiltCriteriaKey, Protocol, ProviderName, RemoteEnvStatus, SelectionTypeInput, SessionStatus, SortDirection, TimeseriesInterval, UnitPosition
+from .enums import AbcampaignStatus, AuthProviderKind, BillingCycle, CompletionGroupBy, CompletionSource, ComputePoolCapability, DatasetKind, DatasetSource, DatasetStatus, DateBucketUnit, ExternalModelProviderName, FeedbackType, GraderTypeEnum, HarmonyStatus, JobArtifactKind, JobArtifactStatus, JobKind, JobStatus, JobStatusOutput, JudgeCapability, MetricAggregation, MetricKind, MetricScoringType, ModelCapabilityFilter, ModelOnline, ModelserviceStatus, OpenAIModel, PrebuiltCriteriaKey, Protocol, ProviderName, RemoteEnvStatus, SelectionTypeInput, SessionStatus, SortDirection, TimeseriesInterval, 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, CustomRecipeData, CustomRecipeDataCreatedBy, CustomRecipeDataLabels, DatasetData, DatasetDataMetricsUsage, DatasetDataMetricsUsageMetric, GraderData, GraderDataGraderConfigCustomConfigOutput, GraderDataGraderConfigJudgeConfigOutput, GraderDataGraderConfigJudgeConfigOutputExamples, GraderDataGraderConfigJudgeConfigOutputExamplesInput, GraderDataGraderConfigJudgeConfigOutputModel, GraderDataGraderConfigPrebuiltConfigOutput, GraderDataGraderConfigPrebuiltConfigOutputModel, GraderDataGraderConfigPrebuiltConfigOutputPrebuiltCriteria, GraderDataGraderConfigRemoteConfigOutput, GraderDataMetric, GraderDataUseCase, HarmonyGroupData, HarmonyGroupDataComputePool, HarmonyGroupDataGpuAllocations, HarmonyGroupDataOnlineModels, JobData, JobDataCreatedBy, JobDataDetails, JobDataDetailsArtifacts, JobDataDetailsArtifactsByproductsDatasetByproducts, JobDataDetailsArtifactsByproductsEvaluationByproducts, JobDataDetailsArtifactsByproductsEvaluationByproductsEvalResults, JobDataDetailsArtifactsByproductsEvaluationByproductsEvalResultsMetric, JobDataDetailsArtifactsByproductsEvaluationByproductsEvalResultsModelService, JobDataDetailsArtifactsByproductsModelByproducts, JobDataRecipe, JobDataStages, JobDataStagesInfoBatchInferenceJobStageOutput, JobDataStagesInfoEvalJobStageOutput, JobDataStagesInfoTrainingJobStageOutput, JobDataUseCase, JobStageOutputData, JobStageOutputDataInfoBatchInferenceJobStageOutput, JobStageOutputDataInfoEvalJobStageOutput, JobStageOutputDataInfoTrainingJobStageOutput, JudgeData, JudgeDataExamples, JudgeDataExamplesInput, JudgeDataMetric, JudgeDataModel, 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
@@ -67,7 +66,7 @@ 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 .me import Me, MeMe
 from .remove_remote_env import RemoveRemoteEnv
 from .remove_team_member import RemoveTeamMember, RemoveTeamMemberRemoveTeamMember
 from .resize_inference_partition import ResizeInferencePartition
@@ -81,4 +80,4 @@ 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', 'AddExternalModel', 'AddExternalModelAddExternalModel', 'AddExternalModelInput', 'AddHFModel', 'AddHFModelImportHfModel', 'AddHFModelInput', 'AddModel', 'AddModelAddModel', 'AddModelAddModelBackbone', 'AddModelInput', 'AddRemoteEnv', 'AddRemoteEnvAddRemoteEnv', 'AnthropicProviderDataInput', 'ApiKeyCreate', 'ArtifactFilter', 'AsyncBaseClientOpenTelemetry', 'AsyncGQLClient', 'AttachModel', 'AttachModelToUseCase', 'AttachModelToUseCaseAttachModel', 'AuthProviderKind', 'BaseClientOpenTelemetry', 'BaseModel', 'BillingCycle', 'CancelABCampaign', 'CancelAllocationInput', 'CancelJob', 'CancelJobCancelJob', 'CapabilityFilter', 'CompletionComparisonFeedbackData', 'CompletionComparisonFeedbackDataModel', 'CompletionComparisonFilterInput', 'CompletionData', 'CompletionDataChatMessages', 'CompletionDataComparisonFeedbacks', 'CompletionDataComparisonFeedbacksMetric', 'CompletionDataComparisonFeedbacksOtherCompletion', 'CompletionDataComparisonFeedbacksPreferedCompletion', 'CompletionDataComparisonFeedbacksUsecase', 'CompletionDataDirectFeedbacks', 'CompletionDataDirectFeedbacksMetric', 'CompletionDataLabels', 'CompletionDataMetadata', 'CompletionDataMetadataUsage', 'CompletionDataModel', 'CompletionFeedbackFilterInput', 'CompletionGroupBy', 'CompletionLabelValue', 'CompletionSource', 'CompletionsByFilters', 'CompletionsById', 'ComputePoolCapability', 'CreateAbCampaign', 'CreateAbCampaignCreateAbCampaign', 'CreateCustomRecipe', 'CreateCustomRecipeCreateCustomRecipe', 'CreateDatasetFromMultipartUpload', 'CreateDatasetFromMultipartUploadCreateDatasetFromMultipartUpload', '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', 'DatasetCompletionQuery', 'DatasetCreate', 'DatasetCreateFromFilters', 'DatasetCreateFromMultipartUpload', 'DatasetData', 'DatasetDataMetricsUsage', 'DatasetDataMetricsUsageMetric', 'DatasetKind', 'DatasetSource', 'DatasetStatus', 'DatasetUploadProcessingStatus', 'DatasetUploadProcessingStatusDatasetUploadProcessingStatus', 'DatasetUploadProcessingStatusInput', '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', 'ExternalModelProviderName', 'FeedbackAddInput', 'FeedbackFilterInput', 'FeedbackType', 'FeedbackUpdateInput', 'FromGroupsQuery', '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', 'GroupSelection', 'GroupSelectionQuery', 'HarmonyGroupData', 'HarmonyGroupDataComputePool', 'HarmonyGroupDataGpuAllocations', 'HarmonyGroupDataOnlineModels', 'HarmonyStatus', 'JobArtifactFilter', 'JobArtifactKind', 'JobData', 'JobDataCreatedBy', 'JobDataDetails', 'JobDataDetailsArtifacts', 'JobDataDetailsArtifactsByproductsDatasetByproducts', 'JobDataDetailsArtifactsByproductsEvaluationByproducts', 'JobDataDetailsArtifactsByproductsEvaluationByproductsEvalResults', 'JobDataDetailsArtifactsByproductsEvaluationByproductsEvalResultsMetric', 'JobDataDetailsArtifactsByproductsEvaluationByproductsEvalResultsModelService', 'JobDataDetailsArtifactsByproductsModelByproducts', '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', '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', 'ModelCapabilityFilter', 'ModelComputeConfigInput', 'ModelData', 'ModelDataAdmin', 'ModelDataAdminUseCases', 'ModelDataComputeConfig', 'ModelFilter', 'ModelOnline', 'ModelPlacementInput', 'ModelProviderDataInput', 'ModelServiceData', 'ModelServiceDataModel', 'ModelServiceDataModelBackbone', 'ModelServiceDisconnect', 'ModelServiceFilter', 'ModelserviceStatus', 'OpenAIModel', 'OpenAIProviderDataInput', 'OrderPair', 'PrebuiltConfigInput', 'PrebuiltCriteriaKey', 'PrebuiltJudgeCreate', 'Protocol', 'ProviderName', 'RemoteConfigInput', 'RemoteEnvCreate', 'RemoteEnvData', 'RemoteEnvStatus', 'RemoveRemoteEnv', 'RemoveTeamMember', 'RemoveTeamMemberRemoveTeamMember', 'ResizeInferencePartition', 'ResizePartitionInput', 'RoleCreate', 'SampleConfig', 'SearchInput', 'SelectionTypeInput', 'SessionStatus', 'ShareUseCase', 'ShareUseCaseShareUseCase', 'SortDirection', 'SystemPromptTemplateCreate', 'SystemPromptTemplateUpdate', 'TeamCreate', 'TeamMemberRemove', 'TeamMemberSet', 'TerminateModel', 'TestRemoteEnv', 'TestRemoteEnvTestRemoteEnvRemoteEnvTestOffline', 'TestRemoteEnvTestRemoteEnvRemoteEnvTestOnline', 'TimeRange', 'TimeseriesInput', 'TimeseriesInterval', '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']
+__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', 'BaseClientOpenTelemetry', 'BaseModel', 'BillingCycle', 'CancelABCampaign', 'CancelAllocationInput', 'CancelJob', 'CancelJobCancelJob', 'CapabilityFilter', 'CompletionComparisonFeedbackData', 'CompletionComparisonFeedbackDataModel', 'CompletionComparisonFilterInput', 'CompletionData', 'CompletionDataChatMessages', 'CompletionDataComparisonFeedbacks', 'CompletionDataComparisonFeedbacksMetric', 'CompletionDataComparisonFeedbacksOtherCompletion', 'CompletionDataComparisonFeedbacksPreferedCompletion', 'CompletionDataComparisonFeedbacksUsecase', 'CompletionDataDirectFeedbacks', 'CompletionDataDirectFeedbacksMetric', 'CompletionDataLabels', 'CompletionDataMetadata', 'CompletionDataMetadataUsage', 'CompletionDataModel', 'CompletionFeedbackFilterInput', 'CompletionGroupBy', 'CompletionLabelValue', 'CompletionSource', 'CompletionsByFilters', 'CompletionsById', 'ComputePoolCapability', 'CreateAbCampaign', 'CreateAbCampaignCreateAbCampaign', 'CreateCustomRecipe', 'CreateCustomRecipeCreateCustomRecipe', 'CreateDatasetFromMultipartUpload', 'CreateDatasetFromMultipartUploadCreateDatasetFromMultipartUpload', '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', 'DatasetCompletionQuery', 'DatasetCreate', 'DatasetCreateFromFilters', 'DatasetCreateFromMultipartUpload', 'DatasetData', 'DatasetDataMetricsUsage', 'DatasetDataMetricsUsageMetric', 'DatasetKind', 'DatasetSource', 'DatasetStatus', 'DatasetUploadProcessingStatus', 'DatasetUploadProcessingStatusDatasetUploadProcessingStatus', 'DatasetUploadProcessingStatusInput', 'DateBucketUnit', 'DeleteCustomRecipe', 'DeleteDataset', 'DeleteGrader', 'DeleteGraderDeleteGrader', 'DeleteJudge', 'DeleteJudgeDeleteJudge', 'DeleteUser', 'DeleteUserDeleteUser', 'DescribeAbCampaign', 'DescribeAbCampaignAbCampaign', 'DescribeAbCampaignAbCampaignReport', 'DescribeDataset', 'DescribeDatasetDataset', 'DescribeInteraction', 'DescribeInteractionCompletion', 'DescribeJob', 'DescribeJobJob', 'DescribeMetric', 'DescribeMetricAdmin', 'DescribeMetricAdminMetric', 'DescribeMetricMetric', 'DescribeModel', 'DescribeModelAdmin', 'DescribeModelAdminModel', 'DescribeModelAdminModelBackbone', 'DescribeModelModel', 'DescribeModelModelBackbone', 'DescribeUseCase', 'DescribeUseCaseUseCase', 'EmojiInput', 'ExternalModelProviderName', 'FeedbackAddInput', 'FeedbackFilterInput', 'FeedbackType', 'FeedbackUpdateInput', 'FromGroupsQuery', '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', 'GroupSelection', 'GroupSelectionQuery', 'HarmonyGroupData', 'HarmonyGroupDataComputePool', 'HarmonyGroupDataGpuAllocations', 'HarmonyGroupDataOnlineModels', 'HarmonyStatus', 'JobArtifactFilter', 'JobArtifactKind', 'JobArtifactStatus', 'JobData', 'JobDataCreatedBy', 'JobDataDetails', 'JobDataDetailsArtifacts', 'JobDataDetailsArtifactsByproductsDatasetByproducts', 'JobDataDetailsArtifactsByproductsEvaluationByproducts', 'JobDataDetailsArtifactsByproductsEvaluationByproductsEvalResults', 'JobDataDetailsArtifactsByproductsEvaluationByproductsEvalResultsMetric', 'JobDataDetailsArtifactsByproductsEvaluationByproductsEvalResultsModelService', 'JobDataDetailsArtifactsByproductsModelByproducts', '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', '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', 'MetricAggregation', 'MetricCreate', 'MetricData', 'MetricDataAdmin', 'MetricDataAdminUseCases', 'MetricGetOrCreate', 'MetricKind', 'MetricLink', 'MetricScoringType', 'MetricTrendInput', 'MetricUnlink', 'MetricWithContextData', 'ModelCapabilityFilter', 'ModelComputeConfigInput', 'ModelData', 'ModelDataAdmin', 'ModelDataAdminUseCases', 'ModelDataComputeConfig', 'ModelFilter', 'ModelOnline', 'ModelPlacementInput', 'ModelProviderDataInput', 'ModelServiceData', 'ModelServiceDataModel', 'ModelServiceDataModelBackbone', 'ModelServiceDisconnect', 'ModelServiceFilter', 'ModelserviceStatus', 'OpenAIModel', 'OpenAIProviderDataInput', 'OrderPair', 'PrebuiltConfigInput', 'PrebuiltCriteriaKey', 'PrebuiltJudgeCreate', 'Protocol', 'ProviderName', 'RemoteConfigInput', 'RemoteEnvCreate', 'RemoteEnvData', 'RemoteEnvStatus', 'RemoveRemoteEnv', 'RemoveTeamMember', 'RemoveTeamMemberRemoveTeamMember', 'ResizeInferencePartition', 'ResizePartitionInput', 'RoleCreate', 'SampleConfig', 'SearchInput', 'SelectionTypeInput', 'SessionStatus', 'ShareUseCase', 'ShareUseCaseShareUseCase', 'SortDirection', 'SystemPromptTemplateCreate', 'SystemPromptTemplateUpdate', 'TeamCreate', 'TeamMemberRemove', 'TeamMemberSet', 'TerminateModel', 'TestRemoteEnv', 'TestRemoteEnvTestRemoteEnvRemoteEnvTestOffline', 'TestRemoteEnvTestRemoteEnvRemoteEnvTestOnline', 'TimeRange', 'TimeseriesInput', 'TimeseriesInterval', '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-0.1.11 → adaptive_sdk-0.1.13}/src/adaptive_sdk/graphql_client/async_client.py

@@ -28,7 +28,6 @@ from .delete_dataset import DeleteDataset
 from .delete_grader import DeleteGrader
 from .delete_judge import DeleteJudge
 from .delete_user import DeleteUser
-from .deploy_model import DeployModel
 from .describe_ab_campaign import DescribeAbCampaign
 from .describe_dataset import DescribeDataset
 from .describe_interaction import DescribeInteraction
@@ -42,7 +41,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, AttachModel, CreateRecipeInput, CursorPageInput, CustomRecipeFilterInput, DatasetCreate, DatasetCreateFromMultipartUpload, DatasetUploadProcessingStatusInput, GraderCreateInput, GraderUpdateInput, JobInput, JudgeCreate, JudgeUpdate, ListCompletionsFilterInput, ListJobsFilterInput, MetricCreate, MetricLink, MetricUnlink, ModelComputeConfigInput, ModelFilter, OrderPair, PrebuiltJudgeCreate, RemoteEnvCreate, 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
@@ -135,13 +134,6 @@ class AsyncGQLClient(AsyncBaseClientOpenTelemetry):
         data = self.get_data(response)
         return UpdateModel.model_validate(data)
 
-    async def deploy_model(self, id_or_key: Any, 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 = await self.execute(query=query, operation_name='DeployModel', variables=variables, **kwargs)
-        data = self.get_data(response)
-        return DeployModel.model_validate(data)
-
     async def terminate_model(self, id_or_key: Any, 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}
@@ -486,7 +478,7 @@ class AsyncGQLClient(AsyncBaseClientOpenTelemetry):
         return ListUsers.model_validate(data)
 
     async def me(self, **kwargs: Any) -> Me:
-        query = gql('\n            query Me {\n              me {\n                ...UserData\n                apiKeys {\n                  key\n                  createdAt\n                }\n              }\n            }\n\n            fragment UserData on User {\n              id\n              email\n              name\n              createdAt\n              teams {\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              }\n            }\n            ')
+        query = gql('\n            query Me {\n              me {\n                ...UserData\n              }\n            }\n\n            fragment UserData on User {\n              id\n              email\n              name\n              createdAt\n              teams {\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              }\n            }\n            ')
         variables: Dict[str, object] = {}
         response = await self.execute(query=query, operation_name='Me', variables=variables, **kwargs)
         data = self.get_data(response)

{adaptive_sdk-0.1.11 → adaptive_sdk-0.1.13}/src/adaptive_sdk/graphql_client/client.py

@@ -28,7 +28,6 @@ from .delete_dataset import DeleteDataset
 from .delete_grader import DeleteGrader
 from .delete_judge import DeleteJudge
 from .delete_user import DeleteUser
-from .deploy_model import DeployModel
 from .describe_ab_campaign import DescribeAbCampaign
 from .describe_dataset import DescribeDataset
 from .describe_interaction import DescribeInteraction
@@ -42,7 +41,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, AttachModel, CreateRecipeInput, CursorPageInput, CustomRecipeFilterInput, DatasetCreate, DatasetCreateFromMultipartUpload, DatasetUploadProcessingStatusInput, GraderCreateInput, GraderUpdateInput, JobInput, JudgeCreate, JudgeUpdate, ListCompletionsFilterInput, ListJobsFilterInput, MetricCreate, MetricLink, MetricUnlink, ModelComputeConfigInput, ModelFilter, OrderPair, PrebuiltJudgeCreate, RemoteEnvCreate, 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
@@ -135,13 +134,6 @@ class GQLClient(BaseClientOpenTelemetry):
         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}
@@ -486,7 +478,7 @@ class GQLClient(BaseClientOpenTelemetry):
         return ListUsers.model_validate(data)
 
     def me(self, **kwargs: Any) -> Me:
-        query = gql('\n            query Me {\n              me {\n                ...UserData\n                apiKeys {\n                  key\n                  createdAt\n                }\n              }\n            }\n\n            fragment UserData on User {\n              id\n              email\n              name\n              createdAt\n              teams {\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              }\n            }\n            ')
+        query = gql('\n            query Me {\n              me {\n                ...UserData\n              }\n            }\n\n            fragment UserData on User {\n              id\n              email\n              name\n              createdAt\n              teams {\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              }\n            }\n            ')
         variables: Dict[str, object] = {}
         response = self.execute(query=query, operation_name='Me', variables=variables, **kwargs)
         data = self.get_data(response)

{adaptive_sdk-0.1.11 → adaptive_sdk-0.1.13}/src/adaptive_sdk/graphql_client/custom_fields.py

@@ -587,6 +587,7 @@ class CustomRecipeFields(GraphQLField):
     id: 'CustomRecipeGraphQLField' = CustomRecipeGraphQLField('id')
     key: 'CustomRecipeGraphQLField' = CustomRecipeGraphQLField('key')
     name: 'CustomRecipeGraphQLField' = CustomRecipeGraphQLField('name')
+    is_multifile: 'CustomRecipeGraphQLField' = CustomRecipeGraphQLField('isMultifile')
     editable: 'CustomRecipeGraphQLField' = CustomRecipeGraphQLField('editable')
     hidden: 'CustomRecipeGraphQLField' = CustomRecipeGraphQLField('hidden')
     builtin: 'CustomRecipeGraphQLField' = CustomRecipeGraphQLField('builtin')
@@ -626,6 +627,8 @@ class CustomRecipeJobDetailsFields(GraphQLField):
     @classmethod
     def artifacts(cls) -> 'JobArtifactFields':
         return JobArtifactFields('artifacts')
+    num_gpus: 'CustomRecipeJobDetailsGraphQLField' = CustomRecipeJobDetailsGraphQLField('numGpus')
+    gpu_duration_ms: 'CustomRecipeJobDetailsGraphQLField' = CustomRecipeJobDetailsGraphQLField('gpuDurationMs')
 
     def fields(self, *subfields: Union[CustomRecipeJobDetailsGraphQLField, 'JobArtifactFields']) -> 'CustomRecipeJobDetailsFields':
         """Subfields should come from the CustomRecipeJobDetailsFields class"""
@@ -1056,6 +1059,7 @@ class JobArtifactFields(GraphQLField):
     job_id: 'JobArtifactGraphQLField' = JobArtifactGraphQLField('jobId')
     name: 'JobArtifactGraphQLField' = JobArtifactGraphQLField('name')
     kind: 'JobArtifactGraphQLField' = JobArtifactGraphQLField('kind')
+    status: 'JobArtifactGraphQLField' = JobArtifactGraphQLField('status')
     uri: 'JobArtifactGraphQLField' = JobArtifactGraphQLField('uri')
     metadata: 'JobArtifactGraphQLField' = JobArtifactGraphQLField('metadata')
     created_at: 'JobArtifactGraphQLField' = JobArtifactGraphQLField('createdAt')
@@ -1417,15 +1421,15 @@ class ModelFields(GraphQLField):
         return MetricWithContextFields('metrics')
 
     @classmethod
-    def use_cases(cls, filter: UseCaseFilter, attached: bool) -> 'UseCaseFields':
+    def use_cases(cls, filter: UseCaseFilter) -> 'UseCaseFields':
         """Return the list of UseCase which use this model"""
-        arguments: Dict[str, Dict[str, Any]] = {'filter': {'type': 'UseCaseFilter!', 'value': filter}, 'attached': {'type': 'Boolean!', 'value': attached}}
+        arguments: Dict[str, Dict[str, Any]] = {'filter': {'type': 'UseCaseFilter!', 'value': filter}}
         cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
         return UseCaseFields('useCases', arguments=cleared_arguments)
 
     @classmethod
-    def model_services(cls, filter: UseCaseFilter, attached: bool) -> 'ModelServiceFields':
-        arguments: Dict[str, Dict[str, Any]] = {'filter': {'type': 'UseCaseFilter!', 'value': filter}, 'attached': {'type': 'Boolean!', 'value': attached}}
+    def model_services(cls, filter: UseCaseFilter) -> 'ModelServiceFields':
+        arguments: Dict[str, Dict[str, Any]] = {'filter': {'type': 'UseCaseFilter!', 'value': filter}}
         cleared_arguments = {key: value for key, value in arguments.items() if value['value'] is not None}
         return ModelServiceFields('modelServices', arguments=cleared_arguments)
     provider_name: 'ModelGraphQLField' = ModelGraphQLField('providerName')
@@ -2193,11 +2197,7 @@ class UserFields(GraphQLField):
     def teams(cls) -> 'TeamWithroleFields':
         return TeamWithroleFields('teams')
 
-    @classmethod
-    def api_keys(cls) -> 'ApiKeyFields':
-        return ApiKeyFields('apiKeys')
-
-    def fields(self, *subfields: Union[UserGraphQLField, 'ApiKeyFields', 'TeamWithroleFields']) -> 'UserFields':
+    def fields(self, *subfields: Union[UserGraphQLField, 'TeamWithroleFields']) -> 'UserFields':
         """Subfields should come from the UserFields class"""
         self._subfields.extend(subfields)
         return self

{adaptive_sdk-0.1.11 → adaptive_sdk-0.1.13}/src/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, ApiKeyCreate, AttachModel, CancelAllocationInput, CreateRecipeInput, CreateToolProviderInput, DatasetCreate, DatasetCreateFromFilters, DatasetCreateFromMultipartUpload, FeedbackAddInput, FeedbackUpdateInput, GraderCreateInput, GraderUpdateInput, JobInput, JudgeCreate, JudgeUpdate, MetricCreate, MetricLink, MetricUnlink, ModelComputeConfigInput, ModelServiceDisconnect, PrebuiltJudgeCreate, RemoteEnvCreate, ResizePartitionInput, RoleCreate, SystemPromptTemplateCreate, SystemPromptTemplateUpdate, TeamCreate, TeamMemberRemove, TeamMemberSet, UpdateCompletion, UpdateModelService, UpdateRecipeInput, UpdateToolProviderInput, UseCaseCreate, UseCaseShares, UseCaseUpdate, UserCreate
 
 class Mutation:
     """@private"""
@@ -140,12 +140,6 @@ 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"""

{adaptive_sdk-0.1.11 → adaptive_sdk-0.1.13}/src/adaptive_sdk/graphql_client/enums.py

@@ -94,6 +94,13 @@ class JobArtifactKind(str, Enum):
     DATASET = 'DATASET'
     MODEL = 'MODEL'
 
+class JobArtifactStatus(str, Enum):
+    """@public"""
+    PENDING = 'PENDING'
+    PROCESSING = 'PROCESSING'
+    READY = 'READY'
+    ERROR = 'ERROR'
+
 class JobKind(str, Enum):
     """@public"""
     TRAINING = 'TRAINING'

{adaptive_sdk-0.1.11 → adaptive_sdk-0.1.13}/src/adaptive_sdk/graphql_client/input_types.py

@@ -50,6 +50,8 @@ class AnthropicProviderDataInput(BaseModel):
 class ApiKeyCreate(BaseModel):
     """@private"""
     user: str
+    expiration_for_previous_keys: Optional[int | str] = Field(alias='expirationForPreviousKeys', default=None)
+    'If not provided, previous keys expire immediately'
 
 class ArtifactFilter(BaseModel):
     """@private"""

adaptive_sdk-0.1.13/src/adaptive_sdk/graphql_client/me.py

@@ -0,0 +1,12 @@
+from typing import Optional
+from .base_model import BaseModel
+from .fragments import UserData
+
+class Me(BaseModel):
+    """@public"""
+    me: Optional['MeMe']
+
+class MeMe(UserData):
+    """@public"""
+    pass
+Me.model_rebuild()

{adaptive_sdk-0.1.11 → adaptive_sdk-0.1.13}/src/adaptive_sdk/resources/jobs.py

@@ -68,12 +68,13 @@ class Jobs(SyncAPIResource, UseCaseResource):  # type: ignore[misc]
     def run(
         self,
         recipe_key: str,
-        args: dict[str, Any],
         num_gpus: int,
+        args: dict[str, Any] | None = None,
         name: str | None = None,
         use_case: str | None = None,
         compute_pool: str | None = None,
     ) -> JobDataPlus:
+        args = args or {}
         job_data = self._gql_client.create_job(
             input=JobInput(
                 recipe=recipe_key,
@@ -139,12 +140,13 @@ class AsyncJobs(AsyncAPIResource, UseCaseResource):  # type: ignore[misc]
     async def run(
         self,
         recipe_key: str,
-        args: dict[str, Any],
         num_gpus: int,
+        args: dict[str, Any] | None = None,
         name: str | None = None,
         use_case: str | None = None,
         compute_pool: str | None = None,
     ) -> JobDataPlus:
+        args = args or {}
         job_data = (
             await self._gql_client.create_job(
                 input=JobInput(

{adaptive_sdk-0.1.11 → adaptive_sdk-0.1.13}/src/adaptive_sdk/resources/models.py

@@ -277,16 +277,6 @@ class Models(SyncAPIResource, UseCaseResource):  # type: ignore[misc]
         )
         return self._gql_client.update_model(input).update_model_service
 
-    def deploy(self, model: str, wait: bool = False) -> str:
-        """
-        Deploy a model, loading it to memory and making it ready for inference.
-
-        Args:
-            model: Model key.
-            wait: If `True`, call block until model is in `Online` state.
-        """
-        return self._gql_client.deploy_model(id_or_key=model, wait=wait).deploy_model
-
     def terminate(self, model: str, force: bool = False) -> str:
         """
         Terminate model, removing it from memory and making it unavailable to all use cases.
@@ -490,18 +480,6 @@ class AsyncModels(AsyncAPIResource, UseCaseResource):  # type: ignore[misc]
         result = await self._gql_client.update_model(input)
         return result.update_model_service
 
-    async def deploy(self, model: str, wait: bool = False) -> str:
-        """
-        Deploy a model, loading it to memory and making it ready for inference.
-
-        Args:
-            model: Model key.
-            wait: If `True`, call block until model is in `Online` state.
-        """
-        return (
-            await self._gql_client.deploy_model(id_or_key=model, wait=wait)
-        ).deploy_model
-
     async def terminate(self, model: str, force: bool = False) -> str:
         """
         Terminate model, removing it from memory and making it unavailable to all use cases.

{adaptive_sdk-0.1.11 → adaptive_sdk-0.1.13}/src/adaptive_sdk/resources/recipes.py

@@ -42,7 +42,8 @@ class Recipes(SyncAPIResource, UseCaseResource):  # type: ignore[misc]
     def upload(
         self,
         path: str,
-        recipe_key: str,
+        recipe_key: str | None = None,
+        entrypoint: str | None = None,
         name: str | None = None,
         description: str | None = None,
         labels: dict[str, str] | None = None,
@@ -50,8 +51,25 @@ class Recipes(SyncAPIResource, UseCaseResource):  # type: ignore[misc]
     ) -> CustomRecipeData:
         """
         Upload a recipe from either a single Python file or a directory (path).
-        If a directory is provided, it must contain a 'main.py' and will be zipped in-memory before upload.
+
+        Args:
+            path: Path to a Python file or directory containing the recipe
+            recipe_key: Optional unique key for the recipe. If not provided, inferred from:
+                - File name (without .py) if path is a file
+                - "dir_name/entrypoint_name" if path is a directory and custom entrypoint is specified
+                - Directory name if path is a directory and no custom entrypoint is specified
+            entrypoint: Optional relative path to the entrypoint file within a directory.
+                If specified, this file will be used as main.py. Cannot be used if
+                main.py already exists in the directory, or if path is a file.
+            name: Optional display name for the recipe
+            description: Optional description
+            labels: Optional key-value labels
+            use_case: Optional use case identifier
         """
+        p = Path(path)
+        if recipe_key is None:
+            recipe_key = _get_recipe_key(p, entrypoint=entrypoint)
+
         inferred_name = name or recipe_key
         label_inputs = [LabelInput(key=k, value=v) for k, v in labels.items()] if labels else None
         input = CreateRecipeInput(
@@ -60,10 +78,13 @@ class Recipes(SyncAPIResource, UseCaseResource):  # type: ignore[misc]
             description=description,
             labels=label_inputs,
         )
-        with _upload_from_path(path) as file_upload:
-            return self._gql_client.create_custom_recipe(
+
+        with _upload_from_path(path, entrypoint=entrypoint) as file_upload:
+            new_recipe = self._gql_client.create_custom_recipe(
                 use_case=self.use_case_key(use_case), input=input, file=file_upload
             ).create_custom_recipe
+        logger.info(f"New recipe created with key `{new_recipe.key}`")
+        return new_recipe
 
     def get(
         self,
@@ -78,6 +99,7 @@ class Recipes(SyncAPIResource, UseCaseResource):  # type: ignore[misc]
         self,
         recipe_key: str,
         path: str | None = None,
+        entrypoint: str | None = None,
         name: str | None = None,
         description: str | None = None,
         labels: Sequence[tuple[str, str]] | None = None,
@@ -91,7 +113,7 @@ class Recipes(SyncAPIResource, UseCaseResource):  # type: ignore[misc]
         )
 
         if path:
-            with _upload_from_path(path) as file_upload:
+            with _upload_from_path(path, entrypoint=entrypoint) as file_upload:
                 return self._gql_client.update_custom_recipe(
                     use_case=self.use_case_key(use_case),
                     id=recipe_key,
@@ -160,12 +182,34 @@ class AsyncRecipes(AsyncAPIResource, UseCaseResource):  # type: ignore[misc]
     async def upload(
         self,
         path: str,
-        recipe_key: str,
+        recipe_key: str | None = None,
+        entrypoint: str | None = None,
         name: str | None = None,
         description: str | None = None,
         labels: Sequence[tuple[str, str]] | None = None,
         use_case: str | None = None,
     ) -> CustomRecipeData:
+        """
+        Upload a recipe from either a single Python file or a directory (path).
+
+        Args:
+            path: Path to a Python file or directory containing the recipe
+            recipe_key: Optional unique key for the recipe. If not provided, inferred from:
+                - File name (without .py) if path is a file
+                - "dir_name/entrypoint_name" if path is a directory and custom entrypoint is specified
+                - Directory name if path is a directory and no custom entrypoint is specified
+            entrypoint: Optional relative path to the entrypoint file within a directory.
+                If specified, this file will be used as main.py. Cannot be used if
+                main.py already exists in the directory, or if path is a file.
+            name: Optional display name for the recipe
+            description: Optional description
+            labels: Optional key-value labels
+            use_case: Optional use case identifier
+        """
+        p = Path(path)
+        if recipe_key is None:
+            recipe_key = _get_recipe_key(p, entrypoint=entrypoint)
+
         inferred_name = name or recipe_key
         label_inputs = [LabelInput(key=k, value=v) for k, v in labels] if labels else None
         input = CreateRecipeInput(
@@ -174,12 +218,14 @@ class AsyncRecipes(AsyncAPIResource, UseCaseResource):  # type: ignore[misc]
             description=description,
             labels=label_inputs,
         )
-        with _upload_from_path(path) as file_upload:
-            return (
+        with _upload_from_path(path, entrypoint=entrypoint) as file_upload:
+            new_recipe = (
                 await self._gql_client.create_custom_recipe(
                     use_case=self.use_case_key(use_case), input=input, file=file_upload
                 )
             ).create_custom_recipe
+            logger.info(f"New recipe create with key {new_recipe.key}")
+            return new_recipe
 
     async def get(
         self,
@@ -194,6 +240,7 @@ class AsyncRecipes(AsyncAPIResource, UseCaseResource):  # type: ignore[misc]
         self,
         recipe_key: str,
         path: str | None = None,
+        entrypoint: str | None = None,
         name: str | None = None,
         description: str | None = None,
         labels: Sequence[tuple[str, str]] | None = None,
@@ -207,7 +254,7 @@ class AsyncRecipes(AsyncAPIResource, UseCaseResource):  # type: ignore[misc]
         )
 
         if path:
-            with _upload_from_path(path) as file_upload:
+            with _upload_from_path(path, entrypoint=entrypoint) as file_upload:
                 return (
                     await self._gql_client.update_custom_recipe(
                         use_case=self.use_case_key(use_case),
@@ -275,6 +322,21 @@ def _count_keys_recursively(data: Any) -> int:
     return count
 
 
+def _get_recipe_key(path: Path, entrypoint: str | None = None) -> str:
+    if path.is_file():
+        recipe_key = path.stem
+    elif path.is_dir():
+        if entrypoint:
+            entrypoint_stem = Path(entrypoint).stem
+            recipe_key = f"{path.name}/{entrypoint_stem}"
+        else:
+            recipe_key = path.name
+    else:
+        raise ValueError(f"Path must be a Python file or directory: {path}")
+
+    return recipe_key
+
+
 def _validate_python_file(path: Path) -> None:
     """Validate that the path exists, is a file and has a .py extension."""
     if not path.exists():
@@ -285,40 +347,53 @@ def _validate_python_file(path: Path) -> None:
         raise ValueError(f"Expected a Python file with .py extension, got: {path}")
 
 
-def _validate_recipe_directory(dir_path: Path) -> None:
-    """Validate that the directory exists and contains a main.py file."""
+def _validate_recipe_directory(dir_path: Path, entrypoint: str | None = None) -> None:
     if not dir_path.exists():
         raise FileNotFoundError(f"Directory not found: {dir_path}")
     if not dir_path.is_dir():
         raise ValueError(f"Expected a directory path, got a file: {dir_path}")
+
     main_py = dir_path / "main.py"
-    if not main_py.exists() or not main_py.is_file():
-        raise FileNotFoundError(f"Directory must contain a 'main.py' file: {dir_path}")
+    main_py_exists = main_py.exists() and main_py.is_file()
+
+    if entrypoint:
+        if main_py_exists:
+            raise ValueError(
+                f"Cannot specify entrypoint when main.py already exists in directory: {dir_path}. "
+                f"Either remove/rename main.py or use it directly without specifying an entrypoint."
+            )
+        entrypoint_path = dir_path / entrypoint
+        if not entrypoint_path.exists() or not entrypoint_path.is_file():
+            raise FileNotFoundError(f"Specified entrypoint file not found: {entrypoint} in directory {dir_path}")
+    else:
+        if not main_py_exists:
+            raise FileNotFoundError(
+                f"Directory must contain a 'main.py' file, or in alternative you must specify an `entrypoint` file"
+            )
 
 
-def _zip_directory_to_bytes_io(dir_path: Path) -> io.BytesIO:
-    """Zip the contents of a directory into an in-memory BytesIO buffer."""
+def _zip_directory_to_bytes_io(dir_path: Path, entrypoint: str | None = None) -> io.BytesIO:
     buffer = io.BytesIO()
     with zipfile.ZipFile(buffer, mode="w", compression=zipfile.ZIP_DEFLATED) as zf:
         for root, _, files in os.walk(dir_path):
             for file_name in files:
                 file_path = Path(root) / file_name
                 arcname = file_path.relative_to(dir_path)
+                if entrypoint and arcname.as_posix() == entrypoint:
+                    arcname = Path("main.py")
+
                 zf.write(file_path, arcname.as_posix())
     buffer.seek(0)
     return buffer
 
 
 @contextmanager
-def _upload_from_path(path: str):
-    """
-    Context manager yielding an Upload object for a Python file or a directory.
-
-    - If path is a .py file, validates and opens it for upload.
-    - If path is a directory, validates it contains main.py, zips contents in-memory.
-    """
+def _upload_from_path(path: str, entrypoint: str | None = None):
     p = Path(path)
+
     if p.is_file():
+        if entrypoint:
+            raise ValueError(f"Entrypoint parameter is not supported for single file recipe uploads")
         _validate_python_file(p)
         filename = p.name
         content_type = mimetypes.guess_type(str(p))[0] or "application/octet-stream"
@@ -328,28 +403,13 @@ def _upload_from_path(path: str):
         finally:
             f.close()
     elif p.is_dir():
-        _validate_recipe_directory(p)
-        # Ensure __init__.py exists at the root of the directory before zipping
-        created_init = False
-        root_init = p / "__init__.py"
+        _validate_recipe_directory(p, entrypoint=entrypoint)
         zip_buffer = None
         try:
-            if not root_init.exists():
-                root_init.touch()
-                created_init = True
-                logger.info(f"Added __init__.py to your directory, as it is required for proper execution of recipe")
-            zip_buffer = _zip_directory_to_bytes_io(p)
-        finally:
-            if created_init:
-                try:
-                    root_init.unlink()
-                    logger.info(f"Cleaned up __init__.py from your directory")
-                except Exception:
-                    logger.error(f"Failed to remove __init__.py from your directory")
-                    pass
-        if zip_buffer is None:
-            raise RuntimeError("Failed to create in-memory zip for directory upload")
-
+            zip_buffer = _zip_directory_to_bytes_io(p, entrypoint=entrypoint)
+        except Exception:
+            logger.error(f"Failed to create in-memory zip for directory upload.")
+            raise
         filename = f"{p.name}.zip"
         try:
             yield Upload(filename=filename, content=zip_buffer, content_type="application/zip")

adaptive_sdk-0.1.11/src/adaptive_sdk/graphql_client/deploy_model.py

@@ -1,6 +0,0 @@
-from pydantic import Field
-from .base_model import BaseModel
-
-class DeployModel(BaseModel):
-    """@public"""
-    deploy_model: str = Field(alias='deployModel')

adaptive_sdk-0.1.11/src/adaptive_sdk/graphql_client/me.py

@@ -1,19 +0,0 @@
-from typing import List, Optional
-from pydantic import Field
-from .base_model import BaseModel
-from .fragments import UserData
-
-class Me(BaseModel):
-    """@public"""
-    me: Optional['MeMe']
-
-class MeMe(UserData):
-    """@public"""
-    api_keys: List['MeMeApiKeys'] = Field(alias='apiKeys')
-
-class MeMeApiKeys(BaseModel):
-    """@public"""
-    key: str
-    created_at: int = Field(alias='createdAt')
-Me.model_rebuild()
-MeMe.model_rebuild()