Fix: raptor don't have attribute chat (#11936)
### 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:
parent
948bc93786
commit
2b260901df
1 changed files with 1 additions and 1 deletions
|
|
@ -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)
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue