31 lines
659 B
Go
31 lines
659 B
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
|
||
|
|
"github.com/yankeguo/zhipu"
|
||
|
|
|
||
|
|
)
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
// 或者手动指定密钥
|
||
|
|
client, err := zhipu.NewClient(zhipu.WithAPIKey("0cf510ebc01599dba2a593069c1bdfbc.nQBQ4skP8xBh7ijU"))
|
||
|
|
|
||
|
|
service := client.ChatCompletion("glm-4-flash").
|
||
|
|
AddMessage(zhipu.ChatCompletionMessage{
|
||
|
|
Role: "user",
|
||
|
|
Content: "你好",
|
||
|
|
}).SetStreamHandler(func(chunk zhipu.ChatCompletionResponse) error {
|
||
|
|
println(chunk.Choices[0].Delta.Content)
|
||
|
|
return nil
|
||
|
|
})
|
||
|
|
|
||
|
|
res, err := service.Do(context.Background())
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
zhipu.GetAPIErrorCode(err) // get the API error code
|
||
|
|
} else {
|
||
|
|
println(res.Choices[0].Message.Content)
|
||
|
|
}
|
||
|
|
}
|