fix bugs

agents/legacy/jar3d_agent.py

@@ -100,7 +100,7 @@ def set_chat_finished(state: State) -> bool:
         final_response_formatted = final_response_formatted.rstrip()
         
         # print(colored(f"\n\n DEBUG: {final_response_formatted}\n\n", 'green'))
-        print(colored(f"\n\n Jar3d👩‍💻: {final_response_formatted}", 'cyan'))
+        print(colored(f"\n\n Jarvis👩‍💻: {final_response_formatted}", 'cyan'))
         state["final_answer"] = f'''{final_response_formatted}'''
     else:
         print(colored("Error: Could not find '>> FINAL ANSWER:' in the response", 'red'))

agents/legacy/jar3d_agent_backup.py

@@ -102,7 +102,7 @@ def set_chat_finished(state: State) -> bool:
         final_response_formatted = final_response_formatted.rstrip()
         
         # print(colored(f"\n\n DEBUG: {final_response_formatted}\n\n", 'green'))
-        print(colored(f"\n\n Jar3d👩‍💻: {final_response_formatted}", 'cyan'))
+        print(colored(f"\n\n Jarvis👩‍💻: {final_response_formatted}", 'cyan'))
         state["final_answer"] = f'''{final_response_formatted}'''
     else:
         print(colored("Error: Could not find '>> FINAL ANSWER:' in the response", 'red'))

models/llms.py

@@ -8,6 +8,7 @@ from utils.logging import log_function, setup_logging
 from tenacity import retry, stop_after_attempt, wait_fixed, retry_if_exception_type
 from config.load_configs import load_config
 from dotenv import load_dotenv
+from requests.packages.urllib3.util.retry import Retry
 load_dotenv()
 
 setup_logging(level=logging.DEBUG)