@agentuity/runtime 0.0.51 → 0.0.53

package/dist/services/session/json.js

@@ -0,0 +1,35 @@
+import { join } from 'node:path';
+import { randomUUID } from 'node:crypto';
+/**
+ * An implementation of the SessionEventProvider which uses JSON logs for delivery
+ */
+export class JSONSessionEventProvider {
+    directory;
+    constructor(directory) {
+        this.directory = directory;
+    }
+    makeFilename(type) {
+        return join(this.directory, `session-${type}.${Date.now()}${randomUUID()}.json`);
+    }
+    /**
+     * called when the session starts
+     *
+     * @param event SessionStartEvent
+     */
+    async start(event) {
+        const filename = this.makeFilename('start');
+        const payload = JSON.stringify({ ...event, timestamp: new Date() }) + '\n';
+        await Bun.file(filename).write(payload);
+    }
+    /**
+     * called when the session completes
+     *
+     * @param event SessionCompleteEvent
+     */
+    async complete(event) {
+        const filename = this.makeFilename('complete');
+        const payload = JSON.stringify({ ...event, timestamp: new Date() }) + '\n';
+        await Bun.file(filename).write(payload);
+    }
+}
+//# sourceMappingURL=json.js.map

package/dist/services/session/json.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"json.js","sourceRoot":"","sources":["../../../src/services/session/json.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAOzC;;GAEG;AACH,MAAM,OAAO,wBAAwB;IAC5B,SAAS,CAAS;IAE1B,YAAY,SAAiB;QAC5B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC5B,CAAC;IACO,YAAY,CAAC,IAA0B;QAC9C,OAAO,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,WAAW,IAAI,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,UAAU,EAAE,OAAO,CAAC,CAAC;IAClF,CAAC;IACD;;;;OAIG;IACH,KAAK,CAAC,KAAK,CAAC,KAAwB;QACnC,MAAM,QAAQ,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAC5C,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,GAAG,KAAK,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,EAAE,CAAC,GAAG,IAAI,CAAC;QAC3E,MAAM,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IACzC,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,QAAQ,CAAC,KAA2B;QACzC,MAAM,QAAQ,GAAG,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC;QAC/C,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,GAAG,KAAK,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,EAAE,CAAC,GAAG,IAAI,CAAC;QAC3E,MAAM,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IACzC,CAAC;CACD"}

package/dist/services/session/local.js

@@ -0,0 +1,22 @@
+/**
+ * An implementation of the SessionEventProvider which is no-op
+ */
+export class LocalSessionEventProvider {
+    /**
+     * called when the session starts
+     *
+     * @param event SessionStartEvent
+     */
+    async start(_event) {
+        // no op
+    }
+    /**
+     * called when the session completes
+     *
+     * @param event SessionCompleteEvent
+     */
+    async complete(_event) {
+        // no op
+    }
+}
+//# sourceMappingURL=local.js.map

package/dist/services/session/local.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"local.js","sourceRoot":"","sources":["../../../src/services/session/local.ts"],"names":[],"mappings":"AAMA;;GAEG;AACH,MAAM,OAAO,yBAAyB;IACrC;;;;OAIG;IACH,KAAK,CAAC,KAAK,CAAC,MAAyB;QACpC,QAAQ;IACT,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,QAAQ,CAAC,MAA4B;QAC1C,QAAQ;IACT,CAAC;CACD"}

package/dist/session.js

@@ -0,0 +1,199 @@
+import { getCookie, setCookie } from 'hono/cookie';
+import { fireEvent } from './app';
+// WeakMap to store event listeners for Thread and Session instances
+const threadEventListeners = new WeakMap();
+const sessionEventListeners = new WeakMap();
+// Helper to fire thread event listeners
+async function fireThreadEvent(thread, eventName) {
+    const listeners = threadEventListeners.get(thread);
+    if (!listeners)
+        return;
+    const callbacks = listeners.get(eventName);
+    if (!callbacks || callbacks.size === 0)
+        return;
+    for (const callback of callbacks) {
+        await callback(eventName, thread);
+    }
+}
+// Helper to fire session event listeners
+async function fireSessionEvent(session, eventName) {
+    const listeners = sessionEventListeners.get(session);
+    if (!listeners)
+        return;
+    const callbacks = listeners.get(eventName);
+    if (!callbacks || callbacks.size === 0)
+        return;
+    for (const callback of callbacks) {
+        await callback(eventName, session);
+    }
+}
+// Generate thread or session ID
+export function generateId(prefix) {
+    const arr = new Uint8Array(16);
+    crypto.getRandomValues(arr);
+    return `${prefix}${prefix ? '_' : ''}${arr.toHex()}`;
+}
+export class DefaultThread {
+    #lastUsed;
+    id;
+    state;
+    provider;
+    constructor(provider, id) {
+        this.provider = provider;
+        this.id = id;
+        this.state = new Map();
+        this.#lastUsed = Date.now();
+    }
+    addEventListener(eventName, callback) {
+        let listeners = threadEventListeners.get(this);
+        if (!listeners) {
+            listeners = new Map();
+            threadEventListeners.set(this, listeners);
+        }
+        let callbacks = listeners.get(eventName);
+        if (!callbacks) {
+            callbacks = new Set();
+            listeners.set(eventName, callbacks);
+        }
+        callbacks.add(callback);
+    }
+    removeEventListener(eventName, callback) {
+        const listeners = threadEventListeners.get(this);
+        if (!listeners)
+            return;
+        const callbacks = listeners.get(eventName);
+        if (!callbacks)
+            return;
+        callbacks.delete(callback);
+    }
+    async fireEvent(eventName) {
+        await fireThreadEvent(this, eventName);
+    }
+    async destroy() {
+        this.provider.destroy(this);
+    }
+    touch() {
+        this.#lastUsed = Date.now();
+    }
+    expired() {
+        return Date.now() - this.#lastUsed >= 3.6e6; // 1 hour
+    }
+}
+export class DefaultSession {
+    id;
+    thread;
+    state;
+    constructor(thread, id) {
+        this.id = id;
+        this.thread = thread;
+        this.state = new Map();
+    }
+    addEventListener(eventName, callback) {
+        let listeners = sessionEventListeners.get(this);
+        if (!listeners) {
+            listeners = new Map();
+            sessionEventListeners.set(this, listeners);
+        }
+        let callbacks = listeners.get(eventName);
+        if (!callbacks) {
+            callbacks = new Set();
+            listeners.set(eventName, callbacks);
+        }
+        callbacks.add(callback);
+    }
+    removeEventListener(eventName, callback) {
+        const listeners = sessionEventListeners.get(this);
+        if (!listeners)
+            return;
+        const callbacks = listeners.get(eventName);
+        if (!callbacks)
+            return;
+        callbacks.delete(callback);
+    }
+    async fireEvent(eventName) {
+        await fireSessionEvent(this, eventName);
+    }
+}
+export class DefaultThreadProvider {
+    threads = new Map();
+    async initialize() {
+        setInterval(() => {
+            for (const [, thread] of this.threads) {
+                if (thread.expired()) {
+                    void (async () => {
+                        try {
+                            await this.destroy(thread);
+                        }
+                        catch (err) {
+                            console.error('Failed to destroy expired thread', err);
+                        }
+                    })();
+                }
+            }
+        }, 60_000).unref();
+    }
+    async restore(ctx) {
+        const cookie = getCookie(ctx);
+        let threadId;
+        if (cookie.atid?.startsWith('thrd_')) {
+            threadId = cookie.atid;
+        }
+        threadId = threadId || generateId('thrd');
+        if (threadId) {
+            setCookie(ctx, 'atid', threadId);
+            const existing = this.threads.get(threadId);
+            if (existing) {
+                return existing;
+            }
+        }
+        const thread = new DefaultThread(this, threadId);
+        this.threads.set(thread.id, thread);
+        await fireEvent('thread.created', thread);
+        return thread;
+    }
+    async save(thread) {
+        if (thread instanceof DefaultThread) {
+            thread.touch();
+        }
+    }
+    async destroy(thread) {
+        if (thread instanceof DefaultThread) {
+            try {
+                await thread.fireEvent('destroyed');
+                await fireEvent('thread.destroyed', thread);
+            }
+            finally {
+                this.threads.delete(thread.id);
+                threadEventListeners.delete(thread);
+            }
+        }
+    }
+}
+export class DefaultSessionProvider {
+    sessions = new Map();
+    async initialize() {
+        // No initialization needed for in-memory provider
+    }
+    async restore(thread, sessionId) {
+        let session = this.sessions.get(sessionId);
+        if (!session) {
+            session = new DefaultSession(thread, sessionId);
+            this.sessions.set(sessionId, session);
+            await fireEvent('session.started', session);
+        }
+        return session;
+    }
+    async save(session) {
+        if (session instanceof DefaultSession) {
+            try {
+                await session.fireEvent('completed');
+                await fireEvent('session.completed', session);
+            }
+            finally {
+                this.sessions.delete(session.id);
+                sessionEventListeners.delete(session);
+            }
+        }
+    }
+}
+//# sourceMappingURL=session.js.map

package/dist/session.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"session.js","sourceRoot":"","sources":["../src/session.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACnD,OAAO,EAAY,SAAS,EAAE,MAAM,OAAO,CAAC;AAwD5C,oEAAoE;AACpE,MAAM,oBAAoB,GAAG,IAAI,OAAO,EAGrC,CAAC;AACJ,MAAM,qBAAqB,GAAG,IAAI,OAAO,EAGtC,CAAC;AAEJ,wCAAwC;AACxC,KAAK,UAAU,eAAe,CAAC,MAAc,EAAE,SAA0B;IACxE,MAAM,SAAS,GAAG,oBAAoB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IACnD,IAAI,CAAC,SAAS;QAAE,OAAO;IAEvB,MAAM,SAAS,GAAG,SAAS,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IAC3C,IAAI,CAAC,SAAS,IAAI,SAAS,CAAC,IAAI,KAAK,CAAC;QAAE,OAAO;IAE/C,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;QAClC,MAAO,QAAgB,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;IAC5C,CAAC;AACF,CAAC;AAED,yCAAyC;AACzC,KAAK,UAAU,gBAAgB,CAAC,OAAgB,EAAE,SAA2B;IAC5E,MAAM,SAAS,GAAG,qBAAqB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IACrD,IAAI,CAAC,SAAS;QAAE,OAAO;IAEvB,MAAM,SAAS,GAAG,SAAS,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IAC3C,IAAI,CAAC,SAAS,IAAI,SAAS,CAAC,IAAI,KAAK,CAAC;QAAE,OAAO;IAE/C,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;QAClC,MAAO,QAAgB,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IAC7C,CAAC;AACF,CAAC;AAED,gCAAgC;AAChC,MAAM,UAAU,UAAU,CAAC,MAAe;IACzC,MAAM,GAAG,GAAG,IAAI,UAAU,CAAC,EAAE,CAAC,CAAC;IAC/B,MAAM,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;IAC5B,OAAO,GAAG,MAAM,GAAG,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,GAAG,CAAC,KAAK,EAAE,EAAE,CAAC;AACtD,CAAC;AAED,MAAM,OAAO,aAAa;IACzB,SAAS,CAAS;IACT,EAAE,CAAS;IACX,KAAK,CAAuB;IAC7B,QAAQ,CAAiB;IAEjC,YAAY,QAAwB,EAAE,EAAU;QAC/C,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,KAAK,GAAG,IAAI,GAAG,EAAE,CAAC;QACvB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC7B,CAAC;IAED,gBAAgB,CAAC,SAA0B,EAAE,QAAkC;QAC9E,IAAI,SAAS,GAAG,oBAAoB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAC/C,IAAI,CAAC,SAAS,EAAE,CAAC;YAChB,SAAS,GAAG,IAAI,GAAG,EAAE,CAAC;YACtB,oBAAoB,CAAC,GAAG,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;QAC3C,CAAC;QACD,IAAI,SAAS,GAAG,SAAS,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACzC,IAAI,CAAC,SAAS,EAAE,CAAC;YAChB,SAAS,GAAG,IAAI,GAAG,EAAE,CAAC;YACtB,SAAS,CAAC,GAAG,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;QACrC,CAAC;QACD,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IACzB,CAAC;IAED,mBAAmB,CAAC,SAA0B,EAAE,QAAkC;QACjF,MAAM,SAAS,GAAG,oBAAoB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACjD,IAAI,CAAC,SAAS;YAAE,OAAO;QACvB,MAAM,SAAS,GAAG,SAAS,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC3C,IAAI,CAAC,SAAS;YAAE,OAAO;QACvB,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IAC5B,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,SAA0B;QACzC,MAAM,eAAe,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;IACxC,CAAC;IAED,KAAK,CAAC,OAAO;QACZ,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IAC7B,CAAC;IAED,KAAK;QACJ,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC7B,CAAC;IAED,OAAO;QACN,OAAO,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,SAAS,IAAI,KAAK,CAAC,CAAC,SAAS;IACvD,CAAC;CACD;AAED,MAAM,OAAO,cAAc;IACjB,EAAE,CAAS;IACX,MAAM,CAAS;IACf,KAAK,CAAuB;IAErC,YAAY,MAAc,EAAE,EAAU;QACrC,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,KAAK,GAAG,IAAI,GAAG,EAAE,CAAC;IACxB,CAAC;IAED,gBAAgB,CAAC,SAA2B,EAAE,QAAmC;QAChF,IAAI,SAAS,GAAG,qBAAqB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAChD,IAAI,CAAC,SAAS,EAAE,CAAC;YAChB,SAAS,GAAG,IAAI,GAAG,EAAE,CAAC;YACtB,qBAAqB,CAAC,GAAG,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;QAC5C,CAAC;QACD,IAAI,SAAS,GAAG,SAAS,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACzC,IAAI,CAAC,SAAS,EAAE,CAAC;YAChB,SAAS,GAAG,IAAI,GAAG,EAAE,CAAC;YACtB,SAAS,CAAC,GAAG,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;QACrC,CAAC;QACD,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IACzB,CAAC;IAED,mBAAmB,CAAC,SAA2B,EAAE,QAAmC;QACnF,MAAM,SAAS,GAAG,qBAAqB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAClD,IAAI,CAAC,SAAS;YAAE,OAAO;QACvB,MAAM,SAAS,GAAG,SAAS,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC3C,IAAI,CAAC,SAAS;YAAE,OAAO;QACvB,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IAC5B,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,SAA2B;QAC1C,MAAM,gBAAgB,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;IACzC,CAAC;CACD;AAED,MAAM,OAAO,qBAAqB;IACzB,OAAO,GAAG,IAAI,GAAG,EAAyB,CAAC;IAEnD,KAAK,CAAC,UAAU;QACf,WAAW,CAAC,GAAG,EAAE;YAChB,KAAK,MAAM,CAAC,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;gBACvC,IAAI,MAAM,CAAC,OAAO,EAAE,EAAE,CAAC;oBACtB,KAAK,CAAC,KAAK,IAAI,EAAE;wBAChB,IAAI,CAAC;4BACJ,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;wBAC5B,CAAC;wBAAC,OAAO,GAAG,EAAE,CAAC;4BACd,OAAO,CAAC,KAAK,CAAC,kCAAkC,EAAE,GAAG,CAAC,CAAC;wBACxD,CAAC;oBACF,CAAC,CAAC,EAAE,CAAC;gBACN,CAAC;YACF,CAAC;QACF,CAAC,EAAE,MAAM,CAAC,CAAC,KAAK,EAAE,CAAC;IACpB,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,GAAiB;QAC9B,MAAM,MAAM,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC;QAC9B,IAAI,QAA4B,CAAC;QAEjC,IAAI,MAAM,CAAC,IAAI,EAAE,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;YACtC,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC;QACxB,CAAC;QAED,QAAQ,GAAG,QAAQ,IAAI,UAAU,CAAC,MAAM,CAAC,CAAC;QAE1C,IAAI,QAAQ,EAAE,CAAC;YACd,SAAS,CAAC,GAAG,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;YACjC,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAC5C,IAAI,QAAQ,EAAE,CAAC;gBACd,OAAO,QAAQ,CAAC;YACjB,CAAC;QACF,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,aAAa,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;QACjD,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;QACpC,MAAM,SAAS,CAAC,gBAAgB,EAAE,MAAM,CAAC,CAAC;QAC1C,OAAO,MAAM,CAAC;IACf,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,MAAc;QACxB,IAAI,MAAM,YAAY,aAAa,EAAE,CAAC;YACrC,MAAM,CAAC,KAAK,EAAE,CAAC;QAChB,CAAC;IACF,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAc;QAC3B,IAAI,MAAM,YAAY,aAAa,EAAE,CAAC;YACrC,IAAI,CAAC;gBACJ,MAAM,MAAM,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;gBACpC,MAAM,SAAS,CAAC,kBAAkB,EAAE,MAAM,CAAC,CAAC;YAC7C,CAAC;oBAAS,CAAC;gBACV,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBAC/B,oBAAoB,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YACrC,CAAC;QACF,CAAC;IACF,CAAC;CACD;AAED,MAAM,OAAO,sBAAsB;IAC1B,QAAQ,GAAG,IAAI,GAAG,EAA0B,CAAC;IAErD,KAAK,CAAC,UAAU;QACf,kDAAkD;IACnD,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAc,EAAE,SAAiB;QAC9C,IAAI,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC3C,IAAI,CAAC,OAAO,EAAE,CAAC;YACd,OAAO,GAAG,IAAI,cAAc,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;YAChD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;YACtC,MAAM,SAAS,CAAC,iBAAiB,EAAE,OAAO,CAAC,CAAC;QAC7C,CAAC;QACD,OAAO,OAAO,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,OAAgB;QAC1B,IAAI,OAAO,YAAY,cAAc,EAAE,CAAC;YACvC,IAAI,CAAC;gBACJ,MAAM,OAAO,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;gBACrC,MAAM,SAAS,CAAC,mBAAmB,EAAE,OAAO,CAAC,CAAC;YAC/C,CAAC;oBAAS,CAAC;gBACV,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBACjC,qBAAqB,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YACvC,CAAC;QACF,CAAC;IACF,CAAC;CACD"}

package/dist/workbench.js

@@ -0,0 +1,38 @@
+import { z } from 'zod';
+import { timingSafeEqual } from 'node:crypto';
+import { getAgents } from './agent';
+export const createWorkbenchMetadataRoute = () => {
+    const authHeader = process.env.AGENTUITY_WORKBENCH_APIKEY
+        ? `Bearer ${process.env.AGENTUITY_WORKBENCH_APIKEY}`
+        : undefined;
+    const agents = getAgents();
+    return async (ctx) => {
+        if (authHeader) {
+            try {
+                const authValue = ctx.req.header('Authorization');
+                if (!authValue ||
+                    !timingSafeEqual(Buffer.from(authValue, 'utf-8'), Buffer.from(authHeader, 'utf-8'))) {
+                    return ctx.text('Unauthorized', { status: 401 });
+                }
+            }
+            catch {
+                // timing safe equals will throw if the input/output lengths are mismatched
+                // so we treat all exceptions as invalid
+                return ctx.text('Unauthorized', { status: 401 });
+            }
+        }
+        const schemas = { agents: {} };
+        // TODO: this is going to only work for zod schema for now. need a way to handle others
+        for (const [, agent] of agents) {
+            schemas.agents[agent.metadata.name] = {
+                schema: {
+                    input: agent.inputSchema ? z.toJSONSchema(agent.inputSchema) : undefined,
+                    output: agent.outputSchema ? z.toJSONSchema(agent.outputSchema) : undefined,
+                },
+                metadata: agent.metadata,
+            };
+        }
+        return ctx.json(schemas);
+    };
+};
+//# sourceMappingURL=workbench.js.map

package/dist/workbench.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"workbench.js","sourceRoot":"","sources":["../src/workbench.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAC9C,OAAO,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AAEpC,MAAM,CAAC,MAAM,4BAA4B,GAAG,GAAY,EAAE;IACzD,MAAM,UAAU,GAAG,OAAO,CAAC,GAAG,CAAC,0BAA0B;QACxD,CAAC,CAAC,UAAU,OAAO,CAAC,GAAG,CAAC,0BAA0B,EAAE;QACpD,CAAC,CAAC,SAAS,CAAC;IACb,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;IAC3B,OAAO,KAAK,EAAE,GAAG,EAAE,EAAE;QACpB,IAAI,UAAU,EAAE,CAAC;YAChB,IAAI,CAAC;gBACJ,MAAM,SAAS,GAAG,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC;gBAClD,IACC,CAAC,SAAS;oBACV,CAAC,eAAe,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,OAAO,CAAC,EAAE,MAAM,CAAC,IAAI,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC,EAClF,CAAC;oBACF,OAAO,GAAG,CAAC,IAAI,CAAC,cAAc,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC;gBAClD,CAAC;YACF,CAAC;YAAC,MAAM,CAAC;gBACR,2EAA2E;gBAC3E,wCAAwC;gBACxC,OAAO,GAAG,CAAC,IAAI,CAAC,cAAc,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC;YAClD,CAAC;QACF,CAAC;QACD,MAAM,OAAO,GAAwC,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;QACpE,uFAAuF;QACvF,KAAK,MAAM,CAAC,EAAE,KAAK,CAAC,IAAI,MAAM,EAAE,CAAC;YAChC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG;gBACrC,MAAM,EAAE;oBACP,KAAK,EAAE,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,SAAS;oBACxE,MAAM,EAAE,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,SAAS;iBAC3E;gBACD,QAAQ,EAAE,KAAK,CAAC,QAAQ;aACxB,CAAC;QACH,CAAC;QACD,OAAO,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC1B,CAAC,CAAC;AACH,CAAC,CAAC"}

package/package.json

@@ -1,14 +1,14 @@
 {
 	"name": "@agentuity/runtime",
-	"version": "0.0.51",
+	"version": "0.0.53",
 	"license": "Apache-2.0",
 	"author": "Agentuity employees and contributors",
 	"type": "module",
-	"main": "./src/index.ts",
+	"main": "./dist/index.js",
 	"types": "./dist/index.d.ts",
 	"exports": {
 		".": {
-			"import": "./src/index.ts",
+			"import": "./dist/index.js",
 			"types": "./dist/index.d.ts"
 		}
 	},
@@ -25,8 +25,8 @@
 		"prepublishOnly": "bun run clean && bun run build"
 	},
 	"dependencies": {
-		"@agentuity/core": "0.0.51",
-		"@agentuity/server": "0.0.51",
+		"@agentuity/core": "0.0.53",
+		"@agentuity/server": "0.0.53",
 		"@opentelemetry/api": "^1.9.0",
 		"@opentelemetry/api-logs": "^0.207.0",
 		"@opentelemetry/auto-instrumentations-node": "^0.66.0",

package/src/_context.ts

@@ -2,17 +2,23 @@ import { AsyncLocalStorage } from 'node:async_hooks';
 import type { Context as HonoContext } from 'hono';
 import type { Tracer } from '@opentelemetry/api';
 import type { KeyValueStorage, ObjectStorage, StreamStorage, VectorStorage } from '@agentuity/core';
-import type { AgentContext, AgentName } from './agent';
+import type { AgentContext, AgentName, AgentRegistry, AgentRunner } from './agent';
 import type { Logger } from './logger';
 import type WaitUntilHandler from './_waituntil';
 import { registerServices } from './_services';
 import type { Thread, Session } from './session';
 
-export interface RequestAgentContextArgs<TAgentMap, TAgent> {
+export interface RequestAgentContextArgs<
+	TAgentMap extends AgentRegistry = AgentRegistry,
+	// eslint-disable-next-line @typescript-eslint/no-explicit-any
+	TCurrent extends AgentRunner<any, any, any> | undefined = AgentRunner<any, any, any> | undefined,
+	// eslint-disable-next-line @typescript-eslint/no-explicit-any
+	TParent extends AgentRunner<any, any, any> | undefined = AgentRunner<any, any, any> | undefined,
+> {
 	sessionId: string;
 	agent: TAgentMap;
-	current: TAgent;
-	parent?: TAgent;
+	current: TCurrent;
+	parent: TParent;
 	agentName: AgentName;
 	logger: Logger;
 	tracer: Tracer;
@@ -21,10 +27,17 @@ export interface RequestAgentContextArgs<TAgentMap, TAgent> {
 	handler: WaitUntilHandler;
 }
 
-export class RequestAgentContext<TAgentMap, TAgent> implements AgentContext {
+export class RequestAgentContext<
+	TAgentMap extends AgentRegistry = AgentRegistry,
+	// eslint-disable-next-line @typescript-eslint/no-explicit-any
+	TCurrent extends AgentRunner<any, any, any> | undefined = AgentRunner<any, any, any> | undefined,
+	// eslint-disable-next-line @typescript-eslint/no-explicit-any
+	TParent extends AgentRunner<any, any, any> | undefined = AgentRunner<any, any, any> | undefined,
+> implements AgentContext<TAgentMap, TCurrent, TParent>
+{
 	agent: TAgentMap;
-	current: TAgent;
-	parent?: TAgent;
+	current: TCurrent;
+	parent: TParent;
 	agentName: AgentName;
 	logger: Logger;
 	sessionId: string;
@@ -38,7 +51,7 @@ export class RequestAgentContext<TAgentMap, TAgent> implements AgentContext {
 	thread: Thread;
 	private handler: WaitUntilHandler;
 
-	constructor(args: RequestAgentContextArgs<TAgentMap, TAgent>) {
+	constructor(args: RequestAgentContextArgs<TAgentMap, TCurrent, TParent>) {
 		this.agent = args.agent;
 		this.current = args.current;
 		this.parent = args.parent;
@@ -58,7 +71,8 @@ export class RequestAgentContext<TAgentMap, TAgent> implements AgentContext {
 	}
 }
 
-const agentAsyncLocalStorage = new AsyncLocalStorage<AgentContext>();
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+const agentAsyncLocalStorage = new AsyncLocalStorage<AgentContext<any, any, any>>();
 const httpAsyncLocalStorage = new AsyncLocalStorage<HonoContext>();
 
 export const inAgentContext = (): boolean => {
@@ -71,7 +85,8 @@ export const inHTTPContext = (): boolean => {
 	return !!context;
 };
 
-export const getAgentContext = (): AgentContext => {
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+export const getAgentContext = (): AgentContext<any, any, any> => {
 	const context = agentAsyncLocalStorage.getStore();
 	if (!context) {
 		throw new Error('AgentContext is not available');
@@ -90,12 +105,18 @@ export const getHTTPContext = (): HonoContext => {
 export const getAgentAsyncLocalStorage = () => agentAsyncLocalStorage;
 export const getHTTPAsyncLocalStorage = () => httpAsyncLocalStorage;
 
-export const runInAgentContext = <TAgentMap, TAgent>(
+export const runInAgentContext = <
+	TAgentMap extends AgentRegistry = AgentRegistry,
+	// eslint-disable-next-line @typescript-eslint/no-explicit-any
+	TCurrent extends AgentRunner<any, any, any> | undefined = AgentRunner<any, any, any> | undefined,
+	// eslint-disable-next-line @typescript-eslint/no-explicit-any
+	TParent extends AgentRunner<any, any, any> | undefined = AgentRunner<any, any, any> | undefined,
+>(
 	ctxObject: Record<string, unknown>,
-	args: RequestAgentContextArgs<TAgentMap, TAgent>,
+	args: RequestAgentContextArgs<TAgentMap, TCurrent, TParent>,
 	next: () => Promise<void>
 ) => {
-	const ctx = new RequestAgentContext(args);
+	const ctx = new RequestAgentContext<TAgentMap, TCurrent, TParent>(args);
 	// eslint-disable-next-line @typescript-eslint/no-explicit-any
 	const _ctx = ctx as any;
 	Object.getOwnPropertyNames(ctx).forEach((k) => {