46 lines
1.7 KiB
Go
46 lines
1.7 KiB
Go
package lib
|
||
|
||
import (
|
||
"log"
|
||
"testing"
|
||
)
|
||
|
||
//测试获取access_token
|
||
func TestErnieBotTurboGenerateAccessToken(t *testing.T) {
|
||
AppID := "35662533"
|
||
APIKey := "Iq1FfkOQIGtMtZqRFxOrvq6T"
|
||
SecretKey := "qbzsoFAUSl8UGt1GkGSDSjENtqsjrOTC"
|
||
m, err := NewErnieBotTurbo(AppID, APIKey, SecretKey)
|
||
log.Println(m.AccessToken, err)
|
||
}
|
||
|
||
//测试流式输出
|
||
func TestErnieBotTurboStreamChat(t *testing.T) {
|
||
AppID := "35662533"
|
||
APIKey := "Iq1FfkOQIGtMtZqRFxOrvq6T"
|
||
SecretKey := "qbzsoFAUSl8UGt1GkGSDSjENtqsjrOTC"
|
||
m, _ := NewErnieBotTurbo(AppID, APIKey, SecretKey)
|
||
system := "假设你是一个程序员，你是程序员老狼的数字分身，你擅长计算机互联网程序开发，你的微信：llike620，官网：gofly.v1kf.com。你开发了唯一客服系统，你了解唯一客服系统的各种问题，你的回答应该是简洁的、不要超过50字。你只能根据知识信息回答问题。你绝对不能暴露自己是个“人工智能、大模型、AI、机器人、GPT”。如果用户问题里有自己联系方式，请回复“稍等，我们马上联系您”。请检测问题语种，并用该语种回答问题."
|
||
prompt := []map[string]string{{"role": "user", "content": system + "我的问题是：你是谁"}}
|
||
res, _ := m.StreamChat(prompt)
|
||
for {
|
||
str, err := m.StreamRecv(res)
|
||
if err != nil {
|
||
log.Println(err)
|
||
break
|
||
}
|
||
log.Println(str, err)
|
||
}
|
||
}
|
||
|
||
//测试Embedding
|
||
func TestErnieBotTurboEmbedding(t *testing.T) {
|
||
AppID := "35662533"
|
||
APIKey := "Iq1FfkOQIGtMtZqRFxOrvq6T"
|
||
SecretKey := "qbzsoFAUSl8UGt1GkGSDSjENtqsjrOTC"
|
||
m, _ := NewErnieBotTurbo(AppID, APIKey, SecretKey)
|
||
prompt := []string{"我的问题是：你是谁"}
|
||
res, err := m.Embedding(prompt)
|
||
log.Println(res, err)
|
||
}
|