main.go

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