annzee
/
PCM
forked from JCCE/PCM
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fit:修改modelarts proto

This commit is contained in:
qiwang 2023-03-02 11:34:33 +08:00
parent 9b104e9e52
commit 069fb33989
5 changed files with 2347 additions and 1723 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
adaptor/AIComputing/AICore/api/etc/aicore-api.yaml
View File

@ -2,7 +2,7 @@ Name: AICore-api
Host: 0.0.0.0
Port: 8888
#rpc
#rpc by wangqi
ModelartsRpcConf:
Endpoints:
- 127.0.0.1:2003

6
adaptor/AIComputing/modelarts/rpc/internal/server/modelartsserver.go
View File

@ -88,6 +88,12 @@ func (s *ModelArtsServer) CreateAlgorithm(ctx context.Context, in *modelarts.Cre
return l.CreateAlgorithm(in)
}
// ListAlgorithms 查询算法
func (s *ModelArtsServer) ListAlgorithms(ctx context.Context, in *modelarts.ListAlgorithmsReq) (*modelarts.ListAlgorithmsResp, error) {
l := logic.NewListAlgorithmsLogic(ctx, s.svcCtx)
return l.ListAlgorithms(in)
}
// export task
func (s *ModelArtsServer) ExportTask(ctx context.Context, in *modelarts.ExportTaskReq) (*modelarts.ExportTaskResp, error) {
l := logic.NewExportTaskLogic(ctx, s.svcCtx)

4011
adaptor/AIComputing/modelarts/rpc/modelarts/modelarts.pb.go
View File

File diff suppressed because it is too large Load Diff

38
adaptor/AIComputing/modelarts/rpc/modelarts/modelarts_grpc.pb.go
View File

@ -44,6 +44,8 @@ type ModelArtsClient interface {
ListTrainingJobConfig(ctx context.Context, in *ListTrainingJobConfigReq, opts ...grpc.CallOption) (*ListTrainingJobConfigResp, error)
// CreateAlgorithm 创建算法
CreateAlgorithm(ctx context.Context, in *CreateAlgorithmReq, opts ...grpc.CallOption) (*CreateAlgorithmResp, error)
// ListAlgorithms 查询算法
ListAlgorithms(ctx context.Context, in *ListAlgorithmsReq, opts ...grpc.CallOption) (*ListAlgorithmsResp, error)
//export task
ExportTask(ctx context.Context, in *ExportTaskReq, opts ...grpc.CallOption) (*ExportTaskResp, error)
GetExportTasksOfDataset(ctx context.Context, in *GetExportTasksOfDatasetReq, opts ...grpc.CallOption) (*GetExportTasksOfDatasetResp, error)
@ -171,6 +173,15 @@ func (c *modelArtsClient) CreateAlgorithm(ctx context.Context, in *CreateAlgorit
return out, nil
}
func (c *modelArtsClient) ListAlgorithms(ctx context.Context, in *ListAlgorithmsReq, opts ...grpc.CallOption) (*ListAlgorithmsResp, error) {
out := new(ListAlgorithmsResp)
err := c.cc.Invoke(ctx, "/modelarts.ModelArts/ListAlgorithms", in, out, opts...)
if err != nil {
return nil, err
}
return out, nil
}
func (c *modelArtsClient) ExportTask(ctx context.Context, in *ExportTaskReq, opts ...grpc.CallOption) (*ExportTaskResp, error) {
out := new(ExportTaskResp)
err := c.cc.Invoke(ctx, "/modelarts.ModelArts/ExportTask", in, out, opts...)
@ -323,6 +334,8 @@ type ModelArtsServer interface {
ListTrainingJobConfig(context.Context, *ListTrainingJobConfigReq) (*ListTrainingJobConfigResp, error)
// CreateAlgorithm 创建算法
CreateAlgorithm(context.Context, *CreateAlgorithmReq) (*CreateAlgorithmResp, error)
// ListAlgorithms 查询算法
ListAlgorithms(context.Context, *ListAlgorithmsReq) (*ListAlgorithmsResp, error)
//export task
ExportTask(context.Context, *ExportTaskReq) (*ExportTaskResp, error)
GetExportTasksOfDataset(context.Context, *GetExportTasksOfDatasetReq) (*GetExportTasksOfDatasetResp, error)
@ -381,6 +394,9 @@ func (UnimplementedModelArtsServer) ListTrainingJobConfig(context.Context, *List
func (UnimplementedModelArtsServer) CreateAlgorithm(context.Context, *CreateAlgorithmReq) (*CreateAlgorithmResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method CreateAlgorithm not implemented")
}
func (UnimplementedModelArtsServer) ListAlgorithms(context.Context, *ListAlgorithmsReq) (*ListAlgorithmsResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method ListAlgorithms not implemented")
}
func (UnimplementedModelArtsServer) ExportTask(context.Context, *ExportTaskReq) (*ExportTaskResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method ExportTask not implemented")
}
@ -634,6 +650,24 @@ func _ModelArts_CreateAlgorithm_Handler(srv interface{}, ctx context.Context, de
return interceptor(ctx, in, info, handler)
}
func _ModelArts_ListAlgorithms_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(ListAlgorithmsReq)
if err := dec(in); err != nil {
return nil, err
}
if interceptor == nil {
return srv.(ModelArtsServer).ListAlgorithms(ctx, in)
}
info := &grpc.UnaryServerInfo{
Server: srv,
FullMethod: "/modelarts.ModelArts/ListAlgorithms",
}
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(ModelArtsServer).ListAlgorithms(ctx, req.(*ListAlgorithmsReq))
}
return interceptor(ctx, in, info, handler)
}
func _ModelArts_ExportTask_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(ExportTaskReq)
if err := dec(in); err != nil {
@ -937,6 +971,10 @@ var ModelArts_ServiceDesc = grpc.ServiceDesc{
MethodName: "CreateAlgorithm",
Handler: _ModelArts_CreateAlgorithm_Handler,
},
{
MethodName: "ListAlgorithms",
Handler: _ModelArts_ListAlgorithms_Handler,
},
{
MethodName: "ExportTask",
Handler: _ModelArts_ExportTask_Handler,

13
adaptor/AIComputing/modelarts/rpc/modelartsclient/modelarts.go
View File

@ -16,6 +16,7 @@ type (
AdvancedConfigAl = modelarts.AdvancedConfigAl
AlgoConfigs = modelarts.AlgoConfigs
Algorithm = modelarts.Algorithm
AlgorithmResponse = modelarts.AlgorithmResponse
Algorithms = modelarts.Algorithms
AnnotationFormatConfig = modelarts.AnnotationFormatConfig
AnnotationsS = modelarts.AnnotationsS
@ -36,6 +37,8 @@ type (
Cpu = modelarts.Cpu
CreateAlgorithmReq = modelarts.CreateAlgorithmReq
CreateAlgorithmResp = modelarts.CreateAlgorithmResp
CreateDataSetReq = modelarts.CreateDataSetReq
CreateDataSetResq = modelarts.CreateDataSetResq
CreateModelReq = modelarts.CreateModelReq
CreateModelRequestInferParams = modelarts.CreateModelRequestInferParams
CreateModelRequestModelApis = modelarts.CreateModelRequestModelApis
@ -102,6 +105,8 @@ type (
JobConfigAl = modelarts.JobConfigAl
JobMetadata = modelarts.JobMetadata
JobResponse = modelarts.JobResponse
ListAlgorithmsReq = modelarts.ListAlgorithmsReq
ListAlgorithmsResp = modelarts.ListAlgorithmsResp
ListClustersReq = modelarts.ListClustersReq
ListClustersResp = modelarts.ListClustersResp
ListImportTasksReq = modelarts.ListImportTasksReq
@ -209,6 +214,8 @@ type (
ListTrainingJobConfig(ctx context.Context, in *ListTrainingJobConfigReq, opts ...grpc.CallOption) (*ListTrainingJobConfigResp, error)
// CreateAlgorithm 创建算法
CreateAlgorithm(ctx context.Context, in *CreateAlgorithmReq, opts ...grpc.CallOption) (*CreateAlgorithmResp, error)
// ListAlgorithms 查询算法
ListAlgorithms(ctx context.Context, in *ListAlgorithmsReq, opts ...grpc.CallOption) (*ListAlgorithmsResp, error)
// export task
ExportTask(ctx context.Context, in *ExportTaskReq, opts ...grpc.CallOption) (*ExportTaskResp, error)
GetExportTasksOfDataset(ctx context.Context, in *GetExportTasksOfDatasetReq, opts ...grpc.CallOption) (*GetExportTasksOfDatasetResp, error)
@ -306,6 +313,12 @@ func (m *defaultModelArts) CreateAlgorithm(ctx context.Context, in *CreateAlgori
return client.CreateAlgorithm(ctx, in, opts...)
}
// ListAlgorithms 查询算法
func (m *defaultModelArts) ListAlgorithms(ctx context.Context, in *ListAlgorithmsReq, opts ...grpc.CallOption) (*ListAlgorithmsResp, error) {
client := modelarts.NewModelArtsClient(m.cli.Conn())
return client.ListAlgorithms(ctx, in, opts...)
}
// export task
func (m *defaultModelArts) ExportTask(ctx context.Context, in *ExportTaskReq, opts ...grpc.CallOption) (*ExportTaskResp, error) {
client := modelarts.NewModelArtsClient(m.cli.Conn())