@lastbrain/ai-ui-react 1.0.68 → 1.0.70

package/src/types.ts

@@ -4,6 +4,9 @@ export type ToastType = {
   code?: string;
 };
 export type UiMode = "modal" | "drawer";
+export type AiSize = "sm" | "md" | "lg";
+export type AiRadius = "none" | "sm" | "md" | "lg" | "full";
+export type AiVariant = "default" | "light";
 
 export interface BaseAiProps {
   baseUrl?: string;

package/src/utils/errorHandler.ts

@@ -8,12 +8,25 @@ export interface ParsedError {
   isUserFriendly: boolean;
 }
 
+type NormalizedErrorLike = {
+  code: string;
+  message: string;
+};
+
+function isNormalizedErrorLike(value: unknown): value is NormalizedErrorLike {
+  if (!value || typeof value !== "object") {
+    return false;
+  }
+  const v = value as Record<string, unknown>;
+  return typeof v.code === "string" && typeof v.message === "string";
+}
+
 /**
  * Parse et uniformise la gestion des erreurs des composants AI
  */
-export function parseAIError(error: any): ParsedError {
+export function parseAIError(error: unknown): ParsedError {
   // Si l'erreur est déjà un objet normalisé
-  if (error?.code && error?.message) {
+  if (isNormalizedErrorLike(error)) {
     return {
       message: getUserFriendlyMessage(error.code, error.message),
       code: error.code,
@@ -160,7 +173,7 @@ export interface ErrorToastCallback {
  * @param showInternalToast Callback interne optionnelle pour afficher un toast dans le composant
  */
 export function handleAIError(
-  error: any,
+  error: unknown,
   onToast?: ErrorToastCallback,
   showInternalToast?: (error: { message: string; code?: string }) => void
 ): void {

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 [];