provider

type AliasesModelProvider interface {
	ModelAlias(model string, models []string) (string, error)
}

type Base64ImageSourceParam struct {
	Data      string `json:"data" format:"byte"`
	MediaType string `json:"media_type"`
	Type      string `json:"type"`
}

type Base64PDFSourceParam struct {
	Data      string `json:"data" format:"byte"`
	MediaType string `json:"media_type"`
	Type      string `json:"type"`
}

type CachedModels struct {
	Models    []Info    `json:"models"`
	Provider  string    `json:"provider"`
	CachedAt  time.Time `json:"cached_at"`
	ExpiresAt time.Time `json:"expires_at"`
}

type ContentBlockParamUnion struct {
	OfText       *TextBlockParam       `json:",omitzero,inline"`
	OfImage      *ImageBlockParam      `json:",omitzero,inline"`
	OfToolUse    *ToolUseBlockParam    `json:",omitzero,inline"`
	OfToolResult *ToolResultBlockParam `json:",omitzero,inline"`
	OfThinking   *ThinkingBlockParam   `json:",omitzero,inline"`
}

type ContentBlockSourceContentUnionParam struct {
	OfString                    *string                               `json:",omitzero,inline"`
	OfContentBlockSourceContent []ContentBlockSourceContentUnionParam `json:",omitzero,inline"`
}

type ContentBlockSourceParam struct {
	Content ContentBlockSourceContentUnionParam `json:"content,omitzero"`
	Type    string                              `json:"type"`
}

type ContentBlockType string

type GenerateContext struct {
	StepID  string
	RunID   string
	Context context.Context
}

type ImageBlockParam struct {
	Source ImageBlockParamSourceUnion `json:"source,omitzero"`
	Type   string                     `json:"type"` // image
}

type ImageBlockParamSourceUnion struct {
	OfBase64 *Base64ImageSourceParam `json:",omitzero,inline"`
	OfURL    *URLImageSourceParam    `json:",omitzero,inline"`
}

type Info struct {
	ID          string   `json:"id"`
	Name        string   `json:"name,omitempty"`
	Provider    string   `json:"provider"`
	CreatedAt   string   `json:"created_at,omitempty"`
	Deprecated  bool     `json:"deprecated,omitempty"`
	Description string   `json:"description,omitempty"`
	Features    []string `json:"features,omitempty"`
}

type LocalModelProvider interface {
	// contains filtered or unexported methods
}

type Message struct {
	Role        string                   `json:"role"`
	IsTruncated bool                     `json:"-"`
	Content     []ContentBlockParamUnion `json:"content"`
}

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

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

type PlainTextSourceParam struct {
	Data      string `json:"data"`
	MediaType string `json:"media_type"`
	Type      string `json:"type"`
}

type Provider interface {
	// Generate generates a response from the model
	Generate(ctx GenerateContext, request *Request, progressChan chan<- pkgEvents.ExecutionEvent) ([]Message, *execcontext.TokenUsage, error)

	// GetName returns the provider name
	GetName() string

	// ListModels dynamically queries available models from the provider API
	ListModels(ctx context.Context) ([]Info, error)

	// Close cleans up resources
	Close() error
}

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

type Request struct {
	Model        string       `json:"model"`
	Messages     []Message    `json:"messages"`
	SystemPrompt string       `json:"system_prompt,omitempty"`
	Temperature  *float64     `json:"temperature,omitempty"`
	MaxTokens    *int         `json:"max_tokens,omitempty"`
	TopP         *float64     `json:"top_p,omitempty"`
	Stop         []string     `json:"stop,omitempty"`
	Tools        []tools.Tool `json:"tools,omitempty"`

	// Additional metadata
	RequestID string                 `json:"request_id,omitempty"`
	Metadata  map[string]interface{} `json:"metadata,omitempty"`
}

type TextBlockParam struct {
	Text string `json:"text"`
	Type string `json:"type"` // text
}

type ThinkingBlockParam struct {
	Signature string `json:"signature"`
	Thinking  string `json:"thinking"`
	Type      string `json:"type"` // thinking
}

type ToolResultBlockParam struct {
	ToolUseID string `json:"tool_use_id"`
	IsError   *bool  `json:"is_error,omitzero"`
	Content   string `json:"content,omitzero"`
	Type      string `json:"type"` // tool_result
}

type ToolUseBlockParam struct {
	ID    string          `json:"id"`
	Input json.RawMessage `json:"input,omitzero"`
	Name  string          `json:"name"`
	Type  string          `json:"type"` // tool_use
}

type URLImageSourceParam struct {
	URL  string `json:"url"`
	Type string `json:"type"`
}

type URLPDFSourceParam struct {
	URL  string `json:"url"`
	Type string `json:"type"`
}