scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-09 03:18:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 109

fixes for model alias

Browse source
This commit is contained in:
Ishaan Jaff 2025-05-19 17:25:33 -07:00
parent 83d647952a
commit 1dcbb331b1
2 changed files with 35 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
ui/litellm-dashboard/src/components/model_dashboard.tsx
View file

@ -1256,7 +1256,8 @@ const ModelDashboard: React.FC<ModelDashboardProps> = ({
{accessToken && (
<ModelAliasManagement
accessToken={accessToken}
availableModels={availableModelGroups}
userID={userID}
userRole={userRole}
onRefresh={handleRefreshClick}
/>
)}

36
ui/litellm-dashboard/src/components/model_dashboard/model_alias_management.tsx
View file

@ -31,24 +31,27 @@ import {
InformationCircleIcon,
RefreshIcon,
} from "@heroicons/react/outline";
import { getRouterSettings, updateRouterSettings } from "../networking";
import { getRouterSettings, updateRouterSettings, modelAvailableCall } from "../networking";
interface ModelAliasManagementProps {
accessToken: string;
availableModels: string[];
onRefresh: () => void;
userID?: string;
userRole?: string;
}
const ModelAliasManagement: React.FC<ModelAliasManagementProps> = ({
accessToken,
availableModels,
onRefresh,
userID,
userRole,
}) => {
const [modelGroupAliases, setModelGroupAliases] = useState<Record<string, string>>({});
const [isLoading, setIsLoading] = useState<boolean>(false);
const [formLoading, setFormLoading] = useState<boolean>(false);
const [addForm] = Form.useForm();
const [editForm] = Form.useForm();
const [availableModels, setAvailableModels] = useState<string[]>([]);
// Modal state
const [isEditModalVisible, setIsEditModalVisible] = useState<boolean>(false);
@ -59,6 +62,33 @@ const ModelAliasManagement: React.FC<ModelAliasManagementProps> = ({
fetchRouterSettings();
}, [accessToken]);
// Fetch available models
useEffect(() => {
fetchAvailableModels();
}, [accessToken, userID, userRole]);
const fetchAvailableModels = async () => {
if (!accessToken || !userID || !userRole) return;
try {
const model_available = await modelAvailableCall(
accessToken,
userID,
userRole
);
const available_model_names = model_available["data"].map(
(element: { id: string }) => element.id
);
console.log("available_model_names for aliases:", available_model_names);
setAvailableModels(available_model_names);
} catch (error) {
console.error("Error fetching available models:", error);
message.error("Failed to load available models");
}
};
const fetchRouterSettings = async () => {
if (!accessToken) return;