MigrateRepos/glm-free-api
1
0
Fork 0
mirror of https://github.com/LLM-Red-Team/glm-free-api.git synced 2024-12-23 01:19:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

优化代码调用输出效果

Browse Source
This commit is contained in:
Vinlic 2024-03-21 11:51:26 +08:00
parent 5158dee32c
commit 888455b56a
1 changed files with 35 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

54
src/api/controllers/chat.ts
View File

@ -479,7 +479,8 @@ async function receiveStream(stream: any): Promise<any> {
created: util.unixTimestamp()
};
let toolCall = false;
// let codeGenerating = false;
let codeGenerating = false;
let codeTemp = '';
let lastExecutionOutput = '';
let textOffset = 0;
const parser = createParser(event => {
@ -518,18 +519,24 @@ async function receiveStream(stream: any): Promise<any> {
toolCall = true;
return innerStr + imageText;
}
// else if(type == 'code' && partStatus == 'init' && !codeGenerating) {
// codeGenerating = true;
// const label = '代码生成中...\n';
// textOffset += label.length;
// return innerStr + label;
// }
// else if(type == 'code' && partStatus == 'finish' && codeGenerating) {
// codeGenerating = false;
// const label = '代码执行中...\n';
// textOffset += label.length;
// return innerStr + label;
// }
else if(type == 'code' && partStatus == 'init') {
let codeHead = '';
if(!codeGenerating) {
codeGenerating = true;
codeHead = '```python\n';
}
const chunk = code.substring(codeTemp.length, code.length);
codeTemp += chunk;
textOffset += codeHead.length + chunk.length;
return innerStr + codeHead + chunk;
}
else if(type == 'code' && partStatus == 'finish' && codeGenerating) {
const codeFooter = '\n```';
codeGenerating = false;
codeTemp = '';
textOffset += codeFooter.length;
return innerStr + codeFooter;
}
else if(type == 'execution_output' && _.isString(content) && partStatus == 'done' && lastExecutionOutput != content) {
lastExecutionOutput = content;
const _content = content.replace(/^\n/, '');
@ -576,6 +583,7 @@ function createTransStream(stream: any, endCallback?: Function) {
let content = '';
let toolCall = false;
let codeGenerating = false;
let codeTemp = '';
let lastExecutionOutput = '';
let textOffset = 0;
!transStream.closed && transStream.write(`data: ${JSON.stringify({
@ -621,17 +629,23 @@ function createTransStream(stream: any, endCallback?: Function) {
toolCall = true;
return innerStr + imageText;
}
else if(type == 'code' && partStatus == 'init' && !codeGenerating) {
else if(type == 'code' && partStatus == 'init') {
let codeHead = '';
if(!codeGenerating) {
codeGenerating = true;
const label = '代码生成中...\n';
textOffset += label.length;
return innerStr + label;
codeHead = '```python\n';
}
const chunk = code.substring(codeTemp.length, code.length);
codeTemp += chunk;
textOffset += codeHead.length + chunk.length;
return innerStr + codeHead + chunk;
}
else if(type == 'code' && partStatus == 'finish' && codeGenerating) {
const codeFooter = '\n```';
codeGenerating = false;
const label = '代码执行中...\n';
textOffset += label.length;
return innerStr + label;
codeTemp = '';
textOffset += codeFooter.length;
return innerStr + codeFooter;
}
else if(type == 'execution_output' && _.isString(content) && partStatus == 'done' && lastExecutionOutput != content) {
lastExecutionOutput = content;