Fix ai name removal

coronasafe · Apr 18, 2024 · 55a526d · 55a526d
1 parent 713afed
commit 55a526d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ayushma/utils/openaiapi.py b/ayushma/utils/openaiapi.py
@@ -345,7 +345,7 @@ def converse(
         response = lang_chain_helper.get_response(
             english_text, reference, chat_history, documents
         )
-        chat_response = response.replace("Ayushma:", "").lstrip()
+        chat_response = response.replace(f"{AI_NAME}:", "").lstrip()
         stats["response_end_time"] = time.time()
         translated_chat_response, url, chat_message = handle_post_response(
             chat_response,
@@ -427,7 +427,7 @@ def converse(
                         break
 
                     chat_response += next_token[0]
-                    chat_response = chat_response.replace(f"{AI_NAME}: ", "")
+                    chat_response = chat_response.replace(f"{AI_NAME}:", "").lstrip()
                     yield create_json_response(
                         local_translated_text,
                         chat.external_id,