@3-/aiapi 0.1.79 → 0.1.80

package/check/SEARCH.js

@@ -16,7 +16,7 @@ export default {
         properties: {
           title: {
             type: TYPE.STRING,
-            description: '章节标题'
+            description: '章节标题（只需要标题，不标注是第几章）'
           },
           md: {
             type: TYPE.STRING,

package/fmt.js

@@ -4,8 +4,8 @@ import fmtJson from './fmtJson.js';
 import partition from './partition.js';
 
 export default async(chat, txt) => {
-  var pli;
-  pli = (await partition(chat, txt));
+  var pli, 题;
+  [题, pli] = (await partition(chat, txt));
   if (!pli.length) {
     return '';
   }

package/package.json

@@ -1 +1 @@
-{"name":"@3-/aiapi","version":"0.1.79","repository":{"type":"git","url":"git+https://atomgit.com/i18n/lib.git"},"homepage":"https://atomgit.com/i18n/lib/tree/dev/aiapi","author":"i18n.site@gmail.com","license":"MulanPSL-2.0","exports":{".":"./lib.js","./*":"./*"},"files":["./*"],"devDependencies":{"@3-/read":"^0.1.4"},"scripts":{},"type":"module","dependencies":{"@3-/retry":"^0.0.2","@3-/rm_cn_space":"^0.1.1","@3-/sleep":"^0.0.4","@3-/txt_li":"^0.1.5","@3-/utf8":"^0.0.4","lodash-es":"^4.17.21"}}
+{"name":"@3-/aiapi","version":"0.1.80","repository":{"type":"git","url":"git+https://atomgit.com/i18n/lib.git"},"homepage":"https://atomgit.com/i18n/lib/tree/dev/aiapi","author":"i18n.site@gmail.com","license":"MulanPSL-2.0","exports":{".":"./lib.js","./*":"./*"},"files":["./*"],"devDependencies":{"@3-/read":"^0.1.4"},"scripts":{},"type":"module","dependencies":{"@3-/retry":"^0.0.2","@3-/rm_cn_space":"^0.1.1","@3-/sleep":"^0.0.4","@3-/txt_li":"^0.1.5","@3-/utf8":"^0.0.4","lodash-es":"^4.17.21"}}

package/partition.js

@@ -1,4 +1,4 @@
-import seg from "./seg.js";
+import { default as seg, segWithTitle } from "./seg.js";
 import txtLi from "@3-/txt_li";
 import rmCnSpace from "@3-/rm_cn_space";
 
@@ -43,8 +43,9 @@ const partition = async (li, title_number, remain_seg) => {
 
 export default async (chat, txt) => {
   if (!txt) {
-    return [];
+    return ["", []];
   }
-  const li = txtLi(txt).map((i) => rmCnSpace(i));
-  return partition(li, await seg(chat, li), repartition(chat, li.length));
+  const li = txtLi(txt).map((i) => rmCnSpace(i)),
+    [title, seg_li] = await segWithTitle(chat, li);
+  return [title, await partition(li, seg_li, repartition(chat, li.length))];
 };

package/researchFmt.js

@@ -10,8 +10,8 @@ import SEARCH from './check/SEARCH.js';
 import refmt from './refmt.js';
 
 export default async(chat, txt) => {
-  var fmt, gen_txt, pli, search;
-  pli = (await partition(chat, txt));
+  var fmt, gen_txt, pli, search, 文章标题;
+  [文章标题, pli] = (await partition(chat, txt));
   if (!pli.length) {
     return [];
   }
@@ -45,5 +45,5 @@ export default async(chat, txt) => {
   if (search.length) {
     fmt += '# 机器投研\n\n' + search.map(refmt).join('\n');
   }
-  return fmt;
+  return [文章标题, fmt];
 };

package/seg.js

@@ -1,34 +1,64 @@
 #!/usr/bin/env -S node --trace-uncaught --expose-gc --unhandled-rejections=strict --experimental-wasm-modules
+var OPT, OPT_WITH_TITLE, gen;
+
 import TYPE from './TYPE.js';
 
-export default async(chat, txt_li) => {
-  var split_li, 提示词;
+OPT = {
+  type: TYPE.OBJECT,
+  required: ['li'],
+  properties: {
+    li: {
+      type: TYPE.ARRAY,
+      description: '拆分文章为多个章节，每个章节包含一系列问答对',
+      minItems: 1,
+      items: {
+        type: TYPE.OBJECT,
+        properties: {
+          题: {
+            description: '章节标题(只写标题，不写序号、注释)',
+            type: TYPE.STRING
+          },
+          行: {
+            description: '该章最后一句的行号',
+            type: TYPE.INTEGER
+          }
+        },
+        required: ['题', '行']
+      }
+    }
+  }
+};
+
+OPT_WITH_TITLE = structuredClone(OPT);
+
+OPT_WITH_TITLE.required.push('题');
+
+OPT_WITH_TITLE.properties.题 = {
+  type: TYPE.STRING,
+  description: '文章总标题（要简短）'
+};
+
+gen = async(opt, chat, txt_li) => {
+  var r, 提示词;
   提示词 = `下文第1列为行号,第2列为对话内容,以tab分隔。
 请划分章节,每章不要太长也不要太短,千字左右一章。
 输出章节标题、每章最后一句号的行号。
 输出格式为json数组:\n` + txt_li.map((i, pos) => {
     return (pos + 1) + '\t' + i.trim();
   }).join('\n');
-  split_li = (await chat(提示词, {
-    type: TYPE.ARRAY,
-    description: '拆分文章为多个章节，每个章节包含一系列问答对',
-    minItems: 1,
-    items: {
-      type: TYPE.OBJECT,
-      properties: {
-        题: {
-          description: '章节标题(只写标题，不写注释)',
-          type: TYPE.STRING
-        },
-        行: {
-          description: '该章最后一句的行号',
-          type: TYPE.INTEGER
-        }
-      },
-      required: ['题', '行']
-    }
-  }, '你是专业资深的杂志编辑'));
-  return split_li.map((i) => {
+  r = (await chat(提示词, opt, '你是专业资深的杂志编辑'));
+  r.li = r.li.map((i) => {
     return [i.题, i.行];
   });
+  return r;
+};
+
+export default async(chat, txt_li) => {
+  return ((await gen(OPT, chat, txt_li))).li;
+};
+
+export const segWithTitle = async(chat, txt_li) => {
+  var r;
+  r = (await gen(OPT_WITH_TITLE, chat, txt_li));
+  return [r.题, r.li];
 };