@3-/aiapi 0.1.79 → 0.1.81

package/check/GEN.js

@@ -7,13 +7,13 @@ export default {
   items: {
     type: TYPE.OBJECT,
     properties: {
-      title: {
+      question: {
         type: TYPE.STRING,
-        description: '简短的标题，概述研究内容'
+        description: '从对话中提炼出的、具有明确事实属性，可通过网络搜索验证的关键行业洞察。必须是可以被证明或者证伪的命题。请避免无法验证的主观形容词'
       },
-      question: {
+      title: {
         type: TYPE.STRING,
-        description: '从对话中提炼出的、具有明确事实属性且可通过外部搜索验证的关键行业洞察，是可以被证明或者证伪的命题。请避免无法验证的主观形容词。'
+        description: '从最少字概述命题（必须是一个命题）作为标题'
       },
       zh: {
         type: TYPE.ARRAY,

package/check/SEARCH.js

@@ -9,14 +9,14 @@ export default {
       description: `请执行以下结构化的研究报告撰写任务：
 1. 章节拆分规划：首先，请根据研究主题的内在逻辑，将报告内容拆分为3-5个主要章节，每个章节需有明确的主题和研究范围。
 2. 研究过程汇报：其次，详细记录并汇报研究实施的完整过程，包括但不限于：研究方法选择、数据收集途径、分析工具使用、关键节点时间线等。
-3. 资料汇总整理：然后，在最后一章中，基于汇总的资料进行严谨的逻辑推理，分析数据间的关联性和规律，最终得出明确、有依据的研究结论。`,
+3. 资料汇总整理：然后，在最后一章中，基于汇总的资料进行严谨的逻辑推理，分析数据间的关联性和规律，最终得出明确、有依据的研究结论，如果无法得出答案，请写存疑。`,
       items: {
         type: TYPE.OBJECT,
         required: ['title', 'md'],
         properties: {
           title: {
             type: TYPE.STRING,
-            description: '章节标题'
+            description: '章节标题（只需要标题，不标注是第几章）'
           },
           md: {
             type: TYPE.STRING,
@@ -27,11 +27,15 @@ export default {
     },
     结论: {
       type: TYPE.STRING,
-      description: '一句话概述研究结论（支持还是不支持命题）'
+      description: '一句话概述研究结论（支持还是不支持命题，如果不确定，请写存疑）'
+    },
+    分: {
+      type: TYPE.INTEGER,
+      description: '1=支持，0=不确定，-1=不支持'
     },
     文献: {
       type: TYPE.ARRAY,
-      description: '参考资料',
+      description: '参考资料（按重要程度排序,重要的在前面）',
       minItems: 1,
       items: {
         type: TYPE.OBJECT,
@@ -43,7 +47,7 @@ export default {
           },
           brief: {
             type: TYPE.STRING,
-            description: '概述文章内容'
+            description: '抽取与命题有关的文章内容、数据'
           },
           url: {
             type: TYPE.STRING,
@@ -53,5 +57,5 @@ export default {
       }
     }
   },
-  required: ['结论', '研究', "文献"]
+  required: ['结论', '研究', "文献", "分"]
 };

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.81","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 [];
   }
@@ -20,12 +20,12 @@ export default async(chat, txt) => {
     return '# ' + title + '\n' + li.join('\n');
   }).join('\n');
   search = Promise.all(((await chat(gen_txt, GEN, "你是专业的风险投资人"))).map(async(i) => {
-    var body, en, question, reason, title, zh, 文献, 研究, 结论;
+    var body, en, question, reason, title, zh, 分, 文献, 研究, 结论;
     // console.log i
     ({title, question, zh, en, reason} = i);
     body = title + '\n研究命题: ' + question + '\n中文搜索词: ' + zh + '\n英文搜索词: ' + en;
     console.log('\n# ' + body);
-    ({结论, 研究, 文献} = (await chat('请基于搜索（包括权威媒体、财报、论文等），对以下命题深度研究:\n' + body, SEARCH, '利用搜索工具进行客观理性的研究分析，以数据为支撑来推演', {
+    ({结论, 研究, 文献, 分} = (await chat('请基于搜索（包括权威媒体、财报、论文等），对以下命题深度研究:\n' + body, SEARCH, '利用搜索工具进行客观理性的研究分析，以数据为支撑来推演', {
       tools: [
         {
           google_search: {}
@@ -35,7 +35,15 @@ export default async(chat, txt) => {
         }
       ]
     })));
-    return '## ' + title.trim() + '\n\n**<u>问题:</u>** ' + question.trim() + '\n\n**<u>结论</u>:** ' + 结论.trim() + '\n\n' + 研究.map(({title, md}) => {
+    title = title.trim();
+    if (分 > 0) {
+      title = '✅ ' + title;
+    } else if (分 < 0) {
+      title = '❌ ' + title;
+    } else {
+      title = '❓ ' + title;
+    }
+    return '## ' + title + '\n\n**<u>命题:</u>** ' + question.trim() + '\n\n**<u>结论</u>:** ' + 结论.trim() + '\n\n' + 研究.map(({title, md}) => {
       return '### ' + title + '\n\n' + md;
     }).join('\n') + '\n\n### 参考资料\n\n' + 文献.map(({title, brief, url}, pos) => {
       return (pos + 1) + '. [' + title + '](' + url + ') : ' + brief;
@@ -45,5 +53,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];
 };