models

type AmazonBedrockModel struct {
	*BaseModel
	Client                BedrockClient          `json:"-"`
	ClientKwargs          map[string]interface{} `json:"client_kwargs"`
	CustomRoleConversions map[string]string      `json:"custom_role_conversions"`
	InferenceConfig       map[string]interface{} `json:"inference_config"`
	GuardrailConfig       map[string]interface{} `json:"guardrail_config"`
	AdditionalModelFields map[string]interface{} `json:"additional_model_fields"`
}

type AudioData struct {
	Data   string `json:"data"`   // Base64 encoded audio data
	Format string `json:"format"` // "mp3", "wav", "ogg", etc.
	URL    string `json:"url,omitempty"`
}

type AzureOpenAIServerModel struct {
	*OpenAIServerModel
	APIVersion   string `json:"api_version"`
	Deployment   string `json:"deployment"`
	ResourceName string `json:"resource_name"`
}

type BaseModel struct {
	FlattenMessagesAsText bool   `json:"flatten_messages_as_text"`
	ToolNameKey           string `json:"tool_name_key"`
	ToolArgumentsKey      string `json:"tool_arguments_key"`
	ModelID               string `json:"model_id"`

	// Deprecated token counting properties (maintained for compatibility)
	LastInputTokenCount  *int `json:"_last_input_token_count,omitempty"`
	LastOutputTokenCount *int `json:"_last_output_token_count,omitempty"`

	CustomParams map[string]interface{} `json:"custom_params,omitempty"`
}

type BedrockClient interface {
	Converse(input map[string]interface{}) (map[string]interface{}, error)
	ConverseStream(input map[string]interface{}) (<-chan map[string]interface{}, error)
}

type ChatMessage struct {
	Role       string                 `json:"role"`
	Content    *string                `json:"content,omitempty"`
	ToolCalls  []ChatMessageToolCall  `json:"tool_calls,omitempty"`
	Raw        interface{}            `json:"-"` // Stores raw output from API
	TokenUsage *monitoring.TokenUsage `json:"token_usage,omitempty"`
}

type ChatMessageStreamDelta struct {
	Content    *string                `json:"content,omitempty"`
	ToolCalls  []ChatMessageToolCall  `json:"tool_calls,omitempty"`
	TokenUsage *monitoring.TokenUsage `json:"token_usage,omitempty"`
}

type ChatMessageToolCall struct {
	Function ChatMessageToolCallDefinition `json:"function"`
	ID       string                        `json:"id"`
	Type     string                        `json:"type"`
}

type ChatMessageToolCallDefinition struct {
	Arguments   interface{} `json:"arguments"`
	Name        string      `json:"name"`
	Description *string     `json:"description,omitempty"`
}

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

type GenerateOptions struct {
	StopSequences     []string               `json:"stop_sequences,omitempty"`
	ResponseFormat    *ResponseFormat        `json:"response_format,omitempty"`     // Enhanced structured format
	ResponseFormatRaw map[string]interface{} `json:"response_format_raw,omitempty"` // Raw format for backwards compatibility
	ToolsToCallFrom   []tools.Tool           `json:"tools_to_call_from,omitempty"`
	Grammar           map[string]string      `json:"grammar,omitempty"`
	MaxTokens         *int                   `json:"max_tokens,omitempty"`
	Temperature       *float64               `json:"temperature,omitempty"`
	TopP              *float64               `json:"top_p,omitempty"`
	TopK              *int                   `json:"top_k,omitempty"`
	FrequencyPenalty  *float64               `json:"frequency_penalty,omitempty"`
	PresencePenalty   *float64               `json:"presence_penalty,omitempty"`
	Seed              *int                   `json:"seed,omitempty"`
	ValidateOutput    bool                   `json:"validate_output"`  // Whether to validate structured output
	RetryOnFailure    bool                   `json:"retry_on_failure"` // Whether to retry on validation failure
	MaxRetries        int                    `json:"max_retries"`      // Maximum number of retries
	CustomParams      map[string]interface{} `json:"custom_params,omitempty"`
}

type ImageURL struct {
	URL    string `json:"url"`
	Detail string `json:"detail,omitempty"` // "low", "high", "auto"
}

type InferenceClientModel struct {
	*BaseModel
	Provider string            `json:"provider"`
	Client   interface{}       `json:"-"` // HTTP client or SDK client
	Token    string            `json:"-"` // API token
	BaseURL  string            `json:"base_url"`
	Headers  map[string]string `json:"headers"`
}

type JSONSchema struct {
	Name        string                 `json:"name"`
	Description string                 `json:"description,omitempty"`
	Schema      map[string]interface{} `json:"schema"`
	Strict      bool                   `json:"strict,omitempty"`
}

type LiteLLMModel struct {
	*BaseModel
	APIBase               string            `json:"api_base"`
	APIKey                string            `json:"-"` // API key (not serialized)
	Headers               map[string]string `json:"headers"`
	Client                interface{}       `json:"-"`       // HTTP client
	Timeout               int               `json:"timeout"` // Request timeout in seconds
	CustomRoleConversions map[string]string `json:"custom_role_conversions"`
	Organization          string            `json:"organization"`
	Project               string            `json:"project"`
}

type MediaContent struct {
	Type      MediaType   `json:"type"`
	Text      *string     `json:"text,omitempty"`
	ImageURL  *ImageURL   `json:"image_url,omitempty"`
	AudioData *AudioData  `json:"audio_data,omitempty"`
	VideoData *VideoData  `json:"video_data,omitempty"`
	Metadata  interface{} `json:"metadata,omitempty"`
}

type MediaType string

type MessageRole string

type Model interface {
	// Generate generates a response from the model
	Generate(
		messages []interface{},
		options *GenerateOptions,
	) (*ChatMessage, error)

	// GenerateStream generates a streaming response (if supported)
	GenerateStream(
		messages []interface{},
		options *GenerateOptions,
	) (<-chan *ChatMessageStreamDelta, error)

	// ParseToolCalls parses tool calls from message content
	ParseToolCalls(message *ChatMessage) (*ChatMessage, error)

	// ToDict converts the model to a dictionary representation
	ToDict() map[string]interface{}

	// GetModelID returns the model identifier
	GetModelID() string

	// SupportsStreaming returns true if the model supports streaming
	SupportsStreaming() bool

	// Close cleans up model resources
	Close() error
}

type ModelType string

type MultimodalMessage struct {
	Role    string          `json:"role"`
	Content []*MediaContent `json:"content"`
}

type MultimodalSupport struct {
	MaxImageSize int64    `json:"max_image_size"` // Maximum image size in bytes
	MaxAudioSize int64    `json:"max_audio_size"` // Maximum audio size in bytes
	MaxVideoSize int64    `json:"max_video_size"` // Maximum video size in bytes
	ImageFormats []string `json:"image_formats"`  // Supported image formats
	AudioFormats []string `json:"audio_formats"`  // Supported audio formats
	VideoFormats []string `json:"video_formats"`  // Supported video formats
}

type OpenAIServerModel struct {
	*BaseModel
	BaseURL string            `json:"base_url"`
	APIKey  string            `json:"-"` // API key (not serialized)
	Headers map[string]string `json:"headers"`
	Client  interface{}       `json:"-"`       // HTTP client
	Timeout int               `json:"timeout"` // Request timeout in seconds
}

type ResponseFormat struct {
	Type        string                 `json:"type"`                  // "json_object", "json_schema", "text"
	JSONSchema  *JSONSchema            `json:"json_schema,omitempty"` // For structured JSON output
	Schema      map[string]interface{} `json:"schema,omitempty"`      // Raw schema definition
	Strict      bool                   `json:"strict,omitempty"`      // Whether to enforce strict schema compliance
	Name        string                 `json:"name,omitempty"`        // Name for the response format
	Description string                 `json:"description,omitempty"` // Description of the format
}

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

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

type StructuredOutput struct {
	Content  interface{}            `json:"content"`  // The parsed content
	Raw      string                 `json:"raw"`      // Raw text output
	Format   *ResponseFormat        `json:"format"`   // The format used
	Valid    bool                   `json:"valid"`    // Whether output is valid according to schema
	Errors   []string               `json:"errors"`   // Validation errors if any
	Metadata map[string]interface{} `json:"metadata"` // Additional metadata
}

type TransformersModel struct {
	*BaseModel
	ModelPath        string                 `json:"model_path"`
	TokenizerPath    string                 `json:"tokenizer_path"`
	Device           string                 `json:"device"`
	TorchDtype       string                 `json:"torch_dtype"`
	ModelKwargs      map[string]interface{} `json:"model_kwargs"`
	TokenizerKwargs  map[string]interface{} `json:"tokenizer_kwargs"`
	StoppingCriteria interface{}            `json:"-"` // StoppingCriteriaList
}

type VideoData struct {
	Data   string `json:"data"`   // Base64 encoded video data
	Format string `json:"format"` // "mp4", "webm", "avi", etc.
	URL    string `json:"url,omitempty"`
}