From c0c898faeb3514181db633b9023abbedc87311b1 Mon Sep 17 00:00:00 2001 From: Collin Dutter Date: Fri, 20 Dec 2024 13:55:44 -0800 Subject: [PATCH] WIP --- griptape/tasks/prompt_task.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/griptape/tasks/prompt_task.py b/griptape/tasks/prompt_task.py index 4ed313bf0..a2fe970e6 100644 --- a/griptape/tasks/prompt_task.py +++ b/griptape/tasks/prompt_task.py @@ -20,7 +20,6 @@ if TYPE_CHECKING: from schema import Schema - from griptape.drivers import BasePromptDriver from griptape.memory import TaskMemory from griptape.memory.structure.base_conversation_memory import BaseConversationMemory from griptape.structures import Structure @@ -35,7 +34,7 @@ class PromptTask(BaseTask, RuleMixin, ActionsSubtaskOriginMixin): # Stop sequence for chain-of-thought in the framework. Using this "token-like" string to make it more unique, # so that it doesn't trigger on accident. RESPONSE_STOP_SEQUENCE = "<|Response|>" - prompt_driver: BasePromptDriver = field( + prompt_driver: str = field( default=Factory(lambda: Defaults.drivers_config.prompt_driver), kw_only=True, metadata={"serializable": True} ) generate_system_template: Callable[[PromptTask], str] = field(