CalvinBackup/llmsecops-research
1
0
Fork 0
mirror of https://github.com/lightbroker/llmsecops-research.git synced 2026-09-24 02:30:56 +02:00
Code Issues Packages Projects Releases Wiki Activity

avoid unclosed curly braces

Browse Source
This commit is contained in:
Adam Wilson
2025-07-26 15:50:10 -06:00
parent 16ba9c15ee
commit 27dad236ef
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/text_generation/adapters/prompt_injection_example_repository.py
+2 -2
View File
@@ -44,13 +44,13 @@ class PromptInjectionExampleRepository(AbstractPromptInjectionExampleRepository)
prompt_keys = ['prompt_injection_prompt'] prompt_keys = ['prompt_injection_prompt']
for key in prompt_keys: for key in prompt_keys:
if key in item: if key in item:
extracted['prompt_injection_prompt'] = item[key] extracted['prompt_injection_prompt'] = item[key].replace('{', '').replace('}', '')
break break
completion_keys = ['prompt_injection_completion'] completion_keys = ['prompt_injection_completion']
for key in completion_keys: for key in completion_keys:
if key in item: if key in item:
extracted['prompt_injection_completion'] = item[key] extracted['prompt_injection_completion'] = item[key].replace('{', '').replace('}', '')
break break
if 'prompt_injection_prompt' in extracted and 'prompt_injection_completion' in extracted: if 'prompt_injection_prompt' in extracted and 'prompt_injection_completion' in extracted: