openai model change

sudesh0sudesh · Aug 6, 2024 · 684a591 · 684a591
1 parent 91d8008
commit 684a591
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/opencve/commands/imports/cve.py b/opencve/commands/imports/cve.py
@@ -110,7 +110,7 @@ def run():
                         prompt = f"CVE SUMMARY:{summary}"
 
                         completion = client.chat.completions.create(
-                            model="gpt-3.5-turbo-0125",
+                            model="gpt-4o-mini-2024-07-18",
                             messages=[
                                 {"role": "system", "content": instruction},
                                 {"role": "user", "content": prompt},

diff --git a/opencve/commands/utils.py b/opencve/commands/utils.py
@@ -94,7 +94,7 @@ def create_cve(cls, cve_json):
                 prompt = f"CVE SUMMARY:{summary}"
 
                 completion = client.chat.completions.create(
-                    model="gpt-3.5-turbo-0125",
+                    model="gpt-4o-mini-2024-07-18",
                     messages=[
                         {"role": "system", "content": instruction},
                         {"role": "user", "content": prompt},