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

Restore missing user prompt for initial facts (#576) 

Co-authored-by: Albert Villanova del Moral <8515462+albertvillanova@users.noreply.github.com>
This commit is contained in:
NeuroWhAI 2025-02-10 18:00:02 +09:00 committed by GitHub
parent 3a90bd7164
commit 75b2a10fbc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 19 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
src/smolagents/agents.py
View File

@ -459,7 +459,20 @@ You have been provided with these additional arguments, that you can access usin
"role": MessageRole.SYSTEM,
"content": [{"type": "text", "text": self.prompt_templates["planning"]["initial_facts"]}],
}
input_messages = [message_prompt_facts]
message_prompt_task = {
"role": MessageRole.USER,
"content": [
{
"type": "text",
"text": f"""Here is the task:
```
{task}
```
Now begin!""",
}
],
}
input_messages = [message_prompt_facts, message_prompt_task]
chat_message_facts: ChatMessage = self.model(input_messages)
answer_facts = chat_message_facts.content

8
tests/test_agents.py
View File

@ -703,12 +703,14 @@ class TestMultiStepAgent:
assert isinstance(planning_step, PlanningStep)
messages = planning_step.model_input_messages
assert isinstance(messages, list)
assert len(messages) == 1
for message in messages:
assert len(messages) == 2
expected_roles = [MessageRole.SYSTEM, MessageRole.USER]
for i, message in enumerate(messages):
assert isinstance(message, dict)
assert "role" in message
assert "content" in message
assert isinstance(message["role"], MessageRole)
assert message["role"] == expected_roles[i]
assert isinstance(message["content"], list)
assert len(message["content"]) == 1
for content in message["content"]:
@ -721,7 +723,7 @@ class TestMultiStepAgent:
assert len(call_args.args) == 1
messages = call_args.args[0]
assert isinstance(messages, list)
assert len(messages) == 1
# assert len(messages) == 1 # TODO
for message in messages:
assert isinstance(message, dict)
assert "role" in message