koishi-plugin-aka-ai-generator 0.2.0 → 0.2.1

package/lib/index.js

@@ -672,18 +672,40 @@ function apply(ctx, config) {
   }
   __name(buildModelMappingIndex, "buildModelMappingIndex");
   function parseStyleCommandModifiers(argv) {
-    const allArgs = [...argv.args || []].map((arg) => typeof arg === "string" ? arg.trim() : "").filter(Boolean);
-    if (!allArgs.length && !argv.rest) return {};
-    const modifiers = {};
-    let remainingArgs = [...allArgs];
-    const promptIndex = remainingArgs.findIndex((arg) => arg.toLowerCase().startsWith("-prompt:"));
-    if (promptIndex >= 0) {
-      const [promptToken, ...afterPrompt] = remainingArgs.slice(promptIndex);
-      const promptHead = promptToken.substring(promptToken.indexOf(":") + 1);
-      modifiers.customPromptSuffix = [promptHead, ...afterPrompt].join(" ").trim();
-      remainingArgs = remainingArgs.slice(0, promptIndex);
+    const rawArgs = [...argv.args || []].map((arg) => typeof arg === "string" ? arg.trim() : "").filter(Boolean);
+    if (!rawArgs.length && !argv.rest) return {};
+    const modifiers = { customAdditions: [] };
+    const flagCandidates = [];
+    let index = 0;
+    while (index < rawArgs.length) {
+      const token = rawArgs[index];
+      if (!token) {
+        index++;
+        continue;
+      }
+      const lower = token.toLowerCase();
+      if (lower.startsWith("-prompt:")) {
+        const promptHead = token.substring(token.indexOf(":") + 1);
+        const restTokens = rawArgs.slice(index + 1);
+        modifiers.customPromptSuffix = [promptHead, ...restTokens].join(" ").trim();
+        break;
+      }
+      if (lower === "-add") {
+        index++;
+        const additionTokens = [];
+        while (index < rawArgs.length && !rawArgs[index].startsWith("-")) {
+          additionTokens.push(rawArgs[index]);
+          index++;
+        }
+        if (additionTokens.length) {
+          modifiers.customAdditions.push(additionTokens.join(" "));
+        }
+        continue;
+      }
+      flagCandidates.push(token);
+      index++;
     }
-    for (const arg of remainingArgs) {
+    for (const arg of flagCandidates) {
       if (!arg.startsWith("-")) continue;
       const key = normalizeSuffix(arg);
       if (!key) continue;
@@ -1087,6 +1109,9 @@ function apply(ctx, config) {
           if (modifiers.modelMapping?.promptSuffix) {
             promptSegments.push(modifiers.modelMapping.promptSuffix);
           }
+          if (modifiers.customAdditions?.length) {
+            promptSegments.push(...modifiers.customAdditions);
+          }
           if (modifiers.customPromptSuffix) {
             promptSegments.push(modifiers.customPromptSuffix);
           }

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-aka-ai-generator",
   "description": "自用AI生成插件(GPTGod & Yunwu)",
-  "version": "0.2.0",
+  "version": "0.2.1",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [