o +i@sfdZddlmZddlZddlmZdZdZdZdd d Z dddZ d ddZ d!ddZ d"ddZ dS)#z $TOOL_NAME <$PARAMETER_NAME>$PARAMETER_VALUE ... Here are the tools available: {formatted_tools} z {tool_name} {tool_description} {formatted_parameters} z {parameter_name} {parameter_type} {parameter_description} parameterdict[str, Any]returnstrcCsNd|vr|dSd|vrtd|diSd|vr"td|diSt|S)NtypeanyOfallOf)jsondumps)rr ]/var/www/html/psymed-ai/venv/lib/python3.10/site-packages/langchain_anthropic/experimental.py _get_type+s rtools list[dict]cCs.dd|D}ddd|D}tj|dS)z=Generate a system message that describes the available tools.c Ss<g|]}|d|dddd|ddDdqS)name description cSs*g|]\}}tj|t||ddqS)r)parameter_nameparameter_typeparameter_description)TOOL_PARAMETER_FORMATformatrget).0rrr r r <sz1get_system_message... parameters properties tool_nametool_descriptionformatted_parameters)joinitemsrtoolr r rr7sz&get_system_message..rcSs(g|]}tj|d|d|ddqS)r r!r"r) TOOL_FORMATrr%r r rrIs)formatted_tools)r#SYSTEM_PROMPT_FORMATr)r tools_datatools_formattedr r rget_system_message5s r,trstr | dict[str, Any]cCsxt|dkr |jp dSi}|D]*}|j|vrt|||j<qt||jts/||jg||j<||jt|q|S)Nr)lentexttag _xml_to_dict isinstancelistappend)r-dchildr r rr3Us   r3invokecs|djt|d}fdd|D}t|dkryt|tsy|d}|D]O\}}||ddvrxd|dd|vrx|dd|ddkrUt|tsU|g||<|dd|dd krxt|trxt| d krxt t | ||<q)t |d d d S)Nr rcsg|] }|dkr|qSrr r%r:r rrlsz)_xml_to_function_call..rrrarrayobject)r argumentsfunction)r?r)findr1r3r0r4rr$r5dictkeysnextitervaluesr r )r9rr>filtered_toolsr&keyvaluer r:r_xml_to_function_callgs6  rIelemlist[dict[str, Any]]cs|d}fdd|DS)zJConvert an XML element and its children into a dictionary of dictionaries.r9csg|]}t|qSr )rI)rr9rr rrsz&_xml_to_tool_calls..)findall)rJrinvokesr rLr_xml_to_tool_callss rO)rrrr)rrrr)r-rrr.)r9rrrrr)rJrrrrrK)__doc__ __future__rr typingrr)r'rrr,r3rIrOr r r rs    !