From a162b6e7c2a20f19d569e59ae33777d93cd42e4d Mon Sep 17 00:00:00 2001
From: wizardchen <wizardchen@tencent.com>
Date: Wed, 26 Nov 2025 22:30:12 +0800
Subject: [PATCH] refactor: Enhance type sanitization for model creation to
 improve security

---
 internal/handler/model.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/internal/handler/model.go b/internal/handler/model.go
index 1bbf0899c..2666bb50c 100644
--- a/internal/handler/model.go
+++ b/internal/handler/model.go
@@ -97,7 +97,7 @@ func (h *ModelHandler) CreateModel(c *gin.Context) {
 	model := &types.Model{
 		TenantID:    tenantID,
 		Name:        secutils.SanitizeForLog(req.Name),
-		Type:        req.Type,
+		Type:        types.ModelType(secutils.SanitizeForLog(string(req.Type))),
 		Source:      req.Source,
 		Description: secutils.SanitizeForLog(req.Description),
 		Parameters:  req.Parameters,