@reverbia/sdk 1.0.0-next.20251208093930 → 1.0.0-next.20251208094446

package/dist/react/{chunk-2WZDKSTC.mjs → chunk-BG7SZT33.mjs}

@@ -16,27 +16,54 @@ Task: "${userMessage}"
 
 Best tool:`;
 }
-function extractParams(userMessage, tool) {
+function buildParamExtractionPrompt(userMessage, paramName, paramDescription) {
+  const desc = paramDescription ? ` (${paramDescription})` : "";
+  return `Extract the value for "${paramName}"${desc} from the user message. Reply with ONLY the extracted value, nothing else.
+
+User message: "${userMessage}"
+
+Value for ${paramName}:`;
+}
+async function extractParams(userMessage, tool, options) {
   const params = {};
-  if (!tool.parameters) return params;
-  for (const param of tool.parameters) {
-    if (param.name === "expression" || param.name === "query") {
-      params[param.name] = userMessage;
-    } else if (param.name === "location" || param.name === "city") {
-      const words = userMessage.split(/\s+/);
-      const capitalizedWords = words.filter(
-        (w) => w.length > 1 && w[0] === w[0].toUpperCase()
+  if (!tool.parameters || tool.parameters.length === 0) return params;
+  const { model, device } = options;
+  try {
+    const pipeline = await getTextGenerationPipeline({
+      model,
+      device,
+      dtype: "q4"
+    });
+    for (const param of tool.parameters) {
+      const prompt = buildParamExtractionPrompt(
+        userMessage,
+        param.name,
+        param.description
       );
-      params[param.name] = capitalizedWords.length > 0 ? capitalizedWords.join(" ") : userMessage;
-    } else if (param.name === "text" || param.name === "input") {
-      params[param.name] = userMessage;
-    } else {
+      const output = await pipeline(prompt, {
+        max_new_tokens: 32,
+        // Allow reasonable length for parameter values
+        temperature: 0,
+        do_sample: false,
+        return_full_text: false
+      });
+      const generatedText = output?.[0]?.generated_text || output?.generated_text || "";
+      const extractedValue = generatedText.trim().split("\n")[0].trim();
+      console.log(
+        `[Tool Selector] Extracted param "${param.name}":`,
+        extractedValue
+      );
+      params[param.name] = extractedValue || userMessage;
+    }
+  } catch (error) {
+    console.error("[Tool Selector] Error extracting params:", error);
+    for (const param of tool.parameters) {
       params[param.name] = userMessage;
     }
   }
   return params;
 }
-function parseToolSelectionResponse(response, tools, userMessage) {
+async function parseToolSelectionResponse(response, tools, userMessage, options) {
   console.log("[Tool Selector] Raw response:", response);
   const cleaned = response.toLowerCase().trim().split(/[\s\n,.]+/)[0].replace(/[^a-z0-9_-]/g, "");
   console.log("[Tool Selector] Parsed tool name:", cleaned);
@@ -51,7 +78,7 @@ function parseToolSelectionResponse(response, tools, userMessage) {
     );
     if (fuzzyTool) {
       console.log(`[Tool Selector] Fuzzy matched tool: ${fuzzyTool.name}`);
-      const params2 = extractParams(userMessage, fuzzyTool);
+      const params2 = await extractParams(userMessage, fuzzyTool, options);
       return {
         toolSelected: true,
         toolName: fuzzyTool.name,
@@ -62,7 +89,7 @@ function parseToolSelectionResponse(response, tools, userMessage) {
     console.warn(`[Tool Selector] Unknown tool: ${cleaned}`);
     return { toolSelected: false };
   }
-  const params = extractParams(userMessage, selectedTool);
+  const params = await extractParams(userMessage, selectedTool, options);
   console.log(`[Tool Selector] Selected tool: ${selectedTool.name}`, params);
   return {
     toolSelected: true,
@@ -103,7 +130,10 @@ async function selectTool(userMessage, tools, options = {}) {
       return { toolSelected: false };
     }
     const generatedText = output?.[0]?.generated_text || output?.generated_text || "";
-    return parseToolSelectionResponse(generatedText, tools, userMessage);
+    return await parseToolSelectionResponse(generatedText, tools, userMessage, {
+      model,
+      device
+    });
   } catch (error) {
     console.error("[Tool Selector] Error:", error);
     return { toolSelected: false };

package/dist/react/index.cjs

@@ -47215,27 +47215,54 @@ Task: "${userMessage}"
 
 Best tool:`;
 }
-function extractParams(userMessage, tool) {
+function buildParamExtractionPrompt(userMessage, paramName, paramDescription) {
+  const desc = paramDescription ? ` (${paramDescription})` : "";
+  return `Extract the value for "${paramName}"${desc} from the user message. Reply with ONLY the extracted value, nothing else.
+
+User message: "${userMessage}"
+
+Value for ${paramName}:`;
+}
+async function extractParams(userMessage, tool, options) {
   const params = {};
-  if (!tool.parameters) return params;
-  for (const param of tool.parameters) {
-    if (param.name === "expression" || param.name === "query") {
-      params[param.name] = userMessage;
-    } else if (param.name === "location" || param.name === "city") {
-      const words = userMessage.split(/\s+/);
-      const capitalizedWords = words.filter(
-        (w) => w.length > 1 && w[0] === w[0].toUpperCase()
+  if (!tool.parameters || tool.parameters.length === 0) return params;
+  const { model, device } = options;
+  try {
+    const pipeline = await getTextGenerationPipeline({
+      model,
+      device,
+      dtype: "q4"
+    });
+    for (const param of tool.parameters) {
+      const prompt = buildParamExtractionPrompt(
+        userMessage,
+        param.name,
+        param.description
       );
-      params[param.name] = capitalizedWords.length > 0 ? capitalizedWords.join(" ") : userMessage;
-    } else if (param.name === "text" || param.name === "input") {
-      params[param.name] = userMessage;
-    } else {
+      const output = await pipeline(prompt, {
+        max_new_tokens: 32,
+        // Allow reasonable length for parameter values
+        temperature: 0,
+        do_sample: false,
+        return_full_text: false
+      });
+      const generatedText = output?.[0]?.generated_text || output?.generated_text || "";
+      const extractedValue = generatedText.trim().split("\n")[0].trim();
+      console.log(
+        `[Tool Selector] Extracted param "${param.name}":`,
+        extractedValue
+      );
+      params[param.name] = extractedValue || userMessage;
+    }
+  } catch (error) {
+    console.error("[Tool Selector] Error extracting params:", error);
+    for (const param of tool.parameters) {
       params[param.name] = userMessage;
     }
   }
   return params;
 }
-function parseToolSelectionResponse(response, tools, userMessage) {
+async function parseToolSelectionResponse(response, tools, userMessage, options) {
   console.log("[Tool Selector] Raw response:", response);
   const cleaned = response.toLowerCase().trim().split(/[\s\n,.]+/)[0].replace(/[^a-z0-9_-]/g, "");
   console.log("[Tool Selector] Parsed tool name:", cleaned);
@@ -47250,7 +47277,7 @@ function parseToolSelectionResponse(response, tools, userMessage) {
     );
     if (fuzzyTool) {
       console.log(`[Tool Selector] Fuzzy matched tool: ${fuzzyTool.name}`);
-      const params2 = extractParams(userMessage, fuzzyTool);
+      const params2 = await extractParams(userMessage, fuzzyTool, options);
       return {
         toolSelected: true,
         toolName: fuzzyTool.name,
@@ -47261,7 +47288,7 @@ function parseToolSelectionResponse(response, tools, userMessage) {
     console.warn(`[Tool Selector] Unknown tool: ${cleaned}`);
     return { toolSelected: false };
   }
-  const params = extractParams(userMessage, selectedTool);
+  const params = await extractParams(userMessage, selectedTool, options);
   console.log(`[Tool Selector] Selected tool: ${selectedTool.name}`, params);
   return {
     toolSelected: true,
@@ -47302,7 +47329,10 @@ async function selectTool(userMessage, tools, options = {}) {
       return { toolSelected: false };
     }
     const generatedText = output?.[0]?.generated_text || output?.generated_text || "";
-    return parseToolSelectionResponse(generatedText, tools, userMessage);
+    return await parseToolSelectionResponse(generatedText, tools, userMessage, {
+      model,
+      device
+    });
   } catch (error) {
     console.error("[Tool Selector] Error:", error);
     return { toolSelected: false };

package/dist/react/index.mjs

@@ -2,7 +2,7 @@ import {
   DEFAULT_TOOL_SELECTOR_MODEL,
   executeTool,
   selectTool
-} from "./chunk-2WZDKSTC.mjs";
+} from "./chunk-BG7SZT33.mjs";
 import "./chunk-Q6FVPTTV.mjs";
 import "./chunk-FBCDBTKJ.mjs";
 
@@ -862,7 +862,7 @@ var webFeatures = null;
 var webFeaturesPromise = isReactNative ? Promise.resolve(null) : Promise.all([
   import("./generation-NG4QVPCR.mjs"),
   import("./constants-WUGUGYE3.mjs"),
-  import("./selector-BWDBB4Y5.mjs")
+  import("./selector-XMR5KL3E.mjs")
 ]).then(([generation, constants, selector]) => {
   webFeatures = {
     generateLocalChatCompletion: generation.generateLocalChatCompletion,

package/dist/react/{selector-BWDBB4Y5.mjs → selector-XMR5KL3E.mjs}

@@ -3,7 +3,7 @@ import {
   executeTool,
   preloadToolSelectorModel,
   selectTool
-} from "./chunk-2WZDKSTC.mjs";
+} from "./chunk-BG7SZT33.mjs";
 import "./chunk-Q6FVPTTV.mjs";
 import "./chunk-FBCDBTKJ.mjs";
 export {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@reverbia/sdk",
-  "version": "1.0.0-next.20251208093930",
+  "version": "1.0.0-next.20251208094446",
   "description": "",
   "main": "./dist/index.cjs",
   "module": "./dist/index.mjs",