11

ai.go

@@ -2,23 +2,33 @@ package l_ai_brand
 
 import (
 	"context"
-	"gitea.cdlsxd.cn/self-tools/l_ai_brand/doubao"
+	"encoding/json"
+
+	"gitea.cdlsxd.cn/self-tools/l_request"
 	"strings"
 )
 
-func GetBrand(ctx context.Context, goodsName, key, model string, brandList []string) (brand string, err error) {
+func GetBrand(ctx context.Context, goodsName string, brandList []string, address string) (brand string, err error) {
 
-	modelObj := doubao.NewDouBao(model, key)
-	text := []string{
-		"根据商品名称，获取对应品牌[QUESTION]" + goodsName + "[/QUESTION]",
-		"-只需要获取一个品牌名称",
-		"-如果无法获取，则返回`失败`",
+	request := map[string]interface{}{
+		"goods":      goodsName,
+		"brand_list": strings.Join(brandList, ","),
 	}
-	if brandList != nil {
-		text = append(text, "-优先从以下品牌选取："+strings.Join(brandList, "、"))
+	req := l_request.Request{
+		Method: "POST",
+		Url:    address,
+		Json:   request,
 	}
-	brand, err = modelObj.GetData(ctx, key, func(input string) (string, error) {
-		return input, nil
-	}, text...)
+	res, err := req.Send()
+	if err != nil {
+		return
+	}
+	var resMap BrandRes
+	err = json.Unmarshal(res.Content, &resMap)
+	brand = resMap.ExtractedBrand
 	return
 }
+
+type BrandRes struct {
+	ExtractedBrand string `json:"extracted_brand"`
+}

ai_test.go

@@ -7,6 +7,6 @@ import (
 )
 
 func TestAddress(t *testing.T) {
-	res, err := GetBrand(context.Background(), "促销 维达（Vinda）抽纸 超韧130抽*24包S码 湿水不易破 卫生纸 纸巾餐巾纸 整箱", "236ba4b6-9daa-4755-b22f-2fd274cd223a", "doubao-1-5-lite-32k-250115", strings.Split("爱奇艺,QQ阅读,优酷,芒果,腾讯,QQ音乐,KFC,奈雪的茶,星巴克,Keep,盒马生鲜,COCO,E袋洗,NN加速器,PP视频,WPS,百度网盘,百度文库,必胜客,哔哩哔哩,得到,滴滴,饿了么,哈根达斯,哈罗出行,京东,酷狗音乐,酷我音乐,懒人听书,良品铺子,埋堆堆,麦当劳,猫眼电影,美团外卖,咪咕视频,咪咕文学,面包新语,南瓜电影,十点读书,搜狐视频,苏宁易购,淘票票,腾讯QQ,天猫,网易云音乐,沃尔玛,喜马拉雅,迅雷,知乎,中国电信,中国联通,中国石化,中国移动,,九阳,奔腾,炊大皇,西屋,小熊,维达（Vinda）", ","))
+	res, err := GetBrand(context.Background(), "促销 维达（Vinda）抽纸 超韧130抽*24包S码 湿水不易破 卫生纸 纸巾餐巾纸 整箱", strings.Split("爱奇艺,QQ阅读,优酷,芒果,腾讯,QQ音乐,KFC,奈雪的茶,星巴克,Keep,盒马生鲜,COCO,E袋洗,NN加速器,PP视频,WPS,百度网盘,百度文库,必胜客,哔哩哔哩,得到,滴滴,饿了么,哈根达斯,哈罗出行,京东,酷狗音乐,酷我音乐,懒人听书,良品铺子,埋堆堆,麦当劳,猫眼电影,美团外卖,咪咕视频,咪咕文学,面包新语,南瓜电影,十点读书,搜狐视频,苏宁易购,淘票票,腾讯QQ,天猫,网易云音乐,沃尔玛,喜马拉雅,迅雷,知乎,中国电信,中国联通,中国石化,中国移动,,九阳,奔腾,炊大皇,西屋,小熊,维达（Vinda）", ","), "http://192.168.6.81:8004/extract_brand")
 	t.Log(res, err)
 }

go.mod

@@ -2,7 +2,10 @@ module gitea.cdlsxd.cn/self-tools/l_ai_brand
 
 go 1.22.2
 
-require github.com/volcengine/volcengine-go-sdk v1.0.187
+require (
+	gitea.cdlsxd.cn/self-tools/l_request v1.0.8
+	github.com/volcengine/volcengine-go-sdk v1.0.187
+)
 
 require (
 	github.com/google/uuid v1.3.0 // indirect