wehub-resource-sync
e071084ebe
govulncheck / govulncheck (push) Has been cancelled
Lint / golangci-lint (push) Has been cancelled
Run Tests / Unit Tests (push) Has been cancelled
Run Tests / Etcd Integration Tests (push) Has been cancelled
Harness (E2E) / Harnesses (mock LLM) (push) Has been cancelled
Harness (E2E) / Provider harnesses (live LLM conformance) (push) Has been cancelled
46 lines
1.2 KiB
Go
46 lines
1.2 KiB
Go
package ai
|
|
|
|
// History is a convenience for accumulating conversation messages
|
|
// with automatic truncation. Use it to build Request.Messages for
|
|
// multi-turn conversations.
|
|
//
|
|
// hist := ai.NewHistory(50)
|
|
// hist.Add("user", "hello")
|
|
// resp, _ := m.Generate(ctx, &ai.Request{Messages: hist.Messages(), Prompt: "next"})
|
|
// hist.Add("assistant", resp.Reply)
|
|
type History struct {
|
|
messages []Message
|
|
limit int
|
|
}
|
|
|
|
// NewHistory creates an empty History. limit controls the maximum
|
|
// number of messages retained (0 = unlimited).
|
|
func NewHistory(limit int) *History {
|
|
return &History{limit: limit}
|
|
}
|
|
|
|
// Add appends a message and truncates if over limit.
|
|
func (h *History) Add(role string, content any) {
|
|
h.messages = append(h.messages, Message{Role: role, Content: content})
|
|
if h.limit > 0 && len(h.messages) > h.limit {
|
|
h.messages = h.messages[len(h.messages)-h.limit:]
|
|
}
|
|
}
|
|
|
|
// Messages returns a copy of the accumulated messages.
|
|
func (h *History) Messages() []Message {
|
|
out := make([]Message, len(h.messages))
|
|
copy(out, h.messages)
|
|
return out
|
|
}
|
|
|
|
// Len returns the number of messages.
|
|
func (h *History) Len() int {
|
|
return len(h.messages)
|
|
}
|
|
|
|
// Reset clears all messages.
|
|
func (h *History) Reset() {
|
|
h.messages = nil
|
|
}
|