mirror of
https://github.com/coaidev/coai.git
synced 2025-05-19 21:10:18 +09:00
5ddfbbc295
feat: update zhipuai chatglm channel api format v3 to v4 and support glm-4 & glm-4v (#147) Co-Authored-By: Minghan Zhang <112773885+zmh-program@users.noreply.github.com>
131 lines
3.9 KiB
Go
131 lines
3.9 KiB
Go
package zhipuai
|
|
|
|
import "chat/globals"
|
|
|
|
const (
|
|
GLM4 = "glm-4"
|
|
GLM4Vision = "glm-4v"
|
|
GLMTurbo = "glm-3-turbo" // GLM3 Turbo
|
|
GLMPro = "chatglm_pro" // GLM3 Pro (deprecated)
|
|
GLMStd = "chatglm_std" // GLM3 Standard (deprecated)
|
|
GLMLite = "chatglm_lite" // GLM3 Lite (deprecated)
|
|
)
|
|
|
|
type ImageUrl struct {
|
|
Url string `json:"url"`
|
|
Detail *string `json:"detail,omitempty"`
|
|
}
|
|
|
|
type MessageContent struct {
|
|
Type string `json:"type"`
|
|
Text *string `json:"text,omitempty"`
|
|
ImageUrl *ImageUrl `json:"image_url,omitempty"`
|
|
}
|
|
|
|
type MessageContents []MessageContent
|
|
|
|
type Message struct {
|
|
Role string `json:"role"`
|
|
Content interface{} `json:"content"`
|
|
Name *string `json:"name,omitempty"`
|
|
FunctionCall *globals.FunctionCall `json:"function_call,omitempty"` // only `function` role
|
|
ToolCallId *string `json:"tool_call_id,omitempty"` // only `tool` role
|
|
ToolCalls *globals.ToolCalls `json:"tool_calls,omitempty"` // only `assistant` role
|
|
}
|
|
|
|
// ChatRequest is the request body for chatglm
|
|
type ChatRequest struct {
|
|
Model string `json:"model"`
|
|
Messages interface{} `json:"messages"`
|
|
MaxToken *int `json:"max_tokens,omitempty"`
|
|
Stream bool `json:"stream"`
|
|
PresencePenalty *float32 `json:"presence_penalty,omitempty"`
|
|
FrequencyPenalty *float32 `json:"frequency_penalty,omitempty"`
|
|
Temperature *float32 `json:"temperature,omitempty"`
|
|
TopP *float32 `json:"top_p,omitempty"`
|
|
Tools *globals.FunctionTools `json:"tools,omitempty"`
|
|
ToolChoice *interface{} `json:"tool_choice,omitempty"` // string or object
|
|
}
|
|
|
|
// CompletionRequest is the request body for chatglm completion
|
|
type CompletionRequest struct {
|
|
Model string `json:"model"`
|
|
Prompt string `json:"prompt"`
|
|
MaxToken *int `json:"max_tokens,omitempty"`
|
|
Stream bool `json:"stream"`
|
|
}
|
|
|
|
// ChatResponse is the native http request body for chatglm
|
|
type ChatResponse struct {
|
|
ID string `json:"id"`
|
|
Object string `json:"object"`
|
|
Created int64 `json:"created"`
|
|
Model string `json:"model"`
|
|
Choices []struct {
|
|
Index int `json:"index"`
|
|
Message globals.Message `json:"message"`
|
|
FinishReason string `json:"finish_reason"`
|
|
} `json:"choices"`
|
|
Error struct {
|
|
Message string `json:"message"`
|
|
} `json:"error"`
|
|
}
|
|
|
|
// ChatStreamResponse is the stream response body for chatglm
|
|
type ChatStreamResponse struct {
|
|
ID string `json:"id"`
|
|
Object string `json:"object"`
|
|
Created int64 `json:"created"`
|
|
Model string `json:"model"`
|
|
Choices []struct {
|
|
Delta globals.Message `json:"delta"`
|
|
Index int `json:"index"`
|
|
FinishReason string `json:"finish_reason"`
|
|
} `json:"choices"`
|
|
}
|
|
|
|
// CompletionResponse is the native http request body / stream response body for chatglm completion
|
|
type CompletionResponse struct {
|
|
ID string `json:"id"`
|
|
Object string `json:"object"`
|
|
Created int64 `json:"created"`
|
|
Model string `json:"model"`
|
|
Choices []struct {
|
|
Text string `json:"text"`
|
|
Index int `json:"index"`
|
|
} `json:"choices"`
|
|
}
|
|
|
|
type ChatStreamErrorResponse struct {
|
|
Error struct {
|
|
Message string `json:"message"`
|
|
Type string `json:"type"`
|
|
Code string `json:"code"`
|
|
} `json:"error"`
|
|
}
|
|
|
|
type ImageSize string
|
|
|
|
// ImageRequest is the request body for chatglm dalle image generation
|
|
type ImageRequest struct {
|
|
Model string `json:"model"`
|
|
Prompt string `json:"prompt"`
|
|
Size ImageSize `json:"size"`
|
|
N int `json:"n"`
|
|
}
|
|
|
|
type ImageResponse struct {
|
|
Data []struct {
|
|
Url string `json:"url"`
|
|
} `json:"data"`
|
|
Error struct {
|
|
Message string `json:"message"`
|
|
} `json:"error"`
|
|
}
|
|
|
|
var (
|
|
ImageSize256 ImageSize = "256x256"
|
|
ImageSize512 ImageSize = "512x512"
|
|
ImageSize1024 ImageSize = "1024x1024"
|
|
)
|