27 lines
556 B
Go
27 lines
556 B
Go
|
package lib
|
||
|
|
|
||
|
|
import (
|
||
|
|
"log"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestChatGLM_ChatStream(t *testing.T) {
|
||
|
|
chatglm, err := NewChatGLM("0ca00a2167291d1ae977c2b88ffdab57.JHEkjkBxyjkFiJ43")
|
||
|
|
if err != nil {
|
||
|
|
log.Println(err)
|
||
|
|
}
|
||
|
|
//res, err := chatglm.ChatGLM6B()
|
||
|
|
//log.Println(res, err)
|
||
|
|
prompt := []map[string]string{{"role": "user", "content": "你是谁？"}}
|
||
|
|
res, err := chatglm.StreamChatGLM6B(prompt, 0.79, 0.7, false)
|
||
|
|
for {
|
||
|
|
str, err := chatglm.StreamRecv(res)
|
||
|
|
if err != nil {
|
||
|
|
log.Println(err)
|
||
|
|
break
|
||
|
|
}
|
||
|
|
log.Println(str, err)
|
||
|
|
}
|
||
|
|
//log.Println(res, err)
|
||
|
|
}
|