shennian 0.2.89 → 0.2.90

package/dist/src/manager/registry.js

@@ -1,269 +1,2 @@
-// @arch docs/features/manager-agent.md
-// @test src/__tests__/manager-runtime.test.ts
-import fs from 'node:fs';
-import path from 'node:path';
-import { randomUUID } from 'node:crypto';
-import { extractPayloadText, isToolPayload } from '@shennian/wire';
-import { resolveShennianPath } from '../config/index.js';
-import { listSessionRecords, readMessages } from '../session/store.js';
-import { listProjectedSessions } from '../session/projection.js';
-const REGISTRY_PATH = resolveShennianPath('manager-registry.json');
-function nowIso() {
-    return new Date().toISOString();
-}
-function emptyRegistry() {
-    return { managers: {}, workers: {}, replyTargets: {} };
-}
-export class ManagerRegistry {
-    load() {
-        try {
-            const parsed = JSON.parse(fs.readFileSync(REGISTRY_PATH, 'utf-8'));
-            return {
-                managers: parsed.managers ?? {},
-                workers: parsed.workers ?? {},
-                replyTargets: parsed.replyTargets ?? {},
-            };
-        }
-        catch {
-            return emptyRegistry();
-        }
-    }
-    save(registry) {
-        fs.mkdirSync(path.dirname(REGISTRY_PATH), { recursive: true });
-        fs.writeFileSync(REGISTRY_PATH, JSON.stringify(registry, null, 2));
-    }
-    upsertManager(input) {
-        const registry = this.load();
-        const existing = registry.managers[input.sessionId];
-        const timestamp = nowIso();
-        const record = {
-            sessionId: input.sessionId,
-            agentSessionId: input.agentSessionId ?? existing?.agentSessionId ?? null,
-            workDir: input.workDir,
-            machineId: input.machineId ?? existing?.machineId ?? null,
-            modelId: input.modelId,
-            defaultWorkerAgentType: input.defaultWorkerAgentType ?? existing?.defaultWorkerAgentType ?? null,
-            defaultWorkerModelId: input.defaultWorkerModelId ?? existing?.defaultWorkerModelId ?? null,
-            status: input.status ?? existing?.status ?? 'idle',
-            managedWorkerSessionIds: existing?.managedWorkerSessionIds ?? [],
-            attachedExternalChannels: existing?.attachedExternalChannels ?? [],
-            createdAt: existing?.createdAt ?? timestamp,
-            updatedAt: timestamp,
-        };
-        registry.managers[input.sessionId] = record;
-        this.save(registry);
-        return record;
-    }
-    getManager(sessionId) {
-        return this.load().managers[sessionId];
-    }
-    addWorker(input) {
-        const registry = this.load();
-        const manager = registry.managers[input.managerSessionId];
-        if (!manager)
-            throw new Error('Manager runtime is not registered');
-        const sessionId = input.sessionId ?? `sess_worker_${randomUUID()}`;
-        const timestamp = nowIso();
-        const record = {
-            sessionId,
-            agentType: input.agentType,
-            workDir: input.workDir,
-            managedBy: input.managerSessionId,
-            status: 'running',
-            createdAt: timestamp,
-            updatedAt: timestamp,
-            lastActivityAt: timestamp,
-            summary: input.summary ?? null,
-            agentSessionId: null,
-            runId: null,
-        };
-        registry.workers[sessionId] = record;
-        if (!manager.managedWorkerSessionIds.includes(sessionId)) {
-            manager.managedWorkerSessionIds.push(sessionId);
-        }
-        manager.updatedAt = timestamp;
-        this.save(registry);
-        return record;
-    }
-    updateWorker(sessionId, patch) {
-        const registry = this.load();
-        const existing = registry.workers[sessionId];
-        if (!existing)
-            return undefined;
-        const record = {
-            ...existing,
-            ...patch,
-            updatedAt: nowIso(),
-        };
-        registry.workers[sessionId] = record;
-        this.save(registry);
-        return record;
-    }
-    listWorkers(managerSessionId, opts = {}) {
-        const registry = this.load();
-        const manager = registry.managers[managerSessionId];
-        if (!manager)
-            return [];
-        const managedWorkers = Object.values(registry.workers)
-            .filter((worker) => worker.workDir === manager.workDir)
-            .filter((worker) => opts.includeManagers || worker.agentType !== 'manager')
-            .map((worker) => this.decorateManagedWorker(worker, opts.runningSessionIds));
-        const managedIds = new Set(managedWorkers.map((worker) => worker.sessionId));
-        const projectSessions = listSessionRecords()
-            .filter((session) => session.workDir === manager.workDir)
-            .filter((session) => opts.includeManagers || session.agentType !== 'manager')
-            .filter((session) => !managedIds.has(session.sessionId))
-            .map((session) => this.sessionRecordToWorker(session, opts.runningSessionIds));
-        const knownIds = new Set([...managedWorkers, ...projectSessions].map((session) => session.sessionId));
-        const projectedSource = opts.projectedSessions ?? listProjectedSessions();
-        const projectedSessions = projectedSource
-            .filter((session) => session.workDir === manager.workDir)
-            .filter((session) => opts.includeManagers || session.agentType !== 'manager')
-            .filter((session) => !knownIds.has(session.id))
-            .map((session) => this.projectedSessionToWorker(session, opts.runningSessionIds));
-        return [...managedWorkers, ...projectSessions, ...projectedSessions]
-            .sort((a, b) => b.lastActivityAt.localeCompare(a.lastActivityAt));
-    }
-    getWorkerForManager(managerSessionId, sessionId) {
-        const registry = this.load();
-        const manager = registry.managers[managerSessionId];
-        const worker = registry.workers[sessionId];
-        if (!manager)
-            return undefined;
-        if (worker) {
-            if (worker.workDir !== manager.workDir)
-                return undefined;
-            if (worker.agentType === 'manager')
-                return undefined;
-            return this.decorateManagedWorker(worker);
-        }
-        const session = listSessionRecords().find((item) => item.sessionId === sessionId);
-        if (!session || session.workDir !== manager.workDir || session.agentType === 'manager')
-            return undefined;
-        return this.sessionRecordToWorker(session);
-    }
-    decorateManagedWorker(worker, runningSessionIds) {
-        const status = runningSessionIds?.has(worker.sessionId) ? 'running' : worker.status;
-        const transcript = this.readTranscriptSummary(worker.sessionId, worker.summary ?? null);
-        const lastMessagePreview = transcript.lastMessagePreview ?? worker.lastMessagePreview ?? worker.summary ?? null;
-        const userGoal = worker.userGoal ?? transcript.userGoal ?? null;
-        return {
-            ...worker,
-            managedBy: worker.managedBy ?? null,
-            status,
-            title: worker.title ?? userGoal ?? lastMessagePreview ?? worker.summary ?? worker.sessionId,
-            userGoal,
-            lastMessagePreview,
-            summary: worker.summary ?? lastMessagePreview,
-            canResume: worker.canResume ?? worker.status !== 'aborted',
-            source: worker.source ?? 'manager_registry',
-            role: 'managed_worker',
-        };
-    }
-    sessionRecordToWorker(session, runningSessionIds) {
-        const transcript = this.readTranscriptSummary(session.sessionId, session.lastMessagePreview ?? null);
-        const lastMessagePreview = transcript.lastMessagePreview ?? session.lastMessagePreview ?? null;
-        const userGoal = transcript.userGoal;
-        const status = runningSessionIds?.has(session.sessionId)
-            ? 'running'
-            : session.status === 'failed'
-                ? 'error'
-                : session.status === 'completed'
-                    ? 'final'
-                    : 'idle';
-        return {
-            sessionId: session.sessionId,
-            agentType: session.agentType,
-            workDir: session.workDir,
-            managedBy: null,
-            status,
-            createdAt: session.createdAt,
-            updatedAt: session.updatedAt,
-            lastActivityAt: session.lastActivityAt,
-            title: userGoal ?? lastMessagePreview ?? session.sessionId,
-            userGoal,
-            lastMessagePreview,
-            summary: lastMessagePreview ?? userGoal ?? null,
-            agentSessionId: session.agentSessionId ?? null,
-            runId: null,
-            canResume: Boolean(session.agentSessionId),
-            source: 'local_index',
-            role: 'project_session',
-        };
-    }
-    projectedSessionToWorker(session, runningSessionIds) {
-        const lastMessagePreview = session.lastMessagePreview ?? null;
-        const userGoal = session.title?.trim() || lastMessagePreview;
-        const status = runningSessionIds?.has(session.id)
-            ? 'running'
-            : session.status === 'failed'
-                ? 'error'
-                : session.status === 'completed'
-                    ? 'final'
-                    : 'idle';
-        return {
-            sessionId: session.id,
-            agentType: session.agentType,
-            workDir: session.workDir,
-            managedBy: null,
-            status,
-            createdAt: session.createdAt,
-            updatedAt: session.updatedAt,
-            lastActivityAt: session.lastActivityAt || session.updatedAt,
-            title: userGoal ?? session.id,
-            userGoal: userGoal ?? null,
-            lastMessagePreview,
-            summary: lastMessagePreview ?? userGoal ?? null,
-            agentSessionId: session.agentSessionId ?? null,
-            runId: session.activity?.runId ?? null,
-            canResume: true,
-            source: 'app_projection',
-            role: 'project_session',
-        };
-    }
-    readTranscriptSummary(sessionId, fallbackPreview) {
-        const messages = readMessages(sessionId, { limit: 200 });
-        if (!messages.length)
-            return { userGoal: null, lastMessagePreview: fallbackPreview };
-        const chronological = [...messages].sort((a, b) => a.ts - b.ts);
-        const firstUser = chronological.find((message) => message.role === 'user');
-        const lastReadable = [...chronological].reverse().find((message) => readableText(message));
-        return {
-            userGoal: firstUser ? clip(readableText(firstUser), 160) : null,
-            lastMessagePreview: clip(readableText(lastReadable), 180) ?? fallbackPreview,
-        };
-    }
-    createReplyTarget(input) {
-        const registry = this.load();
-        const record = {
-            replyTarget: `rt_${randomUUID()}`,
-            managerSessionId: input.managerSessionId,
-            channelId: input.channelId,
-            conversationId: input.conversationId,
-            messageId: input.messageId ?? null,
-            createdAt: nowIso(),
-        };
-        registry.replyTargets[record.replyTarget] = record;
-        this.save(registry);
-        return record;
-    }
-    getReplyTarget(replyTarget) {
-        return this.load().replyTargets[replyTarget];
-    }
-    getLatestReplyTargetForManager(managerSessionId) {
-        return Object.values(this.load().replyTargets)
-            .filter((target) => target.managerSessionId === managerSessionId)
-            .sort((a, b) => b.createdAt.localeCompare(a.createdAt))[0];
-    }
-}
-function readableText(message) {
-    if (!message || isToolPayload(message.payload))
-        return null;
-    const text = extractPayloadText(message.payload).replace(/\r\n/g, '\n').trim();
-    return text || null;
-}
-function clip(text, max) {
-    if (!text)
-        return null;
-    return text.length > max ? `${text.slice(0, max)}...` : text;
-}
+import u from"node:fs";import h from"node:path";import{randomUUID as y}from"node:crypto";import{extractPayloadText as k,isToolPayload as T}from"@shennian/wire";import{resolveShennianPath as A}from"../config/index.js";import{listSessionRecords as I,readMessages as w}from"../session/store.js";import{listProjectedSessions as v}from"../session/projection.js";const g=A("manager-registry.json");function l(){return new Date().toISOString()}function M(){return{managers:{},workers:{},replyTargets:{}}}class B{load(){try{const e=JSON.parse(u.readFileSync(g,"utf-8"));return{managers:e.managers??{},workers:e.workers??{},replyTargets:e.replyTargets??{}}}catch{return M()}}save(e){u.mkdirSync(h.dirname(g),{recursive:!0}),u.writeFileSync(g,JSON.stringify(e,null,2))}upsertManager(e){const t=this.load(),r=t.managers[e.sessionId],a=l(),s={sessionId:e.sessionId,agentSessionId:e.agentSessionId??r?.agentSessionId??null,workDir:e.workDir,machineId:e.machineId??r?.machineId??null,modelId:e.modelId,defaultWorkerAgentType:e.defaultWorkerAgentType??r?.defaultWorkerAgentType??null,defaultWorkerModelId:e.defaultWorkerModelId??r?.defaultWorkerModelId??null,status:e.status??r?.status??"idle",managedWorkerSessionIds:r?.managedWorkerSessionIds??[],attachedExternalChannels:r?.attachedExternalChannels??[],createdAt:r?.createdAt??a,updatedAt:a};return t.managers[e.sessionId]=s,this.save(t),s}getManager(e){return this.load().managers[e]}addWorker(e){const t=this.load(),r=t.managers[e.managerSessionId];if(!r)throw new Error("Manager runtime is not registered");const a=e.sessionId??`sess_worker_${y()}`,s=l(),o={sessionId:a,agentType:e.agentType,workDir:e.workDir,managedBy:e.managerSessionId,status:"running",createdAt:s,updatedAt:s,lastActivityAt:s,summary:e.summary??null,agentSessionId:null,runId:null};return t.workers[a]=o,r.managedWorkerSessionIds.includes(a)||r.managedWorkerSessionIds.push(a),r.updatedAt=s,this.save(t),o}updateWorker(e,t){const r=this.load(),a=r.workers[e];if(!a)return;const s={...a,...t,updatedAt:l()};return r.workers[e]=s,this.save(r),s}listWorkers(e,t={}){const r=this.load(),a=r.managers[e];if(!a)return[];const s=Object.values(r.workers).filter(n=>n.workDir===a.workDir).filter(n=>t.includeManagers||n.agentType!=="manager").map(n=>this.decorateManagedWorker(n,t.runningSessionIds)),o=new Set(s.map(n=>n.sessionId)),d=I().filter(n=>n.workDir===a.workDir).filter(n=>t.includeManagers||n.agentType!=="manager").filter(n=>!o.has(n.sessionId)).map(n=>this.sessionRecordToWorker(n,t.runningSessionIds)),c=new Set([...s,...d].map(n=>n.sessionId)),f=(t.projectedSessions??v()).filter(n=>n.workDir===a.workDir).filter(n=>t.includeManagers||n.agentType!=="manager").filter(n=>!c.has(n.id)).map(n=>this.projectedSessionToWorker(n,t.runningSessionIds));return[...s,...d,...f].sort((n,S)=>S.lastActivityAt.localeCompare(n.lastActivityAt))}getWorkerForManager(e,t){const r=this.load(),a=r.managers[e],s=r.workers[t];if(!a)return;if(s)return s.workDir!==a.workDir||s.agentType==="manager"?void 0:this.decorateManagedWorker(s);const o=I().find(d=>d.sessionId===t);if(!(!o||o.workDir!==a.workDir||o.agentType==="manager"))return this.sessionRecordToWorker(o)}decorateManagedWorker(e,t){const r=t?.has(e.sessionId)?"running":e.status,a=this.readTranscriptSummary(e.sessionId,e.summary??null),s=a.lastMessagePreview??e.lastMessagePreview??e.summary??null,o=e.userGoal??a.userGoal??null;return{...e,managedBy:e.managedBy??null,status:r,title:e.title??o??s??e.summary??e.sessionId,userGoal:o,lastMessagePreview:s,summary:e.summary??s,canResume:e.canResume??e.status!=="aborted",source:e.source??"manager_registry",role:"managed_worker"}}sessionRecordToWorker(e,t){const r=this.readTranscriptSummary(e.sessionId,e.lastMessagePreview??null),a=r.lastMessagePreview??e.lastMessagePreview??null,s=r.userGoal,o=t?.has(e.sessionId)?"running":e.status==="failed"?"error":e.status==="completed"?"final":"idle";return{sessionId:e.sessionId,agentType:e.agentType,workDir:e.workDir,managedBy:null,status:o,createdAt:e.createdAt,updatedAt:e.updatedAt,lastActivityAt:e.lastActivityAt,title:s??a??e.sessionId,userGoal:s,lastMessagePreview:a,summary:a??s??null,agentSessionId:e.agentSessionId??null,runId:null,canResume:!!e.agentSessionId,source:"local_index",role:"project_session"}}projectedSessionToWorker(e,t){const r=e.lastMessagePreview??null,a=e.title?.trim()||r,s=t?.has(e.id)?"running":e.status==="failed"?"error":e.status==="completed"?"final":"idle";return{sessionId:e.id,agentType:e.agentType,workDir:e.workDir,managedBy:null,status:s,createdAt:e.createdAt,updatedAt:e.updatedAt,lastActivityAt:e.lastActivityAt||e.updatedAt,title:a??e.id,userGoal:a??null,lastMessagePreview:r,summary:r??a??null,agentSessionId:e.agentSessionId??null,runId:e.activity?.runId??null,canResume:!0,source:"app_projection",role:"project_session"}}readTranscriptSummary(e,t){const r=w(e,{limit:200});if(!r.length)return{userGoal:null,lastMessagePreview:t};const a=[...r].sort((d,c)=>d.ts-c.ts),s=a.find(d=>d.role==="user"),o=[...a].reverse().find(d=>m(d));return{userGoal:s?p(m(s),160):null,lastMessagePreview:p(m(o),180)??t}}createReplyTarget(e){const t=this.load(),r={replyTarget:`rt_${y()}`,managerSessionId:e.managerSessionId,channelId:e.channelId,conversationId:e.conversationId,messageId:e.messageId??null,createdAt:l()};return t.replyTargets[r.replyTarget]=r,this.save(t),r}getReplyTarget(e){return this.load().replyTargets[e]}getLatestReplyTargetForManager(e){return Object.values(this.load().replyTargets).filter(t=>t.managerSessionId===e).sort((t,r)=>r.createdAt.localeCompare(t.createdAt))[0]}}function m(i){return!i||T(i.payload)?null:k(i.payload).replace(/\r\n/g,`
+`).trim()||null}function p(i,e){return i?i.length>e?`${i.slice(0,e)}...`:i:null}export{B as ManagerRegistry};