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

Fix Extra planning (#362) 

* fix Extra planning
This commit is contained in:
BUI Van Tuan 2025-01-28 09:21:58 +01:00 committed by GitHub
parent 68f3da12ea
commit 3b5c99e87a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 18 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/smolagents/agents.py
View File

@ -265,14 +265,14 @@ class MultiStepAgent:
elif isinstance(step_log, PlanningStep): elif isinstance(step_log, PlanningStep):
thought_message = { thought_message = {
"role": MessageRole.ASSISTANT, "role": MessageRole.ASSISTANT,
"content": "[FACTS LIST]:\n" + step_log.facts.strip(), "content": [{"type": "text", "text": "[FACTS LIST]:\n" + step_log.facts.strip()}],
} }
memory.append(thought_message) memory.append(thought_message)
if not summary_mode: if not summary_mode:
thought_message = { thought_message = {
"role": MessageRole.ASSISTANT, "role": MessageRole.ASSISTANT,
"content": "[PLAN]:\n" + step_log.plan.strip(), "content": [{"type": "text", "text": "[PLAN]:\n" + step_log.plan.strip()}],
} }
memory.append(thought_message) memory.append(thought_message)
@ -700,22 +700,25 @@ Now begin!""",
# Redact updated facts # Redact updated facts
facts_update_system_prompt = { facts_update_system_prompt = {
"role": MessageRole.SYSTEM, "role": MessageRole.SYSTEM,
"content": SYSTEM_PROMPT_FACTS_UPDATE, "content": [{"type": "text", "text": SYSTEM_PROMPT_FACTS_UPDATE}],
} }
facts_update_message = { facts_update_message = {
"role": MessageRole.USER, "role": MessageRole.USER,
"content": USER_PROMPT_FACTS_UPDATE, "content": [{"type": "text", "text": USER_PROMPT_FACTS_UPDATE}],
} }
facts_update = self.model([facts_update_system_prompt] + agent_memory + [facts_update_message]).content facts_update = self.model([facts_update_system_prompt] + agent_memory + [facts_update_message]).content
# Redact updated plan # Redact updated plan
plan_update_message = { plan_update_message = {
"role": MessageRole.SYSTEM, "role": MessageRole.SYSTEM,
"content": SYSTEM_PROMPT_PLAN_UPDATE.format(task=task), "content": [{"type": "text", "text": SYSTEM_PROMPT_PLAN_UPDATE.format(task=task)}],
} }
plan_update_message_user = { plan_update_message_user = {
"role": MessageRole.USER, "role": MessageRole.USER,
"content": USER_PROMPT_PLAN_UPDATE.format( "content": [
{
"type": "text",
"text": USER_PROMPT_PLAN_UPDATE.format(
task=task, task=task,
tool_descriptions=get_tool_descriptions(self.tools, self.tool_description_template), tool_descriptions=get_tool_descriptions(self.tools, self.tool_description_template),
managed_agents_descriptions=(show_agents_descriptions(self.managed_agents)), managed_agents_descriptions=(show_agents_descriptions(self.managed_agents)),
@ -723,6 +726,8 @@ Now begin!""",
remaining_steps=(self.max_steps - step), remaining_steps=(self.max_steps - step),
), ),
} }
],
}
plan_update = self.model( plan_update = self.model(
[plan_update_message] + agent_memory + [plan_update_message_user], [plan_update_message] + agent_memory + [plan_update_message_user],
stop_sequences=["<end_plan>"], stop_sequences=["<end_plan>"],