~hGddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl m Z ddlmZddlmZddlmZddlmZmZerddlmZdgZe d ed efd Zgd Zd ed efdZGddeZ Gdde Z!Gdde Z"Gdde Z#Gdde"Z$Gdde"Z%Gdde Z&ded efdZ'ded e(e)eefeffdZ*d ed e(eefd edzfd!Z+d ed e(eefd efd"Z,d#Z-e-fd$ed%e.d efd&Z/Gd'd(ej`Z1d)Z2d*Z3d+Z4d7d,Z5d efd-Z6d.Z7d/Z8d0eezfd1Z9d2ed efd3Z:d4jwZ<Gd5d6Z=y)8N) lru_cache)BytesIO)Path)dedent) TYPE_CHECKINGAny) AgentLogger AgentError package_namereturnc tjj|y#tjj$rYywxYw)NTF) importlibmetadataversionPackageNotFoundError)r s L/opt/mcp/mcp-sentiment/venv/lib/python3.12/site-packages/smolagents/utils.py_is_package_availabler*s<""<0    2 2s"AA) collectionsdatetime itertoolsmathqueuerandomrestat statisticstime unicodedatatextc6d}tjd||S)zLEscapes square brackets in code segments while preserving Rich styling tags.c|jd}tjdd|}|jrd|dSd|dS)NzQbold|red|green|blue|yellow|magenta|cyan|white|black|italic|dim|\s|#[0-9a-fA-F]{6}z\[z\][])grouprsubstrip)matchcontentcleaneds rreplace_bracketed_contentz7escape_code_brackets..replace_bracketed_contentEsL++a.&& `bdfm &-]]_WIS!HAgYa.Hz \[([^\]]*)\])rr')rr,s rescape_code_bracketsr.BsI 66/#r< __classcell__)r8s@rr r Os!7" Od38nOr-ceZdZdZy)AgentParsingErrorz3Exception raised for errors in parsing in the agentNr;r?r@rAr-rrDrD[s=r-rDceZdZdZy)AgentExecutionError5Exception raised for errors in execution in the agentNrErFr-rrHrHa?r-rHceZdZdZy)AgentMaxStepsErrorrINrErFr-rrLrLgrJr-rLceZdZdZy)AgentToolCallErrorzKException raised for errors when incorrect arguments are passed to the toolNrErFr-rrNrNmsUr-rNceZdZdZy)AgentToolExecutionErrorz1Exception raised for errors when executing a toolNrErFr-rrPrPss;r-rPceZdZdZy)AgentGenerationErrorz6Exception raised for errors in generation in the agentNrErFr-rrRrRys@r-rRobjc F|yt|ttttfryt|trg |j dr|j ds"|j dr1|j dr tj|}t|S|S|St|ttfr|Dcgc] }t|c}St|tr6|jDcic]\}}t|t|c}}St|drOd|j j"i|j$jDcic]\}}|t|c}}St|S#tj$rY|SwxYwcc}wcc}}wcc}}w)z4Recursive function to make objects JSON serializableN{}r$r%__dict___type) isinstancer<intfloatbool startswithendswithjsonloadsmake_json_serializableJSONDecodeErrorlisttupler>itemshasattrr8r;rW)rSparseditemkvs rrarasb { C#sE40 1 c3  NN3'CLL,=3>>RUCV[^[g[ghk[l!ZZ_F1&99 s C$ '9<=&t,== C >AiikJdaA.q11JJ j !//s]`]i]i]o]o]q3rUYUVXYA7Ma7P4P3rss3x''   >J4ss$A#E87F/FF8FF json_blobc  |jd}ttjd|Dcgc]}|j c}d}|||dz}t j |d}||d|fScc}w#t$r tdt j$rH}|j}||dz |d zd k(r td td |d |d||dz |dzdd}~wwxYw)zZExtracts the JSON blob from the input and returns the JSON data and the rest of the input.rUrVr"F)strictNz0The model output does not contain any JSON blob.z}, zmJSON is invalid: you probably tried to provide multiple tool calls in one action. PROVIDE ONLY ONE TOOL CALL.z>The JSON blob you used is invalid due to the following error: z. JSON blob was: z6, decoding failed on that specific part of the blob: 'z'.) findrcrfinditerstartr_r` IndexError ValueErrorrbpos)rkfirst_accolade_indexalast_accolade_indexjson_str json_dataeplaces rparse_json_blobrs (~~c226r{{3 7R2STQqwwyTUWX14G!4KLJJx6 )$9%9:::U MKLL      UQY +v 5 LQCP'[)%!)eai01 5   s)2A<A7 +A<7A<<&C*"AC%%C*code_block_tagsc|dd|d}tj||tj}|rdjd|DSy)z#Extract code from the LLM's output.r(.*?)r"z c3<K|]}|jywr1)r().0r)s r z)extract_code_from_text..s>U5;;=>sN)rfindallDOTALLjoin)rrpatternmatchess rextract_code_from_textrsO!!$%U?1+=*>?Gjj$ 2G{{>g>>> r-ct||}|s t|d}|r|S tj||S#t$rYnwxYwd|vrBd|vr>t t d|dd|dd|d |dd |dd j t t d |dd|dd |d|dd|dd j )a*Extract code blocs from the LLM's output. If a valid code block is passed, it returns it directly. Args: text (`str`): LLM's output text to parse. Returns: `str`: Extracted code block. Raises: ValueError: If no valid code block is found in the text. )z```(?:python|py)z ```finalanswerzI Your code snippet is invalid, because the regex pattern rrr"zQ was not found in it. Here is your code snippet: zs It seems like you're trying to return the final answer, you can do it as follows: zH final_answer("YOUR FINAL ANSWER HERE") z zE Your code snippet is invalid, because the regex pattern zI was not found in it. Here is your code snippet: z Make sure to include code with the correct pattern, for instance: Thoughts: Your thoughts z1 # Your python code here z )rastparse SyntaxErrorrvrr()rrrs rparse_code_blobsrsK%T?;G (/LM  $    $8t+ IIXYZI[H\\abqrsbtauv #$% #$% eg   EETUVEWDXX]^mno^p]qr F Q  ! Q  !    %'  s7 AAi Nr* max_lengthcTt||kr|S|d|dzd|dz|| dzdzS)Nroz3 ..._This content has been truncated to stay below z characters_... )len)r*rs rtruncate_contentrsS 7|z! %jAo &DZLPab czkQ&() * r-ceZdZdZdZdZy) ImportFinderc"t|_yr1)setpackagesr=s rr3zImportFinder.__init__s  r-c|jD];}|jjdd}|jj |=yN.r)namesnamesplitradd)r6nodealias base_packages r visit_ImportzImportFinder.visit_Imports@ZZ ,E ::++C03L MM  l + ,r-c|jr:|jjdd}|jj|yyr)modulerrr)r6rrs rvisit_ImportFromzImportFinder.visit_ImportFrom s9 ;;;;,,S1!4L MM  l + r-N)r;r?r@r3rrrFr-rrrs, ,r-rcdt|tjr |j}t |S)z6Get source code for a method, including bound methods.)rYtypes MethodType__func__ get_source)methods rget_method_sourcers&&%**+ f r-c t|}t|}djd|jdD}djd|jdD}||k(S#ttf$rYywxYw)z)Compare two methods by their source code. c3`K|]&}|jjdr#|(yw@Nr(r]rlines rrz!is_same_method..$eT H_H_`cHdDe$..c3`K|]&}|jjdr#|(ywrrrs rrz!is_same_method..rrF)rrr TypeErrorOSError)method1method2source1source2s ris_same_methodrsy #G,#G,))eW]]4-@ee))eW]]4-@ee'!! w sA(A++A=<A=cPt|rt|r t||S||k(S)z=Compare two class items (methods or attributes) for equality.)callabler)item1item2s r is_same_itemr&s&8E?eU++~r-c  |j}|j}g}|r#|jd|d|jdn|jd|d|jr:|r|j|jk7r|jd|jd|jj Dcic]A\}}|j ds+t|s |rt||rt|||k(s||C}}}|j D]\}}t|trZd|vr+|jdd }|jd |d |dE|jd |d tj|p|jd |d t||r|jd |jj D cic]\}} t| rv|rYt||rMt| t t"fs7t||j$j&| j$j&k7r|t| dr | j(n| } }} | j D]\}} t+| } | j-d} | d}t/|t/|j1z }| Dcgc]}||d } }dj3| Dcgc]}|j5rd |zn|c}} |j| |jd t7}|j9t;j<dj3||j>}g}|r+|jd|j@d|j|D]}|jd||r|jd |jC|dj3|Scc}}wcc} }wcc}wcc}w)z e eS !u} % eY ? ""T$vm_C#HI""T$s4::e3D2E#FG   dV3tE{m< = >2 ,,,,.  D$ D>8T*4, !<=Hd+44<< @U@UU '$ ">dDH G    f"6* $**40 !!_ Z3z'8'8':#;;2>?$VW ? ? Wc"dtDJJL6D=d#J"de =)2 !NM $))K"89:$--KU8#6#6"7x@Q@Q?RST$0WWI./02{# 99[ !!M. (@"ds8AO/B O5 O;P ct|ts"t|stdt|d} t |ddxst j |}t|jS#t$r }|}Yd}~nd}~wwxYw ddl }|j}|s tddj|jjdgj}|s t!dt#j$|}t#j&|D]}t|t"j(t"j*fs.|j,|j.k(sHtdj|j1d|j2d z |j4jcSt!d |j.d #t$r|t $r}||d}~wwxYw) aGet the source code of a class or callable object (e.g.: function, method). First attempts to get the source code using `inspect.getsource`. In a dynamic environment (e.g.: Jupyter, IPython), if this fails, falls back to retrieving the source code from the current interactive shell session. Args: obj: A class or callable object (e.g.: function, method) Returns: str: The source code of the object, dedented and stripped Raises: TypeError: If object is not a class or callable OSError: If source code cannot be retrieved from any source ValueError: If source cannot be found in IPython history Note: TODO: handle Python standard REPL z Expected class or callable, got N __source__rzNo active IPython shell foundrInz&No code cells found in IPython sessionr"zCould not find source code for z in IPython history)rYr:rrrinspect getsourcerr(rIPython get_ipython ImportErrorruser_nsgetrvrrwalkClassDef FunctionDefrr;rlineno end_lineno) rS inspect_errorsourcer}rshell all_cellstreers rrrs( sD !Xc]:49+FGGMlD1KW5F5Fs5Kf~##%%  ###%=> >IIemm//b9:@@B EF Fyy#HHTN kD$s ?@TYYRUR^R^E^dii (=dkkAoPTP_P_(`abhhjj k:3<<.H[\]]  #]"#sCG G%AG8GG-%G((G-ct}|j|dtj|j j dS)NPNG)formatzutf-8)rsavebase64 b64encodegetvaluedecode)imagebuffereds rencode_image_base64r s>yH JJxJ&   H--/ 0 7 7 @@r-c d|S)Nzdata:image/png;base64,rF) base64_images rmake_image_urlr s #L> 22r-folderctj|dttjj |dd5 dddy#1swYyxYw)NT)exist_okz __init__.pyw)osmakedirsopenpathr)rs rmake_init_filersAKK& bggll6=13 7     s A  Arcvt|tr(|jxrtj| SdS)NF)rYr< isidentifierkeyword iskeyword)rs r is_valid_namers4BLTSVBW4    >w'8'8'>#>b]bbr-aimport yaml import os from smolagents import GradioUI, {{ class_name }}, {{ agent_dict['model']['class'] }} # Get current directory path CURRENT_DIR = os.path.dirname(os.path.abspath(__file__)) {% for tool in tools.values() -%} from {{managed_agent_relative_path}}tools.{{ tool.name }} import {{ tool.__class__.__name__ }} as {{ tool.name | camelcase }} {% endfor %} {% for managed_agent in managed_agents.values() -%} from {{managed_agent_relative_path}}managed_agents.{{ managed_agent.name }}.app import agent_{{ managed_agent.name }} {% endfor %} model = {{ agent_dict['model']['class'] }}( {% for key in agent_dict['model']['data'] if key not in ['class', 'last_input_token_count', 'last_output_token_count'] -%} {{ key }}={{ agent_dict['model']['data'][key]|repr }}, {% endfor %}) {% for tool in tools.values() -%} {{ tool.name }} = {{ tool.name | camelcase }}() {% endfor %} with open(os.path.join(CURRENT_DIR, "prompts.yaml"), 'r') as stream: prompt_templates = yaml.safe_load(stream) {{ agent_name }} = {{ class_name }}( model=model, tools=[{% for tool_name in tools.keys() if tool_name != "final_answer" %}{{ tool_name }}{% if not loop.last %}, {% endif %}{% endfor %}], managed_agents=[{% for subagent_name in managed_agents.keys() %}agent_{{ subagent_name }}{% if not loop.last %}, {% endif %}{% endfor %}], {% for attribute_name, value in agent_dict.items() if attribute_name not in ["model", "tools", "prompt_templates", "authorized_imports", "managed_agents", "requirements"] -%} {{ attribute_name }}={{ value|repr }}, {% endfor %}prompt_templates=prompt_templates ) if __name__ == "__main__": GradioUI({{ agent_name }}).launch() c*eZdZdZddedzfdZdZy) RateLimiteraJSimple rate limiter that enforces a minimum delay between consecutive requests. This class is useful for limiting the rate of operations such as API requests, by ensuring that calls to `throttle()` are spaced out by at least a given interval based on the desired requests per minute. If no rate is specified (i.e., `requests_per_minute` is None), rate limiting is disabled and `throttle()` becomes a no-op. Args: requests_per_minute (`float | None`): Maximum number of allowed requests per minute. Use `None` to disable rate limiting. Nrequests_per_minutecT|du|_|jrd|z nd|_d|_y)NgN@g)_enabled _interval _last_call)r6rs rr3zRateLimiter.__init__ s*+47 7;}} 33#r-c|jsytj}||jz }||jkr"tj|j|z tj|_y)z6Pause execution to respect the rate limit, if enabled.N)r rr"r!sleep)r6nowelapseds rthrottlezRateLimiter.throttlesS}} iik' T^^ # JJt~~/ 0))+r-r1)r;r?r@rAr[r3r'rFr-rrrs EDL &r-rr1)>rrimportlib.metadatarimportlib.utilrr_rrrrr functoolsriorpathlibrtextwraprtypingrrsmolagents.memoryr __all__r<r\rBASE_BUILTIN_MODULESr. Exceptionr rDrHrLrNrPrRrardr>rrrMAX_LENGTH_TRUNCATE_CONTENTrZr NodeVisitorrrrrrrr r rrr(AGENT_GRADIO_APP_TEMPLATErrFr-rr6s"   %- .    Ds Ds D O O   *    ,  1  : 4 s uT#s(^S-@'A .uS#X3QU:535sCx5S5p$6Q c s UX ,3??," W"t3#s3#lA 3 3: ccc$H EGIN&&r-