mirror of
https://github.com/songquanpeng/one-api.git
synced 2025-09-29 14:46:38 +08:00
41 lines
1.1 KiB
Go
41 lines
1.1 KiB
Go
package palm
|
|
|
|
import "one-api/types"
|
|
|
|
type PaLMChatMessage struct {
|
|
Author string `json:"author"`
|
|
Content string `json:"content"`
|
|
}
|
|
|
|
type PaLMFilter struct {
|
|
Reason string `json:"reason"`
|
|
Message string `json:"message"`
|
|
}
|
|
|
|
type PaLMPrompt struct {
|
|
Messages []PaLMChatMessage `json:"messages"`
|
|
}
|
|
|
|
type PaLMChatRequest struct {
|
|
Prompt PaLMPrompt `json:"prompt"`
|
|
Temperature float64 `json:"temperature,omitempty"`
|
|
CandidateCount int `json:"candidateCount,omitempty"`
|
|
TopP float64 `json:"topP,omitempty"`
|
|
TopK int `json:"topK,omitempty"`
|
|
}
|
|
|
|
type PaLMError struct {
|
|
Code int `json:"code"`
|
|
Message string `json:"message"`
|
|
Status string `json:"status"`
|
|
}
|
|
|
|
type PaLMChatResponse struct {
|
|
Candidates []PaLMChatMessage `json:"candidates"`
|
|
Messages []types.ChatCompletionMessage `json:"messages"`
|
|
Filters []PaLMFilter `json:"filters"`
|
|
Error PaLMError `json:"error"`
|
|
Usage *types.Usage `json:"usage,omitempty"`
|
|
Model string `json:"model,omitempty"`
|
|
}
|