mirror of
https://github.com/lightbroker/llmsecops-research.git
synced 2026-10-05 15:56:58 +02:00
avoid unclosed curly braces
This commit is contained in:
1 parent
16ba9c15ee
commit
27dad236ef
1 file changed
+2
-2
@@ -44,13 +44,13 @@ class PromptInjectionExampleRepository(AbstractPromptInjectionExampleRepository)
|
||||
prompt_keys = ['prompt_injection_prompt']
|
||||
for key in prompt_keys:
|
||||
if key in item:
|
||||
extracted['prompt_injection_prompt'] = item[key]
|
||||
extracted['prompt_injection_prompt'] = item[key].replace('{', '').replace('}', '')
|
||||
break
|
||||
|
||||
completion_keys = ['prompt_injection_completion']
|
||||
for key in completion_keys:
|
||||
if key in item:
|
||||
extracted['prompt_injection_completion'] = item[key]
|
||||
extracted['prompt_injection_completion'] = item[key].replace('{', '').replace('}', '')
|
||||
break
|
||||
|
||||
if 'prompt_injection_prompt' in extracted and 'prompt_injection_completion' in extracted:
|
||||
|
||||
Reference in new issue
Block a user