koishi-plugin-aka-ai-generator 0.2.2 → 0.2.3

package/lib/index.js

@@ -672,15 +672,23 @@ function apply(ctx, config) {
   }
   __name(buildModelMappingIndex, "buildModelMappingIndex");
   function parseStyleCommandModifiers(argv, imgParam) {
-    const argsList = [...argv.args || []].map((arg) => typeof arg === "string" ? arg.trim() : "").filter(Boolean);
-    const restStr = typeof argv.rest === "string" ? argv.rest.trim() : "";
-    if (restStr) {
-      const restParts = restStr.split(/\s+/).filter(Boolean);
-      argsList.push(...restParts);
-    }
-    if (imgParam && typeof imgParam === "string" && !imgParam.startsWith("http") && !imgParam.startsWith("data:")) {
-      const imgParts = imgParam.split(/\s+/).filter(Boolean);
-      argsList.push(...imgParts);
+    const session = argv.session;
+    let rawText = "";
+    if (session?.content) {
+      const elements = import_koishi.h.parse(session.content);
+      rawText = import_koishi.h.select(elements, "text").map((e) => e.attrs.content).join(" ");
+    }
+    const argsList = rawText ? rawText.split(/\s+/).filter(Boolean) : [...argv.args || []].map((arg) => typeof arg === "string" ? arg.trim() : "").filter(Boolean);
+    if (!rawText) {
+      const restStr = typeof argv.rest === "string" ? argv.rest.trim() : "";
+      if (restStr) {
+        const restParts = restStr.split(/\s+/).filter(Boolean);
+        argsList.push(...restParts);
+      }
+      if (imgParam && typeof imgParam === "string" && !imgParam.startsWith("http") && !imgParam.startsWith("data:")) {
+        const imgParts = imgParam.split(/\s+/).filter(Boolean);
+        argsList.push(...imgParts);
+      }
     }
     if (!argsList.length) return {};
     const modifiers = { customAdditions: [] };
@@ -702,8 +710,15 @@ function apply(ctx, config) {
       if (lower === "-add") {
         index++;
         const additionTokens = [];
-        while (index < argsList.length && !argsList[index].startsWith("-")) {
-          additionTokens.push(argsList[index]);
+        while (index < argsList.length) {
+          const nextToken = argsList[index];
+          if (nextToken.startsWith("-")) {
+            const key = normalizeSuffix(nextToken);
+            if (key && modelMappingIndex.has(key)) break;
+            if (nextToken.toLowerCase().startsWith("-prompt:")) break;
+            if (nextToken.toLowerCase() === "-add") break;
+          }
+          additionTokens.push(nextToken);
           index++;
         }
         if (additionTokens.length) {

package/package.json

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