a bgN@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.0irq/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/tools/json/tool.py z _parse_input..cSs g|]}|rt|n|qSr)isdigitintrrrrrr)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|WSty|}zt |WYd}~Sd}~00dS)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)selfritemsvalrerrrr3+s   z JsonSpec.keysc Cszpt|}|j}|D] }||}qt|trDtt||jkrDWdSt|}t||jkrl|d|jd}|WSty}zt|WYd}~Sd}~00dS)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/r0lenr r&r4r5)r6rr7r8rZstr_valr9rrrvalue?s zJsonSpec.valueN) __name__ __module__ __qualname____doc____annotations__r& classmethodr.r3r;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)rEr3r6rHrIrrr_run`szJsonListKeysTool._run(Optional[AsyncCallbackManagerForToolRun]cs ||SrJrLrKrrr_arungszJsonListKeysTool._arun)N)N r<r=r>r?rCr@rDrLrOrrrrrBUs    rBc@sZeZdZUdZdZded<dZded<ded<ddd dd d d Zddddd ddZd S)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#rENrFrGcCs |j|SrJ)rEr;rKrrrrLzszJsonGetValueTool._runrMcs ||SrJrNrKrrrrOszJsonGetValueTool._arun)N)NrPrrrrrQos    rQ)r? __future__rr*rpathlibrtypingrrrrZpydanticrZlangchain_core.callbacksr r Zlangchain_core.toolsr r"r#rBrQrrrrs     8