Merge pull request #33 from lowcoding/feat/v1.8.8

🐞 fix: openai log
lowcoding · Aug 20, 2024 · d2abc1b · d2abc1b
2 parents 8e961f4 + fd4770c
commit d2abc1b
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/package.json b/package.json
@@ -3,7 +3,7 @@
 	"displayName": "lowcode",
 	"description": "lowcode tool, support ChatGPT and other LLM",
 	"author": "wjkang <ruoxieme@gmail.com>",
-	"version": "1.8.7",
+	"version": "1.8.8",
 	"icon": "asset/icon.png",
 	"publisher": "wjkang",
 	"repository": "https://github.com/lowcoding/lowcode-vscode",

diff --git a/src/utils/openai.ts b/src/utils/openai.ts
@@ -9,7 +9,7 @@ export const createChatCompletion = (options: {
 }) =>
   new Promise<string>((resolve) => {
     let combinedResult = '';
-    const error = '发生错误：';
+    let error = '发生错误：';
     const config = getChatGPTConfig();
     const request = https.request(
       {
@@ -27,6 +27,7 @@ export const createChatCompletion = (options: {
         res.on('data', async (chunk) => {
           const text = new TextDecoder('utf-8').decode(chunk);
           const data = text.split('\n\n').filter((s) => s);
+          console.log(data);
           for (let i = 0; i < data.length; i++) {
             try {
               let element = data[i];
@@ -43,7 +44,7 @@ export const createChatCompletion = (options: {
                 // 处理没有 data 开头
                 element = preDataLast + element;
               }
-              if (element.includes('data: ')) {
+              if (element.startsWith('data: ')) {
                 if (element.includes('[DONE]')) {
                   if (options.handleChunk) {
                     options.handleChunk({ text: '' });
@@ -69,13 +70,18 @@ export const createChatCompletion = (options: {
                 }
               } else {
                 console.log('no includes data: ', element);
+                if (options.handleChunk) {
+                  options.handleChunk({
+                    text: element,
+                  });
+                }
               }
             } catch (e) {
               console.error({
                 e: (e as Error).toString(),
                 element: data[i],
               });
-              // error = (e as Error).toString();
+              error = (e as Error).toString();
             }
           }
         });