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

ui - allow entering custom model names

Browse source
This commit is contained in:
Ishaan Jaff 2024-07-29 15:51:13 -07:00
parent b0da3f12e8
commit 3d88d7f750
1 changed files with 37 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -220,6 +220,9 @@ const handleSubmit = async (
else if (key === "base_model") {
// Add key-value pair to model_info dictionary
modelInfoObj[key] = value;
}
else if (key === "custom_model_name") {
litellmParamsObj["model"] = value;
} else if (key == "litellm_extra_params") {
console.log("litellm_extra_params:", value);
let litellmExtraParams = {};
@ -1718,26 +1721,43 @@ const ModelDashboard: React.FC<ModelDashboardProps> = ({
</Col>
</Row>
<Form.Item
rules={[{ required: true, message: "Required" }]}
label="LiteLLM Model Name(s)"
label="LiteLLM Model Name(s)"
tooltip="Actual model name used for making litellm.completion() call."
className="mb-0"
>
<Form.Item
name="model"
tooltip="Actual model name used for making litellm.completion() call."
className="mb-0"
rules={[{ required: true, message: "Required" }]}
noStyle
>
{ (selectedProvider === Providers.Azure) || (selectedProvider === Providers.OpenAI_Compatible) || (selectedProvider === Providers.Ollama) ? (
<TextInput placeholder={getPlaceholder(selectedProvider.toString())} />
) : providerModels.length > 0 ? (
<MultiSelect value={providerModels}>
{providerModels.map((model, index) => (
<MultiSelectItem key={index} value={model}>
{model}
</MultiSelectItem>
))}
</MultiSelect>
) : (
<TextInput placeholder={getPlaceholder(selectedProvider.toString())} />
)}
<MultiSelect>
<MultiSelectItem value="custom">Custom Model Name (Enter below)</MultiSelectItem>
{providerModels.map((model, index) => (
<MultiSelectItem key={index} value={model}>
{model}
</MultiSelectItem>
))}
</MultiSelect>
</Form.Item>
<Form.Item
noStyle
shouldUpdate={(prevValues, currentValues) => prevValues.model !== currentValues.model}
>
{({ getFieldValue }) => {
const selectedModels = getFieldValue('model') || [];
return selectedModels.includes('custom') && (
<Form.Item
name="custom_model_name"
rules={[{ required: true, message: "Please enter a custom model name" }]}
className="mt-2"
>
<TextInput placeholder="Enter custom model name" />
</Form.Item>
)
}}
</Form.Item>
</Form.Item>
<Row>
<Col span={10}></Col>
<Col span={10}>