Update aeo_ex_generator/aeo_example_generator.py

aeo_ex_generator/aeo_example_generator.py

@@ -342,10 +342,15 @@ If namespace "uco-role" prefix is used then https://ontology.unifiedcyberontolog
         return string.split(":")[0]
     
     
-    def auto_generate(self,planSize:int=3,keywords:str="",nkeywords:str=""):
-        p = f"Generate a deception plan with {planSize} PlannedEvents."
+    def auto_generate(self,planSize:int=3,keywords:str="",nkeywords:str="",contextPrompt:str=""):
+        p = ""
+        if contextPrompt:
+            p += f"Given the following context: {contextPrompt}."
+        p += f"Generate a deception plan with {planSize} PlannedEvents."
         if keywords:
             p+= f"The plan must include specifically {keywords}."
+        else:
+            p+=f"Make sure to include all deception objects in the context."
         if nkeywords:
             p+= f"Do not include {nkeywords}."
         #deception plan description