From d02093dc2465625920d7767471353d7d3695d258 Mon Sep 17 00:00:00 2001 From: Albert Villanova del Moral <8515462+albertvillanova@users.noreply.github.com> Date: Fri, 14 Feb 2025 11:20:53 +0100 Subject: [PATCH] Fix PlanningPromptTemplate and its docstring (#648) --- src/smolagents/agents.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/src/smolagents/agents.py b/src/smolagents/agents.py index 1ce54d1..1fac1d7 100644 --- a/src/smolagents/agents.py +++ b/src/smolagents/agents.py @@ -91,8 +91,7 @@ class PlanningPromptTemplate(TypedDict): Prompt templates for the planning step. Args: - initial_facts_pre_task (`str`): Initial facts pre-task prompt. - initial_facts_task (`str`): Initial facts task prompt. + initial_facts (`str`): Initial facts prompt. initial_plan (`str`): Initial plan prompt. update_facts_pre_messages (`str`): Update facts pre-messages prompt. update_facts_post_messages (`str`): Update facts post-messages prompt. @@ -100,8 +99,7 @@ class PlanningPromptTemplate(TypedDict): update_plan_post_messages (`str`): Update plan post-messages prompt. """ - initial_facts_pre_task: str - initial_facts_task: str + initial_facts: str initial_plan: str update_facts_pre_messages: str update_facts_post_messages: str