From 27dad236ef94ce1617dcd4ec4c2cbde76c2cba0c Mon Sep 17 00:00:00 2001 From: Adam Wilson Date: Sat, 26 Jul 2025 15:50:10 -0600 Subject: [PATCH] avoid unclosed curly braces --- .../adapters/prompt_injection_example_repository.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/text_generation/adapters/prompt_injection_example_repository.py b/src/text_generation/adapters/prompt_injection_example_repository.py index b36eb5ee6..bbec202fa 100644 --- a/src/text_generation/adapters/prompt_injection_example_repository.py +++ b/src/text_generation/adapters/prompt_injection_example_repository.py @@ -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: