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

ui use all_models alias

Browse source
This commit is contained in:
Ishaan Jaff 2024-04-02 15:01:42 -07:00
parent 7946c68f47
commit d46f77fd58
3 changed files with 7 additions and 36 deletions
Download patch file Download diff file Expand all files Collapse all files

16
ui/litellm-dashboard/src/components/create_key_button.tsx
View file

@ -91,19 +91,6 @@ const CreateKey: React.FC<CreateKeyProps> = ({
}
};
const handleModelSelection = (selectedModels: string[]) => {
if (selectedModels.includes("all_models")) {
// Select all models except "All Models"
const allModelsExceptAll = team ? team.models : userModels;
form.setFieldsValue({
models: allModelsExceptAll
});
}
};
const handleCopy = () => {
message.success('API Key copied to clipboard');
};
@ -154,9 +141,8 @@ const CreateKey: React.FC<CreateKeyProps> = ({
mode="multiple"
placeholder="Select models"
style={{ width: "100%" }}
onChange={(selectedModels) => handleModelSelection(selectedModels)}
>
<Option key="all_models" value="all_models">
<Option key="all-models" value="all-models">
All Models
</Option>
{team && team.models ? (

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

@ -115,6 +115,9 @@ const Team: React.FC<TeamProps> = ({
<Input />
</Form.Item>
<Form.Item label="Models" name="models">
<Select2.Option key="all-models" value="all-models">
All Models
</Select2.Option>
<Select2
mode="multiple"
placeholder="Select models"
@ -216,15 +219,7 @@ const handleEditSubmit = async (formValues: Record<string, any>) => {
setIsDeleteModalOpen(true);
};
const handleModelSelection = (selectedModels: string[]) => {
if (selectedModels.includes("all_models")) {
// Select all models except "All Models"
const allModelsExceptAll = userModels.filter(model => model !== "all");
form.setFieldsValue({
models: allModelsExceptAll
});
}
};
const confirmDelete = async () => {
@ -484,9 +479,8 @@ const handleEditSubmit = async (formValues: Record<string, any>) => {
mode="multiple"
placeholder="Select models"
style={{ width: "100%" }}
onChange={(selectedModels) => handleModelSelection(selectedModels)}
>
<Select2.Option key="all_models" value="all_models">
<Select2.Option key="all-models" value="all-models">
All Models
</Select2.Option>
{userModels.map((model) => (

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

@ -100,15 +100,7 @@ const ViewKeyTable: React.FC<ViewKeyTableProps> = ({
}
const handleModelSelection = (selectedModels: string[]) => {
if (selectedModels.includes("all_models")) {
// Select all models except "All Models"
const allModelsExceptAll = selectedTeam ? selectedTeam.models : userModels;
form.setFieldsValue({
models: allModelsExceptAll
});
}
};
const handleOk = () => {
form
@ -155,7 +147,6 @@ const ViewKeyTable: React.FC<ViewKeyTableProps> = ({
mode="multiple"
placeholder="Select models"
style={{ width: "100%" }}
onChange={(selectedModels) => handleModelSelection(selectedModels)}
>
<Option key="all_models" value="all_models">
All Models