diff --git a/opencve/commands/imports/cve.py b/opencve/commands/imports/cve.py index efe92f3..4bb50c2 100644 --- 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 index 346e17f..a59efa4 100644 --- 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},