self-tools
/
l_ai_category
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

1

This commit is contained in:
renzhiyuan 2025-03-27 17:29:34 +08:00
parent 39f4871a42
commit b5bb2fad66
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ai.go
View File

@ -2,6 +2,7 @@ package l_ai_category
import ( import (
"context" "context"
"fmt"
"gitea.cdlsxd.cn/self-tools/l_ai_category/doubao" "gitea.cdlsxd.cn/self-tools/l_ai_category/doubao"
"strconv" "strconv"
) )
@ -11,7 +12,7 @@ func GetCategory(ctx context.Context, goodsName, key, model string, cateJson str
modelObj := doubao.NewDouBao(model, key) modelObj := doubao.NewDouBao(model, key)
text := []string{ text := []string{
"根据商品名称,从json中找到该商品对应的第三级分类[QUESTION]" + goodsName + "[/QUESTION]", "根据商品名称,从json中找到该商品对应的第三级分类[QUESTION]" + goodsName + "[/QUESTION]",
"-只需要返回类型的名称对应的id", "-只需要返回类型的名称对应的id,不用返回其他任何文字",
"-如果无法匹配，则返回数字0", "-如果无法匹配，则返回数字0",
"-以下是 JSON 数据：" + cateJson, "-以下是 JSON 数据：" + cateJson,
} }
@ -23,6 +24,7 @@ func GetCategory(ctx context.Context, goodsName, key, model string, cateJson str
} }
categoryId, err = strconv.Atoi(category) categoryId, err = strconv.Atoi(category)
if err != nil { if err != nil {
err = fmt.Errorf("未找到商品分类")
categoryId = 0 categoryId = 0
} }
return return

4
ai_test.go
View File

@ -9,14 +9,14 @@ import (
) )
const ( const (
dns = "root:lansexiongdi6,@tcp(47.97.27.195:3306)/report?parseTime=True&loc=Local" dns = ""
driver = "mysql" driver = "mysql"
table = "goods_category" table = "goods_category"
) )
func TestCategory(t *testing.T) { func TestCategory(t *testing.T) {
catString := string(CategoryToJsonByte()) catString := string(CategoryToJsonByte())
res, err := GetCategory(context.Background(), "金龙鱼御品珍珠米", "03320e58-6a0b-4061-a22b-902039f2190d", "deepseek-v3-250324", catString) res, err := GetCategory(context.Background(), "思嘉思达 雅格电火锅 3.5L电火锅 SKD-G0020", "03320e58-6a0b-4061-a22b-902039f2190d", "deepseek-v3-250324", catString)
t.Log(res, err) t.Log(res, err)
} }