49 lines
1.2 KiB
Go
49 lines
1.2 KiB
Go
package glm
|
|
|
|
import (
|
|
"catface/app/global/variable"
|
|
"context"
|
|
"errors"
|
|
|
|
"github.com/yankeguo/zhipu"
|
|
|
|
)
|
|
|
|
// ChatWithGLM 封装了与GLM模型进行对话的逻辑
|
|
func Chat(message string) (string, error) {
|
|
service := variable.GlmClient.ChatCompletion("glm-4-flash").
|
|
AddMessage(zhipu.ChatCompletionMessage{
|
|
Role: "user",
|
|
Content: message,
|
|
})
|
|
|
|
res, err := service.Do(context.Background())
|
|
if err != nil {
|
|
apiErrorCode := zhipu.GetAPIErrorCode(err)
|
|
return "", errors.New(apiErrorCode) // 将字符串包装成 error 类型
|
|
}
|
|
|
|
return res.Choices[0].Message.Content, nil
|
|
}
|
|
|
|
// ChatStream 接收一个消息和一个通道，将流式响应发送到通道中
|
|
func ChatStream(message string, ch chan<- string) error {
|
|
service := variable.GlmClient.ChatCompletion("glm-4-flash").
|
|
AddMessage(zhipu.ChatCompletionMessage{Role: "user", Content: message}).
|
|
SetStreamHandler(func(chunk zhipu.ChatCompletionResponse) error {
|
|
content := chunk.Choices[0].Delta.Content
|
|
if content != "" {
|
|
ch <- content // 将内容发送到通道
|
|
}
|
|
return nil
|
|
})
|
|
|
|
// 执行服务调用
|
|
_, err := service.Do(context.Background())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|