feat: add session storage adapter for upcoming web frontend

eslint.config.js

@@ -39,6 +39,8 @@ export default [
                 format: ["camelCase", "PascalCase"],
             }],
 
+            "@typescript-eslint/no-invalid-this": ["error"],
+
             "@stylistic/brace-style": ["error", "allman", { allowSingleLine: true }],
             "@stylistic/quotes": ["error", "double"],
             "@stylistic/comma-dangle": ["error", "always-multiline"],

package.json

@@ -77,6 +77,9 @@
         "ulid": "^3.0.1",
         "write-file-atomic": "^6.0.0"
     },
+    "optionalDependencies": {
+        "@supabase/supabase-js": "^2.57.2"
+    },
     "devDependencies": {
         "@sindresorhus/tsconfig": "^7.0.0",
         "@stylistic/eslint-plugin": "^5.1.0",

src/ai/custom-messages.ts

@@ -7,7 +7,7 @@ type TCustomMessageType = "error" | "tool-progress";
 
 export type TMessageType = ReturnType<BaseMessage["getType"]> | TCustomMessageType;
 
-export class CustomMessage
+export abstract class CustomMessage
 {
     _type: TCustomMessageType;
 
@@ -21,6 +21,8 @@ export class CustomMessage
         return this._type;
     }
 
+    abstract isMessageFinished(): boolean;
+
     static fromObject(obj: any): CustomMessage
     {
         if (obj._type === "error")
@@ -29,7 +31,7 @@ export class CustomMessage
         }
         else if (obj._type === "tool-progress")
         {
-            return new ToolProgressMessage(obj.toolCall, obj.status, obj.content, obj.confirmState);
+            return new ToolProgressMessage(obj.toolCall, obj.status, obj.content, obj.confirmState, obj.info);
         }
 
         throw new Error(`Unknown custom message type: ${obj._type}`);
@@ -61,6 +63,11 @@ class ErrorMessage extends CustomMessage
             };
         }
     }
+
+    override isMessageFinished(): boolean
+    {
+        return true;
+    }
 }
 
 class ToolProgressMessage extends CustomMessage
@@ -71,12 +78,16 @@ class ToolProgressMessage extends CustomMessage
     status: "pending" | "pending-confirmation" | "confirmed" | "declined" | "success" | "error";
     content?: string;
     confirmState: typeof ToolProgressMessage.CONFIRM_STATES[number];
+    info: {
+        description?: string;
+        fileName?: string;
+    };
 
     static isTypeOf = (m: any) => m instanceof ToolProgressMessage;
 
     clone({ status, confirmState }: Pick<ToolProgressMessage, "status" | "confirmState">)
     {
-        return new ToolProgressMessage(this.toolCall, status, this.content, confirmState);
+        return new ToolProgressMessage(this.toolCall, status, this.content, confirmState, this.info);
     }
 
     toggleConfirmState({ direction }: { direction: -1 | 1 })
@@ -124,14 +135,20 @@ class ToolProgressMessage extends CustomMessage
         return toolCallChunks?.map(i => ToolProgressMessage.createFromChunk(i)).filter(i => !!i) || [];
     }
 
-    constructor(toolCall: ToolCall, status?: ToolProgressMessage["status"], content?: string, confirmState?: ToolProgressMessage["confirmState"])
+    constructor(toolCall: ToolCall, status?: ToolProgressMessage["status"], content?: string, confirmState?: ToolProgressMessage["confirmState"], info?: ToolProgressMessage["info"])
     {
         super("tool-progress");
 
         this.toolCall = toolCall;
         this.status = status || "pending";
         this.content = content;
         this.confirmState = confirmState || "no";
+        this.info = info || {};
+    }
+
+    override isMessageFinished(): boolean
+    {
+        return this.status !== "pending" && this.status !== "pending-confirmation";
     }
 }
 
@@ -149,9 +166,38 @@ export const isCustomMessage = (message: any): message is CustomMessage =>
     );
 };
 
+const setMessageIsStreaming = (msg: BaseMessage, isStreaming: boolean) =>
+{
+    msg.additional_kwargs["_coba_message_is_streaming"] = isStreaming;
+    return msg;
+};
+
+const isMessageStreaming = (msg: TMessage): boolean =>
+{
+    return (msg instanceof BaseMessage) && (msg as BaseMessage).additional_kwargs["_coba_message_is_streaming"] === true;
+};
+
+const isMessageFinished = (msg: TMessage): boolean =>
+{
+    if (msg instanceof BaseMessage)
+    {
+        return !isMessageStreaming(msg);
+    }
+
+    return msg.isMessageFinished();
+};
+
+const resetIsStreamingFlag = (msg: TMessage[]): TMessage[] => 
+{
+    return msg.map(m => (m instanceof BaseMessage) ? setMessageIsStreaming(m, false) : m);
+};
+
 export
 {
     ErrorMessage,
+    isMessageFinished,
+    isMessageStreaming,
+    resetIsStreamingFlag,
+    setMessageIsStreaming,
     ToolProgressMessage,
 };
-