gpt

type ChatChoice struct {
	Message ChatMessage `json:"message"`
}

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

type ChatRequest struct {
	Model            string        `json:"model"`
	Messages         []ChatMessage `json:"messages"`
	Stream           bool          `json:"stream"`
	MaxTokens        int64         `json:"max_tokens"`
	FrequencyPenalty float64       `json:"frequency_penalty"`
	PresencePenalty  float64       `json:"presence_penalty"`
	Temperature      float64       `json:"temperature"`
	TopP             float64       `json:"top_p"`
}

type ChatResponse struct {
	Id      string       `json:"id"`
	Choices []ChatChoice `json:"choices"`
}

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

type DoubaoChat struct {
	Api   string
	Model string
	Key   string
}

type DoubaoChatChoice struct {
	Index        int                `json:"index"`
	FinishReason string             `json:"finish_reason"`
	Message      DoubaoChatMessage  `json:"message"`
	Logprobs     DoubaoChatLogprobs `json:"logprobs"`
}

type DoubaoChatFunction struct {
	Name      string `json:"name"`
	Arguments string `json:"arguments"`
}

type DoubaoChatLogprobs struct {
	Content []DoubaoChatTokenLogprob `json:"content"`
}

type DoubaoChatMessage struct {
	Role      string               `json:"role"`
	Content   string               `json:"content"`
	ToolCalls []DoubaoChatToolCall `json:"tool_calls"`
}

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

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

type DoubaoChatTokenLogprob struct {
	Token       string                 `json:"token"`
	Bytes       []int                  `json:"bytes"`
	Logprob     float64                `json:"logprob"`
	TopLogprobs []DoubaoChatTopLogprob `json:"top_logprobs"`
}

type DoubaoChatToolCall struct {
	Id       string             `json:"id"`
	Type     string             `json:"type"`
	Function DoubaoChatFunction `json:"function"`
}

type DoubaoChatTopLogprob struct {
	Token   string  `json:"token"`
	Bytes   []int   `json:"bytes"`
	Logprob float64 `json:"logprob"`
}

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

type Gpt interface {
	Init(context.Context) error
	Deinit(context.Context) error
	Chat(context.Context, *ChatRequest) (ChatResponse, error)
}