@lastbrain/ai-ui-react 1.0.11 → 1.0.12

package/src/utils/modelManagement.ts

@@ -0,0 +1,130 @@
+/**
+ * Utilitaires pour la gestion des modèles IA
+ */
+
+export interface ModelToggleOptions {
+  apiKey?: string; // LB_API_KEY pour l'authentification
+  baseUrl?: string; // URL de base de l'API
+}
+
+/**
+ * Active ou désactive un modèle pour l'utilisateur courant
+ */
+export async function toggleUserModel(
+  modelId: string,
+  isActive: boolean,
+  options: ModelToggleOptions = {}
+): Promise<void> {
+  const { apiKey, baseUrl = "" } = options;
+
+  const headers: Record<string, string> = {
+    "Content-Type": "application/json",
+  };
+
+  // Ajouter la clé API si fournie
+  if (apiKey) {
+    headers["Authorization"] = `Bearer ${apiKey}`;
+  }
+
+  const response = await fetch(
+    `${baseUrl}/api/public/v1/ai/user/models/toggle`,
+    {
+      method: "PUT",
+      headers,
+      body: JSON.stringify({
+        model_id: modelId,
+        is_active: isActive,
+      }),
+    }
+  );
+
+  if (!response.ok) {
+    const error = await response.text();
+    throw new Error(`Erreur lors de la modification du modèle: ${error}`);
+  }
+}
+
+/**
+ * Récupère tous les modèles disponibles avec API key ou via proxy
+ */
+export async function getAvailableModels(
+  options: ModelToggleOptions = {}
+): Promise<
+  Array<{
+    id: string;
+    name: string;
+    description?: string;
+    provider: string;
+    category: "text" | "image" | "audio" | "video";
+    isActive?: boolean;
+    isPro?: boolean;
+    costPer1M?: number;
+  }>
+> {
+  const { apiKey, baseUrl = "" } = options;
+
+  // Si baseUrl est fourni (contexte externe), utiliser la route auth qui sera proxifiée
+  const isExternal = baseUrl && baseUrl !== "";
+  const endpoint = isExternal
+    ? `${baseUrl}/ai/auth/ai-models-available`
+    : `/api/public/v1/ai/models/available`;
+
+  const headers: Record<string, string> = {};
+
+  // Ajouter la clé API seulement si pas de baseUrl (appel direct)
+  if (!isExternal && apiKey) {
+    headers["Authorization"] = `Bearer ${apiKey}`;
+  }
+
+  const response = await fetch(endpoint, {
+    method: "GET",
+    headers,
+    credentials: isExternal ? "include" : "same-origin",
+  });
+
+  if (!response.ok) {
+    throw new Error(
+      `Erreur lors de la récupération des modèles: ${response.status}`
+    );
+  }
+
+  const data = await response.json();
+  return data.models || [];
+}
+
+/**
+ * Récupère les modèles activés par l'utilisateur avec API key
+ */
+export async function getUserModels(
+  options: ModelToggleOptions = {}
+): Promise<string[]> {
+  const { apiKey, baseUrl = "" } = options;
+
+  // Si baseUrl est fourni (contexte externe), utiliser la route auth qui sera proxifiée
+  const isExternal = baseUrl && baseUrl !== "";
+  const endpoint = isExternal
+    ? `${baseUrl}/ai/auth/user-models`
+    : `/api/public/v1/ai/user/models`;
+
+  const headers: Record<string, string> = {};
+
+  // Ajouter la clé API seulement si pas de baseUrl (appel direct)
+  if (!isExternal && apiKey) {
+    headers["Authorization"] = `Bearer ${apiKey}`;
+  }
+
+  const response = await fetch(endpoint, {
+    method: "GET",
+    headers,
+    credentials: isExternal ? "include" : "same-origin",
+  });
+
+  if (!response.ok) {
+    throw new Error(
+      `Erreur lors de la récupération des modèles utilisateur: ${response.status}`
+    );
+  }
+
+  const data = await response.json();
+  return data.enabled_models || [];
+}