Significant-Gravitas
/
AutoGPT
mirror of https://github.com/Significant-Gravitas/AutoGPT.git
1
0
Fork 1
Code Issues Packages Projects Releases Wiki Activity

🐛 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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
autogpt/chat.py
View File

@ -7,10 +7,12 @@ from autogpt.config import Config
from autogpt.llm_utils import create_chat_completion from autogpt.llm_utils import create_chat_completion
from autogpt.logs import logger from autogpt.logs import logger
from plugin_template import Message
cfg = Config() 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. 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 == "": if not plugin_response or plugin_response == "":
continue continue
tokens_to_add = token_counter.count_message_tokens( 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 current_tokens_used + tokens_to_add > send_token_limit:
if cfg.debug_mode: if cfg.debug_mode: