NgNdZddlmZddlZddlZddlmZddlmZm Z m Z m Z ddl m Z ddlmZmZddlmZdd ZGdde ZGddeZGddeZdS)z"Tools for working with JSON specs.) annotationsN)Path)DictListOptionalUnion) BaseModel)AsyncCallbackManagerForToolRunCallbackManagerForToolRun)BaseTooltextstrreturnList[Union[str, int]]c`tjd|}d|D}d|D}|S)zDParse input of the form data["key1"][0]["key2"] into a list of keys.z\[.*?]crg|]4}|dddddd5S)"')replace.0is _/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/tools/json/tool.py z _parse_input..s> C C C1QrT7??3 # # + +C 4 4 C C CcXg|]'}|rt|n|(S)isdigitintrs rrz _parse_input..s/ 5 5 5AQYY[[ '3q666a 5 5 5r)refindall)r _resress r _parse_inputr's> :i & &D C Cd C C CC 5 5 5 5 5C JrcTeZdZUdZded<dZded<edd Zdd ZddZ dS)JsonSpeczBase class for JSON spec.rdict_r"max_value_lengthpathrrc|std|tj|}||S)zCreate a JsonSpec from a file.zFile not found: )r*)existsFileNotFoundErrorjsonloads read_text)clsr-r*s r from_filezJsonSpec.from_file#sW{{}} ?#$=t$=$=>> > 4>>++,,srr rcF t|}|j}|D] }|r||} t|tst d|dt t |S#t$r}t|cYd}~Sd}~wwxYw)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 ValueErrorrlistkeys Exceptionrepr)selfr itemsvalres rr;z JsonSpec.keys+s  &&E*C ! !!a&Cc4((  TdTTTtCHHJJ''(( (   77NNNNNN sA:A== B BB B c t|}|j}|D] }||} t|tr't t ||jkrdSt |}t ||jkr|d|jdz}|S#t$r}t|cYd}~Sd}~wwxYw)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*r7r8lenrr,r<r=)r>r r?r@rstr_valrAs rvaluezJsonSpec.value?s  &&E*C  !f#t$$ WSXX9N)N)NVV#hhG7||d333!"9D$9"9:UBN   77NNNNNN s$AB!:B B?&B:4B?:B?N)r-rrr))r rrr) __name__ __module__ __qualname____doc____annotations__r, classmethodr5r;rEr rrr)r)s}##KKK   [ (rr)cReZdZUdZdZded<dZded<ded< dddZ dddZd S)JsonListKeysToolz%Tool for listing keys in a JSON spec.json_spec_list_keysrnamez 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]rc6|j|SN)rQr;r>rRrSs r_runzJsonListKeysTool._run`s y~~j)))r(Optional[AsyncCallbackManagerForToolRun]c0K||SrVrXrWs r_arunzJsonListKeysTool._arung yy$$$rrVrRrrSrTrrrRrrSrYrr rFrGrHrIrOrJrPrXr\r rrrMrMUs//%D%%%%K NNN <@*****AE%%%%%%%rrMcReZdZUdZdZded<dZded<ded< dddZ dddZd S)JsonGetValueToolz(Tool for getting a value in a JSON spec.json_spec_get_valuerrOz 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"]). rPr)rQNrRrSrTrc6|j|SrV)rQrErWs rrXzJsonGetValueTool._runzs yz***rrYc0K||SrVr[rWs rr\zJsonGetValueTool._arunr]rrVr^r_r`r rrrbrbos22%D%%%%K NNN <@+++++AE%%%%%%%rrb)r rrr)rI __future__rr1r#pathlibrtypingrrrrpydanticr langchain_core.callbacksr r langchain_core.toolsr r'r)rMrbr rrrlsP(("""""" ............*)))))55555y555p%%%%%x%%%4%%%%%x%%%%%r