Refactor LLM admin to use generated API and types

Replaces usage of the custom BackendApi client and legacy types in admin LLM actions and components with generated OpenAPI endpoints and types. Updates API calls, error handling, and type imports throughout the admin LLM dashboard. Also corrects operationId fields in backend routes and OpenAPI spec for consistency.

autogpt_platform/backend/backend/server/v2/admin/llm_routes.py

@@ -388,6 +388,7 @@ async def list_llm_creators():
 @router.get(
     "/creators/{creator_id}",
     summary="Get creator details",
+    operation_id="getV2GetLlmCreatorDetails",
     response_model=llm_model.LlmModelCreator,
 )
 async def get_llm_creator(creator_id: str):

autogpt_platform/backend/backend/server/v2/store/routes.py

@@ -294,6 +294,7 @@ async def get_creators(
 @router.get(
     "/creator/{username}",
     summary="Get creator details",
+    operation_id="getV2GetCreatorDetails",
     tags=["store", "public"],
     response_model=backend.server.v2.store.model.CreatorDetails,
 )

autogpt_platform/frontend/src/app/(platform)/admin/llms/actions.ts

@@ -2,18 +2,35 @@
 
 import { revalidatePath } from "next/cache";
 
-import BackendApi from "@/lib/autogpt-server-api";
-import type {
-  LlmProvider,
-  LlmModel,
-  LlmCreatorsResponse,
-  LlmMigrationsResponse,
-  UpsertLlmProviderRequest,
-  CreateLlmModelRequest,
-  UpdateLlmModelRequest,
-  ToggleLlmModelRequest,
-  UpsertLlmCreatorRequest,
-} from "@/lib/autogpt-server-api/types";
+// Generated API functions
+import {
+  getV2ListLlmProviders,
+  postV2CreateLlmProvider,
+  getV2ListLlmModels,
+  postV2CreateLlmModel,
+  patchV2UpdateLlmModel,
+  patchV2ToggleLlmModelAvailability,
+  deleteV2DeleteLlmModelAndMigrateWorkflows,
+  getV2GetModelUsageCount,
+  getV2ListModelMigrations,
+  postV2RevertAModelMigration,
+  getV2ListModelCreators,
+  postV2CreateModelCreator,
+  patchV2UpdateModelCreator,
+  deleteV2DeleteModelCreator,
+} from "@/app/api/__generated__/endpoints/admin/admin";
+
+// Generated types
+import type { LlmProvidersResponse } from "@/app/api/__generated__/models/llmProvidersResponse";
+import type { LlmModelsResponse } from "@/app/api/__generated__/models/llmModelsResponse";
+import type { UpsertLlmProviderRequest } from "@/app/api/__generated__/models/upsertLlmProviderRequest";
+import type { CreateLlmModelRequest } from "@/app/api/__generated__/models/createLlmModelRequest";
+import type { UpdateLlmModelRequest } from "@/app/api/__generated__/models/updateLlmModelRequest";
+import type { ToggleLlmModelRequest } from "@/app/api/__generated__/models/toggleLlmModelRequest";
+import type { LlmMigrationsResponse } from "@/app/api/__generated__/models/llmMigrationsResponse";
+import type { LlmCreatorsResponse } from "@/app/api/__generated__/models/llmCreatorsResponse";
+import type { UpsertLlmCreatorRequest } from "@/app/api/__generated__/models/upsertLlmCreatorRequest";
+import type { LlmModelUsageResponse } from "@/app/api/__generated__/models/llmModelUsageResponse";
 
 const ADMIN_LLM_PATH = "/admin/llms";
 
@@ -21,9 +38,12 @@ const ADMIN_LLM_PATH = "/admin/llms";
 // Provider Actions
 // =============================================================================
 
-export async function fetchLlmProviders(): Promise<{ providers: LlmProvider[] }> {
-  const api = new BackendApi();
-  return await api.listAdminLlmProviders(true);
+export async function fetchLlmProviders(): Promise<LlmProvidersResponse> {
+  const response = await getV2ListLlmProviders({ include_models: true });
+  if (response.status !== 200) {
+    throw new Error("Failed to fetch LLM providers");
+  }
+  return response.data;
 }
 
 export async function createLlmProviderAction(formData: FormData) {
@@ -45,8 +65,10 @@ export async function createLlmProviderAction(formData: FormData) {
     metadata: {},
   };
 
-  const api = new BackendApi();
-  await api.createAdminLlmProvider(payload);
+  const response = await postV2CreateLlmProvider(payload);
+  if (response.status !== 200) {
+    throw new Error("Failed to create LLM provider");
+  }
   revalidatePath(ADMIN_LLM_PATH);
 }
 
@@ -54,9 +76,12 @@ export async function createLlmProviderAction(formData: FormData) {
 // Model Actions
 // =============================================================================
 
-export async function fetchLlmModels(): Promise<{ models: LlmModel[] }> {
-  const api = new BackendApi();
-  return await api.listAdminLlmModels();
+export async function fetchLlmModels(): Promise<LlmModelsResponse> {
+  const response = await getV2ListLlmModels();
+  if (response.status !== 200) {
+    throw new Error("Failed to fetch LLM models");
+  }
+  return response.data;
 }
 
 export async function createLlmModelAction(formData: FormData) {
@@ -64,9 +89,11 @@ export async function createLlmModelAction(formData: FormData) {
   const creatorId = formData.get("creator_id");
 
   // Fetch provider to get default credentials
-  const api = new BackendApi();
-  const providersResponse = await api.listAdminLlmProviders(false);
-  const provider = providersResponse.providers.find((p) => p.id === providerId);
+  const providersResponse = await getV2ListLlmProviders({ include_models: false });
+  if (providersResponse.status !== 200) {
+    throw new Error("Failed to fetch providers");
+  }
+  const provider = providersResponse.data.providers.find((p) => p.id === providerId);
 
   if (!provider) {
     throw new Error("Provider not found");
@@ -99,7 +126,10 @@ export async function createLlmModelAction(formData: FormData) {
     ],
   };
 
-  await api.createAdminLlmModel(payload);
+  const response = await postV2CreateLlmModel(payload);
+  if (response.status !== 200) {
+    throw new Error("Failed to create LLM model");
+  }
   revalidatePath(ADMIN_LLM_PATH);
 }
 
@@ -146,8 +176,10 @@ export async function updateLlmModelAction(formData: FormData) {
       : undefined,
   };
 
-  const api = new BackendApi();
-  await api.updateAdminLlmModel(modelId, payload);
+  const response = await patchV2UpdateLlmModel(modelId, payload);
+  if (response.status !== 200) {
+    throw new Error("Failed to update LLM model");
+  }
   revalidatePath(ADMIN_LLM_PATH);
 }
 
@@ -165,28 +197,37 @@ export async function toggleLlmModelAction(formData: FormData): Promise<void> {
     custom_credit_cost: customCreditCost ? Number(customCreditCost) : undefined,
   };
 
-  const api = new BackendApi();
-  await api.toggleAdminLlmModel(modelId, payload);
+  const response = await patchV2ToggleLlmModelAvailability(modelId, payload);
+  if (response.status !== 200) {
+    throw new Error("Failed to toggle LLM model");
+  }
   revalidatePath(ADMIN_LLM_PATH);
 }
 
 export async function deleteLlmModelAction(formData: FormData) {
-  try {
-    const modelId = String(formData.get("model_id"));
-    const replacementModelSlug = String(formData.get("replacement_model_slug"));
+  const modelId = String(formData.get("model_id"));
+  const replacementModelSlug = String(formData.get("replacement_model_slug"));
 
-    if (!replacementModelSlug) {
-      throw new Error("Replacement model is required");
-    }
-
-    const api = new BackendApi();
-    const result = await api.deleteAdminLlmModel(modelId, replacementModelSlug);
-    revalidatePath(ADMIN_LLM_PATH);
-    return result;
-  } catch (error) {
-    console.error("Delete model error:", error);
-    throw error instanceof Error ? error : new Error("Failed to delete model");
+  if (!replacementModelSlug) {
+    throw new Error("Replacement model is required");
   }
+
+  const response = await deleteV2DeleteLlmModelAndMigrateWorkflows(modelId, {
+    replacement_model_slug: replacementModelSlug,
+  });
+  if (response.status !== 200) {
+    throw new Error("Failed to delete model");
+  }
+  revalidatePath(ADMIN_LLM_PATH);
+  return response.data;
+}
+
+export async function fetchLlmModelUsage(modelId: string): Promise<LlmModelUsageResponse> {
+  const response = await getV2GetModelUsageCount(modelId);
+  if (response.status !== 200) {
+    throw new Error("Failed to fetch model usage");
+  }
+  return response.data;
 }
 
 // =============================================================================
@@ -196,24 +237,21 @@ export async function deleteLlmModelAction(formData: FormData) {
 export async function fetchLlmMigrations(
   includeReverted: boolean = false
 ): Promise<LlmMigrationsResponse> {
-  const api = new BackendApi();
-  return await api.listAdminLlmMigrations(includeReverted);
+  const response = await getV2ListModelMigrations({ include_reverted: includeReverted });
+  if (response.status !== 200) {
+    throw new Error("Failed to fetch migrations");
+  }
+  return response.data;
 }
 
-export async function revertLlmMigrationAction(
-  formData: FormData
-): Promise<void> {
-  try {
-    const migrationId = String(formData.get("migration_id"));
-    const api = new BackendApi();
-    await api.revertAdminLlmMigration(migrationId);
-    revalidatePath(ADMIN_LLM_PATH);
-  } catch (error) {
-    console.error("Revert migration error:", error);
-    throw error instanceof Error
-      ? error
-      : new Error("Failed to revert migration");
+export async function revertLlmMigrationAction(formData: FormData): Promise<void> {
+  const migrationId = String(formData.get("migration_id"));
+
+  const response = await postV2RevertAModelMigration(migrationId, null);
+  if (response.status !== 200) {
+    throw new Error("Failed to revert migration");
   }
+  revalidatePath(ADMIN_LLM_PATH);
 }
 
 // =============================================================================
@@ -221,8 +259,11 @@ export async function revertLlmMigrationAction(
 // =============================================================================
 
 export async function fetchLlmCreators(): Promise<LlmCreatorsResponse> {
-  const api = new BackendApi();
-  return await api.listAdminLlmCreators();
+  const response = await getV2ListModelCreators();
+  if (response.status !== 200) {
+    throw new Error("Failed to fetch creators");
+  }
+  return response.data;
 }
 
 export async function createLlmCreatorAction(formData: FormData): Promise<void> {
@@ -241,8 +282,10 @@ export async function createLlmCreatorAction(formData: FormData): Promise<void>
     metadata: {},
   };
 
-  const api = new BackendApi();
-  await api.createAdminLlmCreator(payload);
+  const response = await postV2CreateModelCreator(payload);
+  if (response.status !== 200) {
+    throw new Error("Failed to create creator");
+  }
   revalidatePath(ADMIN_LLM_PATH);
 }
 
@@ -263,19 +306,19 @@ export async function updateLlmCreatorAction(formData: FormData): Promise<void>
     metadata: {},
   };
 
-  const api = new BackendApi();
-  await api.updateAdminLlmCreator(creatorId, payload);
+  const response = await patchV2UpdateModelCreator(creatorId, payload);
+  if (response.status !== 200) {
+    throw new Error("Failed to update creator");
+  }
   revalidatePath(ADMIN_LLM_PATH);
 }
 
 export async function deleteLlmCreatorAction(formData: FormData): Promise<void> {
-  try {
-    const creatorId = String(formData.get("creator_id"));
-    const api = new BackendApi();
-    await api.deleteAdminLlmCreator(creatorId);
-    revalidatePath(ADMIN_LLM_PATH);
-  } catch (error) {
-    console.error("Delete creator error:", error);
-    throw error instanceof Error ? error : new Error("Failed to delete creator");
+  const creatorId = String(formData.get("creator_id"));
+
+  const response = await deleteV2DeleteModelCreator(creatorId);
+  if (response.status !== 200) {
+    throw new Error("Failed to delete creator");
   }
+  revalidatePath(ADMIN_LLM_PATH);
 }

autogpt_platform/frontend/src/app/(platform)/admin/llms/components/AddModelModal.tsx

@@ -3,7 +3,8 @@
 import { useState } from "react";
 import { Dialog } from "@/components/molecules/Dialog/Dialog";
 import { Button } from "@/components/atoms/Button/Button";
-import type { LlmProvider, LlmModelCreator } from "@/lib/autogpt-server-api/types";
+import type { LlmProvider } from "@/app/api/__generated__/models/llmProvider";
+import type { LlmModelCreator } from "@/app/api/__generated__/models/llmModelCreator";
 import { createLlmModelAction } from "../actions";
 import { useRouter } from "next/navigation";
 

autogpt_platform/frontend/src/app/(platform)/admin/llms/components/CreatorsTable.tsx

@@ -1,7 +1,7 @@
 "use client";
 
 import { useState } from "react";
-import type { LlmModelCreator } from "@/lib/autogpt-server-api/types";
+import type { LlmModelCreator } from "@/app/api/__generated__/models/llmModelCreator";
 import {
   Table,
   TableBody,

autogpt_platform/frontend/src/app/(platform)/admin/llms/components/DeleteModelModal.tsx

@@ -3,8 +3,8 @@
 import { useState } from "react";
 import { Dialog } from "@/components/molecules/Dialog/Dialog";
 import { Button } from "@/components/atoms/Button/Button";
-import type { LlmModel } from "@/lib/autogpt-server-api/types";
-import { deleteLlmModelAction } from "../actions";
+import type { LlmModel } from "@/app/api/__generated__/models/llmModel";
+import { deleteLlmModelAction, fetchLlmModelUsage } from "../actions";
 
 export function DeleteModelModal({
   model,
@@ -26,9 +26,7 @@ export function DeleteModelModal({
 
   async function fetchUsage() {
     try {
-      const BackendApi = (await import("@/lib/autogpt-server-api")).default;
-      const api = new BackendApi();
-      const usage = await api.getAdminLlmModelUsage(model.id);
+      const usage = await fetchLlmModelUsage(model.id);
       setUsageCount(usage.node_count);
     } catch {
       setUsageCount(null);

autogpt_platform/frontend/src/app/(platform)/admin/llms/components/DisableModelModal.tsx

@@ -3,8 +3,8 @@
 import { useState } from "react";
 import { Dialog } from "@/components/molecules/Dialog/Dialog";
 import { Button } from "@/components/atoms/Button/Button";
-import type { LlmModel } from "@/lib/autogpt-server-api/types";
-import { toggleLlmModelAction } from "../actions";
+import type { LlmModel } from "@/app/api/__generated__/models/llmModel";
+import { toggleLlmModelAction, fetchLlmModelUsage } from "../actions";
 
 export function DisableModelModal({
   model,
@@ -29,9 +29,7 @@ export function DisableModelModal({
 
   async function fetchUsage() {
     try {
-      const BackendApi = (await import("@/lib/autogpt-server-api")).default;
-      const api = new BackendApi();
-      const usage = await api.getAdminLlmModelUsage(model.id);
+      const usage = await fetchLlmModelUsage(model.id);
       setUsageCount(usage.node_count);
     } catch {
       setUsageCount(null);

autogpt_platform/frontend/src/app/(platform)/admin/llms/components/EditModelModal.tsx

@@ -3,11 +3,9 @@
 import { useState } from "react";
 import { Dialog } from "@/components/molecules/Dialog/Dialog";
 import { Button } from "@/components/atoms/Button/Button";
-import type {
-  LlmModel,
-  LlmModelCreator,
-  LlmProvider,
-} from "@/lib/autogpt-server-api/types";
+import type { LlmModel } from "@/app/api/__generated__/models/llmModel";
+import type { LlmModelCreator } from "@/app/api/__generated__/models/llmModelCreator";
+import type { LlmProvider } from "@/app/api/__generated__/models/llmProvider";
 import { updateLlmModelAction } from "../actions";
 
 export function EditModelModal({
@@ -20,7 +18,7 @@ export function EditModelModal({
   creators: LlmModelCreator[];
 }) {
   const [open, setOpen] = useState(false);
-  const cost = model.costs[0];
+  const cost = model.costs?.[0];
   const provider = providers.find((p) => p.id === model.provider_id);
 
   return (

autogpt_platform/frontend/src/app/(platform)/admin/llms/components/LlmRegistryDashboard.tsx

@@ -1,11 +1,9 @@
 "use client";
 
-import type {
-  LlmModel,
-  LlmModelCreator,
-  LlmModelMigration,
-  LlmProvider,
-} from "@/lib/autogpt-server-api/types";
+import type { LlmModel } from "@/app/api/__generated__/models/llmModel";
+import type { LlmModelCreator } from "@/app/api/__generated__/models/llmModelCreator";
+import type { LlmModelMigration } from "@/app/api/__generated__/models/llmModelMigration";
+import type { LlmProvider } from "@/app/api/__generated__/models/llmProvider";
 import { AddProviderModal } from "./AddProviderModal";
 import { AddModelModal } from "./AddModelModal";
 import { AddCreatorModal } from "./AddCreatorModal";

autogpt_platform/frontend/src/app/(platform)/admin/llms/components/MigrationsTable.tsx

@@ -1,7 +1,7 @@
 "use client";
 
 import { useState } from "react";
-import type { LlmModelMigration } from "@/lib/autogpt-server-api/types";
+import type { LlmModelMigration } from "@/app/api/__generated__/models/llmModelMigration";
 import { Button } from "@/components/atoms/Button/Button";
 import {
   Table,

autogpt_platform/frontend/src/app/(platform)/admin/llms/components/ModelsTable.tsx

@@ -1,8 +1,6 @@
-import type {
-  LlmModel,
-  LlmModelCreator,
-  LlmProvider,
-} from "@/lib/autogpt-server-api/types";
+import type { LlmModel } from "@/app/api/__generated__/models/llmModel";
+import type { LlmModelCreator } from "@/app/api/__generated__/models/llmModelCreator";
+import type { LlmProvider } from "@/app/api/__generated__/models/llmProvider";
 import {
   Table,
   TableBody,
@@ -55,7 +53,7 @@ export function ModelsTable({
         </TableHeader>
         <TableBody>
           {models.map((model) => {
-            const cost = model.costs[0];
+            const cost = model.costs?.[0];
             const provider = providerLookup.get(model.provider_id);
             return (
               <TableRow

autogpt_platform/frontend/src/app/(platform)/admin/llms/components/ProviderList.tsx

@@ -6,7 +6,7 @@ import {
   TableHeader,
   TableRow,
 } from "@/components/atoms/Table/Table";
-import type { LlmProvider } from "@/lib/autogpt-server-api/types";
+import type { LlmProvider } from "@/app/api/__generated__/models/llmProvider";
 
 export function ProviderList({ providers }: { providers: LlmProvider[] }) {
   if (!providers.length) {

autogpt_platform/frontend/src/app/api/openapi.json

@@ -2901,7 +2901,7 @@
         "tags": ["v2", "store", "public"],
         "summary": "Get creator details",
         "description": "Get the details of a creator.\n- Creator Details Page",
-        "operationId": "getV2Get creator details",
+        "operationId": "getV2GetCreatorDetails",
         "parameters": [
           {
             "name": "username",
@@ -4716,7 +4716,7 @@
         "tags": ["v2", "admin", "llm", "admin"],
         "summary": "Get creator details",
         "description": "Get details of a specific model creator.",
-        "operationId": "getV2Get creator details",
+        "operationId": "getV2GetLlmCreatorDetails",
         "security": [{ "HTTPBearerJWT": [] }],
         "parameters": [
           {