upstream/mi-gpt
1
0
Fork 0
mirror of https://github.com/idootop/mi-gpt.git synced 2025-04-22 15:42:56 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: 去除掉ai返回内容中的代码块样式

Browse Source
This commit is contained in:
roitium 2024-07-22 12:13:02 +08:00
parent 2f53d99359
commit c985e5b08f
2 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/services/bot/memory/long-term.ts
View File

@ -67,6 +67,10 @@ export class LongTermMemoryAgent {
shortTermMemory: lastOf(newMemories)!.text, 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(); return jsonDecode(res?.content)?.longTermMemories?.toString();
} }
} }

4
src/services/bot/memory/short-term.ts
View File

@ -78,6 +78,10 @@ export class ShortTermMemoryAgent {
.join("\n"), .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(); return jsonDecode(res?.content)?.shortTermMemories?.toString();
} }
} }