ai

type Attr struct {
	Key   string
	Value string
}

type CaptureConfig struct {
	Enabled      bool
	DebugMode    bool
	OutputFile   string
	ProviderName string
}

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

type CapturedAttr struct {
	Key   string `json:"key"`
	Value string `json:"value"`
}

type CapturedError struct {
	Message string `json:"message"`
	Type    string `json:"type"`
}

type CapturedFunction struct {
	Name        string                 `json:"name"`
	Description string                 `json:"description"`
	Parameters  map[string]interface{} `json:"parameters,omitempty"`
}

type CapturedPrompt struct {
	Name        string                 `json:"name"`
	Text        string                 `json:"text"`
	Instruction string                 `json:"instruction"`
	Functions   []CapturedFunction     `json:"functions,omitempty"`
	Context     map[string]interface{} `json:"context,omitempty"`
}

type FunctionDeclaration struct {
	Name        string
	Description string
	Parameters  *Schema
	Response    *Schema
}

type FunctionResponse struct {
	Name     string
	Response map[string]any
}

type Gen interface {
	GenerateContent(ctx context.Context, p Prompt, debug bool, args ...string) (string, error)
	GenerateContentAttr(ctx context.Context, prompt Prompt, debug bool, attrs []Attr) (string, error)
	CountTokens(ctx context.Context, p Prompt, debug bool, args ...string) (*TokenCount, error)
	CountTokensAttr(ctx context.Context, p Prompt, debug bool, attrs []Attr) (*TokenCount, error)
	GetStatus() *Status
}

type HandlerFunc func(ctx context.Context, attr map[string]any) (map[string]any, error)

type Image struct {
	Type     string `yaml:"type"`
	Filename string `yaml:"filename"`
	Data     []byte `yaml:"data"`
}

type Interaction struct {
	ID          string                 `json:"id"`
	Timestamp   time.Time              `json:"timestamp"`
	Prompt      CapturedPrompt         `json:"prompt"`
	Args        []string               `json:"args"`
	Attrs       []CapturedAttr         `json:"attrs,omitempty"`
	Response    string                 `json:"response"`
	Error       *CapturedError         `json:"error,omitempty"`
	Duration    time.Duration          `json:"duration"`
	LLMProvider string                 `json:"llm_provider"`
	Tools       []string               `json:"tools"`
	Context     map[string]interface{} `json:"context,omitempty"`
	Debug       bool                   `json:"debug"`
}

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

type MockGen struct {
	ResponseQueue []string
	CallCounts    map[string]int
	UsedPrompts   []Prompt
	LastAttrs     []Attr
	// contains filtered or unexported fields
}

type Prompt struct {
	Name           string   `yaml:"name"`
	Instruction    string   `yaml:"instruction"`
	Text           string   `yaml:"text"`
	Images         []*Image `yaml:"images"`
	RequiredTools  []string `yaml:"required_tools"`
	Functions      []*FunctionDeclaration
	ResponseSchema *Schema                `yaml:"response_schema"`
	Handlers       map[string]HandlerFunc `yaml:"-"`
	ModelName      string                 `yaml:"model_name"`
	MaxTokens      int32                  `yaml:"max_tokens"`
	Temperature    float32                `yaml:"temperature"`
	TopP           float32                `yaml:"top_p"`
}

type RetryConfig struct {
	Enabled        bool
	MaxRetries     int
	InitialBackoff time.Duration
}

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

type Schema struct {
	Type          Type               `yaml:"type"`
	Format        string             `yaml:"format"`
	Title         string             `yaml:"title"`
	Description   string             `yaml:"description"`
	Nullable      bool               `yaml:"nullable"`
	Items         *Schema            `yaml:"items"`
	MinItems      int64              `yaml:"min_items"`
	MaxItems      int64              `yaml:"max_items"`
	Enum          []string           `yaml:"enum"`
	Properties    map[string]*Schema `yaml:"properties"`
	Required      []string           `yaml:"required"`
	MinProperties int64              `yaml:"min_properties"`
	MaxProperties int64              `yaml:"max_properties"`
	Minimum       float64            `yaml:"minimum"`
	Maximum       float64            `yaml:"maximum"`
	MinLength     int64              `yaml:"min_length"`
	MaxLength     int64              `yaml:"max_length"`
	Pattern       string             `yaml:"pattern"`
}

type Status struct {
	Connected bool
	Model     string
	Backend   string
	Message   string
}

type TokenCount struct {
	TotalTokens  int32
	InputTokens  int32
	OutputTokens int32
}

type Type int32