357 lines
11 KiB
Go
357 lines
11 KiB
Go
package dto
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"one-api/types"
|
|
)
|
|
|
|
type SimpleResponse struct {
|
|
Usage `json:"usage"`
|
|
Error any `json:"error"`
|
|
}
|
|
|
|
// GetOpenAIError 从动态错误类型中提取OpenAIError结构
|
|
func (s *SimpleResponse) GetOpenAIError() *types.OpenAIError {
|
|
return GetOpenAIError(s.Error)
|
|
}
|
|
|
|
type TextResponse struct {
|
|
Id string `json:"id"`
|
|
Object string `json:"object"`
|
|
Created int64 `json:"created"`
|
|
Model string `json:"model"`
|
|
Choices []OpenAITextResponseChoice `json:"choices"`
|
|
Usage `json:"usage"`
|
|
}
|
|
|
|
type OpenAITextResponseChoice struct {
|
|
Index int `json:"index"`
|
|
Message `json:"message"`
|
|
FinishReason string `json:"finish_reason"`
|
|
}
|
|
|
|
type OpenAITextResponse struct {
|
|
Id string `json:"id"`
|
|
Model string `json:"model"`
|
|
Object string `json:"object"`
|
|
Created any `json:"created"`
|
|
Choices []OpenAITextResponseChoice `json:"choices"`
|
|
Error any `json:"error,omitempty"`
|
|
Usage `json:"usage"`
|
|
}
|
|
|
|
// GetOpenAIError 从动态错误类型中提取OpenAIError结构
|
|
func (o *OpenAITextResponse) GetOpenAIError() *types.OpenAIError {
|
|
return GetOpenAIError(o.Error)
|
|
}
|
|
|
|
type OpenAIEmbeddingResponseItem struct {
|
|
Object string `json:"object"`
|
|
Index int `json:"index"`
|
|
Embedding []float64 `json:"embedding"`
|
|
}
|
|
|
|
type OpenAIEmbeddingResponse struct {
|
|
Object string `json:"object"`
|
|
Data []OpenAIEmbeddingResponseItem `json:"data"`
|
|
Model string `json:"model"`
|
|
Usage `json:"usage"`
|
|
}
|
|
|
|
type FlexibleEmbeddingResponseItem struct {
|
|
Object string `json:"object"`
|
|
Index int `json:"index"`
|
|
Embedding any `json:"embedding"`
|
|
}
|
|
|
|
type FlexibleEmbeddingResponse struct {
|
|
Object string `json:"object"`
|
|
Data []FlexibleEmbeddingResponseItem `json:"data"`
|
|
Model string `json:"model"`
|
|
Usage `json:"usage"`
|
|
}
|
|
|
|
type ChatCompletionsStreamResponseChoice struct {
|
|
Delta ChatCompletionsStreamResponseChoiceDelta `json:"delta,omitempty"`
|
|
Logprobs *any `json:"logprobs"`
|
|
FinishReason *string `json:"finish_reason"`
|
|
Index int `json:"index"`
|
|
}
|
|
|
|
type ChatCompletionsStreamResponseChoiceDelta struct {
|
|
Content *string `json:"content,omitempty"`
|
|
ReasoningContent *string `json:"reasoning_content,omitempty"`
|
|
Reasoning *string `json:"reasoning,omitempty"`
|
|
Role string `json:"role,omitempty"`
|
|
ToolCalls []ToolCallResponse `json:"tool_calls,omitempty"`
|
|
}
|
|
|
|
func (c *ChatCompletionsStreamResponseChoiceDelta) SetContentString(s string) {
|
|
c.Content = &s
|
|
}
|
|
|
|
func (c *ChatCompletionsStreamResponseChoiceDelta) GetContentString() string {
|
|
if c.Content == nil {
|
|
return ""
|
|
}
|
|
return *c.Content
|
|
}
|
|
|
|
func (c *ChatCompletionsStreamResponseChoiceDelta) GetReasoningContent() string {
|
|
if c.ReasoningContent == nil && c.Reasoning == nil {
|
|
return ""
|
|
}
|
|
if c.ReasoningContent != nil {
|
|
return *c.ReasoningContent
|
|
}
|
|
return *c.Reasoning
|
|
}
|
|
|
|
func (c *ChatCompletionsStreamResponseChoiceDelta) SetReasoningContent(s string) {
|
|
c.ReasoningContent = &s
|
|
c.Reasoning = &s
|
|
}
|
|
|
|
type ToolCallResponse struct {
|
|
// Index is not nil only in chat completion chunk object
|
|
Index *int `json:"index,omitempty"`
|
|
ID string `json:"id,omitempty"`
|
|
Type any `json:"type"`
|
|
Function FunctionResponse `json:"function"`
|
|
}
|
|
|
|
func (c *ToolCallResponse) SetIndex(i int) {
|
|
c.Index = &i
|
|
}
|
|
|
|
type FunctionResponse struct {
|
|
Description string `json:"description,omitempty"`
|
|
Name string `json:"name,omitempty"`
|
|
// call function with arguments in JSON format
|
|
Parameters any `json:"parameters,omitempty"` // request
|
|
Arguments string `json:"arguments"` // response
|
|
}
|
|
|
|
type ChatCompletionsStreamResponse struct {
|
|
Id string `json:"id"`
|
|
Object string `json:"object"`
|
|
Created int64 `json:"created"`
|
|
Model string `json:"model"`
|
|
SystemFingerprint *string `json:"system_fingerprint"`
|
|
Choices []ChatCompletionsStreamResponseChoice `json:"choices"`
|
|
Usage *Usage `json:"usage"`
|
|
}
|
|
|
|
func (c *ChatCompletionsStreamResponse) IsFinished() bool {
|
|
if len(c.Choices) == 0 {
|
|
return false
|
|
}
|
|
return c.Choices[0].FinishReason != nil && *c.Choices[0].FinishReason != ""
|
|
}
|
|
|
|
func (c *ChatCompletionsStreamResponse) IsToolCall() bool {
|
|
if len(c.Choices) == 0 {
|
|
return false
|
|
}
|
|
return len(c.Choices[0].Delta.ToolCalls) > 0
|
|
}
|
|
|
|
func (c *ChatCompletionsStreamResponse) GetFirstToolCall() *ToolCallResponse {
|
|
if c.IsToolCall() {
|
|
return &c.Choices[0].Delta.ToolCalls[0]
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (c *ChatCompletionsStreamResponse) ClearToolCalls() {
|
|
if !c.IsToolCall() {
|
|
return
|
|
}
|
|
for choiceIdx := range c.Choices {
|
|
for callIdx := range c.Choices[choiceIdx].Delta.ToolCalls {
|
|
c.Choices[choiceIdx].Delta.ToolCalls[callIdx].ID = ""
|
|
c.Choices[choiceIdx].Delta.ToolCalls[callIdx].Type = nil
|
|
c.Choices[choiceIdx].Delta.ToolCalls[callIdx].Function.Name = ""
|
|
}
|
|
}
|
|
}
|
|
|
|
func (c *ChatCompletionsStreamResponse) Copy() *ChatCompletionsStreamResponse {
|
|
choices := make([]ChatCompletionsStreamResponseChoice, len(c.Choices))
|
|
copy(choices, c.Choices)
|
|
return &ChatCompletionsStreamResponse{
|
|
Id: c.Id,
|
|
Object: c.Object,
|
|
Created: c.Created,
|
|
Model: c.Model,
|
|
SystemFingerprint: c.SystemFingerprint,
|
|
Choices: choices,
|
|
Usage: c.Usage,
|
|
}
|
|
}
|
|
|
|
func (c *ChatCompletionsStreamResponse) GetSystemFingerprint() string {
|
|
if c.SystemFingerprint == nil {
|
|
return ""
|
|
}
|
|
return *c.SystemFingerprint
|
|
}
|
|
|
|
func (c *ChatCompletionsStreamResponse) SetSystemFingerprint(s string) {
|
|
c.SystemFingerprint = &s
|
|
}
|
|
|
|
type ChatCompletionsStreamResponseSimple struct {
|
|
Choices []ChatCompletionsStreamResponseChoice `json:"choices"`
|
|
Usage *Usage `json:"usage"`
|
|
}
|
|
|
|
type CompletionsStreamResponse struct {
|
|
Choices []struct {
|
|
Text string `json:"text"`
|
|
FinishReason string `json:"finish_reason"`
|
|
} `json:"choices"`
|
|
}
|
|
|
|
type Usage struct {
|
|
PromptTokens int `json:"prompt_tokens"`
|
|
CompletionTokens int `json:"completion_tokens"`
|
|
TotalTokens int `json:"total_tokens"`
|
|
PromptCacheHitTokens int `json:"prompt_cache_hit_tokens,omitempty"`
|
|
|
|
PromptTokensDetails InputTokenDetails `json:"prompt_tokens_details"`
|
|
CompletionTokenDetails OutputTokenDetails `json:"completion_tokens_details"`
|
|
InputTokens int `json:"input_tokens"`
|
|
OutputTokens int `json:"output_tokens"`
|
|
InputTokensDetails *InputTokenDetails `json:"input_tokens_details"`
|
|
// OpenRouter Params
|
|
Cost any `json:"cost,omitempty"`
|
|
}
|
|
|
|
type InputTokenDetails struct {
|
|
CachedTokens int `json:"cached_tokens"`
|
|
CachedCreationTokens int `json:"-"`
|
|
TextTokens int `json:"text_tokens"`
|
|
AudioTokens int `json:"audio_tokens"`
|
|
ImageTokens int `json:"image_tokens"`
|
|
}
|
|
|
|
type OutputTokenDetails struct {
|
|
TextTokens int `json:"text_tokens"`
|
|
AudioTokens int `json:"audio_tokens"`
|
|
ReasoningTokens int `json:"reasoning_tokens"`
|
|
}
|
|
|
|
type OpenAIResponsesResponse struct {
|
|
ID string `json:"id"`
|
|
Object string `json:"object"`
|
|
CreatedAt int `json:"created_at"`
|
|
Status string `json:"status"`
|
|
Error any `json:"error,omitempty"`
|
|
IncompleteDetails *IncompleteDetails `json:"incomplete_details,omitempty"`
|
|
Instructions string `json:"instructions"`
|
|
MaxOutputTokens int `json:"max_output_tokens"`
|
|
Model string `json:"model"`
|
|
Output []ResponsesOutput `json:"output"`
|
|
ParallelToolCalls bool `json:"parallel_tool_calls"`
|
|
PreviousResponseID string `json:"previous_response_id"`
|
|
Reasoning *Reasoning `json:"reasoning"`
|
|
Store bool `json:"store"`
|
|
Temperature float64 `json:"temperature"`
|
|
ToolChoice string `json:"tool_choice"`
|
|
Tools []map[string]any `json:"tools"`
|
|
TopP float64 `json:"top_p"`
|
|
Truncation string `json:"truncation"`
|
|
Usage *Usage `json:"usage"`
|
|
User json.RawMessage `json:"user"`
|
|
Metadata json.RawMessage `json:"metadata"`
|
|
}
|
|
|
|
// GetOpenAIError 从动态错误类型中提取OpenAIError结构
|
|
func (o *OpenAIResponsesResponse) GetOpenAIError() *types.OpenAIError {
|
|
return GetOpenAIError(o.Error)
|
|
}
|
|
|
|
type IncompleteDetails struct {
|
|
Reasoning string `json:"reasoning"`
|
|
}
|
|
|
|
type ResponsesOutput struct {
|
|
Type string `json:"type"`
|
|
ID string `json:"id"`
|
|
Status string `json:"status"`
|
|
Role string `json:"role"`
|
|
Content []ResponsesOutputContent `json:"content"`
|
|
}
|
|
|
|
type ResponsesOutputContent struct {
|
|
Type string `json:"type"`
|
|
Text string `json:"text"`
|
|
Annotations []interface{} `json:"annotations"`
|
|
}
|
|
|
|
const (
|
|
BuildInToolWebSearchPreview = "web_search_preview"
|
|
BuildInToolFileSearch = "file_search"
|
|
)
|
|
|
|
const (
|
|
BuildInCallWebSearchCall = "web_search_call"
|
|
)
|
|
|
|
const (
|
|
ResponsesOutputTypeItemAdded = "response.output_item.added"
|
|
ResponsesOutputTypeItemDone = "response.output_item.done"
|
|
)
|
|
|
|
// ResponsesStreamResponse 用于处理 /v1/responses 流式响应
|
|
type ResponsesStreamResponse struct {
|
|
Type string `json:"type"`
|
|
Response *OpenAIResponsesResponse `json:"response,omitempty"`
|
|
Delta string `json:"delta,omitempty"`
|
|
Item *ResponsesOutput `json:"item,omitempty"`
|
|
}
|
|
|
|
// GetOpenAIError 从动态错误类型中提取OpenAIError结构
|
|
func GetOpenAIError(errorField any) *types.OpenAIError {
|
|
if errorField == nil {
|
|
return nil
|
|
}
|
|
|
|
switch err := errorField.(type) {
|
|
case types.OpenAIError:
|
|
return &err
|
|
case *types.OpenAIError:
|
|
return err
|
|
case map[string]interface{}:
|
|
// 处理从JSON解析来的map结构
|
|
openaiErr := &types.OpenAIError{}
|
|
if errType, ok := err["type"].(string); ok {
|
|
openaiErr.Type = errType
|
|
}
|
|
if errMsg, ok := err["message"].(string); ok {
|
|
openaiErr.Message = errMsg
|
|
}
|
|
if errParam, ok := err["param"].(string); ok {
|
|
openaiErr.Param = errParam
|
|
}
|
|
if errCode, ok := err["code"]; ok {
|
|
openaiErr.Code = errCode
|
|
}
|
|
return openaiErr
|
|
case string:
|
|
// 处理简单字符串错误
|
|
return &types.OpenAIError{
|
|
Type: "error",
|
|
Message: err,
|
|
}
|
|
default:
|
|
// 未知类型，尝试转换为字符串
|
|
return &types.OpenAIError{
|
|
Type: "unknown_error",
|
|
Message: fmt.Sprintf("%v", err),
|
|
}
|
|
}
|
|
}
|