testing modification of CPE

sudesh0sudesh · Apr 10, 2024 · f96d92b · f96d92b
1 parent 009bf6f
commit f96d92b
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
@@ -46,7 +46,7 @@ def run():
     total_results = 0
     openai_api_key=os.environ.get("OPEN_AI_KEY",None)
     client = OpenAI(api_key=openai_api_key)
-    instruction="Your sole purpose is to extract only a single cpe information in the format 'cpe:2.3:a:vendor:product' from the provided CVE summary without any heading and other information."
+    instruction="Your sole purpose is to extract only a single cpe information in the format 'cpe:2.3:a:vendor:product' from the provided CVE summary; Replace a with h for hardware or o for Operating system based on type of vulnerbility. If there is no match please return 'cpe:2.3:a:::'  Answer without any heading and other information."
 
     while start_index <= total_results:
         url = url_template.format(idx=start_index)

diff --git a/opencve/commands/utils.py b/opencve/commands/utils.py
@@ -55,7 +55,7 @@ def create_change(cls, cve_obj, cve_json, task, events):
     def create_cve(cls, cve_json):
         openai_api_key=os.environ.get("OPEN_AI_KEY",None)
         client = OpenAI(api_key=openai_api_key)
-        instruction="Your sole purpose is to extract only a single cpe information in the format 'cpe:2.3:a:vendor:product' from the provided CVE summary without any heading and other information."
+        instruction="Your sole purpose is to extract only a single cpe information in the format 'cpe:2.3:a:vendor:product' from the provided CVE summary; Replace a with h for hardware or o for Operating system based on type of vulnerbility. If there is no match please return 'cpe:2.3:a:::'  Answer without any heading and other information."
 
         # Takes the CVSS scores
         if "cvssMetricV31" in cve_json["metrics"]: