From 735fcecbabde504cf3586c45b34737ff09abc7a8 Mon Sep 17 00:00:00 2001
From: Yanzhen Yu <yanzhen@smartx.com>
Date: Thu, 21 Dec 2023 16:14:14 +0800
Subject: [PATCH] fix description extraction

---
 prompts/common.ts | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/prompts/common.ts b/prompts/common.ts
index 1f69ba8..71f4c4a 100644
--- a/prompts/common.ts
+++ b/prompts/common.ts
@@ -50,10 +50,11 @@ export async function getCode(
         throw new Error(`invalid code blocks ${JSON.stringify(codeBlocks)}`);
       }
 
+      remove(tree, "code");
       return {
         code: codeBlocks[0],
         usage: chatCompletion.usage,
-        description: toMarkdown(remove(tree, "code")),
+        description: toMarkdown(tree),
       };
     },
     { delay: 100, maxTry: 3 }