~h*ddlZddlZddlmZddlmZmZmZee eZ GddejZ d de ddfd Zy) N) zip_longest)BASE_BUILTIN_MODULES get_source is_valid_nameceZdZdZddeedefdZdZdZ dZ dZ d Z d Z d Zd Zd ZdZdZdZdZdZdZdZy) MethodCheckerz Checks that a method - only uses defined names - contains no local imports (e.g. numpy is ok but local_script is not) class_attributes check_importsct|_i|_i|_t|_t|_||_g|_||_dh|_ t|_ y)NAny) setundefined_namesimports from_importsassigned_names arg_namesr errorsr typing_namesdefined_classes)selfr r s V/opt/mcp/mcp-sentiment/venv/lib/python3.12/site-packages/smolagents/tool_validation.py__init__zMethodChecker.__init__sY"u !e 0 *"G"ucN|jDchc]}|jc}|_|jr/|jj |jj|j r0|jj |j jyycc}w)zCollect function argumentsN)argsargrkwargaddvararg)rnoders rvisit_argumentszMethodChecker.visit_argumentssh-1YY7c#''7 :: NN  tzz~~ . ;; NN  t{{ / 8sB"c|jD]5}|jxs |j}|j|j|<7yN)namesasnamenamer)rr!r' actual_names r visit_ImportzMethodChecker.visit_Import&s9JJ 2D++2K(, DLL % 2rc|jxsd}|jD]7}|jxs |j}||jf|j|<9y)N)moduler%r&r'r)rr!r,r'r(s rvisit_ImportFromzMethodChecker.visit_ImportFrom+sP"JJ AD++2K.4dii-@D  k * Arc|jD]}t|tjr&|jj |j Ct|tjtjfsn|jD]B}t|tjs|jj |j D|j|jyr$) targets isinstanceastNamerridTupleListeltsvisitvaluerr!targetelts r visit_AssignzMethodChecker.visit_Assign1sll 8F&#((+##'' 2FSYY$9:!;;8C!#sxx0++//78  8 4::rc |jD]c}|jst|jtjs5|j j |jje|j|y)z=Track aliases in 'with' statements (the 'y' in 'with X as y')N) items optional_varsr0r1r2rrr3 generic_visit)rr!items r visit_WithzMethodChecker.visit_With;sdJJ CD!!d00#((;''++D,>,>,A,AB C 4 rc|jr%|jj|j|j|y)z 4 rc\|j|j|j|y)z,Track variables in dictionary comprehensionsNrOrDs rvisit_DictCompzMethodChecker.visit_DictCompirQrc\|j|j|j|y)z%Track variables in set comprehensionsNrOrDs r visit_SetCompzMethodChecker.visit_SetCompnrQrct|jtjr|jjdk(s|j |yyNr)r0r8r1r2r3r@rDs rvisit_AttributezMethodChecker.visit_Attributess74::sxx0TZZ]]f5L   t $6Mrcp|jj|j|j|y)zTrack class definitionsN)rrr'r@rDs rvisit_ClassDefzMethodChecker.visit_ClassDefws(   + 4 rclt|jtjr|jt vs|jt vs|j|jvs|jdk(s|j|jvs|j|jvs|j|jvsu|j|jvs\|j|jvsC|j|jvs*|jjd|jdyyyyyyyyyyyyNrzName 'z' is undefined.)r0ctxr1Loadr3_BUILTIN_NAMESrrr rrrrrrappendrDs r visit_NamezMethodChecker.visit_Name|s  dhh )>)772277dnn,77f$77d33377dll*77d///77d11177d///77d222 ""VDGG9O#DE3020+4%-3* *rct|jtjrR|jjt vs5|jjt vs|jj|jvs|jjdk(s|jj|jvs|jj|jvs|jj|jvsw|jj|jvsU|jj|jvs3|jjd|jjd|j|yr\)r0funcr1r2r3r_rrr rrrrrr`r@rDs r visit_CallzMethodChecker.visit_Calls dii * .99<<#7799<<4>>199<<6)99<<4#8#8899<<4<</99<<4#4#4499<<4#6#6699<<4#7#77 ""VDIILL>#IJ 4 rNT)__name__ __module__ __qualname____doc__rstrboolrr"r)r-r<rBrErGrIrMrPrSrUrXrZrardrrr r sp %S %$ %02 A !! #!8! ! ! %! F !rr r returnc Gddtj}|}t|}tj|}|jd}t |tj s td|j|g}|jr||jz }|jr-|jddj|j|jr-|jddj|j|jr-|jddj|j|jD]r}t |tjst!|j"| } | j||| j$D cgc]} d |j&d | c} z }t|r+td |j(d dj|zycc} w)aa Validates that a Tool class follows the proper patterns: 0. Any argument of __init__ should have a default. Args chosen at init are not traceable, so we cannot rebuild the source code for them, thus any important arg should be defined as a class attribute. 1. About the class: - Class attributes should only be strings or dicts - Class attributes cannot be complex attributes 2. About all class methods: - Imports must be from packages, not local files - All methods must be self-contained Raises all errors encountered, if no error returns None. c$eZdZdZdZdZdZy)3validate_tool_attributes..ClassLevelCheckerct|_t|_t|_t|_t|_d|_g|_y)NF)rimported_namescomplex_attributesr non_defaultsnon_literal_defaults in_methodinvalid_attributes)rs rrz.ClassLevelChecker.__init__sC"%%D &)eD #$'ED ! #D (+D %"DN&(D #rc|jdk(r|j||j}d|_|j|||_y)NrT)r'_check_init_function_parametersrvr@)rr! old_contexts rvisit_FunctionDefzEvalidate_tool_attributes..ClassLevelChecker.visit_FunctionDefsAyyJ&44T:..K!DN   t $(DNrc|jry|jD]B}t|tjs|j j |jDtdtj|jDsQ|jD]B}t|tjs|jj |jDt|jddddk(rt|jtjs*|jjd|jdyt|jjt s4|jjd|jjdyt#|jjs4|jjd |jjdyyy) Nc 3K|]j}t|tjtjtjtj tj tjflywr$)r0r1StrNumConstantDictr5Set).0vals r zSvalidate_tool_attributes..ClassLevelChecker.visit_Assign..sG3#''3<<388UXU\U\ ]^sA0A2rr3r+r'z2Class attribute 'name' must be a constant, found ''z0Class attribute 'name' must be a string, found 'z\Class attribute 'name' must be a valid Python identifier and not a reserved keyword, found ')rvr/r0r1r2r rr3allwalkr8rsgetattrrrwr`rjr)rr!r:s rr<z@validate_tool_attributes..ClassLevelChecker.visit_Assigns~~,, 9fchh/))--fii8 9 88DJJ/#ll?F!&#((3//33FII>? t||Ab1V;!$**cll;++225ghlhrhrgsst3uv#DJJ$4$4c:++22J4::K[K[J\\]^'tzz'7'78++22vw{xBxBxHxHwIIJK9t|tjtjtjtjtjtjfr|j j|j yrW)reversedlistrrdefaultsrrtrr0r1r~rrrr5rru)rr!rdefaults rryzSvalidate_tool_attributes..ClassLevelChecker._check_init_function_parameterss (k(499>>:RT\]a]f]f]o]oTp.q)r s ; W?ww&())--cgg6#Gcggsww chhX[X`X`bebibi-jk--11#'':  ;rN)rfrgrhrr{r<ryrlrrClassLevelCheckerrps ) ) < ;rrrzSource code must define a classzKComplex attributes should be defined in __init__, not as class attributes: z, zLParameters in __init__ must have default values, found required parameters: zUParameters in __init__ must have literal default values, found non-literal defaults: )r z- z: zTool validation failed for z:  N)r1 NodeVisitorrparsebodyr0ClassDef ValueErrorr7rwrsr`joinrtru FunctionDefr r rr'rf) clsr rclass_level_checkersourcetree class_noderr!method_checkererrors rvalidate_tool_attributesrs7;COO7;r,- _F 99V D1J j#,, /:;;j) F--%888-- Yyy,??@A C '' Zyy,99:; = // cyy,AABC E U dCOO ,*+>+O+O_lmN   & n>S>STUDII;b0T TF U 6s||nCH499U[K\\]] Us/G=re)r1builtins itertoolsrutilsrrrrvarsr_rr rkrrlrrrsM !BBT(^$O!COOO!dm m m r