This commit is contained in:
stellaHSR 2023-07-22 09:49:49 +08:00
parent 351fd84bc6
commit e32f9e76ed

View file

@ -46,7 +46,7 @@ class Action(ABC):
system_msgs.append(self.prefix)
return await self.llm.aask(prompt, system_msgs)
@retry(stop=stop_after_attempt(1), wait=wait_fixed(1))
@retry(stop=stop_after_attempt(2), wait=wait_fixed(1))
async def _aask_v1(self, prompt: str, output_class_name: str,
output_data_mapping: dict,
system_msgs: Optional[list[str]] = None) -> ActionOutput: