fix(agent): include session history in /btw context
This commit is contained in:
parent
f0ac4d9b5c
commit
20d8c58ca9
2 changed files with 9 additions and 3 deletions
|
|
@ -1555,6 +1555,7 @@ func (al *AgentLoop) askSideQuestion(
|
||||||
var channel, chatID, senderID, senderDisplayName string
|
var channel, chatID, senderID, senderDisplayName string
|
||||||
var media []string
|
var media []string
|
||||||
var activeSkills []string
|
var activeSkills []string
|
||||||
|
var history []providers.Message
|
||||||
var summary string
|
var summary string
|
||||||
if opts != nil {
|
if opts != nil {
|
||||||
channel = opts.Channel
|
channel = opts.Channel
|
||||||
|
|
@ -1570,13 +1571,14 @@ func (al *AgentLoop) askSideQuestion(
|
||||||
Budget: agent.ContextWindow,
|
Budget: agent.ContextWindow,
|
||||||
MaxTokens: agent.MaxTokens,
|
MaxTokens: agent.MaxTokens,
|
||||||
}); err == nil && resp != nil {
|
}); err == nil && resp != nil {
|
||||||
|
history = resp.History
|
||||||
summary = resp.Summary
|
summary = resp.Summary
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
messages := agent.ContextBuilder.BuildMessages(
|
messages := agent.ContextBuilder.BuildMessages(
|
||||||
nil,
|
history,
|
||||||
summary,
|
summary,
|
||||||
question,
|
question,
|
||||||
media,
|
media,
|
||||||
|
|
|
||||||
|
|
@ -286,8 +286,12 @@ func TestProcessMessage_BtwCommandRunsWithoutPersistingHistory(t *testing.T) {
|
||||||
if len(provider.lastMessages) == 0 {
|
if len(provider.lastMessages) == 0 {
|
||||||
t.Fatal("provider did not receive any messages")
|
t.Fatal("provider did not receive any messages")
|
||||||
}
|
}
|
||||||
if len(provider.lastMessages) != 2 {
|
if len(provider.lastMessages) != 4 {
|
||||||
t.Fatalf("provider messages len = %d, want 2 (system + user, no history)", len(provider.lastMessages))
|
t.Fatalf("provider messages len = %d, want 4 (system + prior history + user)", len(provider.lastMessages))
|
||||||
|
}
|
||||||
|
|
||||||
|
if !reflect.DeepEqual(provider.lastMessages[1:3], initialHistory) {
|
||||||
|
t.Fatalf("provider history = %#v, want %#v", provider.lastMessages[1:3], initialHistory)
|
||||||
}
|
}
|
||||||
|
|
||||||
lastMessage := provider.lastMessages[len(provider.lastMessages)-1]
|
lastMessage := provider.lastMessages[len(provider.lastMessages)-1]
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue