Fix: Local model deployment status error

This commit is contained in:
Liujian
2025-02-14 17:24:41 +08:00
parent aef9805bdf
commit 26a50b9a79
3 changed files with 19 additions and 1 deletions
+11
View File
@@ -3,12 +3,15 @@ package ai_local
import (
"context"
"encoding/json"
"errors"
"fmt"
"io"
"math"
"net/http"
"strings"
"gorm.io/gorm"
"github.com/APIParkLab/APIPark/module/router"
"github.com/APIParkLab/APIPark/model/plugin_model"
@@ -194,6 +197,14 @@ func (i *imlLocalModelController) DeployStart(ctx *gin.Context, input *ai_local_
func (i *imlLocalModelController) initAILocalService(ctx context.Context, model string, teamID string) error {
err := i.transaction.Transaction(ctx, func(ctx context.Context) error {
_, err := i.serviceModule.Get(ctx, model)
if err == nil {
return nil
} else {
if !errors.Is(err, gorm.ErrRecordNotFound) {
return err
}
}
catalogueInfo, err := i.catalogueModule.DefaultCatalogue(ctx)
if err != nil {
return err