修复代码生成调用输出

LLM-Red-Team · Dec 13, 2024 · 05ecba5 · 05ecba5
1 parent a969acd
commit 05ecba5
Show file tree

Hide file tree

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

diff --git a/src/api/controllers/chat.ts b/src/api/controllers/chat.ts
@@ -991,7 +991,7 @@ async function receiveStream(stream: any): Promise<any> {
                 textOffset += imageText.length;
                 toolCall = true;
                 return innerStr + imageText;
-              } else if (type == "code" && partStatus == "init") {
+              } else if (type == "code" && status == "init") {
                 let codeHead = "";
                 if (!codeGenerating) {
                   codeGenerating = true;
@@ -1003,7 +1003,7 @@ async function receiveStream(stream: any): Promise<any> {
                 return innerStr + codeHead + chunk;
               } else if (
                 type == "code" &&
-                partStatus == "finish" &&
+                status == "finish" &&
                 codeGenerating
               ) {
                 const codeFooter = "\n```\n";
@@ -1014,7 +1014,7 @@ async function receiveStream(stream: any): Promise<any> {
               } else if (
                 type == "execution_output" &&
                 _.isString(content) &&
-                partStatus == "done" &&
+                status == "finish" &&
                 lastExecutionOutput != content
               ) {
                 lastExecutionOutput = content;
@@ -1154,7 +1154,7 @@ function createTransStream(stream: any, endCallback?: Function) {
               textOffset += imageText.length;
               toolCall = true;
               return innerStr + imageText;
-            } else if (type == "code" && partStatus == "init") {
+            } else if (type == "code" && status == "init") {
               let codeHead = "";
               if (!codeGenerating) {
                 codeGenerating = true;
@@ -1166,7 +1166,7 @@ function createTransStream(stream: any, endCallback?: Function) {
               return innerStr + codeHead + chunk;
             } else if (
               type == "code" &&
-              partStatus == "finish" &&
+              status == "finish" &&
               codeGenerating
             ) {
               const codeFooter = "\n```\n";
@@ -1177,7 +1177,7 @@ function createTransStream(stream: any, endCallback?: Function) {
             } else if (
               type == "execution_output" &&
               _.isString(content) &&
-              partStatus == "done" &&
+              status == "finish" &&
               lastExecutionOutput != content
             ) {
               lastExecutionOutput = content;