diff --git a/builder/deploy/common/utils.go b/builder/deploy/common/utils.go index a893dfb53..b3a0fa763 100644 --- a/builder/deploy/common/utils.go +++ b/builder/deploy/common/utils.go @@ -35,6 +35,8 @@ func GetValidSceneType(deployType int) types.SceneType { return types.SceneModelFinetune case types.ServerlessType: return types.SceneModelServerless + case types.NotebookType: + return types.SceneModelInference default: return types.SceneUnknow } diff --git a/builder/deploy/common/utils_test.go b/builder/deploy/common/utils_test.go index 81f82748d..a218ea553 100644 --- a/builder/deploy/common/utils_test.go +++ b/builder/deploy/common/utils_test.go @@ -23,6 +23,7 @@ func Test_GetValidSceneType(t *testing.T) { types.InferenceType: types.SceneModelInference, types.FinetuneType: types.SceneModelFinetune, types.ServerlessType: types.SceneModelServerless, + types.NotebookType: types.SceneModelInference, -1: types.SceneUnknow, } diff --git a/builder/deploy/deployer.go b/builder/deploy/deployer.go index bcdd200d4..86b2a524a 100644 --- a/builder/deploy/deployer.go +++ b/builder/deploy/deployer.go @@ -995,7 +995,7 @@ func (d *deployer) startAcctForEvaluations(clusterMap map[string]database.Cluste } str, err := json.Marshal(event) if err != nil { - slog.Error("error marshal metering event", slog.Any("event", event), slog.Any("error", err)) + slog.Error("error marshal evaluation metering event", slog.Any("event", event), slog.Any("error", err)) return } err = d.eventPub.PublishMeteringEvent(str)