ihNdZddlmZddlZddlZddlmZddlmZm Z m Z m Z ddl m Z ddlmZmZddlmZdd ZGd d e ZGd d eZGddeZy)z"Tools for working with JSON specs.) annotationsN)Path)DictListOptionalUnion) BaseModel)AsyncCallbackManagerForToolRunCallbackManagerForToolRun)BaseToolctjd|}|Dcgc]'}|ddjddjdd)}}|Dcgc]}|jr t |n|!}}|Scc}wcc}w)zDParse input of the form data["key1"][0]["key2"] into a list of keys.z\[.*?]"')refindallreplaceisdigitint)text_resiress a/var/www/html/dev/engine/venv/lib/python3.12/site-packages/langchain_community/tools/json/tool.py _parse_inputrsu ::i &D>B C1Qr7??3 # + +C 4 CC C14 5AQYY[3q6a ' 5C 5 J D 5s ,A5 $A:cLeZdZUdZded<dZded<ed dZd dZd d Z y ) JsonSpeczBase class for JSON spec.rdict_rmax_value_lengthc|jstd|tj|j }||S)zCreate a JsonSpec from a file.zFile not found: )r )existsFileNotFoundErrorjsonloads read_text)clspathr s r from_filezJsonSpec.from_file#s@{{}#&6tf$=> > 4>>+,c t|}|j}|D] }|s||} t|tst d|dt t |jS#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) rr isinstancedict ValueErrorstrlistkeys Exceptionrepr)selfritemsvalres rr3z JsonSpec.keys+s  &E**C !a&C !c4( %dV+STtCHHJ'( ( 7N s#A)AA)) B2 B<BBc^ t|}|j}|D]}||} t|tr"t t ||j kDryt |}t ||j kDr|d|j dz}|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...) rr r.r/lenr1r"r4r5)r6rr7r8rstr_valr9s rvaluezJsonSpec.value?s  &E**C !f #t$SX9N9N)NV#hG7|d333!"9D$9$9:UBN 7N s$AB6B B, B'!B,'B,N)r*rreturnr)rr1r>r1) __name__ __module__ __qualname____doc____annotations__r" classmethodr+r3r=r,rrrs1# Kc  (r,rcdeZdZUdZdZded<dZded<ded< d d d Z d dd Zy )JsonListKeysToolz%Tool for listing keys in a JSON spec.json_spec_list_keysr1namez 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"]). descriptionrspecNc8|jj|SN)rKr3r6 tool_input run_managers r_runzJsonListKeysTool._run`s yy~~j))r,c,K|j|SwrMrQrNs r_arunzJsonListKeysTool._arung yy$$rMrOr1rPz#Optional[CallbackManagerForToolRun]r>r1rOr1rPz(Optional[AsyncCallbackManagerForToolRun]r>r1 r?r@rArBrIrCrJrQrTrEr,rrGrGUss/%D#%K N <@**9* *AE%%>% %r,rGcdeZdZUdZdZded<dZded<ded< d d d Z d dd Zy )JsonGetValueToolz(Tool for getting a value in a JSON spec.json_spec_get_valuer1rIz 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"]). rJrrKNc8|jj|SrM)rKr=rNs rrQzJsonGetValueTool._runzs yyz**r,c,K|j|SwrMrSrNs rrTzJsonGetValueTool._arunrUrVrMrWrXrYrEr,rr[r[oss2%D#%K N <@++9+ +AE%%>% %r,r[)rr1r>zList[Union[str, int]])rB __future__rr&rpathlibrtypingrrrrpydanticr langchain_core.callbacksr r langchain_core.toolsr rrrGr[rEr,rresQ(" ..*5y5p%x%4%x%r,