g3fiIdZddlmZddlZddlZddlmZmZddlm Z erddl m Z ddZ ddZ d d dd Zejd ej Zed  ddZdZed  ddZddZy)zUtilities for JSON.) annotationsN) TYPE_CHECKINGAny)OutputParserException)Callablec(|jd}tjdd|}tjdd|}tjdd|}tjdd |}|jd |z|jd zS) zReplace newline characters in a regex match with escaped sequences. Args: match: Regex match object containing the string to process. Returns: String with newlines, carriage returns, tabs, and quotes properly escaped. \nz\\nz\rz\\rz\tz\\tz(?E !EKKN 22ct|ttfr|j}t j dt |tjS)atCustom parser for multiline strings. The LLM response for `action_input` may be a multiline string containing unescaped newlines, tabs or quotes. This function replaces those characters with their escaped counterparts. (newlines in JSON must be double-escaped: `\\n`). Returns: The modified string with escaped newlines, tabs and quotes. z("action_input"\:\s*")(.*?)("))flags) isinstancebytes bytearraydecoderrrDOTALL)multiline_strings r_custom_parserr!sD"UI$67+224 66)ii  rFstrictc tj||S#tj$rYnwxYwg}g}d}d}|D]}|}|r |dk(r|sd}no|dk(r|sd}ne|dk(r| }n\d}nY|dk(rd}d}nO|dk(r|jd n8|d k(r|jd n!|d vr|r|d |k(r|j ny|j||r#|r|j |jd|j |rV tjdj ||z|S#tj$r|j YnwxYw|rVtj||S)zParse a JSON string that may be missing closing braces. Args: s: The JSON string to parse. strict: Whether to use strict parsing. Returns: The parsed JSON object as a Python dictionary. rF" r \T{}[]>r'r%N)jsonloadsJSONDecodeErrorappendpopreversejoin)sr new_charsstackis_inside_stringescapedcharnew_chars rparse_partial_jsonr8;s zz!F++      I EG# s{7#( g%+ S[# G S[ LL  S[ LL  Z rd*  "9#@  MMO MMO  ::bggi%&78H H##  MMO   ::a ''s//(D00#EEz```(json)?(.*)parserc t||S#tj$r-tj |}||n|j d}YnwxYwt||S)zParse a JSON string from a Markdown string. Args: json_string: The Markdown string. parser: The parser to use. Defaults to `parse_partial_json`. Returns: The parsed JSON object as a Python dictionary. r9r ) _parse_jsonr*r,_json_markdown_researchr ) json_stringr:rjson_strs rparse_json_markdownrAs`D;v66   D!((5#(-;U[[^ D x //s =AAz `cR|jt}t|}||S)aParse a JSON string, handling special characters and whitespace. Strips whitespace, newlines, and backticks from the start and end of the string, then processes special characters before parsing. Args: json_str: The JSON string to parse. parser: Optional custom parser function. Defaults to `parse_partial_json`. Returns: Parsed JSON object. )strip_json_strip_charsr)r@r:s rr<r<s,$~~/0Hh'H ( rc t|}t |t s&dt |jd}t|||D]}||vsd|d|}t||S#tj$r}d|}t||d}~wwxYw)aParse and check a JSON string from a Markdown string. Checks that it contains the expected keys. Args: text: The Markdown string. expected_keys: The expected keys in the JSON string. Returns: The parsed JSON object as a Python dictionary. Raises: OutputParserException: If the JSON string is invalid or does not contain the expected keys. z Got invalid JSON object. Error: Nz&Expected JSON object (dict), but got: z. ) llm_outputz)Got invalid return object. Expected key `z` to be present, but got )rAr*r,rrdicttype__name__)text expected_keysjson_objemsg error_messagekeys rparse_and_check_json_markdownrQs 0&t, h %4T(^5L5L4MR P $MdCC- h ;C5A**25 (, , - O!   004#C(a/0s A##B 6BB )rz re.Match[str]returnstr)rzstr | bytes | bytearrayrRrS)r1rSrboolrRr)r?rSr:Callable[[str], Any]rRr)r@rSr:rUrRr)rJrSrKz list[str]rRrG)__doc__ __future__rr*rtypingrrlangchain_core.exceptionsrcollections.abcrrrr8compilerr=rArDr<rQrrr]s" %;(3$427N(bBJJ0"))<9K00!50006H26"r