@lastbrain/ai-ui-react 1.0.69 → 1.0.70

package/src/utils/modelManagement.ts

@@ -7,6 +7,51 @@ export interface ModelToggleOptions {
   baseUrl?: string; // URL de base de l'API
 }
 
+interface Provider {
+  models?: ModelInfo[];
+}
+
+interface ProvidersResponse {
+  providers?: Provider[];
+}
+
+interface ModelsResponse {
+  models?: ModelInfo[];
+}
+
+type ModelCategory = "text" | "image" | "audio" | "video";
+
+interface ModelInfo {
+  id: string;
+  name: string;
+  description?: string;
+  provider: string;
+  category: ModelCategory;
+  isActive?: boolean;
+  isPro?: boolean;
+  costPer1M?: number;
+}
+
+function isModelCategory(value: unknown): value is ModelCategory {
+  return (
+    value === "text" ||
+    value === "image" ||
+    value === "audio" ||
+    value === "video"
+  );
+}
+
+function isModelInfo(value: unknown): value is ModelInfo {
+  if (!value || typeof value !== "object") return false;
+  const v = value as Record<string, unknown>;
+  return (
+    typeof v.id === "string" &&
+    typeof v.name === "string" &&
+    typeof v.provider === "string" &&
+    isModelCategory(v.category)
+  );
+}
+
 function isAuthTokenCandidate(value?: string): boolean {
   if (!value) return false;
   const token = value.trim();
@@ -66,18 +111,7 @@ export async function toggleUserModel(
  */
 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;
-  }>
-> {
+): Promise<ModelInfo[]> {
   const { apiKey, baseUrl = "" } = options;
 
   // Déterminer le contexte
@@ -123,11 +157,15 @@ export async function getAvailableModels(
   const data = await response.json();
 
   if (Array.isArray(data?.models)) {
-    return data.models;
+    return data.models.filter(isModelInfo);
   }
   if (Array.isArray(data?.providers)) {
-    return data.providers.flatMap((provider: any) =>
-      Array.isArray(provider.models) ? provider.models : []
+    return (
+      (data as ProvidersResponse).providers?.flatMap((provider: Provider) =>
+        Array.isArray(provider.models)
+          ? provider.models.filter(isModelInfo)
+          : []
+      ) || []
     );
   }
   return [];