@3-/aiapi 0.1.6 → 0.1.8

package/fmtJson.js

@@ -7,9 +7,14 @@ import seg from './seg.js';
 
 export default async(chat, txt) => {
   var gen, i, pli, result, split_li, sum, tmp, txt_li;
+  if (!txt) {
+    return [];
+  }
   txt_li = txt.split('\n');
   split_li = (await seg(chat, txt_li));
-  pli = partition(txt_li, split_li);
+  pli = partition(txt_li, split_li.map((i) => {
+    return [i.题, i.行];
+  }));
   result = [];
   sum = 0;
   tmp = [];
@@ -20,6 +25,7 @@ export default async(chat, txt) => {
     tmp.push(i);
     sum += i.length;
     if (sum > 3000) {
+      console.log(sum);
       gen();
       sum = 0;
       tmp = [];

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@3-/aiapi",
-  "version": "0.1.6",
+  "version": "0.1.8",
   "repository": {
     "type": "git",
     "url": "git+https://atomgit.com/i18n/lib.git"

package/partition.js

@@ -10,9 +10,9 @@ export default (lines, title_number) => {
 
   // 遍历排序后的行号
   for (const [t, row] of order) {
+    console.log(t)
     // 将行号转换为从0开始的数组索引
     const index = row - 1;
-    console.log(title)
     result.push(title + lines.slice(start, index).join('\n'));
     title = '# ' + t + '\n'
     start = index;

package/seg.js

@@ -4,32 +4,31 @@ import TYPE from './TYPE.js';
 export default async(chat, txt_li) => {
   var split_li, 提示词;
   提示词 = `下文第1列为行号,第2列为对话内容,以tab分隔。
-请划分章节,每章以提问开始,包含多组问答。
+请划分章节,每章以提问开始,包含多组问答,每章三千字左右。
 输出章节标题和每章首个问题的行号及原文,输出格式为JSON数组:\n` + txt_li.map((i, pos) => {
     return (pos + 1) + '\t' + i.trim();
   }).join('\n');
   return split_li = (await chat(提示词, {
     type: TYPE.ARRAY,
     description: '章节和行号的列表',
+    minItems: 1,
     items: {
-      type: TYPE.ARRAY,
-      items: false,
-      minItems: 3,
-      maxItems: 3,
-      prefixItems: [
-        {
-          description: '章节标题,1-2个词即可',
+      type: TYPE.OBJECT,
+      properties: {
+        题: {
+          description: '章节标题,要概括章节内容',
           type: TYPE.STRING
         },
-        {
+        行: {
           description: '该章首行行号',
           type: TYPE.INTEGER
         },
-        {
+        文: {
           description: '该章首行提问的原文',
           type: TYPE.STRING
         }
-      ]
+      },
+      required: ['题', '行', '文']
     }
   }, '你是专业资深的秘书'));
 };