ui show litellm model name

This commit is contained in:
Ishaan Jaff 2024-08-08 17:47:59 -07:00
parent 64dddea828
commit cc8282fe29
2 changed files with 41 additions and 1 deletions

View file

@ -844,6 +844,7 @@ const ModelDashboard: React.FC<ModelDashboardProps> = ({
modelData.data[i].provider = provider;
modelData.data[i].input_cost = input_cost;
modelData.data[i].output_cost = output_cost;
modelData.data[i].litellm_model_name = litellm_model_name;
// Convert Cost in terms of Cost per 1M tokens
if (modelData.data[i].input_cost) {
@ -1381,6 +1382,16 @@ const ModelDashboard: React.FC<ModelDashboardProps> = ({
>
Provider
</TableHeaderCell>
<TableHeaderCell
style={{
maxWidth: "150px",
whiteSpace: "normal",
wordBreak: "break-word",
fontSize: "11px",
}}
>
LiteLLM Model
</TableHeaderCell>
{userRole === "Admin" && (
<TableHeaderCell
style={{
@ -1508,6 +1519,34 @@ const ModelDashboard: React.FC<ModelDashboardProps> = ({
>
<p className="text-xs">{model.provider || "-"}</p>
</TableCell>
<TableCell
style={{
maxWidth: "100px",
whiteSpace: "normal",
wordBreak: "break-word",
}}
>
<Tooltip title={model && model.litellm_model_name}>
<pre
style={{
maxWidth: "150px",
whiteSpace: "normal",
wordBreak: "break-word",
}}
className="text-xs"
title={
model && model.litellm_model_name
? model.litellm_model_name
: ""
}
>
{model && model.litellm_model_name
? model.litellm_model_name.slice(0, 20) + (model.litellm_model_name.length > 20 ? "..." : "")
: "-"}
</pre>
</Tooltip>
</TableCell>
{userRole === "Admin" && (
<TableCell
style={{

View file

@ -46,7 +46,8 @@ const ViewUserSpend: React.FC<ViewUserSpendProps> = ({ userID, userRole, accessT
console.log(`userSpend: ${userSpend}`)
let [spend, setSpend] = useState(userSpend !== null ? userSpend : 0.0);
const [maxBudget, setMaxBudget] = useState(selectedTeam ? selectedTeam.max_budget : null);
console.log(`maxBudget: ${maxBudget}, selectedTeam.max_budget: ${selectedTeam.max_budget}, selectedTeam: ${JSON.stringify(selectedTeam)}`)
const team_budget = selectedTeam ? selectedTeam.max_budget : "unlimited";
console.log(`maxBudget: ${maxBudget}, selectedTeam.max_budget: ${team_budget}, selectedTeam: ${JSON.stringify(selectedTeam)}`)
const [userModels, setUserModels] = useState([]);
useEffect(() => {
const fetchData = async () => {