fastapi search

server/chatStream.js

@@ -23,33 +23,44 @@ import { LIGHTNING_BASE } from "./config.js";
 const __dirname = path.dirname(fileURLToPath(import.meta.url));
 const WORKER_PATH = path.join(__dirname, "searchWorker.js");
 
-function gradioSearch(query) {
-  return new Promise((resolve, reject) => {
-    const worker = new Worker(WORKER_PATH, { workerData: { query } });
-
-    const timeout = setTimeout(() => {
-      worker.terminate();
-      reject(new Error("Search timed out after 45s"));
-    }, 45_000);
-
-    worker.on("message", (msg) => {
-      clearTimeout(timeout);
-      if (msg.ok) resolve(msg.result);
-      else reject(new Error(msg.error));
-    });
-
-    worker.on("error", (err) => {
-      clearTimeout(timeout);
-      reject(err);
-    });
-
-    worker.on("exit", (code) => {
-      // By the time this fires the promise is already settled via "message".
-      // Only reject if the worker crashed without posting anything.
-      clearTimeout(timeout);
-      if (code !== 0) reject(new Error(`Search worker exited with code ${code}`));
-    });
+async function gradioSearch(query) {
+  const req = await fetch("https://incognitolm-Web-Search.hf.space/api/search", {
+    method: "POST",
+    headers: { "Content-Type": "application/json" },
+    body: JSON.stringify({ query }),
   });
+  const res = await req.json();
+  if (!req.ok || !res.result) {
+    throw new Error(`Search API error: ${res.error || req.statusText}`);
+  }
+  return res.results[0];
+
+  // return new Promise((resolve, reject) => {
+  //   const worker = new Worker(WORKER_PATH, { workerData: { query } });
+
+  //   const timeout = setTimeout(() => {
+  //     worker.terminate();
+  //     reject(new Error("Search timed out after 45s"));
+  //   }, 45_000);
+
+  //   worker.on("message", (msg) => {
+  //     clearTimeout(timeout);
+  //     if (msg.ok) resolve(msg.result);
+  //     else reject(new Error(msg.error));
+  //   });
+
+  //   worker.on("error", (err) => {
+  //     clearTimeout(timeout);
+  //     reject(err);
+  //   });
+
+  //   worker.on("exit", (code) => {
+  //     // By the time this fires the promise is already settled via "message".
+  //     // Only reject if the worker crashed without posting anything.
+  //     clearTimeout(timeout);
+  //     if (code !== 0) reject(new Error(`Search worker exited with code ${code}`));
+  //   });
+  // });
 }
 
 const SYSTEM_PROMPT =