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

UI - QOL improvements

Browse source
This commit is contained in:
Ishaan Jaff 2024-04-08 20:39:07 -07:00
parent 80694ba79b
commit 46618573ad
2 changed files with 53 additions and 44 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -176,6 +176,9 @@ const ModelDashboard: React.FC<ModelDashboardProps> = ({
}
const handleSubmit = async (formValues: Record<string, any>) => {
try {
const litellmParamsObj: Record<string, any> = {};
const modelInfoObj: Record<string, any> = {};
let modelName: string = "";
@ -201,11 +204,12 @@ const ModelDashboard: React.FC<ModelDashboardProps> = ({
if (key == "litellm_extra_params") {
console.log("litellm_extra_params:", value);
let litellmExtraParams = {};
if (value && value != undefined) {
try {
litellmExtraParams = JSON.parse(value);
}
catch (error) {
message.error("Failed to parse LiteLLM Extra Paras: " + error);
message.error("Failed to parse LiteLLM Extra Params: " + error);
throw new Error("Failed to parse litellm_extra_params: " + error);
}
for (const [key, value] of Object.entries(litellmExtraParams)) {
@ -213,6 +217,7 @@ const ModelDashboard: React.FC<ModelDashboardProps> = ({
}
}
}
}
const new_model: Model = {
"model_name": modelName,
@ -226,8 +231,12 @@ const ModelDashboard: React.FC<ModelDashboardProps> = ({
accessToken,
new_model
);
form.resetFields();
console.log(`response for model create call: ${response["data"]}`);
} catch (error) {
message.error("Failed to create model: " + error);
}
}
const handleOk = () => {
@ -235,7 +244,7 @@ const ModelDashboard: React.FC<ModelDashboardProps> = ({
.validateFields()
.then((values) => {
handleSubmit(values);
form.resetFields();
// form.resetFields();
})
.catch((error) => {
console.error("Validation failed:", error);

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

@ -38,7 +38,7 @@ export const modelCreateCall = async (
const data = await response.json();
console.log("API Response:", data);
message.success("Model created successfully. Wait 60s and refresh.")
message.success("Model created successfully. Wait 60s and refresh on 'All Models' page");
return data;
} catch (error) {
console.error("Failed to create key:", error);