groq_test.go

 1package providertests
 2
 3import (
 4	"net/http"
 5	"os"
 6	"testing"
 7
 8	"charm.land/fantasy"
 9	"charm.land/fantasy/providers/groq"
10	"gopkg.in/dnaeon/go-vcr.v4/pkg/recorder"
11)
12
13func TestGroqCommon(t *testing.T) {
14	testCommon(t, []builderPair{
15		{"groq-kimi-k2-0905", builderGroqProvider, nil, nil},
16	})
17}
18
19func builderGroqProvider(t *testing.T, r *recorder.Recorder) (fantasy.LanguageModel, error) {
20	provider, err := groq.New(
21		groq.WithAPIKey(os.Getenv("FANTASY_GROQ_API_KEY")),
22		groq.WithHTTPClient(&http.Client{Transport: r}),
23	)
24	if err != nil {
25		return nil, err
26	}
27	return provider.LanguageModel(t.Context(), "moonshotai/kimi-k2-instruct-0905")
28}