feat: change model

src/forms.py

@@ -352,7 +352,7 @@ def change_request_form(embed_output):
 
             client = OpenAI(api_key=open_api_key)
             response = client.chat.completions.create(
-                model="gpt-4-0125-preview",
+                model="gpt-4o",
                 messages=[
                     {"role": "system", "content": system_prompt},
                     {"role": "user", "content": user_prompt},

src/generate_query_all.py

@@ -1,11 +1,13 @@
-from openai import OpenAI
 import json
 import os
+
 from dotenv import load_dotenv
+from openai import OpenAI
 from openpyxl import load_workbook
 
+
 def make_json(type, subtype, subtype2, items):
-    name = f'{type}_{subtype}_{subtype2}'
+    name = f"{type}_{subtype}_{subtype2}"
 
     with open("data/template.json", "r") as f:
         template = f.read()
@@ -13,7 +15,7 @@ def make_json(type, subtype, subtype2, items):
     template = json.loads(template)
 
     template["name"] = name
-    #print(template)
+    # print(template)
 
     prompt = f"""
     あなたは製薬会社の優秀なQAです。
@@ -33,7 +35,7 @@ def make_json(type, subtype, subtype2, items):
     """
 
     completion = client.chat.completions.create(
-        model="gpt-4-0125-preview",
+        model="gpt-4o",
         messages=[
             {
                 "role": "user",
@@ -44,7 +46,7 @@ def make_json(type, subtype, subtype2, items):
         temperature=0.7,
     )
 
-    #print(completion.choices[0].message.content)
+    # print(completion.choices[0].message.content)
 
     with open(f"data/questions/{name}.json", "w") as f:
         f.write(completion.choices[0].message.content)
@@ -55,24 +57,24 @@ open_api_key = os.environ.get("OPENAI_API_KEY")
 client = OpenAI(api_key=open_api_key)
 
 
-wb = load_workbook('data/変更リスクアセスメント表2023.12.22更新 .xlsx')
-type = '製造方法・製造設備'
+wb = load_workbook("data/変更リスクアセスメント表2023.12.22更新 .xlsx")
+type = "製造方法・製造設備"
 subtype2 = "x"
-sheet = wb[type] 
+sheet = wb[type]
 subtype = "包装工程のパラメーター変更"
 qs = []
 
 for i in range(40, 52):
-    if sheet[f'C{i}'].value == None or sheet[f'C{i}'].value == subtype:
-        qs.append(sheet[f'F{i}'].value)
-        print(sheet[f'F{i}'].value)
+    if sheet[f"C{i}"].value == None or sheet[f"C{i}"].value == subtype:
+        qs.append(sheet[f"F{i}"].value)
+        print(sheet[f"F{i}"].value)
     else:
         make_json(type, subtype, subtype2, qs)
-        print(f'{type}_{subtype}_{subtype2}_completed')
+        print(f"{type}_{subtype}_{subtype2}_completed")
         qs = []
-        qs.append(sheet[f'F{i}'].value)
-        print(sheet[f'F{i}'].value)
-        subtype = sheet[f'C{i}'].value
+        qs.append(sheet[f"F{i}"].value)
+        print(sheet[f"F{i}"].value)
+        subtype = sheet[f"C{i}"].value
 
 make_json(type, subtype, subtype2, qs)
-print(f'{type}_{subtype}_{subtype2}_completed')
+print(f"{type}_{subtype}_{subtype2}_completed")