Skip to content

Commit 709b8ad

Browse files
QinYuuuuDev Agent
andauthored
Add NotebookType support and update scene validation (#606)
Co-authored-by: Dev Agent <[email protected]>
1 parent 61b8700 commit 709b8ad

File tree

3 files changed

+4
-1
lines changed

3 files changed

+4
-1
lines changed

builder/deploy/common/utils.go

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,8 @@ func GetValidSceneType(deployType int) types.SceneType {
3535
return types.SceneModelFinetune
3636
case types.ServerlessType:
3737
return types.SceneModelServerless
38+
case types.NotebookType:
39+
return types.SceneModelInference
3840
default:
3941
return types.SceneUnknow
4042
}

builder/deploy/common/utils_test.go

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@ func Test_GetValidSceneType(t *testing.T) {
2323
types.InferenceType: types.SceneModelInference,
2424
types.FinetuneType: types.SceneModelFinetune,
2525
types.ServerlessType: types.SceneModelServerless,
26+
types.NotebookType: types.SceneModelInference,
2627
-1: types.SceneUnknow,
2728
}
2829

builder/deploy/deployer.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -995,7 +995,7 @@ func (d *deployer) startAcctForEvaluations(clusterMap map[string]database.Cluste
995995
}
996996
str, err := json.Marshal(event)
997997
if err != nil {
998-
slog.Error("error marshal metering event", slog.Any("event", event), slog.Any("error", err))
998+
slog.Error("error marshal evaluation metering event", slog.Any("event", event), slog.Any("error", err))
999999
return
10001000
}
10011001
err = d.eventPub.PublishMeteringEvent(str)

0 commit comments

Comments
 (0)