feat: 植物识别百科ai助手迁移

app/plant/rpc/internal/config/config.go

@@ -7,14 +7,4 @@ type Config struct {
 	DB struct {
 		DataSource string
 	}
-	Ai struct {
-		EmbeddingApiUrl    string
-		EmbeddingApiKey    string
-		EmbeddingModelName string
-		QdrantUrl          string
-		QdrantApiKey       string
-		QdrantCollection   string
-		VectorDimension    int
-		DailyQuota         int64
-	}
 }

app/plant/rpc/internal/logic/addCareRecordLogic.go

@@ -57,19 +57,29 @@ func (l *AddCareRecordLogic) AddCareRecord(in *plant.AddCareRecordReq) (*plant.C
 
 		// 4. 生成下一期任务（以今天为基准，+period 天）
 		nextDue := time.Now().Truncate(24*time.Hour).AddDate(0, 0, plan.Period)
-		nextTask := plantModel.CareTask{
-			UserID:       in.UserId,
-			PlantID:      in.PlantId,
-			PlanID:       in.PlanId,
-			Name:         plan.Name,
-			Icon:         plan.Icon,
-			TargetAction: plan.TargetAction,
-			DueDate:      nextDue,
-			Status:       1,
-		}
-		if err := tx.Create(&nextTask).Error; err != nil {
+
+		// 检查是否已经存在该计划的待办任务，避免重复生成相同事项的多个待办任务
+		var activeCount int64
+		if err := tx.Model(&plantModel.CareTask{}).
+			Where("plan_id = ? AND status = 1", plan.ID).
+			Count(&activeCount).Error; err != nil {
 			return err
 		}
+		if activeCount == 0 {
+			nextTask := plantModel.CareTask{
+				UserID:       in.UserId,
+				PlantID:      in.PlantId,
+				PlanID:       in.PlanId,
+				Name:         plan.Name,
+				Icon:         plan.Icon,
+				TargetAction: plan.TargetAction,
+				DueDate:      nextDue,
+				Status:       1,
+			}
+			if err := tx.Create(&nextTask).Error; err != nil {
+				return err
+			}
+		}
 
 		// 5. 更新用户 care_count 统计
 		actionMap := map[string]string{

app/plant/rpc/internal/logic/completeTaskLogic.go

@@ -51,14 +51,24 @@ func (l *CompleteTaskLogic) CompleteTask(in *plant.CompleteTaskReq) (*plant.Task
 		today := time.Now()
 		todayZero := time.Date(today.Year(), today.Month(), today.Day(), 0, 0, 0, 0, today.Location())
 		nextDue := todayZero.AddDate(0, 0, plan.Period)
-		newTask := plantModel.CareTask{
-			UserID: plan.UserID, PlantID: plan.PlantID, PlanID: plan.ID,
-			Name: plan.Name, Icon: plan.Icon, TargetAction: plan.TargetAction,
-			DueDate: nextDue, Status: 1,
-		}
-		if err := tx.Create(&newTask).Error; err != nil {
+
+		// 检查是否已经存在该计划的待办任务，避免重复生成相同事项的多个待办任务
+		var activeCount int64
+		if err := tx.Model(&plantModel.CareTask{}).
+			Where("plan_id = ? AND status = 1", plan.ID).
+			Count(&activeCount).Error; err != nil {
 			return err
 		}
+		if activeCount == 0 {
+			newTask := plantModel.CareTask{
+				UserID: plan.UserID, PlantID: plan.PlantID, PlanID: plan.ID,
+				Name: plan.Name, Icon: plan.Icon, TargetAction: plan.TargetAction,
+				DueDate: nextDue, Status: 1,
+			}
+			if err := tx.Create(&newTask).Error; err != nil {
+				return err
+			}
+		}
 		// 4. 保存养护记录
 		record := plantModel.CareRecord{
 			UserID: plan.UserID, PlantID: plan.PlantID, PlanID: plan.ID,

app/plant/rpc/internal/logic/deleteClassifyLogLogic.go

@@ -19,7 +19,7 @@ func NewDeleteClassifyLogLogic(ctx context.Context, svcCtx *svc.ServiceContext)
 }
 
 func (l *DeleteClassifyLogLogic) DeleteClassifyLog(in *plant.IdsReq) (*plant.CommonResp, error) {
-	if err := l.svcCtx.DB.Where("id IN ?", in.Ids).Delete(&plantModel.OcrLog{}).Error; err != nil {
+	if err := l.svcCtx.DB.Where("id IN ?", in.Ids).Delete(&plantModel.ClassifyRecord{}).Error; err != nil {
 		return nil, err
 	}
 	return &plant.CommonResp{Code: 0, Msg: "ok"}, nil

app/plant/rpc/internal/logic/deleteWikiVectorLogic.go

@@ -22,10 +22,14 @@ func NewDeleteWikiVectorLogic(ctx context.Context, svcCtx *svc.ServiceContext) *
 }
 
 func (l *DeleteWikiVectorLogic) DeleteWikiVector(in *plant.SyncWikiVectorReq) (*plant.CommonResp, error) {
-	if l.svcCtx.Config.Ai.QdrantUrl == "" || l.svcCtx.Config.Ai.QdrantCollection == "" {
+	dbCfg, err := getActiveAiConfig(l.svcCtx.DB)
+	if err != nil {
+		return nil, err
+	}
+	if dbCfg.QdrantUrl == "" || dbCfg.QdrantCollection == "" {
 		return nil, errors.New("AI/RAG 未配置 QdrantUrl 或 QdrantCollection")
 	}
-	if err := deleteWikiVector(l.ctx, l.svcCtx.Config, in.WikiId); err != nil {
+	if err := deleteWikiVector(l.ctx, dbCfg, in.WikiId); err != nil {
 		return nil, err
 	}
 	if err := l.svcCtx.DB.Model(&plantModel.Wiki{}).Where("id = ?", in.WikiId).Update("is_vector_synced", false).Error; err != nil {

app/plant/rpc/internal/logic/getAiChatQuotaLogic.go

@@ -26,10 +26,12 @@ func (l *GetAiChatQuotaLogic) GetAiChatQuota(in *plant.GetProfileReq) (*plant.Ai
 	l.svcCtx.DB.Model(&plantModel.AiChatHistory{}).
 		Where("user_id = ? AND created_at >= ?", in.UserId, todayStart).
 		Count(&used)
-	limit := l.svcCtx.Config.Ai.DailyQuota
-	if limit <= 0 {
-		limit = 20
+
+	limit := int64(20)
+	if dbCfg, err := getActiveAiConfig(l.svcCtx.DB); err == nil && dbCfg.DailyQueryLimit > 0 {
+		limit = int64(dbCfg.DailyQueryLimit)
 	}
+
 	remaining := limit - used
 	if remaining < 0 {
 		remaining = 0

app/plant/rpc/internal/logic/getMyClassifyLogLogic.go

@@ -2,6 +2,7 @@ package logic
 
 import (
 	"context"
+	"encoding/json"
 	"github.com/zeromicro/go-zero/core/logx"
 	plantModel "sundynix-micro-go/app/plant/model"
 	"sundynix-micro-go/app/plant/rpc/internal/svc"
@@ -19,14 +20,19 @@ func NewGetMyClassifyLogLogic(ctx context.Context, svcCtx *svc.ServiceContext) *
 }
 
 func (l *GetMyClassifyLogLogic) GetMyClassifyLog(in *plant.GetProfileReq) (*plant.ClassifyLogListResp, error) {
-	var logs []plantModel.OcrLog
-	if err := l.svcCtx.DB.Where("user_id = ?", in.UserId).Order("created_at desc").Limit(50).Find(&logs).Error; err != nil {
+	var records []plantModel.ClassifyRecord
+	if err := l.svcCtx.DB.Where("user_id = ?", in.UserId).Order("created_at desc").Limit(50).Find(&records).Error; err != nil {
 		return nil, err
 	}
-	list := make([]*plant.ClassifyLogInfo, 0, len(logs))
-	for _, og := range logs {
+	list := make([]*plant.ClassifyLogInfo, 0, len(records))
+	for _, og := range records {
+		var imgUrl string
+		if len(og.AllResults) > 0 && og.AllResults[0].BaikeInfo != nil {
+			imgUrl = og.AllResults[0].BaikeInfo.ImageUrl
+		}
+		resultBytes, _ := json.Marshal(og.AllResults)
 		list = append(list, &plant.ClassifyLogInfo{
-			Id: og.ID, UserId: og.UserID, ImageUrl: og.ImageUrl, Result: og.Result,
+			Id: og.ID, UserId: og.UserID, ImageUrl: imgUrl, Result: string(resultBytes),
 			CreatedAt: og.CreatedAt.Format("2006-01-02 15:04:05"),
 		})
 	}

app/plant/rpc/internal/logic/qdrantVector.go

@@ -1,24 +1,24 @@
 package logic
 
 import (
-	"bytes"
 	"context"
-	"crypto/md5"
-	"encoding/hex"
-	"encoding/json"
 	"errors"
 	"fmt"
-	"io"
-	"net/http"
 	"strings"
 
+	"github.com/google/uuid"
+	qdrant "github.com/qdrant/go-client/qdrant"
+	openai "github.com/sashabaranov/go-openai"
+	"google.golang.org/grpc"
+	"google.golang.org/grpc/credentials/insecure"
+	"google.golang.org/grpc/metadata"
+	"gorm.io/gorm"
+
 	plantModel "sundynix-micro-go/app/plant/model"
-	"sundynix-micro-go/app/plant/rpc/internal/config"
 )
 
 func wikiVectorID(wikiID string) string {
-	sum := md5.Sum([]byte("sundynix-plant-wiki:" + wikiID))
-	return hex.EncodeToString(sum[:])
+	return uuid.NewMD5(uuid.NameSpaceOID, []byte(wikiID)).String()
 }
 
 func buildWikiVectorText(w plantModel.Wiki) string {
@@ -30,131 +30,129 @@ func buildWikiVectorText(w plantModel.Wiki) string {
 		w.FloweringShape, w.FlowerDiameter, w.Fruit)
 }
 
-func embeddingModel(c config.Config) string {
-	if c.Ai.EmbeddingModelName != "" {
-		return c.Ai.EmbeddingModelName
+func getActiveAiConfig(db *gorm.DB) (*plantModel.SysAiConfig, error) {
+	var cfg plantModel.SysAiConfig
+	if err := db.Where("is_active = 1").First(&cfg).Error; err != nil {
+		return nil, errors.New("数据库未找到已激活的 AI 配置")
+	}
+	return &cfg, nil
+}
+
+func embeddingModel(cfg *plantModel.SysAiConfig) string {
+	if cfg.EmbeddingModelName != "" {
+		return cfg.EmbeddingModelName
 	}
 	return "text-embedding-3-small"
 }
 
-func createEmbedding(ctx context.Context, c config.Config, text string) ([]float32, error) {
-	body, _ := json.Marshal(map[string]interface{}{
-		"model": embeddingModel(c),
-		"input": text,
+func createEmbedding(ctx context.Context, cfg *plantModel.SysAiConfig, text string) ([]float32, error) {
+	config := openai.DefaultConfig(cfg.EmbeddingApiKey)
+	if cfg.EmbeddingApiUrl != "" {
+		config.BaseURL = cfg.EmbeddingApiUrl
+	}
+	client := openai.NewClientWithConfig(config)
+	resp, err := client.CreateEmbeddings(ctx, openai.EmbeddingRequest{
+		Input: []string{text},
+		Model: openai.EmbeddingModel(embeddingModel(cfg)),
 	})
-	req, err := http.NewRequestWithContext(ctx, http.MethodPost, c.Ai.EmbeddingApiUrl, bytes.NewReader(body))
 	if err != nil {
 		return nil, err
 	}
-	req.Header.Set("Content-Type", "application/json")
-	req.Header.Set("Authorization", "Bearer "+c.Ai.EmbeddingApiKey)
-	resp, err := http.DefaultClient.Do(req)
-	if err != nil {
-		return nil, err
-	}
-	defer resp.Body.Close()
-	if resp.StatusCode < 200 || resp.StatusCode >= 300 {
-		raw, _ := io.ReadAll(io.LimitReader(resp.Body, 4096))
-		return nil, fmt.Errorf("embedding 请求失败: %s %s", resp.Status, strings.TrimSpace(string(raw)))
-	}
-	var parsed struct {
-		Data []struct {
-			Embedding []float32 `json:"embedding"`
-		} `json:"data"`
-	}
-	if err := json.NewDecoder(resp.Body).Decode(&parsed); err != nil {
-		return nil, err
-	}
-	if len(parsed.Data) == 0 || len(parsed.Data[0].Embedding) == 0 {
+	if len(resp.Data) == 0 || len(resp.Data[0].Embedding) == 0 {
 		return nil, errors.New("embedding 响应为空")
 	}
-	return parsed.Data[0].Embedding, nil
+	return resp.Data[0].Embedding, nil
 }
 
-func qdrantURL(c config.Config, path string) string {
-	return strings.TrimRight(c.Ai.QdrantUrl, "/") + path
-}
-
-func doQdrant(ctx context.Context, c config.Config, method, path string, body interface{}) error {
-	var reader io.Reader
-	if body != nil {
-		raw, _ := json.Marshal(body)
-		reader = bytes.NewReader(raw)
+func newQdrantConn(cfg *plantModel.SysAiConfig) (*grpc.ClientConn, context.Context, error) {
+	addr := strings.TrimPrefix(cfg.QdrantUrl, "http://")
+	addr = strings.TrimPrefix(addr, "https://")
+	conn, err := grpc.NewClient(addr, grpc.WithTransportCredentials(insecure.NewCredentials()))
+	if err != nil {
+		return nil, nil, fmt.Errorf("qdrant grpc dial failed: %w", err)
 	}
-	req, err := http.NewRequestWithContext(ctx, method, qdrantURL(c, path), reader)
+	ctx := context.Background()
+	if cfg.QdrantApiKey != "" {
+		ctx = metadata.NewOutgoingContext(ctx, metadata.Pairs("api-key", cfg.QdrantApiKey))
+	}
+	return conn, ctx, nil
+}
+
+func ensureQdrantCollection(cfg *plantModel.SysAiConfig, dim int) error {
+	conn, ctx, err := newQdrantConn(cfg)
 	if err != nil {
 		return err
 	}
-	req.Header.Set("Content-Type", "application/json")
-	if c.Ai.QdrantApiKey != "" {
-		req.Header.Set("api-key", c.Ai.QdrantApiKey)
-	}
-	resp, err := http.DefaultClient.Do(req)
-	if err != nil {
-		return err
-	}
-	defer resp.Body.Close()
-	if resp.StatusCode < 200 || resp.StatusCode >= 300 {
-		raw, _ := io.ReadAll(io.LimitReader(resp.Body, 4096))
-		return fmt.Errorf("qdrant 请求失败: %s %s", resp.Status, strings.TrimSpace(string(raw)))
-	}
-	return nil
-}
-
-func ensureQdrantCollection(ctx context.Context, c config.Config, dim int) error {
-	getReq, err := http.NewRequestWithContext(ctx, http.MethodGet, qdrantURL(c, "/collections/"+c.Ai.QdrantCollection), nil)
-	if err != nil {
-		return err
-	}
-	if c.Ai.QdrantApiKey != "" {
-		getReq.Header.Set("api-key", c.Ai.QdrantApiKey)
-	}
-	if resp, err := http.DefaultClient.Do(getReq); err == nil {
-		_ = resp.Body.Close()
-		if resp.StatusCode >= 200 && resp.StatusCode < 300 {
-			return nil
-		}
-	}
+	defer conn.Close()
 	if dim <= 0 {
-		dim = c.Ai.VectorDimension
+		dim = cfg.VectorDimension
 	}
 	if dim <= 0 {
 		dim = 1536
 	}
-	return doQdrant(ctx, c, http.MethodPut, "/collections/"+c.Ai.QdrantCollection, map[string]interface{}{
-		"vectors": map[string]interface{}{
-			"size":     dim,
-			"distance": "Cosine",
+	collClient := qdrant.NewCollectionsClient(conn)
+	if _, getErr := collClient.Get(ctx, &qdrant.GetCollectionInfoRequest{CollectionName: cfg.QdrantCollection}); getErr == nil {
+		return nil
+	}
+	_, err = collClient.Create(ctx, &qdrant.CreateCollection{
+		CollectionName: cfg.QdrantCollection,
+		VectorsConfig: &qdrant.VectorsConfig{
+			Config: &qdrant.VectorsConfig_Params{
+				Params: &qdrant.VectorParams{Size: uint64(dim), Distance: qdrant.Distance_Cosine},
+			},
 		},
 	})
+	return err
 }
 
-func upsertWikiVector(ctx context.Context, c config.Config, w plantModel.Wiki) error {
+func upsertWikiVector(ctx context.Context, cfg *plantModel.SysAiConfig, w plantModel.Wiki) error {
 	text := buildWikiVectorText(w)
-	vector, err := createEmbedding(ctx, c, text)
+	vector, err := createEmbedding(ctx, cfg, text)
 	if err != nil {
 		return err
 	}
-	if err := ensureQdrantCollection(ctx, c, len(vector)); err != nil {
+	if err := ensureQdrantCollection(cfg, len(vector)); err != nil {
 		return err
 	}
-	return doQdrant(ctx, c, http.MethodPut, "/collections/"+c.Ai.QdrantCollection+"/points?wait=true", map[string]interface{}{
-		"points": []map[string]interface{}{
-			{
-				"id":     wikiVectorID(w.ID),
-				"vector": vector,
-				"payload": map[string]interface{}{
-					"wiki_id":   w.ID,
-					"name":      w.Name,
-					"full_text": text,
+	conn, qdCtx, err := newQdrantConn(cfg)
+	if err != nil {
+		return err
+	}
+	defer conn.Close()
+	ptsClient := qdrant.NewPointsClient(conn)
+
+	_, err = ptsClient.Upsert(qdCtx, &qdrant.UpsertPoints{
+		CollectionName: cfg.QdrantCollection,
+		Points: []*qdrant.PointStruct{{
+			Id:      qdrant.NewID(wikiVectorID(w.ID)),
+			Vectors: qdrant.NewVectors(vector...),
+			Payload: map[string]*qdrant.Value{
+				"wiki_id":   qdrant.NewValueString(w.ID),
+				"name":      qdrant.NewValueString(w.Name),
+				"full_text": qdrant.NewValueString(text),
+			},
+		}},
+	})
+	return err
+}
+
+func deleteWikiVector(ctx context.Context, cfg *plantModel.SysAiConfig, wikiID string) error {
+	conn, qdCtx, err := newQdrantConn(cfg)
+	if err != nil {
+		return err
+	}
+	defer conn.Close()
+	ptsClient := qdrant.NewPointsClient(conn)
+
+	_, err = ptsClient.Delete(qdCtx, &qdrant.DeletePoints{
+		CollectionName: cfg.QdrantCollection,
+		Points: &qdrant.PointsSelector{
+			PointsSelectorOneOf: &qdrant.PointsSelector_Points{
+				Points: &qdrant.PointsIdsList{
+					Ids: []*qdrant.PointId{qdrant.NewID(wikiVectorID(wikiID))},
 				},
 			},
 		},
 	})
-}
-
-func deleteWikiVector(ctx context.Context, c config.Config, wikiID string) error {
-	return doQdrant(ctx, c, http.MethodPost, "/collections/"+c.Ai.QdrantCollection+"/points/delete?wait=true", map[string]interface{}{
-		"points": []string{wikiVectorID(wikiID)},
-	})
+	return err
 }

app/plant/rpc/internal/logic/syncAllWikiVectorLogic.go

@@ -23,7 +23,11 @@ func NewSyncAllWikiVectorLogic(ctx context.Context, svcCtx *svc.ServiceContext)
 }
 
 func (l *SyncAllWikiVectorLogic) SyncAllWikiVector(in *plant.PageReq) (*plant.CommonResp, error) {
-	if l.svcCtx.Config.Ai.EmbeddingApiUrl == "" || l.svcCtx.Config.Ai.QdrantUrl == "" || l.svcCtx.Config.Ai.QdrantCollection == "" {
+	dbCfg, err := getActiveAiConfig(l.svcCtx.DB)
+	if err != nil {
+		return nil, err
+	}
+	if dbCfg.EmbeddingApiUrl == "" || dbCfg.QdrantUrl == "" || dbCfg.QdrantCollection == "" {
 		return nil, errors.New("AI/RAG 未配置 EmbeddingApiUrl、QdrantUrl 或 QdrantCollection")
 	}
 	var wikis []plantModel.Wiki
@@ -32,7 +36,7 @@ func (l *SyncAllWikiVectorLogic) SyncAllWikiVector(in *plant.PageReq) (*plant.Co
 	}
 	success := 0
 	for _, wiki := range wikis {
-		if err := upsertWikiVector(l.ctx, l.svcCtx.Config, wiki); err != nil {
+		if err := upsertWikiVector(l.ctx, dbCfg, wiki); err != nil {
 			l.Logger.Errorf("sync wiki vector failed, wiki_id=%s, err=%v", wiki.ID, err)
 			continue
 		}

app/plant/rpc/internal/logic/syncWikiVectorLogic.go

@@ -25,14 +25,18 @@ func (l *SyncWikiVectorLogic) SyncWikiVector(in *plant.SyncWikiVectorReq) (*plan
 	if in.WikiId == "" {
 		return nil, errors.New("wikiId 不能为空")
 	}
-	if l.svcCtx.Config.Ai.EmbeddingApiUrl == "" || l.svcCtx.Config.Ai.QdrantUrl == "" || l.svcCtx.Config.Ai.QdrantCollection == "" {
+	dbCfg, err := getActiveAiConfig(l.svcCtx.DB)
+	if err != nil {
+		return nil, err
+	}
+	if dbCfg.EmbeddingApiUrl == "" || dbCfg.QdrantUrl == "" || dbCfg.QdrantCollection == "" {
 		return nil, errors.New("AI/RAG 未配置 EmbeddingApiUrl、QdrantUrl 或 QdrantCollection")
 	}
 	var wiki plantModel.Wiki
 	if err := l.svcCtx.DB.Where("id = ?", in.WikiId).First(&wiki).Error; err != nil {
 		return nil, err
 	}
-	if err := upsertWikiVector(l.ctx, l.svcCtx.Config, wiki); err != nil {
+	if err := upsertWikiVector(l.ctx, dbCfg, wiki); err != nil {
 		return nil, err
 	}
 	if err := l.svcCtx.DB.Model(&plantModel.Wiki{}).Where("id = ?", in.WikiId).Update("is_vector_synced", true).Error; err != nil {

app/plant/rpc/internal/svc/serviceContext.go

@@ -40,7 +40,6 @@ func NewServiceContext(c config.Config) *ServiceContext {
 		&plantModel.PostLike{},
 		&plantModel.PostOss{},
 		&plantModel.Topic{},
-		&plantModel.OcrLog{},
 		&plantModel.MediaCheckResult{},
 		&plantModel.ExchangeItem{},
 		&plantModel.ExchangeOrder{},
@@ -50,6 +49,8 @@ func NewServiceContext(c config.Config) *ServiceContext {
 		&plantModel.AiChatHistory{},
 		&plantModel.GrowthRecordOss{},
 		&plantModel.Banner{},
+		&plantModel.SysAiConfig{},
+		&plantModel.ClassifyRecord{},
 	); err != nil {
 		logx.Errorf("数据库迁移失败: %v", err)
 	}