Update openai chat completions api

challenge/gpt_eval.py

@@ -1,23 +1,23 @@
-import openai
 import pickle
 import pdb
 import numpy as np
 import torch
 import json
 import argparse
 from multiprocessing import Pool
+from openai import OpenAI
 
 
 class GPTEvaluation:
  def __init__(self):
- openai.api_key = "you need to use your own openai key for evaluation on your local machine"
+ self.client = OpenAI(api_key="you need to use your own openai key for evaluation on your local machine")
 
  def call_chatgpt(self, chatgpt_messages, max_tokens=40, model="gpt-3.5-turbo"):
- response = openai.ChatCompletion.create(
+ response = self.client.chat.completions.create(
  model=model, messages=chatgpt_messages, temperature=0.6, max_tokens=max_tokens
  )
- reply = response["choices"][0]["message"]["content"]
- total_tokens = response["usage"]["total_tokens"]
+ reply = response.choices[0].message.content
+ total_tokens = response.usage.total_tokens
  return reply, total_tokens
 
  def prepare_chatgpt_message(self, prompt):
@@ -55,4 +55,4 @@ def forward(self, data):
  with Pool(5) as p: # Change the number based on your CPU cores
  scores = p.map(eval.forward, data)
 
- print(scores)
+ print(scores)