Selaa lähdekoodia

change package name

Roc 1 päivä sitten
vanhempi
commit
a9940a3cb9

+ 1 - 1
controllers/rag/abstract.go → controllers/llm/abstract.go

@@ -1,4 +1,4 @@
-package rag
+package llm
 
 import (
 	"encoding/json"

+ 1 - 1
controllers/rag/chat_ws_controller.go → controllers/llm/chat_ws_controller.go

@@ -1,4 +1,4 @@
-package rag
+package llm
 
 import (
 	"eta/eta_api/controllers"

+ 1 - 2
controllers/rag/kb_controller.go → controllers/llm/kb_controller.go

@@ -1,4 +1,4 @@
-package rag
+package llm
 
 import (
 	"encoding/json"
@@ -49,4 +49,3 @@ func (kbctrl *KbController) SearchDocs() {
 	br.Success = true
 	br.Msg = "获取成功"
 }
-

+ 20 - 0
controllers/llm/llm_http/request.go

@@ -0,0 +1,20 @@
+package llm_http
+
+type LLMQuestionReq struct {
+	Question      string `description:"提问"`
+	KnowledgeBase string `description:"知识库"`
+	SessionId     string `description:"会话ID"`
+}
+
+type UserChatReq struct {
+	ChatId    int    `json:"ChatId"`
+	ChatTitle string `json:"ChatTitle" description:"会话名称"`
+}
+
+type UserChatRecordReq struct {
+	Id           int    `json:"Id"`
+	ChatId       int    `json:"ChatId"`
+	Content      string `json:"Content" description:"会话名称"`
+	ChatUserType string `json:"ChatUserType" description:"用户类型"`
+	SendTime     string `json:"SendTime" description:"发送时间"`
+}

+ 13 - 0
controllers/llm/llm_http/response.go

@@ -0,0 +1,13 @@
+package llm_http
+
+import "eta/eta_api/models/llm"
+
+type UserChatListResp struct {
+	TodayList     []llm.UserLlmChatListViewItem
+	YesterdayList []llm.UserLlmChatListViewItem
+	WeekList      []llm.UserLlmChatListViewItem
+}
+type UserChatResp struct {
+	ChatId    int
+	ChatTitle string
+}

+ 1 - 1
controllers/rag/question.go → controllers/llm/question.go

@@ -1,4 +1,4 @@
-package rag
+package llm
 
 import (
 	"encoding/json"

+ 2 - 2
controllers/rag/user_chat_controller.go → controllers/llm/user_chat_controller.go

@@ -1,9 +1,9 @@
-package rag
+package llm
 
 import (
 	"encoding/json"
 	"eta/eta_api/controllers"
-	"eta/eta_api/controllers/rag/llm_http"
+	"eta/eta_api/controllers/llm/llm_http"
 	"eta/eta_api/models"
 	"eta/eta_api/models/llm"
 	llmService "eta/eta_api/services/llm"

+ 1 - 1
controllers/rag/wechat_platform.go → controllers/llm/wechat_platform.go

@@ -1,4 +1,4 @@
-package rag
+package llm
 
 import (
 	"encoding/json"

+ 7 - 7
routers/router.go

@@ -30,8 +30,8 @@ import (
 	"eta/eta_api/controllers/eta_forum"
 	"eta/eta_api/controllers/eta_trial"
 	"eta/eta_api/controllers/fe_calendar"
+	"eta/eta_api/controllers/llm"
 	"eta/eta_api/controllers/material"
-	"eta/eta_api/controllers/rag"
 	"eta/eta_api/controllers/report_approve"
 	"eta/eta_api/controllers/residual_analysis"
 	"eta/eta_api/controllers/roadshow"
@@ -69,12 +69,12 @@ func init() {
 		),
 		web.NSNamespace("/llm",
 			web.NSInclude(
-				&rag.ChatWsController{},
-				&rag.UserChatController{},
-				&rag.KbController{},
-				&rag.WechatPlatformController{},
-				&rag.QuestionController{},
-				&rag.AbstractController{},
+				&llm.ChatWsController{},
+				&llm.UserChatController{},
+				&llm.KbController{},
+				&llm.WechatPlatformController{},
+				&llm.QuestionController{},
+				&llm.AbstractController{},
 			),
 		),
 		web.NSNamespace("/banner",