client

type AnthropicClient struct {
	BaseUrl  string
	ApiKey   string
	Model    string
	Provider string
	Thinking bool
	// contains filtered or unexported fields
}

type AnthropicErrorResponse struct {
	Type  string `json:"type"`
	Error struct {
		Type    string `json:"type"`
		Message string `json:"message"`
	} `json:"error"`
}

type AnthropicResponse struct {
	ID      string `json:"id"`
	Type    string `json:"type"`
	Role    string `json:"role"`
	Content []struct {
		Type string `json:"type"`
		Text string `json:"text"`
	} `json:"content"`
	Model string `json:"model"`
	Usage struct {
		InputTokens  int `json:"input_tokens"`
		OutputTokens int `json:"output_tokens"`
	} `json:"usage"`
}

type AnthropicStreamChunk struct {
	Type  string `json:"type"`
	Index int    `json:"index,omitempty"`
	Delta struct {
		Type        string  `json:"type"`
		Text        string  `json:"text"`
		Thinking    *string `json:"thinking,omitempty"`
		PartialJSON *string `json:"partial_json,omitempty"`
	} `json:"delta,omitempty"`
	Usage *struct {
		InputTokens  int `json:"input_tokens"`
		OutputTokens int `json:"output_tokens"`
	} `json:"usage,omitempty"`
}

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

type ChatCompletionRequest struct {
	Model         string                  `json:"model"`
	Messages      []ChatCompletionMessage `json:"messages"`
	Stream        bool                    `json:"stream,omitempty"`
	StreamOptions *StreamOptions          `json:"stream_options,omitempty"`
	Thinking      *ThinkingOptions        `json:"thinking,omitempty"`
}

type ChatCompletionResponse struct {
	ID      string `json:"id"`
	Object  string `json:"object"`
	Created int64  `json:"created"`
	Model   string `json:"model"`
	Choices []struct {
		Index   int `json:"index"`
		Message struct {
			Role    string `json:"role"`
			Content string `json:"content"`
		} `json:"message"`
		FinishReason string `json:"finish_reason"`
	} `json:"choices"`
	Usage struct {
		PromptTokens            int                      `json:"prompt_tokens"`
		CompletionTokens        int                      `json:"completion_tokens"`
		TotalTokens             int                      `json:"total_tokens"`
		CompletionTokensDetails *CompletionTokensDetails `json:"completion_tokens_details,omitempty"`
	} `json:"usage"`
}

type CompletionTokensDetails struct {
	ReasoningTokens int `json:"reasoning_tokens"`
	ThinkingTokens  int `json:"thinking_tokens"`
}

type ModelClient interface {
	Request(prompt string, stream bool) (*ResponseMetrics, error)
	GetProtocol() string
	GetModel() string
	SetLogger(logger *logger.Logger) // 设置日志记录器
}

type OpenAIClient struct {
	Model    string
	Provider string
	Thinking bool // 是否开启 thinking 模式
	// contains filtered or unexported fields
}

type OpenAIErrorResponse struct {
	Error struct {
		Message string `json:"message"`
		Type    string `json:"type"`
		Param   string `json:"param,omitempty"`
		Code    string `json:"code,omitempty"`
	} `json:"error"`
}

type ResponseMetrics struct {
	// 时间相关指标
	TimeToFirstToken time.Duration // 首个 token 的响应时间 (TTFT)
	TotalTime        time.Duration // 总耗时 (从请求开始到完全结束)

	// 网络连接指标
	DNSTime          time.Duration // DNS解析时间
	ConnectTime      time.Duration // TCP连接建立时间
	TLSHandshakeTime time.Duration // TLS握手时间
	TargetIP         string        // 目标服务器IP地址

	// 内容指标
	PromptTokens     int // 输入 token 数量
	ThinkingTokens   int // 思考/推理 token 数量
	CompletionTokens int // 输出 token 数量 (用于TPS计算)

	// 错误信息
	ErrorMessage string // 错误信息（如果有）
}

type StreamOptions struct {
	IncludeUsage bool `json:"include_usage"`
}

type StreamResponseChunk struct {
	ID      string `json:"id"`
	Object  string `json:"object"`
	Created int64  `json:"created"`
	Model   string `json:"model"`
	Choices []struct {
		Index int `json:"index"`
		Delta struct {
			ThinkingContent *string `json:"reasoning_content,omitempty"`
			Content         string  `json:"content"`
		} `json:"delta"`
		FinishReason *string `json:"finish_reason"`
	} `json:"choices"`
	Usage *struct {
		PromptTokens            int                      `json:"prompt_tokens"`
		CompletionTokens        int                      `json:"completion_tokens"`
		TotalTokens             int                      `json:"total_tokens"`
		CompletionTokensDetails *CompletionTokensDetails `json:"completion_tokens_details,omitempty"`
	} `json:"usage,omitempty"`
}

type ThinkingOptions struct {
	Type string `json:"type"`
}