mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-09 22:31:41 +00:00
add model param to suggest templates endpoint
This commit is contained in:
parent
72f58ef316
commit
3fca6e5a97
1 changed files with 2 additions and 0 deletions
|
|
@ -962,6 +962,7 @@ def _build_comparison_blocked_words(
|
|||
class SuggestTemplatesRequest(BaseModel):
|
||||
attack_examples: List[str] = Field(default_factory=list)
|
||||
description: str = Field(default="")
|
||||
model: Optional[str] = None
|
||||
|
||||
|
||||
@router.post(
|
||||
|
|
@ -990,4 +991,5 @@ async def suggest_policy_templates(
|
|||
templates=templates,
|
||||
attack_examples=data.attack_examples,
|
||||
description=data.description,
|
||||
model=data.model,
|
||||
)
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue