D_iISrSSKJr SSKrSSKrSSKJrJr SSKJ r \(aSSK J r SSjr SSjr S S .SS jjr\R"S \R 5r\S .SSjjrSr\S .SSjjrSSjrg)zUtilities for JSON.) annotationsN) TYPE_CHECKINGAny)OutputParserException)Callablec0URS5n[R"SSU5n[R"SSU5n[R"SSU5n[R"SS U5nURS 5U-URS 5-$) 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 22c[U[[45(aUR5n[R "S[ U[RS9$)aXCustom 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!sG"UI$677+224 66)ii  rFstrictcH[R"XS9$![Ra Of=f/n/nSnSnUHnUnU(a1US:Xa U(dSnOUS:Xa U(dSnOuUS:XaU(+nOgSnOdUS:XaSnSnOYUS:XaURS 5 OAUS :XaURS 5 O)US ;a#U(aUS U:XaUR 5 O gURU5 M U(a(U(aUR 5 URS5 UR 5 U(aZ[R"SR X#-5US9$![Ra UR 5 Of=fU(aMZ[R"XS9$)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!++      I EG s{7#( g%+ S[# G S[ LL  S[ LL  Z rd*  "9@  MMO MMO  ::bggi&78H H##  MMO  ) ::a ''s--7%E$FFz```(json)?(.*)parserc[XS9$![Ra. [R U5nUcUOUR S5nOf=f[X1S9$)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;66   D!((5#(-;U[[^ D x //s ?A  A z `cRUR[5n[U5nU"U5$)anParse 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([U5n[ U[ 5(d"S[ U5RS3n[XPS9eUHnXb;dM SUSU3n[U5e U$![RanSU3n[U5UeSnAff=f)aoParse 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 $MCC  ;C5A**25 (, ,  O!   004#C(a/0s A''B;B  B)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(bJJ0"))<9K00!50006H26"r