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

adaptive_sdk/graphql_client/__init__.py

@@ -38,7 +38,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 +67,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 +81,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', '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', '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/graphql_client/async_client.py

@@ -486,7 +486,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/graphql_client/client.py

@@ -486,7 +486,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/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/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/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/graphql_client/me.py

@@ -1,5 +1,4 @@
-from typing import List, Optional
-from pydantic import Field
+from typing import Optional
 from .base_model import BaseModel
 from .fragments import UserData
 
@@ -9,11 +8,5 @@ class Me(BaseModel):
 
 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()
+    pass
+Me.model_rebuild()

adaptive_sdk/resources/compute_pools.py

@@ -36,7 +36,7 @@ class ComputePools(SyncAPIResource, UseCaseResource):
         """
         cps = self.list()
         found_cp = False
-        for cp in cps.compute_pools:
+        for cp in cps:
             if cp.key == compute_pool_key:
                 selected_cp = cp
                 found_cp = True
@@ -75,7 +75,7 @@ class AsyncComputePools(AsyncAPIResource, UseCaseResource):
         """
         cps = await self.list()
         found_cp = False
-        for cp in cps.compute_pools:
+        for cp in cps:
             if cp.key == compute_pool_key:
                 selected_cp = cp
                 found_cp = True

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/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,
@@ -160,12 +181,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 +217,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,
@@ -275,6 +320,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 +345,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 +401,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.10.dist-info → adaptive_sdk-0.1.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: adaptive-sdk
-Version: 0.1.10
+Version: 0.1.12
 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.10.dist-info → adaptive_sdk-0.1.12.dist-info}/RECORD

@@ -13,20 +13,20 @@ adaptive_sdk/external/reward_client.py,sha256=TSVdi437McUkMModDQV6ZswduaDmmG1Bvk
 adaptive_sdk/external/reward_server.py,sha256=yT8vqAEIoaq8nFZYaku5IoK0_7fX9uX_dfF1CxcDKnA,7488
 adaptive_sdk/external/reward_types.py,sha256=aZmTolT0OjgObo-64zJkarUKOY4RdxHnsQt8AKAaq1w,1710
 adaptive_sdk/external/websocket_utils.py,sha256=QN-K1IRbqe1LdQTz6vXhVgn-R2SjOB80NjGD2w_LaAo,1284
-adaptive_sdk/graphql_client/__init__.py,sha256=gAbg4sm_anABglsmVQH_emoq5pCu_SvLUHfjVX4rxBE,21564
+adaptive_sdk/graphql_client/__init__.py,sha256=Kdi29ZNGnz0uBvRqxY1WKJkmh-R5QEoQEOiQc2poPUo,21576
 adaptive_sdk/graphql_client/add_external_model.py,sha256=9VpQHlJMm5t_Ja_SX2MkYcSG1IQpem6mrdXMfNTa1oo,352
 adaptive_sdk/graphql_client/add_hf_model.py,sha256=aC2IcYftepe28Hi01Kg0w3A7bjKrStWasx6aXiqghJU,312
 adaptive_sdk/graphql_client/add_model.py,sha256=Uyhpxkziz1Pv2S7Q0wOaKGK4krjmEf2PeLK0yEs89Hs,461
 adaptive_sdk/graphql_client/add_remote_env.py,sha256=X94F94tnMzuX9TC_Ieowngw35gJxnKL2YST2sqEJ7CY,328
 adaptive_sdk/graphql_client/async_base_client_open_telemetry.py,sha256=XSRJGEcyfWnFjrDz4Un6xfQWOxr7jWto1vYjbYTQeo0,19761
-adaptive_sdk/graphql_client/async_client.py,sha256=hnEGHgc7StUSfNSmTIPQKTWUG7C8GktYqynhOZmRGTs,119145
+adaptive_sdk/graphql_client/async_client.py,sha256=v84UFrgxoVqC0A-XbwQz4kLVBEtGsuGyVHWJFzio4Kc,119047
 adaptive_sdk/graphql_client/attach_model_to_use_case.py,sha256=WyERM4wxKrUS1u9VZ3FUasZu5AVSbRswzy9jmxssTFM,361
 adaptive_sdk/graphql_client/base_client_open_telemetry.py,sha256=IV96gRr5FuH-dWMU5PBvQhTwEsV7udsXU-Dyh-Mx-4k,9398
 adaptive_sdk/graphql_client/base_model.py,sha256=2xxKA4sIdlGPIezw06XP9bseSxBURU7nivgt_gL44iA,621
 adaptive_sdk/graphql_client/base_operation.py,sha256=aooq1M4r79cvMoganZ2RvQ3-v0io22fGLOVfl3UBqPQ,4635
 adaptive_sdk/graphql_client/cancel_ab_campaign.py,sha256=JAEpmedz0jOu90U3KR0PYCMAhf2_E6h6WOT30HSE91k,176
 adaptive_sdk/graphql_client/cancel_job.py,sha256=3ZLoqrULi1mp5A5i4rD-gXliKhD8038IPfvCNBg0dPs,291
-adaptive_sdk/graphql_client/client.py,sha256=rGq0Uy4iiO79-ufnegJFkUPPcK9PzKcD-bY95YPh3rw,118200
+adaptive_sdk/graphql_client/client.py,sha256=XwV4lEzQThVK31hPBmqwO5ncGqdgM-lXnr-tBRisvO8,118102
 adaptive_sdk/graphql_client/create_ab_campaign.py,sha256=___2iFSGnWKkjXz_MfxKUxi1EoQcSBv9AX8S7YoUeVw,374
 adaptive_sdk/graphql_client/create_custom_recipe.py,sha256=eIVLDHbdFG2qWjoZBLC3Xs40Fjy6L-u6wrABV5ibUxo,382
 adaptive_sdk/graphql_client/create_dataset_from_multipart_upload.py,sha256=eoqOfEviCFww5ElTknknV0qOpFTJQmQ1BeGLsRQ5iHc,730
@@ -39,7 +39,7 @@ adaptive_sdk/graphql_client/create_role.py,sha256=6aTdNOZxavMyjkH-g01uYOZgpjYWcA
 adaptive_sdk/graphql_client/create_team.py,sha256=6Alt1ralE1-Xvp2wrEaLUHMW5RtiFqz2fIsUYE_2LbM,370
 adaptive_sdk/graphql_client/create_use_case.py,sha256=sekD76jWCo3zNCfMsBGhVYfNSIK4JPPBz9066BOt49g,332
 adaptive_sdk/graphql_client/create_user.py,sha256=gurD0kZgncXt1HBr7Oo5AkK5ubqFKpJvaR1rn506gHo,301
-adaptive_sdk/graphql_client/custom_fields.py,sha256=T461vWHgAhZdjk2fH7luhb5GuLA06dNnEg8D9x_8EcY,99817
+adaptive_sdk/graphql_client/custom_fields.py,sha256=s77Stl8__yrPDAimCp2zoKefZdT6YxAwwN5atsfMw5Y,99937
 adaptive_sdk/graphql_client/custom_mutations.py,sha256=meo96-odIhk_NoFL_uDj78xguLOD-_DIYW2bj6ilvII,25341
 adaptive_sdk/graphql_client/custom_queries.py,sha256=dVMj82YFJg9wjcnZdmnCbEKEaLiL8QUgpCjr5ONuQAU,18242
 adaptive_sdk/graphql_client/custom_typing_fields.py,sha256=yT_a4kZOMIccqIm-sAG0zu70c-plQtQrZ6ynmfxjlDU,20786
@@ -59,13 +59,13 @@ adaptive_sdk/graphql_client/describe_metric_admin.py,sha256=_SKKwnFhZnbOTT97elEr
 adaptive_sdk/graphql_client/describe_model.py,sha256=UnsOnAyBjNsnkJaS4q5uwkSSvInHwRqUj3XqAoO0yO4,434
 adaptive_sdk/graphql_client/describe_model_admin.py,sha256=XUt_CBSMw1HRleUEWZn2snYt2BNSux_siqrVlwtqH-w,484
 adaptive_sdk/graphql_client/describe_use_case.py,sha256=WW0QkTmdfggN8YBUNGi8ShrP_fr2jXPR6Fer6jlQxu0,353
-adaptive_sdk/graphql_client/enums.py,sha256=SHZSmlL85LtwXJ32fOmto9N07mtAR6a3t4XbRrhGyuA,5108
+adaptive_sdk/graphql_client/enums.py,sha256=dneXg7eIX_UzVNfdImhs_Z1JPulePP-zl_XqKk7Q1Qw,5257
 adaptive_sdk/graphql_client/exceptions.py,sha256=NiC6v-5S7aRDlvQTcHH3K5KvxWvk-c-PkIQQHkipTB8,2268
 adaptive_sdk/graphql_client/fragments.py,sha256=NMata4RWlGeR5rYJouo-PqRFGB2tN_9VBh3bBW6K8w4,22485
 adaptive_sdk/graphql_client/get_custom_recipe.py,sha256=7qxBZGQTqpc69k-NwzgFctaHWaRz0tHl7YlVSsEad6U,383
 adaptive_sdk/graphql_client/get_grader.py,sha256=kubHDBtUcq6mZtUR5_Of0QbjnGUPSYuavF3_xwmwbY8,233
 adaptive_sdk/graphql_client/get_judge.py,sha256=urEnHW3XfURi5GAFBPfbqzOZGQDxgsGRA6nZmUKmoMA,224
-adaptive_sdk/graphql_client/input_types.py,sha256=lFY7mKgDHHRNcXTsu_0QqTaG3wWTUoMw7sjlRWqGi88,20048
+adaptive_sdk/graphql_client/input_types.py,sha256=BLT_xkluyu6O8w0wZGBgpZjgPsA_II02WAnk63UNpAA,20215
 adaptive_sdk/graphql_client/link_metric.py,sha256=EDH67ckBzzc6MYIGfsmgZRBnjqxLsCGwFUaFMXPEsBY,327
 adaptive_sdk/graphql_client/list_ab_campaigns.py,sha256=SIbU6I2OQkNHt0Gw6YStoiiwJHUk2rfXnpoGLzrFjxc,379
 adaptive_sdk/graphql_client/list_compute_pools.py,sha256=4Qli5Foxm3jztbUAL5gbwqtcrElwwlC4LGJMOMBI6Cc,782
@@ -88,7 +88,7 @@ adaptive_sdk/graphql_client/list_use_cases.py,sha256=fbEicXAKDTgsBwBVwtd68hMEPk-
 adaptive_sdk/graphql_client/list_users.py,sha256=9LCNz49jqxrUapHyOdnzs-ZtU1xsejqoXUmE8vsr-qY,256
 adaptive_sdk/graphql_client/load_dataset.py,sha256=P_h3wPTT4E5mbgJoR5jX1_5GhDLIJsA6lmYPT27VxY8,323
 adaptive_sdk/graphql_client/lock_grader.py,sha256=cCISIKjFaSakf-tr4oEWebieT5yrpvGxdS_R_E4iww8,305
-adaptive_sdk/graphql_client/me.py,sha256=0I4fqW8wtO-jQSsSym_dtWggdbEa3m71wH955-AswwI,440
+adaptive_sdk/graphql_client/me.py,sha256=oR_m5QRalokWb9lZYSNgDzOBcNLuAgBIXZh7zfBerqw,227
 adaptive_sdk/graphql_client/remove_remote_env.py,sha256=lmEwudWc6bkU1ev2g9_QEOgkJT_n7n9P5wGm5AV2MY8,173
 adaptive_sdk/graphql_client/remove_team_member.py,sha256=H75WLcB4bM0yctfmmzztxSAzxvbOwWmkVZhO5DUuh1g,350
 adaptive_sdk/graphql_client/resize_inference_partition.py,sha256=osrAGQGYCFjAyfkftkxYW4Wv2VzMq_oW8i0x3_B8aKg,200
@@ -110,16 +110,16 @@ adaptive_sdk/resources/__init__.py,sha256=-cHNzI-hg1u-GGe-fVvV_L4JOgat2zJuKnC3E2
 adaptive_sdk/resources/abtests.py,sha256=9PCPjxuWwY9ec88ewnq54gkoELq5U5iaBmHhzLCAsFU,7698
 adaptive_sdk/resources/base_resource.py,sha256=D9adWSFxiDB7chVstDuBu1jcuXkE71UQJexnWENpC4A,1497
 adaptive_sdk/resources/chat.py,sha256=k-jh_IZfKwqoWNN2P5BhAM3P_f5HLybVZp35pUadwso,11954
-adaptive_sdk/resources/compute_pools.py,sha256=v9Igu7HnEOIfGhDuwl2cSzDQaxc4FNsgXjufFWtoAhc,3469
+adaptive_sdk/resources/compute_pools.py,sha256=_c_eDkXyqKm53ZcfL91EtcuitYqd7ds_3Uz4PUY-TLw,3441
 adaptive_sdk/resources/datasets.py,sha256=sgGP2BwhaezaGei8xXoH0aKHyZFc64ZvIllxFUKNTd8,13648
 adaptive_sdk/resources/embeddings.py,sha256=-ov_EChHU6PJJOJRtDlCo4sYyr9hwyvRjnBhub8QNFg,3922
 adaptive_sdk/resources/feedback.py,sha256=lujqwFIhxi6iovL8JWL05Kr-gkzR4QEwUXZbTx33raA,14116
 adaptive_sdk/resources/graders.py,sha256=ekQQ5fqmLZpZHeLr6iUm6m45wDevoDJdj3mG-axR-m8,29014
 adaptive_sdk/resources/interactions.py,sha256=9A0aKyfE5dhMj-rj6NOiF7kxAl89SXksFsRJXXjPGK8,10810
-adaptive_sdk/resources/jobs.py,sha256=dN19HgOvGEq9O5hhgDZlWhi15noXo4qHLT_fd-qjrAU,5318
+adaptive_sdk/resources/jobs.py,sha256=3ueoHp_58ELIAWfRfiAp51Jt7hazMmbfngaep2wnPn8,5398
 adaptive_sdk/resources/models.py,sha256=f64QQndni3ETRaBICDUyP2Aka9R5Z_RKEyxqZcsv3p0,18314
 adaptive_sdk/resources/permissions.py,sha256=ckO-oacWkvgcwXBK5iW_8qaK-g0SHXpEEy1qZy5lrB0,1053
-adaptive_sdk/resources/recipes.py,sha256=f4Q_noEAjx-LGkgzLOU6UtgFTK75cQu0cYBakNjUCv0,13224
+adaptive_sdk/resources/recipes.py,sha256=ookHUdWpBv2w1RmkmQgxAs7E2VGK9dyb28FOK1sD9eE,15991
 adaptive_sdk/resources/roles.py,sha256=fD1F5Gd3ddkATsU2aFj7japTJVZngVgqt6TXskBQEOA,2218
 adaptive_sdk/resources/teams.py,sha256=KItuOfqKKyFSRFoEAF7rAXb8nbIqRkwRxjEbACjeEoY,1476
 adaptive_sdk/resources/use_cases.py,sha256=a_Ls3kbYgByJMlOYM_UGi4MTiW4wU7m057budrgfEE0,9014
@@ -127,6 +127,6 @@ adaptive_sdk/resources/users.py,sha256=SoGWwdDCdhK4KjYOcAws-ZWlW7Edii7D3Vxfdu-NZ
 adaptive_sdk/rest/__init__.py,sha256=P9uhkOoc9cgUkJ5MBoV5soPgQWSkvPrTwHzPGX7i5tY,610
 adaptive_sdk/rest/base_model.py,sha256=P06TNhnqXa6JEje_B_94vAa5zqPYIVxMZAp6aZ4d80U,516
 adaptive_sdk/rest/rest_types.py,sha256=aSN7901_1goByBEl7Ka6onU2ia-_RhlmtH9suFkWkic,8859
-adaptive_sdk-0.1.10.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
-adaptive_sdk-0.1.10.dist-info/METADATA,sha256=I37LRW4IFFOwkSmVuEAh103DMmy9G8E1YUyJfjFaoQo,1447
-adaptive_sdk-0.1.10.dist-info/RECORD,,
+adaptive_sdk-0.1.12.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
+adaptive_sdk-0.1.12.dist-info/METADATA,sha256=XMmkasZbLbFIG-tmV-OtbSwt-Tgj4xjgRcFSeYBkhBw,1447
+adaptive_sdk-0.1.12.dist-info/RECORD,,