phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 03:04:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

(Admin UI) - Test Key Tab - Allow using UI Session instead of manually creating a virtual key (#7348)

Browse source 
* ui fix - allow searching model list + fix bug on filtering

* qa fix - use correct provider name for azure_text

* ui wrap content onto next line

* ui fix - allow selecting current UI session when logging in

* ui session budgets
This commit is contained in:
Ishaan Jaff 2024-12-21 13:14:15 -08:00 • committed by GitHub
parent b52783445e
commit ce41cd977c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 41 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

33
ui/litellm-dashboard/src/components/chat_ui.tsx
View file

@ -88,6 +88,7 @@ const ChatUI: React.FC<ChatUIProps> = ({
userRole,
userID,
}) => {
const [apiKeySource, setApiKeySource] = useState<'session' | 'custom'>('session');
const [apiKey, setApiKey] = useState("");
const [inputMessage, setInputMessage] = useState("");
const [chatHistory, setChatHistory] = useState<any[]>([]);
@ -167,7 +168,14 @@ const ChatUI: React.FC<ChatUIProps> = ({
const handleSendMessage = async () => {
if (inputMessage.trim() === "") return;
if (!apiKey || !token || !userRole || !userID) {
if (!token || !userRole || !userID) {
return;
}
const effectiveApiKey = apiKeySource === 'session' ? accessToken : apiKey;
if (!effectiveApiKey) {
message.error("Please provide an API key or select Current UI Session");
return;
}
@ -182,7 +190,7 @@ const ChatUI: React.FC<ChatUIProps> = ({
inputMessage,
(chunk) => updateUI("assistant", chunk),
selectedModel,
apiKey
effectiveApiKey
);
}
} catch (error) {
@ -223,8 +231,25 @@ const ChatUI: React.FC<ChatUIProps> = ({
<div className="sm:max-w-2xl">
<Grid numItems={2}>
<Col>
<Text>API Key</Text>
<TextInput placeholder="Type API Key here" type="password" onValueChange={setApiKey} value={apiKey}/>
<Text>API Key Source</Text>
<Select
defaultValue="session"
style={{ width: "100%" }}
onChange={(value) => setApiKeySource(value as "session" | "custom")}
options={[
{ value: 'session', label: 'Current UI Session' },
{ value: 'custom', label: 'Virtual Key' },
]}
/>
{apiKeySource === 'custom' && (
<TextInput
className="mt-2"
placeholder="Enter custom API key"
type="password"
onValueChange={setApiKey}
value={apiKey}
/>
)}
</Col>
<Col className="mx-2">
<Text>Select Model:</Text>