zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-08-18 14:28:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: unable to search agent log by ID (#18373)

Browse Source 
### Summary

As title
This commit is contained in:
Haruko386
2026-08-17 18:42:45 +08:00
committed by GitHub
parent 68da250f8f
commit 5a5f600ae3
2 changed files with 74 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
internal/dao/chat_session.go
View File

@@ -221,10 +221,17 @@ func (dao *ChatSessionDAO) ListAgentSessions(ctx context.Context, db *gorm.DB, p
if params.Keywords != "" {
keywords := strings.ToLower(params.Keywords)
escapedKeywords := strings.Trim(strconv.QuoteToASCII(keywords), `"`)
keywordPattern := "%" + keywords + "%"
if escapedKeywords == keywords {
query = query.Where("LOWER(message) LIKE ?", "%"+keywords+"%")
query = query.Where("(LOWER(id) LIKE ? OR LOWER(name) LIKE ? OR LOWER(message) LIKE ?)", keywordPattern, keywordPattern, keywordPattern)
} else {
query = query.Where("(LOWER(message) LIKE ? OR LOWER(message) LIKE ?)", "%"+keywords+"%", "%"+escapedKeywords+"%")
query = query.Where(
"(LOWER(id) LIKE ? OR LOWER(name) LIKE ? OR LOWER(message) LIKE ? OR LOWER(message) LIKE ?)",
keywordPattern,
keywordPattern,
keywordPattern,
"%"+escapedKeywords+"%",
)
}
}

65
internal/dao/chat_session_test.go
View File

@@ -67,6 +67,29 @@ func createAgentSessionForDAOTest(t *testing.T, db *gorm.DB, id, agentID, userID
}
}
func createNamedAgentSessionForDAOTest(t *testing.T, db *gorm.DB, id, agentID, userID, name string, message json.RawMessage, updateTime int64) {
t.Helper()
updateDate := time.UnixMilli(updateTime).Local()
session := &entity.API4Conversation{
ID: id,
Name: &name,
DialogID: agentID,
UserID: userID,
Message: message,
Reference: json.RawMessage(`[]`),
BaseModel: entity.BaseModel{
CreateTime: &updateTime,
CreateDate: &updateDate,
UpdateTime: &updateTime,
UpdateDate: &updateDate,
},
}
if err := db.Create(session).Error; err != nil {
t.Fatalf("failed to create session %s: %v", id, err)
}
}
func createChatSessionForDAOTest(t *testing.T, db *gorm.DB, id, chatID, name string, updateTime int64) {
t.Helper()
@@ -205,3 +228,45 @@ func TestChatSessionDAOListAgentSessionsFiltersAndPaginates(t *testing.T) {
t.Fatalf("expected user-1, got %s", sessions[0].UserID)
}
}
func TestChatSessionDAOListAgentSessionsSearchesIDNameAndMessage(t *testing.T) {
db := setupChatSessionDAOTestDB(t)
createNamedAgentSessionForDAOTest(t, db, "release-session-id", "agent-1", "user-1", "plain", json.RawMessage(`[{"content":"ordinary"}]`), 1000)
createNamedAgentSessionForDAOTest(t, db, "session-title", "agent-1", "user-1", "Release Notes", json.RawMessage(`[{"content":"ordinary"}]`), 2000)
createNamedAgentSessionForDAOTest(t, db, "session-message", "agent-1", "user-1", "plain", json.RawMessage(`[{"content":"release details"}]`), 3000)
createNamedAgentSessionForDAOTest(t, db, "other-agent-release", "agent-2", "user-1", "Release Notes", json.RawMessage(`[{"content":"release details"}]`), 4000)
ctx := t.Context()
total, sessions, err := NewChatSessionDAO().ListAgentSessions(ctx, db, ListAgentSessionsParams{
AgentID: "agent-1",
Keywords: "release",
Page: 1,
PageSize: 10,
OrderBy: "update_time",
Desc: false,
})
if err != nil {
t.Fatalf("ListAgentSessions failed: %v", err)
}
if total != 3 {
t.Fatalf("expected total 3, got %d", total)
}
gotIDs := make([]string, 0, len(sessions))
for _, session := range sessions {
gotIDs = append(gotIDs, session.ID)
if session.DialogID != "agent-1" {
t.Fatalf("session %s leaked from agent %s", session.ID, session.DialogID)
}
}
wantIDs := []string{"release-session-id", "session-title", "session-message"}
if len(gotIDs) != len(wantIDs) {
t.Fatalf("ids = %v, want %v", gotIDs, wantIDs)
}
for i := range wantIDs {
if gotIDs[i] != wantIDs[i] {
t.Fatalf("ids = %v, want %v", gotIDs, wantIDs)
}
}
}