AI Package

ai.WithModelName("googleai/gemini-2.0-flash")

model := genkit.LookupModel(g, "googleai/gemini-2.0-flash")
ai.WithModel(model)

ai.WithPrompt("Tell me a joke about programming.")

ai.WithSystem("You are a helpful coding assistant.")

ai.WithMessages([]*ai.Message{
    {Role: ai.RoleUser, Content: []*ai.Part{ai.NewTextPart("Hello")}},
    {Role: ai.RoleModel, Content: []*ai.Part{ai.NewTextPart("Hi!")}},
})

ai.WithTools(weatherTool, calculatorTool)

ai.WithToolChoice(ai.ToolChoiceRequired) // Force tool use
ai.WithToolChoice(ai.ToolChoiceNone)     // Disable tools
ai.WithToolChoice(ai.ToolChoiceAuto)     // Let model decide

ai.WithConfig(&ai.GenerationCommonConfig{
    Temperature: 0.7,
    MaxOutputTokens: 1000,
    TopP: 0.9,
})

type Person struct {
    Name string `json:"name"`
    Age  int    `json:"age"`
}

ai.WithOutputType(Person{})

ai.WithOutputSchema(map[string]any{
    "type": "object",
    "properties": map[string]any{
        "name": map[string]any{"type": "string"},
    },
})

ai.WithDocs([]*ai.Document{
    ai.Document.FromText("Context document 1"),
    ai.Document.FromText("Context document 2"),
})

ai.WithStreaming(func(ctx context.Context, chunk *ai.ModelResponseChunk) error {
    fmt.Print(chunk.Text())
    return nil
})

type Message struct {
    Role     Role
    Content  []*Part
    Metadata map[string]any
}

// Create parts
ai.NewTextPart("Hello")
ai.NewMediaPart("image/png", imageData)
ai.NewToolRequestPart(name, input)
ai.NewToolResponsePart(name, output)

type ModelResponse struct {
    Message      *Message
    FinishReason FinishReason
    Usage        *GenerationUsage
    Request      *ModelRequest
    Custom       map[string]any
}

// Methods
resp.Text() string
resp.Output(v any) error

type ModelRequest struct {
    Messages []*Message
    Tools    []*ToolDefinition
    Config   any
    Output   *OutputConfig
}

type GenerationCommonConfig struct {
    Temperature     float64
    MaxOutputTokens int
    TopK            int
    TopP            float64
    StopSequences   []string
}