internal/llm/provider_more_test.go


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

package llm

import "testing"

func TestWithOptions_Apply(t *testing.T) {
	o := Options{}
	WithModel("m")(&o)
	WithTemperature(0.7)(&o)
	WithMaxTokens(123)(&o)
	WithStop("END")(&o)
	if o.Model != "m" || o.Temperature != 0.7 || o.MaxTokens != 123 || len(o.Stop) != 1 || o.Stop[0] != "END" {
		t.Fatalf("options not applied correctly: %+v", o)
	}
}

func TestNewFromConfig_Success_OpenAI(t *testing.T) {
	// OpenAI success
	oc := Config{Provider: "openai", OpenAIBaseURL: "http://x", OpenAIModel: "gpt"}
	c, err := NewFromConfig(oc, "KEY", "", "")
	if err != nil || c == nil || c.Name() != "openai" || c.DefaultModel() == "" {
		t.Fatalf("openai new: %v %v", c, err)
	}
}