scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-29 06:33:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 200

fix(ui): drop the list row's stale org_id when a key save returns its organization

Browse source
This commit is contained in:
mateo-berri 2026-10-20 19:51:59 -07:00
parent be501494b5
commit d712b29914
2 changed files with 42 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

39
ui/litellm-dashboard/src/components/templates/KeyInfoView.handleKeyUpdate.test.tsx
View file

@ -655,3 +655,42 @@ describe("KeyInfoView handleKeyUpdate soft_budget", () => {
expect(keyUpdateCallMock).not.toHaveBeenCalled();
});
});
describe("KeyInfoView handleKeyUpdate organization", () => {
it("should show no organization after a save clears the one a key list row carried as org_id", async () => {
const adminAuth = {
accessToken: "access_abc",
userId: "user_1",
userRole: "Admin",
premiumUser: true,
token: "token_123",
userEmail: "test@example.com",
disabledPersonalKeyCreation: false,
showSSOBanner: false,
};
keyUpdateCallMock.mockResolvedValueOnce({ ...baseKeyData, organization_id: null });
mockUseAuthorized.mockReturnValue(adminAuth);
render(
<KeyInfoView
keyId="tok_123"
onClose={() => {}}
keyData={{ ...baseKeyData, org_id: "org-1" } as any}
onKeyDataUpdate={() => {}}
teams={[]}
/>,
);
fireEvent.click(screen.getByText("Settings"));
expect(screen.getByText("Organization", { selector: "p" }).nextElementSibling).toHaveTextContent("org-1");
fireEvent.click(screen.getByText("Edit Settings"));
(globalThis as any).__TEST_FORM_VALUES = { token: "tok_123", organization_id: null };
fireEvent.click(screen.getByText("Mock Submit"));
await waitFor(() => {
expect(screen.getByText("Organization", { selector: "p" }).nextElementSibling).toHaveTextContent("Not Set");
});
});
});

4
ui/litellm-dashboard/src/components/templates/key_info_view.tsx
View file

@ -398,7 +398,9 @@ export default function KeyInfoView({
const newKeyValues = await keyUpdateCall(accessToken, formValues);
setCurrentKeyData((prevData) => (prevData ? { ...prevData, ...newKeyValues } : undefined));
setCurrentKeyData((prevData) =>
prevData ? { ...prevData, ...newKeyValues, org_id: newKeyValues.organization_id } : undefined,
);
void queryClient.invalidateQueries({ queryKey: keyKeys.all });
if (onKeyDataUpdate) {