From 285afad993ce7ba66f87dd76ae59a24e9cc17422 Mon Sep 17 00:00:00 2001 From: ikechan8370 Date: Sat, 1 Jul 2023 21:42:12 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8Dwebsite=20tool?= =?UTF-8?q?=E7=9A=84bug=EF=BC=88=E5=95=8A=EF=BC=89?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- utils/tools/WebsiteTool.js | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/utils/tools/WebsiteTool.js b/utils/tools/WebsiteTool.js index b32e8b0..864288e 100644 --- a/utils/tools/WebsiteTool.js +++ b/utils/tools/WebsiteTool.js @@ -60,13 +60,14 @@ export class WebsiteTool extends AbstractTool { .replace('', '') // 去除声明 let maxModelTokens = getMaxModelTokens(Config.model) text = text.slice(0, Math.min(text.length, maxModelTokens - 1600)) + let completionParams = { + model: Config.model + } let api = new ChatGPTAPI({ apiBaseUrl: Config.openAiBaseUrl, apiKey: Config.apiKey, debug: false, - completionParams: { - model: Config.model - }, + completionParams, fetch: (url, options = {}) => { const defaultOptions = Config.proxy ? { @@ -81,7 +82,7 @@ export class WebsiteTool extends AbstractTool { }, maxModelTokens }) - const htmlContentSummaryRes = await api.sendMessage(`去除与主体内容无关的部分,从中整理出主体内容并转换成md格式,不需要主观描述性的语言与冗余的空白行。${text}`) + const htmlContentSummaryRes = await api.sendMessage(`去除与主体内容无关的部分,从中整理出主体内容并转换成md格式,不需要主观描述性的语言与冗余的空白行。${text}`, { completionParams }) let htmlContentSummary = htmlContentSummaryRes.text return `this is the main content of website:\n ${htmlContentSummary}` } catch (err) {