cmd

type Completion struct {
	Data Data `json:"data"`
}

type Data struct {
	Text string `json:"text"`
}

type ImageGenerationConfig struct {
	NumberOfImages int     `json:"numberOfImages"`
	Quality        string  `json:"quality"`
	CfgScale       float64 `json:"cfgScale"`
	Height         int     `json:"height"`
	Width          int     `json:"width"`
	Seed           int64   `json:"seed"`
}

type PromptFn func(string) string

type Request struct {
	Prompt      string  `json:"prompt"`
	MaxTokens   int     `json:"maxTokens"`
	Temperature float64 `json:"temperature,omitempty"`

	StopSequences []string `json:"stop_sequences,omitempty"`
}

type Response struct {
	Completions []Completion `json:"completions"`
}

type TextToImageParams struct {
	Text string `json:"text"`
}

type TitanImageRequest struct {
	TaskType              string                `json:"taskType"`
	TextToImageParams     TextToImageParams     `json:"textToImageParams"`
	ImageGenerationConfig ImageGenerationConfig `json:"imageGenerationConfig"`
}

type TitanImageResponse struct {
	Images []string `json:"images"`
}