rc-plugin/utils/llm-util.js
zhiyu1998 65fb73405b feat: 添加对R文档的查询支持,使用方式为#R文档 你的问题
- 在query.js中添加了对R文档查询的处理函数intelligentDoc
- 引入了新的常量和函数以支持R文档的查询和内容获取
- 移除了不再使用的PearAPI相关代码
- 在utils/llm-util.js中添加了新的函数llmRead和deepSeekChat以支持文档内容获取和对话生成
- 在constants/query.js中添加了新的常量以支持R文档查询
2024-09-10 22:21:44 +08:00

39 lines
994 B
JavaScript

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;
}