@workbench-kit/runtime 0.0.2-prototype.0.1.2 → 0.0.2-prototype.0.2.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@workbench-kit/runtime",
-  "version": "0.0.2-prototype.0.1.2",
+  "version": "0.0.2-prototype.0.2.2",
   "private": false,
   "type": "module",
   "exports": {
@@ -14,7 +14,7 @@
     "!src/**/*.stories.tsx"
   ],
   "dependencies": {
-    "@workbench-kit/contracts": "0.0.2-prototype.0.1.2"
+    "@workbench-kit/contracts": "0.0.2-prototype.0.2.2"
   },
   "description": "Framework-neutral Workbench Kit runtime events and mock runtime utilities.",
   "publishConfig": {

package/src/chatMessages.ts

@@ -0,0 +1,38 @@
+import type { RuntimeChatMessage, WorkbenchRuntimeEvent } from './types';
+
+export function upsertRuntimeChatMessage(
+  messages: readonly RuntimeChatMessage[],
+  message: RuntimeChatMessage,
+): RuntimeChatMessage[] {
+  const messageIndex = messages.findIndex((current) => current.id === message.id);
+  if (messageIndex === -1) {
+    return [...messages, message];
+  }
+
+  if (messages[messageIndex] === message) {
+    return [...messages];
+  }
+
+  return messages.map((current, index) => (index === messageIndex ? message : current));
+}
+
+export function reduceRuntimeChatMessages(
+  messages: readonly RuntimeChatMessage[],
+  event: WorkbenchRuntimeEvent,
+): RuntimeChatMessage[] {
+  if (event.type !== 'message' && event.type !== 'message-delta') {
+    return [...messages];
+  }
+
+  return upsertRuntimeChatMessage(messages, event.message);
+}
+
+export function deriveRuntimeChatMessages(
+  events: readonly WorkbenchRuntimeEvent[],
+  initialMessages: readonly RuntimeChatMessage[] = [],
+): RuntimeChatMessage[] {
+  return events.reduce<RuntimeChatMessage[]>(
+    (messages, event) => reduceRuntimeChatMessages(messages, event),
+    [...initialMessages],
+  );
+}

package/src/index.ts

@@ -1,4 +1,9 @@
 export { createMockWorkbenchRuntime } from './mockRuntime';
+export {
+  deriveRuntimeChatMessages,
+  reduceRuntimeChatMessages,
+  upsertRuntimeChatMessage,
+} from './chatMessages';
 export type {
   MockRuntimeResponsePlan,
   MockWorkbenchRuntime,

package/src/mockRuntime.ts

@@ -23,9 +23,7 @@ export interface MockWorkbenchRuntimeOptions {
   initialStatus?: RuntimeStatus;
   now?: () => string;
   response?:
-    | false
-    | MockRuntimeResponsePlan
-    | ((message: RuntimeChatMessage) => MockRuntimeResponsePlan);
+    false | MockRuntimeResponsePlan | ((message: RuntimeChatMessage) => MockRuntimeResponsePlan);
 }
 
 export interface MockWorkbenchRuntime {