File tree 1 file changed +10
-7
lines changed
1 file changed +10
-7
lines changed Original file line number Diff line number Diff line change 4
4
import os
5
5
6
6
load_dotenv ()
7
- openAI_email = os .getenv ("OPENAI_EMAIL" )
8
- openAI_password = os .getenv ("OPENAI_PASSWORD" )
9
- session_token = os .getenv ("SESSION_TOKEN" )
10
- openAI_key = os .getenv ("OPENAI_API_KEY" )
11
- openAI_engine = os .getenv ("OPENAI_ENGINE" )
7
+ OPENAI_EMAIL = os .getenv ("OPENAI_EMAIL" )
8
+ OPENAI_PASSWORD = os .getenv ("OPENAI_PASSWORD" )
9
+ SESSION_TOKEN = os .getenv ("SESSION_TOKEN" )
10
+ OPENAI_API_KEY = os .getenv ("OPENAI_API_KEY" )
11
+ ENGINE = os .getenv ("OPENAI_ENGINE" )
12
+ CHAT_MODEL = os .getenv ("CHAT_MODEL" )
12
13
13
- unofficial_chatbot = AsyncChatbot (config = {"email" :openAI_email , "password" :openAI_password , "session_token" :session_token })
14
- offical_chatbot = Chatbot (api_key = openAI_key )
14
+ if CHAT_MODEL == "UNOFFICIAL" :
15
+ unofficial_chatbot = AsyncChatbot (config = {"email" :OPENAI_EMAIL , "password" :OPENAI_PASSWORD , "session_token" :SESSION_TOKEN })
16
+ elif CHAT_MODEL == "OFFICIAL" :
17
+ offical_chatbot = Chatbot (api_key = OPENAI_API_KEY , engine = ENGINE )
15
18
16
19
async def official_handle_response (message ) -> str :
17
20
responseMessage = offical_chatbot .ask (message )
You can’t perform that action at this time.
0 commit comments