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