From 3fca6e5a9792fc3b870569e702d8b8b91efd0dfa Mon Sep 17 00:00:00 2001
From: Ishaan Jaffer <ishaanjaffer0324@gmail.com>
Date: Thu, 19 Feb 2026 11:55:37 -0800
Subject: [PATCH] add model param to suggest templates endpoint

---
 .../proxy/management_endpoints/policy_endpoints/endpoints.py    | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/litellm/proxy/management_endpoints/policy_endpoints/endpoints.py b/litellm/proxy/management_endpoints/policy_endpoints/endpoints.py
index 4aa33f0a473..efecd172d79 100644
--- a/litellm/proxy/management_endpoints/policy_endpoints/endpoints.py
+++ b/litellm/proxy/management_endpoints/policy_endpoints/endpoints.py
@@ -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,
     )