github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-05 21:12:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

Implement str one

Browse Source
This commit is contained in:
Nuno Campos
2023-09-28 20:20:47 +01:00
parent f672b39cc9
commit 63f2ef8d1c
1 changed files with 29 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
libs/langchain/langchain/output_parsers/json.py
View File

@@ -172,15 +172,12 @@ class SimpleJsonOutputParser(BaseOutputParser[Any]):
return "simple_json_output_parser" return "simple_json_output_parser"
class PartialJsonOutputParser(BaseCumulativeTransformOutputParser[Any]): class PartialFunctionsJsonOutputParser(BaseCumulativeTransformOutputParser[Any]):
from_function_args: bool = False
@property @property
def _type(self) -> str: def _type(self) -> str:
return "partial_json" return "partial_functions_json"
def parse_result(self, result: List[Generation]) -> Any: def parse_result(self, result: List[Generation]) -> Any:
if self.from_function_args:
if len(result) != 1: if len(result) != 1:
raise OutputParserException( raise OutputParserException(
f"Expected exactly one result, but got {len(result)}" f"Expected exactly one result, but got {len(result)}"
@@ -202,3 +199,12 @@ class PartialJsonOutputParser(BaseCumulativeTransformOutputParser[Any]):
def parse(self, text: str) -> Any: def parse(self, text: str) -> Any:
pass pass
class PartialJsonOutputParser(BaseCumulativeTransformOutputParser[Any]):
@property
def _type(self) -> str:
return "partial_functions_json"
def parse(self, text: str) -> Any:
return parse_json_markdown(text)