phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

Merge pull request #2828 from BerriAI/ui_fix_edit_key_showing_models

Browse source 
(ui) fix bug when editing keys - selecting models
This commit is contained in:
Ishaan Jaff 2024-04-03 20:52:43 -07:00 • committed by GitHub
commit e154c5d834
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 31 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ui/litellm-dashboard/src/components/user_dashboard.tsx
View file

@ -217,10 +217,10 @@ const UserDashboard: React.FC<UserDashboardProps> = ({
<ViewKeyTable
userID={userID}
userRole={userRole}
accessToken={accessToken}
selectedTeam={selectedTeam ? selectedTeam : null}
data={keys}
userModels={userModels}
setData={setKeys}
/>
<CreateKey

39
ui/litellm-dashboard/src/components/view_key_table.tsx
View file

@ -1,6 +1,6 @@
"use client";
import React, { useEffect, useState } from "react";
import { keyDeleteCall } from "./networking";
import { keyDeleteCall, modelAvailableCall } from "./networking";
import { InformationCircleIcon, StatusOnlineIcon, TrashIcon, PencilAltIcon } from "@heroicons/react/outline";
import { keySpendLogsCall, PredictedSpendLogsCall, keyUpdateCall } from "./networking";
import {
@ -47,9 +47,9 @@ interface EditKeyModalProps {
// Define the props type
interface ViewKeyTableProps {
userID: string;
userRole: string | null;
accessToken: string;
selectedTeam: any | null;
userModels: string[];
data: any[] | null;
setData: React.Dispatch<React.SetStateAction<any[] | null>>;
}
@ -73,9 +73,9 @@ interface ItemData {
const ViewKeyTable: React.FC<ViewKeyTableProps> = ({
userID,
userRole,
accessToken,
selectedTeam,
userModels,
data,
setData,
}) => {
@ -91,18 +91,39 @@ const ViewKeyTable: React.FC<ViewKeyTableProps> = ({
const [editModalVisible, setEditModalVisible] = useState(false);
const [selectedToken, setSelectedToken] = useState<ItemData | null>(null);
const [userModels, setUserModels] = useState([]);
useEffect(() => {
const fetchUserModels = async () => {
try {
if (userID === null) {
return;
}
if (accessToken !== null && userRole !== null) {
const model_available = await modelAvailableCall(accessToken, userID, userRole);
let available_model_names = model_available["data"].map(
(element: { id: string }) => element.id
);
console.log("available_model_names:", available_model_names);
setUserModels(available_model_names);
}
} catch (error) {
console.error("Error fetching user models:", error);
}
};
fetchUserModels();
}, [accessToken, userID, userRole]);
const EditKeyModal: React.FC<EditKeyModalProps> = ({ visible, onCancel, token, onSubmit }) => {
const [form] = Form.useForm();
// check token.models length == 0
if (token.models.length == 0 && selectedTeam) {
token.models = selectedTeam.models;
}
console.log("in edit key modal:", token);
console.log("in edit key modal, team:", selectedTeam);
const handleOk = () => {
form
.validateFields()