@@ -1,6 +1,13 @@
package cluster
package cluster
import (
import (
"errors"
"fmt"
"io/ioutil"
"path"
"strings"
"time"
"code.gitea.io/gitea/entity"
"code.gitea.io/gitea/entity"
"code.gitea.io/gitea/manager/client/grampus"
"code.gitea.io/gitea/manager/client/grampus"
"code.gitea.io/gitea/models"
"code.gitea.io/gitea/models"
@@ -15,12 +22,6 @@ import (
"code.gitea.io/gitea/services/ai_task_service/schedule"
"code.gitea.io/gitea/services/ai_task_service/schedule"
"code.gitea.io/gitea/services/ai_task_service/storage_helper"
"code.gitea.io/gitea/services/ai_task_service/storage_helper"
"code.gitea.io/gitea/services/cloudbrain/cloudbrainTask"
"code.gitea.io/gitea/services/cloudbrain/cloudbrainTask"
"errors"
"fmt"
"io/ioutil"
"path"
"strings"
"time"
)
)
type C2NetClusterAdapter struct {
type C2NetClusterAdapter struct {
@@ -266,7 +267,12 @@ func convertToCreateNoteBookTaskResponse(res *models.GrampusNotebookRestartRespo
}
}
}
}
func (c C2NetClusterAdapter) DeleteNoteBook(entity.JobIdAndVersionId) error {
func (c C2NetClusterAdapter) DeleteNoteBook(opts entity.JobIdAndVersionId) error {
_, err := grampus.DeleteJob(opts.JobID, string(models.JobTypeDebug))
if err != nil {
log.Error("DeleteNoteBook(%s) failed:%v", opts, err)
return err
}
return nil
return nil
}
}
@@ -588,9 +594,15 @@ func convertGrampus2TrainRes(res *models.CreateGrampusJobResponse) *entity.Creat
}
}
}
}
func (c C2NetClusterAdapter) DeleteTrainJob(entity.JobIdAndVersionId) error {
func (c C2NetClusterAdapter) DeleteTrainJob(opts entity.JobIdAndVersionId) error {
_, err := grampus.DeleteJob(opts.JobID)
if err != nil {
log.Error("Delete train job(%s) failed:%v", opts, err)
return err
}
return nil
return nil
}
}
func (c C2NetClusterAdapter) StopTrainJob(opts entity.JobIdAndVersionId) error {
func (c C2NetClusterAdapter) StopTrainJob(opts entity.JobIdAndVersionId) error {
_, err := grampus.StopJob(opts.JobID, string(models.JobTypeDebug))
_, err := grampus.StopJob(opts.JobID, string(models.JobTypeDebug))
if err != nil {
if err != nil {