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