gpt

type ChatChoice struct {
	Index        int         `json:"index"`
	Message      ChatMessage `json:"message"`
	LogProbs     bool        `json:"log_probs"`
	FinishReason string      `json:"finish_reason"`
}

type ChatMessage struct {
	Role    string `json:"role"`
	Content string `json:"content"`
}

type ChatMessages struct {
	Role    string `json:"role"`
	Content string `json:"content"`
}

type ChatPrompt struct {
	Messages []ChatMessages `json:"messages"`
}

type ChatPromptTokensDetails struct {
	CachedTokens int `json:"cached_tokens"`
}

type ChatResponse struct {
	Id      string       `json:"id"`
	Object  string       `json:"object"`
	Created int64        `json:"created"`
	Model   string       `json:"model"`
	Choices []ChatChoice `json:"choices"`
	Usage   ChatUsage    `json:"usage"`
}

type ChatUsage struct {
	PromptTokens        int                     `json:"prompt_tokens"`
	CompletionTokens    int                     `json:"completion_tokens"`
	TotalTokens         int                     `json:"total_tokens"`
	PromptTokensDetails ChatPromptTokensDetails `json:"prompt_tokens_details"`
}

type Config struct {
	Provider string
	Api      string
	Model    string
	Key      string
}

type DoubaoChat struct {
	Api   string
	Model string
	Key   string
}

type DoubaoVision struct {
	Api   string
	Model string
	Key   string
}

type Gpt interface {
	Init(context.Context) error
	Deinit(context.Context) error
	Run(context.Context, string) ([]string, error)
}

type VisionChoice struct {
	Index        int           `json:"index"`
	Message      VisionMessage `json:"message"`
	LogProbs     bool          `json:"log_probs"`
	FinishReason string        `json:"finish_reason"`
}

type VisionContent struct {
	Type     string         `json:"type"`
	Text     string         `json:"text"`
	ImageUrl VisionImageUrl `json:"image_url"`
}

type VisionImageUrl struct {
	Url string `json:"url"`
}

type VisionMessage struct {
	Role    string `json:"role"`
	Content string `json:"content"`
}

type VisionMessages struct {
	Role    string          `json:"role"`
	Content []VisionContent `json:"content"`
}

type VisionPrompt struct {
	Messages []VisionMessages `json:"messages"`
}

type VisionResponse struct {
	Id      string         `json:"id"`
	Object  string         `json:"object"`
	Created int64          `json:"created"`
	Model   string         `json:"model"`
	Choices []VisionChoice `json:"choices"`
	Usage   VisionUsage    `json:"usage"`
}

type VisionUsage struct {
	PromptTokens     int `json:"prompt_tokens"`
	CompletionTokens int `json:"completion_tokens"`
	TotalTokens      int `json:"total_tokens"`
}