mirror of
https://github.com/coaidev/coai.git
synced 2025-05-19 13:00:14 +09:00
35 lines
794 B
Go
35 lines
794 B
Go
package zhipuai
|
|
|
|
import "chat/globals"
|
|
|
|
const (
|
|
ChatGLMTurbo = "chatglm_turbo"
|
|
ChatGLMPro = "chatglm_pro"
|
|
ChatGLMStd = "chatglm_std"
|
|
ChatGLMLite = "chatglm_lite"
|
|
)
|
|
|
|
type Payload struct {
|
|
ApiKey string `json:"api_key"`
|
|
Exp int64 `json:"exp"`
|
|
TimeStamp int64 `json:"timestamp"`
|
|
}
|
|
|
|
type ChatRequest struct {
|
|
Prompt []globals.Message `json:"prompt"`
|
|
Temperature *float32 `json:"temperature,omitempty"`
|
|
TopP *float32 `json:"top_p,omitempty"`
|
|
Ref *ChatRef `json:"ref,omitempty"`
|
|
}
|
|
|
|
type ChatRef struct {
|
|
Enable *bool `json:"enable,omitempty"`
|
|
SearchQuery *string `json:"search_query,omitempty"`
|
|
}
|
|
|
|
type Occurrence struct {
|
|
Code int `json:"code"`
|
|
Msg string `json:"msg"`
|
|
Success bool `json:"success"`
|
|
}
|