api

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

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

type Config struct {
	ApiKey string `json:"apiKey"`
}

type ImagesGenerationsReq struct {
	Prompt         string `json:"prompt"`
	N              int    `json:"n"`
	Size           string `json:"size"`
	ResponseFormat string `json:"response_format"`
	User           string `json:"user"`
}

type ImagesGenerationsRes struct {
	Created int64                      `json:"created"`
	Data    []ImagesGenerationsResData `json:"data"`
}

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

type ReqChatCompletions struct {
	Model    string                   `json:"model"`
	Messages []ChatCompletionsMessage `json:"messages"`
	Stream   bool                     `json:"stream"`
}

type Request[RequestBody any, ResponseResult any] struct {
	Url          string
	Method       string
	Headers      map[string]string
	Body         *RequestBody
	Response     *ResponseResult
	StatusCode   int
	HttpResponse *http.Response
	ApiKey       string
}

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

type ResChatCompletionsChoice struct {
	Index        int                    `json:"index"`
	Message      ChatCompletionsMessage `json:"message"`
	FinishReason string                 `json:"finish_reason"`
}