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}