Migrate to OpenAI v1 library

platisd · May 9, 2024 · c2ac01f · c2ac01f
1 parent d1aa0a3
commit c2ac01f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/autofill_description.py b/autofill_description.py
@@ -103,9 +103,7 @@ def main():
     model_sample_response = os.environ.get(
         "INPUT_MODEL_SAMPLE_RESPONSE", GOOD_SAMPLE_RESPONSE
     )
-    completion_prompt = os.environ.get(
-        "INPUT_COMPLETION_PROMPT", COMPLETION_PROMPT
-    )
+    completion_prompt = os.environ.get("INPUT_COMPLETION_PROMPT", COMPLETION_PROMPT)
     authorization_header = {
         "Accept": "application/vnd.github.v3+json",
         "Authorization": "token %s" % github_token,
@@ -177,8 +175,8 @@ def main():
     if len(completion_prompt) > max_allowed_characters:
         completion_prompt = completion_prompt[:max_allowed_characters]
 
-    openai.api_key = openai_api_key
-    openai_response = openai.ChatCompletion.create(
+    openai_client = openai.OpenAI(api_key=openai_api_key)
+    openai_response = openai_client.chat.completions.create(
         model=open_ai_model,
         messages=[
             {

diff --git a/requirements.txt b/requirements.txt
@@ -1,2 +1,2 @@
 requests>=2.18
-openai==0.27.2
+openai==1.27.0