52 lines
1.3 KiB
Go
52 lines
1.3 KiB
Go
package test
|
||
|
||
import (
|
||
"catface/app/global/variable"
|
||
_ "catface/bootstrap"
|
||
"context"
|
||
"testing"
|
||
|
||
"github.com/yankeguo/zhipu"
|
||
)
|
||
|
||
func TestGlmMessageStore(t *testing.T) {
|
||
glmClient, err := zhipu.NewClient(zhipu.WithAPIKey(variable.ConfigYml.GetString("Glm.ApiKey")))
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
|
||
service := glmClient.ChatCompletion("glm-4-flash").AddMessage(zhipu.ChatCompletionMessage{
|
||
Role: "user",
|
||
Content: "请你记一下我说的数字：2",
|
||
})
|
||
|
||
res, err := service.Do(context.Background())
|
||
if err != nil {
|
||
apiErrorCode := zhipu.GetAPIErrorCode(err)
|
||
t.Fatal(apiErrorCode)
|
||
}
|
||
t.Log(res.Choices[0].Message.Content)
|
||
|
||
messages := service.GetMessages()
|
||
for _, message := range messages {
|
||
t.Log(message.(zhipu.ChatCompletionMessage).Role, message.(zhipu.ChatCompletionMessage).Content)
|
||
}
|
||
|
||
service.AddMessage(zhipu.ChatCompletionMessage{
|
||
Role: "user",
|
||
Content: "现在请你复述我刚才说的数字。",
|
||
})
|
||
res, err = service.Do(context.Background())
|
||
if err != nil {
|
||
apiErrorCode := zhipu.GetAPIErrorCode(err)
|
||
t.Fatal(apiErrorCode)
|
||
}
|
||
|
||
messages = service.GetMessages()
|
||
for _, message := range messages {
|
||
t.Log(message.(zhipu.ChatCompletionMessage).Role, message.(zhipu.ChatCompletionMessage).Content)
|
||
}
|
||
|
||
t.Log(res.Choices[0].Message.Content)
|
||
}
|