react-brai 1.2.3

package/copy-dts.js

@@ -0,0 +1,29 @@
+// copy-dts.js
+const fs = require("fs");
+const path = require("path");
+
+function copyDts(src, dest) {
+  if (!fs.existsSync(dest)) fs.mkdirSync(dest);
+
+  const entries = fs.readdirSync(src, { withFileTypes: true });
+
+  for (const entry of entries) {
+    const srcPath = path.join(src, entry.name);
+    const destPath = path.join(dest, entry.name);
+
+    if (entry.isDirectory()) {
+      copyDts(srcPath, destPath);
+    } else if (entry.name.endsWith(".d.ts") || entry.name.endsWith(".d.mts")) {
+      fs.copyFileSync(srcPath, destPath);
+      console.log(`Saved type definition: ${entry.name}`);
+    }
+  }
+}
+
+try {
+  copyDts("dist-raw", "dist");
+  console.log("✅ Type definitions copied successfully.");
+} catch (err) {
+  console.error("❌ Failed to copy types:", err);
+  process.exit(1);
+}

package/dist/index.d.mts

@@ -0,0 +1,57 @@
+import { InitProgressReport } from '@mlc-ai/web-llm';
+export { InitProgressReport } from '@mlc-ai/web-llm';
+import * as react_cai from 'react-cai';
+
+type ModelID = "Llama-3-8B-Instruct-q4f16_1-MLC" | "Gemma-2b-it-q4f16_1-MLC" | string;
+interface Message {
+    role: "system" | "user" | "assistant";
+    content: string;
+}
+interface WorkerMessage {
+    type: "READY" | "PROGRESS" | "TOKEN" | "DONE" | "ERROR";
+    payload?: any;
+}
+interface CustomModelRecord {
+    model: string;
+    model_id: string;
+    model_lib: string;
+}
+interface ModelRecord {
+    model: string;
+    model_id: string;
+    model_lib: string;
+}
+interface WorkerPayload {
+    type: "LOAD" | "GENERATE";
+    modelId?: string | ModelRecord;
+    messages?: any[];
+    options?: {
+        context_window_size?: number;
+    };
+}
+
+declare function useLocalAI(): {
+    isSupported: boolean | null;
+    loadModel: (model: string | CustomModelRecord, options?: {
+        contextWindow?: number;
+    }) => void;
+    chat: (messages: Message[]) => void;
+    isReady: boolean;
+    isLoading: boolean;
+    response: string;
+    progress: InitProgressReport | null;
+    error: string | null;
+};
+
+type TransformerStatus = "idle" | "loading" | "ready" | "error";
+declare function useUniversalAI(): {
+    chat: (messages: react_cai.Message[]) => void;
+    response: string;
+    isLlmReady: boolean;
+    loadPipeline: (task: string, model: string) => string;
+    runPipeline: (task: string, model: string, input: any, options?: {}) => Promise<any>;
+    pipelines: Record<string, TransformerStatus>;
+    progress: Record<string, number>;
+};
+
+export { type CustomModelRecord, type Message, type ModelID, type ModelRecord, type WorkerMessage, type WorkerPayload, useLocalAI, useUniversalAI };

package/dist/index.d.ts

@@ -0,0 +1,57 @@
+import { InitProgressReport } from '@mlc-ai/web-llm';
+export { InitProgressReport } from '@mlc-ai/web-llm';
+import * as react_cai from 'react-cai';
+
+type ModelID = "Llama-3-8B-Instruct-q4f16_1-MLC" | "Gemma-2b-it-q4f16_1-MLC" | string;
+interface Message {
+    role: "system" | "user" | "assistant";
+    content: string;
+}
+interface WorkerMessage {
+    type: "READY" | "PROGRESS" | "TOKEN" | "DONE" | "ERROR";
+    payload?: any;
+}
+interface CustomModelRecord {
+    model: string;
+    model_id: string;
+    model_lib: string;
+}
+interface ModelRecord {
+    model: string;
+    model_id: string;
+    model_lib: string;
+}
+interface WorkerPayload {
+    type: "LOAD" | "GENERATE";
+    modelId?: string | ModelRecord;
+    messages?: any[];
+    options?: {
+        context_window_size?: number;
+    };
+}
+
+declare function useLocalAI(): {
+    isSupported: boolean | null;
+    loadModel: (model: string | CustomModelRecord, options?: {
+        contextWindow?: number;
+    }) => void;
+    chat: (messages: Message[]) => void;
+    isReady: boolean;
+    isLoading: boolean;
+    response: string;
+    progress: InitProgressReport | null;
+    error: string | null;
+};
+
+type TransformerStatus = "idle" | "loading" | "ready" | "error";
+declare function useUniversalAI(): {
+    chat: (messages: react_cai.Message[]) => void;
+    response: string;
+    isLlmReady: boolean;
+    loadPipeline: (task: string, model: string) => string;
+    runPipeline: (task: string, model: string, input: any, options?: {}) => Promise<any>;
+    pipelines: Record<string, TransformerStatus>;
+    progress: Record<string, number>;
+};
+
+export { type CustomModelRecord, type Message, type ModelID, type ModelRecord, type WorkerMessage, type WorkerPayload, useLocalAI, useUniversalAI };