@ariaflowagents/cf-agent 0.1.0

package/README.md

@@ -0,0 +1,40 @@
+# @ariaflowagents/cf-agent
+
+Cloudflare Agents base class for hosting AriaFlow Runtime on Durable Objects.
+
+## Why Cloudflare Agents
+
+- Batteries-included state with automatic client sync and embedded SQLite.
+- Real-time WebSockets for streaming agent updates.
+- Extensible edge runtime with Durable Objects for global, low-latency sessions.
+
+## Usage
+
+```ts
+import { AriaFlowChatAgent } from '@ariaflowagents/cf-agent';
+import { Runtime, type AgentConfig } from '@ariaflowagents/core';
+
+export class SupportAgent extends AriaFlowChatAgent {
+  async createRuntimeConfig() {
+    const support: AgentConfig = {
+      id: 'support',
+      name: 'Support',
+      systemPrompt: 'You help users with support questions.',
+    };
+
+    return {
+      agents: [support],
+      defaultAgentId: 'support',
+    };
+  }
+}
+```
+
+## Notes
+
+- `createRuntimeConfig()` is called once on startup and should return a `Runtime`
+  or a `HarnessConfig` used to build one.
+- The base class streams AriaFlow parts over WebSockets and syncs `activeAgentId`
+  into Agent state on handoffs.
+- `GET /info` is handled in `onRequest` for simple metadata.
+- `CloudflareSQLiteStore` is exported from this package for custom session wiring.

package/dist/AriaFlowChatAgent.d.ts

@@ -0,0 +1,24 @@
+import { Agent, type Connection, type WSMessage } from 'agents';
+import { Runtime, type HarnessConfig, type HarnessHooks, type HarnessStreamPart } from '@ariaflowagents/core';
+export type AriaFlowRuntimeFactory = HarnessConfig | Runtime;
+export type AriaFlowChatPayload = {
+    type?: string;
+    text?: string;
+    message?: string;
+    content?: string;
+    sessionId?: string;
+    userId?: string;
+};
+export declare abstract class AriaFlowChatAgent<Env = unknown, State = unknown> extends Agent<Env, State> {
+    protected runtime: Runtime;
+    abstract createRuntimeConfig(): Promise<AriaFlowRuntimeFactory> | AriaFlowRuntimeFactory;
+    protected getContextId(): string | null;
+    protected getSessionId(_connection: Connection, _payload: AriaFlowChatPayload): string;
+    protected getHooks(): HarnessHooks;
+    protected shouldBroadcast(): boolean;
+    protected sendPart(connection: Connection, part: HarnessStreamPart): void;
+    onStart(): Promise<void>;
+    onMessage(connection: Connection, message: WSMessage): Promise<void>;
+    onRequest(request: Request): Promise<Response>;
+}
+//# sourceMappingURL=AriaFlowChatAgent.d.ts.map

package/dist/AriaFlowChatAgent.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"AriaFlowChatAgent.d.ts","sourceRoot":"","sources":["../src/AriaFlowChatAgent.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,KAAK,UAAU,EAAE,KAAK,SAAS,EAAE,MAAM,QAAQ,CAAC;AAChE,OAAO,EAAE,OAAO,EAAE,KAAK,aAAa,EAAE,KAAK,YAAY,EAAE,KAAK,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AAG9G,MAAM,MAAM,sBAAsB,GAAG,aAAa,GAAG,OAAO,CAAC;AAE7D,MAAM,MAAM,mBAAmB,GAAG;IAChC,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB,CAAC;AA6DF,8BAAsB,iBAAiB,CAAC,GAAG,GAAG,OAAO,EAAE,KAAK,GAAG,OAAO,CAAE,SAAQ,KAAK,CACnF,GAAG,EACH,KAAK,CACN;IACC,SAAS,CAAC,OAAO,EAAG,OAAO,CAAC;IAE5B,QAAQ,CAAC,mBAAmB,IAAI,OAAO,CAAC,sBAAsB,CAAC,GAAG,sBAAsB;IAExF,SAAS,CAAC,YAAY,IAAI,MAAM,GAAG,IAAI;IAKvC,SAAS,CAAC,YAAY,CAAC,WAAW,EAAE,UAAU,EAAE,QAAQ,EAAE,mBAAmB,GAAG,MAAM;IAItF,SAAS,CAAC,QAAQ,IAAI,YAAY;IA0BlC,SAAS,CAAC,eAAe,IAAI,OAAO;IAIpC,SAAS,CAAC,QAAQ,CAAC,UAAU,EAAE,UAAU,EAAE,IAAI,EAAE,iBAAiB,GAAG,IAAI;IAYnE,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAmBxB,SAAS,CAAC,UAAU,EAAE,UAAU,EAAE,OAAO,EAAE,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC;IAmCpE,SAAS,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC;CAYrD"}

package/dist/AriaFlowChatAgent.js

@@ -0,0 +1,153 @@
+import { Agent } from 'agents';
+import { Runtime } from '@ariaflowagents/core';
+import { CloudflareSQLiteStore } from './stores/CloudflareSQLiteStore.js';
+const mergeHooks = (base, extra) => {
+    if (!base)
+        return extra ?? {};
+    if (!extra)
+        return base;
+    const merged = { ...base };
+    for (const [key, hook] of Object.entries(extra)) {
+        const hookKey = key;
+        const existing = merged[hookKey];
+        if (!hook)
+            continue;
+        if (existing) {
+            merged[hookKey] = (async (...args) => {
+                await existing(...args);
+                await hook(...args);
+            });
+        }
+        else {
+            merged[hookKey] = hook;
+        }
+    }
+    return merged;
+};
+const coerceToString = (message) => {
+    if (typeof message === 'string') {
+        return message;
+    }
+    if (message instanceof ArrayBuffer) {
+        return new TextDecoder().decode(message);
+    }
+    if (ArrayBuffer.isView(message)) {
+        return new TextDecoder().decode(message.buffer);
+    }
+    return null;
+};
+const parsePayload = (raw) => {
+    const asString = coerceToString(raw);
+    if (!asString)
+        return null;
+    try {
+        return JSON.parse(asString);
+    }
+    catch {
+        return { text: asString };
+    }
+};
+export class AriaFlowChatAgent extends Agent {
+    runtime;
+    getContextId() {
+        const ctx = this.ctx;
+        return ctx?.id?.toString?.() ?? null;
+    }
+    getSessionId(_connection, _payload) {
+        return _payload.sessionId ?? this.getContextId() ?? _connection.id;
+    }
+    getHooks() {
+        return {
+            onHandoff: async (_context, _from, to, reason) => {
+                if (this.setState) {
+                    const state = this.state;
+                    this.setState({
+                        ...state,
+                        activeAgentId: to,
+                        lastHandoffReason: reason,
+                        updatedAt: Date.now(),
+                    });
+                }
+            },
+            onAgentStart: async (_context, agentId) => {
+                if (this.setState) {
+                    const state = this.state;
+                    this.setState({
+                        ...state,
+                        activeAgentId: agentId,
+                        updatedAt: Date.now(),
+                    });
+                }
+            },
+        };
+    }
+    shouldBroadcast() {
+        return true;
+    }
+    sendPart(connection, part) {
+        const payload = JSON.stringify(part);
+        const broadcaster = this;
+        if (this.shouldBroadcast() && typeof broadcaster.broadcast === 'function') {
+            broadcaster.broadcast(payload);
+            return;
+        }
+        connection.send(payload);
+    }
+    async onStart() {
+        await super.onStart();
+        const config = await this.createRuntimeConfig();
+        if (config instanceof Runtime) {
+            this.runtime = config;
+            return;
+        }
+        const hooks = mergeHooks(config.hooks, this.getHooks());
+        const sessionStore = config.sessionStore ?? new CloudflareSQLiteStore(this.sql);
+        this.runtime = new Runtime({
+            ...config,
+            hooks,
+            sessionStore,
+        });
+    }
+    async onMessage(connection, message) {
+        const payload = parsePayload(message);
+        if (!payload)
+            return;
+        if (payload.type === 'ping') {
+            connection.send(JSON.stringify({ type: 'pong' }));
+            return;
+        }
+        const input = payload.text ?? payload.message ?? payload.content;
+        if (typeof input !== 'string' || !input.trim()) {
+            connection.send(JSON.stringify({ type: 'error', error: 'Message content required' }));
+            return;
+        }
+        const sessionId = this.getSessionId(connection, payload);
+        try {
+            for await (const part of this.runtime.stream({
+                input,
+                sessionId,
+                userId: payload.userId,
+            })) {
+                this.sendPart(connection, part);
+            }
+        }
+        catch (error) {
+            connection.send(JSON.stringify({
+                type: 'error',
+                error: error.message,
+            }));
+        }
+    }
+    async onRequest(request) {
+        const url = new URL(request.url);
+        if (url.pathname.endsWith('/info')) {
+            return Response.json({
+                agentId: this.getContextId(),
+                runtimeReady: Boolean(this.runtime),
+                timestamp: Date.now(),
+            });
+        }
+        return super.onRequest(request);
+    }
+}
+//# sourceMappingURL=AriaFlowChatAgent.js.map

package/dist/AriaFlowChatAgent.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"AriaFlowChatAgent.js","sourceRoot":"","sources":["../src/AriaFlowChatAgent.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAmC,MAAM,QAAQ,CAAC;AAChE,OAAO,EAAE,OAAO,EAAiE,MAAM,sBAAsB,CAAC;AAC9G,OAAO,EAAE,qBAAqB,EAAE,MAAM,mCAAmC,CAAC;AAqB1E,MAAM,UAAU,GAAG,CAAC,IAAmB,EAAE,KAAoB,EAAgB,EAAE;IAC7E,IAAI,CAAC,IAAI;QAAE,OAAO,KAAK,IAAI,EAAE,CAAC;IAC9B,IAAI,CAAC,KAAK;QAAE,OAAO,IAAI,CAAC;IAExB,MAAM,MAAM,GAAG,EAAE,GAAG,IAAI,EAAwC,CAAC;IAEjE,KAAK,MAAM,CAAC,GAAG,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QAChD,MAAM,OAAO,GAAG,GAAc,CAAC;QAC/B,MAAM,QAAQ,GAAG,MAAM,CAAC,OAAO,CAAuB,CAAC;QACvD,IAAI,CAAC,IAAI;YAAE,SAAS;QAEpB,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,GAAG,IAAW,EAAE,EAAE;gBAC1C,MAAM,QAAQ,CAAC,GAAG,IAAI,CAAC,CAAC;gBACxB,MAAO,IAAe,CAAC,GAAG,IAAI,CAAC,CAAC;YAClC,CAAC,CAA0B,CAAC;QAC9B,CAAC;aAAM,CAAC;YACN,MAAM,CAAC,OAAO,CAAC,GAAG,IAA6B,CAAC;QAClD,CAAC;IACH,CAAC;IAED,OAAO,MAAsB,CAAC;AAChC,CAAC,CAAC;AAEF,MAAM,cAAc,GAAG,CAAC,OAAkB,EAAiB,EAAE;IAC3D,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;QAChC,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,IAAI,OAAO,YAAY,WAAW,EAAE,CAAC;QACnC,OAAO,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC3C,CAAC;IAED,IAAI,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;QAChC,OAAO,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IAClD,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC,CAAC;AAEF,MAAM,YAAY,GAAG,CAAC,GAAc,EAA8B,EAAE;IAClE,MAAM,QAAQ,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC;IACrC,IAAI,CAAC,QAAQ;QAAE,OAAO,IAAI,CAAC;IAE3B,IAAI,CAAC;QACH,OAAO,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAwB,CAAC;IACrD,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;IAC5B,CAAC;AACH,CAAC,CAAC;AAEF,MAAM,OAAgB,iBAAkD,SAAQ,KAG/E;IACW,OAAO,CAAW;IAIlB,YAAY;QACpB,MAAM,GAAG,GAAI,IAAiE,CAAC,GAAG,CAAC;QACnF,OAAO,GAAG,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,IAAI,IAAI,CAAC;IACvC,CAAC;IAES,YAAY,CAAC,WAAuB,EAAE,QAA6B;QAC3E,OAAO,QAAQ,CAAC,SAAS,IAAI,IAAI,CAAC,YAAY,EAAE,IAAI,WAAW,CAAC,EAAE,CAAC;IACrE,CAAC;IAES,QAAQ;QAChB,OAAO;YACL,SAAS,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE;gBAC/C,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;oBAClB,MAAM,KAAK,GAAG,IAAI,CAAC,KAAgC,CAAC;oBACpD,IAAI,CAAC,QAAQ,CAAC;wBACZ,GAAG,KAAK;wBACR,aAAa,EAAE,EAAE;wBACjB,iBAAiB,EAAE,MAAM;wBACzB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;qBACb,CAAC,CAAC;gBACd,CAAC;YACH,CAAC;YACD,YAAY,EAAE,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE;gBACxC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;oBAClB,MAAM,KAAK,GAAG,IAAI,CAAC,KAAgC,CAAC;oBACpD,IAAI,CAAC,QAAQ,CAAC;wBACZ,GAAG,KAAK;wBACR,aAAa,EAAE,OAAO;wBACtB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;qBACb,CAAC,CAAC;gBACd,CAAC;YACH,CAAC;SACF,CAAC;IACJ,CAAC;IAES,eAAe;QACvB,OAAO,IAAI,CAAC;IACd,CAAC;IAES,QAAQ,CAAC,UAAsB,EAAE,IAAuB;QAChE,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QACrC,MAAM,WAAW,GAAG,IAAmC,CAAC;QAExD,IAAI,IAAI,CAAC,eAAe,EAAE,IAAI,OAAO,WAAW,CAAC,SAAS,KAAK,UAAU,EAAE,CAAC;YAC1E,WAAW,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;YAC/B,OAAO;QACT,CAAC;QAED,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,OAAO;QACX,MAAM,KAAK,CAAC,OAAO,EAAE,CAAC;QAEtB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,mBAAmB,EAAE,CAAC;QAChD,IAAI,MAAM,YAAY,OAAO,EAAE,CAAC;YAC9B,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;YACtB,OAAO;QACT,CAAC;QAED,MAAM,KAAK,GAAG,UAAU,CAAC,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;QACxD,MAAM,YAAY,GAAG,MAAM,CAAC,YAAY,IAAI,IAAI,qBAAqB,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAEhF,IAAI,CAAC,OAAO,GAAG,IAAI,OAAO,CAAC;YACzB,GAAG,MAAM;YACT,KAAK;YACL,YAAY;SACb,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,UAAsB,EAAE,OAAkB;QACxD,MAAM,OAAO,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC;QACtC,IAAI,CAAC,OAAO;YAAE,OAAO;QAErB,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC5B,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;YAClD,OAAO;QACT,CAAC;QAED,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,IAAI,OAAO,CAAC,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC;QACjE,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,EAAE,CAAC;YAC/C,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,0BAA0B,EAAE,CAAC,CAAC,CAAC;YACtF,OAAO;QACT,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QAEzD,IAAI,CAAC;YACH,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC;gBAC3C,KAAK;gBACL,SAAS;gBACT,MAAM,EAAE,OAAO,CAAC,MAAM;aACvB,CAAC,EAAE,CAAC;gBACH,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;YAClC,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,UAAU,CAAC,IAAI,CACb,IAAI,CAAC,SAAS,CAAC;gBACb,IAAI,EAAE,OAAO;gBACb,KAAK,EAAG,KAAe,CAAC,OAAO;aAChC,CAAC,CACH,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,OAAgB;QAC9B,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QACjC,IAAI,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;YACnC,OAAO,QAAQ,CAAC,IAAI,CAAC;gBACnB,OAAO,EAAE,IAAI,CAAC,YAAY,EAAE;gBAC5B,YAAY,EAAE,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC;gBACnC,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;aACtB,CAAC,CAAC;QACL,CAAC;QAED,OAAO,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IAClC,CAAC;CACF"}

package/dist/index.d.ts

@@ -0,0 +1,4 @@
+export { AriaFlowChatAgent } from './AriaFlowChatAgent.js';
+export type { AriaFlowChatPayload, AriaFlowRuntimeFactory } from './AriaFlowChatAgent.js';
+export { CloudflareSQLiteStore } from './stores/CloudflareSQLiteStore.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAC3D,YAAY,EAAE,mBAAmB,EAAE,sBAAsB,EAAE,MAAM,wBAAwB,CAAC;AAC1F,OAAO,EAAE,qBAAqB,EAAE,MAAM,mCAAmC,CAAC"}

package/dist/index.js

@@ -0,0 +1,3 @@
+export { AriaFlowChatAgent } from './AriaFlowChatAgent.js';
+export { CloudflareSQLiteStore } from './stores/CloudflareSQLiteStore.js';
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAE3D,OAAO,EAAE,qBAAqB,EAAE,MAAM,mCAAmC,CAAC"}

package/dist/stores/CloudflareSQLiteStore.d.ts

@@ -0,0 +1,12 @@
+import type { Session, SessionStore } from '@ariaflowagents/core';
+export declare class CloudflareSQLiteStore implements SessionStore {
+    private sql;
+    constructor(sql: any);
+    private init;
+    get(id: string): Promise<Session | null>;
+    save(session: Session): Promise<void>;
+    delete(id: string): Promise<void>;
+    list(userId?: string): Promise<Session[]>;
+    cleanup(maxAgeMs: number): Promise<number>;
+}
+//# sourceMappingURL=CloudflareSQLiteStore.d.ts.map

package/dist/stores/CloudflareSQLiteStore.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"CloudflareSQLiteStore.d.ts","sourceRoot":"","sources":["../../src/stores/CloudflareSQLiteStore.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AAiClE,qBAAa,qBAAsB,YAAW,YAAY;IAC5C,OAAO,CAAC,GAAG;gBAAH,GAAG,EAAE,GAAG;IAI5B,OAAO,CAAC,IAAI;IAUN,GAAG,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,GAAG,IAAI,CAAC;IAaxC,IAAI,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC;IAWrC,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAIjC,IAAI,CAAC,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC;IAmBzC,OAAO,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;CAajD"}

package/dist/stores/CloudflareSQLiteStore.js

@@ -0,0 +1,99 @@
+function reviveSession(raw) {
+    const session = { ...raw };
+    session.createdAt = new Date(session.createdAt);
+    session.updatedAt = new Date(session.updatedAt);
+    session.handoffHistory = (session.handoffHistory ?? []).map(record => ({
+        ...record,
+        timestamp: new Date(record.timestamp),
+    }));
+    if (session.metadata) {
+        session.metadata = {
+            ...session.metadata,
+            createdAt: new Date(session.metadata.createdAt),
+            lastActiveAt: new Date(session.metadata.lastActiveAt),
+            handoffHistory: (session.metadata.handoffHistory ?? []).map(record => ({
+                ...record,
+                timestamp: new Date(record.timestamp),
+            })),
+        };
+    }
+    session.agentStates = Object.fromEntries(Object.entries(session.agentStates ?? {}).map(([agentId, state]) => [
+        agentId,
+        {
+            ...state,
+            lastActive: new Date(state.lastActive),
+        },
+    ]));
+    return session;
+}
+export class CloudflareSQLiteStore {
+    sql;
+    constructor(sql) {
+        this.sql = sql;
+        this.init();
+    }
+    init() {
+        this.sql `
+      CREATE TABLE IF NOT EXISTS ariaflow_sessions (
+        id TEXT PRIMARY KEY,
+        data TEXT NOT NULL,
+        updated_at DATETIME DEFAULT CURRENT_TIMESTAMP
+      )
+    `;
+    }
+    async get(id) {
+        const rows = this.sql `SELECT data FROM ariaflow_sessions WHERE id = ${id}`;
+        if (rows && rows.length > 0) {
+            try {
+                return reviveSession(JSON.parse(rows[0].data));
+            }
+            catch (error) {
+                console.error('Failed to parse session data from SQLite', error);
+                return null;
+            }
+        }
+        return null;
+    }
+    async save(session) {
+        const data = JSON.stringify(session);
+        this.sql `
+      INSERT INTO ariaflow_sessions (id, data, updated_at)
+      VALUES (${session.id}, ${data}, CURRENT_TIMESTAMP)
+      ON CONFLICT(id) DO UPDATE SET
+        data = excluded.data,
+        updated_at = excluded.updated_at
+    `;
+    }
+    async delete(id) {
+        this.sql `DELETE FROM ariaflow_sessions WHERE id = ${id}`;
+    }
+    async list(userId) {
+        const rows = (this.sql `SELECT data FROM ariaflow_sessions` ?? []);
+        const sessions = rows
+            .map(row => {
+            try {
+                return reviveSession(JSON.parse(row.data));
+            }
+            catch {
+                return null;
+            }
+        })
+            .filter((session) => session !== null);
+        if (!userId) {
+            return sessions;
+        }
+        return sessions.filter((session) => session.userId === userId);
+    }
+    async cleanup(maxAgeMs) {
+        const cutoff = new Date(Date.now() - maxAgeMs).toISOString();
+        const result = this.sql `
+      DELETE FROM ariaflow_sessions
+      WHERE updated_at < ${cutoff}
+    `;
+        if (typeof result?.changes === 'number') {
+            return result.changes;
+        }
+        return 0;
+    }
+}
+//# sourceMappingURL=CloudflareSQLiteStore.js.map

package/dist/stores/CloudflareSQLiteStore.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"CloudflareSQLiteStore.js","sourceRoot":"","sources":["../../src/stores/CloudflareSQLiteStore.ts"],"names":[],"mappings":"AAEA,SAAS,aAAa,CAAC,GAAY;IACjC,MAAM,OAAO,GAAG,EAAE,GAAG,GAAG,EAAa,CAAC;IACtC,OAAO,CAAC,SAAS,GAAG,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAChD,OAAO,CAAC,SAAS,GAAG,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAChD,OAAO,CAAC,cAAc,GAAG,CAAC,OAAO,CAAC,cAAc,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QACrE,GAAG,MAAM;QACT,SAAS,EAAE,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC;KACtC,CAAC,CAAC,CAAC;IACJ,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;QACrB,OAAO,CAAC,QAAQ,GAAG;YACjB,GAAG,OAAO,CAAC,QAAQ;YACnB,SAAS,EAAE,IAAI,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC;YAC/C,YAAY,EAAE,IAAI,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAC;YACrD,cAAc,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,cAAc,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBACrE,GAAG,MAAM;gBACT,SAAS,EAAE,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC;aACtC,CAAC,CAAC;SACJ,CAAC;IACJ,CAAC;IACD,OAAO,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,CACtC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,WAAW,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC;QAClE,OAAO;QACP;YACE,GAAG,KAAK;YACR,UAAU,EAAE,IAAI,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC;SACvC;KACF,CAAC,CACH,CAAC;IACF,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,MAAM,OAAO,qBAAqB;IACZ;IAApB,YAAoB,GAAQ;QAAR,QAAG,GAAH,GAAG,CAAK;QAC1B,IAAI,CAAC,IAAI,EAAE,CAAC;IACd,CAAC;IAEO,IAAI;QACV,IAAI,CAAC,GAAG,CAAA;;;;;;KAMP,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,EAAU;QAClB,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAA,iDAAiD,EAAE,EAAE,CAAC;QAC3E,IAAI,IAAI,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5B,IAAI,CAAC;gBACH,OAAO,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;YACjD,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,0CAA0C,EAAE,KAAK,CAAC,CAAC;gBACjE,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,OAAgB;QACzB,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QACrC,IAAI,CAAC,GAAG,CAAA;;gBAEI,OAAO,CAAC,EAAE,KAAK,IAAI;;;;KAI9B,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,EAAU;QACrB,IAAI,CAAC,GAAG,CAAA,4CAA4C,EAAE,EAAE,CAAC;IAC3D,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,MAAe;QACxB,MAAM,IAAI,GAAG,CAAC,IAAI,CAAC,GAAG,CAAA,oCAAoC,IAAI,EAAE,CAA4B,CAAC;QAC7F,MAAM,QAAQ,GAAc,IAAI;aAC7B,GAAG,CAAC,GAAG,CAAC,EAAE;YACT,IAAI,CAAC;gBACH,OAAO,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC;YAC7C,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC,CAAC;aACD,MAAM,CAAC,CAAC,OAAO,EAAsB,EAAE,CAAC,OAAO,KAAK,IAAI,CAAC,CAAC;QAE7D,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,OAAO,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAgB,EAAE,EAAE,CAAC,OAAO,CAAC,MAAM,KAAK,MAAM,CAAC,CAAC;IAC1E,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,QAAgB;QAC5B,MAAM,MAAM,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;QAC7D,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,CAAA;;2BAEA,MAAM;KAC5B,CAAC;QAEF,IAAI,OAAO,MAAM,EAAE,OAAO,KAAK,QAAQ,EAAE,CAAC;YACxC,OAAO,MAAM,CAAC,OAAO,CAAC;QACxB,CAAC;QAED,OAAO,CAAC,CAAC;IACX,CAAC;CACF"}

package/package.json

@@ -0,0 +1,38 @@
+{
+  "name": "@ariaflowagents/cf-agent",
+  "version": "0.1.0",
+  "description": "Cloudflare Agents base class for hosting AriaFlow Runtime on Durable Objects",
+  "type": "module",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "default": "./dist/index.js"
+    }
+  },
+  "dependencies": {
+    "agents": "^0.3.5"
+  },
+  "peerDependencies": {
+    "@ariaflowagents/core": "^0.1.0",
+    "agents": "^0.3.5"
+  },
+  "devDependencies": {
+    "@ariaflowagents/core": "file:../ariaflow-core",
+    "agents": "^0.3.5",
+    "typescript": "^5.3.0"
+  },
+  "scripts": {
+    "build": "tsc -p tsconfig.json",
+    "clean": "rm -rf dist",
+    "prepublishOnly": "npm run build"
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "files": [
+    "dist",
+    "README.md"
+  ]
+}