From d23ada30d72fe44467770d82389243cc3f7cb254 Mon Sep 17 00:00:00 2001 From: Taylor Beeston Date: Mon, 17 Apr 2023 12:41:17 -0700 Subject: [PATCH] :bug: Fix on_planning --- autogpt/chat.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/autogpt/chat.py b/autogpt/chat.py index 22fe636c..e7354fc1 100644 --- a/autogpt/chat.py +++ b/autogpt/chat.py @@ -7,10 +7,12 @@ from autogpt.config import Config from autogpt.llm_utils import create_chat_completion from autogpt.logs import logger +from plugin_template import Message + cfg = Config() -def create_chat_message(role, content): +def create_chat_message(role, content) -> Message: """ Create a chat message with the given role and content. @@ -145,7 +147,7 @@ def chat_with_ai( if not plugin_response or plugin_response == "": continue tokens_to_add = token_counter.count_message_tokens( - [plugin_response], model + [create_chat_message("system", plugin_response)], model ) if current_tokens_used + tokens_to_add > send_token_limit: if cfg.debug_mode: