🐛 Fix on_planning

pull/2531/head
Taylor Beeston 2023-04-17 12:41:17 -07:00
parent dea5000a01
commit d23ada30d7
No known key found for this signature in database
GPG Key ID: 5627A4571915CEE9
1 changed files with 4 additions and 2 deletions

View File

@ -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: