diff --git a/src/pydantic_censoring.py b/src/pydantic_censoring.py index edddb1d..c0145ba 100644 --- a/src/pydantic_censoring.py +++ b/src/pydantic_censoring.py @@ -70,12 +70,11 @@ async def generate_placeholders(model: Model, censored_words: Set[str]) -> List[ placeholder_agent = Agent( model, output_type=PlaceholderMapping, - system_prompt=PromptManager.get_prompt('generate_placeholders', {}) + system_prompt=PromptManager.get_prompt('generate_placeholders') ) # Convert set to sorted list for consistent ordering words_list = sorted(list(censored_words)) - result = await placeholder_agent.run(words_list) return result.output.mappings