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

Append attachments content to last message (#17993)

Browse Source
This commit is contained in:
Wang Qi
2026-08-07 16:04:19 +08:00
committed by GitHub
parent 3f78c156d4
commit 993b41b7b1
5 changed files with 9 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
test/unit_test/rag/advanced_rag/test_agentic_rag.py
View File

@@ -1,5 +1,3 @@
from copy import deepcopy
import pytest
from rag.advanced_rag.agentic_rag import RAGTools
@@ -13,11 +11,10 @@ class FakeChatModel:
@pytest.mark.asyncio
async def test_rag_tool_adds_text_attachment_as_evidence(monkeypatch):
async def test_rag_tool_adds_text_attachment_to_user_question(monkeypatch):
captured = {}
async def fake_run_agentic_rag(tools, messages):
captured["kbinfos"] = deepcopy(tools.kbinfos)
captured["messages"] = messages
yield "answer"
@@ -26,6 +23,4 @@ async def test_rag_tool_adds_text_attachment_as_evidence(monkeypatch):
tools = RAGTools([], FakeChatModel(), text_attachments_content="attached facts")
assert await tools.rag("What is attached?") == "answer"
assert captured["messages"] == [{"role": "user", "content": "What is attached?"}]
assert captured["kbinfos"]["chunks"][0]["docnm_kwd"] == "Chat attachment"
assert captured["kbinfos"]["chunks"][0]["content_with_weight"] == "attached facts"
assert captured["messages"] == [{"role": "user", "content": "What is attached?attached facts"}]