cerebras_test.go

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