pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2025-12-19 20:16:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix: raptor don't have attribute chat (#11936)

Browse Source 
### What problem does this PR solve?

Raptor don't have attribute chat.

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Yongteng Lei
2025-12-12 20:08:18 +08:00
committed by GitHub
parent 948bc93786
commit 2b260901df
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
rag/raptor.py
View File

@ -63,7 +63,7 @@ class RecursiveAbstractiveProcessing4TreeOrganizedRetrieval:
last_exc = None last_exc = None
for attempt in range(3): for attempt in range(3):
try: try:
response = await asyncio.to_thread(self._llm_model.chat, system, history, gen_conf) response = await self._llm_model.async_chat(system, history, gen_conf)
response = re.sub(r"^.*</think>", "", response, flags=re.DOTALL) response = re.sub(r"^.*</think>", "", response, flags=re.DOTALL)
if response.find("**ERROR**") >= 0: if response.find("**ERROR**") >= 0:
raise Exception(response) raise Exception(response)