aljaz/Auto-GPT
1
0
Fork 0
mirror of https://github.com/aljazceru/Auto-GPT.git synced 2025-12-18 14:34:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

Update chat.py

Browse Source 
Adds code separation between code blocks.
This commit is contained in:
Andres Caicedo
2023-04-04 11:02:01 +02:00
parent 238824553a
commit 500b2b4836
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/chat.py
View File

@@ -73,6 +73,7 @@ def chat_with_ai(
message_to_add = full_message_history[next_message_to_add_index] message_to_add = full_message_history[next_message_to_add_index]
tokens_to_add = token_counter.count_message_tokens([message_to_add], model) tokens_to_add = token_counter.count_message_tokens([message_to_add], model)
if current_tokens_used + tokens_to_add > send_token_limit: if current_tokens_used + tokens_to_add > send_token_limit:
break break
@@ -98,13 +99,16 @@ def chat_with_ai(
print(f"Send Token Count: {current_tokens_used}") print(f"Send Token Count: {current_tokens_used}")
print(f"Tokens remaining for response: {tokens_remaining}") print(f"Tokens remaining for response: {tokens_remaining}")
print("------------ CONTEXT SENT TO AI ---------------") print("------------ CONTEXT SENT TO AI ---------------")
for message in current_context: for message in current_context:
# Skip printing the prompt # Skip printing the prompt
if message["role"] == "system" and message["content"] == prompt: if message["role"] == "system" and message["content"] == prompt:
continue continue
print( print(
f"{message['role'].capitalize()}: {message['content']}") f"{message['role'].capitalize()}: {message['content']}")
print() print()
print("----------- END OF CONTEXT ----------------") print("----------- END OF CONTEXT ----------------")
# TODO: use a model defined elsewhere, so that model can contain temperature and other settings we care about # TODO: use a model defined elsewhere, so that model can contain temperature and other settings we care about