package l_ai_category

import (
	"context"
	"gitea.cdlsxd.cn/self-tools/l_ai_category/doubao"
	"strconv"
)

func GetCategory(ctx context.Context, goodsName, key, model string, cateJson string) (categoryId int, err error) {

	modelObj := doubao.NewDouBao(model, key)
	text := []string{
		"根据商品名称,从json中找到该商品对应的第三级分类[QUESTION]" + goodsName + "[/QUESTION]",
		"-只需要返回类型的名称对应的id",
		"-如果无法匹配，则返回数字0",
		"-以下是 JSON 数据：" + cateJson,
	}
	category, err := modelObj.GetData(ctx, key, func(input string) (string, error) {
		return input, nil
	}, text...)
	if err != nil {
		return
	}
	categoryId, err = strconv.Atoi(category)
	if err != nil {
		categoryId = 0
	}
	return
}