diff --git a/ui/litellm-dashboard/src/components/networking.tsx b/ui/litellm-dashboard/src/components/networking.tsx index 7055f1e44be..0acae1c2e8f 100644 --- a/ui/litellm-dashboard/src/components/networking.tsx +++ b/ui/litellm-dashboard/src/components/networking.tsx @@ -5602,7 +5602,8 @@ export const enrichPolicyTemplateStream = async ( guardrailDefinitions: any[]; }) => void, onError?: (error: string) => void, - options?: { instruction?: string; existingCompetitors?: string[] } + options?: { instruction?: string; existingCompetitors?: string[] }, + onStatus?: (message: string) => void ) => { const url = proxyBaseUrl ? `${proxyBaseUrl}/policy/templates/enrich/stream` @@ -5647,6 +5648,8 @@ export const enrichPolicyTemplateStream = async ( const event = JSON.parse(line.slice(6)); if (event.type === "competitor") { onCompetitor(event.name); + } else if (event.type === "status") { + onStatus?.(event.message); } else if (event.type === "done") { onDone(event); } else if (event.type === "error") {