U h,@sdZddlmZddlZddlZddlmZddlmZm Z m Z m Z ddl m Z ddlmZmZddlmZd d d d d ZGddde ZGdddeZGdddeZdS)z"Tools for working with JSON specs.) annotationsN)Path)DictListOptionalUnion) BaseModel)AsyncCallbackManagerForToolRunCallbackManagerForToolRun)BaseToolstrzList[Union[str, int]]textreturncCs,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.0irG/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/tools/json/tool.py sz _parse_input..cSs g|]}|rt|n|qSr)isdigitintrrrrrs)refindall)rZ_resresrrr _parse_inputs r!c@sZeZdZUdZded<dZded<edddd d Zd d d d dZd d d ddZ dS)JsonSpeczBase class for JSON spec.rdict_rmax_value_lengthr)pathrcCs.|std|t|}||dS)zCreate a JsonSpec from a file.zFile not found: )r#)existsFileNotFoundErrorjsonloads read_text)clsr&r#rrr from_file#szJsonSpec.from_filer r c CszPt|}|j}|D]}|r||}qt|ts@td|dtt|WStk r~}zt |WYSd}~XYnXdS)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)selfritemsvalrerrrr2+s   z JsonSpec.keysc Cszpt|}|j}|D] }||}qt|trDtt||jkrDWdSt|}t||jkrl|d|jd}|WStk r}zt|WYSd}~XYnXdS)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)r5rr6r7rZstr_valr8rrrvalue?s zJsonSpec.valueN) __name__ __module__ __qualname____doc____annotations__r% classmethodr-r2r:rrrrr"s  r"c@sZeZdZUdZdZded<dZded<ded<ddd dd d d Zddddd ddZd 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#Optional[CallbackManagerForToolRun] tool_input run_managerrcCs |j|SN)rDr2r5rGrHrrr_run`szJsonListKeysTool._run(Optional[AsyncCallbackManagerForToolRun]cs ||SrIrKrJrrr_arungszJsonListKeysTool._arun)N)N r;r<r=r>rBr?rCrKrNrrrrrAUs    rAc@sZeZdZUdZdZded<dZded<ded<ddd dd d d Zddddd ddZd S)JsonGetValueToolz(Tool for getting a value in a JSON spec.Zjson_spec_get_valuer rBz 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"]). rCr"rDNrErFcCs |j|SrI)rDr:rJrrrrKzszJsonGetValueTool._runrLcs ||SrIrMrJrrrrNszJsonGetValueTool._arun)N)NrOrrrrrPos    rP)r> __future__rr)rpathlibrtypingrrrrZlangchain_core.pydantic_v1rZlangchain_core.callbacksr r Zlangchain_core.toolsr r!r"rArPrrrrs     8