github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-26 13:59:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Async support for OpenAIFunctionsAgentOutputParser (#11140)

Browse Source
This commit is contained in:
William FH
2023-09-28 06:42:59 -07:00
committed by GitHub
parent ce08f436db
commit 8ae9b71e41
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
libs/langchain/langchain/agents/output_parsers/openai_functions.py
View File

@@ -1,3 +1,4 @@
import asyncio
import json import json
from json import JSONDecodeError from json import JSONDecodeError
from typing import List, Union from typing import List, Union
@@ -80,5 +81,12 @@ class OpenAIFunctionsAgentOutputParser(AgentOutputParser):
message = result[0].message message = result[0].message
return self._parse_ai_message(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]: def parse(self, text: str) -> Union[AgentAction, AgentFinish]:
raise ValueError("Can only parse messages") raise ValueError("Can only parse messages")