adaptive-sdk 0.1.2__py3-none-any.whl → 0.1.3__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- adaptive_sdk/graphql_client/__init__.py +8 -7
- adaptive_sdk/graphql_client/add_hf_model.py +7 -1
- adaptive_sdk/graphql_client/async_client.py +15 -7
- adaptive_sdk/graphql_client/client.py +15 -7
- adaptive_sdk/graphql_client/custom_fields.py +154 -39
- adaptive_sdk/graphql_client/custom_mutations.py +2 -2
- adaptive_sdk/graphql_client/custom_queries.py +16 -6
- adaptive_sdk/graphql_client/custom_typing_fields.py +42 -7
- adaptive_sdk/graphql_client/describe_job.py +12 -0
- adaptive_sdk/graphql_client/enums.py +11 -5
- adaptive_sdk/graphql_client/fragments.py +23 -23
- adaptive_sdk/graphql_client/get_custom_recipe.py +2 -1
- adaptive_sdk/graphql_client/input_types.py +10 -2
- adaptive_sdk/graphql_client/list_compute_pools.py +3 -3
- adaptive_sdk/graphql_client/list_harmony_groups.py +13 -0
- adaptive_sdk/resources/jobs.py +16 -14
- adaptive_sdk/resources/models.py +77 -33
- adaptive_sdk/resources/recipes.py +159 -96
- adaptive_sdk/rest/rest_types.py +61 -37
- {adaptive_sdk-0.1.2.dist-info → adaptive_sdk-0.1.3.dist-info}/METADATA +1 -1
- {adaptive_sdk-0.1.2.dist-info → adaptive_sdk-0.1.3.dist-info}/RECORD +22 -21
- adaptive_sdk/graphql_client/list_partitions.py +0 -12
- {adaptive_sdk-0.1.2.dist-info → adaptive_sdk-0.1.3.dist-info}/WHEEL +0 -0
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
from .async_base_client_open_telemetry import AsyncBaseClientOpenTelemetry
|
|
2
2
|
from .async_client import AsyncGQLClient
|
|
3
3
|
from .add_external_model import AddExternalModel, AddExternalModelAddExternalModel
|
|
4
|
-
from .add_hf_model import AddHFModel
|
|
4
|
+
from .add_hf_model import AddHFModel, AddHFModelImportHfModel
|
|
5
5
|
from .add_model import AddModel, AddModelAddModel, AddModelAddModelBackbone
|
|
6
6
|
from .add_remote_env import AddRemoteEnv, AddRemoteEnvAddRemoteEnv
|
|
7
7
|
from .attach_model_to_use_case import AttachModelToUseCase, AttachModelToUseCaseAttachModel
|
|
@@ -30,32 +30,33 @@ from .deploy_model import DeployModel
|
|
|
30
30
|
from .describe_ab_campaign import DescribeAbCampaign, DescribeAbCampaignAbCampaign, DescribeAbCampaignAbCampaignReport
|
|
31
31
|
from .describe_dataset import DescribeDataset, DescribeDatasetDataset
|
|
32
32
|
from .describe_interaction import DescribeInteraction, DescribeInteractionCompletion
|
|
33
|
+
from .describe_job import DescribeJob, DescribeJobJob
|
|
33
34
|
from .describe_metric import DescribeMetric, DescribeMetricMetric
|
|
34
35
|
from .describe_metric_admin import DescribeMetricAdmin, DescribeMetricAdminMetric
|
|
35
36
|
from .describe_model import DescribeModel, DescribeModelModel, DescribeModelModelBackbone
|
|
36
37
|
from .describe_model_admin import DescribeModelAdmin, DescribeModelAdminModel, DescribeModelAdminModelBackbone
|
|
37
38
|
from .describe_use_case import DescribeUseCase, DescribeUseCaseUseCase
|
|
38
|
-
from .enums import AbcampaignStatus, AuthProviderKind, CompletionGroupBy, CompletionSource, CompletionSourceOutput, ComputePoolCapability, DatasetKind, DatasetSource, DateBucketUnit, EvaluationJobStatus, EvaluationType, ExternalModelProviderName, FeedbackType, FeedbackTypeOutput, GraderTypeEnum, JobArtifactKind, JobKind, JobStatus, JobStatusOutput, JudgeCapability, MetricAggregation, MetricKind, MetricScoringType, ModelKindFilter, ModelOnline, ModelserviceStatus, OpenAIModel,
|
|
39
|
+
from .enums import AbcampaignStatus, AuthProviderKind, BillingCycle, CompletionGroupBy, CompletionSource, CompletionSourceOutput, ComputePoolCapability, DatasetKind, DatasetSource, DateBucketUnit, EvaluationJobStatus, EvaluationType, ExternalModelProviderName, FeedbackType, FeedbackTypeOutput, GraderTypeEnum, HarmonyStatus, JobArtifactKind, JobKind, JobStatus, JobStatusOutput, JudgeCapability, MetricAggregation, MetricKind, MetricScoringType, ModelKindFilter, ModelOnline, ModelserviceStatus, OpenAIModel, PrebuiltCriteriaKey, Protocol, ProviderName, RemoteEnvStatus, SelectionTypeOutput, SortDirection, TimeseriesInterval, TrainingJobStatus, TrainingMetadataOutputAlignmentMethod, TrainingMetadataOutputTrainingType, UnitPosition
|
|
39
40
|
from .exceptions import GraphQLClientError, GraphQLClientGraphQLError, GraphQLClientGraphQLMultiError, GraphQLClientHttpError, GraphQLClientInvalidResponseError
|
|
40
|
-
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, JobData, JobDataCreatedBy, JobDataDetails, JobDataDetailsArtifacts, JobDataRecipe, JobDataStages, JobDataStagesInfoBatchInferenceJobStageOutput, JobDataStagesInfoEvalJobStageOutput, JobDataStagesInfoTrainingJobStageOutput, JobDataUseCase, JobStageOutputData, JobStageOutputDataInfoBatchInferenceJobStageOutput, JobStageOutputDataInfoEvalJobStageOutput, JobStageOutputDataInfoTrainingJobStageOutput, JudgeData, JudgeDataExamples, JudgeDataExamplesInput, JudgeDataMetric, JudgeDataModel, ListCompletionsFilterOutputData, ListCompletionsFilterOutputDataFeedbacks, ListCompletionsFilterOutputDataLabels, ListCompletionsFilterOutputDataTimerange, MetricData, MetricDataAdmin, MetricDataAdminUseCases, MetricWithContextData, ModelData, ModelDataAdmin, ModelDataAdminUseCases, ModelDataComputeConfig, ModelServiceData, ModelServiceDataModel, ModelServiceDataModelBackbone,
|
|
41
|
+
from .fragments import AbCampaignCreateData, AbCampaignDetailData, AbCampaignDetailDataMetric, AbCampaignDetailDataModels, AbCampaignDetailDataUseCase, AbCampaignReportData, AbCampaignReportDataVariants, AbCampaignReportDataVariantsComparisons, AbCampaignReportDataVariantsComparisonsVariant, AbCampaignReportDataVariantsInterval, AbCampaignReportDataVariantsVariant, CompletionComparisonFeedbackData, CompletionComparisonFeedbackDataModel, CompletionData, CompletionDataChatMessages, CompletionDataComparisonFeedbacks, CompletionDataComparisonFeedbacksMetric, CompletionDataComparisonFeedbacksOtherCompletion, CompletionDataComparisonFeedbacksPreferedCompletion, CompletionDataComparisonFeedbacksUsecase, CompletionDataDirectFeedbacks, CompletionDataDirectFeedbacksMetric, CompletionDataLabels, CompletionDataMetadata, CompletionDataMetadataUsage, CompletionDataModel, CustomRecipeData, CustomRecipeDataCreatedBy, CustomRecipeDataLabels, DatasetData, DatasetDataMetricsUsage, DatasetDataMetricsUsageMetric, GraderData, GraderDataGraderConfigCustomConfigOutput, GraderDataGraderConfigJudgeConfigOutput, GraderDataGraderConfigJudgeConfigOutputExamples, GraderDataGraderConfigJudgeConfigOutputExamplesInput, GraderDataGraderConfigJudgeConfigOutputModel, GraderDataGraderConfigPrebuiltConfigOutput, GraderDataGraderConfigPrebuiltConfigOutputModel, GraderDataGraderConfigPrebuiltConfigOutputPrebuiltCriteria, GraderDataGraderConfigRemoteConfigOutput, GraderDataMetric, GraderDataUseCase, HarmonyGroupData, HarmonyGroupDataComputePool, HarmonyGroupDataOnlineModels, JobData, JobDataCreatedBy, JobDataDetails, JobDataDetailsArtifacts, JobDataRecipe, JobDataStages, JobDataStagesInfoBatchInferenceJobStageOutput, JobDataStagesInfoEvalJobStageOutput, JobDataStagesInfoTrainingJobStageOutput, JobDataUseCase, JobStageOutputData, JobStageOutputDataInfoBatchInferenceJobStageOutput, JobStageOutputDataInfoEvalJobStageOutput, JobStageOutputDataInfoTrainingJobStageOutput, JudgeData, JudgeDataExamples, JudgeDataExamplesInput, JudgeDataMetric, JudgeDataModel, ListCompletionsFilterOutputData, ListCompletionsFilterOutputDataFeedbacks, ListCompletionsFilterOutputDataLabels, ListCompletionsFilterOutputDataTimerange, MetricData, MetricDataAdmin, MetricDataAdminUseCases, MetricWithContextData, ModelData, ModelDataAdmin, ModelDataAdminUseCases, ModelDataComputeConfig, ModelServiceData, ModelServiceDataModel, ModelServiceDataModelBackbone, RemoteEnvData, UseCaseData, UseCaseDataMetrics, UseCaseDataModelServices, UseCaseDataShares, UseCaseDataSharesRole, UseCaseDataSharesTeam, UserData, UserDataTeams, UserDataTeamsRole, UserDataTeamsTeam
|
|
41
42
|
from .get_custom_recipe import GetCustomRecipe, GetCustomRecipeCustomRecipe
|
|
42
43
|
from .get_grader import GetGrader, GetGraderGrader
|
|
43
44
|
from .get_judge import GetJudge, GetJudgeJudge
|
|
44
|
-
from .input_types import AbcampaignCreate, AbCampaignFilter, AddExternalModelInput, AddHFModelInput, AddModelInput, AnthropicProviderDataInput, ApiKeyCreate, ArtifactFilter, AttachModel, AzureProviderDataInput, CompletionComparisonFilterInput, CompletionFeedbackFilterInput, CompletionLabelValue, CreateRecipeInput, CreateToolProviderInput, CursorPageInput, CustomConfigInput, CustomRecipeFilterInput, DatasetCreate, DatasetGenerate, DatasetGenerationConfig, EmojiInput, FeedbackAddInput, FeedbackFilterInput, FeedbackUpdateInput, GoogleProviderDataInput, GraderConfigInput, GraderCreateInput, GraderUpdateInput, JobArtifactFilter, JobInput, JudgeConfigInput, JudgeCreate, JudgeExampleInput, JudgeExampleInputTurnEntry, JudgeUpdate, LabelFilter, LabelInput, ListCompletionsFilterInput, ListJobsFilterInput, MetricCreate, MetricGetOrCreate, MetricLink, MetricTrendInput, MetricUnlink, ModelComputeConfigInput, ModelFilter, ModelPlacementInput, ModelProviderDataInput, ModelServiceDisconnect, ModelServiceFilter, NvidiaProviderDataInput, OpenAIProviderDataInput, OrderPair, PrebuiltConfigInput, PrebuiltJudgeCreate, RagdataGenerationConfig, RemoteConfigInput, RemoteEnvCreate, RoleCreate, SystemPromptTemplateCreate, SystemPromptTemplateUpdate, TeamCreate, TeamMemberRemove, TeamMemberSet, TimeRange, TimeseriesInput, UnitConfigInput, UpdateCompletion, UpdateModelService, UpdateRecipeInput, UpdateToolProviderInput, UsageFilterInput, UsagePerUseCaseFilterInput, UseCaseCreate, UseCaseFilter, UseCaseMetadataInput, UseCaseSettingsInput, UseCaseShareInput, UseCaseShares, UseCaseUpdate, UserCreate, UserCreateTeamWithRole, WidgetInput
|
|
45
|
+
from .input_types import AbcampaignCreate, AbCampaignFilter, AddExternalModelInput, AddHFModelInput, AddModelInput, AnthropicProviderDataInput, ApiKeyCreate, ArtifactFilter, AttachModel, AzureProviderDataInput, CompletionComparisonFilterInput, CompletionFeedbackFilterInput, CompletionLabelValue, CreateRecipeInput, CreateToolProviderInput, CursorPageInput, CustomConfigInput, CustomRecipeFilterInput, DatasetCreate, DatasetGenerate, DatasetGenerationConfig, EmojiInput, FeedbackAddInput, FeedbackFilterInput, FeedbackUpdateInput, GlobalUsageFilterInput, GoogleProviderDataInput, GraderConfigInput, GraderCreateInput, GraderUpdateInput, JobArtifactFilter, JobInput, JudgeConfigInput, JudgeCreate, JudgeExampleInput, JudgeExampleInputTurnEntry, JudgeUpdate, LabelFilter, LabelInput, ListCompletionsFilterInput, ListJobsFilterInput, MetricCreate, MetricGetOrCreate, MetricLink, MetricTrendInput, MetricUnlink, ModelComputeConfigInput, ModelFilter, ModelPlacementInput, ModelProviderDataInput, ModelServiceDisconnect, ModelServiceFilter, NvidiaProviderDataInput, OpenAIProviderDataInput, OrderPair, PrebuiltConfigInput, PrebuiltJudgeCreate, RagdataGenerationConfig, RemoteConfigInput, RemoteEnvCreate, RoleCreate, SystemPromptTemplateCreate, SystemPromptTemplateUpdate, TeamCreate, TeamMemberRemove, TeamMemberSet, TimeRange, TimeseriesInput, UnitConfigInput, UpdateCompletion, UpdateModelService, UpdateRecipeInput, UpdateToolProviderInput, UsageFilterInput, UsagePerUseCaseFilterInput, UseCaseCreate, UseCaseFilter, UseCaseMetadataInput, UseCaseSettingsInput, UseCaseShareInput, UseCaseShares, UseCaseUpdate, UserCreate, UserCreateTeamWithRole, WidgetInput
|
|
45
46
|
from .link_metric import LinkMetric, LinkMetricLinkMetric
|
|
46
47
|
from .list_ab_campaigns import ListAbCampaigns, ListAbCampaignsAbCampaigns
|
|
47
|
-
from .list_compute_pools import ListComputePools, ListComputePoolsComputePools,
|
|
48
|
+
from .list_compute_pools import ListComputePools, ListComputePoolsComputePools, ListComputePoolsComputePoolsHarmonyGroups
|
|
48
49
|
from .list_custom_recipes import ListCustomRecipes, ListCustomRecipesCustomRecipes
|
|
49
50
|
from .list_datasets import ListDatasets, ListDatasetsDatasets
|
|
50
51
|
from .list_graders import ListGraders, ListGradersGraders
|
|
51
52
|
from .list_grouped_interactions import ListGroupedInteractions, ListGroupedInteractionsCompletionsGrouped, ListGroupedInteractionsCompletionsGroupedNodes, ListGroupedInteractionsCompletionsGroupedNodesCompletions, ListGroupedInteractionsCompletionsGroupedNodesCompletionsNodes, ListGroupedInteractionsCompletionsGroupedNodesDirectFeedbacksStats, ListGroupedInteractionsCompletionsGroupedNodesDirectFeedbacksStatsMetric, ListGroupedInteractionsCompletionsGroupedPageInfo
|
|
53
|
+
from .list_harmony_groups import ListHarmonyGroups, ListHarmonyGroupsHarmonyGroups
|
|
52
54
|
from .list_interactions import ListInteractions, ListInteractionsCompletions, ListInteractionsCompletionsNodes, ListInteractionsCompletionsPageInfo
|
|
53
55
|
from .list_jobs import ListJobs, ListJobsJobs, ListJobsJobsNodes, ListJobsJobsPageInfo
|
|
54
56
|
from .list_judge_versions import ListJudgeVersions, ListJudgeVersionsJudgeVersions
|
|
55
57
|
from .list_judges import ListJudges, ListJudgesJudges
|
|
56
58
|
from .list_metrics import ListMetrics, ListMetricsMetrics
|
|
57
59
|
from .list_models import ListModels, ListModelsModels, ListModelsModelsBackbone
|
|
58
|
-
from .list_partitions import ListPartitions, ListPartitionsPartitions
|
|
59
60
|
from .list_permissions import ListPermissions
|
|
60
61
|
from .list_remote_envs import ListRemoteEnvs, ListRemoteEnvsRemoteEnvs
|
|
61
62
|
from .list_roles import ListRoles, ListRolesRoles
|
|
@@ -77,4 +78,4 @@ from .update_judge import UpdateJudge, UpdateJudgeUpdateJudge
|
|
|
77
78
|
from .update_model import UpdateModel, UpdateModelUpdateModelService
|
|
78
79
|
from .update_model_compute_config import UpdateModelComputeConfig, UpdateModelComputeConfigUpdateModelComputeConfig
|
|
79
80
|
from .update_user import UpdateUser, UpdateUserSetTeamMember, UpdateUserSetTeamMemberRole, UpdateUserSetTeamMemberTeam, UpdateUserSetTeamMemberUser
|
|
80
|
-
__all__ = ['AbCampaignCreateData', 'AbCampaignDetailData', 'AbCampaignDetailDataMetric', 'AbCampaignDetailDataModels', 'AbCampaignDetailDataUseCase', 'AbCampaignFilter', 'AbCampaignReportData', 'AbCampaignReportDataVariants', 'AbCampaignReportDataVariantsComparisons', 'AbCampaignReportDataVariantsComparisonsVariant', 'AbCampaignReportDataVariantsInterval', 'AbCampaignReportDataVariantsVariant', 'AbcampaignCreate', 'AbcampaignStatus', 'AddExternalModel', 'AddExternalModelAddExternalModel', 'AddExternalModelInput', 'AddHFModel', 'AddHFModelInput', 'AddModel', 'AddModelAddModel', 'AddModelAddModelBackbone', 'AddModelInput', 'AddRemoteEnv', 'AddRemoteEnvAddRemoteEnv', 'AnthropicProviderDataInput', 'ApiKeyCreate', 'ArtifactFilter', 'AsyncBaseClientOpenTelemetry', 'AsyncGQLClient', 'AttachModel', 'AttachModelToUseCase', 'AttachModelToUseCaseAttachModel', 'AuthProviderKind', 'AzureProviderDataInput', 'BaseClientOpenTelemetry', 'BaseModel', 'CancelABCampaign', 'CancelJob', 'CancelJobCancelJob', 'CompletionComparisonFeedbackData', 'CompletionComparisonFeedbackDataModel', 'CompletionComparisonFilterInput', 'CompletionData', 'CompletionDataChatMessages', 'CompletionDataComparisonFeedbacks', 'CompletionDataComparisonFeedbacksMetric', 'CompletionDataComparisonFeedbacksOtherCompletion', 'CompletionDataComparisonFeedbacksPreferedCompletion', 'CompletionDataComparisonFeedbacksUsecase', 'CompletionDataDirectFeedbacks', 'CompletionDataDirectFeedbacksMetric', 'CompletionDataLabels', 'CompletionDataMetadata', 'CompletionDataMetadataUsage', 'CompletionDataModel', 'CompletionFeedbackFilterInput', 'CompletionGroupBy', 'CompletionLabelValue', 'CompletionSource', 'CompletionSourceOutput', 'ComputePoolCapability', 'CreateAbCampaign', 'CreateAbCampaignCreateAbCampaign', 'CreateCustomRecipe', 'CreateCustomRecipeCreateCustomRecipe', 'CreateGrader', 'CreateGraderCreateGrader', 'CreateJob', 'CreateJobCreateJob', 'CreateJudge', 'CreateJudgeCreateJudge', 'CreateMetric', 'CreateMetricCreateMetric', 'CreatePrebuiltJudge', 'CreatePrebuiltJudgeCreatePrebuiltJudge', 'CreateRecipeInput', 'CreateRole', 'CreateRoleCreateRole', 'CreateTeam', 'CreateTeamCreateTeam', 'CreateToolProviderInput', 'CreateUseCase', 'CreateUseCaseCreateUseCase', 'CreateUser', 'CreateUserCreateUser', 'CursorPageInput', 'CustomConfigInput', 'CustomRecipeData', 'CustomRecipeDataCreatedBy', 'CustomRecipeDataLabels', 'CustomRecipeFilterInput', 'DatasetCreate', 'DatasetData', 'DatasetDataMetricsUsage', 'DatasetDataMetricsUsageMetric', 'DatasetGenerate', 'DatasetGenerationConfig', 'DatasetKind', 'DatasetSource', 'DateBucketUnit', 'DeleteCustomRecipe', 'DeleteDataset', 'DeleteGrader', 'DeleteGraderDeleteGrader', 'DeleteJudge', 'DeleteJudgeDeleteJudge', 'DeleteUser', 'DeleteUserDeleteUser', 'DeployModel', 'DescribeAbCampaign', 'DescribeAbCampaignAbCampaign', 'DescribeAbCampaignAbCampaignReport', 'DescribeDataset', 'DescribeDatasetDataset', 'DescribeInteraction', 'DescribeInteractionCompletion', 'DescribeMetric', 'DescribeMetricAdmin', 'DescribeMetricAdminMetric', 'DescribeMetricMetric', 'DescribeModel', 'DescribeModelAdmin', 'DescribeModelAdminModel', 'DescribeModelAdminModelBackbone', 'DescribeModelModel', 'DescribeModelModelBackbone', 'DescribeUseCase', 'DescribeUseCaseUseCase', 'EmojiInput', 'EvaluationJobStatus', 'EvaluationType', 'ExternalModelProviderName', 'FeedbackAddInput', 'FeedbackFilterInput', 'FeedbackType', 'FeedbackTypeOutput', 'FeedbackUpdateInput', 'GQLClient', 'GetCustomRecipe', 'GetCustomRecipeCustomRecipe', 'GetGrader', 'GetGraderGrader', 'GetJudge', 'GetJudgeJudge', 'GoogleProviderDataInput', 'GraderConfigInput', 'GraderCreateInput', 'GraderData', 'GraderDataGraderConfigCustomConfigOutput', 'GraderDataGraderConfigJudgeConfigOutput', 'GraderDataGraderConfigJudgeConfigOutputExamples', 'GraderDataGraderConfigJudgeConfigOutputExamplesInput', 'GraderDataGraderConfigJudgeConfigOutputModel', 'GraderDataGraderConfigPrebuiltConfigOutput', 'GraderDataGraderConfigPrebuiltConfigOutputModel', 'GraderDataGraderConfigPrebuiltConfigOutputPrebuiltCriteria', 'GraderDataGraderConfigRemoteConfigOutput', 'GraderDataMetric', 'GraderDataUseCase', 'GraderTypeEnum', 'GraderUpdateInput', 'GraphQLClientError', 'GraphQLClientGraphQLError', 'GraphQLClientGraphQLMultiError', 'GraphQLClientHttpError', 'GraphQLClientInvalidResponseError', 'JobArtifactFilter', 'JobArtifactKind', 'JobData', 'JobDataCreatedBy', 'JobDataDetails', 'JobDataDetailsArtifacts', 'JobDataRecipe', 'JobDataStages', 'JobDataStagesInfoBatchInferenceJobStageOutput', 'JobDataStagesInfoEvalJobStageOutput', 'JobDataStagesInfoTrainingJobStageOutput', 'JobDataUseCase', 'JobInput', 'JobKind', 'JobStageOutputData', 'JobStageOutputDataInfoBatchInferenceJobStageOutput', 'JobStageOutputDataInfoEvalJobStageOutput', 'JobStageOutputDataInfoTrainingJobStageOutput', 'JobStatus', 'JobStatusOutput', 'JudgeCapability', 'JudgeConfigInput', 'JudgeCreate', 'JudgeData', 'JudgeDataExamples', 'JudgeDataExamplesInput', 'JudgeDataMetric', 'JudgeDataModel', 'JudgeExampleInput', 'JudgeExampleInputTurnEntry', 'JudgeUpdate', 'LabelFilter', 'LabelInput', 'LinkMetric', 'LinkMetricLinkMetric', 'ListAbCampaigns', 'ListAbCampaignsAbCampaigns', 'ListCompletionsFilterInput', 'ListCompletionsFilterOutputData', 'ListCompletionsFilterOutputDataFeedbacks', 'ListCompletionsFilterOutputDataLabels', 'ListCompletionsFilterOutputDataTimerange', 'ListComputePools', 'ListComputePoolsComputePools', '
|
|
81
|
+
__all__ = ['AbCampaignCreateData', 'AbCampaignDetailData', 'AbCampaignDetailDataMetric', 'AbCampaignDetailDataModels', 'AbCampaignDetailDataUseCase', 'AbCampaignFilter', 'AbCampaignReportData', 'AbCampaignReportDataVariants', 'AbCampaignReportDataVariantsComparisons', 'AbCampaignReportDataVariantsComparisonsVariant', 'AbCampaignReportDataVariantsInterval', 'AbCampaignReportDataVariantsVariant', 'AbcampaignCreate', 'AbcampaignStatus', 'AddExternalModel', 'AddExternalModelAddExternalModel', 'AddExternalModelInput', 'AddHFModel', 'AddHFModelImportHfModel', 'AddHFModelInput', 'AddModel', 'AddModelAddModel', 'AddModelAddModelBackbone', 'AddModelInput', 'AddRemoteEnv', 'AddRemoteEnvAddRemoteEnv', 'AnthropicProviderDataInput', 'ApiKeyCreate', 'ArtifactFilter', 'AsyncBaseClientOpenTelemetry', 'AsyncGQLClient', 'AttachModel', 'AttachModelToUseCase', 'AttachModelToUseCaseAttachModel', 'AuthProviderKind', 'AzureProviderDataInput', 'BaseClientOpenTelemetry', 'BaseModel', 'BillingCycle', 'CancelABCampaign', 'CancelJob', 'CancelJobCancelJob', 'CompletionComparisonFeedbackData', 'CompletionComparisonFeedbackDataModel', 'CompletionComparisonFilterInput', 'CompletionData', 'CompletionDataChatMessages', 'CompletionDataComparisonFeedbacks', 'CompletionDataComparisonFeedbacksMetric', 'CompletionDataComparisonFeedbacksOtherCompletion', 'CompletionDataComparisonFeedbacksPreferedCompletion', 'CompletionDataComparisonFeedbacksUsecase', 'CompletionDataDirectFeedbacks', 'CompletionDataDirectFeedbacksMetric', 'CompletionDataLabels', 'CompletionDataMetadata', 'CompletionDataMetadataUsage', 'CompletionDataModel', 'CompletionFeedbackFilterInput', 'CompletionGroupBy', 'CompletionLabelValue', 'CompletionSource', 'CompletionSourceOutput', 'ComputePoolCapability', 'CreateAbCampaign', 'CreateAbCampaignCreateAbCampaign', 'CreateCustomRecipe', 'CreateCustomRecipeCreateCustomRecipe', 'CreateGrader', 'CreateGraderCreateGrader', 'CreateJob', 'CreateJobCreateJob', 'CreateJudge', 'CreateJudgeCreateJudge', 'CreateMetric', 'CreateMetricCreateMetric', 'CreatePrebuiltJudge', 'CreatePrebuiltJudgeCreatePrebuiltJudge', 'CreateRecipeInput', 'CreateRole', 'CreateRoleCreateRole', 'CreateTeam', 'CreateTeamCreateTeam', 'CreateToolProviderInput', 'CreateUseCase', 'CreateUseCaseCreateUseCase', 'CreateUser', 'CreateUserCreateUser', 'CursorPageInput', 'CustomConfigInput', 'CustomRecipeData', 'CustomRecipeDataCreatedBy', 'CustomRecipeDataLabels', 'CustomRecipeFilterInput', 'DatasetCreate', 'DatasetData', 'DatasetDataMetricsUsage', 'DatasetDataMetricsUsageMetric', 'DatasetGenerate', 'DatasetGenerationConfig', 'DatasetKind', 'DatasetSource', 'DateBucketUnit', 'DeleteCustomRecipe', 'DeleteDataset', 'DeleteGrader', 'DeleteGraderDeleteGrader', 'DeleteJudge', 'DeleteJudgeDeleteJudge', 'DeleteUser', 'DeleteUserDeleteUser', 'DeployModel', 'DescribeAbCampaign', 'DescribeAbCampaignAbCampaign', 'DescribeAbCampaignAbCampaignReport', 'DescribeDataset', 'DescribeDatasetDataset', 'DescribeInteraction', 'DescribeInteractionCompletion', 'DescribeJob', 'DescribeJobJob', 'DescribeMetric', 'DescribeMetricAdmin', 'DescribeMetricAdminMetric', 'DescribeMetricMetric', 'DescribeModel', 'DescribeModelAdmin', 'DescribeModelAdminModel', 'DescribeModelAdminModelBackbone', 'DescribeModelModel', 'DescribeModelModelBackbone', 'DescribeUseCase', 'DescribeUseCaseUseCase', 'EmojiInput', 'EvaluationJobStatus', 'EvaluationType', 'ExternalModelProviderName', 'FeedbackAddInput', 'FeedbackFilterInput', 'FeedbackType', 'FeedbackTypeOutput', 'FeedbackUpdateInput', 'GQLClient', 'GetCustomRecipe', 'GetCustomRecipeCustomRecipe', 'GetGrader', 'GetGraderGrader', 'GetJudge', 'GetJudgeJudge', 'GlobalUsageFilterInput', 'GoogleProviderDataInput', 'GraderConfigInput', 'GraderCreateInput', 'GraderData', 'GraderDataGraderConfigCustomConfigOutput', 'GraderDataGraderConfigJudgeConfigOutput', 'GraderDataGraderConfigJudgeConfigOutputExamples', 'GraderDataGraderConfigJudgeConfigOutputExamplesInput', 'GraderDataGraderConfigJudgeConfigOutputModel', 'GraderDataGraderConfigPrebuiltConfigOutput', 'GraderDataGraderConfigPrebuiltConfigOutputModel', 'GraderDataGraderConfigPrebuiltConfigOutputPrebuiltCriteria', 'GraderDataGraderConfigRemoteConfigOutput', 'GraderDataMetric', 'GraderDataUseCase', 'GraderTypeEnum', 'GraderUpdateInput', 'GraphQLClientError', 'GraphQLClientGraphQLError', 'GraphQLClientGraphQLMultiError', 'GraphQLClientHttpError', 'GraphQLClientInvalidResponseError', 'HarmonyGroupData', 'HarmonyGroupDataComputePool', 'HarmonyGroupDataOnlineModels', 'HarmonyStatus', 'JobArtifactFilter', 'JobArtifactKind', 'JobData', 'JobDataCreatedBy', 'JobDataDetails', 'JobDataDetailsArtifacts', 'JobDataRecipe', 'JobDataStages', 'JobDataStagesInfoBatchInferenceJobStageOutput', 'JobDataStagesInfoEvalJobStageOutput', 'JobDataStagesInfoTrainingJobStageOutput', 'JobDataUseCase', 'JobInput', 'JobKind', 'JobStageOutputData', 'JobStageOutputDataInfoBatchInferenceJobStageOutput', 'JobStageOutputDataInfoEvalJobStageOutput', 'JobStageOutputDataInfoTrainingJobStageOutput', 'JobStatus', 'JobStatusOutput', 'JudgeCapability', 'JudgeConfigInput', 'JudgeCreate', 'JudgeData', 'JudgeDataExamples', 'JudgeDataExamplesInput', 'JudgeDataMetric', 'JudgeDataModel', 'JudgeExampleInput', 'JudgeExampleInputTurnEntry', 'JudgeUpdate', 'LabelFilter', 'LabelInput', 'LinkMetric', 'LinkMetricLinkMetric', 'ListAbCampaigns', 'ListAbCampaignsAbCampaigns', 'ListCompletionsFilterInput', 'ListCompletionsFilterOutputData', 'ListCompletionsFilterOutputDataFeedbacks', 'ListCompletionsFilterOutputDataLabels', 'ListCompletionsFilterOutputDataTimerange', 'ListComputePools', 'ListComputePoolsComputePools', 'ListComputePoolsComputePoolsHarmonyGroups', 'ListCustomRecipes', 'ListCustomRecipesCustomRecipes', 'ListDatasets', 'ListDatasetsDatasets', 'ListGraders', 'ListGradersGraders', 'ListGroupedInteractions', 'ListGroupedInteractionsCompletionsGrouped', 'ListGroupedInteractionsCompletionsGroupedNodes', 'ListGroupedInteractionsCompletionsGroupedNodesCompletions', 'ListGroupedInteractionsCompletionsGroupedNodesCompletionsNodes', 'ListGroupedInteractionsCompletionsGroupedNodesDirectFeedbacksStats', 'ListGroupedInteractionsCompletionsGroupedNodesDirectFeedbacksStatsMetric', 'ListGroupedInteractionsCompletionsGroupedPageInfo', 'ListHarmonyGroups', 'ListHarmonyGroupsHarmonyGroups', 'ListInteractions', 'ListInteractionsCompletions', 'ListInteractionsCompletionsNodes', 'ListInteractionsCompletionsPageInfo', 'ListJobs', 'ListJobsFilterInput', 'ListJobsJobs', 'ListJobsJobsNodes', 'ListJobsJobsPageInfo', 'ListJudgeVersions', 'ListJudgeVersionsJudgeVersions', 'ListJudges', 'ListJudgesJudges', 'ListMetrics', 'ListMetricsMetrics', 'ListModels', 'ListModelsModels', 'ListModelsModelsBackbone', 'ListPermissions', 'ListRemoteEnvs', 'ListRemoteEnvsRemoteEnvs', 'ListRoles', 'ListRolesRoles', 'ListTeams', 'ListTeamsTeams', 'ListUseCases', 'ListUseCasesUseCases', 'ListUsers', 'ListUsersUsers', 'LoadDataset', 'LoadDatasetCreateDataset', 'LockGrader', 'LockGraderLockGrader', 'Me', 'MeMe', 'MeMeApiKeys', 'MetricAggregation', 'MetricCreate', 'MetricData', 'MetricDataAdmin', 'MetricDataAdminUseCases', 'MetricGetOrCreate', 'MetricKind', 'MetricLink', 'MetricScoringType', 'MetricTrendInput', 'MetricUnlink', 'MetricWithContextData', 'ModelComputeConfigInput', 'ModelData', 'ModelDataAdmin', 'ModelDataAdminUseCases', 'ModelDataComputeConfig', 'ModelFilter', 'ModelKindFilter', 'ModelOnline', 'ModelPlacementInput', 'ModelProviderDataInput', 'ModelServiceData', 'ModelServiceDataModel', 'ModelServiceDataModelBackbone', 'ModelServiceDisconnect', 'ModelServiceFilter', 'ModelserviceStatus', 'NvidiaProviderDataInput', 'OpenAIModel', 'OpenAIProviderDataInput', 'OrderPair', 'PrebuiltConfigInput', 'PrebuiltCriteriaKey', 'PrebuiltJudgeCreate', 'Protocol', 'ProviderName', 'RagdataGenerationConfig', 'RemoteConfigInput', 'RemoteEnvCreate', 'RemoteEnvData', 'RemoteEnvStatus', 'RemoveRemoteEnv', 'RemoveTeamMember', 'RemoveTeamMemberRemoveTeamMember', 'RoleCreate', 'SelectionTypeOutput', 'ShareUseCase', 'ShareUseCaseShareUseCase', 'SortDirection', 'SystemPromptTemplateCreate', 'SystemPromptTemplateUpdate', 'TeamCreate', 'TeamMemberRemove', 'TeamMemberSet', 'TerminateModel', 'TestRemoteEnv', 'TestRemoteEnvTestRemoteEnvRemoteEnvTestOffline', 'TestRemoteEnvTestRemoteEnvRemoteEnvTestOnline', 'TimeRange', 'TimeseriesInput', 'TimeseriesInterval', 'TrainingJobStatus', 'TrainingMetadataOutputAlignmentMethod', 'TrainingMetadataOutputTrainingType', 'UnitConfigInput', 'UnitPosition', 'UnlinkMetric', 'UpdateCompletion', 'UpdateCustomRecipe', 'UpdateCustomRecipeUpdateCustomRecipe', 'UpdateGrader', 'UpdateGraderUpdateGrader', 'UpdateJudge', 'UpdateJudgeUpdateJudge', 'UpdateModel', 'UpdateModelComputeConfig', 'UpdateModelComputeConfigUpdateModelComputeConfig', 'UpdateModelService', 'UpdateModelUpdateModelService', 'UpdateRecipeInput', 'UpdateToolProviderInput', 'UpdateUser', 'UpdateUserSetTeamMember', 'UpdateUserSetTeamMemberRole', 'UpdateUserSetTeamMemberTeam', 'UpdateUserSetTeamMemberUser', 'Upload', 'UsageFilterInput', 'UsagePerUseCaseFilterInput', 'UseCaseCreate', 'UseCaseData', 'UseCaseDataMetrics', 'UseCaseDataModelServices', 'UseCaseDataShares', 'UseCaseDataSharesRole', 'UseCaseDataSharesTeam', 'UseCaseFilter', 'UseCaseMetadataInput', 'UseCaseSettingsInput', 'UseCaseShareInput', 'UseCaseShares', 'UseCaseUpdate', 'UserCreate', 'UserCreateTeamWithRole', 'UserData', 'UserDataTeams', 'UserDataTeamsRole', 'UserDataTeamsTeam', 'WidgetInput']
|
|
@@ -1,6 +1,12 @@
|
|
|
1
1
|
from pydantic import Field
|
|
2
2
|
from .base_model import BaseModel
|
|
3
|
+
from .fragments import JobData
|
|
3
4
|
|
|
4
5
|
class AddHFModel(BaseModel):
|
|
5
6
|
"""@public"""
|
|
6
|
-
import_hf_model:
|
|
7
|
+
import_hf_model: 'AddHFModelImportHfModel' = Field(alias='importHfModel')
|
|
8
|
+
|
|
9
|
+
class AddHFModelImportHfModel(JobData):
|
|
10
|
+
"""@public"""
|
|
11
|
+
pass
|
|
12
|
+
AddHFModel.model_rebuild()
|
|
@@ -30,6 +30,7 @@ from .deploy_model import DeployModel
|
|
|
30
30
|
from .describe_ab_campaign import DescribeAbCampaign
|
|
31
31
|
from .describe_dataset import DescribeDataset
|
|
32
32
|
from .describe_interaction import DescribeInteraction
|
|
33
|
+
from .describe_job import DescribeJob
|
|
33
34
|
from .describe_metric import DescribeMetric
|
|
34
35
|
from .describe_metric_admin import DescribeMetricAdmin
|
|
35
36
|
from .describe_model import DescribeModel
|
|
@@ -47,13 +48,13 @@ from .list_custom_recipes import ListCustomRecipes
|
|
|
47
48
|
from .list_datasets import ListDatasets
|
|
48
49
|
from .list_graders import ListGraders
|
|
49
50
|
from .list_grouped_interactions import ListGroupedInteractions
|
|
51
|
+
from .list_harmony_groups import ListHarmonyGroups
|
|
50
52
|
from .list_interactions import ListInteractions
|
|
51
53
|
from .list_jobs import ListJobs
|
|
52
54
|
from .list_judge_versions import ListJudgeVersions
|
|
53
55
|
from .list_judges import ListJudges
|
|
54
56
|
from .list_metrics import ListMetrics
|
|
55
57
|
from .list_models import ListModels
|
|
56
|
-
from .list_partitions import ListPartitions
|
|
57
58
|
from .list_permissions import ListPermissions
|
|
58
59
|
from .list_remote_envs import ListRemoteEnvs
|
|
59
60
|
from .list_roles import ListRoles
|
|
@@ -188,7 +189,7 @@ class AsyncGQLClient(AsyncBaseClientOpenTelemetry):
|
|
|
188
189
|
return DeleteDataset.model_validate(data)
|
|
189
190
|
|
|
190
191
|
async def add_hf_model(self, input: AddHFModelInput, **kwargs: Any) -> AddHFModel:
|
|
191
|
-
query = gql('\n mutation AddHFModel($input: AddHFModelInput!) {\n importHfModel(input: $input)\n }\n ')
|
|
192
|
+
query = gql('\n mutation AddHFModel($input: AddHFModelInput!) {\n importHfModel(input: $input) {\n ...JobData\n }\n }\n\n fragment CustomRecipeData on CustomRecipe {\n id\n key\n name\n content\n contentHash\n editable\n global\n builtin\n inputSchema\n jsonSchema\n description\n labels {\n key\n value\n }\n createdAt\n updatedAt\n createdBy {\n id\n name\n email\n }\n }\n\n fragment JobData on Job {\n id\n name\n status\n createdAt\n createdBy {\n id\n name\n }\n startedAt\n endedAt\n durationMs\n progress\n error\n kind\n stages {\n name\n status\n info {\n __typename\n ... on TrainingJobStageOutput {\n monitoringLink\n totalNumSamples\n processedNumSamples\n checkpoints\n }\n ... on EvalJobStageOutput {\n totalNumSamples\n processedNumSamples\n }\n ... on BatchInferenceJobStageOutput {\n totalNumSamples\n processedNumSamples\n }\n }\n }\n useCase {\n id\n key\n name\n }\n recipe {\n ...CustomRecipeData\n }\n details {\n args\n recipeHash\n artifacts {\n id\n name\n kind\n uri\n metadata\n createdAt\n }\n }\n }\n ')
|
|
192
193
|
variables: Dict[str, object] = {'input': input}
|
|
193
194
|
response = await self.execute(query=query, operation_name='AddHFModel', variables=variables, **kwargs)
|
|
194
195
|
data = self.get_data(response)
|
|
@@ -495,15 +496,15 @@ class AsyncGQLClient(AsyncBaseClientOpenTelemetry):
|
|
|
495
496
|
data = self.get_data(response)
|
|
496
497
|
return ListPermissions.model_validate(data)
|
|
497
498
|
|
|
498
|
-
async def
|
|
499
|
-
query = gql('\n query
|
|
499
|
+
async def list_harmony_groups(self, **kwargs: Any) -> ListHarmonyGroups:
|
|
500
|
+
query = gql('\n query ListHarmonyGroups {\n harmonyGroups {\n ...HarmonyGroupData\n }\n }\n\n fragment HarmonyGroupData on HarmonyGroup {\n id\n key\n computePool {\n key\n name\n }\n status\n url\n worldSize\n gpuTypes\n createdAt\n onlineModels {\n ...ModelData\n }\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n kind\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n ')
|
|
500
501
|
variables: Dict[str, object] = {}
|
|
501
|
-
response = await self.execute(query=query, operation_name='
|
|
502
|
+
response = await self.execute(query=query, operation_name='ListHarmonyGroups', variables=variables, **kwargs)
|
|
502
503
|
data = self.get_data(response)
|
|
503
|
-
return
|
|
504
|
+
return ListHarmonyGroups.model_validate(data)
|
|
504
505
|
|
|
505
506
|
async def list_compute_pools(self, **kwargs: Any) -> ListComputePools:
|
|
506
|
-
query = gql('\n query ListComputePools {\n computePools {\n id\n key\n name\n createdAt\n capabilities\n
|
|
507
|
+
query = gql('\n query ListComputePools {\n computePools {\n id\n key\n name\n createdAt\n capabilities\n harmonyGroups {\n ...HarmonyGroupData\n }\n }\n }\n\n fragment HarmonyGroupData on HarmonyGroup {\n id\n key\n computePool {\n key\n name\n }\n status\n url\n worldSize\n gpuTypes\n createdAt\n onlineModels {\n ...ModelData\n }\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n kind\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n ')
|
|
507
508
|
variables: Dict[str, object] = {}
|
|
508
509
|
response = await self.execute(query=query, operation_name='ListComputePools', variables=variables, **kwargs)
|
|
509
510
|
data = self.get_data(response)
|
|
@@ -572,6 +573,13 @@ class AsyncGQLClient(AsyncBaseClientOpenTelemetry):
|
|
|
572
573
|
data = self.get_data(response)
|
|
573
574
|
return ListJobs.model_validate(data)
|
|
574
575
|
|
|
576
|
+
async def describe_job(self, id: Any, **kwargs: Any) -> DescribeJob:
|
|
577
|
+
query = gql('\n query DescribeJob($id: UUID!) {\n job(id: $id) {\n ...JobData\n }\n }\n\n fragment CustomRecipeData on CustomRecipe {\n id\n key\n name\n content\n contentHash\n editable\n global\n builtin\n inputSchema\n jsonSchema\n description\n labels {\n key\n value\n }\n createdAt\n updatedAt\n createdBy {\n id\n name\n email\n }\n }\n\n fragment JobData on Job {\n id\n name\n status\n createdAt\n createdBy {\n id\n name\n }\n startedAt\n endedAt\n durationMs\n progress\n error\n kind\n stages {\n name\n status\n info {\n __typename\n ... on TrainingJobStageOutput {\n monitoringLink\n totalNumSamples\n processedNumSamples\n checkpoints\n }\n ... on EvalJobStageOutput {\n totalNumSamples\n processedNumSamples\n }\n ... on BatchInferenceJobStageOutput {\n totalNumSamples\n processedNumSamples\n }\n }\n }\n useCase {\n id\n key\n name\n }\n recipe {\n ...CustomRecipeData\n }\n details {\n args\n recipeHash\n artifacts {\n id\n name\n kind\n uri\n metadata\n createdAt\n }\n }\n }\n ')
|
|
578
|
+
variables: Dict[str, object] = {'id': id}
|
|
579
|
+
response = await self.execute(query=query, operation_name='DescribeJob', variables=variables, **kwargs)
|
|
580
|
+
data = self.get_data(response)
|
|
581
|
+
return DescribeJob.model_validate(data)
|
|
582
|
+
|
|
575
583
|
async def execute_custom_operation(self, *fields: GraphQLField, operation_type: OperationType, operation_name: str) -> Dict[str, Any]:
|
|
576
584
|
selections = self._build_selection_set(fields)
|
|
577
585
|
combined_variables = self._combine_variables(fields)
|
|
@@ -30,6 +30,7 @@ from .deploy_model import DeployModel
|
|
|
30
30
|
from .describe_ab_campaign import DescribeAbCampaign
|
|
31
31
|
from .describe_dataset import DescribeDataset
|
|
32
32
|
from .describe_interaction import DescribeInteraction
|
|
33
|
+
from .describe_job import DescribeJob
|
|
33
34
|
from .describe_metric import DescribeMetric
|
|
34
35
|
from .describe_metric_admin import DescribeMetricAdmin
|
|
35
36
|
from .describe_model import DescribeModel
|
|
@@ -47,13 +48,13 @@ from .list_custom_recipes import ListCustomRecipes
|
|
|
47
48
|
from .list_datasets import ListDatasets
|
|
48
49
|
from .list_graders import ListGraders
|
|
49
50
|
from .list_grouped_interactions import ListGroupedInteractions
|
|
51
|
+
from .list_harmony_groups import ListHarmonyGroups
|
|
50
52
|
from .list_interactions import ListInteractions
|
|
51
53
|
from .list_jobs import ListJobs
|
|
52
54
|
from .list_judge_versions import ListJudgeVersions
|
|
53
55
|
from .list_judges import ListJudges
|
|
54
56
|
from .list_metrics import ListMetrics
|
|
55
57
|
from .list_models import ListModels
|
|
56
|
-
from .list_partitions import ListPartitions
|
|
57
58
|
from .list_permissions import ListPermissions
|
|
58
59
|
from .list_remote_envs import ListRemoteEnvs
|
|
59
60
|
from .list_roles import ListRoles
|
|
@@ -188,7 +189,7 @@ class GQLClient(BaseClientOpenTelemetry):
|
|
|
188
189
|
return DeleteDataset.model_validate(data)
|
|
189
190
|
|
|
190
191
|
def add_hf_model(self, input: AddHFModelInput, **kwargs: Any) -> AddHFModel:
|
|
191
|
-
query = gql('\n mutation AddHFModel($input: AddHFModelInput!) {\n importHfModel(input: $input)\n }\n ')
|
|
192
|
+
query = gql('\n mutation AddHFModel($input: AddHFModelInput!) {\n importHfModel(input: $input) {\n ...JobData\n }\n }\n\n fragment CustomRecipeData on CustomRecipe {\n id\n key\n name\n content\n contentHash\n editable\n global\n builtin\n inputSchema\n jsonSchema\n description\n labels {\n key\n value\n }\n createdAt\n updatedAt\n createdBy {\n id\n name\n email\n }\n }\n\n fragment JobData on Job {\n id\n name\n status\n createdAt\n createdBy {\n id\n name\n }\n startedAt\n endedAt\n durationMs\n progress\n error\n kind\n stages {\n name\n status\n info {\n __typename\n ... on TrainingJobStageOutput {\n monitoringLink\n totalNumSamples\n processedNumSamples\n checkpoints\n }\n ... on EvalJobStageOutput {\n totalNumSamples\n processedNumSamples\n }\n ... on BatchInferenceJobStageOutput {\n totalNumSamples\n processedNumSamples\n }\n }\n }\n useCase {\n id\n key\n name\n }\n recipe {\n ...CustomRecipeData\n }\n details {\n args\n recipeHash\n artifacts {\n id\n name\n kind\n uri\n metadata\n createdAt\n }\n }\n }\n ')
|
|
192
193
|
variables: Dict[str, object] = {'input': input}
|
|
193
194
|
response = self.execute(query=query, operation_name='AddHFModel', variables=variables, **kwargs)
|
|
194
195
|
data = self.get_data(response)
|
|
@@ -495,15 +496,15 @@ class GQLClient(BaseClientOpenTelemetry):
|
|
|
495
496
|
data = self.get_data(response)
|
|
496
497
|
return ListPermissions.model_validate(data)
|
|
497
498
|
|
|
498
|
-
def
|
|
499
|
-
query = gql('\n query
|
|
499
|
+
def list_harmony_groups(self, **kwargs: Any) -> ListHarmonyGroups:
|
|
500
|
+
query = gql('\n query ListHarmonyGroups {\n harmonyGroups {\n ...HarmonyGroupData\n }\n }\n\n fragment HarmonyGroupData on HarmonyGroup {\n id\n key\n computePool {\n key\n name\n }\n status\n url\n worldSize\n gpuTypes\n createdAt\n onlineModels {\n ...ModelData\n }\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n kind\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n ')
|
|
500
501
|
variables: Dict[str, object] = {}
|
|
501
|
-
response = self.execute(query=query, operation_name='
|
|
502
|
+
response = self.execute(query=query, operation_name='ListHarmonyGroups', variables=variables, **kwargs)
|
|
502
503
|
data = self.get_data(response)
|
|
503
|
-
return
|
|
504
|
+
return ListHarmonyGroups.model_validate(data)
|
|
504
505
|
|
|
505
506
|
def list_compute_pools(self, **kwargs: Any) -> ListComputePools:
|
|
506
|
-
query = gql('\n query ListComputePools {\n computePools {\n id\n key\n name\n createdAt\n capabilities\n
|
|
507
|
+
query = gql('\n query ListComputePools {\n computePools {\n id\n key\n name\n createdAt\n capabilities\n harmonyGroups {\n ...HarmonyGroupData\n }\n }\n }\n\n fragment HarmonyGroupData on HarmonyGroup {\n id\n key\n computePool {\n key\n name\n }\n status\n url\n worldSize\n gpuTypes\n createdAt\n onlineModels {\n ...ModelData\n }\n }\n\n fragment ModelData on Model {\n id\n key\n name\n online\n isExternal\n providerName\n isAdapter\n isTraining\n createdAt\n kind\n size\n computeConfig {\n tp\n kvCacheLen\n maxSeqLen\n }\n }\n ')
|
|
507
508
|
variables: Dict[str, object] = {}
|
|
508
509
|
response = self.execute(query=query, operation_name='ListComputePools', variables=variables, **kwargs)
|
|
509
510
|
data = self.get_data(response)
|
|
@@ -572,6 +573,13 @@ class GQLClient(BaseClientOpenTelemetry):
|
|
|
572
573
|
data = self.get_data(response)
|
|
573
574
|
return ListJobs.model_validate(data)
|
|
574
575
|
|
|
576
|
+
def describe_job(self, id: Any, **kwargs: Any) -> DescribeJob:
|
|
577
|
+
query = gql('\n query DescribeJob($id: UUID!) {\n job(id: $id) {\n ...JobData\n }\n }\n\n fragment CustomRecipeData on CustomRecipe {\n id\n key\n name\n content\n contentHash\n editable\n global\n builtin\n inputSchema\n jsonSchema\n description\n labels {\n key\n value\n }\n createdAt\n updatedAt\n createdBy {\n id\n name\n email\n }\n }\n\n fragment JobData on Job {\n id\n name\n status\n createdAt\n createdBy {\n id\n name\n }\n startedAt\n endedAt\n durationMs\n progress\n error\n kind\n stages {\n name\n status\n info {\n __typename\n ... on TrainingJobStageOutput {\n monitoringLink\n totalNumSamples\n processedNumSamples\n checkpoints\n }\n ... on EvalJobStageOutput {\n totalNumSamples\n processedNumSamples\n }\n ... on BatchInferenceJobStageOutput {\n totalNumSamples\n processedNumSamples\n }\n }\n }\n useCase {\n id\n key\n name\n }\n recipe {\n ...CustomRecipeData\n }\n details {\n args\n recipeHash\n artifacts {\n id\n name\n kind\n uri\n metadata\n createdAt\n }\n }\n }\n ')
|
|
578
|
+
variables: Dict[str, object] = {'id': id}
|
|
579
|
+
response = self.execute(query=query, operation_name='DescribeJob', variables=variables, **kwargs)
|
|
580
|
+
data = self.get_data(response)
|
|
581
|
+
return DescribeJob.model_validate(data)
|
|
582
|
+
|
|
575
583
|
def execute_custom_operation(self, *fields: GraphQLField, operation_type: OperationType, operation_name: str) -> Dict[str, Any]:
|
|
576
584
|
selections = self._build_selection_set(fields)
|
|
577
585
|
combined_variables = self._combine_variables(fields)
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
from typing import Any, Dict, Optional, Union
|
|
2
2
|
from .base_operation import GraphQLField
|
|
3
|
-
from .custom_typing_fields import AbcampaignGraphQLField, AbReportGraphQLField, AbVariantReportComparisonGraphQLField, AbVariantReportGraphQLField, ActivityGraphQLField, ActivityOutputGraphQLField, AdaptBuiltinRecipeConfigOutputGraphQLField, AdaptCustomRecipeConfigOutputGraphQLField, AdaptRequestConfigOutputUnion, ApiKeyGraphQLField, ArtifactByproductsUnion, AuthProviderGraphQLField, BaseTrainingParamsOutputGraphQLField, BatchInferenceJobStageOutputGraphQLField, ChatMessageGraphQLField, ComparisonFeedbackGraphQLField, CompletionConnectionGraphQLField, CompletionEdgeGraphQLField, CompletionFeedbackFilterOutputGraphQLField, CompletionGraphQLField, CompletionGroupDataConnectionGraphQLField, CompletionGroupDataEdgeGraphQLField, CompletionGroupDataGraphQLField, CompletionGroupFeedbackStatsGraphQLField, CompletionHistoryEntryOuputGraphQLField, CompletionLabelFilterOutputGraphQLField, CompletionLabelGraphQLField, CompletionMetadataGraphQLField, ComputePoolGraphQLField, CustomConfigOutputGraphQLField, CustomRecipeGraphQLField, CustomRecipeJobDetailsGraphQLField, DatasetGraphQLField, DatasetMetricUsageGraphQLField, DatasetValidationOutputGraphQLField, DeleteConfirmGraphQLField, DirectFeedbackGraphQLField, DpotrainingParamsOutputGraphQLField, EmojiGraphQLField, EvalJobStageOutputGraphQLField, EvaluationAnswerRelevancyRecipeGraphQLField, EvaluationByproductsGraphQLField, EvaluationContextRelevancyRecipeGraphQLField, EvaluationCustomRecipeGraphQLField, EvaluationFaithfulnessRecipeGraphQLField, EvaluationJobGraphQLField, EvaluationRecipeUnion, EvaluationResultGraphQLField, GpuAllocationGraphQLField, GraderConfigUnion, GraderGraphQLField, GrpotrainingParamsOutputGraphQLField, GuidelineGraphQLField, GuidelinesTrainingParamsOutputGraphQLField, InteractionOutputGraphQLField, IntervalGraphQLField, JobArtifactGraphQLField, JobConnectionGraphQLField, JobEdgeGraphQLField, JobGraphQLField, JobStageInfoOutputUnion, JobStageOutputGraphQLField, JudgeConfigOutputGraphQLField, JudgeExampleGraphQLField, JudgeGraphQLField, JudgeTrainingParamsOutputGraphQLField, LabelGraphQLField, LabelKeyUsageGraphQLField, LabelUsageGraphQLField, LabelValueUsageGraphQLField, ListCompletionsFilterOutputGraphQLField, MetaObjectGraphQLField, MetricActivityGraphQLField, MetricGraphQLField, MetricTrainingParamsMetadataOutputUnion, MetricTrainingParamsOutputGraphQLField, MetricWithContextGraphQLField, ModelComputeConfigOutputGraphQLField, ModelGraphQLField, ModelPlacementOutputGraphQLField, ModelServiceGraphQLField, PageInfoGraphQLField,
|
|
3
|
+
from .custom_typing_fields import AbcampaignGraphQLField, AbReportGraphQLField, AbVariantReportComparisonGraphQLField, AbVariantReportGraphQLField, ActivityGraphQLField, ActivityOutputGraphQLField, AdaptBuiltinRecipeConfigOutputGraphQLField, AdaptCustomRecipeConfigOutputGraphQLField, AdaptRequestConfigOutputUnion, ApiKeyGraphQLField, ArtifactByproductsUnion, AuthProviderGraphQLField, BaseTrainingParamsOutputGraphQLField, BatchInferenceJobStageOutputGraphQLField, BillingUsageGraphQLField, ChatMessageGraphQLField, ComparisonFeedbackGraphQLField, CompletionConnectionGraphQLField, CompletionEdgeGraphQLField, CompletionFeedbackFilterOutputGraphQLField, CompletionGraphQLField, CompletionGroupDataConnectionGraphQLField, CompletionGroupDataEdgeGraphQLField, CompletionGroupDataGraphQLField, CompletionGroupFeedbackStatsGraphQLField, CompletionHistoryEntryOuputGraphQLField, CompletionLabelFilterOutputGraphQLField, CompletionLabelGraphQLField, CompletionMetadataGraphQLField, ComputePoolGraphQLField, ContractGraphQLField, CustomConfigOutputGraphQLField, CustomRecipeGraphQLField, CustomRecipeJobDetailsGraphQLField, DatasetGraphQLField, DatasetMetricUsageGraphQLField, DatasetValidationOutputGraphQLField, DeleteConfirmGraphQLField, DirectFeedbackGraphQLField, DpotrainingParamsOutputGraphQLField, EmojiGraphQLField, EvalJobStageOutputGraphQLField, EvaluationAnswerRelevancyRecipeGraphQLField, EvaluationByproductsGraphQLField, EvaluationContextRelevancyRecipeGraphQLField, EvaluationCustomRecipeGraphQLField, EvaluationFaithfulnessRecipeGraphQLField, EvaluationJobGraphQLField, EvaluationRecipeUnion, EvaluationResultGraphQLField, GlobalUsageGraphQLField, GpuAllocationGraphQLField, GraderConfigUnion, GraderGraphQLField, GrpotrainingParamsOutputGraphQLField, GuidelineGraphQLField, GuidelinesTrainingParamsOutputGraphQLField, HarmonyGroupGraphQLField, InteractionOutputGraphQLField, IntervalGraphQLField, JobArtifactGraphQLField, JobConnectionGraphQLField, JobEdgeGraphQLField, JobGraphQLField, JobStageInfoOutputUnion, JobStageOutputGraphQLField, JudgeConfigOutputGraphQLField, JudgeExampleGraphQLField, JudgeGraphQLField, JudgeTrainingParamsOutputGraphQLField, LabelGraphQLField, LabelKeyUsageGraphQLField, LabelUsageGraphQLField, LabelValueUsageGraphQLField, ListCompletionsFilterOutputGraphQLField, MetaObjectGraphQLField, MetricActivityGraphQLField, MetricGraphQLField, MetricTrainingParamsMetadataOutputUnion, MetricTrainingParamsOutputGraphQLField, MetricWithContextGraphQLField, ModelComputeConfigOutputGraphQLField, ModelGraphQLField, ModelPlacementOutputGraphQLField, ModelServiceGraphQLField, PageInfoGraphQLField, PpotrainingParamsOutputGraphQLField, PrebuiltConfigDefinitionGraphQLField, PrebuiltConfigOutputGraphQLField, PrebuiltCriteriaGraphQLField, ProviderListGraphQLField, RemoteConfigOutputGraphQLField, RemoteEnvGraphQLField, RemoteEnvTestOfflineGraphQLField, RemoteEnvTestOnlineGraphQLField, RewardServerTrainingParamsOutputGraphQLField, RoleGraphQLField, SampleConfigOutputGraphQLField, SampleDatasourceCompletionsOutputGraphQLField, SampleDatasourceDatasetOutputGraphQLField, SampleDatasourceOutputUnion, ScalarMetricConfigOutputGraphQLField, SessionGraphQLField, SettingsGraphQLField, SfttrainingParamsOutputGraphQLField, ShareGraphQLField, SystemPromptTemplateGraphQLField, TeamGraphQLField, TeamMemberGraphQLField, TeamWithroleGraphQLField, TimeRangeOutputGraphQLField, TimeseriesGraphQLField, ToolProviderGraphQLField, TrainingConfigOutputGraphQLField, TrainingJobGraphQLField, TrainingJobStageOutputGraphQLField, TrainingMetadataOutputGraphQLField, TrainingMetadataOutputParametersUnion, TrainingObjectiveOutputUnion, TrendResultGraphQLField, UnitConfigGraphQLField, UsageAggregateItemGraphQLField, UsageAggregatePerUseCaseItemGraphQLField, UsageGraphQLField, UsageStatsByModelGraphQLField, UsageStatsGraphQLField, UseCaseGraphQLField, UseCaseItemGraphQLField, UseCaseMetadataGraphQLField, UserGraphQLField, WidgetGraphQLField
|
|
4
4
|
from .input_types import AbCampaignFilter, CursorPageInput, FeedbackFilterInput, ListCompletionsFilterInput, MetricTrendInput, ModelServiceFilter, OrderPair, TimeRange, TimeseriesInput, UseCaseFilter
|
|
5
5
|
|
|
6
6
|
class AbReportFields(GraphQLField):
|
|
@@ -237,6 +237,29 @@ class BatchInferenceJobStageOutputFields(GraphQLField):
|
|
|
237
237
|
self._alias = alias
|
|
238
238
|
return self
|
|
239
239
|
|
|
240
|
+
class BillingUsageFields(GraphQLField):
|
|
241
|
+
"""@private"""
|
|
242
|
+
now: 'BillingUsageGraphQLField' = BillingUsageGraphQLField('now')
|
|
243
|
+
start: 'BillingUsageGraphQLField' = BillingUsageGraphQLField('start')
|
|
244
|
+
end: 'BillingUsageGraphQLField' = BillingUsageGraphQLField('end')
|
|
245
|
+
|
|
246
|
+
@classmethod
|
|
247
|
+
def usage(cls) -> 'UsageStatsFields':
|
|
248
|
+
return UsageStatsFields('usage')
|
|
249
|
+
|
|
250
|
+
@classmethod
|
|
251
|
+
def projection(cls) -> 'UsageStatsFields':
|
|
252
|
+
return UsageStatsFields('projection')
|
|
253
|
+
|
|
254
|
+
def fields(self, *subfields: Union[BillingUsageGraphQLField, 'UsageStatsFields']) -> 'BillingUsageFields':
|
|
255
|
+
"""Subfields should come from the BillingUsageFields class"""
|
|
256
|
+
self._subfields.extend(subfields)
|
|
257
|
+
return self
|
|
258
|
+
|
|
259
|
+
def alias(self, alias: str) -> 'BillingUsageFields':
|
|
260
|
+
self._alias = alias
|
|
261
|
+
return self
|
|
262
|
+
|
|
240
263
|
class ChatMessageFields(GraphQLField):
|
|
241
264
|
"""@private"""
|
|
242
265
|
role: 'ChatMessageGraphQLField' = ChatMessageGraphQLField('role')
|
|
@@ -573,15 +596,15 @@ class ComputePoolFields(GraphQLField):
|
|
|
573
596
|
created_at: 'ComputePoolGraphQLField' = ComputePoolGraphQLField('createdAt')
|
|
574
597
|
|
|
575
598
|
@classmethod
|
|
576
|
-
def
|
|
577
|
-
return
|
|
599
|
+
def all_harmony_groups(cls) -> 'HarmonyGroupFields':
|
|
600
|
+
return HarmonyGroupFields('all_harmony_groups')
|
|
578
601
|
|
|
579
602
|
@classmethod
|
|
580
|
-
def
|
|
581
|
-
return
|
|
603
|
+
def harmony_groups(cls) -> 'HarmonyGroupFields':
|
|
604
|
+
return HarmonyGroupFields('harmony_groups')
|
|
582
605
|
capabilities: 'ComputePoolGraphQLField' = ComputePoolGraphQLField('capabilities')
|
|
583
606
|
|
|
584
|
-
def fields(self, *subfields: Union[ComputePoolGraphQLField, '
|
|
607
|
+
def fields(self, *subfields: Union[ComputePoolGraphQLField, 'HarmonyGroupFields']) -> 'ComputePoolFields':
|
|
585
608
|
"""Subfields should come from the ComputePoolFields class"""
|
|
586
609
|
self._subfields.extend(subfields)
|
|
587
610
|
return self
|
|
@@ -590,6 +613,28 @@ class ComputePoolFields(GraphQLField):
|
|
|
590
613
|
self._alias = alias
|
|
591
614
|
return self
|
|
592
615
|
|
|
616
|
+
class ContractFields(GraphQLField):
|
|
617
|
+
"""@private"""
|
|
618
|
+
quota: 'ContractGraphQLField' = ContractGraphQLField('quota')
|
|
619
|
+
start_date: 'ContractGraphQLField' = ContractGraphQLField('startDate')
|
|
620
|
+
end_date: 'ContractGraphQLField' = ContractGraphQLField('endDate')
|
|
621
|
+
cycle: 'ContractGraphQLField' = ContractGraphQLField('cycle')
|
|
622
|
+
|
|
623
|
+
@classmethod
|
|
624
|
+
def usage(cls, *, now: Optional[int | str]=None) -> 'BillingUsageFields':
|
|
625
|
+
arguments: Dict[str, Dict[str, Any]] = {'now': {'type': 'InputDatetime', 'value': now}}
|
|
626
|
+
cleared_arguments = {key: value for (key, value) in arguments.items() if value['value'] is not None}
|
|
627
|
+
return BillingUsageFields('usage', arguments=cleared_arguments)
|
|
628
|
+
|
|
629
|
+
def fields(self, *subfields: Union[ContractGraphQLField, 'BillingUsageFields']) -> 'ContractFields':
|
|
630
|
+
"""Subfields should come from the ContractFields class"""
|
|
631
|
+
self._subfields.extend(subfields)
|
|
632
|
+
return self
|
|
633
|
+
|
|
634
|
+
def alias(self, alias: str) -> 'ContractFields':
|
|
635
|
+
self._alias = alias
|
|
636
|
+
return self
|
|
637
|
+
|
|
593
638
|
class CustomConfigOutputFields(GraphQLField):
|
|
594
639
|
"""@private"""
|
|
595
640
|
description: 'CustomConfigOutputGraphQLField' = CustomConfigOutputGraphQLField('description')
|
|
@@ -609,6 +654,7 @@ class CustomRecipeFields(GraphQLField):
|
|
|
609
654
|
key: 'CustomRecipeGraphQLField' = CustomRecipeGraphQLField('key')
|
|
610
655
|
name: 'CustomRecipeGraphQLField' = CustomRecipeGraphQLField('name')
|
|
611
656
|
editable: 'CustomRecipeGraphQLField' = CustomRecipeGraphQLField('editable')
|
|
657
|
+
hidden: 'CustomRecipeGraphQLField' = CustomRecipeGraphQLField('hidden')
|
|
612
658
|
builtin: 'CustomRecipeGraphQLField' = CustomRecipeGraphQLField('builtin')
|
|
613
659
|
global_: 'CustomRecipeGraphQLField' = CustomRecipeGraphQLField('global')
|
|
614
660
|
created_at: 'CustomRecipeGraphQLField' = CustomRecipeGraphQLField('createdAt')
|
|
@@ -982,6 +1028,35 @@ class EvaluationResultFields(GraphQLField):
|
|
|
982
1028
|
self._alias = alias
|
|
983
1029
|
return self
|
|
984
1030
|
|
|
1031
|
+
class GlobalUsageFields(GraphQLField):
|
|
1032
|
+
"""@private"""
|
|
1033
|
+
|
|
1034
|
+
@classmethod
|
|
1035
|
+
def total(cls) -> 'UsageStatsFields':
|
|
1036
|
+
return UsageStatsFields('total')
|
|
1037
|
+
|
|
1038
|
+
@classmethod
|
|
1039
|
+
def adaptive_models(cls) -> 'UsageStatsFields':
|
|
1040
|
+
return UsageStatsFields('adaptive_models')
|
|
1041
|
+
|
|
1042
|
+
@classmethod
|
|
1043
|
+
def external_models(cls) -> 'UsageStatsFields':
|
|
1044
|
+
return UsageStatsFields('external_models')
|
|
1045
|
+
|
|
1046
|
+
@classmethod
|
|
1047
|
+
def by_model(cls) -> 'UsageStatsByModelFields':
|
|
1048
|
+
return UsageStatsByModelFields('by_model')
|
|
1049
|
+
signature: 'GlobalUsageGraphQLField' = GlobalUsageGraphQLField('signature')
|
|
1050
|
+
|
|
1051
|
+
def fields(self, *subfields: Union[GlobalUsageGraphQLField, 'UsageStatsByModelFields', 'UsageStatsFields']) -> 'GlobalUsageFields':
|
|
1052
|
+
"""Subfields should come from the GlobalUsageFields class"""
|
|
1053
|
+
self._subfields.extend(subfields)
|
|
1054
|
+
return self
|
|
1055
|
+
|
|
1056
|
+
def alias(self, alias: str) -> 'GlobalUsageFields':
|
|
1057
|
+
self._alias = alias
|
|
1058
|
+
return self
|
|
1059
|
+
|
|
985
1060
|
class GpuAllocationFields(GraphQLField):
|
|
986
1061
|
"""@private"""
|
|
987
1062
|
name: 'GpuAllocationGraphQLField' = GpuAllocationGraphQLField('name')
|
|
@@ -1073,6 +1148,39 @@ class GuidelinesTrainingParamsOutputFields(GraphQLField):
|
|
|
1073
1148
|
self._alias = alias
|
|
1074
1149
|
return self
|
|
1075
1150
|
|
|
1151
|
+
class HarmonyGroupFields(GraphQLField):
|
|
1152
|
+
"""@private"""
|
|
1153
|
+
id: 'HarmonyGroupGraphQLField' = HarmonyGroupGraphQLField('id')
|
|
1154
|
+
key: 'HarmonyGroupGraphQLField' = HarmonyGroupGraphQLField('key')
|
|
1155
|
+
|
|
1156
|
+
@classmethod
|
|
1157
|
+
def compute_pool(cls) -> 'ComputePoolFields':
|
|
1158
|
+
return ComputePoolFields('compute_pool')
|
|
1159
|
+
status: 'HarmonyGroupGraphQLField' = HarmonyGroupGraphQLField('status')
|
|
1160
|
+
url: 'HarmonyGroupGraphQLField' = HarmonyGroupGraphQLField('url')
|
|
1161
|
+
world_size: 'HarmonyGroupGraphQLField' = HarmonyGroupGraphQLField('worldSize')
|
|
1162
|
+
gpu_total: 'HarmonyGroupGraphQLField' = HarmonyGroupGraphQLField('gpuTotal')
|
|
1163
|
+
gpu_allocated: 'HarmonyGroupGraphQLField' = HarmonyGroupGraphQLField('gpuAllocated')
|
|
1164
|
+
|
|
1165
|
+
@classmethod
|
|
1166
|
+
def gpu_allocations(cls) -> 'GpuAllocationFields':
|
|
1167
|
+
return GpuAllocationFields('gpu_allocations')
|
|
1168
|
+
gpu_types: 'HarmonyGroupGraphQLField' = HarmonyGroupGraphQLField('gpuTypes')
|
|
1169
|
+
created_at: 'HarmonyGroupGraphQLField' = HarmonyGroupGraphQLField('createdAt')
|
|
1170
|
+
|
|
1171
|
+
@classmethod
|
|
1172
|
+
def online_models(cls) -> 'ModelFields':
|
|
1173
|
+
return ModelFields('online_models')
|
|
1174
|
+
|
|
1175
|
+
def fields(self, *subfields: Union[HarmonyGroupGraphQLField, 'ComputePoolFields', 'GpuAllocationFields', 'ModelFields']) -> 'HarmonyGroupFields':
|
|
1176
|
+
"""Subfields should come from the HarmonyGroupFields class"""
|
|
1177
|
+
self._subfields.extend(subfields)
|
|
1178
|
+
return self
|
|
1179
|
+
|
|
1180
|
+
def alias(self, alias: str) -> 'HarmonyGroupFields':
|
|
1181
|
+
self._alias = alias
|
|
1182
|
+
return self
|
|
1183
|
+
|
|
1076
1184
|
class InteractionOutputFields(GraphQLField):
|
|
1077
1185
|
"""@private"""
|
|
1078
1186
|
value: 'InteractionOutputGraphQLField' = InteractionOutputGraphQLField('value')
|
|
@@ -1701,39 +1809,6 @@ class PageInfoFields(GraphQLField):
|
|
|
1701
1809
|
self._alias = alias
|
|
1702
1810
|
return self
|
|
1703
1811
|
|
|
1704
|
-
class PartitionFields(GraphQLField):
|
|
1705
|
-
"""@private"""
|
|
1706
|
-
id: 'PartitionGraphQLField' = PartitionGraphQLField('id')
|
|
1707
|
-
key: 'PartitionGraphQLField' = PartitionGraphQLField('key')
|
|
1708
|
-
|
|
1709
|
-
@classmethod
|
|
1710
|
-
def compute_pool(cls) -> 'ComputePoolFields':
|
|
1711
|
-
return ComputePoolFields('compute_pool')
|
|
1712
|
-
status: 'PartitionGraphQLField' = PartitionGraphQLField('status')
|
|
1713
|
-
url: 'PartitionGraphQLField' = PartitionGraphQLField('url')
|
|
1714
|
-
world_size: 'PartitionGraphQLField' = PartitionGraphQLField('worldSize')
|
|
1715
|
-
gpu_total: 'PartitionGraphQLField' = PartitionGraphQLField('gpuTotal')
|
|
1716
|
-
gpu_allocated: 'PartitionGraphQLField' = PartitionGraphQLField('gpuAllocated')
|
|
1717
|
-
|
|
1718
|
-
@classmethod
|
|
1719
|
-
def gpu_allocations(cls) -> 'GpuAllocationFields':
|
|
1720
|
-
return GpuAllocationFields('gpu_allocations')
|
|
1721
|
-
gpu_types: 'PartitionGraphQLField' = PartitionGraphQLField('gpuTypes')
|
|
1722
|
-
created_at: 'PartitionGraphQLField' = PartitionGraphQLField('createdAt')
|
|
1723
|
-
|
|
1724
|
-
@classmethod
|
|
1725
|
-
def online_models(cls) -> 'ModelFields':
|
|
1726
|
-
return ModelFields('online_models')
|
|
1727
|
-
|
|
1728
|
-
def fields(self, *subfields: Union[PartitionGraphQLField, 'ComputePoolFields', 'GpuAllocationFields', 'ModelFields']) -> 'PartitionFields':
|
|
1729
|
-
"""Subfields should come from the PartitionFields class"""
|
|
1730
|
-
self._subfields.extend(subfields)
|
|
1731
|
-
return self
|
|
1732
|
-
|
|
1733
|
-
def alias(self, alias: str) -> 'PartitionFields':
|
|
1734
|
-
self._alias = alias
|
|
1735
|
-
return self
|
|
1736
|
-
|
|
1737
1812
|
class PpotrainingParamsOutputFields(GraphQLField):
|
|
1738
1813
|
"""@private"""
|
|
1739
1814
|
kl_div_coeff: 'PpotrainingParamsOutputGraphQLField' = PpotrainingParamsOutputGraphQLField('klDivCoeff')
|
|
@@ -2350,6 +2425,46 @@ class UsageAggregatePerUseCaseItemFields(GraphQLField):
|
|
|
2350
2425
|
self._alias = alias
|
|
2351
2426
|
return self
|
|
2352
2427
|
|
|
2428
|
+
class UsageStatsFields(GraphQLField):
|
|
2429
|
+
"""@private"""
|
|
2430
|
+
total_tokens: 'UsageStatsGraphQLField' = UsageStatsGraphQLField('totalTokens')
|
|
2431
|
+
interactions: 'UsageStatsGraphQLField' = UsageStatsGraphQLField('interactions')
|
|
2432
|
+
prompt_tokens: 'UsageStatsGraphQLField' = UsageStatsGraphQLField('promptTokens')
|
|
2433
|
+
completion_tokens: 'UsageStatsGraphQLField' = UsageStatsGraphQLField('completionTokens')
|
|
2434
|
+
|
|
2435
|
+
def fields(self, *subfields: UsageStatsGraphQLField) -> 'UsageStatsFields':
|
|
2436
|
+
"""Subfields should come from the UsageStatsFields class"""
|
|
2437
|
+
self._subfields.extend(subfields)
|
|
2438
|
+
return self
|
|
2439
|
+
|
|
2440
|
+
def alias(self, alias: str) -> 'UsageStatsFields':
|
|
2441
|
+
self._alias = alias
|
|
2442
|
+
return self
|
|
2443
|
+
|
|
2444
|
+
class UsageStatsByModelFields(GraphQLField):
|
|
2445
|
+
"""@private"""
|
|
2446
|
+
|
|
2447
|
+
@classmethod
|
|
2448
|
+
def model(cls) -> 'ModelFields':
|
|
2449
|
+
return ModelFields('model')
|
|
2450
|
+
total_tokens: 'UsageStatsByModelGraphQLField' = UsageStatsByModelGraphQLField('totalTokens')
|
|
2451
|
+
interactions: 'UsageStatsByModelGraphQLField' = UsageStatsByModelGraphQLField('interactions')
|
|
2452
|
+
prompt_tokens: 'UsageStatsByModelGraphQLField' = UsageStatsByModelGraphQLField('promptTokens')
|
|
2453
|
+
completion_tokens: 'UsageStatsByModelGraphQLField' = UsageStatsByModelGraphQLField('completionTokens')
|
|
2454
|
+
|
|
2455
|
+
@classmethod
|
|
2456
|
+
def timeseries(cls) -> 'UsageAggregateItemFields':
|
|
2457
|
+
return UsageAggregateItemFields('timeseries')
|
|
2458
|
+
|
|
2459
|
+
def fields(self, *subfields: Union[UsageStatsByModelGraphQLField, 'ModelFields', 'UsageAggregateItemFields']) -> 'UsageStatsByModelFields':
|
|
2460
|
+
"""Subfields should come from the UsageStatsByModelFields class"""
|
|
2461
|
+
self._subfields.extend(subfields)
|
|
2462
|
+
return self
|
|
2463
|
+
|
|
2464
|
+
def alias(self, alias: str) -> 'UsageStatsByModelFields':
|
|
2465
|
+
self._alias = alias
|
|
2466
|
+
return self
|
|
2467
|
+
|
|
2353
2468
|
class UseCaseFields(GraphQLField):
|
|
2354
2469
|
"""@private"""
|
|
2355
2470
|
id: 'UseCaseGraphQLField' = UseCaseGraphQLField('id')
|
|
@@ -146,10 +146,10 @@ class Mutation:
|
|
|
146
146
|
return ModelFields(field_name='addModel', arguments=cleared_arguments)
|
|
147
147
|
|
|
148
148
|
@classmethod
|
|
149
|
-
def import_hf_model(cls, input: AddHFModelInput) ->
|
|
149
|
+
def import_hf_model(cls, input: AddHFModelInput) -> JobFields:
|
|
150
150
|
arguments: Dict[str, Dict[str, Any]] = {'input': {'type': 'AddHFModelInput!', 'value': input}}
|
|
151
151
|
cleared_arguments = {key: value for (key, value) in arguments.items() if value['value'] is not None}
|
|
152
|
-
return
|
|
152
|
+
return JobFields(field_name='importHfModel', arguments=cleared_arguments)
|
|
153
153
|
|
|
154
154
|
@classmethod
|
|
155
155
|
def update_model_compute_config(cls, id_or_key: str, input: ModelComputeConfigInput) -> ModelFields:
|