|
@@ -126,7 +126,7 @@ func (manager *ConnectionManager) HandleMessage(userID int, sessionID string, me
|
|
|
var resp *http.Response
|
|
|
labels := llm.GetDAFHandlerInstance().FindTextTagLabels(userMessage.Query)
|
|
|
if len(labels) > 0 {
|
|
|
- articles, findErr := rag.GetArticleByTags(labels, 15, 5)
|
|
|
+ articles, findErr := rag.GetArticleByTags(labels, 15, 10)
|
|
|
if findErr != nil {
|
|
|
utils.FileLog.Warn("没有搜索到相关的研报内容,执行RAG对话 ")
|
|
|
resp, err = llmService.KnowledgeBaseChat(userMessage.Query, userMessage.KbName, session.History)
|