Merge pull request #24 from smallcloudai/openai-api-update

Openai api update
smallcloudai · May 16, 2024 · ff64a57 · ff64a57
2 parents 5bf93ab + 02733aa
commit ff64a57
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 9 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "refact-chat-js",
-  "version": "1.0.1-alpha.0",
+  "version": "1.0.1-alpha.1",
   "type": "module",
   "license": "BSD-3-Clause",
   "files": [

diff --git a/src/hooks/useEventBusForChat/index.ts b/src/hooks/useEventBusForChat/index.ts
@@ -75,17 +75,19 @@ function formatChatResponse(
       return acc.concat([[cur.delta.role, cur.delta.content]]);
     }
 
-    if (acc.length === 0) {
+    if (acc.length === 0 && cur.delta.role) {
       return acc.concat([[cur.delta.role, cur.delta.content]]);
     }
 
     const lastMessage = acc[acc.length - 1];
     if (lastMessage[0] === "assistant") {
       const last = acc.slice(0, -1);
       const currentMessage = lastMessage[1];
-      return last.concat([
-        [cur.delta.role, currentMessage + cur.delta.content],
-      ]);
+      return last.concat([["assistant", currentMessage + cur.delta.content]]);
+    }
+
+    if (cur.delta.role === null) {
+      return acc;
     }
 
     return acc.concat([[cur.delta.role, cur.delta.content]]);

diff --git a/src/services/refact.ts b/src/services/refact.ts
@@ -62,11 +62,11 @@ export function isChatContextFileMessage(
 }
 
 interface BaseDelta {
-  role: ChatRole;
+  role: ChatRole | null;
 }
 
 interface AssistantDelta extends BaseDelta {
-  role: "assistant";
+  role: "assistant" | null;
   content: string;
 }
 interface ChatContextFileDelta extends BaseDelta {