gen

type FullRequest struct {
	Request
	Prompts []prompt.Prompt `json:"prompts"`
}

type Gen interface {
	Generator(options ...Option) *Generator
}

type Generator struct {
	Prompter Prompter
	Request  Request
}

type Model struct {
	Provider    string `json:"provider"`
	Name        string `json:"name"`
	Description string `json:"description,omitempty"`

	InputContentTypes []string `json:"input_content_types,omitempty"`

	InputMaxToken  int `json:"input_max_token,omitempty"`
	OutputMaxToken int `json:"output_max_token,omitempty"`

	SupportTools            bool `json:"support_tools,omitempty"`
	SupportStructuredOutput bool `json:"support_structured_output,omitempty"`
}

type Option func(generator *Generator) *Generator

type Prompter interface {
	SetRequest(request Request)
	Prompt(prompts ...prompt.Prompt) (*Response, error)
}

type Request struct {
	Model Model `json:"model"`

	SystemPrompt string `json:"system_prompt"`

	StopSequences []string `json:"stop_sequences"`
	TopP          float64  `json:"top_p"`
	Temperature   float64  `json:"temperature"`
	MaxTokens     int      `json:"max_tokens"`

	OutputSchema *schema.JSON `json:"output_schema"`

	Tools      []tools.Tool `json:"tools"`
	ToolConfig *tools.Tool  `json:"tool"`
}

type Response struct {
	Texts []string     `json:"texts,omitempty"`
	Tools []tools.Call `json:"tools,omitempty"`

	Metadata models.Metadata `json:"metadata,omitempty"`
}