o ZhN@sdZddlmZddlZddlZddlmZddlmZm Z m Z m Z ddl m Z ddlmZmZddlmZdd dZGddde ZGdddeZGdddeZdS)z"Tools for working with JSON specs.) annotationsN)Path)DictListOptionalUnion) BaseModel)AsyncCallbackManagerForToolRunCallbackManagerForToolRun)BaseTooltextstrreturnList[Union[str, int]]cCs,td|}dd|D}dd|D}|S)zDParse input of the form data["key1"][0]["key2"] into a list of keys.z\[.*?]cSs(g|]}|ddddddqS)"')replace.0irZ/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/tools/json/tool.py s(z _parse_input..cSs g|] }|r t|n|qSr)isdigitintrrrrrs )refindall)r Z_resresrrr _parse_inputs r!c@sHeZdZUdZded<dZded<edd d ZdddZdddZ dS)JsonSpeczBase class for JSON spec.rdict_rmax_value_lengthpathrrcCs.|s td|t|}||dS)zCreate a JsonSpec from a file.zFile not found: )r#)existsFileNotFoundErrorjsonloads read_text)clsr&r#rrr from_file#s zJsonSpec.from_filer r c Cs~z(t|}|j}|D]}|r||}q t|ts td|dtt|WSty>}z t |WYd}~Sd}~ww)zReturn the keys of the dict at the given path. Args: text: Python representation of the path to the dict (e.g. data["key1"][0]["key2"]). zValue at path `z(` is not a dict, get the value directly.N) r!r# isinstancedict ValueErrorr listkeys Exceptionrepr)selfr itemsvalrerrrr2+s   z JsonSpec.keysc Csz8t|}|j}|D]}||}q t|tr"tt||jkr"WdSt|}t||jkr6|d|jd}|WStyN}z t|WYd}~Sd}~ww)zReturn the value of the dict at the given path. Args: text: Python representation of the path to the dict (e.g. data["key1"][0]["key2"]). z=Value is a large dictionary, should explore its keys directlyNz...) r!r#r.r/lenr r%r3r4)r5r r6r7rZstr_valr8rrrvalue?s zJsonSpec.valueN)r&rrr")r r rr ) __name__ __module__ __qualname____doc____annotations__r% classmethodr-r2r:rrrrr"s    r"c@NeZdZUdZdZded<dZded<ded< ddddZ ddddZd S)JsonListKeysToolz%Tool for listing keys in a JSON spec.Zjson_spec_list_keysr namez Can be used to list all keys at a given path. Before calling this you should be SURE that the path to this exists. The input is a text representation of the path to the dict in Python syntax (e.g. data["key1"][0]["key2"]). descriptionr"specN tool_input run_manager#Optional[CallbackManagerForToolRun]rcC |j|SN)rEr2r5rFrGrrr_run` zJsonListKeysTool._run(Optional[AsyncCallbackManagerForToolRun]c ||SrJrLrKrrr_arung zJsonListKeysTool._arunrJrFr rGrHrr rFr rGrNrr r;r<r=r>rCr?rDrLrQrrrrrBU     rBc@rA)JsonGetValueToolz(Tool for getting a value in a JSON spec.Zjson_spec_get_valuer rCz Can be used to see value in string format at a given path. Before calling this you should be SURE that the path to this exists. The input is a text representation of the path to the dict in Python syntax (e.g. data["key1"][0]["key2"]). rDr"rENrFrGrHrcCrIrJ)rEr:rKrrrrLzrMzJsonGetValueTool._runrNcrOrJrPrKrrrrQrRzJsonGetValueTool._arunrJrSrTrUrrrrrWorVrW)r r rr)r> __future__rr)rpathlibrtypingrrrrZpydanticrZlangchain_core.callbacksr r Zlangchain_core.toolsr r!r"rBrWrrrrs      8