chore: cleaup mps v1 billing (#507)

* chore: cleaup mps v1 billing

* chore: remove legacy file upload path

* chore: implement review comments

ui/src/app/billing/page.tsx

@@ -116,7 +116,7 @@ export default function BillingPage() {
     const router = useRouter();
     const searchParams = useSearchParams();
     const auth = useAuth();
-    const { config } = useAppConfig();
+    const { config, loading: configLoading } = useAppConfig();
     const [credits, setCredits] = useState<MpsBillingCreditsResponse | null>(null);
     const [loading, setLoading] = useState(true);
     const [refreshing, setRefreshing] = useState(false);
@@ -125,9 +125,9 @@ export default function BillingPage() {
         () => getPageFromSearchParams(searchParams),
     );
 
-    const isBillingV2 = credits?.billing_version === "v2";
-    const isOssMode = config?.deploymentMode === "oss";
-    const canPurchaseCredits = isBillingV2 && !isOssMode;
+    const hasAppConfig = !configLoading && config !== null;
+    const isOssMode = hasAppConfig && config.deploymentMode === "oss";
+    const canPurchaseCredits = hasAppConfig && config.deploymentMode !== "oss";
     const totalQuota = credits?.total_quota ?? 0;
     const remainingCredits = credits?.remaining_credits ?? 0;
     const usedCredits = credits?.total_credits_used ?? 0;
@@ -229,7 +229,7 @@ export default function BillingPage() {
         }
     };
 
-    if (loading) {
+    if (loading || configLoading) {
         return (
             <div className="container mx-auto p-6 space-y-6">
                 <div className="space-y-2">
@@ -301,7 +301,7 @@ export default function BillingPage() {
             <div className="grid gap-4 md:grid-cols-2">
                 <Card>
                     <CardHeader className="pb-2">
-                        <CardDescription>{isBillingV2 ? "Credit balance" : "Credits remaining"}</CardDescription>
+                        <CardDescription>{isOssMode ? "Credits remaining" : "Credit balance"}</CardDescription>
                         <CardTitle className="flex items-center gap-2 text-3xl">
                             <CircleDollarSign className="h-6 w-6 text-muted-foreground" />
                             {formatCredits(remainingCredits)}
@@ -319,13 +319,13 @@ export default function BillingPage() {
                     </CardHeader>
                     <CardContent>
                         <p className="text-sm text-muted-foreground">
-                            {isBillingV2 ? "Total ledger debits" : "Current allocation usage"}
+                            {isOssMode ? "Current allocation usage" : "Total ledger debits"}
                         </p>
                     </CardContent>
                 </Card>
             </div>
 
-            {isBillingV2 ? (
+            {!isOssMode ? (
                 <Card>
                     <CardHeader>
                         <CardTitle>Credit Ledger</CardTitle>

ui/src/app/model-configurations/page.tsx

@@ -2,24 +2,12 @@
 import ModelConfigurationV2 from "@/components/ModelConfigurationV2";
 import { SETTINGS_DOCUMENTATION_URLS } from "@/constants/documentation";
 
-interface ServiceConfigurationPageProps {
-    searchParams?: Promise<{
-        action?: string | string[];
-    }>;
-}
-
-export default async function ServiceConfigurationPage({ searchParams }: ServiceConfigurationPageProps) {
-    const params = searchParams ? await searchParams : {};
-    const action = Array.isArray(params.action) ? params.action[0] : params.action;
-
+export default function ServiceConfigurationPage() {
     return (
         <div className="min-h-screen">
             <div className="container mx-auto px-4 py-8">
                 <div className="max-w-4xl mx-auto">
-                    <ModelConfigurationV2
-                        docsUrl={SETTINGS_DOCUMENTATION_URLS.modelOverrides}
-                        initialAction={action}
-                    />
+                    <ModelConfigurationV2 docsUrl={SETTINGS_DOCUMENTATION_URLS.modelOverrides} />
                 </div>
             </div>
         </div>

ui/src/app/workflow/[workflowId]/components/ModelConfigurationDialog.tsx

@@ -1,54 +0,0 @@
-import { ServiceConfigurationForm } from "@/components/ServiceConfigurationForm";
-import {
-    Dialog,
-    DialogContent,
-    DialogDescription,
-    DialogHeader,
-    DialogTitle,
-} from "@/components/ui/dialog";
-import type { WorkflowConfigurations } from "@/types/workflow-configurations";
-
-interface ModelConfigurationDialogProps {
-    open: boolean;
-    onOpenChange: (open: boolean) => void;
-    workflowConfigurations: WorkflowConfigurations | null;
-    workflowName: string;
-    onSave: (configurations: WorkflowConfigurations, workflowName: string) => Promise<void>;
-}
-
-export const ModelConfigurationDialog = ({
-    open,
-    onOpenChange,
-    workflowConfigurations,
-    workflowName,
-    onSave,
-}: ModelConfigurationDialogProps) => {
-    return (
-        <Dialog open={open} onOpenChange={onOpenChange}>
-            <DialogContent className="max-w-2xl max-h-[85vh] overflow-y-auto">
-                <DialogHeader>
-                    <DialogTitle>Model Configuration</DialogTitle>
-                    <DialogDescription>
-                        Override global model settings for this workflow. Toggle individual services to customize.
-                    </DialogDescription>
-                </DialogHeader>
-
-                <ServiceConfigurationForm
-                    mode="override"
-                    currentOverrides={workflowConfigurations?.model_overrides}
-                    submitLabel="Save"
-                    onSave={async (config) => {
-                        await onSave(
-                            {
-                                ...workflowConfigurations,
-                                model_overrides: config.model_overrides as WorkflowConfigurations["model_overrides"],
-                            } as WorkflowConfigurations,
-                            workflowName,
-                        );
-                        onOpenChange(false);
-                    }}
-                />
-            </DialogContent>
-        </Dialog>
-    );
-};

ui/src/app/workflow/[workflowId]/settings/page.tsx

@@ -25,7 +25,6 @@ import {
 } from "@/components/AIModelConfigurationV2Editor";
 import { FlowEdge, FlowNode } from "@/components/flow/types";
 import { LLMConfigSelector } from "@/components/LLMConfigSelector";
-import { ServiceConfigurationForm } from "@/components/ServiceConfigurationForm";
 import SpinLoader from "@/components/SpinLoader";
 import { Button } from "@/components/ui/button";
 import { Calendar } from "@/components/ui/calendar";
@@ -1223,17 +1222,7 @@ function WorkflowModelOverridesSection({
         setOverrideEnabled(Boolean(workflowConfigurations.model_configuration_v2_override));
     }, [workflowConfigurations.model_configuration_v2_override]);
 
-    const source = organizationModelConfiguration?.source || "empty";
-    const isV2 = source === "organization_v2";
-
-    const saveLegacyOverrides = async (config: Record<string, unknown>) => {
-        const nextConfigurations = withoutModelConfigurationOverrides(workflowConfigurations);
-        const modelOverrides = config.model_overrides as WorkflowConfigurations["model_overrides"] | undefined;
-        if (modelOverrides) {
-            nextConfigurations.model_overrides = modelOverrides;
-        }
-        await onSave(nextConfigurations, workflowName);
-    };
+    const hasOrgConfiguration = organizationModelConfiguration?.source === "organization_v2";
 
     const saveV2Override = async (configuration: OrganizationAiModelConfigurationV2) => {
         const nextConfigurations = withoutModelConfigurationOverrides(workflowConfigurations);
@@ -1261,9 +1250,7 @@ function WorkflowModelOverridesSection({
                     Model Overrides
                 </CardTitle>
                 <CardDescription>
-                    {isV2
-                        ? "Override the full organization model configuration for this workflow."
-                        : "Override global model settings for this workflow. Toggle individual services to customize."}{" "}
+                    Override the full organization model configuration for this workflow.{" "}
                     <a href={SETTINGS_DOCUMENTATION_URLS.modelOverrides} target="_blank" rel="noopener noreferrer" className="inline-flex items-center gap-0.5 underline">Learn more <ExternalLink className="h-3 w-3" /></a>
                 </CardDescription>
             </CardHeader>
@@ -1281,28 +1268,18 @@ function WorkflowModelOverridesSection({
                     </div>
                 )}
 
-                {!modelConfigurationLoading && !modelConfigurationError && !isV2 && (
-                    <>
-                        {source === "legacy_user_v1" && (
-                            <div className="flex flex-col gap-3 rounded-md border bg-muted/30 p-4 sm:flex-row sm:items-center sm:justify-between">
-                                <p className="text-sm text-muted-foreground">
-                                    This workflow is using legacy model overrides. Migrate organization model configuration to use v2 overrides.
-                                </p>
-                                <Button type="button" variant="outline" size="sm" asChild>
-                                    <Link href="/model-configurations?action=migrate_to_v2">Migrate to v2</Link>
-                                </Button>
-                            </div>
-                        )}
-                        <ServiceConfigurationForm
-                            mode="override"
-                            currentOverrides={workflowConfigurations.model_overrides}
-                            submitLabel="Save Model Overrides"
-                            onSave={saveLegacyOverrides}
-                        />
-                    </>
+                {!modelConfigurationLoading && !modelConfigurationError && !hasOrgConfiguration && (
+                    <div className="flex flex-col gap-3 rounded-md border bg-muted/30 p-4 sm:flex-row sm:items-center sm:justify-between">
+                        <p className="text-sm text-muted-foreground">
+                            Set up your organization model configuration before overriding it per workflow.
+                        </p>
+                        <Button type="button" variant="outline" size="sm" asChild>
+                            <Link href="/model-configurations">Configure Models</Link>
+                        </Button>
+                    </div>
                 )}
 
-                {!modelConfigurationLoading && !modelConfigurationError && isV2 && modelConfigurationDefaults && organizationModelConfiguration && (
+                {!modelConfigurationLoading && !modelConfigurationError && hasOrgConfiguration && modelConfigurationDefaults && organizationModelConfiguration && (
                     <>
                         <div className="flex items-center justify-between rounded-md border p-4">
                             <div className="space-y-0.5">

ui/src/client/types.gen.ts

@@ -3307,10 +3307,6 @@ export type MpsBillingAccountResponse = {
  * MPSBillingCreditsResponse
  */
 export type MpsBillingCreditsResponse = {
-    /**
-     * Billing Version
-     */
-    billing_version: 'legacy' | 'v2';
     /**
      * Total Credits Used
      */
@@ -3436,24 +3432,6 @@ export type MpsCreditPurchaseUrlResponse = {
     checkout_url: string;
 };
 
-/**
- * MPSCreditsResponse
- */
-export type MpsCreditsResponse = {
-    /**
-     * Total Credits Used
-     */
-    total_credits_used: number;
-    /**
-     * Remaining Credits
-     */
-    remaining_credits: number;
-    /**
-     * Total Quota
-     */
-    total_quota: number;
-};
-
 /**
  * McpRefreshResponse
  *
@@ -12031,45 +12009,6 @@ export type GetCurrentPeriodUsageApiV1OrganizationsUsageCurrentPeriodGetResponse
 
 export type GetCurrentPeriodUsageApiV1OrganizationsUsageCurrentPeriodGetResponse = GetCurrentPeriodUsageApiV1OrganizationsUsageCurrentPeriodGetResponses[keyof GetCurrentPeriodUsageApiV1OrganizationsUsageCurrentPeriodGetResponses];
 
-export type GetMpsCreditsApiV1OrganizationsUsageMpsCreditsGetData = {
-    body?: never;
-    headers?: {
-        /**
-         * Authorization
-         */
-        authorization?: string | null;
-        /**
-         * X-Api-Key
-         */
-        'X-API-Key'?: string | null;
-    };
-    path?: never;
-    query?: never;
-    url: '/api/v1/organizations/usage/mps-credits';
-};
-
-export type GetMpsCreditsApiV1OrganizationsUsageMpsCreditsGetErrors = {
-    /**
-     * Not found
-     */
-    404: unknown;
-    /**
-     * Validation Error
-     */
-    422: HttpValidationError;
-};
-
-export type GetMpsCreditsApiV1OrganizationsUsageMpsCreditsGetError = GetMpsCreditsApiV1OrganizationsUsageMpsCreditsGetErrors[keyof GetMpsCreditsApiV1OrganizationsUsageMpsCreditsGetErrors];
-
-export type GetMpsCreditsApiV1OrganizationsUsageMpsCreditsGetResponses = {
-    /**
-     * Successful Response
-     */
-    200: MpsCreditsResponse;
-};
-
-export type GetMpsCreditsApiV1OrganizationsUsageMpsCreditsGetResponse = GetMpsCreditsApiV1OrganizationsUsageMpsCreditsGetResponses[keyof GetMpsCreditsApiV1OrganizationsUsageMpsCreditsGetResponses];
-
 export type GetBillingCreditsApiV1OrganizationsBillingCreditsGetData = {
     body?: never;
     headers?: {

ui/src/components/ModelConfigurationV2.tsx

@@ -1,12 +1,11 @@
 "use client";
 
-import { ExternalLink, RefreshCw } from "lucide-react";
+import { ExternalLink } from "lucide-react";
 import { useEffect, useRef, useState } from "react";
 
 import {
     getModelConfigurationV2ApiV1OrganizationsModelConfigurationsV2Get,
     getModelConfigurationV2DefaultsApiV1OrganizationsModelConfigurationsV2DefaultsGet,
-    migrateModelConfigurationV2ApiV1OrganizationsModelConfigurationsV2MigratePost,
     saveModelConfigurationV2ApiV1OrganizationsModelConfigurationsV2Put,
 } from "@/client/sdk.gen";
 import type {
@@ -14,47 +13,22 @@ import type {
     OrganizationAiModelConfigurationV2,
 } from "@/client/types.gen";
 import { AIModelConfigurationV2Editor, type ModelConfigurationDefaultsV2 } from "@/components/AIModelConfigurationV2Editor";
-import { ServiceConfigurationForm } from "@/components/ServiceConfigurationForm";
-import {
-    AlertDialog,
-    AlertDialogCancel,
-    AlertDialogContent,
-    AlertDialogDescription,
-    AlertDialogFooter,
-    AlertDialogHeader,
-    AlertDialogTitle,
-} from "@/components/ui/alert-dialog";
-import { Badge } from "@/components/ui/badge";
-import { Button } from "@/components/ui/button";
 import { Skeleton } from "@/components/ui/skeleton";
 import { useUserConfig } from "@/context/UserConfigContext";
 import { detailFromError } from "@/lib/apiError";
 import { useAuth } from "@/lib/auth";
 
-export default function ModelConfigurationV2({
-    docsUrl,
-    initialAction,
-}: {
-    docsUrl?: string;
-    initialAction?: string;
-}) {
+export default function ModelConfigurationV2({ docsUrl }: { docsUrl?: string }) {
     const auth = useAuth();
-    const { refreshConfig, saveUserConfig } = useUserConfig();
+    const { refreshConfig } = useUserConfig();
     const hasFetched = useRef(false);
-    const hasAppliedInitialMigrationAction = useRef(false);
 
     const [defaults, setDefaults] = useState<ModelConfigurationDefaultsV2 | null>(null);
     const [response, setResponse] = useState<OrganizationAiModelConfigurationResponse | null>(null);
     const [loading, setLoading] = useState(true);
-    const [migrating, setMigrating] = useState(false);
-    const [migrationDialogOpen, setMigrationDialogOpen] = useState(false);
     const [error, setError] = useState<string | null>(null);
     const [notice, setNotice] = useState<string | null>(null);
 
-    const applyResponse = (nextResponse: OrganizationAiModelConfigurationResponse) => {
-        setResponse(nextResponse);
-    };
-
     useEffect(() => {
         if (auth.loading || !auth.user || hasFetched.current) return;
         hasFetched.current = true;
@@ -85,7 +59,7 @@ export default function ModelConfigurationV2({
                 return;
             }
             setDefaults(nextDefaults);
-            applyResponse(configResult.data);
+            setResponse(configResult.data);
             setLoading(false);
         };
 
@@ -93,14 +67,6 @@ export default function ModelConfigurationV2({
 
     }, [auth.loading, auth.user]);
 
-    useEffect(() => {
-        if (hasAppliedInitialMigrationAction.current) return;
-        if (initialAction !== "migrate_to_v2") return;
-        if (loading || response?.source !== "legacy_user_v1") return;
-        hasAppliedInitialMigrationAction.current = true;
-        setMigrationDialogOpen(true);
-    }, [initialAction, loading, response?.source]);
-
     const saveConfiguration = async (configuration: OrganizationAiModelConfigurationV2) => {
         if (!defaults) return;
         setError(null);
@@ -117,50 +83,11 @@ export default function ModelConfigurationV2({
             throw new Error("Failed to save model configuration");
         }
 
-        applyResponse(result.data);
+        setResponse(result.data);
         await refreshConfig();
         setNotice("Model configuration saved");
     };
 
-    const migrateConfiguration = async () => {
-        if (!defaults) return;
-        setMigrating(true);
-        setError(null);
-        setNotice(null);
-
-        const result = await migrateModelConfigurationV2ApiV1OrganizationsModelConfigurationsV2MigratePost();
-        if (result.error) {
-            setError(detailFromError(result.error, "Failed to migrate model configuration"));
-        } else if (!result.data) {
-            setError("Failed to migrate model configuration");
-        } else {
-            applyResponse(result.data);
-            await refreshConfig();
-            setNotice("Configuration migrated to v2");
-            setMigrationDialogOpen(false);
-        }
-        setMigrating(false);
-    };
-
-    const migrationWarningDialog = (
-        <AlertDialog open={migrationDialogOpen} onOpenChange={setMigrationDialogOpen}>
-            <AlertDialogContent>
-                <AlertDialogHeader>
-                    <AlertDialogTitle>Migrate model configuration to v2?</AlertDialogTitle>
-                    <AlertDialogDescription>
-                        Your configurations will be migrated to v2. After migration, check your global configuration and workflow model overrides, then run a test call to make sure everything is working.
-                    </AlertDialogDescription>
-                </AlertDialogHeader>
-                <AlertDialogFooter>
-                    <AlertDialogCancel disabled={migrating}>Cancel</AlertDialogCancel>
-                    <Button type="button" onClick={migrateConfiguration} disabled={migrating}>
-                        {migrating ? "Migrating..." : "Migrate to v2"}
-                    </Button>
-                </AlertDialogFooter>
-            </AlertDialogContent>
-        </AlertDialog>
-    );
-
     if (loading) {
         return (
             <div className="w-full max-w-4xl mx-auto space-y-6">
@@ -171,68 +98,6 @@ export default function ModelConfigurationV2({
         );
     }
 
-    const source = response?.source || "empty";
-
-    if (source !== "organization_v2") {
-        return (
-            <div className="w-full max-w-4xl mx-auto space-y-6">
-                <div className="flex flex-col gap-3 sm:flex-row sm:items-start sm:justify-between">
-                    <div>
-                        <div className="flex items-center gap-2">
-                            <h1 className="text-3xl font-bold">AI Models Configuration</h1>
-                            <Badge variant="outline">
-                                {source === "legacy_user_v1" ? "legacy" : "v1"}
-                            </Badge>
-                        </div>
-                        <p className="mt-2 text-sm text-muted-foreground">
-                            Configure your AI model, voice, and transcription services.{" "}
-                            {docsUrl && (
-                                <a href={docsUrl} target="_blank" rel="noopener noreferrer" className="inline-flex items-center gap-0.5 underline">
-                                    Learn more <ExternalLink className="h-3 w-3" />
-                                </a>
-                            )}
-                        </p>
-                    </div>
-                    {source === "legacy_user_v1" && (
-                        <Button type="button" variant="outline" onClick={() => setMigrationDialogOpen(true)} disabled={migrating}>
-                            <RefreshCw className="mr-2 h-4 w-4" />
-                            {migrating ? "Migrating..." : "Migrate to v2"}
-                        </Button>
-                    )}
-                </div>
-
-                {error && (
-                    <div className="rounded-md border border-destructive/40 bg-destructive/10 px-4 py-3 text-sm text-destructive">
-                        {error}
-                    </div>
-                )}
-                {notice && (
-                    <div className="rounded-md border border-green-500/40 bg-green-500/10 px-4 py-3 text-sm text-green-700 dark:text-green-300">
-                        {notice}
-                    </div>
-                )}
-
-                <ServiceConfigurationForm
-                    mode="global"
-                    onSave={async (config) => {
-                        setError(null);
-                        setNotice(null);
-                        await saveUserConfig(config as Parameters<typeof saveUserConfig>[0]);
-                        await refreshConfig();
-                        if (defaults) {
-                            const configResult = await getModelConfigurationV2ApiV1OrganizationsModelConfigurationsV2Get();
-                            if (configResult.data) {
-                                applyResponse(configResult.data);
-                            }
-                        }
-                        setNotice("Configuration saved");
-                    }}
-                />
-                {migrationWarningDialog}
-            </div>
-        );
-    }
-
     return (
         <div className="w-full max-w-4xl mx-auto space-y-6">
             <div className="flex flex-col gap-3 sm:flex-row sm:items-start sm:justify-between">
@@ -268,7 +133,6 @@ export default function ModelConfigurationV2({
                     onSave={saveConfiguration}
                 />
             )}
-            {migrationWarningDialog}
         </div>
     );
 }

ui/src/components/ServiceConfiguration.tsx

@@ -1,37 +0,0 @@
-"use client";
-
-import { ExternalLink } from "lucide-react";
-
-import { ServiceConfigurationForm } from "@/components/ServiceConfigurationForm";
-import { useUserConfig } from "@/context/UserConfigContext";
-
-interface ServiceConfigurationProps {
-    docsUrl?: string;
-}
-
-export default function ServiceConfiguration({ docsUrl }: ServiceConfigurationProps) {
-    const { saveUserConfig } = useUserConfig();
-
-    return (
-        <div className="w-full max-w-2xl mx-auto">
-            <div className="mb-6">
-                <h1 className="text-3xl font-bold mb-2">AI Models Configuration</h1>
-                <p className="text-muted-foreground">
-                    Configure your AI model, voice, and transcription services.{" "}
-                    {docsUrl && (
-                        <a href={docsUrl} target="_blank" rel="noopener noreferrer" className="inline-flex items-center gap-0.5 underline">
-                            Learn more <ExternalLink className="h-3 w-3" />
-                        </a>
-                    )}
-                </p>
-            </div>
-
-            <ServiceConfigurationForm
-                mode="global"
-                onSave={async (config) => {
-                    await saveUserConfig(config);
-                }}
-            />
-        </div>
-    );
-}

ui/src/components/billing/DograhCreditsCard.tsx

@@ -1,122 +0,0 @@
-"use client";
-
-import { UserRound } from "lucide-react";
-import posthog from "posthog-js";
-import { useCallback, useEffect, useState } from "react";
-
-import { getMpsCreditsApiV1OrganizationsUsageMpsCreditsGet } from "@/client/sdk.gen";
-import type { MpsCreditsResponse } from "@/client/types.gen";
-import { BuyCreditsControl } from "@/components/billing/BuyCreditsControl";
-import { Button } from "@/components/ui/button";
-import { Card, CardContent, CardDescription, CardHeader, CardTitle } from "@/components/ui/card";
-import { Progress } from "@/components/ui/progress";
-import { PostHogEvent } from "@/constants/posthog-events";
-import { useLeadForms } from "@/context/LeadFormsContext";
-import { useAuth } from "@/lib/auth";
-
-export function DograhCreditsCard() {
-  const auth = useAuth();
-  const { openHireExpert, openEnterprise } = useLeadForms();
-  const [mpsCredits, setMpsCredits] = useState<MpsCreditsResponse | null>(null);
-  const [isLoadingCredits, setIsLoadingCredits] = useState(true);
-
-  const fetchMpsCredits = useCallback(async () => {
-    if (!auth.isAuthenticated) return;
-    try {
-      const response = await getMpsCreditsApiV1OrganizationsUsageMpsCreditsGet();
-      // The generated client resolves to { data, error } and does NOT throw on
-      // 4xx/5xx (see ui/AGENTS.md) — check error explicitly.
-      if (response.error) {
-        console.error("Failed to fetch MPS credits:", response.error);
-      } else if (response.data) {
-        setMpsCredits(response.data);
-      }
-    } catch (error) {
-      console.error("Failed to fetch MPS credits:", error);
-    } finally {
-      setIsLoadingCredits(false);
-    }
-  }, [auth.isAuthenticated]);
-
-  useEffect(() => {
-    if (auth.isAuthenticated) {
-      fetchMpsCredits();
-    }
-  }, [auth.isAuthenticated, fetchMpsCredits]);
-
-  return (
-    <Card className="mb-6">
-      <CardHeader>
-        <CardTitle>Dograh Model Credits</CardTitle>
-        <CardDescription>
-          These track usage of Dograh models using Dograh Service Keys.
-        </CardDescription>
-      </CardHeader>
-      <CardContent>
-        {isLoadingCredits ? (
-          <div className="animate-pulse space-y-4">
-            <div className="h-4 bg-muted rounded w-1/4"></div>
-            <div className="h-8 bg-muted rounded"></div>
-            <div className="h-4 bg-muted rounded w-1/3"></div>
-          </div>
-        ) : mpsCredits ? (
-          <div className="space-y-4">
-            <div className="flex justify-between items-baseline">
-              <div>
-                <p className="text-2xl font-bold">
-                  {mpsCredits.total_credits_used.toFixed(2)}{" "}
-                  <span className="text-lg font-normal text-muted-foreground">
-                    / {mpsCredits.total_quota.toFixed(2)}
-                  </span>
-                </p>
-                <p className="text-sm text-muted-foreground">Credits Used</p>
-              </div>
-              <div className="text-right">
-                <p className="text-lg font-semibold">{mpsCredits.remaining_credits.toFixed(2)}</p>
-                <p className="text-sm text-muted-foreground">Remaining</p>
-              </div>
-            </div>
-
-            {mpsCredits.total_quota > 0 && (
-              <Progress value={Math.min(100, (mpsCredits.total_credits_used / mpsCredits.total_quota) * 100)} className="h-3" />
-            )}
-          </div>
-        ) : (
-          <p className="text-muted-foreground">
-            No Dograh service keys configured. Set up a service key in your model configuration to see usage.
-          </p>
-        )}
-
-        {/* Footer CTAs — self-serve + done-for-you side by side, with the
-            custom-pricing link directly beneath. */}
-        <div className="mt-6 space-y-4 border-t pt-4">
-          <div className="space-y-1">
-            <p className="text-sm font-medium">Running low?</p>
-            <p className="text-sm text-muted-foreground">Top up instantly, or have us build it for you.</p>
-          </div>
-          <div className="flex flex-col gap-2 sm:flex-row">
-            <BuyCreditsControl className="w-full sm:flex-1" />
-            <Button
-              variant="outline"
-              className="w-full gap-2 sm:flex-1"
-              onClick={() => openHireExpert("billing_card")}
-            >
-              <UserRound className="h-4 w-4" />
-              Hire an Expert
-            </Button>
-          </div>
-          <button
-            type="button"
-            onClick={() => {
-              posthog.capture(PostHogEvent.CUSTOM_PRICING_CLICKED);
-              openEnterprise("billing_custom_pricing");
-            }}
-            className="block text-xs text-muted-foreground underline decoration-dashed underline-offset-4 hover:text-foreground"
-          >
-            Book a Strategy Call: custom pricing for committed volume
-          </button>
-        </div>
-      </CardContent>
-    </Card>
-  );
-}

ui/src/context/OrgConfigContext.tsx

@@ -3,7 +3,7 @@
 import { createContext, ReactNode, useCallback, useContext, useEffect, useRef, useState } from 'react';
 
 import { client } from '@/client/client.gen';
-import { getCurrentOrganizationContextApiV1OrganizationsContextGet, getUserConfigurationsApiV1UserConfigurationsUserGet, updateUserConfigurationsApiV1UserConfigurationsUserPut } from '@/client/sdk.gen';
+import { getCurrentOrganizationContextApiV1OrganizationsContextGet, getUserConfigurationsApiV1UserConfigurationsUserGet } from '@/client/sdk.gen';
 import type { OrganizationContextResponse, UserConfigurationRequestResponseSchema } from '@/client/types.gen';
 import { setupAuthInterceptor } from '@/lib/apiClient';
 import type { AuthUser } from '@/lib/auth';
@@ -22,7 +22,6 @@ interface OrganizationPricing {
 interface OrgConfigContextType {
     orgContext: OrganizationContextResponse | null;
     userConfig: UserConfigurationRequestResponseSchema | null;
-    saveUserConfig: (userConfig: UserConfigurationRequestResponseSchema) => Promise<void>;
     loading: boolean;
     error: Error | null;
     refreshConfig: () => Promise<void>;
@@ -133,33 +132,6 @@ export function OrgConfigProvider({ children }: { children: ReactNode }) {
         fetchConfig();
     }, [auth.loading, auth.isAuthenticated, fetchConfig]);
 
-    const saveUserConfig = useCallback(async (userConfigRequest: UserConfigurationRequestResponseSchema) => {
-        if (!authRef.current.isAuthenticated) throw new Error('No authentication available');
-        const response = await updateUserConfigurationsApiV1UserConfigurationsUserPut({
-            body: {
-                ...userConfig,
-                ...userConfigRequest,
-            } as UserConfigurationRequestResponseSchema,
-        });
-        if (response.error) {
-            let msg = 'Failed to save user configuration';
-            const detail = (response.error as unknown as { detail?: string | { errors: { model: string; message: string }[] } }).detail;
-            if (typeof detail === 'string') {
-                msg = detail;
-            } else if (Array.isArray(detail)) {
-                msg = detail
-                    .map((e: { model: string; message: string }) => `${e.model}: ${e.message}`)
-                    .join('\n');
-            }
-            throw new Error(msg);
-        }
-
-        if (response.data) {
-            setUserConfig(response.data);
-            setOrganizationPricing(pricingFromUserConfig(response.data));
-        }
-    }, [userConfig]);
-
     const refreshConfig = useCallback(async () => {
         await fetchConfig();
     }, [fetchConfig]);
@@ -169,7 +141,6 @@ export function OrgConfigProvider({ children }: { children: ReactNode }) {
             value={{
                 orgContext,
                 userConfig,
-                saveUserConfig,
                 loading,
                 error,
                 refreshConfig,