floraachy
/
pcm-coordinator
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

updated imageinference api 

Former-commit-id: 37f69eb747ca83ea4f303a71d77c9621efd05472
This commit is contained in:
tzwang 2024-06-19 19:22:46 +08:00
parent cfecebc3b8
commit 5a0d3a9428
2 changed files with 40 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
api/desc/inference/inference.api
View File

@ -3,38 +3,33 @@ syntax = "v1"
type ( type (
InferOption { InferOption {
TaskName string `json:"taskName"` TaskName string `json:"taskName"`
// AdapterId string `json:"adapterId"` TaskType string `json:"taskType"`
// AiClusterIds []string `json:"aiClusterIds"` AdapterId string `json:"adapterId"`
// ResourceType string `json:"resourceType"` AiClusterIds []string `json:"aiClusterIds"`
// ComputeCard string `json:"card"` ResourceType string `json:"resourceType"`
// Tops float64 `json:"Tops,optional"` ComputeCard string `json:"card"`
// TaskType string `json:"taskType"` Strategy string `json:"strategy"`
// Datasets string `json:"datasets"` StaticWeightMap map[string]int32 `json:"staticWeightMap,optional"`
// Algorithm string `json:"algorithm"` Params []string `json:"params,optional"`
// Strategy string `json:"strategy"` Envs []string `json:"envs,optional"`
// StaticWeightMap map[string]int32 `json:"staticWeightMap,optional"` Cmd string `json:"cmd,optional"`
// Params []string `json:"params,optional"` Replica int32 `json:"replicas,optional"`
// Envs []string `json:"envs,optional"`
// Cmd string `json:"cmd,optional"`
// Replica int32 `json:"replicas"`
} }
/******************image inference*************************/ /******************image inference*************************/
ImageInferenceReq { ImageInferenceReq {
InferOption *InferOption `json:"inferOption,optional"` InferOption *InferOption `json:"inferOption,optional"`
Images []string `form:"images"`
} }
ImageInferenceResp { ImageInferenceResp {
InferResults []*ImageResult `json:"result"`
} }
// InferResult { ImageResult {
// ClusterId string `json:"clusterId"` ClusterId string `json:"clusterId"`
// TaskId string `json:"taskId"` ClusterName string `json:"clusterName"`
// Card string `json:"card"` ImageName string `json:"imageName"`
// Strategy string `json:"strategy"` Card string `json:"card"`
// JobId string `json:"jobId"` ImageResult string `json:"imageResult"`
// Replica int32 `json:"replica"` }
// Msg string `json:"msg"`
// }
) )

23
api/internal/types/types.go
View File

@ -5880,13 +5880,32 @@ type Category struct {
} }
type InferOption struct { type InferOption struct {
TaskName string `json:"taskName"` TaskName string `json:"taskName"`
TaskType string `json:"taskType"`
AdapterId string `json:"adapterId"`
AiClusterIds []string `json:"aiClusterIds"`
ResourceType string `json:"resourceType"`
ComputeCard string `json:"card"`
Strategy string `json:"strategy"`
StaticWeightMap map[string]int32 `json:"staticWeightMap,optional"`
Params []string `json:"params,optional"`
Envs []string `json:"envs,optional"`
Cmd string `json:"cmd,optional"`
Replica int32 `json:"replicas,optional"`
} }
type ImageInferenceReq struct { type ImageInferenceReq struct {
InferOption *InferOption `json:"inferOption,optional"` InferOption *InferOption `json:"inferOption,optional"`
Images []string `form:"images"`
} }
type ImageInferenceResp struct { type ImageInferenceResp struct {
InferResults []*ImageResult `json:"result"`
}
type ImageResult struct {
ClusterId string `json:"clusterId"`
ClusterName string `json:"clusterName"`
ImageName string `json:"imageName"`
Card string `json:"card"`
ImageResult string `json:"imageResult"`
} }