KeyArgo
/
smolagents
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Small tweaks to Gradio demo

This commit is contained in:
Freddy Boulton 2025-01-07 08:04:11 -05:00 committed by GitHub
parent 7b191952f8
commit 7f3a88c597
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/smolagents/gradio_ui.py
View File

@ -23,7 +23,7 @@ from .types import AgentAudio, AgentImage, AgentText, handle_agent_output_types
def pull_messages_from_step(step_log: AgentStep, test_mode: bool = True): def pull_messages_from_step(step_log: AgentStep, test_mode: bool = True):
"""Extract ChatMessage objects from agent steps""" """Extract ChatMessage objects from agent steps"""
if isinstance(step_log, ActionStep): if isinstance(step_log, ActionStep):
yield gr.ChatMessage(role="assistant", content=step_log.llm_output) yield gr.ChatMessage(role="assistant", content=step_log.llm_output or "")
if step_log.tool_call is not None: if step_log.tool_call is not None:
used_code = step_log.tool_call.name == "code interpreter" used_code = step_log.tool_call.name == "code interpreter"
content = step_log.tool_call.arguments content = step_log.tool_call.arguments
@ -36,7 +36,7 @@ def pull_messages_from_step(step_log: AgentStep, test_mode: bool = True):
) )
if step_log.observations is not None: if step_log.observations is not None:
yield gr.ChatMessage( yield gr.ChatMessage(
role="assistant", content=f"```\n{step_log.observations}\n```" role="assistant", content=step_log.observations
) )
if step_log.error is not None: if step_log.error is not None:
yield gr.ChatMessage( yield gr.ChatMessage(
@ -65,7 +65,7 @@ def stream_to_gradio(
if isinstance(final_answer, AgentText): if isinstance(final_answer, AgentText):
yield gr.ChatMessage( yield gr.ChatMessage(
role="assistant", role="assistant",
content=f"**Final answer:**\n```\n{final_answer.to_string()}\n```", content=f"**Final answer:**\n{final_answer.to_string()}\n",
) )
elif isinstance(final_answer, AgentImage): elif isinstance(final_answer, AgentImage):
yield gr.ChatMessage( yield gr.ChatMessage(