commit

app.py

@@ -5,6 +5,38 @@ import openai
 class OpenAIUtils:
     
     @classmethod
+    def use_openai_chatgpt_base(cls, prompt):
+        """
+        Use OpenAI's GPT-3 model, Davinci, to generate text based on the given prompt
+        """
+        openai.api_key = os.getenv('OPENAI_API_KEY')
+
+        try:
+            # res = openai.ChatCompletion.create(
+            res = openai.Completion.create(
+                model="gpt-3.5-turbo",
+                prompt=prompt,
+                max_tokens=1024,
+                temperature=0.5,
+                messages=[
+                        {"role": "system", "content": "You are a helpful assistant."},
+                        # {"role": "user", "content": "Who won the world series in 2020?"},
+                        # {"role": "assistant", "content": "The Los Angeles Dodgers won the World Series in 2020."},
+                        # {"role": "user", "content": "Where was it played?"}
+                    ]
+                )
+            # res = openai.Completion.create(
+            #     model="text-davinci-003",
+            #     prompt=prompt,
+            #     max_tokens=1024,
+            #     temperature=0.5
+            # )
+            string = res.choices[0].text
+        except openai.error.InvalidRequestError as e:
+            print(f"Error: {e}")
+            return None
+        return string
+    @classmethod
     def use_openai_davinci_base(cls, prompt):
         """
         Use OpenAI's GPT-3 model, Davinci, to generate text based on the given prompt
@@ -43,7 +75,8 @@ class CalqOffer:
 【待遇】
         """
         print(prompt)
-        return OpenAIUtils.use_openai_davinci_base(prompt)
+        # return OpenAIUtils.use_openai_davinci_base(prompt)
+        return OpenAIUtils.use_openai_chatgpt_base(prompt)
 
     def post_process_mail_reply(offer_reply):
         remove_list = ["（箇条書き）"]

requirements.txt

@@ -1 +1 @@
-openai
+openai==0.27.0