generation

type Catalog struct {
	Models []Model
}

type Chain struct {
	// contains filtered or unexported fields
}

type Error struct {
	Kind      ErrorKind
	Operation string
	Provider  string
	Model     string
	Retryable bool
	Err       error
}

type ErrorKind string

type Generator interface {
	Generate(context.Context, Request) (Result, error)
	Models(context.Context, ModelQuery) (Catalog, error)
}

type GenkitGenerator struct {
	// contains filtered or unexported fields
}

type GenkitModel struct {
	Model firebaseai.Model
	Info  Model
}

type Message struct {
	Role    Role
	Content string
}

type Model struct {
	ID                       string
	Provider                 string
	Name                     string
	Purposes                 []Purpose
	ContextWindowTokens      int
	SupportsStructuredOutput bool
	Cloud                    bool
}

type ModelQuery struct {
	Purpose Purpose
}

type Purpose string

type Request struct {
	Purpose         Purpose
	Locale          string
	System          string
	Prompt          string
	Messages        []Message
	ModelID         string
	AffinityKey     string
	StructuredHint  string
	MaxOutputTokens int
	Temperature     float64
}

type Result struct {
	Text         string
	Provider     string
	Model        string
	FinishReason string
	Usage        *Usage
	Latency      time.Duration
}

type Role string

type Usage struct {
	InputTokens  int
	OutputTokens int
	TotalTokens  int
}