mp_common

type Header struct {
	Key   string `json:"key"`
	Value string `json:"value"`
}

type IEmbeddingReq interface {
	Data() map[string]interface{}
}

type IEmbeddingResp interface {
	String() string
	Data() (map[string]interface{}, bool)
}

type ILLMReq interface {
	Stream() bool
	Data() map[string]interface{}
	OpenAIReq() (*OpenAILLMReq, bool)
}

type ILLMResp interface {
	String() string
	Data() (map[string]interface{}, bool)
	OpenAIResp() (*OpenAILLMResp, bool)
}

type IRerankReq interface {
	Data() map[string]interface{}
}

type IRerankResp interface {
	String() string
	Data() (interface{}, bool)
}

type OpenAILLMReq struct {
	Model    string      `json:"model" validate:"required"`
	Messages []OpenAIMsg `json:"messages" validate:"required"`

	Stream         bool `json:"stream,omitempty"`
	MaxTokens      int  `json:"max_tokens,omitempty"`
	EnableThinking bool `json:"enable_thinking,omitempty"`

	Temperature float64 `json:"temperature,omitempty"`
	TopP        float64 `json:"top_p,omitempty"`
	TopK        int     `json:"top_k,omitempty"`
	MinP        float64 `json:"min_p,omitempty"`

	PresencePenalty   float64 `json:"presence_penalty,omitempty"`
	FrequencyPenalty  float64 `json:"frequency_penalty,omitempty"`
	RepetitionPenalty float64 `json:"repetition_penalty,omitempty"`

	Seed        int  `json:"seed,omitempty"`
	Logprobs    bool `json:"logprobs,omitempty"`
	TopLogprobs int  `json:"top_logprobs,omitempty"`
	N           int  `json:"n,omitempty"`

	ResponseFormat *OpenAIResponseFormat `json:"response_format,omitempty"`
	Stop           string                `json:"stop,omitempty"`

	User string `json:"user,omitempty"`
}

type OpenAILLMResp struct {
	ID      string             `json:"id"`      // 唯一标识
	Object  string             `json:"object"`  // 固定为 "chat.completion"
	Created int                `json:"created"` // 时间戳（秒）
	Model   string             `json:"model"`   // 使用的模型
	Choices []OpenAIRespChoice `json:"choices"` // 生成结果列表
	Usage   OpenAIRespUsage    `json:"usage"`   // token 使用统计
}

type OpenAIMsg struct {
	Role    string `json:"role" validate:"required"` // "system" | "user" | "assistant" | "function(已弃用)"
	Content string `json:"content" validate:"required"`
}

type OpenAIRespChoice struct {
	Index        int                  `json:"index"`             // 选项索引
	Message      *OpenAIRespChoiceMsg `json:"message,omitempty"` // 非流式生成的消息
	Delta        *OpenAIRespChoiceMsg `json:"delta,omitempty"`   // 流式生成的消息
	FinishReason string               `json:"finish_reason"`     // 停止原因
}

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

type OpenAIRespUsage struct {
	CompletionTokens int `json:"completion_tokens"` // 输出 token 数
	PromptTokens     int `json:"prompt_tokens"`     // 输入 token 数
	TotalTokens      int `json:"total_tokens"`      // 总 token 数
}

type OpenAIResponseFormat struct {
	Type string `json:"type"` // "text" | "json"
}