From 3296b4721171efdbdcdd0e00ef38e0817585ff98 Mon Sep 17 00:00:00 2001
From: renzhiyuan <465386466@qq.com>
Date: Thu, 27 Mar 2025 23:14:15 +0800
Subject: [PATCH] 1

---
 ai.go      | 2 +-
 ai_test.go | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/ai.go b/ai.go
index 6df5bab..084af66 100644
--- a/ai.go
+++ b/ai.go
@@ -25,7 +25,7 @@ type QuesStruct struct {
 }
 
 func GetCategory(ctx context.Context, goodsName, key, chatModel string, catePath []map[int][]*CategoryDic) (cate3 int, err error) {
-	cate1Json, _ := json.Marshal(catePath[0][0])
+	cate1Json, _ := json.Marshal(catePath[0][1])
 	cate1, err := ques(ctx, goodsName, key, chatModel, cate1Json)
 	if err != nil {
 		return
diff --git a/ai_test.go b/ai_test.go
index 6985068..01f6b33 100644
--- a/ai_test.go
+++ b/ai_test.go
@@ -16,7 +16,7 @@ const (
 
 func TestCategory(t *testing.T) {
 	path := getPath()
-	res, err := GetCategory(context.Background(), "倍轻松头部按摩器\niDream 3（尊享版", "914ccf1d-c002-4fad-a431-f291f5e0d2ad", modelType, path)
+	res, err := GetCategory(context.Background(), "MM 便携装湿厕纸 40包*10片@默认", "03320e58-6a0b-4061-a22b-902039f2190d", modelType, path)
 	t.Log(res, err)
 }