42 lines
896 B
Go
42 lines
896 B
Go
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)
|
|
}
|