Revert to previous openai version

lm-sys · Nov 25, 2023 · 4d8a4f8 · 4d8a4f8
1 parent 549e809
commit 4d8a4f8
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/fastchat/llm_judge/common.py b/fastchat/llm_judge/common.py
@@ -408,16 +408,16 @@ def chat_compeletion_openai(model, conv, temperature, max_tokens, api_dict=None)
     for _ in range(API_MAX_RETRY):
         try:
             messages = conv.to_openai_api_messages()
-            response = openai.chat.completions.create(
+            response = openai.ChatCompletion.create(
                 model=model,
                 messages=messages,
                 n=1,
                 temperature=temperature,
                 max_tokens=max_tokens,
             )
-            output = response.choices[0].message.content
+            output = response["choices"][0]["message"]["content"]
             break
-        except openai.OpenAIError as e:
+        except openai.error.OpenAIError as e:
             print(type(e), e)
             time.sleep(API_RETRY_SLEEP)