change openai

.idea/Data-Copilot.iml

@@ -5,4 +5,8 @@
     <orderEntry type="inheritedJdk" />
     <orderEntry type="sourceFolder" forTests="false" />
   </component>
+  <component name="PyDocumentationSettings">
+    <option name="format" value="GOOGLE" />
+    <option name="myDocStringFormat" value="Google" />
+  </component>
 </module>

app.py

@@ -50,8 +50,8 @@ class Client:
     def run(self, messages):
 
             # self.stop = False
-        self.OPENAI_KEY = '' #os.getenv('OPENAI_API')
-        self.OPENAI_API_KEY_AZURE = os.getenv('OPENAI_API_KEY_AZURE')
+        self.OPENAI_KEY = os.getenv('OPENAI_API')
+        self.OPENAI_API_KEY_AZURE = ''#os.getenv('OPENAI_API_KEY_AZURE')
         self.OPENAI_API_BASED_AZURE = os.getenv('OPENAI_API_BASED_AZURE')
         self.OPENAI_ENGINE_AZURE = os.getenv('OPENAI_ENGINE_AZURE')
         gen = gradio_interface(messages, self.OPENAI_KEY, self.OPENAI_API_KEY_AZURE, self.OPENAI_API_BASED_AZURE, self.OPENAI_ENGINE_AZURE)

lab_gpt4_call.py

@@ -148,6 +148,7 @@ def send_official_call(query, openai_key='', api_base='', engine=''):
     start = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(start))
     print(start)
     openai.api_key  = openai_key
+    openai.api_base = api_base
 
     response = openai.ChatCompletion.create(
         # engine="gpt35",

main.py

@@ -370,7 +370,7 @@ def gradio_interface(query, openai_key, openai_key_azure, api_base, engine):
     try:
         if openai_key.startswith('sk') and openai_key_azure == '':
             print('send_official_call')
-            thread = MyThread(target=run, args=(query, add_to_queue, send_official_call, openai_key))
+            thread = MyThread(target=run, args=(query, add_to_queue, send_official_call, openai_key, api_base))
         elif openai_key =='' and len(openai_key_azure)>0:
             print('send_chat_request_Azure')
             thread = MyThread(target=run, args=(query, add_to_queue, send_chat_request_Azure, openai_key_azure, api_base, engine))