From 4063483b87518b7c6a1c55b45f5dcb09c179f82e Mon Sep 17 00:00:00 2001 From: Wlad Date: Wed, 12 Apr 2023 10:56:54 +0200 Subject: [PATCH] add my fixes --- scripts/chat.py | 8 ++++---- scripts/main.py | 4 ++-- scripts/speak.py | 12 +++++++++--- 3 files changed, 15 insertions(+), 9 deletions(-) diff --git a/scripts/chat.py b/scripts/chat.py index 23e5b501..91bb2412 100644 --- a/scripts/chat.py +++ b/scripts/chat.py @@ -64,14 +64,14 @@ def chat_with_ai( model = cfg.fast_llm_model # TODO: Change model from hardcode to argument # Reserve 1000 tokens for the response - if cfg.debug: + if cfg.debug_mode: print(f"Token limit: {token_limit}") send_token_limit = token_limit - 1000 - relevant_memory = permanent_memory.get_relevant(str(full_message_history[-5:]), 10) + relevant_memory = permanent_memory.get_relevant(str(full_message_history[-9:]), 10) - if cfg.debug: + if cfg.debug_mode: print('Memory Stats: ', permanent_memory.get_stats()) next_message_to_add_index, current_tokens_used, insertion_index, current_context = generate_context( @@ -110,7 +110,7 @@ def chat_with_ai( # assert tokens_remaining >= 0, "Tokens remaining is negative. This should never happen, please submit a bug report at https://www.github.com/Torantulino/Auto-GPT" # Debug print the current context - if cfg.debug: + if cfg.debug_mode: print(f"Token limit: {token_limit}") print(f"Send Token Count: {current_tokens_used}") print(f"Tokens remaining for response: {tokens_remaining}") diff --git a/scripts/main.py b/scripts/main.py index 4e950d8b..1d878afe 100644 --- a/scripts/main.py +++ b/scripts/main.py @@ -67,7 +67,7 @@ def print_to_console( print() def attempt_to_fix_json_by_finding_outermost_brackets(json_string): - if cfg.speak_mode and cfg.debug: + if cfg.speak_mode and cfg.debug_mode: speak.say_text("I have received an invalid JSON response from the OpenAI API. Trying to fix it now.") print_to_console("Attempting to fix JSON by finding outermost brackets\n", Fore.RED, "") @@ -81,7 +81,7 @@ def attempt_to_fix_json_by_finding_outermost_brackets(json_string): # Extract the valid JSON object from the string json_string = json_match.group(0) print_to_console("Apparently json was fixed.", Fore.GREEN,"") - if cfg.speak_mode and cfg.debug: + if cfg.speak_mode and cfg.debug_mode: speak.say_text("Apparently json was fixed.") else: raise ValueError("No valid JSON object found") diff --git a/scripts/speak.py b/scripts/speak.py index 08b0c1c9..1f636a4d 100644 --- a/scripts/speak.py +++ b/scripts/speak.py @@ -46,15 +46,21 @@ def gtts_speech(text): playsound("speech.mp3", True) os.remove("speech.mp3") -def macos_tts_speech(text): - os.system(f'say "{text}"') +def macos_tts_speech(text, voice_index=0): + if voice_index == 0: + os.system(f'say "{text}"') + else: + if voice_index == 1: + os.system(f'say -v "Ava (Premium)" "{text}"') + else: + os.system(f'say -v Samantha "{text}"') def say_text(text, voice_index=0): def speak(): if not cfg.elevenlabs_api_key: if cfg.use_mac_os_tts == 'True': - macos_tts_speech(text) + macos_tts_speech(text, voice_index) else: gtts_speech(text) else: