@agentuity/runtime 0.0.43 → 0.0.44

package/src/services/session/local.ts

@@ -0,0 +1,28 @@
+import {
+	type SessionEventProvider,
+	type SessionStartEvent,
+	type SessionCompleteEvent,
+} from '@agentuity/core';
+
+/**
+ * An implementation of the SessionEventProvider which is no-op
+ */
+export class LocalSessionEventProvider implements SessionEventProvider {
+	/**
+	 * called when the session starts
+	 *
+	 * @param event SessionStartEvent
+	 */
+	async start(_event: SessionStartEvent): Promise<void> {
+		// no op
+	}
+
+	/**
+	 * called when the session completes
+	 *
+	 * @param event SessionCompleteEvent
+	 */
+	async complete(_event: SessionCompleteEvent): Promise<void> {
+		// no op
+	}
+}

package/src/session.ts

@@ -0,0 +1,284 @@
+/* eslint-disable @typescript-eslint/no-explicit-any */
+/** biome-ignore-all lint/suspicious/noExplicitAny: anys are great */
+import type { Context } from 'hono';
+import { getCookie, setCookie } from 'hono/cookie';
+import { type Env, fireEvent } from './app';
+
+export type ThreadEventName = 'destroyed';
+export type SessionEventName = 'completed';
+
+type ThreadEventCallback<T extends Thread> = (
+	eventName: 'destroyed',
+	thread: T
+) => Promise<void> | void;
+
+type SessionEventCallback<T extends Session> = (
+	eventName: 'completed',
+	session: T
+) => Promise<void> | void;
+
+export interface Thread {
+	id: string;
+	state: Map<string, unknown>;
+	addEventListener(
+		eventName: 'destroyed',
+		callback: (eventName: 'destroyed', thread: Thread) => Promise<void> | void
+	): void;
+	removeEventListener(
+		eventName: 'destroyed',
+		callback: (eventName: 'destroyed', thread: Thread) => Promise<void> | void
+	): void;
+	destroy(): Promise<void>;
+}
+
+export interface Session {
+	id: string;
+	thread: Thread;
+	state: Map<string, unknown>;
+	addEventListener(
+		eventName: 'completed',
+		callback: (eventName: 'completed', session: Session) => Promise<void> | void
+	): void;
+	removeEventListener(
+		eventName: 'completed',
+		callback: (eventName: 'completed', session: Session) => Promise<void> | void
+	): void;
+}
+
+export interface ThreadProvider {
+	initialize(): Promise<void>;
+	restore(ctx: Context<Env>): Promise<Thread>;
+	save(thread: Thread): Promise<void>;
+	destroy(thread: Thread): Promise<void>;
+}
+
+export interface SessionProvider {
+	initialize(): Promise<void>;
+	restore(thread: Thread, sessionId: string): Promise<Session>;
+	save(session: Session): Promise<void>;
+}
+
+// WeakMap to store event listeners for Thread and Session instances
+const threadEventListeners = new WeakMap<
+	Thread,
+	Map<ThreadEventName, Set<ThreadEventCallback<any>>>
+>();
+const sessionEventListeners = new WeakMap<
+	Session,
+	Map<SessionEventName, Set<SessionEventCallback<any>>>
+>();
+
+// Helper to fire thread event listeners
+async function fireThreadEvent(thread: Thread, eventName: ThreadEventName): Promise<void> {
+	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 as any)(eventName, thread);
+	}
+}
+
+// Helper to fire session event listeners
+async function fireSessionEvent(session: Session, eventName: SessionEventName): Promise<void> {
+	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 as any)(eventName, session);
+	}
+}
+
+// Generate thread or session ID
+export function generateId(prefix?: string): string {
+	const arr = new Uint8Array(16);
+	crypto.getRandomValues(arr);
+	return `${prefix}${prefix ? '_' : ''}${arr.toHex()}`;
+}
+
+export class DefaultThread implements Thread {
+	#lastUsed: number;
+	readonly id: string;
+	readonly state: Map<string, unknown>;
+	private provider: ThreadProvider;
+
+	constructor(provider: ThreadProvider, id: string) {
+		this.provider = provider;
+		this.id = id;
+		this.state = new Map();
+		this.#lastUsed = Date.now();
+	}
+
+	addEventListener(eventName: ThreadEventName, callback: ThreadEventCallback<any>): void {
+		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: ThreadEventName, callback: ThreadEventCallback<any>): void {
+		const listeners = threadEventListeners.get(this);
+		if (!listeners) return;
+		const callbacks = listeners.get(eventName);
+		if (!callbacks) return;
+		callbacks.delete(callback);
+	}
+
+	async fireEvent(eventName: ThreadEventName): Promise<void> {
+		await fireThreadEvent(this, eventName);
+	}
+
+	async destroy(): Promise<void> {
+		this.provider.destroy(this);
+	}
+
+	touch() {
+		this.#lastUsed = Date.now();
+	}
+
+	expired() {
+		return Date.now() - this.#lastUsed >= 3.6e6; // 1 hour
+	}
+}
+
+export class DefaultSession implements Session {
+	readonly id: string;
+	readonly thread: Thread;
+	readonly state: Map<string, unknown>;
+
+	constructor(thread: Thread, id: string) {
+		this.id = id;
+		this.thread = thread;
+		this.state = new Map();
+	}
+
+	addEventListener(eventName: SessionEventName, callback: SessionEventCallback<any>): void {
+		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: SessionEventName, callback: SessionEventCallback<any>): void {
+		const listeners = sessionEventListeners.get(this);
+		if (!listeners) return;
+		const callbacks = listeners.get(eventName);
+		if (!callbacks) return;
+		callbacks.delete(callback);
+	}
+
+	async fireEvent(eventName: SessionEventName): Promise<void> {
+		await fireSessionEvent(this, eventName);
+	}
+}
+
+export class DefaultThreadProvider implements ThreadProvider {
+	private threads = new Map<string, DefaultThread>();
+
+	async initialize(): Promise<void> {
+		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: Context<Env>): Promise<Thread> {
+		const cookie = getCookie(ctx);
+		let threadId: string | undefined;
+
+		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: Thread): Promise<void> {
+		if (thread instanceof DefaultThread) {
+			thread.touch();
+		}
+	}
+
+	async destroy(thread: Thread): Promise<void> {
+		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 implements SessionProvider {
+	private sessions = new Map<string, DefaultSession>();
+
+	async initialize(): Promise<void> {
+		// No initialization needed for in-memory provider
+	}
+
+	async restore(thread: Thread, sessionId: string): Promise<Session> {
+		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: Session): Promise<void> {
+		if (session instanceof DefaultSession) {
+			try {
+				await session.fireEvent('completed');
+				await fireEvent('session.completed', session);
+			} finally {
+				this.sessions.delete(session.id);
+				sessionEventListeners.delete(session);
+			}
+		}
+	}
+}

package/dist/_unauthenticated.d.ts

@@ -1,26 +0,0 @@
-import type { KeyValueStorage, ObjectStorage, StreamStorage, VectorStorage, DataResult, KeyValueStorageSetParams, ObjectResult, ObjectStorePutParams, CreatePublicURLParams, Stream, CreateStreamProps, ListStreamsParams, ListStreamsResponse, VectorUpsertParams, VectorUpsertResult, VectorResult, VectorSearchResultWithDocument, VectorSearchParams, VectorSearchResult } from '@agentuity/core';
-export declare class UnauthenticatedKeyValueStorage implements KeyValueStorage {
-    get<T>(_name: string, _key: string): Promise<DataResult<T>>;
-    set<T = unknown>(_name: string, _key: string, _value: T, _params?: KeyValueStorageSetParams): Promise<void>;
-    delete(_name: string, _key: string): Promise<void>;
-}
-export declare class UnauthenticatedObjectStorage implements ObjectStorage {
-    get(_bucket: string, _key: string): Promise<ObjectResult>;
-    put(_bucket: string, _key: string, _data: Uint8Array | ArrayBuffer | ReadableStream, _params?: ObjectStorePutParams): Promise<void>;
-    delete(_bucket: string, _key: string): Promise<boolean>;
-    createPublicURL(_bucket: string, _key: string, _params?: CreatePublicURLParams): Promise<string>;
-}
-export declare class UnauthenticatedStreamStorage implements StreamStorage {
-    create(_name: string, _props?: CreateStreamProps): Promise<Stream>;
-    list(_params?: ListStreamsParams): Promise<ListStreamsResponse>;
-    delete(_id: string): Promise<void>;
-}
-export declare class UnauthenticatedVectorStorage implements VectorStorage {
-    upsert(_name: string, ..._documents: VectorUpsertParams[]): Promise<VectorUpsertResult[]>;
-    get<T extends Record<string, unknown> = Record<string, unknown>>(_name: string, _key: string): Promise<VectorResult<T>>;
-    getMany<T extends Record<string, unknown> = Record<string, unknown>>(_name: string, ..._keys: string[]): Promise<Map<string, VectorSearchResultWithDocument<T>>>;
-    search<T extends Record<string, unknown> = Record<string, unknown>>(_name: string, _params: VectorSearchParams<T>): Promise<VectorSearchResult<T>[]>;
-    delete(_name: string, ..._keys: string[]): Promise<number>;
-    exists(_name: string): Promise<boolean>;
-}
-//# sourceMappingURL=_unauthenticated.d.ts.map

package/dist/_unauthenticated.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"_unauthenticated.d.ts","sourceRoot":"","sources":["../src/_unauthenticated.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACX,eAAe,EACf,aAAa,EACb,aAAa,EACb,aAAa,EACb,UAAU,EACV,wBAAwB,EACxB,YAAY,EACZ,oBAAoB,EACpB,qBAAqB,EACrB,MAAM,EACN,iBAAiB,EACjB,iBAAiB,EACjB,mBAAmB,EACnB,kBAAkB,EAClB,kBAAkB,EAClB,YAAY,EACZ,8BAA8B,EAC9B,kBAAkB,EAClB,kBAAkB,EAClB,MAAM,iBAAiB,CAAC;AAYzB,qBAAa,8BAA+B,YAAW,eAAe;IAC/D,GAAG,CAAC,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;IAI3D,GAAG,CAAC,CAAC,GAAG,OAAO,EACpB,KAAK,EAAE,MAAM,EACb,IAAI,EAAE,MAAM,EACZ,MAAM,EAAE,CAAC,EACT,OAAO,CAAC,EAAE,wBAAwB,GAChC,OAAO,CAAC,IAAI,CAAC;IAIV,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;CAGxD;AAED,qBAAa,4BAA6B,YAAW,aAAa;IAC3D,GAAG,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC;IAIzD,GAAG,CACR,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,KAAK,EAAE,UAAU,GAAG,WAAW,GAAG,cAAc,EAChD,OAAO,CAAC,EAAE,oBAAoB,GAC5B,OAAO,CAAC,IAAI,CAAC;IAIV,MAAM,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAIvD,eAAe,CACpB,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,CAAC,EAAE,qBAAqB,GAC7B,OAAO,CAAC,MAAM,CAAC;CAGlB;AAED,qBAAa,4BAA6B,YAAW,aAAa;IAC3D,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,iBAAiB,GAAG,OAAO,CAAC,MAAM,CAAC;IAIlE,IAAI,CAAC,OAAO,CAAC,EAAE,iBAAiB,GAAG,OAAO,CAAC,mBAAmB,CAAC;IAI/D,MAAM,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;CAGxC;AAED,qBAAa,4BAA6B,YAAW,aAAa;IAC3D,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,UAAU,EAAE,kBAAkB,EAAE,GAAG,OAAO,CAAC,kBAAkB,EAAE,CAAC;IAIzF,GAAG,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACpE,KAAK,EAAE,MAAM,EACb,IAAI,EAAE,MAAM,GACV,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAIrB,OAAO,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACxE,KAAK,EAAE,MAAM,EACb,GAAG,KAAK,EAAE,MAAM,EAAE,GAChB,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,8BAA8B,CAAC,CAAC,CAAC,CAAC,CAAC;IAIpD,MAAM,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACvE,KAAK,EAAE,MAAM,EACb,OAAO,EAAE,kBAAkB,CAAC,CAAC,CAAC,GAC5B,OAAO,CAAC,kBAAkB,CAAC,CAAC,CAAC,EAAE,CAAC;IAI7B,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC;IAI1D,MAAM,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;CAG7C"}

package/src/_unauthenticated.ts

@@ -1,126 +0,0 @@
-import type {
-	KeyValueStorage,
-	ObjectStorage,
-	StreamStorage,
-	VectorStorage,
-	DataResult,
-	KeyValueStorageSetParams,
-	ObjectResult,
-	ObjectStorePutParams,
-	CreatePublicURLParams,
-	Stream,
-	CreateStreamProps,
-	ListStreamsParams,
-	ListStreamsResponse,
-	VectorUpsertParams,
-	VectorUpsertResult,
-	VectorResult,
-	VectorSearchResultWithDocument,
-	VectorSearchParams,
-	VectorSearchResult,
-} from '@agentuity/core';
-
-const UNAUTHENTICATED_ERROR =
-	'You must authenticate to use Agentuity services. Set the AGENTUITY_SDK_KEY environment variable or provide custom service implementations in your app config.';
-
-class UnauthenticatedError extends Error {
-	constructor() {
-		super(UNAUTHENTICATED_ERROR);
-		this.name = 'UnauthenticatedError';
-	}
-}
-
-export class UnauthenticatedKeyValueStorage implements KeyValueStorage {
-	async get<T>(_name: string, _key: string): Promise<DataResult<T>> {
-		throw new UnauthenticatedError();
-	}
-
-	async set<T = unknown>(
-		_name: string,
-		_key: string,
-		_value: T,
-		_params?: KeyValueStorageSetParams
-	): Promise<void> {
-		throw new UnauthenticatedError();
-	}
-
-	async delete(_name: string, _key: string): Promise<void> {
-		throw new UnauthenticatedError();
-	}
-}
-
-export class UnauthenticatedObjectStorage implements ObjectStorage {
-	async get(_bucket: string, _key: string): Promise<ObjectResult> {
-		throw new UnauthenticatedError();
-	}
-
-	async put(
-		_bucket: string,
-		_key: string,
-		_data: Uint8Array | ArrayBuffer | ReadableStream,
-		_params?: ObjectStorePutParams
-	): Promise<void> {
-		throw new UnauthenticatedError();
-	}
-
-	async delete(_bucket: string, _key: string): Promise<boolean> {
-		throw new UnauthenticatedError();
-	}
-
-	async createPublicURL(
-		_bucket: string,
-		_key: string,
-		_params?: CreatePublicURLParams
-	): Promise<string> {
-		throw new UnauthenticatedError();
-	}
-}
-
-export class UnauthenticatedStreamStorage implements StreamStorage {
-	async create(_name: string, _props?: CreateStreamProps): Promise<Stream> {
-		throw new UnauthenticatedError();
-	}
-
-	async list(_params?: ListStreamsParams): Promise<ListStreamsResponse> {
-		throw new UnauthenticatedError();
-	}
-
-	async delete(_id: string): Promise<void> {
-		throw new UnauthenticatedError();
-	}
-}
-
-export class UnauthenticatedVectorStorage implements VectorStorage {
-	async upsert(_name: string, ..._documents: VectorUpsertParams[]): Promise<VectorUpsertResult[]> {
-		throw new UnauthenticatedError();
-	}
-
-	async get<T extends Record<string, unknown> = Record<string, unknown>>(
-		_name: string,
-		_key: string
-	): Promise<VectorResult<T>> {
-		throw new UnauthenticatedError();
-	}
-
-	async getMany<T extends Record<string, unknown> = Record<string, unknown>>(
-		_name: string,
-		..._keys: string[]
-	): Promise<Map<string, VectorSearchResultWithDocument<T>>> {
-		throw new UnauthenticatedError();
-	}
-
-	async search<T extends Record<string, unknown> = Record<string, unknown>>(
-		_name: string,
-		_params: VectorSearchParams<T>
-	): Promise<VectorSearchResult<T>[]> {
-		throw new UnauthenticatedError();
-	}
-
-	async delete(_name: string, ..._keys: string[]): Promise<number> {
-		throw new UnauthenticatedError();
-	}
-
-	async exists(_name: string): Promise<boolean> {
-		throw new UnauthenticatedError();
-	}
-}