coder · ssncferreira · Aug 26, 2026 · Aug 25, 2026 · Aug 26, 2026 · EhabY
diff --git a/site/src/pages/GroupsPage/GroupMemberBudgetCells.stories.tsx b/site/src/pages/GroupsPage/GroupMemberBudgetCells.stories.tsx
@@ -287,29 +287,20 @@ export const ResolvingGroupName: Story = {
 	},
 };
 
-/** An effective group that can't be resolved, standing in for another org's. */
-export const NotAttributedUnknownGroup: Story = {
+export const NotAttributedOtherOrganization: Story = {
 	args: {
 		spend: {
 			...mockSpend,
 			group_spend_micros: 456_000_000,
-			effective_group_id: "external-group",
+			effective_group_id: null,
+			group_budget: null,
 		},
 	},
-	parameters: {
-		queries: [
-			{
-				key: getGroupByIdQueryKey("external-group", { exclude_members: true }),
-				data: null,
-			},
-		],
-	},
 	play: async ({ canvasElement }) => {
 		const canvas = within(canvasElement);
 		const cell = await canvas.findByTestId(testId);
 		await expect(cell).toHaveTextContent("\u2014");
 		await expect(cell).not.toHaveTextContent("$456");
-		// The group cell shows an em-dash + info instead of naming the group.
 		const groupCell = canvas.getAllByRole("cell")[1];
 		await expect(groupCell).toHaveTextContent("\u2014");
 		await userEvent.click(

diff --git a/site/src/pages/GroupsPage/GroupMemberBudgetCells.test.ts b/site/src/pages/GroupsPage/GroupMemberBudgetCells.test.ts
@@ -16,8 +16,10 @@ describe("effectiveBudgetGroup", () => {
 		expect(effectiveBudgetGroup(undefined, group)).toEqual({ kind: "none" });
 	});
 
-	it("is other without a governing group (budget in another org)", () => {
-		expect(effectiveBudgetGroup(mockSpend, group)).toEqual({ kind: "other" });
+	it("is other organization when the effective group is masked", () => {
+		expect(effectiveBudgetGroup(mockSpend, group)).toEqual({
+			kind: "otherOrg",
+		});
 	});
 
 	it("is everyone for the org-wide Everyone group", () => {
@@ -53,6 +55,6 @@ describe("effectiveBudgetGroup", () => {
 				{ ...mockSpend, effective_group_id: "group-2" },
 				group,
 			),
-		).toEqual({ kind: "other" });
+		).toEqual({ kind: "otherGroup" });
 	});
 });
diff --git a/site/src/pages/GroupsPage/GroupMemberBudgetCells.tsx b/site/src/pages/GroupsPage/GroupMemberBudgetCells.tsx
@@ -26,7 +26,7 @@ export const GroupMemberBudgetCells: FC<{
 	spend: GroupMemberAISpend | undefined;
 }> = ({ group, userID, spend }) => {
 	const effective = effectiveBudgetGroup(spend, group);
-	const fromOtherGroup = effective.kind === "other";
+	const fromOtherGroup = effective.kind === "otherGroup";
 
 	// A null effective_group_id is a group in another org that can't be
 	// fetched, so only resolve the name when an ID exists.
@@ -64,30 +64,32 @@ export const GroupMemberBudgetCells: FC<{
 		case "this":
 			budgetGroup = <Badge size="sm">{badgeName(groupName)}</Badge>;
 			break;
-		case "other": {
+		case "otherGroup":
 			// Wait for the name to resolve rather than flashing the fallback.
 			if (isResolvingGroupName) {
 				budgetGroup = <Spinner loading size="sm" />;
-			} else if (effectiveGroupName) {
-				budgetGroup = <Badge size="sm">{badgeName(effectiveGroupName)}</Badge>;
 			} else {
-				// The group can't be resolved (another org), so it can't be named.
-				budgetGroup = (
-					<LabelWithInfo label={EM_DASH} message={OTHER_ORG_MESSAGE} />
+				budgetGroup = effectiveGroupName ? (
+					<Badge size="sm">{badgeName(effectiveGroupName)}</Badge>
+				) : (
+					EM_DASH
 				);
 			}
 			break;
-		}
+		case "otherOrg":
+			budgetGroup = (
+				<LabelWithInfo label={EM_DASH} message={OTHER_ORG_MESSAGE} />
+			);
+			break;
 	}
 
 	let budget: ReactNode = EM_DASH;
-	if (spend && fromOtherGroup) {
+	if (spend && effective.kind === "otherOrg") {
+		budget = <LabelWithInfo label={EM_DASH} message={OTHER_ORG_MESSAGE} />;
+	} else if (spend && effective.kind === "otherGroup") {
 		if (isResolvingGroupName) {
 			budget = <Spinner loading size="sm" />;
-		} else if (!effectiveGroupName) {
-			// The spend hides entirely when the governing group can't be resolved.
-			budget = <LabelWithInfo label={EM_DASH} message={OTHER_ORG_MESSAGE} />;
-		} else {
+		} else if (effectiveGroupName) {
 			budget = (
 				<div className="flex flex-col gap-0.5">
 					<span className="flex items-center gap-1">
@@ -171,21 +173,25 @@ type EffectiveBudgetGroup =
 	| { kind: "none" }
 	| { kind: "everyone" }
 	| { kind: "this" }
-	| { kind: "other" };
+	| { kind: "otherGroup" }
+	| { kind: "otherOrg" };
 
 /**
- * Resolves which group governs a member's AI budget. "none" means no budget
- * data loaded; "everyone" is the org-wide fallback when no named group sets a
- * budget. A null effective group means the budget resolves to a group in
- * another organization, so it can't be shown here.
+ * Resolves which group governs a member's AI budget:
+ *
+ * - "none": spend data is not loaded.
+ * - "everyone": the Everyone group governs the budget.
+ * - "this": the viewed group governs the budget.
+ * - "otherGroup": another group in this organization governs the budget.
+ * - "otherOrg": a group in another organization governs the budget.
  */
 export function effectiveBudgetGroup(
 	spend: GroupMemberAISpend | undefined,
 	group: Pick<Group, "id" | "organization_id">,
 ): EffectiveBudgetGroup {
 	const groupId = spend?.effective_group_id ?? null;
 	if (groupId === null) {
-		return spend === undefined ? { kind: "none" } : { kind: "other" };
+		return spend === undefined ? { kind: "none" } : { kind: "otherOrg" };
 	}
 	// Everyone shares the org's id; checked first so it wins when the viewed
 	// group is Everyone itself.
@@ -195,7 +201,7 @@ export function effectiveBudgetGroup(
 	if (groupId === group.id) {
 		return { kind: "this" };
 	}
-	return { kind: "other" };
+	return { kind: "otherGroup" };
 }
 
 const LabelWithInfo: FC<{ label: ReactNode; message: ReactNode }> = ({

diff --git a/site/src/pages/GroupsPage/GroupMembersPage.tsx b/site/src/pages/GroupsPage/GroupMembersPage.tsx
@@ -238,8 +238,8 @@ const GroupMemberRow: FC<GroupMemberRowProps> = ({
 	onManageAIBudget,
 	onRemove,
 }) => {
-	const budgetFromOtherGroup =
-		effectiveBudgetGroup(member.spend, group).kind === "other";
+	const budgetFromOtherOrganization =
+		effectiveBudgetGroup(member.spend, group).kind === "otherOrg";
 
 	return (
 		<TableRow key={member.id}>
@@ -288,7 +288,7 @@ const GroupMemberRow: FC<GroupMemberRowProps> = ({
 							{showAIBudget && (
 								<DropdownMenuItem
 									onClick={onManageAIBudget}
-									disabled={budgetFromOtherGroup}
+									disabled={budgetFromOtherOrganization}
 								>
 									Manage AI budget
 								</DropdownMenuItem>

diff --git a/site/src/pages/GroupsPage/GroupPage.stories.tsx b/site/src/pages/GroupsPage/GroupPage.stories.tsx
@@ -439,7 +439,7 @@ export const WithoutMemberAIBudgetColumn: Story = {
 	},
 };
 
-export const AIBudgetActionDisabledForOtherGroup: Story = {
+export const AIBudgetActionEnabledForOtherGroup: Story = {
 	parameters: {
 		features: ["aibridge"],
 		queries: [
@@ -489,7 +489,7 @@ export const AIBudgetActionDisabledForOtherGroup: Story = {
 		const menuItem = await body.findByRole("menuitem", {
 			name: "Manage AI budget",
 		});
-		await expect(menuItem).toHaveAttribute("aria-disabled", "true");
+		await expect(menuItem).not.toHaveAttribute("aria-disabled", "true");
 	},
 };
 
@@ -929,14 +929,14 @@ export const AIBudgetShowcase: Story = {
 		await expect(manageItem).not.toHaveAttribute("aria-disabled", "true");
 		await userEvent.keyboard("{Escape}");
 
-		// Another named group does disable it.
+		// Another group in the same organization keeps the action enabled.
 		const otherGroupMenu = await canvas.findAllByRole("button", {
 			name: "Open menu",
 		});
 		await userEvent.click(otherGroupMenu[7]);
-		const disabledItem = await body.findByRole("menuitem", {
+		const otherGroupItem = await body.findByRole("menuitem", {
 			name: "Manage AI budget",
 		});
-		await expect(disabledItem).toHaveAttribute("aria-disabled", "true");
+		await expect(otherGroupItem).not.toHaveAttribute("aria-disabled", "true");
 	},
 };