复制项目

pkg/common/convert/agent.go

@@ -0,0 +1,41 @@
+package convert
+
+import (
+	"time"
+
+	"git.imall.cloud/openim/chat/pkg/common/db/table/bot"
+	pbbot "git.imall.cloud/openim/chat/pkg/protocol/bot"
+	"github.com/openimsdk/tools/utils/datautil"
+)
+
+func DB2PBAgent(a *bot.Agent) *pbbot.Agent {
+	return &pbbot.Agent{
+		UserID:     a.UserID,
+		Nickname:   a.NickName,
+		FaceURL:    a.FaceURL,
+		Url:        a.Url,
+		Key:        a.Key,
+		Identity:   a.Identity,
+		Model:      a.Model,
+		Prompts:    a.Prompts,
+		CreateTime: a.CreateTime.UnixMilli(),
+	}
+}
+
+func PB2DBAgent(a *pbbot.Agent) *bot.Agent {
+	return &bot.Agent{
+		UserID:     a.UserID,
+		NickName:   a.Nickname,
+		FaceURL:    a.FaceURL,
+		Key:        a.Key,
+		Url:        a.Url,
+		Identity:   a.Identity,
+		Model:      a.Model,
+		Prompts:    a.Prompts,
+		CreateTime: time.UnixMilli(a.CreateTime),
+	}
+}
+
+func BatchDB2PBAgent(a []*bot.Agent) []*pbbot.Agent {
+	return datautil.Batch(DB2PBAgent, a)
+}