test
This commit is contained in:
@@ -70,12 +70,11 @@ async def generate_placeholders(model: Model, censored_words: Set[str]) -> List[
|
|||||||
placeholder_agent = Agent(
|
placeholder_agent = Agent(
|
||||||
model,
|
model,
|
||||||
output_type=PlaceholderMapping,
|
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
|
# Convert set to sorted list for consistent ordering
|
||||||
words_list = sorted(list(censored_words))
|
words_list = sorted(list(censored_words))
|
||||||
|
|
||||||
result = await placeholder_agent.run(words_list)
|
result = await placeholder_agent.run(words_list)
|
||||||
return result.output.mappings
|
return result.output.mappings
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user