去除更多引用标签

LLM-Red-Team · May 14, 2024 · baf616e · baf616e
1 parent 0a5dbd1
commit baf616e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "step-free-api",
-  "version": "0.0.15",
+  "version": "0.0.16",
   "description": "Stepchat Free API Server",
   "type": "module",
   "main": "dist/index.js",

diff --git a/src/api/controllers/chat.ts b/src/api/controllers/chat.ts
@@ -491,7 +491,7 @@ async function receiveStream(model: string, convId: string, stream: any) {
       } else if (result.textEvent && result.textEvent.text)
         data.choices[0].message.content += result.textEvent.text;
       else if (result.doneEvent) {
-        data.choices[0].message.content = data.choices[0].message.content.replace(/<web_[0-9a-zA-Z]+>/g, '');
+        data.choices[0].message.content = data.choices[0].message.content.replace(/<(web|url|unknown)_[0-9a-zA-Z]+>/g, '');
         data.choices[0].message.content += refContent
           ? `\n\n搜索结果来自：\n${refContent.replace(/\n$/, "")}`
           : "";