Update shortGPT/gpt/gpt_utils.py

shortGPT/gpt/gpt_utils.py

@@ -8,15 +8,15 @@ import tiktoken
 import yaml
 
 from shortGPT.config.api_db import ApiKeyManager
+openai.base_url="https://api.pawan.krd/cosmosrp/v1"
 
-
-def num_tokens_from_messages(texts, model="cosmosrp"):
+def num_tokens_from_messages(texts, model="gpt-3.5-turbo-0301"):
     """Returns the number of tokens used by a list of messages."""
     try:
         encoding = tiktoken.encoding_for_model(model)
     except KeyError:
         encoding = tiktoken.get_encoding("cl100k_base")
-    if model == "gpt-3.5-turbo-0301":  # note: future models may deviate from this
+    if model != "gpt-3.5-turbo-0301":  # note: future models may deviate from this
         if isinstance(texts, str):
             texts = [texts]
         score = 0
@@ -82,12 +82,12 @@ def gpt3Turbo_completion(chat_prompt="", system="You are an AI that can give the
                     {"role": "system", "content": system},
                     {"role": "user", "content": chat_prompt}
                 ]
-            response = openai.ChatCompletion.create(
+            response = openai.chat.completions.create(
                 model=model,
                 messages=messages,
                 max_tokens=max_tokens,
                 temperature=temp)
-            text = response['choices'][0]['message']['content'].strip()
+            text = response.choices[0].message.content.strip()
             if remove_nl:
                 text = re.sub('\s+', ' ', text)
             filename = '%s_gpt3.txt' % time()
@@ -101,4 +101,4 @@ def gpt3Turbo_completion(chat_prompt="", system="You are an AI that can give the
             if retry >= max_retry:
                 raise Exception("GPT3 error: %s" % oops)
             print('Error communicating with OpenAI:', oops)
-            sleep(1)
+            sleep(1)