adaptive-sdk 0.1.10__tar.gz → 0.1.12__tar.gz

{adaptive_sdk-0.1.10 → adaptive_sdk-0.1.12}/PKG-INFO

@@ -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 → adaptive_sdk-0.1.12}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "flit_core.buildapi"
 
 [project]
 name = "adaptive-sdk"
-version = "0.1.10"
+version = "0.1.12"
 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.10 → adaptive_sdk-0.1.12}/src/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-0.1.10 → adaptive_sdk-0.1.12}/src/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-0.1.10 → adaptive_sdk-0.1.12}/src/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-0.1.10 → adaptive_sdk-0.1.12}/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.10 → adaptive_sdk-0.1.12}/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.10 → adaptive_sdk-0.1.12}/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.12/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.10 → adaptive_sdk-0.1.12}/src/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-0.1.10 → adaptive_sdk-0.1.12}/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.10 → adaptive_sdk-0.1.12}/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,
@@ -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/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()