gmakstutis/ragflow
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix: raptor don't have chat

Browse source
This commit is contained in:
yongtenglei 2025-12-12 19:39:07 +08:00
parent 948bc93786
commit ba5cb108d2
1 changed files with 1 additions and 1 deletions
Show stats 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)