forked from JointCloud/pcm-modelarts
276 lines
12 KiB
Go
276 lines
12 KiB
Go
// Code generated by goctl. DO NOT EDIT.
|
|
// Source: pcm-modelarts.proto
|
|
|
|
package server
|
|
|
|
import (
|
|
"context"
|
|
|
|
"gitlink.org.cn/JointCloud/pcm-modelarts/internal/logic/modelartsservice"
|
|
"gitlink.org.cn/JointCloud/pcm-modelarts/internal/svc"
|
|
"gitlink.org.cn/JointCloud/pcm-modelarts/modelarts"
|
|
)
|
|
|
|
type ModelArtsServiceServer struct {
|
|
svcCtx *svc.ServiceContext
|
|
modelarts.UnimplementedModelArtsServiceServer
|
|
}
|
|
|
|
func NewModelArtsServiceServer(svcCtx *svc.ServiceContext) *ModelArtsServiceServer {
|
|
return &ModelArtsServiceServer{
|
|
svcCtx: svcCtx,
|
|
}
|
|
}
|
|
|
|
// get modelarts Token
|
|
func (s *ModelArtsServiceServer) GetToken(ctx context.Context, in *modelarts.TokenReq) (*modelarts.TokenResp, error) {
|
|
l := modelartsservicelogic.NewGetTokenLogic(ctx, s.svcCtx)
|
|
return l.GetToken(in)
|
|
}
|
|
|
|
// get modelarts Token
|
|
func (s *ModelArtsServiceServer) GetDatasetList(ctx context.Context, in *modelarts.DataSetReq) (*modelarts.DataSetResp, error) {
|
|
l := modelartsservicelogic.NewGetDatasetListLogic(ctx, s.svcCtx)
|
|
return l.GetDatasetList(in)
|
|
}
|
|
|
|
// create DateSet
|
|
func (s *ModelArtsServiceServer) CreateDataSet(ctx context.Context, in *modelarts.CreateDataSetReq) (*modelarts.CreateDataSetResq, error) {
|
|
l := modelartsservicelogic.NewCreateDataSetLogic(ctx, s.svcCtx)
|
|
return l.CreateDataSet(in)
|
|
}
|
|
|
|
// create DateSet
|
|
func (s *ModelArtsServiceServer) DeleteDataSet(ctx context.Context, in *modelarts.DeleteDataSetReq) (*modelarts.DeleteDataSetResq, error) {
|
|
l := modelartsservicelogic.NewDeleteDataSetLogic(ctx, s.svcCtx)
|
|
return l.DeleteDataSet(in)
|
|
}
|
|
|
|
// creat task 创建导入任务
|
|
func (s *ModelArtsServiceServer) CreateTask(ctx context.Context, in *modelarts.ImportTaskDataReq) (*modelarts.ImportTaskDataResp, error) {
|
|
l := modelartsservicelogic.NewCreateTaskLogic(ctx, s.svcCtx)
|
|
return l.CreateTask(in)
|
|
}
|
|
|
|
// get taskList 查询数据集导入任务列表
|
|
func (s *ModelArtsServiceServer) GetImportTaskList(ctx context.Context, in *modelarts.ListImportTasksReq) (*modelarts.ListImportTasksResp, error) {
|
|
l := modelartsservicelogic.NewGetImportTaskListLogic(ctx, s.svcCtx)
|
|
return l.GetImportTaskList(in)
|
|
}
|
|
|
|
// ListTrainingJobs 查询训练作业列表
|
|
func (s *ModelArtsServiceServer) GetListTrainingJobs(ctx context.Context, in *modelarts.ListTrainingJobsreq) (*modelarts.ListTrainingJobsresp, error) {
|
|
l := modelartsservicelogic.NewGetListTrainingJobsLogic(ctx, s.svcCtx)
|
|
return l.GetListTrainingJobs(in)
|
|
}
|
|
|
|
// GetTrainingJobs 查询训练详情
|
|
func (s *ModelArtsServiceServer) GetTrainingJobs(ctx context.Context, in *modelarts.DetailTrainingJobsReq) (*modelarts.JobResponse, error) {
|
|
l := modelartsservicelogic.NewGetTrainingJobsLogic(ctx, s.svcCtx)
|
|
return l.GetTrainingJobs(in)
|
|
}
|
|
|
|
// CreateTrainingJob 创建训练作业
|
|
func (s *ModelArtsServiceServer) CreateTrainingJob(ctx context.Context, in *modelarts.CreateTrainingJobReq) (*modelarts.CreateTrainingJobResp, error) {
|
|
l := modelartsservicelogic.NewCreateTrainingJobLogic(ctx, s.svcCtx)
|
|
return l.CreateTrainingJob(in)
|
|
}
|
|
|
|
// DeleteTrainingJobConfig 删除训练作业
|
|
func (s *ModelArtsServiceServer) DeleteTrainingJob(ctx context.Context, in *modelarts.DeleteTrainingJobReq) (*modelarts.DeleteTrainingJobResp, error) {
|
|
l := modelartsservicelogic.NewDeleteTrainingJobLogic(ctx, s.svcCtx)
|
|
return l.DeleteTrainingJob(in)
|
|
}
|
|
|
|
// CreateTrainingJobConfig 创建训练作业参数
|
|
func (s *ModelArtsServiceServer) CreateTrainingJobConfig(ctx context.Context, in *modelarts.CreateTrainingJobConfigReq) (*modelarts.CreateTrainingJobConfigResp, error) {
|
|
l := modelartsservicelogic.NewCreateTrainingJobConfigLogic(ctx, s.svcCtx)
|
|
return l.CreateTrainingJobConfig(in)
|
|
}
|
|
|
|
// DeleteTrainingJobConfig 删除训练作业参数
|
|
func (s *ModelArtsServiceServer) DeleteTrainingJobConfig(ctx context.Context, in *modelarts.DeleteTrainingJobConfigReq) (*modelarts.DeleteTrainingJobConfigResp, error) {
|
|
l := modelartsservicelogic.NewDeleteTrainingJobConfigLogic(ctx, s.svcCtx)
|
|
return l.DeleteTrainingJobConfig(in)
|
|
}
|
|
|
|
// ListTrainingJobConfig 查询训练作业参数
|
|
func (s *ModelArtsServiceServer) ListTrainingJobConfig(ctx context.Context, in *modelarts.ListTrainingJobConfigReq) (*modelarts.ListTrainingJobConfigResp, error) {
|
|
l := modelartsservicelogic.NewListTrainingJobConfigLogic(ctx, s.svcCtx)
|
|
return l.ListTrainingJobConfig(in)
|
|
}
|
|
|
|
// CreateAlgorithm 创建算法
|
|
func (s *ModelArtsServiceServer) CreateAlgorithm(ctx context.Context, in *modelarts.CreateAlgorithmReq) (*modelarts.CreateAlgorithmResp, error) {
|
|
l := modelartsservicelogic.NewCreateAlgorithmLogic(ctx, s.svcCtx)
|
|
return l.CreateAlgorithm(in)
|
|
}
|
|
|
|
// ListAlgorithms 查询算法
|
|
func (s *ModelArtsServiceServer) ListAlgorithms(ctx context.Context, in *modelarts.ListAlgorithmsReq) (*modelarts.ListAlgorithmsResp, error) {
|
|
l := modelartsservicelogic.NewListAlgorithmsLogic(ctx, s.svcCtx)
|
|
return l.ListAlgorithms(in)
|
|
}
|
|
|
|
// DeleteAlgorithms 删除算法
|
|
func (s *ModelArtsServiceServer) DeleteAlgorithms(ctx context.Context, in *modelarts.DeleteAlgorithmsReq) (*modelarts.DeleteAlgorithmsResp, error) {
|
|
l := modelartsservicelogic.NewDeleteAlgorithmsLogic(ctx, s.svcCtx)
|
|
return l.DeleteAlgorithms(in)
|
|
}
|
|
|
|
// ShowAlgorithmByUuid 展示算法详情
|
|
func (s *ModelArtsServiceServer) ShowAlgorithmByUuid(ctx context.Context, in *modelarts.ShowAlgorithmByUuidReq) (*modelarts.ShowAlgorithmByUuidResp, error) {
|
|
l := modelartsservicelogic.NewShowAlgorithmByUuidLogic(ctx, s.svcCtx)
|
|
return l.ShowAlgorithmByUuid(in)
|
|
}
|
|
|
|
// training-job-flavors 获取训练作业支持的公共规格
|
|
func (s *ModelArtsServiceServer) GetTrainingJobFlavors(ctx context.Context, in *modelarts.TrainingJobFlavorsReq) (*modelarts.TrainingJobFlavorsResp, error) {
|
|
l := modelartsservicelogic.NewGetTrainingJobFlavorsLogic(ctx, s.svcCtx)
|
|
return l.GetTrainingJobFlavors(in)
|
|
}
|
|
|
|
// GET ai-engines 查询作业引擎规格
|
|
func (s *ModelArtsServiceServer) GetAiEnginesList(ctx context.Context, in *modelarts.ListAiEnginesReq) (*modelarts.ListAiEnginesResp, error) {
|
|
l := modelartsservicelogic.NewGetAiEnginesListLogic(ctx, s.svcCtx)
|
|
return l.GetAiEnginesList(in)
|
|
}
|
|
|
|
// 查询训练作业指定任务的日志（预览）
|
|
func (s *ModelArtsServiceServer) GetTrainingJobLogsPreview(ctx context.Context, in *modelarts.GetTrainingJobLogsPreviewReq) (*modelarts.GetTrainingJobLogsPreviewResp, error) {
|
|
l := modelartsservicelogic.NewGetTrainingJobLogsPreviewLogic(ctx, s.svcCtx)
|
|
return l.GetTrainingJobLogsPreview(in)
|
|
}
|
|
|
|
// export task
|
|
func (s *ModelArtsServiceServer) ExportTask(ctx context.Context, in *modelarts.ExportTaskReq) (*modelarts.ExportTaskDataResp, error) {
|
|
l := modelartsservicelogic.NewExportTaskLogic(ctx, s.svcCtx)
|
|
return l.ExportTask(in)
|
|
}
|
|
|
|
func (s *ModelArtsServiceServer) GetExportTasksOfDataset(ctx context.Context, in *modelarts.GetExportTasksOfDatasetReq) (*modelarts.GetExportTasksOfDatasetResp, error) {
|
|
l := modelartsservicelogic.NewGetExportTasksOfDatasetLogic(ctx, s.svcCtx)
|
|
return l.GetExportTasksOfDataset(in)
|
|
}
|
|
|
|
func (s *ModelArtsServiceServer) GetExportTaskStatusOfDataset(ctx context.Context, in *modelarts.GetExportTaskStatusOfDatasetReq) (*modelarts.GetExportTaskStatusOfDatasetResp, error) {
|
|
l := modelartsservicelogic.NewGetExportTaskStatusOfDatasetLogic(ctx, s.svcCtx)
|
|
return l.GetExportTaskStatusOfDataset(in)
|
|
}
|
|
|
|
// processor task
|
|
func (s *ModelArtsServiceServer) CreateProcessorTask(ctx context.Context, in *modelarts.CreateProcessorTaskReq) (*modelarts.CreateProcessorTaskResp, error) {
|
|
l := modelartsservicelogic.NewCreateProcessorTaskLogic(ctx, s.svcCtx)
|
|
return l.CreateProcessorTask(in)
|
|
}
|
|
|
|
func (s *ModelArtsServiceServer) DescribeProcessorTask(ctx context.Context, in *modelarts.DescribeProcessorTaskReq) (*modelarts.DescribeProcessorTaskResp, error) {
|
|
l := modelartsservicelogic.NewDescribeProcessorTaskLogic(ctx, s.svcCtx)
|
|
return l.DescribeProcessorTask(in)
|
|
}
|
|
|
|
// model management
|
|
func (s *ModelArtsServiceServer) CreateModel(ctx context.Context, in *modelarts.CreateModelReq) (*modelarts.CreateModelResp, error) {
|
|
l := modelartsservicelogic.NewCreateModelLogic(ctx, s.svcCtx)
|
|
return l.CreateModel(in)
|
|
}
|
|
|
|
func (s *ModelArtsServiceServer) DeleteModel(ctx context.Context, in *modelarts.DeleteModelReq) (*modelarts.DeleteModelResp, error) {
|
|
l := modelartsservicelogic.NewDeleteModelLogic(ctx, s.svcCtx)
|
|
return l.DeleteModel(in)
|
|
}
|
|
|
|
func (s *ModelArtsServiceServer) ListModels(ctx context.Context, in *modelarts.ListModelReq) (*modelarts.ListModelResp, error) {
|
|
l := modelartsservicelogic.NewListModelsLogic(ctx, s.svcCtx)
|
|
return l.ListModels(in)
|
|
}
|
|
|
|
func (s *ModelArtsServiceServer) ShowModels(ctx context.Context, in *modelarts.ShowModelReq) (*modelarts.ShowModelResp, error) {
|
|
l := modelartsservicelogic.NewShowModelsLogic(ctx, s.svcCtx)
|
|
return l.ShowModels(in)
|
|
}
|
|
|
|
// service management
|
|
func (s *ModelArtsServiceServer) CreateService(ctx context.Context, in *modelarts.CreateServiceReq) (*modelarts.CreateServiceResp, error) {
|
|
l := modelartsservicelogic.NewCreateServiceLogic(ctx, s.svcCtx)
|
|
return l.CreateService(in)
|
|
}
|
|
|
|
func (s *ModelArtsServiceServer) ListServices(ctx context.Context, in *modelarts.ListServicesReq) (*modelarts.ListServicesResp, error) {
|
|
l := modelartsservicelogic.NewListServicesLogic(ctx, s.svcCtx)
|
|
return l.ListServices(in)
|
|
}
|
|
|
|
func (s *ModelArtsServiceServer) ShowService(ctx context.Context, in *modelarts.ShowServiceReq) (*modelarts.ShowServiceResp, error) {
|
|
l := modelartsservicelogic.NewShowServiceLogic(ctx, s.svcCtx)
|
|
return l.ShowService(in)
|
|
}
|
|
|
|
func (s *ModelArtsServiceServer) DeleteService(ctx context.Context, in *modelarts.DeleteServiceReq) (*modelarts.DeleteServiceResp, error) {
|
|
l := modelartsservicelogic.NewDeleteServiceLogic(ctx, s.svcCtx)
|
|
return l.DeleteService(in)
|
|
}
|
|
|
|
func (s *ModelArtsServiceServer) ListClusters(ctx context.Context, in *modelarts.ListClustersReq) (*modelarts.ListClustersResp, error) {
|
|
l := modelartsservicelogic.NewListClustersLogic(ctx, s.svcCtx)
|
|
return l.ListClusters(in)
|
|
}
|
|
|
|
// notebook task
|
|
func (s *ModelArtsServiceServer) ListNotebook(ctx context.Context, in *modelarts.ListNotebookReq) (*modelarts.ListNotebookResp, error) {
|
|
l := modelartsservicelogic.NewListNotebookLogic(ctx, s.svcCtx)
|
|
return l.ListNotebook(in)
|
|
}
|
|
|
|
func (s *ModelArtsServiceServer) CreateNotebook(ctx context.Context, in *modelarts.CreateNotebookReq) (*modelarts.CreateNotebookResp, error) {
|
|
l := modelartsservicelogic.NewCreateNotebookLogic(ctx, s.svcCtx)
|
|
return l.CreateNotebook(in)
|
|
}
|
|
|
|
func (s *ModelArtsServiceServer) StartNotebook(ctx context.Context, in *modelarts.StartNotebookReq) (*modelarts.StartNotebookResp, error) {
|
|
l := modelartsservicelogic.NewStartNotebookLogic(ctx, s.svcCtx)
|
|
return l.StartNotebook(in)
|
|
}
|
|
|
|
func (s *ModelArtsServiceServer) StopNotebook(ctx context.Context, in *modelarts.StopNotebookReq) (*modelarts.StopNotebookResp, error) {
|
|
l := modelartsservicelogic.NewStopNotebookLogic(ctx, s.svcCtx)
|
|
return l.StopNotebook(in)
|
|
}
|
|
|
|
func (s *ModelArtsServiceServer) GetNotebookStorage(ctx context.Context, in *modelarts.GetNotebookStorageReq) (*modelarts.GetNotebookStorageResp, error) {
|
|
l := modelartsservicelogic.NewGetNotebookStorageLogic(ctx, s.svcCtx)
|
|
return l.GetNotebookStorage(in)
|
|
}
|
|
|
|
func (s *ModelArtsServiceServer) MountNotebookStorage(ctx context.Context, in *modelarts.MountNotebookStorageReq) (*modelarts.MountNotebookStorageResp, error) {
|
|
l := modelartsservicelogic.NewMountNotebookStorageLogic(ctx, s.svcCtx)
|
|
return l.MountNotebookStorage(in)
|
|
}
|
|
|
|
// visualization-jobs
|
|
func (s *ModelArtsServiceServer) GetVisualizationJob(ctx context.Context, in *modelarts.GetVisualizationJobReq) (*modelarts.GetVisualizationJobResp, error) {
|
|
l := modelartsservicelogic.NewGetVisualizationJobLogic(ctx, s.svcCtx)
|
|
return l.GetVisualizationJob(in)
|
|
}
|
|
|
|
func (s *ModelArtsServiceServer) CreateVisualizationJob(ctx context.Context, in *modelarts.CreateVisualizationJobReq) (*modelarts.CreateVisualizationJobResp, error) {
|
|
l := modelartsservicelogic.NewCreateVisualizationJobLogic(ctx, s.svcCtx)
|
|
return l.CreateVisualizationJob(in)
|
|
}
|
|
|
|
// reasoning
|
|
func (s *ModelArtsServiceServer) ImageReasoning(ctx context.Context, in *modelarts.ImageReasoningReq) (*modelarts.ImageReasoningResp, error) {
|
|
l := modelartsservicelogic.NewImageReasoningLogic(ctx, s.svcCtx)
|
|
return l.ImageReasoning(in)
|
|
}
|
|
|
|
func (s *ModelArtsServiceServer) ImageReasoningUrl(ctx context.Context, in *modelarts.ImageReasoningUrlReq) (*modelarts.ImageReasoningUrlResp, error) {
|
|
l := modelartsservicelogic.NewImageReasoningUrlLogic(ctx, s.svcCtx)
|
|
return l.ImageReasoningUrl(in)
|
|
}
|
|
|
|
func (s *ModelArtsServiceServer) ChatglmReasoning(ctx context.Context, in *modelarts.ChatglmReasoningReq) (*modelarts.ChatglmReasoningResp, error) {
|
|
l := modelartsservicelogic.NewChatglmReasoningLogic(ctx, s.svcCtx)
|
|
return l.ChatglmReasoning(in)
|
|
}
|