From 347ad047f9eb0e2109cd3abc1fecaccd1b6e01c3 Mon Sep 17 00:00:00 2001
From: creamlike1024 <creamlike1024@outlook.com>
Date: Fri, 31 Oct 2025 19:45:37 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E4=BF=9D=E5=AD=98=E9=87=8D=E5=AE=9A?=
 =?UTF-8?q?=E5=90=91=E4=BF=A1=E6=81=AF=E5=88=B0=20task.Properties?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 model/task.go       | 5 ++++-
 relay/relay_task.go | 5 +++++
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/model/task.go b/model/task.go
index a8c3a7d4d..6c5ade536 100644
--- a/model/task.go
+++ b/model/task.go
@@ -72,7 +72,10 @@ func (t *Task) GetData(v any) error {
 }
 
 type Properties struct {
-	Input string `json:"input"`
+	Input             string `json:"input"`
+	IsModelMapped     bool   `json:"is_model_mapped,omitempty"`
+	OriginModelName   string `json:"origin_model_name,omitempty"`
+	UpstreamModelName string `json:"upstream_model_name,omitempty"`
 }
 
 func (m *Properties) Scan(val interface{}) error {
diff --git a/relay/relay_task.go b/relay/relay_task.go
index db543319a..148b4ac38 100644
--- a/relay/relay_task.go
+++ b/relay/relay_task.go
@@ -214,6 +214,11 @@ func RelayTaskSubmit(c *gin.Context, info *relaycommon.RelayInfo) (taskErr *dto.
 	task.Quota = quota
 	task.Data = taskData
 	task.Action = info.Action
+	task.Properties = model.Properties{
+		IsModelMapped:     info.IsModelMapped,
+		UpstreamModelName: info.UpstreamModelName,
+		OriginModelName:   info.OriginModelName,
+	}
 	err = task.Insert()
 	if err != nil {
 		taskErr = service.TaskErrorWrapper(err, "insert_task_failed", http.StatusInternalServerError)