scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-09 03:18:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 108

wire up AI suggestion modal in policies page

Browse source
This commit is contained in:
Ishaan Jaffer 2026-02-19 10:59:47 -08:00
parent e432c211ae
commit 38ca018595
1 changed files with 22 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
ui/litellm-dashboard/src/components/policies/index.tsx
View file

@ -13,6 +13,7 @@ import PolicyTestPanel from "./policy_test_panel";
import PolicyTemplates from "./policy_templates";
import GuardrailSelectionModal from "./guardrail_selection_modal";
import TemplateParameterModal from "./template_parameter_modal";
import AiSuggestionModal from "./ai_suggestion_modal";
import {
getPoliciesList,
deletePolicyCall,
@ -63,6 +64,8 @@ const PoliciesPanel: React.FC<PoliciesPanelProps> = ({
const [isParameterModalOpen, setIsParameterModalOpen] = useState(false);
const [isEnrichingTemplate, setIsEnrichingTemplate] = useState(false);
const [pendingTemplate, setPendingTemplate] = useState<any>(null);
const [isAiSuggestionModalOpen, setIsAiSuggestionModalOpen] = useState(false);
const [loadedTemplates, setLoadedTemplates] = useState<any[]>([]);
const isAdmin = userRole ? isAdminRole(userRole) : false;
@ -369,7 +372,12 @@ const PoliciesPanel: React.FC<PoliciesPanelProps> = ({
closable
className="mb-6"
/>
<PolicyTemplates onUseTemplate={handleUseTemplate} accessToken={accessToken} />
<PolicyTemplates
onUseTemplate={handleUseTemplate}
onOpenAiSuggestion={() => setIsAiSuggestionModalOpen(true)}
onTemplatesLoaded={setLoadedTemplates}
accessToken={accessToken}
/>
</TabPanel>
<TabPanel>
@ -569,6 +577,19 @@ const PoliciesPanel: React.FC<PoliciesPanelProps> = ({
</TabPanels>
</TabGroup>
<AiSuggestionModal
visible={isAiSuggestionModalOpen}
onSelectTemplates={(selectedTemplates) => {
setIsAiSuggestionModalOpen(false);
if (selectedTemplates.length > 0) {
handleUseTemplate(selectedTemplates[0]);
}
}}
onCancel={() => setIsAiSuggestionModalOpen(false)}
accessToken={accessToken}
allTemplates={loadedTemplates}
/>
{showFlowBuilder && (
<FlowBuilderPage
onBack={() => {