wip(chat-db): track the end point for the current message tree.

Author: MarcMcIntosh

refact-agent/gui/src/components/MessageNode/MessageNode.tsx

@@ -1,5 +1,6 @@
-import React from "react";
+import React, { useEffect, useMemo } from "react";
 import {
+  chatDbMessageSliceActions,
   CMessageNode,
   isUserCMessageNode,
   UserCMessageNode,
@@ -19,6 +20,7 @@ import { ArrowLeftIcon, ArrowRightIcon } from "@radix-ui/react-icons";
 import { PlainText } from "../ChatContent/PlainText";
 import { ContextFiles } from "../ChatContent/ContextFiles";
 import { GroupedDiffs } from "../ChatContent/DiffContent";
+import { useAppDispatch } from "../../hooks";
 
 const ElementForNodeMessage: React.FC<{ message: ChatMessage }> = ({
   message,
@@ -57,6 +59,23 @@ const ElementForNodeMessage: React.FC<{ message: ChatMessage }> = ({
 
 export type MessageNodeProps = { children?: CMessageNode };
 export const MessageNode: React.FC<MessageNodeProps> = ({ children }) => {
+  const dispatch = useAppDispatch();
+
+  useEffect(() => {
+    if (children?.children.length === 0) {
+      const action = chatDbMessageSliceActions.setEnd({
+        number: children.message.cmessage_num,
+        alt: children.message.cmessage_alt,
+      });
+      dispatch(action);
+    }
+  }, [
+    children?.children.length,
+    children?.message.cmessage_num,
+    children?.message.cmessage_alt,
+    dispatch,
+  ]);
+
   if (!children) return null;
   return (
     <>

refact-agent/gui/src/features/ChatDB/chatDbMessagesSlice.ts

@@ -34,6 +34,8 @@ type InitialState = {
   messageList: CMessage[];
   loading: boolean;
   error: null | string;
+  endNumber: number;
+  endAlt: number;
 };
 
 const createChatThread = (): CThreadDefault => {
@@ -52,6 +54,8 @@ const initialState: InitialState = {
   messageList: [],
   loading: false,
   error: null,
+  endNumber: 0,
+  endAlt: 0,
 };
 
 function parseCMessageFromChatDBToCMessage(
@@ -98,6 +102,10 @@ export const chatDbMessageSlice = createSlice({
         });
       }
     },
+    setEnd: (state, action: PayloadAction<{ number: number; alt: number }>) => {
+      state.endNumber = action.payload.number;
+      state.endAlt = action.payload.alt;
+    },
   },
 
   selectors: {

refact-agent/gui/src/services/refact/chatdb.ts

@@ -1,11 +1,16 @@
 import { createAsyncThunk } from "@reduxjs/toolkit/react";
 import { AppDispatch, RootState } from "../../app/store";
-import { CHAT_DB_THREADS_SUB, CHAT_DB_MESSAGES_SUB } from "./consts";
+import {
+  CHAT_DB_THREADS_SUB,
+  CHAT_DB_MESSAGES_SUB,
+  CHAT_DB_MESSAGES_UPDATE,
+} from "./consts";
 import { consumeStream } from "../../features/Chat/Thread/utils";
 import {
   isCThreadSubResponseUpdate,
   isCThreadSubResponseDelete,
   isCMessageUpdateResponse,
+  CMessage,
 } from "./types";
 import { chatDbActions } from "../../features/ChatDB/chatDbSlice";
 import {
@@ -136,18 +141,20 @@ export const subscribeToThreadMessagesThunk = createAppAsyncThunk<
       if (!reader) return;
 
       const onAbort = () => {
-        // console.log("knowledge stream aborted");
+        console.log("message stream aborted");
       };
 
       const onChunk = (chunk: Record<string, unknown>) => {
-        // console.log("cmessages chunks");
-        // console.log({ chunk });
+        console.log("cmessages chunks");
+        console.log({ chunk });
         if (isCMessageUpdateResponse(chunk)) {
           const action = chatDbMessageSliceActions.updateMessage({
             threadId: cthreadId,
             message: chunk.cmessage_rec,
           });
           thunkApi.dispatch(action);
+        } else {
+          console.log("invalid chunk");
         }
       };
 
@@ -160,6 +167,28 @@ export const subscribeToThreadMessagesThunk = createAppAsyncThunk<
     });
 });
 
+// How does this add a context file ? or images ?
+export function updateCMessage(
+  cmessages: CMessage[],
+  port = 8001,
+  apiKey?: string | null,
+) {
+  const url = `http://127.0.0.1:${port}${CHAT_DB_MESSAGES_UPDATE}`;
+  const headers = new Headers();
+  headers.append("Content-Type", "application/json");
+  if (apiKey) {
+    headers.append("Authorization", `Bearer ${apiKey}`);
+  }
+
+  return fetch(url, {
+    method: "POST",
+    headers,
+    redirect: "follow",
+    cache: "no-cache",
+    body: JSON.stringify(cmessages),
+  });
+}
+
 // Types for the API
 
 // export interface Chore {

refact-agent/gui/src/services/refact/consts.ts

@@ -43,3 +43,4 @@ export const KNOWLEDGE_CREATE_URL = "/v1/trajectory-save";
 // Chatdblinks
 export const CHAT_DB_THREADS_SUB = "/db_v1/cthreads-sub";
 export const CHAT_DB_MESSAGES_SUB = "/db_v1/cmessages-sub";
+export const CHAT_DB_MESSAGES_UPDATE = "/db_v1//cmessages-update";