Archives/langchain
2
0
Fork 1
mirror of https://github.com/hwchase17/langchain synced 2024-11-04 06:00:26 +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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 0 deletions
Show Stats 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")