main.go

 1package main
 2
 3import (
 4	"context"
 5	"fmt"
 6
 7	"github.com/charmbracelet/crush/internal/ai"
 8	"github.com/charmbracelet/crush/internal/ai/providers"
 9)
10
11func main() {
12	provider := providers.NewOpenAIProvider(providers.WithOpenAIApiKey("$OPENAI_API_KEY"))
13	model := provider.LanguageModel("gpt-4o")
14
15	response, err := model.Generate(context.Background(), ai.Call{
16		Prompt: ai.Prompt{
17			ai.NewUserMessage("Hello"),
18		},
19		Temperature: ai.FloatOption(0.7),
20	})
21	if err != nil {
22		fmt.Println(err)
23		return
24	}
25
26	fmt.Println("Assistant: ", response.Content.Text())
27	fmt.Println("Usage:", response.Usage)
28}