@polka-codes/cli-shared 0.8.6 → 0.8.7

package/dist/index.js

@@ -36939,6 +36939,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",
@@ -38153,17 +38154,47 @@ var getAvailableTools = ({
 };
 
 // ../core/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 = 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((p2) => p2.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;
     }
@@ -38207,15 +38238,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;
+              }
+            }
           }
         }
       }
@@ -38268,12 +38313,13 @@ var toolInfoPrompt = (tool, toolNamePrefix, parameterPrefix) => `
 Description: ${tool.description}
 
 Parameters:
-${tool.parameters.map((param) => `- ${parameterPrefix}${param.name}: (${param.required ? "required" : "optional"}) ${param.description}`).join(`
+${tool.parameters.map((param) => `- ${parameterPrefix}${param.name}: (${param.required ? "required" : "optional"})${param.allowMultiple ? " (multiple allowed)" : ""} ${param.description}`).join(`
 `)}
 
 Usage:
 <${toolNamePrefix}${tool.name}>
-${tool.parameters.map((param) => `<${parameterPrefix}${param.name}>${param.usageValue}</${parameterPrefix}${param.name}>`).join(`
+${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(`
 `)}
 </${toolNamePrefix}${tool.name}>`;
 var toolInfoExamplesPrompt = (tool, example, toolNamePrefix, parameterPrefix) => `

package/package.json

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