Fix: raptor don't have chat

This commit is contained in:
yongtenglei 2025-12-12 19:39:07 +08:00
parent 948bc93786
commit ba5cb108d2

View file

@ -63,7 +63,7 @@ class RecursiveAbstractiveProcessing4TreeOrganizedRetrieval:
last_exc = None
for attempt in range(3):
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)
if response.find("**ERROR**") >= 0:
raise Exception(response)