Async support for OpenAIFunctionsAgentOutputParser (#11140)

pull/11113/head^2
William FH 11 months ago committed by GitHub
parent ce08f436db
commit 8ae9b71e41
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -1,3 +1,4 @@
import asyncio
import json
from json import JSONDecodeError
from typing import List, Union
@ -80,5 +81,12 @@ class OpenAIFunctionsAgentOutputParser(AgentOutputParser):
message = result[0].message
return self._parse_ai_message(message)
async def aparse_result(
self, result: List[Generation]
) -> Union[AgentAction, AgentFinish]:
return await asyncio.get_running_loop().run_in_executor(
None, self.parse_result, result
)
def parse(self, text: str) -> Union[AgentAction, AgentFinish]:
raise ValueError("Can only parse messages")

Loading…
Cancel
Save