chore: refactor

Zero6992 · Mar 14, 2023 · c8b1ba5 · c8b1ba5
1 parent 1b6462f
commit c8b1ba5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/responses.py b/src/responses.py
@@ -22,7 +22,6 @@ def get_chatbot_model(model_name: str) -> Union[AsyncChatbot, Chatbot]:
         return AsyncChatbot(config={"email": openai_email, "password": openai_password, "session_token": session_token})
     elif model_name == "OFFICIAL":
         openai_api_key = os.getenv("OPENAI_API_KEY")
-        print(openai_api_key)
         engine = os.getenv("OPENAI_ENGINE")
         return Chatbot(api_key=openai_api_key, engine=engine)
 
@@ -39,6 +38,7 @@ async def unofficial_handle_response(message) -> str:
 
 # resets conversation and asks chatGPT the prompt for a persona
 async def switch_persona(persona) -> None:
+    CHAT_MODEL = os.getenv("CHAT_MODEL")
     if CHAT_MODEL ==  "UNOFFICIAL":
         chatbot.reset_chat()
         async for response in chatbot.ask(personas.PERSONAS.get(persona)):