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}