scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-11 22:51:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 154

fix(key_edit_view.tsx): fix showing available models on key edit for non-team key

Browse source
This commit is contained in:
Krrish Dholakia 2025-03-14 20:48:30 -07:00
parent 6ed995952f
commit a93ef15b55
1 changed files with 17 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
ui/litellm-dashboard/src/components/key_edit_view.tsx
View file

@ -2,7 +2,7 @@ import React, { useState, useEffect } from "react";
import { Form, Input, InputNumber, Select } from "antd";
import { Button, TextInput } from "@tremor/react";
import { KeyResponse } from "./key_team_helpers/key_list";
import { getTeamModels } from "../components/create_key_button";
import { fetchTeamModels } from "../components/create_key_button";
import { modelAvailableCall } from "./networking";
interface KeyEditViewProps {
@ -45,31 +45,36 @@ export function KeyEditView({
const [form] = Form.useForm();
const [userModels, setUserModels] = useState<string[]>([]);
const team = teams?.find(team => team.team_id === keyData.team_id);
const availableModels = getTeamModels(team, userModels);
const [availableModels, setAvailableModels] = useState<string[]>([]);
useEffect(() => {
const fetchUserModels = async () => {
const fetchModels = async () => {
if (!userID || !userRole || !accessToken) return;
try {
if (accessToken && userID && userRole) {
if (keyData.team_id === null) {
// Fetch user models if no team
const model_available = await modelAvailableCall(
accessToken,
userID,
userID,
userRole
);
let available_model_names = model_available["data"].map(
const available_model_names = model_available["data"].map(
(element: { id: string }) => element.id
);
console.log("available_model_names:", available_model_names);
setUserModels(available_model_names);
setAvailableModels(available_model_names);
} else if (team?.team_id) {
// Fetch team models if team exists
const models = await fetchTeamModels(userID, userRole, accessToken, team.team_id);
setAvailableModels(Array.from(new Set([...team.models, ...models])));
}
} catch (error) {
console.error("Error fetching user models:", error);
console.error("Error fetching models:", error);
}
};
fetchUserModels();
}, []);
fetchModels();
}, [userID, userRole, accessToken, team, keyData.team_id]);
// Convert API budget duration to form format
const getBudgetDuration = (duration: string | null) => {