scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-05 02:41:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 49

feat(ui): pass selectedGuardrails from sidebar to RealtimePlayground WebSocket URL

Browse source
This commit is contained in:
Ishaan Jaffer 2026-02-25 21:30:45 -08:00
parent 596b8794a6
commit ed6a61fb5b
2 changed files with 8 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
ui/litellm-dashboard/src/components/playground/chat_ui/ChatUI.tsx
View file

@ -1832,6 +1832,7 @@ const ChatUI: React.FC<ChatUIProps> = ({
accessToken={apiKeySource === "session" ? accessToken || "" : apiKey}
selectedModel={selectedModel || ""}
customProxyBaseUrl={customProxyBaseUrl || undefined}
selectedGuardrails={selectedGuardrails.length > 0 ? selectedGuardrails : undefined}
/>
) : (
<>

9
ui/litellm-dashboard/src/components/playground/chat_ui/RealtimePlayground.tsx
View file

@ -24,12 +24,14 @@ interface RealtimePlaygroundProps {
accessToken: string;
selectedModel: string;
customProxyBaseUrl?: string;
selectedGuardrails?: string[];
}
const RealtimePlayground: React.FC<RealtimePlaygroundProps> = ({
accessToken,
selectedModel,
customProxyBaseUrl,
selectedGuardrails,
}) => {
const [messages, setMessages] = useState<RealtimeMessage[]>([]);
const [inputText, setInputText] = useState("");
@ -107,7 +109,10 @@ const RealtimePlayground: React.FC<RealtimePlaygroundProps> = ({
const baseUrl = customProxyBaseUrl || getProxyBaseUrl();
const wsBase = baseUrl.replace(/^http/, "ws");
const url = `${wsBase}/v1/realtime?model=${encodeURIComponent(selectedModel)}`;
let url = `${wsBase}/v1/realtime?model=${encodeURIComponent(selectedModel)}`;
if (selectedGuardrails && selectedGuardrails.length > 0) {
url += `&guardrails=${encodeURIComponent(selectedGuardrails.join(","))}`;
}
const ws = new WebSocket(url, ["realtime", `openai-insecure-api-key.${accessToken}`]);
@ -197,7 +202,7 @@ const RealtimePlayground: React.FC<RealtimePlaygroundProps> = ({
addMessage("status", `Connection failed: ${err.message}`);
setIsConnecting(false);
}
}, [accessToken, selectedModel, selectedVoice, customProxyBaseUrl, addMessage, appendAssistantText, playAudioChunk]);
}, [accessToken, selectedModel, selectedVoice, customProxyBaseUrl, selectedGuardrails, addMessage, appendAssistantText, playAudioChunk]);
const disconnect = useCallback(() => {
stopRecording();