~hR h d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZm Z m
Z
d dl m Z m
Z
d dlmZmZ d dlmZmZmZ d dlmZ d ee
z d
ed efd Zd
ed efdZded efdZd d eded efdZ d d e
ded efdZ d ed efdZ d d ee
z ez defdZ d dede dz dede!dz d ef
dZ" G d d Z#d d gZ$y) N)Path) Generator)
AgentAudio
AgentImage AgentText)MultiStepAgent PlanningStep)
ActionStepFinalAnswerStep)ChatMessageStreamDelta MessageRoleagglomerate_stream_deltas)_is_package_availablestep_log step_namereturnc 0 d| d}| j 5|d| j j dd| j j dz
}|| j j r-dt t
| j j d dndz
}d | d
}|S ) zHGet a footnote string for a step log with duration and token information**z | Input tokens: ,z | Output tokens: z
| Duration: s z/z ) token_usage input_tokens
output_tokenstimingdurationroundfloat)r r
step_footnotestep_footnote_contents P/opt/mcp/mcp-sentiment/venv/lib/python3.12/site-packages/smolagents/gradio_ui.pyget_step_footnote_contentr# s 2&M',X-A-A-N-Nq,QQcdldxdx eG eG HI dJ K K
U]UdUdUmUm}U5 1I1I+JA%N$OqQsuuM OP]^f i model_outputc | sy| j } t j dd| } t j dd| } t j dd| } | j S )z
Clean up model output by removing trailing tags and extra backticks.
Args:
model_output (`str`): Raw model output.
Returns:
`str`: Cleaned model output.
r z```\s*z```z\s*```z```\s*\n\s*)stripresub)r% s r" _clean_model_outputr* ' s` %%'L66-ulCL66-ulCL662E<HL r$ contentc | j } t j dd| } t j dd| } | j } | j d sd| d} | S )z
Format code content as Python code block if it's not already formatted.
Args:
content (`str`): Code content to format.
Returns:
`str`: Code content formatted as a Python code block.
z```.*?\nr z\s*\s*z ```pythonz
```python
z
```)r' r( r)
startswith)r+ s r" _format_code_contentr. ; s` mmoGff["g.Gff("g6GmmoG k * y. Nr$ skip_model_outputsc # K ddl }d| j }|s*|j t j d| dddi |sHt | dd r;t
| j }|j t j |ddi t | d
g r| j d }|j d k( }|j }t |t r%t |j d t | }nt | j }|r t! | }|j t j |d
|j dd } | t | dd rx| j" j r^| j" j }
|
rBt% j& dd |
}
|j t j d|
dddd t | dg ri| j( D ]Z } t+ | j- } |j t j | d| j/ d d dddd \ t | dd r:|j t j t | j0 ddd |j t j t3 | | ddi |j t j d ddi yw) a;
Process an [`ActionStep`] and yield appropriate Gradio ChatMessage objects.
Args:
step_log ([`ActionStep`]): ActionStep to process.
skip_model_outputs (`bool`): Whether to skip model outputs.
Yields:
`gradio.ChatMessage`: Gradio ChatMessages representing the action step.
r NzStep r statusdoneroler+ metadatar% r
tool_callspython_interpreteransweru 🛠️ Used tool )titler1 observationsz^Execution logs:\s*z```bash
u 📝 Execution Logsobservations_imageszimage/.path mime_typeu 🖼️ Output Imageerroru
💥 Error-----)gradio step_number ChatMessager
ASSISTANTgetattrr* r% r6 name arguments
isinstancedictstrgetr' r. r: r( r) r< r to_stringsplitrB r# )
r r/ grrE r% first_tool_call used_codeargsr+ parent_message_tool log_contentimage
path_images
r" _process_action_steprY P s (../0K nn+"7"72k]RTAUaikq`rnss '(NB"G*8+@+@A nn+"7"7 YaciXjnkk x r*"--a0#((,@@ (( dD !$((8SY78G$ioo'G *73G !nn &&-o.B.B-CD -
"! x ,1F1F1L1L1N ++113 &&!7[IK.. **#K=3#8FK !
x.3 11 E#E*446J.. **!+F:CSCSTWCXY[C\B]:^_#9VL !
x$'nn &&HNN0CXdpvNw
..
"
")(K@F#
..k 3 3WPXZ`Oa.
bbs K+K-c # p K ddl }|sV|j t j dddi |j t j | j ddi |j t j t | d ddi |j t j dddi yw) z
Process a [`PlanningStep`] and yield appropriate gradio.ChatMessage objects.
Args:
step_log ([`PlanningStep`]): PlanningStep to process.
Yields:
`gradio.ChatMessage`: Gradio ChatMessages representing the planning step.
r Nz**Planning step**r1 r2 r3 z
Planning steprC )rD rF r
rG planr# )r r/ rQ s r" _process_planning_stepr\ s nn+"7"7AT`hjp_qnrrnn+"7"7 ZbdjYknll
..
"
")(ODF#
..k 3 3WPXZ`Oa.
bbs B4B6c # B K ddl }| j }t |t r9|j t
j d|j dddi yt |t r8|j t
j |j dd ddi yt |t r8|j t
j |j d
d ddi y|j t
j d t | ddi yw) a
Process a [`FinalAnswerStep`] and yield appropriate gradio.ChatMessage objects.
Args:
step_log ([`FinalAnswerStep`]): FinalAnswerStep to process.
Yields:
`gradio.ChatMessage`: Gradio ChatMessages representing the final answer.
r Nz**Final answer:**
r; r1 r2 r3 z image/pngr? z audio/wavz**Final answer:** ) rD outputrK r rF r
rG rO r r rM )r rQ final_answers r" _process_final_answer_stepr` s ??L, *nn &&),*@*@*B)C2F '
L* -nn && ) 3 3 5KP '
L* -nn && ) 3 3 5KP '
nn &&2DS EVDW0Xdlntcu
s D D c # L K t d s t d t | t rt | | E d{ yt | t
rt
| | E d{ yt | t rt | E d{ yt dt | 7 e7 B7 w)ac Extract Gradio ChatMessage objects from agent steps with proper nesting.
Args:
step_log: The step log to display as gr.ChatMessage objects.
skip_model_outputs: If True, skip the model outputs when creating the gr.ChatMessage objects:
This is used for instance when streaming model outputs have already been displayed.
rD UPlease install 'gradio' extra to use the GradioUI: `pip install 'smolagents[gradio]'`NzUnsupported step type: ) r ModuleNotFoundErrorrK r
rY r r\ r r`
ValueErrortype)r r/ s r" pull_messages_from_steprf s ! *! c
(J''2DEEE Hl +)(4FGGG Ho .-h77724>2BCDD
FG7s3 6B$B $B$ B #B$B" B$ B$"B$task task_imagesreset_agent_memoryadditional_argsc # t K t d s t d g }| j ||d|| D ] }t |t t
z t z r$t |t | dd D ] }| g }Et |t sV|j | t | j }| yw) zaRuns an agent with the given task and streams the messages from the agent as gradio ChatMessages.rD rb T)imagesstreamresetrj stream_outputsF)r/ N)
r rc runrK r
r r rf rH r appendr render_as_markdown) agentrg rh ri rj accumulated_eventseventmessagetexts r" stream_to_gradiorx s ! *! c
8: [5GYh eZ, 6 H I2#*52BE#J
"$
5
6 % %e ,,-?@SSUDJ s BB81B8c T e Zd ZdZd
dededz defdZd ZddZ d Z
dd
efd Z d Z y)GradioUIa
Gradio interface for interacting with a [`MultiStepAgent`].
This class provides a web interface to interact with the agent in real-time, allowing users to submit prompts, upload files, and receive responses in a chat-like format.
It can reset the agent's memory at the start of each interaction if desired.
It supports file uploads, which are saved to a specified folder.
It uses the [`gradio.Chatbot`] component to display the conversation history.
This class requires the `gradio` extra to be installed: `smolagents[gradio]`.
Args:
agent ([`MultiStepAgent`]): The agent to interact with.
file_upload_folder (`str`, *optional*): The folder where uploaded files will be saved.
If not provided, file uploads are disabled.
reset_agent_memory (`bool`, *optional*, defaults to `False`): Whether to reset the agent's memory at the start of each interaction.
If `True`, the agent will not remember previous interactions.
Raises:
ModuleNotFoundError: If the `gradio` extra is not installed.
Example:
```python
from smolagents import CodeAgent, GradioUI, InferenceClientModel
model = InferenceClientModel(model_id="meta-llama/Meta-Llama-3.1-8B-Instruct")
agent = CodeAgent(tools=[], model=model)
gradio_ui = GradioUI(agent, file_upload_folder="uploads", reset_agent_memory=True)
gradio_ui.launch()
```
Nrs file_upload_folderri c L t d s t d || _ | t | nd | _ || _ t
|d xs d| _ t
|dd | _ | j 9| j j s | j j dd y y y )NrD rb rI zAgent interface descriptionT)parentsexist_ok) r rc rs r r{ ri rH rI r} existsmkdir)selfrs r{ ri s r" __init__zGradioUI.__init__6 s $X.%g
>P>\$'9":bf "4 E6*?.? "5-> " " .**113''--dT-J 4 /r$ c # K dd l }d|vr| j |d< |j |j d|ddi | t |d || j
D ] }t
||j r$d|d j d< |j | nt
|t ru|j d d
j d d }|d j d d
k( r ||d _
n3|j |j t j |dd
i | | y # t $ r&}| |j dt | d }~ww xY ww)Nr rs userr1 r2 r3 )rg ri r> z\>pendingzError in interaction: )rD rs rq rF rx ri rK r5 rM replacer+ r
rG ExceptionError)r promptmessages
session_staterQ msges r" interact_with_agentz GradioUI.interact_with_agentD sU - '%)ZZM' " > OOBNNRZ\bQcN d eN' g&VH_H_
c2>>26N(( 3CF8<= = >s) E DD, +E , E5!EEE c p ddl }||j dd |fS |g d}t j j |j
d j
}||vr|j dd |fS t j j |j
}t j d
d | }t j j | j t j j | }t j |j
| |j d | d ||gz fS )
a
Upload a file and add it to the list of uploaded files in the session state.
The file is saved to the `self.file_upload_folder` folder.
If the file type is not allowed, it returns a message indicating the disallowed file type.
Args:
file (`gradio.File`): The uploaded file.
file_uploads_log (`list`): A list to log uploaded files.
allowed_file_types (`list`, *optional*): List of allowed file extensions. Defaults to [".pdf", ".docx", ".txt"].
r NzNo file uploadedT)valuevisible)z.pdfz.docxz.txt zFile type disallowed)r z[^\w\-.]_zFile uploaded: )rD Textboxosr@ splitextrI lowerbasenamer( r) joinr{ shutilcopy) r filefile_uploads_logallowed_file_typesrQ file_ext
original_namesanitized_name file_paths r" upload_filezGradioUI.upload_filed s <::$6:EGWW W %!: 77##DII.q1779 - -:: 4d:CEUU U ((3
m
GGLL!8!8"'':J:J>:Z[ DIIy)zzOI;7zFHX\e[fHfffr$ c ` dd l }|t | dkD rd| ndz d|j d fS )Nr zI
You have been provided with these files, which might be helpful or not: r F interactive)rD lenButton)r
text_inputr rQ s r" log_user_messagezGradioUI.log_user_message sN
'(1, ]]m\no
II%I (
r$ sharec J | j j dd|d| y)z
Launch the Gradio app with the agent interface.
Args:
share (`bool`, defaults to `True`): Whether to share the app publicly.
**kwargs: Additional keyword arguments to pass to the Gradio launch method.
T)debugr N )
create_applaunch)r r kwargss r" r zGradioUI.launch s% ! Ct5CFCr$ c
dd l
j dd 5 }
j i }
j g }
j g }
j 5
j d| j
j
dd j d| j rd | j nd
z
j 5
j d d
j d
ddd }
j dd }d d d | j G
j d }
j ddd }|j | j ||g||g
j! d d d d
j# ddd dd d d dd d d dd d!d"dd d#d$dd g% } j% | j& ||g||g j) | j* || |g| g j)
fd&d ||g |j- | j& ||g|||g j) | j* || |g| g j)
fd'd ||g d d d |S # 1 sw Y YxY w# 1 sw Y xY w# 1 sw Y S xY w)(Nr oceanT)theme fill_heightz# r zw
> This web ui allows you to interact with a `smolagents` agent that can use tools and execute steps to complete tasks.z
**Agent description:**
r z**Your request**) container z Chat MessageFz@Enter your prompt here and press Shift+Enter or press the button)lineslabelr placeholderSubmitprimary)variantz
Upload a file)r z
Upload Status)r r r z
Agentr )Nzhhttps://huggingface.co/datasets/huggingface/documentation-images/resolve/main/smolagents/mascot_smol.pngr z$$)leftrightdisplay$z\[z\]z\(z\))r re
avatar_images
resizeablescalelatex_delimitersc N j dd j d fS NTz:Enter your prompt here and press Shift+Enter or the button)r r r r r rQ s r" z%GradioUI.create_app.. 1 JJ$(6r II$I/ r$ c N j dd j d fS r r r s r" r z%GradioUI.create_app.. r r$ )rD BlocksStateSidebarMarkdownrI r
capitalizer} Groupr r r{ Filechanger HTMLChatbotsubmitr thenr click) r demor stored_messagesr r
submit_btnr
upload_statuschatbotrQ s @r" r zGradioUI.create_app s
YYW$Y
7 T 4 HHRLM hhrlO !xx|
**34??AB OOLPL\L\5d6F6F5GHbdf XXZ HKK 2dKC!# ,"'$f ", "J "$8Y!GJH **6"$'''"@K$&JJ_RWafJ$gM && (($&67&(89 ]7
@ jj "UtD !DUC "UtD "UuE " ! G$
%% -. *j9
d4++ow
-VY`Xabcgcg Z( d
%% -. *j9
d4++ow
-VY`Xabcgcg Z( dWT l SH H
T l s? AI5 A,I) @"gH
DD DYr$ rz rx )F)NFN)%r r( r pathlibr typingr smolagents.agent_typesr r r smolagents.agentsr r smolagents.memoryr
r smolagents.modelsr r
r smolagents.utilsr rM r# r* r. r rY r\ r` rf listrL rx rz __all__r r$ r" r sJ
D D : 9 \ \ 2!
\(A !c !VY ! c c (# # *Sc: Sc4 ScT] Sclc\ ct cXa c."
"
Y "
JEj<&?/&Q Egk E2 $ $#'
D[
>b bJ z
*r$