diff --git a/site/src/pages/AISettingsPage/ModelsPage/ModelsPageView.stories.tsx b/site/src/pages/AISettingsPage/ModelsPage/ModelsPageView.stories.tsx
index 39994900f0..85bf8b6635 100644
--- a/site/src/pages/AISettingsPage/ModelsPage/ModelsPageView.stories.tsx
+++ b/site/src/pages/AISettingsPage/ModelsPage/ModelsPageView.stories.tsx
@@ -65,6 +65,9 @@ export const Empty: Story = {
play: async ({ canvasElement }) => {
const canvas = within(canvasElement);
await expect(canvas.getByText("No models configured")).toBeInTheDocument();
+ await expect(
+ canvas.getAllByRole("button", { name: /add model/i }).length,
+ ).toBe(2);
},
};
diff --git a/site/src/pages/AISettingsPage/ModelsPage/ModelsPageView.tsx b/site/src/pages/AISettingsPage/ModelsPage/ModelsPageView.tsx
index 2cf3f67f24..b4375e7ab3 100644
--- a/site/src/pages/AISettingsPage/ModelsPage/ModelsPageView.tsx
+++ b/site/src/pages/AISettingsPage/ModelsPage/ModelsPageView.tsx
@@ -35,9 +35,10 @@ import { ModelRow } from "./components/ModelRow";
const MODELS_PAGE_SIZE = 10;
-const AddModelDropdown: FC<{ providerStates: readonly ProviderState[] }> = ({
- providerStates,
-}) => {
+const AddModelDropdown: FC<{
+ providerStates: readonly ProviderState[];
+ align?: "start" | "end";
+}> = ({ providerStates, align = "end" }) => {
const navigate = useNavigate();
const manageableProviderStates = providerStates.filter(
canManageProviderModels,
@@ -52,7 +53,7 @@ const AddModelDropdown: FC<{ providerStates: readonly ProviderState[] }> = ({
-
+
Select a provider
@@ -140,6 +141,12 @@ const ModelsPageView: FC = ({
+ }
/>
) : (
pagedItems.map((model) => (