@agentshouse/mdruntime 0.1.0-cfb2ae0107054 → 0.1.0-cfcbb238a5fa0

package/dist/conversation.js

@@ -22,6 +22,17 @@ export function unheldView(conversation) {
         chatDraft: null,
     };
 }
+export const SESSION_STARTED = 'Session started';
+const NEW_CHAT = 'new:';
+export function newChat(agent) {
+    return `${NEW_CHAT}${agent}`;
+}
+function creationOf(input) {
+    if (typeof input !== 'object' || input === null)
+        return null;
+    const { agent_id: agent, operation_id: operation } = input;
+    return typeof agent === 'string' && typeof operation === 'string' ? { agent, operation } : null;
+}
 function startedTurn(turn, operation) {
     if (operation.op === 'begin')
         return { turn, started: operation.at, parts: [], context: null };
@@ -78,11 +89,15 @@ function ordered(left, right) {
         return left.identity < right.identity ? -1 : 1;
     return 0;
 }
-function merged(held, arriving) {
+function windowOf(held, arriving, limit) {
     const byIdentity = new Map(held.map((message) => [message.identity, message]));
     for (const message of arriving)
         byIdentity.set(message.identity, message);
-    return [...byIdentity.values()].sort(ordered);
+    const next = [...byIdentity.values()].sort(ordered);
+    return limit === undefined ? next : next.slice(Math.max(0, next.length - limit));
+}
+function newestOf(arriving, from) {
+    return arriving.reduce((top, message) => (top === null || message.position > top ? message.position : top), from);
 }
 function acceptedMessage(value) {
     if (typeof value !== 'object' || value === null)
@@ -141,14 +156,14 @@ function historyPages(result) {
     }
     return pages;
 }
-function turnOf(result) {
+function answered(result, field) {
     if (result.kind !== 'action')
         return null;
     const answer = result.result;
     if (typeof answer !== 'object' || answer === null)
         return null;
-    const record = answer;
-    return typeof record.turn === 'string' ? record.turn : null;
+    const value = answer[field];
+    return typeof value === 'string' ? value : null;
 }
 function limitOf(options) {
     const { limit } = options;
@@ -159,11 +174,18 @@ function limitOf(options) {
     }
     return limit;
 }
+const EXPIRED = { kind: 'rejected', code: 'expired', fields: {}, retryable: false };
 function unwritten(draft) {
     return draft.text === '' && draft.files.length === 0;
 }
 function pending(draft) {
-    return draft.submitted === undefined ? {} : { submitted: draft.submitted };
+    return {
+        ...(draft.submitted === undefined ? {} : { submitted: draft.submitted }),
+        ...(draft.moves === undefined ? {} : { moves: draft.moves }),
+    };
+}
+function empty(draft) {
+    return unwritten(draft) && draft.submitted === undefined && draft.moves === undefined;
 }
 function withSubmitted(draft, submitted) {
     const { submitted: _held, ...unsent } = draft;
@@ -178,15 +200,18 @@ function returned(draft, submission, others) {
     }, others);
 }
 export function createConversationStore(deps) {
-    const messages = new Map();
+    const stored = new Set();
     const reads = new Map();
+    const partials = new Set();
     const drafts = new Map();
-    const sending = new Set();
+    const sending = new Map();
     const behind = new Set();
+    const catching = new Map();
     const gaps = new Map();
     const lives = new Map();
     const views = new Map();
     const listing = new Map();
+    const creations = new Set();
     const listWatchers = new Set();
     let listKnown = false;
     const agentWatchers = new Set();
@@ -198,22 +223,17 @@ export function createConversationStore(deps) {
     function liveOf(conversation) {
         return lives.get(conversation) ?? QUIET;
     }
-    function shownMessages(conversation, limit) {
-        const held = messages.get(conversation) ?? [];
-        const shown = limit === undefined ? held : held.slice(Math.max(0, held.length - limit));
-        return shown.map(({ identity, turn, text, message }) => ({
-            identity,
-            turn,
-            text,
-            message: structuredClone(message),
-        }));
-    }
-    function viewOf(conversation, limit, shown = shownMessages(conversation, limit)) {
+    function viewOf(conversation, held) {
         const draft = drafts.get(conversation);
         const facts = liveOf(conversation);
         return {
             conversation,
-            messages: shown,
+            messages: held.map(({ identity, turn, text, message }) => ({
+                identity,
+                turn,
+                text,
+                message,
+            })),
             turn: structuredClone(facts.turn),
             questions: structuredClone(facts.questions),
             process: facts.process === null ? null : { ...facts.process },
@@ -224,23 +244,30 @@ export function createConversationStore(deps) {
                 : { text: draft.text, files: [...draft.files], revision: draft.revision },
         };
     }
-    function emit(conversation, messagesChanged = true) {
+    function emit(conversation) {
         if (deps.closed())
             return;
         for (const view of views.values()) {
-            if (view.conversation !== conversation)
+            if (view.conversation !== conversation || view.window === null)
                 continue;
-            if (messagesChanged)
-                view.messages = null;
             const ready = [...view.watchers].filter((watcher) => watcher.ready);
             if (ready.length === 0)
                 continue;
-            view.messages ??= shownMessages(conversation, view.limit);
-            const value = viewOf(conversation, view.limit, view.messages);
+            const value = viewOf(conversation, view.window);
             for (const watcher of ready)
                 watcher.notify(value);
         }
     }
+    function slide(conversation, arriving) {
+        for (const view of views.values()) {
+            if (view.conversation !== conversation || view.window === null)
+                continue;
+            view.window = arriving === null ? [] : windowOf(view.window, arriving, view.limit);
+        }
+    }
+    function changesOf(conversation, arriving) {
+        return arriving.map(({ identity, text }) => ({ conversation, identity, text }));
+    }
     function stateOf(conversation) {
         const facts = liveOf(conversation);
         if (facts.questions.length > 0)
@@ -297,7 +324,10 @@ export function createConversationStore(deps) {
         const declaration = deps.agents();
         if (declaration === undefined)
             return;
-        const { documents } = await deps.documents(declaration.locator);
+        const held = await deps.documents(declaration);
+        if ('pending' in held)
+            return;
+        const { documents } = held;
         const names = new Map();
         const retired = new Set();
         for (const document of documents) {
@@ -320,6 +350,12 @@ export function createConversationStore(deps) {
         if (named)
             emitList();
         emitAgents();
+        const ended = [...drafts.keys()].filter(retiredChat);
+        if (ended.length > 0)
+            await remove(ended);
+    }
+    function retiredChat(conversation) {
+        return (conversation.startsWith(NEW_CHAT) && retiredAgents.has(conversation.slice(NEW_CHAT.length)));
     }
     function agentsReady() {
         agentsRead ??= readAgents().catch(() => {
@@ -329,7 +365,7 @@ export function createConversationStore(deps) {
     }
     function live(conversation, change) {
         setLive(conversation, { ...liveOf(conversation), ...change });
-        emit(conversation, false);
+        emit(conversation);
     }
     function relayed(conversation, turn, operation) {
         const held = liveOf(conversation).turn;
@@ -357,21 +393,25 @@ export function createConversationStore(deps) {
     }
     async function remove(leaving) {
         await deps.durably(async (local) => {
-            await local.dropConversations(leaving);
             const removed = [];
-            const changed = [];
             for (const conversation of leaving) {
-                for (const message of messages.get(conversation) ?? []) {
-                    removed.push({ conversation, identity: message.identity });
+                for (const identity of await local.messageIdentities(conversation)) {
+                    removed.push({ conversation, identity, text: null });
                 }
-                if (messages.has(conversation) || drafts.has(conversation) || lives.has(conversation)) {
+            }
+            await local.dropConversations(leaving);
+            const changed = [];
+            for (const conversation of leaving) {
+                if (stored.has(conversation) || drafts.has(conversation) || lives.has(conversation)) {
                     changed.push(conversation);
                 }
-                messages.delete(conversation);
+                stored.delete(conversation);
                 reads.delete(conversation);
+                partials.delete(conversation);
                 drafts.delete(conversation);
                 behind.delete(conversation);
                 setLive(conversation, null);
+                slide(conversation, null);
             }
             if (changed.length > 0)
                 deps.publish(conversationChangeSet(removed, changed));
@@ -393,23 +433,21 @@ export function createConversationStore(deps) {
             await local.keepMessages({
                 conversation,
                 messages: [message],
-                ...(advances ? { read: message.position } : {}),
+                ...(advances ? { read: { at: message.position, partial: false } } : {}),
             });
-            messages.set(conversation, merged(messages.get(conversation) ?? [], [message]));
+            stored.add(conversation);
             if (advances)
                 reads.set(conversation, message.position);
             sealTurns(conversation, new Set([message.turn]));
-            deps.publish(conversationChangeSet([{ conversation, identity: message.identity }], [conversation]));
+            slide(conversation, [message]);
+            deps.publish(conversationChangeSet(changesOf(conversation, [message]), [conversation]));
         });
     }
-    async function readHistories(conversations) {
+    async function readHistories(conversations, ended) {
         const generations = new Map(conversations.map((one) => [one, gaps.get(one) ?? 0]));
-        const starts = new Map(conversations.map((one) => [one, reads.get(one)]));
-        const arriving = new Map();
-        const complete = new Set();
         let asking = conversations.map((conversation) => ({
             conversation,
-            after: starts.get(conversation) ?? undefined,
+            after: reads.get(conversation) ?? undefined,
         }));
         while (asking.length > 0) {
             const result = await deps.call('conversation.history', {
@@ -421,43 +459,64 @@ export function createConversationStore(deps) {
             const pages = historyPages(result);
             if (pages === null)
                 return result;
-            const asked = new Set(asking.map(({ conversation }) => conversation));
+            const asked = new Map(asking.map(({ conversation, after }) => [conversation, after]));
             const following = [];
             for (const page of pages) {
                 if (!asked.has(page.conversation))
                     continue;
-                arriving.set(page.conversation, [
-                    ...(arriving.get(page.conversation) ?? []),
-                    ...page.messages,
-                ]);
-                if (page.next === undefined)
-                    complete.add(page.conversation);
-                else
+                await keepPage(page.conversation, page.messages, asked.get(page.conversation) ?? null, page.next, generations.get(page.conversation) ?? 0);
+                if (page.next !== undefined) {
                     following.push({ conversation: page.conversation, after: page.next });
+                }
+            }
+            for (const { conversation } of asking) {
+                if (!following.some((next) => next.conversation === conversation))
+                    ended(conversation);
             }
             asking = following;
         }
-        for (const [conversation, held] of arriving) {
-            if (!complete.has(conversation))
-                continue;
-            await keepHistory(conversation, held, starts.get(conversation), generations.get(conversation) ?? 0);
-        }
         return null;
     }
-    function keepHistory(conversation, arriving, from, generation) {
+    function caughtUp(conversations) {
+        const ends = new Map();
+        for (const conversation of conversations) {
+            const held = new Promise((end) => ends.set(conversation, end));
+            catching.set(conversation, held);
+            void held.then(() => {
+                if (catching.get(conversation) === held)
+                    catching.delete(conversation);
+            });
+        }
+        const ended = (conversation) => ends.get(conversation)?.();
+        const read = readHistories(conversations, ended);
+        const endAll = () => {
+            for (const conversation of conversations)
+                ended(conversation);
+        };
+        read.then(endAll, endAll);
+        return read;
+    }
+    function keepPage(conversation, arriving, from, next, generation) {
         return deps.durably(async (local) => {
             if (!replicated.has(conversation))
                 return;
-            const newest = arriving.reduce((top, message) => (top === null || message.position > top ? message.position : top), from ?? null);
+            const whole = next === undefined;
+            const cursor = next ?? newestOf(arriving, from);
+            const partial = !whole && (partials.has(conversation) || !reads.has(conversation));
             const sealing = await sealed(local, conversation, new Set(arriving.map((message) => message.turn)));
-            await local.keepMessages({ conversation, messages: arriving, read: newest });
-            messages.set(conversation, merged(messages.get(conversation) ?? [], arriving));
-            reads.set(conversation, newest);
+            await local.keepMessages({ conversation, messages: arriving, read: { at: cursor, partial } });
+            stored.add(conversation);
+            reads.set(conversation, cursor);
+            if (partial)
+                partials.add(conversation);
+            else
+                partials.delete(conversation);
             sealTurns(conversation, new Set(arriving.map((message) => message.turn)));
-            if ((gaps.get(conversation) ?? 0) === generation)
+            if (whole && (gaps.get(conversation) ?? 0) === generation)
                 behind.delete(conversation);
-            if (arriving.length > 0 || sealing) {
-                deps.publish(conversationChangeSet(arriving.map((message) => ({ conversation, identity: message.identity })), [conversation]));
+            slide(conversation, arriving);
+            if (arriving.length > 0 || sealing || whole) {
+                deps.publish(conversationChangeSet(changesOf(conversation, arriving), [conversation]));
             }
         });
     }
@@ -470,7 +529,7 @@ export function createConversationStore(deps) {
             if (held === undefined)
                 return;
             setLive(conversation, { ...QUIET, turn: held.turn });
-            emit(conversation, false);
+            emit(conversation);
             return;
         }
         if (event.kind === 'turn') {
@@ -504,12 +563,12 @@ export function createConversationStore(deps) {
             emit(conversation);
     }
     async function kept(local, conversation, following) {
-        if (following === null || (unwritten(following) && following.submitted === undefined)) {
+        if (following === null || empty(following)) {
             await local.dropChatDraft(conversation);
             drafts.delete(conversation);
         }
         else {
-            await local.keepChatDraft(following);
+            await local.keepChatDrafts([following]);
             drafts.set(conversation, following);
         }
     }
@@ -522,9 +581,32 @@ export function createConversationStore(deps) {
         await kept(local, conversation, withSubmitted(held, left));
         return true;
     }
+    function moved(agent, operation, to) {
+        const from = newChat(agent);
+        return deps.durably(async (local) => {
+            const held = drafts.get(from);
+            const earlier = (held?.moves ?? []).filter((one) => !deps.expired(one));
+            if (earlier.includes(operation))
+                return;
+            const blank = {
+                conversation: from,
+                text: '',
+                files: [],
+                revision: chatDraftRevision(),
+            };
+            const { moves: _answered, ...unsent } = held ?? blank;
+            const taken = drafts.has(to) || empty(unsent) ? [] : [{ ...unsent, conversation: to }];
+            const left = { ...(taken.length === 0 ? unsent : blank), moves: [...earlier, operation] };
+            await local.keepChatDrafts([left, ...taken]);
+            drafts.set(from, left);
+            for (const one of taken)
+                drafts.set(to, one);
+            deps.publish(conversationChangeSet([], [from, to]));
+        });
+    }
     function withDraft(conversation, next) {
         return deps.durably(async (local) => {
-            const following = next(drafts.get(conversation));
+            const following = await next(drafts.get(conversation), local);
             if (following === undefined)
                 return;
             await kept(local, conversation, following);
@@ -533,8 +615,10 @@ export function createConversationStore(deps) {
     }
     async function submitted(conversation, operation, draft) {
         let submission;
+        let earlier = false;
         await withDraft(conversation, (current) => {
             submission = current?.submitted?.find((one) => one.operation === operation);
+            earlier = submission !== undefined;
             if (submission !== undefined)
                 return undefined;
             const said = draft ?? current;
@@ -552,10 +636,19 @@ export function createConversationStore(deps) {
                 submitted: [...(current?.submitted ?? []), submission],
             };
         });
-        return submission;
+        return { submission: submission, earlier };
+    }
+    function forgotten(conversation, submission) {
+        return withDraft(conversation, (current) => {
+            const held = current?.submitted ?? [];
+            if (current === undefined || !held.includes(submission))
+                return undefined;
+            return withSubmitted(current, held.filter((one) => one !== submission));
+        });
     }
     async function dispatched(conversation, submission) {
-        sending.add(submission.operation);
+        const dispatch = {};
+        sending.set(submission.operation, dispatch);
         try {
             const files = await Promise.all(submission.files.map(async (file) => ({ ...(await declaredFile(file)), blob: file.blob })));
             const result = await invoke('conversation.send', conversation, {
@@ -565,17 +658,19 @@ export function createConversationStore(deps) {
             });
             if (result.kind === 'revoked-authorization')
                 return result;
-            if (result.kind !== 'action' && result.retryable)
+            if (result.kind !== 'action' && result.retryable) {
+                deps.unanswered();
                 return result;
-            const turn = turnOf(result);
-            await withDraft(conversation, (current) => {
+            }
+            const turn = answered(result, 'turn');
+            await withDraft(conversation, async (current, local) => {
                 const held = current?.submitted?.find((one) => one.operation === submission.operation);
                 if (current === undefined || held === undefined || held.turn !== null)
                     return undefined;
                 const others = current.submitted.filter((one) => one !== held);
                 if (result.kind !== 'action')
                     return returned(current, held, others);
-                const arrived = (messages.get(conversation) ?? []).some((message) => message.turn === turn);
+                const arrived = turn !== null && (await local.turnArrived(conversation, turn));
                 return withSubmitted(current, turn === null || arrived
                     ? others
                     : current.submitted.map((one) => (one === held ? { ...held, turn } : one)));
@@ -583,7 +678,8 @@ export function createConversationStore(deps) {
             return result;
         }
         finally {
-            sending.delete(submission.operation);
+            if (sending.get(submission.operation) === dispatch)
+                sending.delete(submission.operation);
         }
     }
     function invoke(name, conversation, extra = {}) {
@@ -596,14 +692,27 @@ export function createConversationStore(deps) {
         },
         async read(conversation, options = {}) {
             const limit = limitOf(options);
-            return deps.durably(() => Promise.resolve(viewOf(conversation, limit)));
+            return deps.durably(async (local) => structuredClone(viewOf(conversation, await local.messageWindow(conversation, limit))));
         },
-        create(input) {
-            return deps.call('conversation.create', input);
+        async create(input) {
+            const result = await deps.call('conversation.create', input);
+            const conversation = answered(result, 'conversation');
+            if (conversation === null)
+                return result;
+            if (!listing.has(conversation))
+                creations.add(conversation);
+            const creation = creationOf(input);
+            if (creation !== null)
+                await moved(creation.agent, creation.operation, conversation);
+            return result;
         },
         async send(conversation, operation = changeBatchIdentity(), draft) {
             if (deps.online()) {
-                return dispatched(conversation, await submitted(conversation, operation, draft));
+                const { submission, earlier } = await submitted(conversation, operation, draft);
+                if (!earlier || !deps.expired(operation))
+                    return dispatched(conversation, submission);
+                await forgotten(conversation, submission);
+                return EXPIRED;
             }
             if (draft !== undefined) {
                 await withDraft(conversation, (current) => current?.submitted?.some((one) => one.operation === operation) === true
@@ -644,7 +753,7 @@ export function createConversationStore(deps) {
                 text: draft.text,
                 files: [...draft.files],
                 revision,
-                ...(current?.submitted === undefined ? {} : { submitted: current.submitted }),
+                ...(current === undefined ? {} : pending(current)),
             }));
         },
         discard(conversation) {
@@ -662,18 +771,19 @@ export function createConversationStore(deps) {
     return {
         surface,
         chatDraft,
-        hydrate(held, stored) {
-            messages.clear();
+        hydrate(held, kept) {
+            stored.clear();
             reads.clear();
-            for (const view of views.values())
-                view.messages = null;
+            partials.clear();
             for (const entry of held) {
-                messages.set(entry.conversation, [...entry.messages].sort(ordered));
+                stored.add(entry.conversation);
                 if (entry.read !== undefined)
                     reads.set(entry.conversation, entry.read);
+                if (entry.partial)
+                    partials.add(entry.conversation);
             }
             drafts.clear();
-            for (const record of stored)
+            for (const record of kept)
                 drafts.set(record.conversation, record);
         },
         async reconcile() {
@@ -681,8 +791,10 @@ export function createConversationStore(deps) {
                 return;
             const named = new Set(listing.keys());
             const leaving = [
-                ...new Set([...replicated, ...messages.keys(), ...reads.keys(), ...drafts.keys()]),
-            ].filter((conversation) => !named.has(conversation));
+                ...new Set([...replicated, ...stored, ...reads.keys(), ...drafts.keys()]),
+            ].filter((conversation) => !named.has(conversation) &&
+                !creations.has(conversation) &&
+                (!conversation.startsWith(NEW_CHAT) || retiredChat(conversation)));
             for (const conversation of named) {
                 if (!replicated.has(conversation))
                     behind.add(conversation);
@@ -693,12 +805,20 @@ export function createConversationStore(deps) {
                 await remove(leaving);
         },
         async catchUp(progress) {
-            const reading = [...behind].filter((conversation) => replicated.has(conversation)).sort();
-            if (reading.length > 0 && !deps.closed()) {
-                const failed = await readHistories(reading);
+            const owed = (conversation) => behind.has(conversation) && replicated.has(conversation);
+            const read = async (conversations) => {
+                if (conversations.length === 0 || deps.closed())
+                    return;
+                const failed = await caughtUp(conversations);
                 if (failed?.kind === 'transport-uncertainty')
                     progress.reachable = false;
-            }
+            };
+            const asking = [...behind].filter(owed).sort();
+            const awaited = asking.filter((conversation) => catching.has(conversation));
+            await Promise.all([
+                read(asking.filter((conversation) => !catching.has(conversation))),
+                Promise.all(awaited.map((conversation) => catching.get(conversation))).then(() => read(awaited.filter((conversation) => owed(conversation) && !catching.has(conversation)))),
+            ]);
             if (behind.size > 0)
                 progress.complete = false;
         },
@@ -709,6 +829,10 @@ export function createConversationStore(deps) {
                         continue;
                     if (deps.closed())
                         return;
+                    if (deps.expired(submission.operation)) {
+                        await forgotten(conversation, submission);
+                        continue;
+                    }
                     const result = await dispatched(conversation, submission);
                     if (result.kind === 'revoked-authorization')
                         return;
@@ -716,16 +840,29 @@ export function createConversationStore(deps) {
                         progress.reachable = false;
                         return;
                     }
+                    if (result.kind !== 'action' && result.retryable) {
+                        progress.complete = false;
+                        return;
+                    }
                 }
             }
         },
         current() {
             return behind.size === 0;
         },
+        known() {
+            return listKnown;
+        },
         reconnected() {
             for (const conversation of replicated)
                 gap(conversation);
         },
+        reopened() {
+            sending.clear();
+        },
+        lost() {
+            catching.clear();
+        },
         watch(conversation, options, deliver) {
             const limit = limitOf(options);
             const key = JSON.stringify([conversation, limit ?? null]);
@@ -733,18 +870,20 @@ export function createConversationStore(deps) {
                 conversation,
                 limit,
                 watchers: new Set(),
-                messages: null,
+                window: null,
             };
             views.set(key, view);
             const watcher = { notify: deliver, ready: false };
             view.watchers.add(watcher);
             void deps
-                .durably(() => Promise.resolve())
+                .durably(async (local) => {
+                view.window ??= await local.messageWindow(conversation, limit);
+            })
                 .then(() => {
                 if (deps.closed() || !view.watchers.has(watcher))
                     return;
                 watcher.ready = true;
-                watcher.notify(viewOf(conversation, limit));
+                watcher.notify(viewOf(conversation, view.window ?? []));
             })
                 .catch(() => { });
             return () => {
@@ -791,12 +930,15 @@ export function createConversationStore(deps) {
             if (event.kind === 'list' || event.kind === 'listed' || event.kind === 'unlisted') {
                 if (event.kind === 'list') {
                     listing.clear();
-                    for (const entry of event.entries)
+                    for (const entry of event.entries) {
                         listing.set(entry.conversation, { ...entry });
+                        creations.delete(entry.conversation);
+                    }
                     listKnown = true;
                 }
                 else if (event.kind === 'listed') {
                     listing.set(event.entry.conversation, { ...event.entry });
+                    creations.delete(event.entry.conversation);
                 }
                 else {
                     listing.delete(event.conversation);
@@ -810,12 +952,11 @@ export function createConversationStore(deps) {
             heard(event);
             return false;
         },
-        messages() {
-            return [...messages].flatMap(([conversation, held]) => held.map(({ identity, text }) => ({ conversation, identity, text })));
+        searchable() {
+            return [...(listKnown ? listing.keys() : stored)].sort();
         },
-        message(conversation, identity) {
-            const held = messages.get(conversation)?.find((message) => message.identity === identity);
-            return held === undefined ? null : { conversation, identity, text: held.text };
+        acquired(conversation) {
+            return reads.has(conversation) && !partials.has(conversation);
         },
         close() {
             views.clear();