mirror of
https://github.com/songquanpeng/one-api.git
synced 2025-09-30 15:16:39 +08:00
60 lines
1.7 KiB
Go
60 lines
1.7 KiB
Go
package zhipu_v4
|
|
|
|
import (
|
|
"one-api/relay/channel/openai"
|
|
"time"
|
|
)
|
|
|
|
type Message struct {
|
|
Role string `json:"role,omitempty"`
|
|
Content string `json:"content,omitempty"`
|
|
ToolCalls any `json:"tool_calls,omitempty"`
|
|
ToolCallId any `json:"tool_call_id,omitempty"`
|
|
}
|
|
|
|
type Request struct {
|
|
Model string `json:"model"`
|
|
Stream bool `json:"stream,omitempty"`
|
|
Messages []Message `json:"messages"`
|
|
Temperature float64 `json:"temperature,omitempty"`
|
|
TopP float64 `json:"top_p,omitempty"`
|
|
MaxTokens int `json:"max_tokens,omitempty"`
|
|
Stop []string `json:"stop,omitempty"`
|
|
RequestId string `json:"request_id,omitempty"`
|
|
Tools any `json:"tools,omitempty"`
|
|
ToolChoice any `json:"tool_choice,omitempty"`
|
|
}
|
|
|
|
type TextResponseChoice struct {
|
|
Index int `json:"index"`
|
|
Message `json:"message"`
|
|
FinishReason string `json:"finish_reason"`
|
|
}
|
|
|
|
type Response struct {
|
|
Id string `json:"id"`
|
|
Created int64 `json:"created"`
|
|
Model string `json:"model"`
|
|
TextResponseChoices []TextResponseChoice `json:"choices"`
|
|
openai.Usage `json:"usage"`
|
|
openai.Error `json:"error"`
|
|
}
|
|
|
|
type StreamResponseChoice struct {
|
|
Index int `json:"index,omitempty"`
|
|
Delta Message `json:"delta"`
|
|
FinishReason *string `json:"finish_reason,omitempty"`
|
|
}
|
|
|
|
type StreamResponse struct {
|
|
Id string `json:"id"`
|
|
Created int64 `json:"created"`
|
|
Choices []StreamResponseChoice `json:"choices"`
|
|
openai.Usage `json:"usage"`
|
|
}
|
|
|
|
type tokenData struct {
|
|
Token string
|
|
ExpiryTime time.Time
|
|
}
|