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

idootop · Jul 22, 2024 · c985e5b · c985e5b
1 parent 2f53d99
commit c985e5b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git 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
@@ -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();
   }
 }