From c985e5b08fd0f92a376678220884a944d949d65e Mon Sep 17 00:00:00 2001 From: roitium Date: Mon, 22 Jul 2024 12:13:02 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E5=8E=BB=E9=99=A4=E6=8E=89ai=E8=BF=94?= =?UTF-8?q?=E5=9B=9E=E5=86=85=E5=AE=B9=E4=B8=AD=E7=9A=84=E4=BB=A3=E7=A0=81?= =?UTF-8?q?=E5=9D=97=E6=A0=B7=E5=BC=8F?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/services/bot/memory/long-term.ts | 4 ++++ src/services/bot/memory/short-term.ts | 4 ++++ 2 files changed, 8 insertions(+) diff --git a/src/services/bot/memory/long-term.ts b/src/services/bot/memory/long-term.ts index fe62553..751ece8 100644 --- a/src/services/bot/memory/long-term.ts +++ b/src/services/bot/memory/long-term.ts @@ -67,6 +67,10 @@ export class LongTermMemoryAgent { shortTermMemory: lastOf(newMemories)!.text, }), }); + // 如果返回内容是个markdown代码块,就让他变回普通json + res?.content?.trim(); + if (res?.content?.startsWith("```json")) {res.content = res?.content?.replace("```json", "");} + if (res?.content?.endsWith("```")) {res.content = res?.content?.replace("```", "");} return jsonDecode(res?.content)?.longTermMemories?.toString(); } } diff --git a/src/services/bot/memory/short-term.ts b/src/services/bot/memory/short-term.ts index 6e039f5..85767ef 100644 --- a/src/services/bot/memory/short-term.ts +++ b/src/services/bot/memory/short-term.ts @@ -78,6 +78,10 @@ export class ShortTermMemoryAgent { .join("\n"), }), }); + // 如果返回内容是个markdown代码块,就让他变回普通json + res?.content?.trim(); + if (res?.content?.startsWith("```json")) {res.content = res?.content?.replace("```json", "");} + if (res?.content?.endsWith("```")) {res.content = res?.content?.replace("```", "");} return jsonDecode(res?.content)?.shortTermMemories?.toString(); } }