package main

import (
	"fmt"
	"gitee.com/jn-qq/big-model/common"
	"gitee.com/jn-qq/big-model/openai"
)

func main() {
	client := openai.NewOpenAiClient(
		//common.Model{
		//	Url:       "https://aicloud.oneainexus.cn:30013/llminference/deepseekv3/v1/chat/completions",
		//	ApiKey:    "sk-8zRMtnd7Z4VraVvZFa478cD7B215499c8959Ce2e0e582730",
		//	ModelType: "deepseekv3",
		//	IsReason:  false,
		//},
		common.Model{
			Url:       "https://aicloud.oneainexus.cn:30013/llminference/deepseekr1-qwen32b/v1/chat/completions",
			ApiKey:    "sk-8zRMtnd7Z4VraVvZFa478cD7B215499c8959Ce2e0e582730",
			ModelType: "deepseekr1-qwen32b",
			IsReason:  false,
		},
	).SetStream(true)

	response, err := client.Chat("你好，你叫什么名字？")
	if err != nil {
		panic(err)
	}
	fmt.Println("答案：", response.GetAnswer())
	fmt.Println("思考：", response.GetThink())
}