mirror of
https://github.com/Jerryplusy/rc-plugin.git
synced 2025-10-14 08:09:19 +00:00
✨ feat: 添加对R文档的查询支持，使用方式为#R文档 你的问题
- 在query.js中添加了对R文档查询的处理函数intelligentDoc - 引入了新的常量和函数以支持R文档的查询和内容获取 - 移除了不再使用的PearAPI相关代码 - 在utils/llm-util.js中添加了新的函数llmRead和deepSeekChat以支持文档内容获取和对话生成 - 在constants/query.js中添加了新的常量以支持R文档查询
This commit is contained in:
zhiyu1998
parent
b57c9212bf
commit
65fb73405b
@ -2,12 +2,25 @@ import axios from "axios";
|
||||
import _ from "lodash";
|
||||
import fetch from "node-fetch";
|
||||
// 常量
|
||||
import { CAT_LIMIT, COMMON_USER_AGENT } from "../constants/constant.js";
|
||||
import { LINUX_AI_PROMPT, LINUX_QUERY, REDIS_YUNZAI_LINUX } from "../constants/query.js";
|
||||
import { CAT_LIMIT, COMMON_USER_AGENT, SUMMARY_PROMPT } from "../constants/constant.js";
|
||||
import {
|
||||
LINUX_AI_PROMPT,
|
||||
LINUX_QUERY,
|
||||
RDOC_AI_PROMPT,
|
||||
RDOC_LINK,
|
||||
REDIS_YUNZAI_LINUX,
|
||||
REDIS_YUNZAI_RDOC
|
||||
} from "../constants/query.js";
|
||||
// 配置文件
|
||||
import config from "../model/config.js";
|
||||
import { deepSeekChat, llmRead } from "../utils/llm-util.js";
|
||||
import { OpenaiBuilder } from "../utils/openai-builder.js";
|
||||
import { redisExistAndGetKey, redisExistAndInsertObject, redisExistAndUpdateObject } from "../utils/redis-util.js";
|
||||
import {
|
||||
redisExistAndGetKey,
|
||||
redisExistAndInsertObject,
|
||||
redisExistAndUpdateObject,
|
||||
redisSetKey
|
||||
} from "../utils/redis-util.js";
|
||||
import { textArrayToMakeForward } from "../utils/yunzai-util.js";
|
||||
|
||||
export class query extends plugin {
|
||||
@ -46,6 +59,10 @@ export class query extends plugin {
|
||||
{
|
||||
reg: "^#(linux|Linux)(.*)",
|
||||
fnc: "linuxQuery"
|
||||
},
|
||||
{
|
||||
reg: "^#R文档(.*)",
|
||||
fnc: "intelligentDoc",
|
||||
}
|
||||
],
|
||||
});
|
||||
@ -338,6 +355,42 @@ export class query extends plugin {
|
||||
return parsedData;
|
||||
}
|
||||
|
||||
async intelligentDoc(e) {
|
||||
const question = e.msg.replace("#R文档", "").trim();
|
||||
const rPluginDocument = await redisExistAndGetKey(REDIS_YUNZAI_RDOC);
|
||||
if (question === "") {
|
||||
e.reply("请输入要查询的文档内容！\n例如：#R文档 如何玩转BBDown");
|
||||
return;
|
||||
} else if (question === "更新" || rPluginDocument?.content === undefined) {
|
||||
e.reply("更新文档中...");
|
||||
const content = await llmRead(RDOC_LINK);
|
||||
logger.info(content);
|
||||
await redisSetKey(REDIS_YUNZAI_RDOC, {
|
||||
content
|
||||
})
|
||||
e.reply("文档更新完成！");
|
||||
}
|
||||
let kimiAns, model = "DeepSeek";
|
||||
if (this.aiBaseURL && this.aiApiKey) {
|
||||
const builder = await new OpenaiBuilder()
|
||||
.setBaseURL(this.aiBaseURL)
|
||||
.setApiKey(this.aiApiKey)
|
||||
.setModel(this.aiModel)
|
||||
.setPrompt(rPluginDocument)
|
||||
.build();
|
||||
const kimiResp = await builder.kimi(RDOC_AI_PROMPT.replace("{}", question));
|
||||
kimiAns = kimiResp.ans;
|
||||
model = kimiResp.model;
|
||||
} else {
|
||||
logger.info(RDOC_AI_PROMPT.replace("{}", question));
|
||||
kimiAns = await deepSeekChat(RDOC_AI_PROMPT.replace("{}", question), rPluginDocument.content);
|
||||
logger.info(kimiAns)
|
||||
}
|
||||
const Msg = await Bot.makeForwardMsg(textArrayToMakeForward(e, [`「R插件 x ${ model }」联合为您总结内容：`, kimiAns]));
|
||||
await e.reply(Msg);
|
||||
return;
|
||||
}
|
||||
|
||||
// 删除标签
|
||||
removeTag(title) {
|
||||
const titleRex = /<[^>]+>/g;
|
||||
|
||||
@ -42,8 +42,6 @@ import {
|
||||
NETEASE_API_CN,
|
||||
NETEASE_SONG_DOWNLOAD,
|
||||
NETEASE_TEMP_API,
|
||||
PearAPI_CRAWLER,
|
||||
PearAPI_DEEPSEEK,
|
||||
QISHUI_MUSIC_TEMP_API,
|
||||
QQ_MUSIC_TEMP_API,
|
||||
TWITTER_TWEET_INFO,
|
||||
@ -95,6 +93,7 @@ import {
|
||||
import GeneralLinkAdapter from "../utils/general-link-adapter.js";
|
||||
import { LagrangeAdapter } from "../utils/lagrange-adapter.js";
|
||||
import { contentEstimator } from "../utils/link-share-summary-util.js";
|
||||
import { deepSeekChat, llmRead } from "../utils/llm-util.js";
|
||||
import { getDS } from "../utils/mihoyo.js";
|
||||
import { OpenaiBuilder } from "../utils/openai-builder.js";
|
||||
import { redisExistKey, redisGetKey, redisSetKey } from "../utils/redis-util.js";
|
||||
@ -1729,28 +1728,10 @@ export class tools extends plugin {
|
||||
* @returns {Promise<void>}
|
||||
*/
|
||||
async tempSummary(name, summaryLink, e) {
|
||||
const llmCrawler = await fetch(PearAPI_CRAWLER.replace("{}", summaryLink));
|
||||
const content = await (await llmCrawler.json())?.data;
|
||||
const content = await llmRead(summaryLink);
|
||||
const titleMatch = content.match(/Title:\s*(.*?)\n/)?.[1];
|
||||
e.reply(`${ this.identifyPrefix } 识别：${ name } - ${ titleMatch }，正在为您总结，请稍等...`, true);
|
||||
const deepseekFreeSummary = await fetch(PearAPI_DEEPSEEK, {
|
||||
method: "POST",
|
||||
headers: {
|
||||
"Content-Type": "application/json",
|
||||
},
|
||||
body: JSON.stringify({
|
||||
"messages": [
|
||||
{
|
||||
"role": "system",
|
||||
"content": SUMMARY_PROMPT
|
||||
},
|
||||
{
|
||||
"role": "user",
|
||||
"content": content,
|
||||
}]
|
||||
}),
|
||||
});
|
||||
const summary = (await deepseekFreeSummary.json())?.message;
|
||||
const summary = await deepSeekChat(content, SUMMARY_PROMPT);
|
||||
const Msg = await Bot.makeForwardMsg(textArrayToMakeForward(e, [`「R插件 x DeepSeek」联合为您总结内容：`, summary]));
|
||||
await e.reply(Msg);
|
||||
}
|
||||
|
||||
@ -4,6 +4,8 @@
|
||||
*/
|
||||
export const LINUX_QUERY = "https://api.pearktrue.cn/api/linux/?keyword={}"
|
||||
|
||||
export const RDOC_LINK = "https://gitee.com/kyrzy0416/rconsole-plugin/raw/docs/posts/QA%E5%AE%98%E6%96%B9%E8%A7%A3%E7%AD%94.md";
|
||||
|
||||
export const LINUX_AI_PROMPT = "- Role: Linux命令专家\n" +
|
||||
"- Background: 用户需要从特定的Linux命令网站获取常见命令的用法，希望得到简洁明了的回复。\n" +
|
||||
"- Profile: 你是一位对Linux命令有深入理解的专家，能够快速从网站中提取关键信息，并以简洁的方式呈现给用户。\n" +
|
||||
@ -22,8 +24,22 @@ export const LINUX_AI_PROMPT = "- Role: Linux命令专家\n" +
|
||||
" - 例子3: 命令 'cp' — 复制文件或目录。\n" +
|
||||
"- Initialization: 在第一次对话中，请直接输出以下：您好！我是Linux命令专家，我将为您提供简洁明了的Linux命令用法。";
|
||||
|
||||
export const RDOC_AI_PROMPT = "我有一些有关R插件的问题需要询问你，问题是：“{}”，回答的时候尽量大部分内容都来自文档，比如：询问你关于“Apple Music 和 Spotify 使用说明”，你只需回答：“" +
|
||||
"AM解析和Spotify解析需要使用两个依赖freyr、atomicparsley，现在只以Debian系统为例：\n" +
|
||||
"npm install -g freyr\n" +
|
||||
"# 或者你有yarn的话可以使用\n" +
|
||||
"yarn global add freyr\n" +
|
||||
"# 接着安装它的依赖\n" +
|
||||
"apt-get install atomicparsley”";
|
||||
|
||||
/**
|
||||
* Linux命令缓存
|
||||
* @type {string}
|
||||
*/
|
||||
export const REDIS_YUNZAI_LINUX = "Yz:rconsole:query:linux";
|
||||
|
||||
/**
|
||||
* 文档文档需要的数据
|
||||
* @type {string}
|
||||
*/
|
||||
export const REDIS_YUNZAI_RDOC = "Yz:rconsole:query:rdoc";
|
||||
|
||||
38
utils/llm-util.js
Normal file
38
utils/llm-util.js
Normal file
@ -0,0 +1,38 @@
|
||||
import { PearAPI_CRAWLER, PearAPI_DEEPSEEK } from "../constants/tools.js";
|
||||
|
||||
/**
|
||||
* LLM 爬虫
|
||||
* @param summaryLink
|
||||
* @returns {Promise<string>}
|
||||
*/
|
||||
export async function llmRead(summaryLink) {
|
||||
const llmCrawler = await fetch(PearAPI_CRAWLER.replace("{}", summaryLink));
|
||||
return (await llmCrawler.json())?.data;
|
||||
}
|
||||
|
||||
/**
|
||||
* DeepSeek对话
|
||||
* @param content
|
||||
* @param prompt
|
||||
* @returns {Promise<string>}
|
||||
*/
|
||||
export async function deepSeekChat(content, prompt) {
|
||||
const deepseekFreeSummary = await fetch(PearAPI_DEEPSEEK, {
|
||||
method: "POST",
|
||||
headers: {
|
||||
"Content-Type": "application/json",
|
||||
},
|
||||
body: JSON.stringify({
|
||||
"messages": [
|
||||
{
|
||||
"role": "system",
|
||||
"content": prompt
|
||||
},
|
||||
{
|
||||
"role": "user",
|
||||
"content": content,
|
||||
}]
|
||||
}),
|
||||
});
|
||||
return (await deepseekFreeSummary.json())?.message;
|
||||
}
|
||||
Loading…
x
Reference in New Issue
Block a user