|
- package ai_task_entity
-
- import (
- "code.gitea.io/gitea/models"
- "code.gitea.io/gitea/modules/timeutil"
- )
-
- type CreateNoteBookTaskRequest struct {
- Name string
- Tasks []NoteBookTask
- }
-
- type NoteBookTask struct {
- AutoStopDuration int
- Name string
- Capacity int
- CenterID []string
- Code []ContainerData
- Datasets []ContainerData
- PreTrainModel []ContainerData
- ImageId string
- ImageUrl string
- ResourceSpecId string
- Spec *models.Specification
- }
-
- type CreateNoteBookTaskResponse struct {
- StartedAt int64
- RunSec int64
- CompletedAt int64
- CreatedAt int64
- UpdatedAt int64
- Desc string
- JobID string
- Name string
- Status string
- UserID string
- }
-
- type RestartNoteBookTaskResponse struct {
- JobId string `json:"newId"`
- Status string `json:"status"`
- }
-
- type CreateTrainTaskRequest struct {
- Name string
- Tasks []TrainTask
- }
-
- type QueryTaskResponse struct {
- StartedAt timeutil.TimeStamp `json:"started_at"`
- CompletedAt timeutil.TimeStamp `json:"completed_at"`
- JobId string `json:"job_id"`
- Status string `json:"status"`
- Url string `json:"url"`
- Token string `json:"token"`
- CenterId string `json:"center_id"`
- CenterName string `json:"center_name"`
- }
-
- func ConvertGrampusNotebookResponse(job models.GrampusNotebookInfo) *QueryTaskResponse {
- if len(job.Tasks) == 0 {
- return nil
- }
- task := job.Tasks[0]
- centerId := ""
- if len(task.CenterID) > 0 {
- centerId = task.CenterID[0]
- }
- centerName := ""
- if len(task.CenterName) > 0 {
- centerName = task.CenterName[0]
- }
- var url, token string
- if len(job.Tasks) > 0 {
- url = job.Tasks[0].Url
- token = job.Tasks[0].Token
- }
- return &QueryTaskResponse{
- StartedAt: timeutil.TimeStamp(job.StartedAt),
- CompletedAt: timeutil.TimeStamp(job.CompletedAt),
- Status: job.Status,
- CenterId: centerId,
- CenterName: centerName,
- Url: url,
- Token: token,
- JobId: job.JobID,
- }
- }
- func ConvertGrampusTrainResponse(job models.GrampusJobInfo) *QueryTaskResponse {
- if len(job.Tasks) == 0 {
- return nil
- }
- task := job.Tasks[0]
- centerId := ""
- if len(task.CenterID) > 0 {
- centerId = task.CenterID[0]
- }
- centerName := ""
- if len(task.CenterName) > 0 {
- centerName = task.CenterName[0]
- }
- return &QueryTaskResponse{
- StartedAt: timeutil.TimeStamp(job.StartedAt),
- CompletedAt: timeutil.TimeStamp(job.CompletedAt),
- Status: job.Status,
- CenterId: centerId,
- CenterName: centerName,
- JobId: job.JobID,
- }
- }
-
- func ConvertCloudbrainOneQueryNotebookByNameResponse(result models.JobResultInListPayload) *QueryTaskResponse {
- if result.State == "" {
- return nil
- }
- return &QueryTaskResponse{
- StartedAt: timeutil.TimeStamp(result.CreatedTime / 1000),
- CompletedAt: timeutil.TimeStamp(result.CompletedTime / 1000),
- Status: result.State,
- JobId: result.Id,
- }
- }
-
- func ConvertCloudbrainOneNotebookResponse(result models.JobResultPayload) *QueryTaskResponse {
- if result.JobStatus.State == "" {
- return nil
- }
- return &QueryTaskResponse{
- StartedAt: timeutil.TimeStamp(result.JobStatus.CreatedTime / 1000),
- CompletedAt: timeutil.TimeStamp(result.JobStatus.CompletedTime / 1000),
- Status: result.JobStatus.State,
- JobId: result.ID,
- }
- }
-
- type ClusterLog struct {
- }
-
- type TrainTask struct {
- Command string `json:"command"`
- Name string `json:"name"`
- ImageId string `json:"imageId"`
- ImageUrl string `json:"imageUrl"`
- ResourceSpecId string `json:"resourceSpecId"`
- CenterID []string `json:"centerID"`
- ReplicaNum int `json:"replicaNum"`
- Datasets []ContainerData `json:"datasets"`
- Models []ContainerData `json:"models"`
- Code ContainerData `json:"code"`
- BootFile string `json:"bootFile"`
- OutPut ContainerData `json:"output"`
- Params models.Parameters
- Spec *models.Specification
- }
-
- type CreateTrainTaskResponse struct {
- StartedAt int64 `json:"startedAt"`
- RunSec int64 `json:"runSec"`
- CompletedAt int64 `json:"completedAt"`
- CreatedAt int64 `json:"createdAt"`
- UpdatedAt int64 `json:"updatedAt"`
- Desc string `json:"desc"`
- JobID string `json:"id"`
- Name string `json:"name"`
- Status string `json:"status"`
- UserID string `json:"userId"`
- }
-
- type ClusterType string
-
- const (
- OpenICloudbrainOne ClusterType = "OpenICloudbrainOne"
- OpenICloudbrainTwo ClusterType = "OpenICloudbrainTwo"
- C2Net ClusterType = "C2Net"
- )
-
- func (t ClusterType) GetParentCluster() string {
- switch t {
- case OpenICloudbrainTwo, OpenICloudbrainOne:
- return models.OpenICluster
- case C2Net:
- return models.C2NetCluster
- }
- return ""
- }
-
- func (t ClusterType) GetCloudbrainType() int {
- switch t {
- case OpenICloudbrainOne:
- return models.TypeCloudBrainOne
- case OpenICloudbrainTwo:
- return models.TypeCloudBrainTwo
- case C2Net:
- return models.TypeC2Net
- }
- return -1
- }
-
- func GetClusterTypeFromCloudbrainType(t int) ClusterType {
- switch t {
- case models.TypeCloudBrainOne:
- return OpenICloudbrainOne
- case models.TypeCloudBrainTwo:
- return OpenICloudbrainTwo
- case models.TypeC2Net:
- return C2Net
- }
- return ""
- }
|