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)
|
|||
|
|
}
|