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}