fix: message depth

Dogtiti · May 3, 2023 · 1563665 · 1563665
1 parent d7784d7
commit 1563665
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 22 deletions.
diff --git a/__tests__/create-model.test.ts b/__tests__/create-model.test.ts
@@ -8,21 +8,21 @@ describe("createModel", () => {
       customApiKey: "test_api_key",
       customTemperature: 0.222,
       customModelName: "Custom_Model",
-      maxTokens: 1234,
+      customMaxTokens: 1234,
     };
 
     const model = createModel(customSettings);
 
     expect(model.temperature).toBe(customSettings.customTemperature);
     expect(model.modelName).toBe(customSettings.customModelName);
-    expect(model.maxTokens).toBe(customSettings.maxTokens);
+    expect(model.maxTokens).toBe(customSettings.customMaxTokens);
   });
 
   test("should use default settings when API key is not provided", () => {
     const customSettings = {
       customTemperature: 0.222,
       customModelName: "Custom_Model",
-      maxTokens: 1234,
+      customMaxTokens: 1234,
     };
 
     const model = createModel(customSettings);

diff --git a/src/components/AutonomousAgent.ts b/src/components/AutonomousAgent.ts
@@ -196,11 +196,10 @@ class AutonomousAgent {
         };
         return task;
       });
-      //FIXME
-      // this.tasks = newTasks.concat(this.tasks);
+      this.tasks = newTasks.concat(this.tasks);
       for (const task of newTasks) {
         await new Promise((r) => setTimeout(r, TIMOUT_SHORT));
-        this.tasks.push(task);
+        // this.tasks.push(task);
         this.sendMessage(task);
       }
 
@@ -436,8 +435,6 @@ class AutonomousAgent {
   sendErrorMessage(error: string) {
     this.sendMessage({ type: MESSAGE_TYPE_SYSTEM, value: error });
   }
-
-
 }
 
 const testConnection = async (modelSettings: ModelSettings) => {

diff --git a/src/components/ChatWindow.tsx b/src/components/ChatWindow.tsx
@@ -96,7 +96,7 @@ const ChatWindow = ({
       return depth;
     }
     const index = messages.findLastIndex(
-      (i) => i.parentTaskId && i.taskId === message.taskId
+      (i: Message) => i.parentTaskId && i.taskId === message.taskId
     );
     if (index > -1) {
       const { parentTaskId } = messages[index] as Message;

diff --git a/src/components/stores/messageStore.ts b/src/components/stores/messageStore.ts
@@ -39,21 +39,20 @@ const createMessageSlice: StateCreator<
       const newTask = { ...newMessage };
       newMessage = { ...newMessage };
       set((state) => {
-        //FIXME
-        // const preMessages = state.messages;
-        // const index = preMessages.findLastIndex(
-        //   (message) => message?.taskId === newMessage?.parentTaskId
-        // );
-        // const messagesCopy = [...preMessages];
-        // messagesCopy.splice(
-        //   index > -1 ? index + 1 : messagesCopy.length,
-        //   0,
-        //   newMessage
-        // );
+        const preMessages = state.messages;
+        const index = preMessages.findLastIndex(
+          (message) => message?.taskId === newMessage?.taskId
+        );
+        const messagesCopy = [...preMessages];
+        messagesCopy.splice(
+          index > -1 ? index + 1 : messagesCopy.length,
+          0,
+          newMessage
+        );
         return {
           ...state,
-          // messages: messagesCopy,
-          messages: [...state.messages, newMessage],
+          messages: messagesCopy,
+          // messages: [...state.messages, newMessage],
           tasks:
             isTask(newTask) && !isExistingTask(newTask)
               ? [...state.tasks, newTask]