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

(UI) - allow assigning wildcard models to a team / key (#8041)

Browse source 
* fix message.error

* fix add return_wildcard_routes

* ui edit modelAvailableCall

* fetchAvailableModelsForTeamOrKey

* ui set all models for a team

* ui define common helpers

* edit create key button

* fix viewing model display names

* fix editing team models

* update gitignore

* add jest testing for ui

* Revert "add jest testing for ui"

This reverts commit 98f9a3ebfd.
This commit is contained in:
Ishaan Jaff 2025-01-27 18:06:22 -08:00 • committed by GitHub
parent 3d77e154f2
commit d0cf0a55bb
9 changed files with 142 additions and 41 deletions
Download patch file Download diff file Expand all files Collapse all files

28
ui/litellm-dashboard/src/components/teams.tsx
View file

@ -21,6 +21,7 @@ import {
message,
Tooltip
} from "antd";
import { fetchAvailableModelsForTeamOrKey, getModelDisplayName } from "./key_team_helpers/fetch_available_models_team_key";
import { Select, SelectItem } from "@tremor/react";
import { InfoCircleOutlined } from '@ant-design/icons';
import { getGuardrailsList } from "./networking";
@ -134,7 +135,7 @@ const Team: React.FC<TeamProps> = ({
const [isTeamModalVisible, setIsTeamModalVisible] = useState(false);
const [isAddMemberModalVisible, setIsAddMemberModalVisible] = useState(false);
const [isEditMemberModalVisible, setIsEditMemberModalVisible] = useState(false);
const [userModels, setUserModels] = useState([]);
const [userModels, setUserModels] = useState<string[]>([]);
const [isDeleteModalOpen, setIsDeleteModalOpen] = useState(false);
const [teamToDelete, setTeamToDelete] = useState<string | null>(null);
const [selectedEditMember, setSelectedEditMember] = useState<null | TeamMember>(null);
@ -236,7 +237,7 @@ const Team: React.FC<TeamProps> = ({
{userModels &&
userModels.map((model) => (
<Select2.Option key={model} value={model}>
{model}
{getModelDisplayName(model)}
</Select2.Option>
))}
</Select2>
@ -405,21 +406,12 @@ const Team: React.FC<TeamProps> = ({
useEffect(() => {
const fetchUserModels = async () => {
try {
if (userID === null || userRole === null) {
if (userID === null || userRole === null || accessToken === null) {
return;
}
if (accessToken !== 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);
const models = await fetchAvailableModelsForTeamOrKey(userID, userRole, accessToken);
if (models) {
setUserModels(models);
}
} catch (error) {
console.error("Error fetching user models:", error);
@ -715,8 +707,8 @@ const Team: React.FC<TeamProps> = ({
>
<Text>
{model.length > 30
? `${model.slice(0, 30)}...`
: model}
? `${getModelDisplayName(model).slice(0, 30)}...`
: getModelDisplayName(model)}
</Text>
</Badge>
)
@ -875,7 +867,7 @@ const Team: React.FC<TeamProps> = ({
</Select2.Option>
{userModels.map((model) => (
<Select2.Option key={model} value={model}>
{model}
{getModelDisplayName(model)}
</Select2.Option>
))}
</Select2>