1
0
mirror of https://github.com/fumiama/deepinfra.git synced 2026-06-05 00:32:46 +08:00

feat(chat): add param isusersystem to Modelize

This commit is contained in:
源文雨
2025-03-30 21:51:28 +09:00
parent 577f83008e
commit 71ec2f7c08

View File

@@ -72,8 +72,11 @@ func (l *Log) Add(grp int64, usr, txt string, isbot, isatme bool) {
l.m[grp] = msgs[:len(msgs)-1]
}
func (l *Log) Modelize(p model.Protocol, grp int64, sysp string) deepinfra.Model {
m := p.System(sysp)
func (l *Log) Modelize(p model.Protocol, grp int64, sysp string, isusersystem bool) deepinfra.Model {
m := p
if sysp != "" && !isusersystem {
m.System(sysp)
}
l.mu.RLock()
defer l.mu.RUnlock()
sz := len(l.m[grp])
@@ -82,10 +85,14 @@ func (l *Log) Modelize(p model.Protocol, grp int64, sysp string) deepinfra.Model
}
for i, msg := range l.m[grp] {
if i%2 == 0 { // is user
if i == 0 && isusersystem {
_ = m.User(sysp + "\n\n" + msg.String())
continue
}
_ = m.User(msg.String())
} else {
_ = m.Assistant(msg.String())
continue
}
_ = m.Assistant(msg.String())
}
return m
}