VectorSearch 向量搜索
(c *gin.Context)
| 48 | |
| 49 | // VectorSearch 向量搜索 |
| 50 | func VectorSearch(c *gin.Context) { |
| 51 | eid := config.GetEID(c) |
| 52 | |
| 53 | // 解析请求体 |
| 54 | var req rag.SearchRequest |
| 55 | if err := c.ShouldBindJSON(&req); err != nil { |
| 56 | c.JSON(http.StatusBadRequest, gin.H{ |
| 57 | "success": false, |
| 58 | "message": "请求参数错误", |
| 59 | "error": err.Error(), |
| 60 | }) |
| 61 | return |
| 62 | } |
| 63 | |
| 64 | // 强制设置为向量搜索 |
| 65 | req.SearchType = "vector" |
| 66 | |
| 67 | // 创建搜索服务 |
| 68 | searchService := rag.NewSearchService(model.DB) |
| 69 | |
| 70 | // 执行搜索 |
| 71 | response, err := searchService.Search(eid, &req, nil) |
| 72 | if err != nil { |
| 73 | c.JSON(http.StatusInternalServerError, gin.H{ |
| 74 | "success": false, |
| 75 | "message": "向量搜索失败", |
| 76 | "error": err.Error(), |
| 77 | }) |
| 78 | return |
| 79 | } |
| 80 | |
| 81 | c.JSON(http.StatusOK, gin.H{ |
| 82 | "success": true, |
| 83 | "message": "向量搜索成功", |
| 84 | "data": response, |
| 85 | }) |
| 86 | } |
| 87 | |
| 88 | // FulltextSearch 全文搜索 |
| 89 | func FulltextSearch(c *gin.Context) { |