OpenAI

import "github.com/nuulab/goflow/pkg/llm/openai"
 
// Default: gpt-4o
llm := openai.New("")
 
// Specify model
llm := openai.New("", openai.WithModel("gpt-4o-mini"))
 
// With custom options
llm := openai.New("sk-your-key",
    openai.WithModel("gpt-4o"),
    openai.WithTimeout(60*time.Second),
)
 
// Azure OpenAI Support
llm := openai.New("your-azure-key",
    openai.WithBaseURL("https://your-resource.openai.azure.com/openai/deployments/gpt-4o"),
)

// Simple text completion
response, err := llm.Generate(ctx, "What is Go?")
 
// Chat completion with history
response, err := llm.GenerateChat(ctx, []core.Message{
    {Role: core.RoleSystem, Content: "You are a helpful assistant."},
    {Role: core.RoleUser, Content: "What is Go?"},
})
 
// With generation options
response, err := llm.Generate(ctx, "Tell me a joke",
    core.WithTemperature(0.9),
    core.WithMaxTokens(100),
)

// Stream text response
stream, err := llm.Stream(ctx, "Write a story about a robot")
for chunk := range stream {
    fmt.Print(chunk)
}
 
// Stream chat response
stream, err := llm.StreamChat(ctx, messages)
for chunk := range stream {
    fmt.Print(chunk)
}