From c173a699083c5063a9c2bafdf661e3290633a2ec Mon Sep 17 00:00:00 2001 From: Bagatur <22008038+baskaryan@users.noreply.github.com> Date: Wed, 24 Jan 2024 16:57:16 -0800 Subject: [PATCH] langchain[patch]: oai tools output parser nit (#16540) allow positional init args --- libs/langchain/langchain/output_parsers/openai_tools.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/libs/langchain/langchain/output_parsers/openai_tools.py b/libs/langchain/langchain/output_parsers/openai_tools.py index bea4b12a3c2..39fd05974a9 100644 --- a/libs/langchain/langchain/output_parsers/openai_tools.py +++ b/libs/langchain/langchain/output_parsers/openai_tools.py @@ -72,6 +72,10 @@ class JsonOutputKeyToolsParser(JsonOutputToolsParser): return_single: bool = False """Whether to return only the first tool call.""" + def __init__(self, key_name: str, **kwargs: Any) -> None: + """Allow init with positional args.""" + super().__init__(key_name=key_name, **kwargs) + def parse_result(self, result: List[Generation], *, partial: bool = False) -> Any: results = super().parse_result(result) results = [res for res in results if res["type"] == self.key_name]