Merge pull request #95 from BaranEkin/main

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)