通过扩展maxkb的知识库管理和ai对话能力,实现一键生成分析报告及word文档;
本次优化点在于:
- 实现用户输入多个问题,一键生成分析报告,并提供word文件将内容下载下来
- word中支持markdown格式,能下载各种图片、报表等复杂格式
- 能够支持动态的提示词,即每个问题都可以自带提示词,来微调大模型回复;
先看下操作界面:
如上图所示,可以输入多个问题,一键生成分析报告;
将大模型分析的结果变成word文档,提供给用户下载
下面来说下实现方案:
第一步:修改maxkb的前端代码,增加一个页面和菜单; 由于maxkb是docker部署,所以第一步是进入docker文件目录中;可以通过以下方法进去:
- docker ps #查看maxkb的containerId
- docker exec -it dbb76df3d062 /bin/bash #dbb76df3d062是maxkb的containerId
- 上述命令进入了/opt/maxkb/app 目录,这里有所有maxkb的源码信息,我们把ui目录拷贝出来
- 在ui的src目录更新代码,更新完成后,执行 npm build (注意:node版本用最新版)
- 将生成的dist 目录,上传到 docker的 /opt/maxkb/app/ui 里面,替换原来的dist
- 执行 rm -rf /opt/maxkb/app/data/cache/default_file_cache 删除缓存
- 重启maxkb或者容器,重启 maxkb 命令为 mkctl restart ; 重启容器可以用 docker restart dbb76df3d062 #dbb76df3d062是maxkb的containerId
第二步:实现提示词动态输入,这个要改下后端代码:
- 在chat_views.py 中增加个参数支持传入 self_tip
- 在chat_api.py 中增加参数 self_tip
- 在 chat_message_serializers.py 中增加动态修改逻辑
这样改动后,maxkb可以允许前端传入提示词,并指导大模型回答了
第三步:将回答内容变成word并支持markdown格式的样式输出;
这一步我扩展了个api接口,用了php的代码来做的(当然也可以用其他语言),具体实现方法如下:
// 引入 autoload.php 文件,用于自动加载 PHPOffice 类
require_once 'vendor/autoload.php';
use \PhpOffice\PhpWord\IOFactory;
// 模拟数据
$data = $ary;
// 创建Word文档
$phpWord = new PhpOffice\PhpWord\PhpWord();
$section = $phpWord->addSection();
$html_o = '
<div style="text-align:center;width:100%;font-size:32px;font-weight:bold">XXX矿区研究报告</div>
';
// 将HTML添加到Word文档
\PhpOffice\PhpWord\Shared\Html::addHtml($section, $html_o);
$section = $phpWord->addSection();
// 将HTML添加到Word文档
\PhpOffice\PhpWord\Shared\Html::addHtml($section, $html);
$j = 0;
$html = '<html><header>
<style>img {width:100%;}</style>
</header>
<body>';
foreach ($data as $k => $v) {
$j = $j + 1;
// Markdown内容
$markdown = <<<MD
$v
MD;
// 解析Markdown为HTML
$html .= '<span style="font-size:16px;font-weight:bold">' . $j . '、' . $k . '</span><br/>';
// 解析Markdown为HTML
$parsedown = new Parsedown();
$html .= $parsedown->text($markdown);
$html .= '<br/><br/>';
}
$html.= '</body></html>';
$html = str_replace('<img src="/api/image/','<img width="600" src="'.$host.'/api/image/', $html);
$html = str_replace('<img src="https://api/image/','<img width="600" src="'.$host.'/api/image/', $html);
// 将HTML添加到Word文档
\PhpOffice\PhpWord\Shared\Html::addHtml($section, $html);
// 保存生成的新文档,文件名为 output.docx
$path = 'uploads/report_' . time() . '.docx';
// 保存Word文档
$xmlWriter = IOFactory::createWriter($phpWord, 'Word2007');
$xmlWriter->save($path);
以上就完成了整个文档的输出; 如果您有类似需求,通过ai自动生成文档或者分析报告,可以跟我联系;