github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-25 08:03:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

Backwards compat

Browse Source
This commit is contained in:
Nuno Campos 2023-09-29 14:18:38 +01:00
parent 4e28a7a513
commit 091d8845d5
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -3,7 +3,7 @@ from __future__ import annotations
import json
import re
from json import JSONDecodeError
from typing import Any, List, Optional
from typing import Any, Callable, List, Optional
import jsonpatch
@ -103,7 +103,9 @@ def parse_partial_json(s: str) -> Any:
return None
def parse_json_markdown(json_string: str) -> dict:
def parse_json_markdown(
json_string: str, parser: Callable[[str], Any] = json.loads
) -> dict:
"""
Parse a JSON string from a Markdown string.
@ -130,7 +132,7 @@ def parse_json_markdown(json_string: str) -> dict:
json_str = _custom_parser(json_str)
# Parse the JSON string into a Python dictionary
parsed = parse_partial_json(json_str)
parsed = parser(json_str)
return parsed
@ -216,4 +218,4 @@ class PartialJsonOutputParser(BaseCumulativeTransformOutputParser[Any]):
return jsonpatch.make_patch(prev, next).patch
def parse(self, text: str) -> Any:
return parse_json_markdown(text)
return parse_json_markdown(text, parse_partial_json)