internal/lsp/testhelper_capture_llm_test.go


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18

package lsp

import (
    "context"

    "codeberg.org/snonux/hexai/internal/llm"
)

// captureLLM captures messages sent to Chat for assertions.
type captureLLM struct{ msgs []llm.Message }

func (c *captureLLM) Chat(_ context.Context, m []llm.Message, _ ...llm.RequestOption) (string, error) {
    c.msgs = append([]llm.Message{}, m...)
    return "OK", nil
}
func (*captureLLM) Name() string         { return "cap" }
func (*captureLLM) DefaultModel() string { return "m" }