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

This commit is contained in:
Blizzard
2026-05-24 01:41:22 +08:00
parent ae6d03d351
commit 076ed1509b
29 changed files with 1121 additions and 372 deletions
+1
View File
@@ -39,6 +39,7 @@ func NewProxyRouter(upstreams []config.Upstream) *ProxyRouter {
target := targetURL // 显式捕获循环变量 target := targetURL // 显式捕获循环变量
proxy := &httputil.ReverseProxy{ proxy := &httputil.ReverseProxy{
FlushInterval: -1, // 立即将数据刷新到客户端,确保 SSE 等流式接口能够实时响应
Rewrite: func(pr *httputil.ProxyRequest) { Rewrite: func(pr *httputil.ProxyRequest) {
pr.SetXForwarded() pr.SetXForwarded()
pr.Out.URL.Scheme = target.Scheme pr.Out.URL.Scheme = target.Scheme
-14
View File
@@ -34,17 +34,3 @@ DB:
BaiduOcr: BaiduOcr:
ApiKey: hpBfjwy8ifv3qswYGYjUCNKN ApiKey: hpBfjwy8ifv3qswYGYjUCNKN
SecretKey: i5aXZdM4XZVuDroBslL0f3uIuwbAyXFS SecretKey: i5aXZdM4XZVuDroBslL0f3uIuwbAyXFS
# OpenAI-compatible AI/RAG 配置。未配置 ChatApiUrl/ChatApiKey 时,AI 问答返回明确错误。
Ai:
ChatApiUrl:
ChatApiKey:
ChatModelName:
EmbeddingApiUrl:
EmbeddingApiKey:
EmbeddingModelName:
QdrantUrl:
QdrantApiKey:
QdrantCollection:
VectorDimension: 0
DailyQuota: 20
-13
View File
@@ -22,17 +22,4 @@ type Config struct {
ApiKey string ApiKey string
SecretKey string SecretKey string
} `json:",optional"` } `json:",optional"`
Ai struct {
ChatApiUrl string `json:",optional"`
ChatApiKey string `json:",optional"`
ChatModelName string `json:",optional"`
EmbeddingApiUrl string `json:",optional"`
EmbeddingApiKey string `json:",optional"`
EmbeddingModelName string `json:",optional"`
QdrantUrl string `json:",optional"`
QdrantApiKey string `json:",optional"`
QdrantCollection string `json:",optional"`
VectorDimension int `json:",optional"`
DailyQuota int64 `json:",optional"`
} `json:",optional"`
} }
@@ -2,14 +2,20 @@ package legacy
import ( import (
"context" "context"
"encoding/base64"
"encoding/json" "encoding/json"
"fmt" "fmt"
"io"
"net/http" "net/http"
"net/url"
"strings"
"time" "time"
filePb "sundynix-micro-go/app/file/rpc/file" filePb "sundynix-micro-go/app/file/rpc/file"
aiLogic "sundynix-micro-go/app/plant/api/internal/logic/ai"
"sundynix-micro-go/app/plant/api/internal/logic/complete" "sundynix-micro-go/app/plant/api/internal/logic/complete"
plantLogic "sundynix-micro-go/app/plant/api/internal/logic/myPlant" plantLogic "sundynix-micro-go/app/plant/api/internal/logic/myPlant"
ocrLogic "sundynix-micro-go/app/plant/api/internal/logic/ocr"
postLogic "sundynix-micro-go/app/plant/api/internal/logic/post" postLogic "sundynix-micro-go/app/plant/api/internal/logic/post"
topicLogic "sundynix-micro-go/app/plant/api/internal/logic/topic" topicLogic "sundynix-micro-go/app/plant/api/internal/logic/topic"
wikiLogic "sundynix-micro-go/app/plant/api/internal/logic/wiki" wikiLogic "sundynix-micro-go/app/plant/api/internal/logic/wiki"
@@ -892,3 +898,227 @@ func MediaCheckCallbackHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
response.Ok(w) response.Ok(w)
} }
} }
func AiChatStreamHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
query := r.URL.Query().Get("query")
if query == "" {
query = r.URL.Query().Get("question")
}
if query == "" {
response.Fail(w, "query 不能为空")
return
}
userId := fmt.Sprintf("%v", r.Context().Value("userId"))
header := w.Header()
header.Set("Content-Type", "text/event-stream")
header.Set("Cache-Control", "no-cache")
header.Set("Connection", "keep-alive")
header.Set("Transfer-Encoding", "chunked")
w.WriteHeader(http.StatusOK)
err := aiLogic.StreamChat(r.Context(), svcCtx, userId, query, w)
if err != nil {
_, _ = fmt.Fprintf(w, "data: [ERROR] %v\n\n", err)
if flusher, ok := w.(http.Flusher); ok {
flusher.Flush()
}
} else {
_, _ = fmt.Fprint(w, "data: [DONE]\n\n")
if flusher, ok := w.(http.Flusher); ok {
flusher.Flush()
}
}
}
}
func getBaiduAccessToken(apiKey, secretKey string) (string, error) {
tokenURL := fmt.Sprintf(
"https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=%s&client_secret=%s",
apiKey, secretKey,
)
resp, err := http.Post(tokenURL, "application/x-www-form-urlencoded", nil)
if err != nil {
return "", err
}
defer resp.Body.Close()
body, _ := io.ReadAll(resp.Body)
var tokenObj struct {
AccessToken string `json:"access_token"`
}
if err = json.Unmarshal(body, &tokenObj); err != nil || tokenObj.AccessToken == "" {
return "", fmt.Errorf("解析百度 token 失败")
}
return tokenObj.AccessToken, nil
}
func ClassifyPlantHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
contentType := r.Header.Get("Content-Type")
if !strings.Contains(contentType, "multipart/form-data") {
var req types.OcrReq
if err := httpx.Parse(r, &req); err != nil {
_ = json.NewDecoder(r.Body).Decode(&req)
}
if req.ImageUrl == "" {
response.Fail(w, "接收文件失败: imageUrl 不能为空并且必须使用 multipart/form-data 上传文件")
return
}
l := ocrLogic.NewOcrClassifyLogic(r.Context(), svcCtx)
resp, err := l.OcrClassify(&req)
if err != nil {
response.Fail(w, err.Error())
} else {
response.OkWithData(w, resp)
}
return
}
file, _, err := r.FormFile("file")
if err != nil {
response.Fail(w, "接收文件失败: "+err.Error())
return
}
defer file.Close()
fileBytes, err := io.ReadAll(file)
if err != nil {
response.Fail(w, "读取文件失败: "+err.Error())
return
}
apiKey := svcCtx.Config.BaiduOcr.ApiKey
secretKey := svcCtx.Config.BaiduOcr.SecretKey
if apiKey == "" || secretKey == "" {
response.Fail(w, "百度 OCR 未配置 ApiKey/SecretKey")
return
}
accessToken, err := getBaiduAccessToken(apiKey, secretKey)
if err != nil {
response.Fail(w, err.Error())
return
}
base64Str := base64.StdEncoding.EncodeToString(fileBytes)
escapedBase64 := url.QueryEscape(base64Str)
payload := strings.NewReader("image=" + escapedBase64 + "&baike_num=1")
apiURL := "https://aip.baidubce.com/rest/2.0/image-classify/v1/plant?access_token=" + accessToken
classifyReq, err := http.NewRequest("POST", apiURL, payload)
if err != nil {
response.Fail(w, "创建请求失败: "+err.Error())
return
}
classifyReq.Header.Set("Content-Type", "application/x-www-form-urlencoded")
client := &http.Client{}
classifyResp, err := client.Do(classifyReq)
if err != nil {
response.Fail(w, "调用百度植物识别接口失败: "+err.Error())
return
}
defer classifyResp.Body.Close()
body, err := io.ReadAll(classifyResp.Body)
if err != nil {
response.Fail(w, "读取识别结果失败: "+err.Error())
return
}
var baiduResp struct {
LogId uint64 `json:"log_id"`
Result []struct {
Score float64 `json:"score"`
Name string `json:"name"`
BaikeInfo *struct {
BaikeUrl string `json:"baike_url"`
ImageUrl string `json:"image_url"`
Description string `json:"description"`
} `json:"baike_info"`
} `json:"result"`
}
_ = json.Unmarshal(body, &baiduResp)
if baiduResp.LogId > 0 {
var dbResults plantModel.ResultsArray = make(plantModel.ResultsArray, 0, len(baiduResp.Result))
for _, item := range baiduResp.Result {
var baikeInfo *plantModel.BaikeInfo
if item.BaikeInfo != nil {
baikeInfo = &plantModel.BaikeInfo{
BaikeUrl: item.BaikeInfo.BaikeUrl,
ImageUrl: item.BaikeInfo.ImageUrl,
Description: item.BaikeInfo.Description,
}
}
dbResults = append(dbResults, plantModel.ResultItem{
Score: item.Score,
Name: item.Name,
BaikeInfo: baikeInfo,
})
}
userID := fmt.Sprintf("%v", r.Context().Value("userId"))
record := plantModel.ClassifyRecord{
UserID: userID,
LogID: baiduResp.LogId,
AllResults: dbResults,
}
if errDb := svcCtx.DB.Create(&record).Error; errDb != nil {
fmt.Printf("植物识别记录写入数据库失败: %v\n", errDb)
}
}
var result interface{}
if err = json.Unmarshal(body, &result); err != nil {
response.Fail(w, "解析识别结果失败: "+err.Error())
return
}
response.OkWithData(w, result)
}
}
// AddCarePlanHandler 兼容旧小程序的批量添加养护计划
func AddCarePlanHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
var req struct {
CarePlan []struct {
PlantID string `json:"plantId"`
Name string `json:"name"`
Period int32 `json:"period"`
Icon string `json:"icon"`
TargetAction string `json:"targetAction"`
} `json:"carePlan"`
}
if err := httpx.Parse(r, &req); err != nil {
response.Fail(w, "解析请求参数失败: "+err.Error())
return
}
userId := fmt.Sprintf("%v", r.Context().Value("userId"))
for _, item := range req.CarePlan {
if item.PlantID == "" {
response.Fail(w, "plantId 不能为空")
return
}
_, err := svcCtx.PlantRpc.AddCarePlan(r.Context(), &plantPb.AddCarePlanReq{
UserId: userId,
PlantId: item.PlantID,
Name: item.Name,
Icon: item.Icon,
Period: item.Period,
TargetAction: item.TargetAction,
})
if err != nil {
response.Fail(w, "添加养护计划失败: "+err.Error())
return
}
}
response.Ok(w)
}
}
+12 -2
View File
@@ -573,7 +573,7 @@ func RegisterHandlers(server *rest.Server, serverCtx *svc.ServiceContext) {
{Method: http.MethodPost, Path: "/update", Handler: myPlant.UpdatePlantHandler(serverCtx)}, {Method: http.MethodPost, Path: "/update", Handler: myPlant.UpdatePlantHandler(serverCtx)},
{Method: http.MethodPost, Path: "/deletePlant", Handler: myPlant.DeletePlantHandler(serverCtx)}, {Method: http.MethodPost, Path: "/deletePlant", Handler: myPlant.DeletePlantHandler(serverCtx)},
{Method: http.MethodPost, Path: "/deletePlan", Handler: myPlant.DeleteCarePlanHandler(serverCtx)}, {Method: http.MethodPost, Path: "/deletePlan", Handler: myPlant.DeleteCarePlanHandler(serverCtx)},
{Method: http.MethodPost, Path: "/plan/add", Handler: myPlant.AddCarePlanHandler(serverCtx)}, {Method: http.MethodPost, Path: "/plan/add", Handler: legacy.AddCarePlanHandler(serverCtx)},
{Method: http.MethodGet, Path: "/plan/delete", Handler: legacy.DeletePlanHandler(serverCtx)}, {Method: http.MethodGet, Path: "/plan/delete", Handler: legacy.DeletePlanHandler(serverCtx)},
{Method: http.MethodGet, Path: "/todayTask", Handler: myPlant.GetTodayTaskListHandler(serverCtx)}, {Method: http.MethodGet, Path: "/todayTask", Handler: myPlant.GetTodayTaskListHandler(serverCtx)},
{Method: http.MethodPost, Path: "/completeTask", Handler: complete.CompleteTaskHandler(serverCtx)}, {Method: http.MethodPost, Path: "/completeTask", Handler: complete.CompleteTaskHandler(serverCtx)},
@@ -604,7 +604,6 @@ func RegisterHandlers(server *rest.Server, serverCtx *svc.ServiceContext) {
{Method: http.MethodPost, Path: "/topic/page", Handler: topic.GetTopicListHandler(serverCtx)}, {Method: http.MethodPost, Path: "/topic/page", Handler: topic.GetTopicListHandler(serverCtx)},
{Method: http.MethodGet, Path: "/topic/detail", Handler: legacy.TopicDetailHandler(serverCtx)}, {Method: http.MethodGet, Path: "/topic/detail", Handler: legacy.TopicDetailHandler(serverCtx)},
{Method: http.MethodPost, Path: "/classify/plant", Handler: ocr.OcrClassifyHandler(serverCtx)},
{Method: http.MethodPost, Path: "/classify/myClassifyLog", Handler: ocr.GetMyClassifyLogHandler(serverCtx)}, {Method: http.MethodPost, Path: "/classify/myClassifyLog", Handler: ocr.GetMyClassifyLogHandler(serverCtx)},
{Method: http.MethodPost, Path: "/classify/deleteClassifyLog", Handler: ocr.DeleteClassifyLogHandler(serverCtx)}, {Method: http.MethodPost, Path: "/classify/deleteClassifyLog", Handler: ocr.DeleteClassifyLogHandler(serverCtx)},
@@ -626,4 +625,15 @@ func RegisterHandlers(server *rest.Server, serverCtx *svc.ServiceContext) {
rest.WithJwt(serverCtx.Config.Auth.AccessSecret), rest.WithJwt(serverCtx.Config.Auth.AccessSecret),
rest.WithPrefix("/api/plant"), rest.WithPrefix("/api/plant"),
) )
// SSE 流式问答 & 百度 OCR 识别:需要禁用 go-zero 默认超时(TimeoutHandler 会中断长连接)
server.AddRoutes(
[]rest.Route{
{Method: http.MethodGet, Path: "/chat/stream", Handler: legacy.AiChatStreamHandler(serverCtx)},
{Method: http.MethodPost, Path: "/classify/plant", Handler: legacy.ClassifyPlantHandler(serverCtx)},
},
rest.WithJwt(serverCtx.Config.Auth.AccessSecret),
rest.WithPrefix("/api/plant"),
rest.WithTimeout(0),
)
} }
+123 -154
View File
@@ -1,131 +1,101 @@
package ai package ai
import ( import (
"bufio"
"bytes"
"context" "context"
"encoding/json"
"errors" "errors"
"fmt" "fmt"
"io" "io"
"net/http" "net/http"
"strings" "strings"
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"
"sundynix-micro-go/app/plant/api/internal/svc" "sundynix-micro-go/app/plant/api/internal/svc"
plantModel "sundynix-micro-go/app/plant/model"
plantPb "sundynix-micro-go/app/plant/rpc/plant" plantPb "sundynix-micro-go/app/plant/rpc/plant"
) )
type chatMessage struct { func getActiveAiConfig(svcCtx *svc.ServiceContext) (*plantModel.SysAiConfig, error) {
Role string `json:"role"` var cfg plantModel.SysAiConfig
Content string `json:"content"` err := svcCtx.DB.Where("is_active = 1").First(&cfg).Error
if err != nil {
return nil, errors.New("AI/RAG 问答服务暂未激活或数据库配置缺失")
}
return &cfg, nil
} }
type chatRequest struct { func chatModel(dbCfg *plantModel.SysAiConfig) string {
Model string `json:"model,omitempty"` if dbCfg.ChatModelName != "" {
Messages []chatMessage `json:"messages"` return dbCfg.ChatModelName
Stream bool `json:"stream"`
}
func chatModel(svcCtx *svc.ServiceContext) string {
if svcCtx.Config.Ai.ChatModelName != "" {
return svcCtx.Config.Ai.ChatModelName
} }
return "gpt-4o-mini" return "gpt-4o-mini"
} }
func requestBody(svcCtx *svc.ServiceContext, question string, stream bool) ([]byte, error) { func newQdrantConn(cfg *plantModel.SysAiConfig) (*grpc.ClientConn, context.Context, error) {
systemPrompt := "你是一个专业的植物百科助手。回答规则:基于知识库信息回答,不够则结合通用知识;不要使用 Markdown;用纯文本分段;回答简洁专业、条理清晰。" addr := strings.TrimPrefix(cfg.QdrantUrl, "http://")
if ctxText := retrieveRAGContext(context.Background(), svcCtx, question); ctxText != "" { addr = strings.TrimPrefix(addr, "https://")
systemPrompt += "\n--- 知识库 ---\n" + ctxText + "\n--------------" conn, err := grpc.NewClient(addr, grpc.WithTransportCredentials(insecure.NewCredentials()))
if err != nil {
return nil, nil, fmt.Errorf("qdrant grpc dial failed: %w", err)
} }
return json.Marshal(chatRequest{ ctx := context.Background()
Model: chatModel(svcCtx), if cfg.QdrantApiKey != "" {
Messages: []chatMessage{ ctx = metadata.NewOutgoingContext(ctx, metadata.Pairs("api-key", cfg.QdrantApiKey))
{Role: "system", Content: systemPrompt}, }
{Role: "user", Content: question}, return conn, ctx, nil
},
Stream: stream,
})
} }
func retrieveRAGContext(ctx context.Context, svcCtx *svc.ServiceContext, question string) string { func retrieveRAGContext(ctx context.Context, svcCtx *svc.ServiceContext, dbCfg *plantModel.SysAiConfig, question string) string {
c := svcCtx.Config.Ai if dbCfg.EmbeddingApiUrl == "" || dbCfg.EmbeddingApiKey == "" || dbCfg.QdrantUrl == "" || dbCfg.QdrantCollection == "" {
if c.EmbeddingApiUrl == "" || c.EmbeddingApiKey == "" || c.QdrantUrl == "" || c.QdrantCollection == "" {
return ""
}
body, _ := json.Marshal(map[string]interface{}{
"model": c.EmbeddingModelName,
"input": question,
})
req, err := http.NewRequestWithContext(ctx, http.MethodPost, c.EmbeddingApiUrl, bytes.NewReader(body))
if err != nil {
return ""
}
req.Header.Set("Content-Type", "application/json")
req.Header.Set("Authorization", "Bearer "+c.EmbeddingApiKey)
resp, err := http.DefaultClient.Do(req)
if err != nil {
return ""
}
defer resp.Body.Close()
var emb struct {
Data []struct {
Embedding []float32 `json:"embedding"`
} `json:"data"`
}
if resp.StatusCode < 200 || resp.StatusCode >= 300 || json.NewDecoder(resp.Body).Decode(&emb) != nil || len(emb.Data) == 0 {
return "" return ""
} }
searchBody, _ := json.Marshal(map[string]interface{}{ config := openai.DefaultConfig(dbCfg.EmbeddingApiKey)
"vector": emb.Data[0].Embedding, if dbCfg.EmbeddingApiUrl != "" {
"limit": 3, config.BaseURL = dbCfg.EmbeddingApiUrl
"with_payload": true, }
client := openai.NewClientWithConfig(config)
embResp, err := client.CreateEmbeddings(ctx, openai.EmbeddingRequest{
Input: []string{question},
Model: openai.EmbeddingModel(dbCfg.EmbeddingModelName),
}) })
searchReq, err := http.NewRequestWithContext(ctx, http.MethodPost, strings.TrimRight(c.QdrantUrl, "/")+"/collections/"+c.QdrantCollection+"/points/search", bytes.NewReader(searchBody))
if err != nil { if err != nil {
return "" return ""
} }
searchReq.Header.Set("Content-Type", "application/json")
if c.QdrantApiKey != "" { conn, qdCtx, connErr := newQdrantConn(dbCfg)
searchReq.Header.Set("api-key", c.QdrantApiKey) if connErr != nil {
}
searchResp, err := http.DefaultClient.Do(searchReq)
if err != nil {
return "" return ""
} }
defer searchResp.Body.Close() defer conn.Close()
var parsed struct {
Result []struct { ptsClient := qdrant.NewPointsClient(conn)
Payload map[string]interface{} `json:"payload"` searchRes, searchErr := ptsClient.Search(qdCtx, &qdrant.SearchPoints{
} `json:"result"` CollectionName: dbCfg.QdrantCollection,
} Vector: embResp.Data[0].Embedding,
if searchResp.StatusCode < 200 || searchResp.StatusCode >= 300 || json.NewDecoder(searchResp.Body).Decode(&parsed) != nil { Limit: 3,
WithPayload: &qdrant.WithPayloadSelector{
SelectorOptions: &qdrant.WithPayloadSelector_Enable{Enable: true},
},
})
if searchErr != nil {
return "" return ""
} }
var b strings.Builder var b strings.Builder
for _, item := range parsed.Result { for _, pt := range searchRes.GetResult() {
if text, ok := item.Payload["full_text"].(string); ok && text != "" { if txt, ok := pt.GetPayload()["full_text"]; ok {
b.WriteString(text) b.WriteString(txt.GetStringValue())
b.WriteString("\n") b.WriteString("\n")
} }
} }
return b.String() return b.String()
} }
func newChatRequest(ctx context.Context, svcCtx *svc.ServiceContext, body []byte) (*http.Request, error) {
if svcCtx.Config.Ai.ChatApiUrl == "" || svcCtx.Config.Ai.ChatApiKey == "" {
return nil, errors.New("AI/RAG 未配置 ChatApiUrl 或 ChatApiKey")
}
req, err := http.NewRequestWithContext(ctx, http.MethodPost, svcCtx.Config.Ai.ChatApiUrl, bytes.NewReader(body))
if err != nil {
return nil, err
}
req.Header.Set("Content-Type", "application/json")
req.Header.Set("Authorization", "Bearer "+svcCtx.Config.Ai.ChatApiKey)
return req, nil
}
func SaveHistory(ctx context.Context, svcCtx *svc.ServiceContext, userID, question, answer string) { func SaveHistory(ctx context.Context, svcCtx *svc.ServiceContext, userID, question, answer string) {
if userID == "" || question == "" || answer == "" { if userID == "" || question == "" || answer == "" {
return return
@@ -136,102 +106,100 @@ func SaveHistory(ctx context.Context, svcCtx *svc.ServiceContext, userID, questi
} }
func ChatCompletion(ctx context.Context, svcCtx *svc.ServiceContext, userID, question string) (string, error) { func ChatCompletion(ctx context.Context, svcCtx *svc.ServiceContext, userID, question string) (string, error) {
if err := ensureQuota(ctx, svcCtx, userID); err != nil { dbCfg, err := getActiveAiConfig(svcCtx)
return "", err
}
body, err := requestBody(svcCtx, question, false)
if err != nil { if err != nil {
return "", err return "", err
} }
req, err := newChatRequest(ctx, svcCtx, body) if err := ensureQuota(ctx, svcCtx, userID, dbCfg); err != nil {
if err != nil {
return "", err return "", err
} }
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("AI 请求失败: %s %s", resp.Status, strings.TrimSpace(string(raw)))
}
var parsed struct { systemPrompt := "你是一个专业的植物百科助手。回答规则:基于知识库信息回答,不够则结合通用知识;不要使用 Markdown;用纯文本分段;回答简洁专业、条理清晰。"
Choices []struct { if ctxText := retrieveRAGContext(ctx, svcCtx, dbCfg, question); ctxText != "" {
Message struct { systemPrompt += "\n--- 知识库 ---\n" + ctxText + "\n--------------"
Content string `json:"content"`
} `json:"message"`
} `json:"choices"`
} }
if err := json.NewDecoder(resp.Body).Decode(&parsed); err != nil {
config := openai.DefaultConfig(dbCfg.ChatApiKey)
if dbCfg.ChatApiUrl != "" {
config.BaseURL = dbCfg.ChatApiUrl
}
client := openai.NewClientWithConfig(config)
resp, err := client.CreateChatCompletion(ctx, openai.ChatCompletionRequest{
Model: chatModel(dbCfg),
Messages: []openai.ChatCompletionMessage{
{Role: openai.ChatMessageRoleSystem, Content: systemPrompt},
{Role: openai.ChatMessageRoleUser, Content: question},
},
})
if err != nil {
return "", err return "", err
} }
if len(parsed.Choices) == 0 || parsed.Choices[0].Message.Content == "" { if len(resp.Choices) == 0 || resp.Choices[0].Message.Content == "" {
return "", errors.New("AI 响应为空") return "", errors.New("AI 响应为空")
} }
answer := parsed.Choices[0].Message.Content answer := resp.Choices[0].Message.Content
SaveHistory(ctx, svcCtx, userID, question, answer) SaveHistory(ctx, svcCtx, userID, question, answer)
return answer, nil return answer, nil
} }
func StreamChat(ctx context.Context, svcCtx *svc.ServiceContext, userID, question string, w io.Writer) error { func StreamChat(ctx context.Context, svcCtx *svc.ServiceContext, userID, question string, w io.Writer) error {
if err := ensureQuota(ctx, svcCtx, userID); err != nil { dbCfg, err := getActiveAiConfig(svcCtx)
return err
}
body, err := requestBody(svcCtx, question, true)
if err != nil { if err != nil {
return err return err
} }
req, err := newChatRequest(ctx, svcCtx, body) if err := ensureQuota(ctx, svcCtx, userID, dbCfg); err != nil {
if err != nil {
return err return err
} }
resp, err := http.DefaultClient.Do(req)
if err != nil { systemPrompt := "你是一个专业的植物百科助手。回答规则:基于知识库信息回答,不够则结合通用知识;不要使用 Markdown;用纯文本分段;回答简洁专业、条理清晰。"
return err if ctxText := retrieveRAGContext(ctx, svcCtx, dbCfg, question); ctxText != "" {
} systemPrompt += "\n--- 知识库 ---\n" + ctxText + "\n--------------"
defer resp.Body.Close() }
if resp.StatusCode < 200 || resp.StatusCode >= 300 {
raw, _ := io.ReadAll(io.LimitReader(resp.Body, 4096)) config := openai.DefaultConfig(dbCfg.ChatApiKey)
return fmt.Errorf("AI 请求失败: %s %s", resp.Status, strings.TrimSpace(string(raw))) if dbCfg.ChatApiUrl != "" {
} config.BaseURL = dbCfg.ChatApiUrl
}
client := openai.NewClientWithConfig(config)
stream, err := client.CreateChatCompletionStream(ctx, openai.ChatCompletionRequest{
Model: chatModel(dbCfg),
Messages: []openai.ChatCompletionMessage{
{Role: openai.ChatMessageRoleSystem, Content: systemPrompt},
{Role: openai.ChatMessageRoleUser, Content: question},
},
Stream: true,
})
if err != nil {
return err
}
defer stream.Close()
var answer strings.Builder var answer strings.Builder
scanner := bufio.NewScanner(resp.Body) for {
for scanner.Scan() { resp, recvErr := stream.Recv()
line := scanner.Text() if errors.Is(recvErr, io.EOF) {
_, _ = fmt.Fprintln(w, line) break
}
if recvErr != nil {
return recvErr
}
if len(resp.Choices) > 0 {
content := resp.Choices[0].Delta.Content
if content != "" {
_, _ = fmt.Fprintf(w, "data: %s\n\n", content)
if flusher, ok := w.(http.Flusher); ok { if flusher, ok := w.(http.Flusher); ok {
flusher.Flush() flusher.Flush()
} }
answer.WriteString(content)
}
}
}
if !strings.HasPrefix(line, "data: ") {
continue
}
data := strings.TrimSpace(strings.TrimPrefix(line, "data: "))
if data == "[DONE]" {
continue
}
var chunk struct {
Choices []struct {
Delta struct {
Content string `json:"content"`
} `json:"delta"`
} `json:"choices"`
}
if json.Unmarshal([]byte(data), &chunk) == nil && len(chunk.Choices) > 0 {
answer.WriteString(chunk.Choices[0].Delta.Content)
}
}
if err := scanner.Err(); err != nil {
return err
}
SaveHistory(ctx, svcCtx, userID, question, answer.String()) SaveHistory(ctx, svcCtx, userID, question, answer.String())
return nil return nil
} }
func ensureQuota(ctx context.Context, svcCtx *svc.ServiceContext, userID string) error { func ensureQuota(ctx context.Context, svcCtx *svc.ServiceContext, userID string, dbCfg *plantModel.SysAiConfig) error {
if userID == "" { if userID == "" {
return nil return nil
} }
@@ -239,8 +207,9 @@ func ensureQuota(ctx context.Context, svcCtx *svc.ServiceContext, userID string)
if err != nil { if err != nil {
return err return err
} }
if quota.Limit > 0 && quota.Remaining <= 0 { limit := int64(dbCfg.DailyQueryLimit)
return fmt.Errorf("今日问答次数已达上限(%d次),明天再来吧", quota.Limit) if limit > 0 && quota.Remaining <= 0 {
return fmt.Errorf("今日问答次数已达上限(%d次),明天再来吧", limit)
} }
return nil return nil
} }
@@ -3,10 +3,13 @@ package myPlant
import ( import (
"context" "context"
"fmt" "fmt"
"github.com/zeromicro/go-zero/core/logx"
"sundynix-micro-go/app/plant/api/internal/svc" "sundynix-micro-go/app/plant/api/internal/svc"
"sundynix-micro-go/app/plant/api/internal/types" "sundynix-micro-go/app/plant/api/internal/types"
"sundynix-micro-go/app/plant/rpc/plant" plantModel "sundynix-micro-go/app/plant/model"
"github.com/zeromicro/go-zero/core/logx"
"gorm.io/gorm"
) )
type AddGrowthRecordLogic struct { type AddGrowthRecordLogic struct {
@@ -21,8 +24,36 @@ func NewAddGrowthRecordLogic(ctx context.Context, svcCtx *svc.ServiceContext) *A
func (l *AddGrowthRecordLogic) AddGrowthRecord(req *types.GrowthRecordReq) error { func (l *AddGrowthRecordLogic) AddGrowthRecord(req *types.GrowthRecordReq) error {
userId := fmt.Sprintf("%v", l.ctx.Value("userId")) userId := fmt.Sprintf("%v", l.ctx.Value("userId"))
_, err := l.svcCtx.PlantRpc.AddGrowthRecord(l.ctx, &plant.AddGrowthRecordReq{ imgIds := req.ImgIds
UserId: userId, PlantId: req.PlantId, Content: req.Content, ImgIds: req.ImgIds, if len(imgIds) == 0 && len(req.OssIds) > 0 {
imgIds = req.OssIds
}
err := l.svcCtx.DB.Transaction(func(tx *gorm.DB) error {
record := plantModel.GrowthRecord{
UserID: userId,
PlantID: req.PlantId,
Name: req.Name,
Tag: req.Tag,
Desc: req.Desc,
Content: req.Content,
}
if err := tx.Create(&record).Error; err != nil {
return err
}
// 保存图片关联
if len(imgIds) > 0 {
relations := make([]plantModel.GrowthRecordOss, 0, len(imgIds))
for _, ossId := range imgIds {
relations = append(relations, plantModel.GrowthRecordOss{
GrowthRecordID: record.ID, OssID: ossId,
})
}
if err := tx.Create(&relations).Error; err != nil {
return err
}
}
return nil
}) })
return err return err
} }
@@ -3,9 +3,11 @@ package ocr
import ( import (
"context" "context"
"fmt" "fmt"
"time"
"github.com/zeromicro/go-zero/core/logx" "github.com/zeromicro/go-zero/core/logx"
"sundynix-micro-go/app/plant/api/internal/svc" "sundynix-micro-go/app/plant/api/internal/svc"
plantPb "sundynix-micro-go/app/plant/rpc/plant" plantModel "sundynix-micro-go/app/plant/model"
) )
type GetMyClassifyLogLogic struct { type GetMyClassifyLogLogic struct {
@@ -14,11 +16,53 @@ type GetMyClassifyLogLogic struct {
svcCtx *svc.ServiceContext svcCtx *svc.ServiceContext
} }
type ClassifyRecordResp struct {
List []ClassifyRecordInfo `json:"list"`
Total int64 `json:"total"`
}
type ClassifyRecordInfo struct {
ID string `json:"id"`
UserID string `json:"userId"`
LogID uint64 `json:"logId"`
AllResults plantModel.ResultsArray `json:"allResults"`
CreatedAt string `json:"createdAt"`
CreatedAtStr string `json:"createdAtStr"`
}
func NewGetMyClassifyLogLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetMyClassifyLogLogic { func NewGetMyClassifyLogLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetMyClassifyLogLogic {
return &GetMyClassifyLogLogic{Logger: logx.WithContext(ctx), ctx: ctx, svcCtx: svcCtx} return &GetMyClassifyLogLogic{Logger: logx.WithContext(ctx), ctx: ctx, svcCtx: svcCtx}
} }
func (l *GetMyClassifyLogLogic) GetMyClassifyLog() (*plantPb.ClassifyLogListResp, error) { func (l *GetMyClassifyLogLogic) GetMyClassifyLog() (*ClassifyRecordResp, error) {
userId := fmt.Sprintf("%v", l.ctx.Value("userId")) userId := fmt.Sprintf("%v", l.ctx.Value("userId"))
return l.svcCtx.PlantRpc.GetMyClassifyLog(l.ctx, &plantPb.GetProfileReq{UserId: userId})
var records []plantModel.ClassifyRecord
var total int64
db := l.svcCtx.DB.Model(&plantModel.ClassifyRecord{}).Where("user_id = ?", userId)
if err := db.Count(&total).Error; err != nil {
return nil, err
}
if err := db.Order("created_at desc").Limit(50).Find(&records).Error; err != nil {
return nil, err
}
list := make([]ClassifyRecordInfo, 0, len(records))
for _, og := range records {
list = append(list, ClassifyRecordInfo{
ID: og.ID,
UserID: og.UserID,
LogID: og.LogID,
AllResults: og.AllResults,
CreatedAt: og.CreatedAt.Format(time.RFC3339),
CreatedAtStr: og.CreatedAt.Format("2006-01-02 15:04:05"),
})
}
return &ClassifyRecordResp{
List: list,
Total: total,
}, nil
} }
@@ -14,6 +14,7 @@ import (
"sundynix-micro-go/app/plant/api/internal/svc" "sundynix-micro-go/app/plant/api/internal/svc"
"sundynix-micro-go/app/plant/api/internal/types" "sundynix-micro-go/app/plant/api/internal/types"
plantModel "sundynix-micro-go/app/plant/model"
"github.com/zeromicro/go-zero/core/logx" "github.com/zeromicro/go-zero/core/logx"
) )
@@ -83,7 +84,51 @@ func (l *OcrClassifyLogic) OcrClassify(req *types.OcrReq) (interface{}, error) {
return nil, fmt.Errorf("读取识别结果失败: %w", err) return nil, fmt.Errorf("读取识别结果失败: %w", err)
} }
// 3. 直接返回百度原始结果(前端自行解析 result 字段) // 3. 解析为结构化识别结果并写入 ClassifyRecord 表
var baiduResp struct {
LogId uint64 `json:"log_id"`
Result []struct {
Score float64 `json:"score"`
Name string `json:"name"`
BaikeInfo *struct {
BaikeUrl string `json:"baike_url"`
ImageUrl string `json:"image_url"`
Description string `json:"description"`
} `json:"baike_info"`
} `json:"result"`
}
_ = json.Unmarshal(body, &baiduResp)
if baiduResp.LogId > 0 {
var dbResults plantModel.ResultsArray = make(plantModel.ResultsArray, 0, len(baiduResp.Result))
for _, item := range baiduResp.Result {
var baikeInfo *plantModel.BaikeInfo
if item.BaikeInfo != nil {
baikeInfo = &plantModel.BaikeInfo{
BaikeUrl: item.BaikeInfo.BaikeUrl,
ImageUrl: item.BaikeInfo.ImageUrl,
Description: item.BaikeInfo.Description,
}
}
dbResults = append(dbResults, plantModel.ResultItem{
Score: item.Score,
Name: item.Name,
BaikeInfo: baikeInfo,
})
}
userID := fmt.Sprintf("%v", l.ctx.Value("userId"))
record := plantModel.ClassifyRecord{
UserID: userID,
LogID: baiduResp.LogId,
AllResults: dbResults,
}
if errDb := l.svcCtx.DB.Create(&record).Error; errDb != nil {
l.Logger.Errorf("植物识别记录写入数据库失败: %v", errDb)
}
}
// 4. 直接返回百度原始结果
var result interface{} var result interface{}
if err = json.Unmarshal(body, &result); err != nil { if err = json.Unmarshal(body, &result); err != nil {
return nil, fmt.Errorf("解析识别结果失败: %w", err) return nil, fmt.Errorf("解析识别结果失败: %w", err)
@@ -31,12 +31,16 @@ func NewCreatePostLogic(ctx context.Context, svcCtx *svc.ServiceContext) *Create
func (l *CreatePostLogic) CreatePost(req *types.CreatePostReq) error { func (l *CreatePostLogic) CreatePost(req *types.CreatePostReq) error {
userId := fmt.Sprintf("%v", l.ctx.Value("userId")) userId := fmt.Sprintf("%v", l.ctx.Value("userId"))
imgIds := req.ImgIds
if len(imgIds) == 0 && len(req.OssIds) > 0 {
imgIds = req.OssIds
}
_, err := l.svcCtx.PlantRpc.CreatePost(l.ctx, &plant.CreatePostReq{ _, err := l.svcCtx.PlantRpc.CreatePost(l.ctx, &plant.CreatePostReq{
UserId: userId, UserId: userId,
Title: req.Title, Title: req.Title,
Content: req.Content, Content: req.Content,
Location: req.Location, Location: req.Location,
ImgIds: req.ImgIds, ImgIds: imgIds,
TopicId: req.TopicId, TopicId: req.TopicId,
}) })
return err return err
@@ -3,10 +3,15 @@ package userProfile
import ( import (
"context" "context"
"fmt" "fmt"
"github.com/zeromicro/go-zero/core/logx" "time"
filePb "sundynix-micro-go/app/file/rpc/file"
"sundynix-micro-go/app/plant/api/internal/svc" "sundynix-micro-go/app/plant/api/internal/svc"
"sundynix-micro-go/app/plant/api/internal/types" "sundynix-micro-go/app/plant/api/internal/types"
plantPb "sundynix-micro-go/app/plant/rpc/plant" plantModel "sundynix-micro-go/app/plant/model"
"github.com/zeromicro/go-zero/core/logx"
"gorm.io/gorm"
) )
type GetMyStarsLogic struct { type GetMyStarsLogic struct {
@@ -19,7 +24,348 @@ func NewGetMyStarsLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetMyS
return &GetMyStarsLogic{Logger: logx.WithContext(ctx), ctx: ctx, svcCtx: svcCtx} return &GetMyStarsLogic{Logger: logx.WithContext(ctx), ctx: ctx, svcCtx: svcCtx}
} }
func (l *GetMyStarsLogic) GetMyStars(req *types.PageReq) (*plantPb.UserStarListResp, error) { func (l *GetMyStarsLogic) GetMyStars(req *types.PageReq) (interface{}, error) {
userId := fmt.Sprintf("%v", l.ctx.Value("userId")) userId := fmt.Sprintf("%v", l.ctx.Value("userId"))
return l.svcCtx.PlantRpc.GetMyStars(l.ctx, &plantPb.GetProfileReq{UserId: userId})
db := l.svcCtx.DB.Model(&plantModel.UserStar{}).Where("user_id = ?", userId)
if req.Class == 1 {
db = db.Where("type = ?", "wiki")
} else if req.Class == 2 {
db = db.Where("type = ?", "post")
}
var total int64
if err := db.Count(&total).Error; err != nil {
return nil, err
}
pageSize := req.PageSize
if pageSize <= 0 {
pageSize = 10
}
page := req.Current
if page <= 0 {
page = 1
}
offset := (page - 1) * pageSize
var stars []*plantModel.UserStar
if err := db.Limit(pageSize).Offset(offset).Order("created_at desc").Find(&stars).Error; err != nil {
return nil, err
}
// 提取 TargetID
var wikiIds []string
var postIds []string
for _, s := range stars {
if s.Type == "wiki" {
wikiIds = append(wikiIds, s.TargetID)
} else if s.Type == "post" {
postIds = append(postIds, s.TargetID)
}
}
// 1. 查询 Wiki 详情
wikiMap := make(map[string]map[string]interface{})
if len(wikiIds) > 0 {
var wikis []*plantModel.Wiki
if err := l.svcCtx.DB.Where("id IN ?", wikiIds).Find(&wikis).Error; err == nil {
// 查本地 WikiOss
type rel struct {
WikiID string `gorm:"column:wiki_id"`
OssID string `gorm:"column:oss_id"`
}
var rels []rel
l.svcCtx.DB.Table("sundynix_plant_wiki_oss").Where("wiki_id IN ?", wikiIds).Find(&rels)
wikiOssMap := make(map[string][]string)
var allOssIds []string
for _, r := range rels {
wikiOssMap[r.WikiID] = append(wikiOssMap[r.WikiID], r.OssID)
allOssIds = append(allOssIds, r.OssID)
}
// 通过 FileRpc 获取图片信息
fileMap := l.fetchFileMap(allOssIds)
for _, w := range wikis {
ossIds := wikiOssMap[w.ID]
imgList := l.imgListToList(fileMap, ossIds)
wikiMap[w.ID] = map[string]interface{}{
"id": w.ID, "name": w.Name, "latinName": w.LatinName,
"aliases": w.Aliases, "genus": w.Genus, "difficulty": w.Difficulty,
"isHot": w.IsHot, "growthHabit": w.GrowthHabit,
"lightIntensity": w.LightIntensity, "classId": w.ClassID,
"createdAt": w.CreatedAt.Format("2006-01-02 15:04:05"),
"hasStar": 1, // 既然在这个列表中,说明一定是被收藏的
"imgList": imgList,
}
}
}
}
// 2. 查询 Post 详情
postMap := make(map[string]map[string]interface{})
if len(postIds) > 0 {
var posts []*plantModel.Post
if err := l.svcCtx.DB.
Preload("CommentList", func(db *gorm.DB) *gorm.DB {
return db.Order("created_at asc")
}).
Preload("LikeList").
Where("id IN ?", postIds).Find(&posts).Error; err == nil {
// 查帖子图片
postImgMap := l.queryPostImages(postIds)
// 查用户信息
allUserIds := l.collectPostUserIds(posts)
userMap := l.queryUserMap(allUserIds)
// 点赞收藏状态
likedSet, starredSet := l.queryLikeStarStatus(userId, postIds)
for _, p := range posts {
item := map[string]interface{}{
"id": p.ID, "title": p.Title, "content": p.Content,
"userId": p.UserID, "location": p.Location,
"viewCount": p.ViewCount, "commentCount": p.CommentCount,
"likeCount": p.LikeCount, "starCount": p.StarCount,
"hasReviewed": p.HasReviewed,
"createdAt": p.CreatedAt.Format(time.RFC3339),
"updatedAt": p.UpdatedAt.Format(time.RFC3339),
"createdAtStr": p.CreatedAt.Format("2006-01-02 15:04:05"),
"hasLiked": 0, "hasStar": 0,
"imgList": postImgMap[p.ID],
"publisher": l.buildPublisherInfo(userMap, p.UserID),
"commentList": l.buildCommentList(userMap, p.CommentList),
"likeList": l.buildLikeList(userMap, p.LikeList),
"starList": []map[string]interface{}{},
}
if likedSet[p.ID] {
item["hasLiked"] = 1
}
if starredSet[p.ID] {
item["hasStar"] = 1
}
postMap[p.ID] = item
}
}
}
// 3. 按照 Stars 排序组装最终 List
var list []map[string]interface{}
for _, s := range stars {
item := map[string]interface{}{
"id": s.ID, "userId": s.UserID, "targetId": s.TargetID, "type": s.Type,
"createdAt": s.CreatedAt.Format("2006-01-02 15:04:05"),
}
if s.Type == "wiki" {
if w, ok := wikiMap[s.TargetID]; ok {
item["wiki"] = w
list = append(list, item)
}
} else if s.Type == "post" {
if p, ok := postMap[s.TargetID]; ok {
item["post"] = p
list = append(list, item)
}
}
}
if list == nil {
list = []map[string]interface{}{}
}
return map[string]interface{}{
"list": list,
"total": total,
"page": page,
"pageSize": pageSize,
}, nil
}
func (l *GetMyStarsLogic) fetchFileMap(ids []string) map[string]map[string]interface{} {
result := make(map[string]map[string]interface{})
if len(ids) == 0 {
return result
}
resp, err := l.svcCtx.FileRpc.GetFilesByIds(l.ctx, &filePb.GetFilesByIdsReq{Ids: ids})
if err != nil || resp == nil {
return result
}
for _, f := range resp.Files {
result[f.Id] = map[string]interface{}{
"id": f.Id, "name": f.Name, "url": f.Url, "tag": f.Tag,
"key": f.Key, "suffix": f.Suffix, "md5": f.Md5,
"createdAt": time.Unix(f.CreatedAt, 0).Format(time.RFC3339),
"updatedAt": time.Unix(f.CreatedAt, 0).Format(time.RFC3339),
"createdAtStr": time.Unix(f.CreatedAt, 0).Format("2006-01-02 15:04:05"),
}
}
return result
}
func (l *GetMyStarsLogic) imgListToList(fileMap map[string]map[string]interface{}, ossIds []string) []map[string]interface{} {
var list []map[string]interface{}
for _, id := range ossIds {
if img, ok := fileMap[id]; ok {
list = append(list, img)
}
}
if list == nil {
list = []map[string]interface{}{}
}
return list
}
func (l *GetMyStarsLogic) queryPostImages(postIds []string) map[string][]map[string]interface{} {
type rel struct {
PostID string `gorm:"column:post_id"`
OssID string `gorm:"column:oss_id"`
}
var rels []rel
l.svcCtx.DB.Table("sundynix_plant_post_oss").Where("post_id IN ?", postIds).Find(&rels)
var allOssIds []string
pidMap := make(map[string][]string)
for _, r := range rels {
pidMap[r.PostID] = append(pidMap[r.PostID], r.OssID)
allOssIds = append(allOssIds, r.OssID)
}
fileInfos := l.fetchFileMap(allOssIds)
result := make(map[string][]map[string]interface{})
for pid, ids := range pidMap {
var imgs []map[string]interface{}
for _, oid := range ids {
if info, ok := fileInfos[oid]; ok {
imgs = append(imgs, info)
}
}
if imgs == nil {
imgs = []map[string]interface{}{}
}
result[pid] = imgs
}
for _, pid := range postIds {
if _, ok := result[pid]; !ok {
result[pid] = []map[string]interface{}{}
}
}
return result
}
func (l *GetMyStarsLogic) collectPostUserIds(posts []*plantModel.Post) []string {
set := make(map[string]bool)
for _, p := range posts {
set[p.UserID] = true
for _, c := range p.CommentList {
set[c.UserID] = true
}
for _, l := range p.LikeList {
set[l.UserID] = true
}
}
var ids []string
for id := range set {
ids = append(ids, id)
}
return ids
}
func (l *GetMyStarsLogic) queryUserMap(ids []string) map[string]map[string]interface{} {
result := make(map[string]map[string]interface{})
if len(ids) == 0 {
return result
}
type userRow struct {
ID string `gorm:"column:id"`
NickName string `gorm:"column:nick_name"`
Name string `gorm:"column:name"`
AvatarID string `gorm:"column:avatar_id"`
}
var rows []userRow
l.svcCtx.DB.Table("sundynix_user").Where("id IN ?", ids).Find(&rows)
var avatarIds []string
for _, row := range rows {
if row.AvatarID != "" {
avatarIds = append(avatarIds, row.AvatarID)
}
}
avatarMap := l.fetchFileMap(avatarIds)
for _, row := range rows {
avatarData := map[string]interface{}{}
if av, ok := avatarMap[row.AvatarID]; ok {
avatarData = av
}
result[row.ID] = map[string]interface{}{
"id": row.ID, "nickName": row.NickName, "name": row.Name,
"avatarId": row.AvatarID, "avatar": avatarData,
}
}
return result
}
func (l *GetMyStarsLogic) queryLikeStarStatus(userId string, postIds []string) (likedSet, starredSet map[string]bool) {
likedSet = make(map[string]bool)
starredSet = make(map[string]bool)
if len(postIds) == 0 {
return
}
type rel struct {
PostID string `gorm:"column:post_id"`
}
var likes []rel
l.svcCtx.DB.Table("sundynix_plant_post_like").Where("post_id IN ? AND user_id = ?", postIds, userId).Find(&likes)
for _, l := range likes {
likedSet[l.PostID] = true
}
var stars []rel
l.svcCtx.DB.Table("sundynix_plant_user_star").Where("target_id IN ? AND user_id = ? AND type = 'post'", postIds, userId).Find(&stars)
for _, s := range stars {
starredSet[s.PostID] = true
}
return
}
func (l *GetMyStarsLogic) buildPublisherInfo(userMap map[string]map[string]interface{}, userId string) map[string]interface{} {
if u, ok := userMap[userId]; ok {
return u
}
return map[string]interface{}{
"id": userId, "nickName": "", "name": "", "avatarId": "", "avatar": map[string]interface{}{},
}
}
func (l *GetMyStarsLogic) buildCommentList(userMap map[string]map[string]interface{}, comments []*plantModel.PostComment) []map[string]interface{} {
var list []map[string]interface{}
for _, c := range comments {
list = append(list, map[string]interface{}{
"id": c.ID, "postId": c.PostID, "userId": c.UserID,
"content": c.Content, "parentId": c.ParentID,
"createdAt": c.CreatedAt.Format(time.RFC3339),
"updatedAt": c.UpdatedAt.Format(time.RFC3339),
"createdAtStr": c.CreatedAt.Format("2006-01-02 15:04:05"),
"commentator": l.buildPublisherInfo(userMap, c.UserID),
})
}
if list == nil {
list = []map[string]interface{}{}
}
return list
}
func (l *GetMyStarsLogic) buildLikeList(userMap map[string]map[string]interface{}, likes []*plantModel.PostLike) []map[string]interface{} {
var list []map[string]interface{}
for _, like := range likes {
list = append(list, map[string]interface{}{
"id": like.ID, "postId": like.PostID, "userId": like.UserID,
"liker": l.buildPublisherInfo(userMap, like.UserID),
})
}
if list == nil {
list = []map[string]interface{}{}
}
return list
} }
@@ -29,9 +29,13 @@ func NewCreateWikiClassLogic(ctx context.Context, svcCtx *svc.ServiceContext) *C
} }
func (l *CreateWikiClassLogic) CreateWikiClass(req *types.WikiClassReq) error { func (l *CreateWikiClassLogic) CreateWikiClass(req *types.WikiClassReq) error {
icon := req.Icon
if icon == "" && req.OssId != "" {
icon = req.OssId
}
_, err := l.svcCtx.PlantRpc.CreateWikiClass(l.ctx, &plant.CreateWikiClassReq{ _, err := l.svcCtx.PlantRpc.CreateWikiClass(l.ctx, &plant.CreateWikiClassReq{
Name: req.Name, Name: req.Name,
Icon: req.Icon, Icon: icon,
}) })
return err return err
} }
@@ -19,8 +19,12 @@ func NewUpdateWikiClassLogic(ctx context.Context, svcCtx *svc.ServiceContext) *U
} }
func (l *UpdateWikiClassLogic) UpdateWikiClass(req *types.UpdateWikiClassReq) error { func (l *UpdateWikiClassLogic) UpdateWikiClass(req *types.UpdateWikiClassReq) error {
icon := req.Icon
if icon == "" && req.OssId != "" {
icon = req.OssId
}
_, err := l.svcCtx.PlantRpc.UpdateWikiClass(l.ctx, &plantPb.UpdateWikiClassReq{ _, err := l.svcCtx.PlantRpc.UpdateWikiClass(l.ctx, &plantPb.UpdateWikiClassReq{
Id: req.Id, Name: req.Name, Icon: req.Icon, Id: req.Id, Name: req.Name, Icon: icon,
}) })
return err return err
} }
+4
View File
@@ -88,6 +88,7 @@ type CreatePostReq struct {
Content string `json:"content"` Content string `json:"content"`
Location string `json:"location,optional"` Location string `json:"location,optional"`
ImgIds []string `json:"imgIds,optional"` ImgIds []string `json:"imgIds,optional"`
OssIds []string `json:"ossIds,optional"` // 向下兼容旧版
TopicId string `json:"topicId,optional"` TopicId string `json:"topicId,optional"`
} }
@@ -180,6 +181,7 @@ type PageReq struct {
Current int `json:"current,optional"` Current int `json:"current,optional"`
PageSize int `json:"pageSize,optional"` PageSize int `json:"pageSize,optional"`
Keyword string `json:"keyword,optional"` Keyword string `json:"keyword,optional"`
Class int `json:"class,optional"` // 分类过滤: 0 全部, 1 百科, 2 动态
} }
type PlantListReq struct { type PlantListReq struct {
@@ -292,6 +294,7 @@ type UpdateWikiClassReq struct {
Id string `json:"id"` Id string `json:"id"`
Name string `json:"name,optional"` Name string `json:"name,optional"`
Icon string `json:"icon,optional"` Icon string `json:"icon,optional"`
OssId string `json:"ossId,optional"` // 向下兼容旧版
} }
type UpdateWikiReq struct { type UpdateWikiReq struct {
@@ -337,6 +340,7 @@ type UpdatePlanReq struct {
type WikiClassReq struct { type WikiClassReq struct {
Name string `json:"name"` Name string `json:"name"`
Icon string `json:"icon,optional"` Icon string `json:"icon,optional"`
OssId string `json:"ossId,optional"` // 向下兼容旧版
} }
// ========== Banner ========== // ========== Banner ==========
+75 -13
View File
@@ -1,6 +1,9 @@
package model package model
import ( import (
"database/sql/driver"
"encoding/json"
"errors"
"sundynix-micro-go/common/model" "sundynix-micro-go/common/model"
"time" "time"
) )
@@ -247,19 +250,6 @@ type Topic struct {
func (Topic) TableName() string { return "sundynix_plant_topic" } func (Topic) TableName() string { return "sundynix_plant_topic" }
// ========== OCR ==========
// OcrLog OCR识别记录
type OcrLog struct {
model.BaseModel
UserID string `gorm:"size:50;index;column:user_id" json:"userId"`
ImageUrl string `gorm:"size:500;column:image_url" json:"imageUrl"`
Result string `gorm:"type:text;column:result" json:"result"`
LogID uint64 `gorm:"column:log_id;index" json:"logId"`
}
func (OcrLog) TableName() string { return "sundynix_plant_ocr_log" }
// ========== 积分商城 ========== // ========== 积分商城 ==========
// ExchangeItem status: 1=上架 2=下架 // ExchangeItem status: 1=上架 2=下架
@@ -391,3 +381,75 @@ type MediaCheckResult struct {
} }
func (MediaCheckResult) TableName() string { return "sundynix_plant_media_check_result" } func (MediaCheckResult) TableName() string { return "sundynix_plant_media_check_result" }
// ========== AI RAG 配置 ==========
type SysAiConfig struct {
model.BaseModel
IsActive int `gorm:"column:is_active;type:tinyint;default:0;comment:是否激活(1是0否)" json:"isActive"`
QdrantUrl string `gorm:"column:qdrant_url;type:varchar(255);comment:Qdrant接口地址" json:"qdrantUrl"`
QdrantApiKey string `gorm:"column:qdrant_api_key;type:varchar(255);comment:Qdrant密钥" json:"qdrantApiKey"`
QdrantCollection string `gorm:"column:qdrant_collection;type:varchar(100);comment:Qdrant集合名" json:"qdrantCollection"`
VectorDimension int `gorm:"column:vector_dimension;type:int;comment:向量维度" json:"vectorDimension"`
ChatProvider string `gorm:"column:chat_provider;type:varchar(50);comment:对话模型供应商" json:"chatProvider"`
ChatApiUrl string `gorm:"column:chat_api_url;type:varchar(255);comment:对话模型接口地址" json:"chatApiUrl"`
ChatApiKey string `gorm:"column:chat_api_key;type:varchar(255);comment:对话模型ApiKey" json:"chatApiKey"`
ChatModelName string `gorm:"column:chat_model_name;type:varchar(100);comment:对话模型名称" json:"chatModelName"`
EmbeddingProvider string `gorm:"column:embedding_provider;type:varchar(50);comment:Embedding模型供应商" json:"embeddingProvider"`
EmbeddingApiUrl string `gorm:"column:embedding_api_url;type:varchar(255);comment:Embedding模型接口地址" json:"embeddingApiUrl"`
EmbeddingApiKey string `gorm:"column:embedding_api_key;type:varchar(255);comment:Embedding模型ApiKey" json:"embeddingApiKey"`
EmbeddingModelName string `gorm:"column:embedding_model_name;type:varchar(100);comment:Embedding模型名称" json:"embeddingModelName"`
DailyQueryLimit int `gorm:"column:daily_query_limit;type:int;default:20;comment:每用户每日问答上限(0=不限)" json:"dailyQueryLimit"`
}
func (SysAiConfig) TableName() string { return "sundynix_plant_ai_config" }
// ========== 植物识别记录 ==========
type BaikeInfo struct {
BaikeUrl string `json:"baike_url"` // 百度百科链接
ImageUrl string `json:"image_url"` // 植物图片链接
Description string `json:"description"` // 植物百科描述文本
}
type ResultItem struct {
Score float64 `json:"score"` // 匹配相似度得分(0-1
Name string `json:"name"` // 植物名称
BaikeInfo *BaikeInfo `json:"baike_info"` // 植物百科信息
}
type ResultsArray []ResultItem
// Scan 实现 sql.Scanner 接口:JSON String -> Go Struct (读库)
func (r *ResultsArray) Scan(value interface{}) error {
if value == nil {
*r = make([]ResultItem, 0)
return nil
}
bytes, ok := value.([]byte)
if !ok {
if str, ok := value.(string); ok {
bytes = []byte(str)
} else {
return errors.New("type assertion to []byte/string failed")
}
}
return json.Unmarshal(bytes, r)
}
// Value 实现 driver.Valuer 接口:Go Struct -> JSON String (存库)
func (r ResultsArray) Value() (driver.Value, error) {
if len(r) == 0 {
return "[]", nil
}
return json.Marshal(r)
}
type ClassifyRecord struct {
model.BaseModel
UserID string `gorm:"size:50;index;column:user_id" json:"userId"`
LogID uint64 `gorm:"column:log_id;index" json:"logId"`
AllResults ResultsArray `gorm:"type:json;column:all_results" json:"allResults"`
}
func (ClassifyRecord) TableName() string { return "sundynix_plant_classify_record" }
-10
View File
@@ -10,13 +10,3 @@ Etcd:
DB: DB:
DataSource: root:root@tcp(192.168.100.127:3307)/sundynix_micro_go?charset=utf8mb4&parseTime=True&loc=Local DataSource: root:root@tcp(192.168.100.127:3307)/sundynix_micro_go?charset=utf8mb4&parseTime=True&loc=Local
Ai:
EmbeddingApiUrl:
EmbeddingApiKey:
EmbeddingModelName:
QdrantUrl:
QdrantApiKey:
QdrantCollection:
VectorDimension: 0
DailyQuota: 20
-10
View File
@@ -7,14 +7,4 @@ type Config struct {
DB struct { DB struct {
DataSource string DataSource string
} }
Ai struct {
EmbeddingApiUrl string
EmbeddingApiKey string
EmbeddingModelName string
QdrantUrl string
QdrantApiKey string
QdrantCollection string
VectorDimension int
DailyQuota int64
}
} }
@@ -57,6 +57,15 @@ func (l *AddCareRecordLogic) AddCareRecord(in *plant.AddCareRecordReq) (*plant.C
// 4. 生成下一期任务(以今天为基准,+period 天) // 4. 生成下一期任务(以今天为基准,+period 天)
nextDue := time.Now().Truncate(24*time.Hour).AddDate(0, 0, plan.Period) nextDue := time.Now().Truncate(24*time.Hour).AddDate(0, 0, plan.Period)
// 检查是否已经存在该计划的待办任务,避免重复生成相同事项的多个待办任务
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{ nextTask := plantModel.CareTask{
UserID: in.UserId, UserID: in.UserId,
PlantID: in.PlantId, PlantID: in.PlantId,
@@ -70,6 +79,7 @@ func (l *AddCareRecordLogic) AddCareRecord(in *plant.AddCareRecordReq) (*plant.C
if err := tx.Create(&nextTask).Error; err != nil { if err := tx.Create(&nextTask).Error; err != nil {
return err return err
} }
}
// 5. 更新用户 care_count 统计 // 5. 更新用户 care_count 统计
actionMap := map[string]string{ actionMap := map[string]string{
@@ -51,6 +51,15 @@ func (l *CompleteTaskLogic) CompleteTask(in *plant.CompleteTaskReq) (*plant.Task
today := time.Now() today := time.Now()
todayZero := time.Date(today.Year(), today.Month(), today.Day(), 0, 0, 0, 0, today.Location()) todayZero := time.Date(today.Year(), today.Month(), today.Day(), 0, 0, 0, 0, today.Location())
nextDue := todayZero.AddDate(0, 0, plan.Period) nextDue := todayZero.AddDate(0, 0, plan.Period)
// 检查是否已经存在该计划的待办任务,避免重复生成相同事项的多个待办任务
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{ newTask := plantModel.CareTask{
UserID: plan.UserID, PlantID: plan.PlantID, PlanID: plan.ID, UserID: plan.UserID, PlantID: plan.PlantID, PlanID: plan.ID,
Name: plan.Name, Icon: plan.Icon, TargetAction: plan.TargetAction, Name: plan.Name, Icon: plan.Icon, TargetAction: plan.TargetAction,
@@ -59,6 +68,7 @@ func (l *CompleteTaskLogic) CompleteTask(in *plant.CompleteTaskReq) (*plant.Task
if err := tx.Create(&newTask).Error; err != nil { if err := tx.Create(&newTask).Error; err != nil {
return err return err
} }
}
// 4. 保存养护记录 // 4. 保存养护记录
record := plantModel.CareRecord{ record := plantModel.CareRecord{
UserID: plan.UserID, PlantID: plan.PlantID, PlanID: plan.ID, UserID: plan.UserID, PlantID: plan.PlantID, PlanID: plan.ID,
@@ -19,7 +19,7 @@ func NewDeleteClassifyLogLogic(ctx context.Context, svcCtx *svc.ServiceContext)
} }
func (l *DeleteClassifyLogLogic) DeleteClassifyLog(in *plant.IdsReq) (*plant.CommonResp, error) { 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 nil, err
} }
return &plant.CommonResp{Code: 0, Msg: "ok"}, nil return &plant.CommonResp{Code: 0, Msg: "ok"}, nil
@@ -22,10 +22,14 @@ func NewDeleteWikiVectorLogic(ctx context.Context, svcCtx *svc.ServiceContext) *
} }
func (l *DeleteWikiVectorLogic) DeleteWikiVector(in *plant.SyncWikiVectorReq) (*plant.CommonResp, error) { 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") 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 return nil, err
} }
if err := l.svcCtx.DB.Model(&plantModel.Wiki{}).Where("id = ?", in.WikiId).Update("is_vector_synced", false).Error; err != nil { if err := l.svcCtx.DB.Model(&plantModel.Wiki{}).Where("id = ?", in.WikiId).Update("is_vector_synced", false).Error; err != nil {
@@ -26,10 +26,12 @@ func (l *GetAiChatQuotaLogic) GetAiChatQuota(in *plant.GetProfileReq) (*plant.Ai
l.svcCtx.DB.Model(&plantModel.AiChatHistory{}). l.svcCtx.DB.Model(&plantModel.AiChatHistory{}).
Where("user_id = ? AND created_at >= ?", in.UserId, todayStart). Where("user_id = ? AND created_at >= ?", in.UserId, todayStart).
Count(&used) Count(&used)
limit := l.svcCtx.Config.Ai.DailyQuota
if limit <= 0 { limit := int64(20)
limit = 20 if dbCfg, err := getActiveAiConfig(l.svcCtx.DB); err == nil && dbCfg.DailyQueryLimit > 0 {
limit = int64(dbCfg.DailyQueryLimit)
} }
remaining := limit - used remaining := limit - used
if remaining < 0 { if remaining < 0 {
remaining = 0 remaining = 0
@@ -2,6 +2,7 @@ package logic
import ( import (
"context" "context"
"encoding/json"
"github.com/zeromicro/go-zero/core/logx" "github.com/zeromicro/go-zero/core/logx"
plantModel "sundynix-micro-go/app/plant/model" plantModel "sundynix-micro-go/app/plant/model"
"sundynix-micro-go/app/plant/rpc/internal/svc" "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) { func (l *GetMyClassifyLogLogic) GetMyClassifyLog(in *plant.GetProfileReq) (*plant.ClassifyLogListResp, error) {
var logs []plantModel.OcrLog var records []plantModel.ClassifyRecord
if err := l.svcCtx.DB.Where("user_id = ?", in.UserId).Order("created_at desc").Limit(50).Find(&logs).Error; err != nil { if err := l.svcCtx.DB.Where("user_id = ?", in.UserId).Order("created_at desc").Limit(50).Find(&records).Error; err != nil {
return nil, err return nil, err
} }
list := make([]*plant.ClassifyLogInfo, 0, len(logs)) list := make([]*plant.ClassifyLogInfo, 0, len(records))
for _, og := range logs { 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{ 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"), CreatedAt: og.CreatedAt.Format("2006-01-02 15:04:05"),
}) })
} }
+98 -100
View File
@@ -1,24 +1,24 @@
package logic package logic
import ( import (
"bytes"
"context" "context"
"crypto/md5"
"encoding/hex"
"encoding/json"
"errors" "errors"
"fmt" "fmt"
"io"
"net/http"
"strings" "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" plantModel "sundynix-micro-go/app/plant/model"
"sundynix-micro-go/app/plant/rpc/internal/config"
) )
func wikiVectorID(wikiID string) string { func wikiVectorID(wikiID string) string {
sum := md5.Sum([]byte("sundynix-plant-wiki:" + wikiID)) return uuid.NewMD5(uuid.NameSpaceOID, []byte(wikiID)).String()
return hex.EncodeToString(sum[:])
} }
func buildWikiVectorText(w plantModel.Wiki) string { func buildWikiVectorText(w plantModel.Wiki) string {
@@ -30,131 +30,129 @@ func buildWikiVectorText(w plantModel.Wiki) string {
w.FloweringShape, w.FlowerDiameter, w.Fruit) w.FloweringShape, w.FlowerDiameter, w.Fruit)
} }
func embeddingModel(c config.Config) string { func getActiveAiConfig(db *gorm.DB) (*plantModel.SysAiConfig, error) {
if c.Ai.EmbeddingModelName != "" { var cfg plantModel.SysAiConfig
return c.Ai.EmbeddingModelName 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" return "text-embedding-3-small"
} }
func createEmbedding(ctx context.Context, c config.Config, text string) ([]float32, error) { func createEmbedding(ctx context.Context, cfg *plantModel.SysAiConfig, text string) ([]float32, error) {
body, _ := json.Marshal(map[string]interface{}{ config := openai.DefaultConfig(cfg.EmbeddingApiKey)
"model": embeddingModel(c), if cfg.EmbeddingApiUrl != "" {
"input": text, 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 { if err != nil {
return nil, err return nil, err
} }
req.Header.Set("Content-Type", "application/json") if len(resp.Data) == 0 || len(resp.Data[0].Embedding) == 0 {
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 {
return nil, errors.New("embedding 响应为空") 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 { func newQdrantConn(cfg *plantModel.SysAiConfig) (*grpc.ClientConn, context.Context, error) {
return strings.TrimRight(c.Ai.QdrantUrl, "/") + path addr := strings.TrimPrefix(cfg.QdrantUrl, "http://")
} addr = strings.TrimPrefix(addr, "https://")
conn, err := grpc.NewClient(addr, grpc.WithTransportCredentials(insecure.NewCredentials()))
func doQdrant(ctx context.Context, c config.Config, method, path string, body interface{}) error { if err != nil {
var reader io.Reader return nil, nil, fmt.Errorf("qdrant grpc dial failed: %w", err)
if body != nil {
raw, _ := json.Marshal(body)
reader = bytes.NewReader(raw)
} }
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 { if err != nil {
return err return err
} }
req.Header.Set("Content-Type", "application/json") defer conn.Close()
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
}
}
if dim <= 0 { if dim <= 0 {
dim = c.Ai.VectorDimension dim = cfg.VectorDimension
} }
if dim <= 0 { if dim <= 0 {
dim = 1536 dim = 1536
} }
return doQdrant(ctx, c, http.MethodPut, "/collections/"+c.Ai.QdrantCollection, map[string]interface{}{ collClient := qdrant.NewCollectionsClient(conn)
"vectors": map[string]interface{}{ if _, getErr := collClient.Get(ctx, &qdrant.GetCollectionInfoRequest{CollectionName: cfg.QdrantCollection}); getErr == nil {
"size": dim, return nil
"distance": "Cosine", }
_, 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) text := buildWikiVectorText(w)
vector, err := createEmbedding(ctx, c, text) vector, err := createEmbedding(ctx, cfg, text)
if err != nil { if err != nil {
return err return err
} }
if err := ensureQdrantCollection(ctx, c, len(vector)); err != nil { if err := ensureQdrantCollection(cfg, len(vector)); err != nil {
return err return err
} }
return doQdrant(ctx, c, http.MethodPut, "/collections/"+c.Ai.QdrantCollection+"/points?wait=true", map[string]interface{}{ conn, qdCtx, err := newQdrantConn(cfg)
"points": []map[string]interface{}{ if err != nil {
{ return err
"id": wikiVectorID(w.ID), }
"vector": vector, defer conn.Close()
"payload": map[string]interface{}{ ptsClient := qdrant.NewPointsClient(conn)
"wiki_id": w.ID,
"name": w.Name, _, err = ptsClient.Upsert(qdCtx, &qdrant.UpsertPoints{
"full_text": text, 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, c config.Config, wikiID string) error { func deleteWikiVector(ctx context.Context, cfg *plantModel.SysAiConfig, wikiID string) error {
return doQdrant(ctx, c, http.MethodPost, "/collections/"+c.Ai.QdrantCollection+"/points/delete?wait=true", map[string]interface{}{ conn, qdCtx, err := newQdrantConn(cfg)
"points": []string{wikiVectorID(wikiID)}, 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))},
},
},
},
}) })
return err
} }
@@ -23,7 +23,11 @@ func NewSyncAllWikiVectorLogic(ctx context.Context, svcCtx *svc.ServiceContext)
} }
func (l *SyncAllWikiVectorLogic) SyncAllWikiVector(in *plant.PageReq) (*plant.CommonResp, error) { 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") return nil, errors.New("AI/RAG 未配置 EmbeddingApiUrl、QdrantUrl 或 QdrantCollection")
} }
var wikis []plantModel.Wiki var wikis []plantModel.Wiki
@@ -32,7 +36,7 @@ func (l *SyncAllWikiVectorLogic) SyncAllWikiVector(in *plant.PageReq) (*plant.Co
} }
success := 0 success := 0
for _, wiki := range wikis { 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) l.Logger.Errorf("sync wiki vector failed, wiki_id=%s, err=%v", wiki.ID, err)
continue continue
} }
@@ -25,14 +25,18 @@ func (l *SyncWikiVectorLogic) SyncWikiVector(in *plant.SyncWikiVectorReq) (*plan
if in.WikiId == "" { if in.WikiId == "" {
return nil, errors.New("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") return nil, errors.New("AI/RAG 未配置 EmbeddingApiUrl、QdrantUrl 或 QdrantCollection")
} }
var wiki plantModel.Wiki var wiki plantModel.Wiki
if err := l.svcCtx.DB.Where("id = ?", in.WikiId).First(&wiki).Error; err != nil { if err := l.svcCtx.DB.Where("id = ?", in.WikiId).First(&wiki).Error; err != nil {
return nil, err 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 return nil, err
} }
if err := l.svcCtx.DB.Model(&plantModel.Wiki{}).Where("id = ?", in.WikiId).Update("is_vector_synced", true).Error; err != nil { if err := l.svcCtx.DB.Model(&plantModel.Wiki{}).Where("id = ?", in.WikiId).Update("is_vector_synced", true).Error; err != nil {
+2 -1
View File
@@ -40,7 +40,6 @@ func NewServiceContext(c config.Config) *ServiceContext {
&plantModel.PostLike{}, &plantModel.PostLike{},
&plantModel.PostOss{}, &plantModel.PostOss{},
&plantModel.Topic{}, &plantModel.Topic{},
&plantModel.OcrLog{},
&plantModel.MediaCheckResult{}, &plantModel.MediaCheckResult{},
&plantModel.ExchangeItem{}, &plantModel.ExchangeItem{},
&plantModel.ExchangeOrder{}, &plantModel.ExchangeOrder{},
@@ -50,6 +49,8 @@ func NewServiceContext(c config.Config) *ServiceContext {
&plantModel.AiChatHistory{}, &plantModel.AiChatHistory{},
&plantModel.GrowthRecordOss{}, &plantModel.GrowthRecordOss{},
&plantModel.Banner{}, &plantModel.Banner{},
&plantModel.SysAiConfig{},
&plantModel.ClassifyRecord{},
); err != nil { ); err != nil {
logx.Errorf("数据库迁移失败: %v", err) logx.Errorf("数据库迁移失败: %v", err)
} }
+4 -1
View File
@@ -18,7 +18,10 @@ type BaseModel struct {
// BeforeCreate 创建前自动生成雪花ID // BeforeCreate 创建前自动生成雪花ID
func (m *BaseModel) BeforeCreate(db *gorm.DB) (err error) { func (m *BaseModel) BeforeCreate(db *gorm.DB) (err error) {
db.Statement.SetColumn("id", uniqueid.GenerateID()) if m.ID == "" {
m.ID = uniqueid.GenerateID()
}
db.Statement.SetColumn("id", m.ID)
return return
} }
BIN
View File
Binary file not shown.