diff --git a/ui/litellm-dashboard/src/components/view_logs/LogDetailsDrawer/CollapsibleMessage.test.tsx b/ui/litellm-dashboard/src/components/view_logs/LogDetailsDrawer/CollapsibleMessage.test.tsx
index 1477febd967..3aa8d21c31c 100644
--- a/ui/litellm-dashboard/src/components/view_logs/LogDetailsDrawer/CollapsibleMessage.test.tsx
+++ b/ui/litellm-dashboard/src/components/view_logs/LogDetailsDrawer/CollapsibleMessage.test.tsx
@@ -37,4 +37,18 @@ describe("CollapsibleMessage", () => {
await user.click(screen.getByText("SYSTEM"));
expect(screen.getByText("Toggle me")).toBeInTheDocument();
});
+
+ it("should expand with Enter and collapse with Space from the keyboard", async () => {
+ const user = userEvent.setup();
+ render();
+
+ expect(screen.getByText("Toggle me")).not.toBeVisible();
+
+ await user.tab();
+ await user.keyboard("{Enter}");
+ expect(screen.getByText("Toggle me")).toBeVisible();
+
+ await user.keyboard(" ");
+ expect(screen.getByText("Toggle me")).not.toBeVisible();
+ });
});
diff --git a/ui/litellm-dashboard/src/components/view_logs/LogDetailsDrawer/HistoryTree.test.tsx b/ui/litellm-dashboard/src/components/view_logs/LogDetailsDrawer/HistoryTree.test.tsx
index 10d6bfa0847..0726a9f4b82 100644
--- a/ui/litellm-dashboard/src/components/view_logs/LogDetailsDrawer/HistoryTree.test.tsx
+++ b/ui/litellm-dashboard/src/components/view_logs/LogDetailsDrawer/HistoryTree.test.tsx
@@ -41,4 +41,22 @@ describe("HistoryTree", () => {
expect(screen.getByText("Hello")).toBeInTheDocument();
expect(screen.getByText("Hi there")).toBeInTheDocument();
});
+
+ it("should expand with Enter and collapse with Space from the keyboard", async () => {
+ const user = userEvent.setup();
+ const messages: ParsedMessage[] = [
+ { role: "user", content: "Hello" },
+ { role: "assistant", content: "Hi there" },
+ ];
+ render();
+
+ expect(screen.getByText("Hello")).not.toBeVisible();
+
+ await user.tab();
+ await user.keyboard("{Enter}");
+ expect(screen.getByText("Hello")).toBeVisible();
+
+ await user.keyboard(" ");
+ expect(screen.getByText("Hello")).not.toBeVisible();
+ });
});