react-brai 1.0.0 → 1.0.1

package/package.json

@@ -1,20 +1,15 @@
 {
   "name": "react-brai",
-  "version": "1.0.0",
+  "version": "1.0.1",
   "main": "./dist/index.js",
   "module": "./dist/index.mjs",
   "types": "./dist/index.d.ts",
- "scripts": {
-  "build:worker": "tsup src/worker.ts --format esm --no-splitting --clean --dts",
-
-  "embed": "node scripts/embed-worker.js",
-  
-
-  "build:lib": "tsup src/index.ts --format cjs,esm --dts --external react",
-  
-
-  "build": "npm run build:worker && npm run embed && npm run build:lib"
-},
+  "scripts": {
+    "build:worker": "tsup src/worker.ts --format esm --no-splitting --clean --dts",
+    "embed": "node scripts/embed-worker.js",
+    "build:lib": "tsup src/index.ts --format cjs,esm --dts --external react",
+    "build": "npm run build:worker && npm run embed && npm run build:lib"
+  },
   "keywords": [],
   "author": "",
   "license": "ISC",
@@ -25,6 +20,7 @@
   "devDependencies": {
     "@types/node": "^25.0.3",
     "@types/react": "^19.2.7",
+    "@webgpu/types": "^0.1.68",
     "react": "^19.2.3",
     "tsup": "^8.5.1",
     "typescript": "^5.9.3"

package/src/useLocalAI.ts

@@ -7,6 +7,8 @@ import { WORKER_CODE } from "./worker-embedded";
 export function useLocalAI() {
     const workerRef = useRef<Worker | null>(null);
 
+    const [isSupported, setIsSupported] = useState<boolean | null>(null);
+
     const [isReady, setIsReady] = useState(false);
     const [isLoading, setIsLoading] = useState(false);
     const [response, setResponse] = useState("");
@@ -15,6 +17,15 @@ export function useLocalAI() {
 
     // 1. AUTOMATIC INITIALIZATION (No more initWorker)
     useEffect(() => {
+        if (!navigator.gpu) {
+            console.error("WebGPU is not supported on this device.");
+            setIsSupported(false);
+            setError("WebGPU is not supported on this device.");
+            return; // Stop here, don't load the worker
+        }
+        
+        setIsSupported(true);
+
         if (!WORKER_CODE) return;
 
         // A. Create the worker from the embedded string
@@ -71,6 +82,7 @@ export function useLocalAI() {
 
     return {
         // initWorker is gone!
+        isSupported,
         loadModel,
         chat,
         isReady,

package/tsconfig.json

@@ -12,7 +12,8 @@
     "skipLibCheck": true,
     "forceConsistentCasingInFileNames": true,
     "incremental": false,  // <--- THIS IS THE FIX
-    "jsx": "react"
+    "jsx": "react",
+    "types":["@webgpu/types"]
   },
   "include": ["src"],
   "exclude": ["node_modules", "dist"]

package/dist/index.d.mts

@@ -1,29 +0,0 @@
-import { InitProgressReport } from '@mlc-ai/web-llm';
-export { InitProgressReport } from '@mlc-ai/web-llm';
-
-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 WorkerPayload {
-    type: "LOAD" | "GENERATE";
-    modelId?: string;
-    messages?: Message[];
-}
-
-declare function useLocalAI(): {
-    loadModel: (modelId: string) => void;
-    chat: (messages: Message[]) => void;
-    isReady: boolean;
-    isLoading: boolean;
-    response: string;
-    progress: InitProgressReport | null;
-    error: string | null;
-};
-
-export { type Message, type ModelID, type WorkerMessage, type WorkerPayload, useLocalAI };

package/dist/index.d.ts

@@ -1,29 +0,0 @@
-import { InitProgressReport } from '@mlc-ai/web-llm';
-export { InitProgressReport } from '@mlc-ai/web-llm';
-
-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 WorkerPayload {
-    type: "LOAD" | "GENERATE";
-    modelId?: string;
-    messages?: Message[];
-}
-
-declare function useLocalAI(): {
-    loadModel: (modelId: string) => void;
-    chat: (messages: Message[]) => void;
-    isReady: boolean;
-    isLoading: boolean;
-    response: string;
-    progress: InitProgressReport | null;
-    error: string | null;
-};
-
-export { type Message, type ModelID, type WorkerMessage, type WorkerPayload, useLocalAI };