gen

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

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

type Generator struct {
	Prompter Prompter
	Request  Request
}

type Model struct {
	Provider string `json:"provider"`
	Name     string `json:"name"`

	Config map[string]any `json:"config,omitempty"`

	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 {
	Context context.Context `json:"-"`

	Model        Model  `json:"model"`
	SystemPrompt string `json:"system_prompt,omitempty"`

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

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

	TopP             *float64 `json:"top_p,omitempty"`
	TopK             *int     `json:"top_k,omitempty"`
	Temperature      *float64 `json:"temperature,omitempty"`
	MaxTokens        *int     `json:"max_tokens,omitempty"`
	FrequencyPenalty *float64 `json:"frequency_penalty,omitempty"`
	PresencePenalty  *float64 `json:"presence_penalty,omitempty"`
	StopSequences    []string `json:"stop_sequences,omitempty"`
}

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

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