feat: 百科知识库存入向量

router/plant/ai_chat_router.go

@@ -0,0 +1,15 @@
+package plant
+
+import "github.com/gin-gonic/gin"
+
+type AiChatRouter struct{}
+
+func (s *AiChatRouter) InitAiChatRouter(Router *gin.RouterGroup) (R gin.IRoutes) {
+	aiChatRouter := Router.Group("plant/chat")
+	aiChatApi := aiChatApi
+	{
+		aiChatRouter.GET("stream", aiChatApi.ChatStreamPlant) // SSE 对话流
+		aiChatRouter.POST("sync", aiChatApi.SyncWikiToQdrant) // 后台知识库同步
+	}
+	return aiChatRouter
+}

router/plant/enter.go

@@ -14,6 +14,7 @@ type RouterGroup struct {
 	UserProfileRouter
 	CallbackRouter
 	ExchangeRouter
+	AiChatRouter
 }
 
 // 初始化路由
@@ -29,4 +30,5 @@ var (
 	badgeConfigApi = v1.ApiGroupApp.PlantApiGroup.BadgeConfigApi
 	callbackApi    = v1.ApiGroupApp.PlantApiGroup.CallbackApi
 	exchangeApi    = v1.ApiGroupApp.PlantApiGroup.ExchangeApi
+	aiChatApi      = v1.ApiGroupApp.PlantApiGroup.AiChatApi
 )

router/plant/wiki_router.go

@@ -13,6 +13,8 @@ func (p *WikiRouter) InitWikiRouter(Router *gin.RouterGroup) {
 		wikiRouter.GET("/detail", wikiApi.WikiDetail)
 		wikiRouter.POST("/delete", wikiApi.DeleteWiki)
 		wikiRouter.POST("/uploadImg", wikiApi.UploadImg)
+		wikiRouter.POST("/sync-qdrant", wikiApi.SyncWikiQdrant)
+		wikiRouter.POST("/delete-qdrant", wikiApi.DeleteWikiQdrant)
 
 		//用户端
 		wikiRouter.GET("/star", wikiApi.StarWiki) //收藏或者取消收藏

router/system/enter.go

@@ -10,6 +10,7 @@ type SysRouterGroup struct {
 	MenuRouter
 	OperationRecordRouter
 	OssRouter
+	SysAiConfigRouter
 }
 
 // 初始化路由
@@ -21,4 +22,5 @@ var (
 	menuApi            = v1.ApiGroupApp.SystemApiGroup.MenuApi
 	operationRecordApi = v1.ApiGroupApp.SystemApiGroup.OperationRecordApi
 	ossApi             = v1.ApiGroupApp.SystemApiGroup.OssApi
+	sysAiConfigApi     = v1.ApiGroupApp.SystemApiGroup.SysAiConfigApi
 )

router/system/sys_ai_config_router.go

@@ -0,0 +1,17 @@
+package system
+
+import "github.com/gin-gonic/gin"
+
+type SysAiConfigRouter struct{}
+
+func (s *SysAiConfigRouter) InitSysAiConfigRouter(Router *gin.RouterGroup) (R gin.IRoutes) {
+	sysAiConfigRouter := Router.Group("aiConfig")
+	sysAiConfigApi := sysAiConfigApi
+	{
+		sysAiConfigRouter.POST("create", sysAiConfigApi.CreateAiConfig) // 创建配置
+		sysAiConfigRouter.PUT("update", sysAiConfigApi.UpdateAiConfig)  // 更新配置
+		sysAiConfigRouter.POST("setActive", sysAiConfigApi.SetActive)   // 设置激活状态
+		sysAiConfigRouter.GET("list", sysAiConfigApi.GetList)           // 获取列表
+	}
+	return sysAiConfigRouter
+}