mirror of
https://github.com/linux-do/new-api.git
synced 2025-09-17 16:06:38 +08:00
79 lines
2.2 KiB
Go
79 lines
2.2 KiB
Go
package baidu
|
|
|
|
import (
|
|
"one-api/dto"
|
|
"time"
|
|
)
|
|
|
|
type BaiduMessage struct {
|
|
Role string `json:"role"`
|
|
Content string `json:"content"`
|
|
}
|
|
|
|
type BaiduChatRequest struct {
|
|
Messages []BaiduMessage `json:"messages"`
|
|
Temperature float64 `json:"temperature,omitempty"`
|
|
TopP float64 `json:"top_p,omitempty"`
|
|
PenaltyScore float64 `json:"penalty_score,omitempty"`
|
|
Stream bool `json:"stream,omitempty"`
|
|
System string `json:"system,omitempty"`
|
|
DisableSearch bool `json:"disable_search,omitempty"`
|
|
EnableCitation bool `json:"enable_citation,omitempty"`
|
|
MaxOutputTokens int `json:"max_output_tokens,omitempty"`
|
|
UserId string `json:"user_id,omitempty"`
|
|
}
|
|
|
|
type Error struct {
|
|
ErrorCode int `json:"error_code"`
|
|
ErrorMsg string `json:"error_msg"`
|
|
}
|
|
|
|
type BaiduChatResponse struct {
|
|
Id string `json:"id"`
|
|
Object string `json:"object"`
|
|
Created int64 `json:"created"`
|
|
Result string `json:"result"`
|
|
IsTruncated bool `json:"is_truncated"`
|
|
NeedClearHistory bool `json:"need_clear_history"`
|
|
Usage dto.Usage `json:"usage"`
|
|
Error
|
|
}
|
|
|
|
type BaiduChatStreamResponse struct {
|
|
BaiduChatResponse
|
|
SentenceId int `json:"sentence_id"`
|
|
IsEnd bool `json:"is_end"`
|
|
}
|
|
|
|
type BaiduEmbeddingRequest struct {
|
|
Input []string `json:"input"`
|
|
}
|
|
|
|
type BaiduEmbeddingData struct {
|
|
Object string `json:"object"`
|
|
Embedding []float64 `json:"embedding"`
|
|
Index int `json:"index"`
|
|
}
|
|
|
|
type BaiduEmbeddingResponse struct {
|
|
Id string `json:"id"`
|
|
Object string `json:"object"`
|
|
Created int64 `json:"created"`
|
|
Data []BaiduEmbeddingData `json:"data"`
|
|
Usage dto.Usage `json:"usage"`
|
|
Error
|
|
}
|
|
|
|
type BaiduAccessToken struct {
|
|
AccessToken string `json:"access_token"`
|
|
Error string `json:"error,omitempty"`
|
|
ErrorDescription string `json:"error_description,omitempty"`
|
|
ExpiresIn int64 `json:"expires_in,omitempty"`
|
|
ExpiresAt time.Time `json:"-"`
|
|
}
|
|
|
|
type BaiduTokenResponse struct {
|
|
ExpiresIn int `json:"expires_in"`
|
|
AccessToken string `json:"access_token"`
|
|
}
|