1

ai.go

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

ai_test.go

@@ -9,14 +9,14 @@ import (
 )
 
 const (
-	dns    = "root:lansexiongdi6,@tcp(47.97.27.195:3306)/report?parseTime=True&loc=Local"
+	dns    = ""
 	driver = "mysql"
 	table  = "goods_category"
 )
 
 func TestCategory(t *testing.T) {
 	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)
 }