#5346 模型流控开关修改

Merged
ychao_1983 merged 5 commits from zouap_dev into V20240402 1 month ago
  1. +2
    -2
      routers/api/v1/repo/attachments.go
  2. +1
    -2
      services/ai_task_service/task/opt_handler.go

+ 2
- 2
routers/api/v1/repo/attachments.go View File

@@ -94,7 +94,7 @@ func NewMultipart(ctx *context.APIContext) {
defer mutex.Unlock()
datasetId := ctx.QueryInt64("dataset_id")
fileName := ctx.Query("file_name")
re, err := routeRepo.NewMultipartForApi(ctx.Context, true)
re, err := routeRepo.NewMultipartForApi(ctx.Context, !ignore)
if err != nil {
ctx.JSON(200, map[string]string{
"result_code": "-1",
@@ -186,7 +186,7 @@ func NewModelMultipart(ctx *context.APIContext) {
modelMutex.Lock()
defer modelMutex.Unlock()
fileName := ctx.Query("file_name")
re, err := routeRepo.NewModelMultipartForApi(ctx.Context, true)
re, err := routeRepo.NewModelMultipartForApi(ctx.Context, !ignore)
if err != nil {
ctx.JSON(200, map[string]string{
"result_code": "-1",


+ 1
- 2
services/ai_task_service/task/opt_handler.go View File

@@ -734,11 +734,10 @@ func (DefaultCreationHandler) CheckNotebookCount(ctx *context.CreationContext) *
}

func (DefaultCreationHandler) CheckImageAvailable(ctx *context.CreationContext) *response.BizError {
log.Info("start to gcu CheckImageAvailable 1")
if ctx.Request.SourceCloudbrainId <= 0 {
return nil
}
log.Info("start to gcu CheckImageAvailable 2")
log.Info("start to gcu CheckImageAvailable")
if ctx.Request.ComputeSource.Name == models.GCU {
image_name := ctx.Request.ImageUrl
log.Info("image_name=" + image_name)


Loading…
Cancel
Save