feat: update OpenAI lib and support for gpt-4o-mini

iarahealth · Aug 22, 2024 · bf29b5a · bf29b5a
1 parent e90d120
commit bf29b5a
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -3,7 +3,7 @@ deep-translator==1.11.4
 transformers==4.9.1
 sentencepiece==0.1.99
 tenacity==8.2.3
-openai==1.3.5
+openai==1.42.0
 nlpaug==1.1.11
 nltk==3.6.6
 num2words==0.5.13

diff --git a/run_txt_gen.py b/run_txt_gen.py
@@ -47,7 +47,7 @@
  parser.add_argument(
  "--model",
  type=str,
- default="gpt-3.5-turbo-0125",
+ default="gpt-4o-mini",
  help="ChatGPT model to use",
  )
  parser.add_argument(

diff --git a/text/gen.py b/text/gen.py
@@ -13,13 +13,14 @@
  # https://platform.openai.com/docs/models/gpt-3-5
  "gpt-4-turbo": 128000,
  "gpt-3.5-turbo-0125": 16385,
+ "gpt-4o-mini": 128000,
 }
 
 
 # https://github.com/openai/openai-cookbook/blob/main/examples/How_to_handle_rate_limits.ipynb
 @retry(wait=wait_random_exponential(min=1, max=60), stop=stop_after_attempt(6))
 def make_chatgpt_query(
- client, query: str, return_type: str, model: str = "gpt-3.5-turbo-0125"
+ client, query: str, return_type: str, model: str = "gpt-4o-mini"
 ) -> List[str]:
  """
  Makes a query to the ChatGPT model and returns the generated response.