Zǻi-%ddlmZddlmZddlmZmZmZmZm Z m Z m Z ddl m Z mZmZGddeeZGdde ZGd d eZGd d eZGd deZGddeZGddeZGddeZGddeZy))ABC)Enum)AnyDictListCallableOptionalUnionClassVar) BaseModelFieldmodel_validatorc(eZdZdZdZdZdZdZdZdZ y) ParameterTypez6Enum for parameter types supported in tool parameters.stringintegernumberbooleanobjectarrayN) __name__ __module__ __qualname____doc__STRINGINTEGERNUMBERBOOLEANOBJECTARRAYQ/opt/lhia/marcimex/agent/venv/lib/python3.12/site-packages/neo4j_graphrag/tool.pyrrs#@ FG FG F Er"rcteZdZUdZeed<dZeed<ee ed<de ee ffdZ e de ee fddfd Zy ) ToolParameterz2Base class for all tool parameters using Pydantic. descriptionFrequiredtypereturnc8|j|jd}|S)zr!r"r#r%r%s^<Hd = !!c3h 0T#s(^000r"r%cveZdZUdZej Zeeed<dZ e e e ed<de e efffd ZxZS)r3zString parameter for tools.r(Nenumr)cZt|}|jr|j|d<|S)NrD)superr.rDr,r- __class__s r#r.zStringParameter.model_dump_tool=s)(* 99!YYF6N r")rrrrrrr(r r@rDr rr?rrr. __classcell__rHs@r#r3r37sG%$1$8$8D(= !8 $D(49 $c3hr"r3ceZdZUdZej Zeeed<dZ e e ed<dZ e e ed<de eefffd ZxZS)r4zInteger parameter for tools.r(Nminimummaximumr)ct|}|j|j|d<|j|j|d<|SNrLrMrFr.rLrMrGs r#r.z IntegerParameter.model_dump_toolKE(* << # $ F9  << # $ F9  r")rrrrrrr(r r@rLr intrMrr?rr.rIrJs@r#r4r4DsP&$1$9$9D(= !9!GXc]!!GXc]!c3hr"r4ceZdZUdZej Zeeed<dZ e e ed<dZ e e ed<de eefffd ZxZS)r5zNumber parameter for tools.r(NrLrMr)ct|}|j|j|d<|j|j|d<|SrOrPrGs r#r.zNumberParameter.model_dump_tool[rQr")rrrrrrr(r r@rLr floatrMrr?rr.rIrJs@r#r5r5TsP%$1$8$8D(= !8#GXe_##GXe_#c3hr"r5c:eZdZUdZej Zeeed<y)r6zBoolean parameter for tools.r(N) rrrrrrr(r r@r!r"r#r6r6ds&$1$9$9D(= !9r"r6ceZdZUdZej Zeeed<ded<dZ e e ed<dZ e e ed<e d ed eeefd eeeffd Zd eeefffd Ze d ddZxZS)r8zArray parameter for tools.r(r%itemsN min_items max_itemsbeforemodevaluesr)cx|jd}t|trtj ||d<|S)NrX)r1 isinstancedictr%r>)r:r^rXs r#_preprocess_itemsz ArrayParameter._preprocess_itemsrs5 7# eT "+55e> %!%F:  >> %!%F:  r"afterct|jtsat|jtr&tj |j|_|St dt |jt |jtur2tj |jj|_|S)Nz+Items must be a ToolParameter or dict, got )r`rXr%rar>r2r( model_dumpr,s r#validate_itemszArrayParameter.validate_itemss$**m4$**d+*44TZZ@  !A$tzzBRAST$**  .&001F1F1HIDJ r")r)r8)rrrrrr r(r r@rYr rRrZrrBrar?rrbrr.rjrIrJs@r#r8r8js$$1$7$7D(= !7 #Ix}##Ix}#(#tCH~$sCx.$c3h'" # r"r8ceZdZUdZej Zeeed<e e e fed<e e Zee ed<dZeed<ed ed ee efd ee effd Zdd ee e d e e efffd Zed ddZxZS)r7zObject parameter for tools.r( properties)default_factoryrequired_propertiesTadditional_propertiesr[r\r^r)c|jd}t|trMi}|jD]3\}}t|trtj |||</|||<5||d<|S)Nrl)r1r`rarXr%r>)r:r^props new_propskvs r#_preprocess_propertiesz&ObjectParameter._preprocess_propertiessr <( eT "I  %1a�#:#:1#=IaL#$IaL  % $-F<  r"excludec|xsg}i}|jjD]\}}||vr |j||<t| }||d<|jrd|vr|j|d<d|vr|j |d<|S)Nrlr'roadditionalProperties)rlrXr.rFrnro)r,rvproperties_dictnameparamr-rHs r#r.zObjectParameter.model_dump_tools-R*,??002 r2r(rh)r,validated_propertiesrzr{s r#validate_propertiesz#ObjectParameter.validate_propertiess!??002 3KD%e]3eT*1>1H1H1O(.$#D6)OPTUZP[}]e --:-D-DUEUEUEW-X$T*-2$T* 3/ r"N)r)r7)rrrrrrr(r r@rr?r%r listrnrrorArrBrarrur r.r~rIrJs@r#r7r7s%$1$8$8D(= !8S-'((%*4%@c@"&4&(# DcN tCH~ $ xS ':d3PS8n&'"#r"r7ceZdZUdZeed<eed<eeed<ede fed< dded ed ede fd ee ee ee fffd Z d efdZ d efdZddeeed e ee ffdZde d e fdZd efdZd efdZy)ToolzWAbstract base class defining the interface for all tools in the neo4j-graphrag library._name _description _parameters. _execute_funcNrzr& execute_func parametersc||_||_||_t|trt j ||_yt|t r||_y|d|_ytdt|jd|)Nz7Parameters must be None, dict, or ObjectParameter, got z: ) rrrr`rar7r9r TypeErrorr(r)r,rzr&rrs r#__init__z Tool.__init__s ') j$ '.==jID   O 4)D   #D J'001J<A r"r)c|jS)zWGet the name of the tool. Returns: str: Name of the tool. )rris r#get_namez Tool.get_names zzr"c|jS)zvGet a detailed description of what the tool does. Returns: str: Description of the tool. )rris r#get_descriptionzTool.get_descriptions    r"rvcT|jr|jj|SiS)zGet the parameters the tool accepts in a dictionary format suitable for LLM providers. Returns: Dict[str, Any]: Dictionary containing parameter schema information. )rr.)r,rvs r#get_parameterszTool.get_parameterss)   ##33G< < r"kwargsc &|jdi|S)zExecute the tool with the given query and additional parameters. Args: **kwargs (Any): Additional parameters for the tool. Returns: Any: The result of the tool execution. r!)r)r,rs r#executez Tool.execute s"t!!+F++r"cVd|jd|jd|jdS)Nz Tool(name=z, description=z , parameters=))rrrris r#__str__z Tool.__str__s1DJJ<~d6G6G5H VZVfVfUgghiir"c"|jSr)rris r#__repr__z Tool.__repr__s||~r"r)rrrrr?r@r r7rrr rrrrrrrrrr!r"r#rrsa J/**CH%%HL sCx(  U?DcN#BCD .#!!htCy&9T#s(^ , , ,jj#r"rN)abcrrDrtypingrrrrr r r pydanticr r rr?rr%r3r4r5r6r8r7rr!r"r#rsGGG66C"0I"0J m  }  m  :}: &]&R;m;|H3Hr"