@polka-codes/core 0.8.5 → 0.8.7

package/dist/_tsup-dts-rollup.d.ts

@@ -780,6 +780,7 @@ declare const _default: {
             readonly description: "One or more follow-up questions you need answered before you can continue.";
             readonly required: true;
             readonly allowMultiple: true;
+            readonly usageValue: "questions here";
             readonly children: [{
                 readonly name: "prompt";
                 readonly description: "The text of the question.";
@@ -841,6 +842,7 @@ declare const _default: {
         readonly description: "One or more follow-up questions you need answered before you can continue.";
         readonly required: true;
         readonly allowMultiple: true;
+        readonly usageValue: "questions here";
         readonly children: [{
             readonly name: "prompt";
             readonly description: "The text of the question.";
@@ -2586,6 +2588,7 @@ export declare const toolInfo: {
         readonly description: "One or more follow-up questions you need answered before you can continue.";
         readonly required: true;
         readonly allowMultiple: true;
+        readonly usageValue: "questions here";
         readonly children: [{
             readonly name: "prompt";
             readonly description: "The text of the question.";

package/dist/index.js

@@ -1010,6 +1010,14 @@ var getStringArray = (args, name, defaultValue) => {
   if (ret === "") {
     return [];
   }
+  if (Array.isArray(ret)) {
+    for (const item of ret) {
+      if (typeof item !== "string") {
+        throw new Error(`Invalid argument type: ${name} ${item}`);
+      }
+    }
+    return ret;
+  }
   if (typeof ret !== "string") {
     throw new Error(`Invalid argument type: ${name} ${ret}`);
   }
@@ -1079,6 +1087,7 @@ var toolInfo = {
       description: "One or more follow-up questions you need answered before you can continue.",
       required: true,
       allowMultiple: true,
+      usageValue: "questions here",
       children: [
         {
           name: "prompt",
@@ -2302,16 +2311,46 @@ var getAvailableTools = ({
 };
 
 // src/Agent/parseAssistantMessage.ts
-function parseNestedParameters(content, parameterPrefix) {
+function parseNestedParameters(content, parameterPrefix, childrenParams) {
   const result = {};
   const nestedParamRegex = new RegExp(`<${parameterPrefix}([^>]+)>([\\s\\S]*?)<\\/${parameterPrefix}\\1>`, "gs");
+  const arrayParams = /* @__PURE__ */ new Set();
+  if (childrenParams) {
+    for (const param of childrenParams) {
+      if (param.allowMultiple) {
+        arrayParams.add(param.name);
+      }
+    }
+  }
   while (true) {
     const match = nestedParamRegex.exec(content);
     if (match === null) break;
     const paramName = match[1];
     const paramContent = match[2].trim();
-    if (paramContent.includes(`<${parameterPrefix}`)) {
-      result[paramName] = parseNestedParameters(paramContent, parameterPrefix);
+    const childParam = childrenParams?.find((p) => p.name === paramName);
+    if (paramContent.includes(`<${parameterPrefix}`) && childParam?.children) {
+      const nestedResult = parseNestedParameters(paramContent, parameterPrefix, childParam.children);
+      if (arrayParams.has(paramName)) {
+        if (!result[paramName]) {
+          result[paramName] = [];
+        }
+        if (Array.isArray(result[paramName])) {
+          result[paramName].push(nestedResult);
+        } else {
+          result[paramName] = [nestedResult];
+        }
+      } else {
+        result[paramName] = nestedResult;
+      }
+    } else if (arrayParams.has(paramName)) {
+      if (!result[paramName]) {
+        result[paramName] = [];
+      }
+      if (Array.isArray(result[paramName])) {
+        result[paramName].push(paramContent);
+      } else {
+        result[paramName] = [paramContent];
+      }
     } else {
       result[paramName] = paramContent;
     }
@@ -2353,15 +2392,29 @@ function parseAssistantMessage(assistantMessage, tools, toolNamePrefix) {
           paramMatches.push(paramMatch[1].trim());
         }
         if (paramMatches.length > 0) {
-          if (paramMatches.length === 1) {
+          if (paramMatches.length === 1 && !param.allowMultiple) {
             const paramContent = paramMatches[0];
-            if (paramContent.includes(`<${parameterPrefix}`)) {
-              params[param.name] = parseNestedParameters(paramContent, parameterPrefix);
+            if (paramContent.includes(`<${parameterPrefix}`) && param.children) {
+              params[param.name] = parseNestedParameters(paramContent, parameterPrefix, param.children);
             } else {
               params[param.name] = paramContent;
             }
           } else {
-            params[param.name] = paramMatches;
+            if (param.allowMultiple) {
+              params[param.name] = paramMatches.map((paramContent) => {
+                if (paramContent.includes(`<${parameterPrefix}`) && param.children) {
+                  return parseNestedParameters(paramContent, parameterPrefix, param.children);
+                }
+                return paramContent;
+              });
+            } else {
+              const paramContent = paramMatches[0];
+              if (paramContent.includes(`<${parameterPrefix}`) && param.children) {
+                params[param.name] = parseNestedParameters(paramContent, parameterPrefix, param.children);
+              } else {
+                params[param.name] = paramContent;
+              }
+            }
           }
         }
       }
@@ -2412,11 +2465,16 @@ var toolInfoPrompt = (tool, toolNamePrefix, parameterPrefix) => `
 Description: ${tool.description}
 
 Parameters:
-${tool.parameters.map((param) => `- ${parameterPrefix}${param.name}: (${param.required ? "required" : "optional"}) ${param.description}`).join("\n")}
+${tool.parameters.map(
+  (param) => `- ${parameterPrefix}${param.name}: (${param.required ? "required" : "optional"})${param.allowMultiple ? " (multiple allowed)" : ""} ${param.description}`
+).join("\n")}
 
 Usage:
 <${toolNamePrefix}${tool.name}>
-${tool.parameters.map((param) => `<${parameterPrefix}${param.name}>${param.usageValue}</${parameterPrefix}${param.name}>`).join("\n")}
+${tool.parameters.map(
+  (param) => param.allowMultiple ? `<${parameterPrefix}${param.name}>${param.usageValue}</${parameterPrefix}${param.name}>
+<${parameterPrefix}${param.name}>${param.usageValue}</${parameterPrefix}${param.name}>` : `<${parameterPrefix}${param.name}>${param.usageValue}</${parameterPrefix}${param.name}>`
+).join("\n")}
 </${toolNamePrefix}${tool.name}>`;
 var toolInfoExamplesPrompt = (tool, example, toolNamePrefix, parameterPrefix) => `
 ## Example: ${example.description}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@polka-codes/core",
-  "version": "0.8.5",
+  "version": "0.8.7",
   "license": "AGPL-3.0",
   "author": "github@polka.codes",
   "type": "module",