@projectaria/cf-agents 0.1.1 → 0.1.2

package/dist/AriaCFChatAgent.d.ts

@@ -0,0 +1,37 @@
+/**
+ * AriaCFChatAgent - Main Agent Class for ARIA on Cloudflare
+ *
+ * This class extends Cloudflare's AIChatAgent and integrates ARIA's
+ * agent system for structured conversations.
+ */
+import { AIChatAgent } from "@cloudflare/ai-chat";
+import { type AgentContext } from "agents";
+import { type Agent as AriaAgent, type AgentSession, type ToolRegistry, VoltRuntime } from "@projectaria/aria-agents";
+import type { StreamTextOnFinishCallback, ToolSet, LanguageModel } from "ai";
+import type { AriaCFChatAgentConfig, AriaCFState, AriaCFChatAgentConfigResolved } from "./types";
+import { CloudflareSessionStore } from "./CloudflareSessionStore";
+export declare abstract class AriaCFChatAgent<Env = any, State extends AriaCFState = AriaCFState> extends AIChatAgent<Env, State> {
+    protected readonly ariaAgent: AriaAgent;
+    protected readonly voltRuntime: VoltRuntime;
+    protected readonly config: AriaCFChatAgentConfigResolved;
+    protected readonly sessionStore: CloudflareSessionStore;
+    readonly initialState: State;
+    constructor(ctx: AgentContext, env: Env);
+    protected resolveConfig(): AriaCFChatAgentConfigResolved;
+    protected abstract getConfig(): AriaCFChatAgentConfig;
+    protected createAriaAgent(): AriaAgent;
+    protected createVoltRuntime(): VoltRuntime;
+    protected getModel(): LanguageModel;
+    protected getModelId(): string;
+    protected getTools(): ToolRegistry;
+    onChatMessage(onFinish: StreamTextOnFinishCallback<ToolSet>, _options?: {
+        abortSignal?: AbortSignal;
+    }): Promise<Response | undefined>;
+    protected updateStateFromSession(ariaSession: AgentSession): void;
+    onRequest(request: Request): Promise<Response>;
+    protected getCloudflareTools(): ToolRegistry;
+    onStart(): void;
+    onDestroy(): void;
+}
+export declare function createAriaCFChatAgent<Env = any>(config: AriaCFChatAgentConfig): new (ctx: AgentContext, env: Env) => AriaCFChatAgent<Env>;
+//# sourceMappingURL=AriaCFChatAgent.d.ts.map

package/dist/AriaCFChatAgent.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"AriaCFChatAgent.d.ts","sourceRoot":"","sources":["../src/AriaCFChatAgent.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAClD,OAAO,EAAE,KAAK,YAAY,EAAc,MAAM,QAAQ,CAAC;AACvD,OAAO,EAEL,KAAK,KAAK,IAAI,SAAS,EACvB,KAAK,YAAY,EAEjB,KAAK,YAAY,EACjB,WAAW,EAKZ,MAAM,0BAA0B,CAAC;AAClC,OAAO,KAAK,EAAE,0BAA0B,EAAE,OAAO,EAAE,aAAa,EAAE,MAAM,IAAI,CAAC;AAG7E,OAAO,KAAK,EACV,qBAAqB,EACrB,WAAW,EACX,6BAA6B,EAC9B,MAAM,SAAS,CAAC;AAIjB,OAAO,EAAE,sBAAsB,EAAE,MAAM,0BAA0B,CAAC;AAUlE,8BAAsB,eAAe,CACnC,GAAG,GAAG,GAAG,EACT,KAAK,SAAS,WAAW,GAAG,WAAW,CACvC,SAAQ,WAAW,CAAC,GAAG,EAAE,KAAK,CAAC;IAC/B,SAAS,CAAC,QAAQ,CAAC,SAAS,EAAE,SAAS,CAAC;IACxC,SAAS,CAAC,QAAQ,CAAC,WAAW,EAAE,WAAW,CAAC;IAC5C,SAAS,CAAC,QAAQ,CAAC,MAAM,EAAE,6BAA6B,CAAC;IACzD,SAAS,CAAC,QAAQ,CAAC,YAAY,EAAE,sBAAsB,CAAC;IAExD,QAAQ,CAAC,YAAY,EAAE,KAAK,CAEjB;gBAEC,GAAG,EAAE,YAAY,EAAE,GAAG,EAAE,GAAG;IAsBvC,SAAS,CAAC,aAAa,IAAI,6BAA6B;IAaxD,SAAS,CAAC,QAAQ,CAAC,SAAS,IAAI,qBAAqB;IAErD,SAAS,CAAC,eAAe,IAAI,SAAS;IAWtC,SAAS,CAAC,iBAAiB,IAAI,WAAW;IAc1C,SAAS,CAAC,QAAQ,IAAI,aAAa;IAInC,SAAS,CAAC,UAAU,IAAI,MAAM;IAQ9B,SAAS,CAAC,QAAQ,IAAI,YAAY;IAInB,aAAa,CAC1B,QAAQ,EAAE,0BAA0B,CAAC,OAAO,CAAC,EAC7C,QAAQ,CAAC,EAAE;QAAE,WAAW,CAAC,EAAE,WAAW,CAAA;KAAE,GACvC,OAAO,CAAC,QAAQ,GAAG,SAAS,CAAC;IA4BhC,SAAS,CAAC,sBAAsB,CAAC,WAAW,EAAE,YAAY,GAAG,IAAI;IASlD,SAAS,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC;IAe7D,SAAS,CAAC,kBAAkB,IAAI,YAAY;IAmD5C,OAAO,IAAI,IAAI;IAYf,SAAS,IAAI,IAAI;CAElB;AAED,wBAAgB,qBAAqB,CAAC,GAAG,GAAG,GAAG,EAC7C,MAAM,EAAE,qBAAqB,GAC5B,KAAK,GAAG,EAAE,YAAY,EAAE,GAAG,EAAE,GAAG,KAAK,eAAe,CAAC,GAAG,CAAC,CAU3D"}

package/dist/AriaCFChatAgent.js

@@ -0,0 +1,200 @@
+/**
+ * AriaCFChatAgent - Main Agent Class for ARIA on Cloudflare
+ *
+ * This class extends Cloudflare's AIChatAgent and integrates ARIA's
+ * agent system for structured conversations.
+ */
+import { AIChatAgent } from "@cloudflare/ai-chat";
+import { createAgent, VoltRuntime, CombinedAnalysisService, ProgressionEngine, PromptBuilder, } from "@projectaria/aria-agents";
+import { z } from "zod";
+import { cloudflareToAriaSession, extractLastUserInput } from "./message-format";
+import { createARIAStreamResponse } from "./aria-to-ui-stream";
+import { CloudflareSessionStore } from "./CloudflareSessionStore";
+const ARIA_CF_INITIAL_STATE = {
+    ariaSessionId: null,
+    workingMemory: {},
+    agentConfig: null,
+    lastActivity: null,
+    activeFlow: null,
+};
+export class AriaCFChatAgent extends AIChatAgent {
+    ariaAgent;
+    voltRuntime;
+    config;
+    sessionStore;
+    initialState = {
+        ...ARIA_CF_INITIAL_STATE,
+    };
+    constructor(ctx, env) {
+        super(ctx, env);
+        this.config = this.resolveConfig();
+        this.sessionStore = new CloudflareSessionStore(ctx);
+        this.ariaAgent = this.createAriaAgent();
+        this.voltRuntime = this.createVoltRuntime();
+        const modelId = this.getModelId();
+        const agentName = "aria-agent";
+        const agentId = this.ariaAgent.id;
+        this.setState({
+            ...this.initialState,
+            agentConfig: {
+                id: agentId,
+                name: agentName,
+                modelId,
+            },
+        });
+    }
+    resolveConfig() {
+        const config = this.getConfig();
+        return {
+            flows: config.flows ?? new Map(),
+            model: config.model,
+            instructions: config.instructions ?? "You are a helpful AI assistant.",
+            guidelines: config.guidelines ?? [],
+            tools: config.tools ?? {},
+            historyLimit: config.historyLimit ?? 50,
+            enablePersistence: config.enablePersistence ?? true,
+        };
+    }
+    createAriaAgent() {
+        return createAgent({
+            id: this.name,
+            name: this.name,
+            model: this.config.model,
+            instructions: this.config.instructions,
+            guidelines: this.config.guidelines,
+            tools: this.getTools(),
+        });
+    }
+    createVoltRuntime() {
+        const analysisService = new CombinedAnalysisService(this.config.model, new PromptBuilder({ guidelines: this.config.guidelines }));
+        return new VoltRuntime({
+            flows: this.config.flows,
+            analysisService,
+            progressionEngine: new ProgressionEngine(),
+            sessionStore: this.config.enablePersistence ? this.sessionStore : undefined,
+        });
+    }
+    getModel() {
+        return this.config.model;
+    }
+    getModelId() {
+        const model = this.config.model;
+        if (typeof model === "object" && model !== null) {
+            return model.modelId ?? "unknown";
+        }
+        return "unknown";
+    }
+    getTools() {
+        return this.config.tools;
+    }
+    async onChatMessage(onFinish, _options) {
+        const messages = this.messages;
+        const lastMessage = messages[messages.length - 1];
+        if (!lastMessage) {
+            return new Response("No message", { status: 400 });
+        }
+        const userText = extractLastUserInput(messages);
+        const ariaSession = cloudflareToAriaSession(messages, this.state);
+        if (this.config.enablePersistence) {
+            await this.voltRuntime.startSession(ariaSession.sessionId);
+        }
+        const result = await this.ariaAgent.streamText({
+            input: userText,
+            session: ariaSession,
+        });
+        const session = result.session;
+        if (session) {
+            this.updateStateFromSession(session);
+        }
+        return createARIAStreamResponse(result.fullStream, onFinish);
+    }
+    updateStateFromSession(ariaSession) {
+        this.setState({
+            ...this.state,
+            ariaSessionId: ariaSession.sessionId,
+            workingMemory: ariaSession.workingMemory,
+            lastActivity: new Date().toISOString(),
+        });
+    }
+    async onRequest(request) {
+        const url = new URL(request.url);
+        if (url.pathname === "/health") {
+            return Response.json({
+                status: "healthy",
+                agent: this.name,
+                model: this.getModelId(),
+                lastActivity: this.state.lastActivity,
+            });
+        }
+        return super.onRequest(request);
+    }
+    getCloudflareTools() {
+        return {
+            scheduleTask: {
+                description: "Schedule a task to run in the future.",
+                inputSchema: z.object({
+                    when: z.union([z.number(), z.string()]),
+                    callback: z.string(),
+                    data: z.any().optional(),
+                }),
+                execute: async (args) => {
+                    const when = args.when;
+                    const callback = args.callback;
+                    const data = args.data;
+                    let scheduleWhen = when;
+                    if (typeof when === "string" && !when.includes("*")) {
+                        const date = new Date(when);
+                        if (!isNaN(date.getTime())) {
+                            scheduleWhen = date;
+                        }
+                    }
+                    // schedule() is a Cloudflare Agent method with specific signature
+                    const result = await this.schedule(scheduleWhen, callback, data);
+                    return { taskId: result.id, type: result.type };
+                },
+            },
+            cancelTask: {
+                description: "Cancel a scheduled task by ID",
+                inputSchema: z.object({
+                    taskId: z.string(),
+                }),
+                execute: async (args) => {
+                    const taskId = args.taskId;
+                    const success = await this.cancelSchedule(taskId);
+                    return { success };
+                },
+            },
+            getSchedules: {
+                description: "Get all currently scheduled tasks",
+                inputSchema: z.object({}),
+                execute: async () => {
+                    const schedules = this.getSchedules();
+                    return { schedules };
+                },
+            },
+        };
+    }
+    onStart() {
+        if (this.config.enablePersistence) {
+            this.sessionStore.initialize();
+        }
+        if (this.state.ariaSessionId && this.config.enablePersistence) {
+            this.voltRuntime.startSession(this.state.ariaSessionId).catch((error) => {
+                console.error("Failed to resume session:", error);
+            });
+        }
+    }
+    onDestroy() {
+    }
+}
+export function createAriaCFChatAgent(config) {
+    return class extends AriaCFChatAgent {
+        constructor(ctx, env) {
+            super(ctx, env);
+        }
+        getConfig() {
+            return config;
+        }
+    };
+}
+//# sourceMappingURL=AriaCFChatAgent.js.map

package/dist/AriaCFChatAgent.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"AriaCFChatAgent.js","sourceRoot":"","sources":["../src/AriaCFChatAgent.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAElD,OAAO,EACL,WAAW,EAKX,WAAW,EACX,uBAAuB,EACvB,iBAAiB,EACjB,aAAa,GAEd,MAAM,0BAA0B,CAAC;AAElC,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAQxB,OAAO,EAAE,uBAAuB,EAAE,oBAAoB,EAAE,MAAM,kBAAkB,CAAC;AACjF,OAAO,EAAE,wBAAwB,EAAE,MAAM,qBAAqB,CAAC;AAC/D,OAAO,EAAE,sBAAsB,EAAE,MAAM,0BAA0B,CAAC;AAElE,MAAM,qBAAqB,GAAgB;IACzC,aAAa,EAAE,IAAI;IACnB,aAAa,EAAE,EAAE;IACjB,WAAW,EAAE,IAAI;IACjB,YAAY,EAAE,IAAI;IAClB,UAAU,EAAE,IAAI;CACjB,CAAC;AAEF,MAAM,OAAgB,eAGpB,SAAQ,WAAuB;IACZ,SAAS,CAAY;IACrB,WAAW,CAAc;IACzB,MAAM,CAAgC;IACtC,YAAY,CAAyB;IAE/C,YAAY,GAAU;QAC7B,GAAG,qBAAqB;KAChB,CAAC;IAEX,YAAY,GAAiB,EAAE,GAAQ;QACrC,KAAK,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QAEhB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,aAAa,EAAE,CAAC;QACnC,IAAI,CAAC,YAAY,GAAG,IAAI,sBAAsB,CAAC,GAAuB,CAAC,CAAC;QACxE,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,eAAe,EAAE,CAAC;QACxC,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE5C,MAAM,OAAO,GAAW,IAAI,CAAC,UAAU,EAAE,CAAC;QAC1C,MAAM,SAAS,GAAG,YAAY,CAAC;QAC/B,MAAM,OAAO,GAAW,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;QAE1C,IAAI,CAAC,QAAQ,CAAC;YACZ,GAAG,IAAI,CAAC,YAAY;YACpB,WAAW,EAAE;gBACX,EAAE,EAAE,OAAO;gBACX,IAAI,EAAE,SAAS;gBACf,OAAO;aACR;SACO,CAAC,CAAC;IACd,CAAC;IAES,aAAa;QACrB,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;QAChC,OAAO;YACL,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,IAAI,GAAG,EAAE;YAChC,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,YAAY,EAAE,MAAM,CAAC,YAAY,IAAI,iCAAiC;YACtE,UAAU,EAAE,MAAM,CAAC,UAAU,IAAI,EAAE;YACnC,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,EAAE;YACzB,YAAY,EAAE,MAAM,CAAC,YAAY,IAAI,EAAE;YACvC,iBAAiB,EAAE,MAAM,CAAC,iBAAiB,IAAI,IAAI;SACpD,CAAC;IACJ,CAAC;IAIS,eAAe;QACvB,OAAO,WAAW,CAAC;YACjB,EAAE,EAAE,IAAI,CAAC,IAAI;YACb,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK;YACxB,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY;YACtC,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU;YAClC,KAAK,EAAE,IAAI,CAAC,QAAQ,EAAa;SAClC,CAAC,CAAC;IACL,CAAC;IAES,iBAAiB;QACzB,MAAM,eAAe,GAAG,IAAI,uBAAuB,CACjD,IAAI,CAAC,MAAM,CAAC,KAAK,EACjB,IAAI,aAAa,CAAC,EAAE,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC,CAC1D,CAAC;QAEF,OAAO,IAAI,WAAW,CAAC;YACrB,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK;YACxB,eAAe;YACf,iBAAiB,EAAE,IAAI,iBAAiB,EAAE;YAC1C,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC,CAAE,IAAI,CAAC,YAAwC,CAAC,CAAC,CAAC,SAAS;SACzG,CAAC,CAAC;IACL,CAAC;IAES,QAAQ;QAChB,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;IAC3B,CAAC;IAES,UAAU;QAClB,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;QAChC,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;YAChD,OAAQ,KAA8B,CAAC,OAAO,IAAI,SAAS,CAAC;QAC9D,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAES,QAAQ;QAChB,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;IAC3B,CAAC;IAEQ,KAAK,CAAC,aAAa,CAC1B,QAA6C,EAC7C,QAAwC;QAExC,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC;QAC/B,MAAM,WAAW,GAAG,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAElD,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,OAAO,IAAI,QAAQ,CAAC,YAAY,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC;QACrD,CAAC;QAED,MAAM,QAAQ,GAAG,oBAAoB,CAAC,QAAQ,CAAC,CAAC;QAChD,MAAM,WAAW,GAAG,uBAAuB,CAAC,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;QAElE,IAAI,IAAI,CAAC,MAAM,CAAC,iBAAiB,EAAE,CAAC;YAClC,MAAM,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;QAC7D,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC;YAC7C,KAAK,EAAE,QAAQ;YACf,OAAO,EAAE,WAAW;SACrB,CAAC,CAAC;QAEH,MAAM,OAAO,GAAI,MAAqC,CAAC,OAAO,CAAC;QAC/D,IAAI,OAAO,EAAE,CAAC;YACZ,IAAI,CAAC,sBAAsB,CAAC,OAAO,CAAC,CAAC;QACvC,CAAC;QAED,OAAO,wBAAwB,CAAC,MAAM,CAAC,UAAgC,EAAE,QAAQ,CAAC,CAAC;IACrF,CAAC;IAES,sBAAsB,CAAC,WAAyB;QACxD,IAAI,CAAC,QAAQ,CAAC;YACZ,GAAG,IAAI,CAAC,KAAK;YACb,aAAa,EAAE,WAAW,CAAC,SAAS;YACpC,aAAa,EAAE,WAAW,CAAC,aAAa;YACxC,YAAY,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SAC9B,CAAC,CAAC;IACd,CAAC;IAEQ,KAAK,CAAC,SAAS,CAAC,OAAgB;QACvC,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAEjC,IAAI,GAAG,CAAC,QAAQ,KAAK,SAAS,EAAE,CAAC;YAC/B,OAAO,QAAQ,CAAC,IAAI,CAAC;gBACnB,MAAM,EAAE,SAAS;gBACjB,KAAK,EAAE,IAAI,CAAC,IAAI;gBAChB,KAAK,EAAE,IAAI,CAAC,UAAU,EAAE;gBACxB,YAAY,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY;aACtC,CAAC,CAAC;QACL,CAAC;QAED,OAAO,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IAClC,CAAC;IAES,kBAAkB;QAC1B,OAAO;YACL,YAAY,EAAE;gBACZ,WAAW,EAAE,uCAAuC;gBACpD,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC;oBACpB,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;oBACvC,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;oBACpB,IAAI,EAAE,CAAC,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE;iBACzB,CAAC;gBACF,OAAO,EAAE,KAAK,EAAE,IAA6B,EAAE,EAAE;oBAC/C,MAAM,IAAI,GAAG,IAAI,CAAC,IAA8B,CAAC;oBACjD,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAkB,CAAC;oBACzC,MAAM,IAAI,GAAG,IAAI,CAAC,IAA2C,CAAC;oBAE9D,IAAI,YAAY,GAA2B,IAAI,CAAC;oBAChD,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;wBACpD,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,CAAC;wBAC5B,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC;4BAC3B,YAAY,GAAG,IAAI,CAAC;wBACtB,CAAC;oBACH,CAAC;oBAED,kEAAkE;oBAClE,MAAM,MAAM,GAAG,MAAO,IAAqL,CAAC,QAAQ,CAAC,YAAY,EAAE,QAAQ,EAAE,IAAI,CAAC,CAAC;oBACnP,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC;gBAClD,CAAC;aACF;YAED,UAAU,EAAE;gBACV,WAAW,EAAE,+BAA+B;gBAC5C,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC;oBACpB,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;iBACnB,CAAC;gBACF,OAAO,EAAE,KAAK,EAAE,IAA6B,EAAE,EAAE;oBAC/C,MAAM,MAAM,GAAG,IAAI,CAAC,MAAgB,CAAC;oBACrC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;oBAClD,OAAO,EAAE,OAAO,EAAE,CAAC;gBACrB,CAAC;aACF;YAED,YAAY,EAAE;gBACZ,WAAW,EAAE,mCAAmC;gBAChD,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;gBACzB,OAAO,EAAE,KAAK,IAAI,EAAE;oBAClB,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC;oBACtC,OAAO,EAAE,SAAS,EAAE,CAAC;gBACvB,CAAC;aACF;SACF,CAAC;IACJ,CAAC;IAED,OAAO;QACL,IAAI,IAAI,CAAC,MAAM,CAAC,iBAAiB,EAAE,CAAC;YAClC,IAAI,CAAC,YAAY,CAAC,UAAU,EAAE,CAAC;QACjC,CAAC;QAED,IAAI,IAAI,CAAC,KAAK,CAAC,aAAa,IAAI,IAAI,CAAC,MAAM,CAAC,iBAAiB,EAAE,CAAC;YAC9D,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,KAAK,CAAC,CAAC,KAAY,EAAE,EAAE;gBAC7E,OAAO,CAAC,KAAK,CAAC,2BAA2B,EAAE,KAAK,CAAC,CAAC;YACpD,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,SAAS;IACT,CAAC;CACF;AAED,MAAM,UAAU,qBAAqB,CACnC,MAA6B;IAE7B,OAAO,KAAM,SAAQ,eAAoB;QACvC,YAAY,GAAiB,EAAE,GAAQ;YACrC,KAAK,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QAClB,CAAC;QAES,SAAS;YACjB,OAAO,MAAM,CAAC;QAChB,CAAC;KACF,CAAC;AACJ,CAAC"}

package/dist/CloudflareSessionStore.d.ts

@@ -0,0 +1,37 @@
+/**
+ * Cloudflare Session Store Implementation
+ *
+ * Provides session persistence using Cloudflare's built-in SQLite storage.
+ */
+import type { Session } from "@projectaria/aria-agents";
+import type { Agent } from "agents";
+export interface SessionStore {
+    get(id: string): Promise<{
+        session: Session;
+    } | undefined>;
+    create(session: Session): Promise<void>;
+    update(session: Session): Promise<void>;
+    delete(id: string): Promise<void>;
+    list(limit?: number, offset?: number): Promise<Array<{
+        session: Session;
+    }>>;
+    count(): Promise<number>;
+}
+export declare class CloudflareSessionStore {
+    private static readonly TABLE_NAME;
+    private readonly agent;
+    private initialized;
+    constructor(agent: Agent);
+    initialize(): void;
+    get(id: string): Promise<{
+        session: Session;
+    } | undefined>;
+    create(session: Session): Promise<void>;
+    update(session: Session): Promise<void>;
+    delete(id: string): Promise<void>;
+    list(limit?: number, offset?: number): Promise<Array<{
+        session: Session;
+    }>>;
+    count(): Promise<number>;
+}
+//# sourceMappingURL=CloudflareSessionStore.d.ts.map

package/dist/CloudflareSessionStore.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"CloudflareSessionStore.d.ts","sourceRoot":"","sources":["../src/CloudflareSessionStore.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,0BAA0B,CAAC;AACxD,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,QAAQ,CAAC;AAEpC,MAAM,WAAW,YAAY;IAC3B,GAAG,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC;QAAE,OAAO,EAAE,OAAO,CAAA;KAAE,GAAG,SAAS,CAAC,CAAC;IAC3D,MAAM,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACxC,MAAM,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACxC,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAClC,IAAI,CAAC,KAAK,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC;QAAE,OAAO,EAAE,OAAO,CAAA;KAAE,CAAC,CAAC,CAAC;IAC5E,KAAK,IAAI,OAAO,CAAC,MAAM,CAAC,CAAC;CAC1B;AAED,qBAAa,sBAAsB;IACjC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAmB;IACrD,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAQ;IAC9B,OAAO,CAAC,WAAW,CAAS;gBAEhB,KAAK,EAAE,KAAK;IAIxB,UAAU,IAAI,IAAI;IAoBZ,GAAG,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC;QAAE,OAAO,EAAE,OAAO,CAAA;KAAE,GAAG,SAAS,CAAC;IAuB1D,MAAM,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC;IASvC,MAAM,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC;IAWvC,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IASjC,IAAI,CACR,KAAK,GAAE,MAAY,EACnB,MAAM,GAAE,MAAU,GACjB,OAAO,CAAC,KAAK,CAAC;QAAE,OAAO,EAAE,OAAO,CAAA;KAAE,CAAC,CAAC;IAwBjC,KAAK,IAAI,OAAO,CAAC,MAAM,CAAC;CAU/B"}

package/dist/CloudflareSessionStore.js

@@ -0,0 +1,101 @@
+/**
+ * Cloudflare Session Store Implementation
+ *
+ * Provides session persistence using Cloudflare's built-in SQLite storage.
+ */
+export class CloudflareSessionStore {
+    static TABLE_NAME = "aria_sessions";
+    agent;
+    initialized = false;
+    constructor(agent) {
+        this.agent = agent;
+    }
+    initialize() {
+        if (this.initialized)
+            return;
+        this.agent.sql `
+      CREATE TABLE IF NOT EXISTS ${CloudflareSessionStore.TABLE_NAME} (
+        session_id TEXT PRIMARY KEY,
+        data TEXT NOT NULL,
+        created_at INTEGER DEFAULT (unixepoch()),
+        updated_at INTEGER DEFAULT (unixepoch())
+      )
+    `;
+        this.agent.sql `
+      CREATE INDEX IF NOT EXISTS idx_${CloudflareSessionStore.TABLE_NAME}_updated
+      ON ${CloudflareSessionStore.TABLE_NAME}(updated_at)
+    `;
+        this.initialized = true;
+    }
+    async get(id) {
+        this.initialize();
+        const results = this.agent.sql `
+      SELECT data FROM ${CloudflareSessionStore.TABLE_NAME}
+      WHERE session_id = ${id}
+    `;
+        const firstResult = results[0];
+        if (!firstResult) {
+            return undefined;
+        }
+        try {
+            const session = JSON.parse(firstResult.data);
+            return { session };
+        }
+        catch (error) {
+            console.error(`Failed to parse session ${id}:`, error);
+            return undefined;
+        }
+    }
+    async create(session) {
+        this.initialize();
+        this.agent.sql `
+      INSERT INTO ${CloudflareSessionStore.TABLE_NAME} (session_id, data)
+      VALUES (${session.id}, ${JSON.stringify(session)})
+    `;
+    }
+    async update(session) {
+        this.initialize();
+        this.agent.sql `
+      UPDATE ${CloudflareSessionStore.TABLE_NAME}
+      SET data = ${JSON.stringify(session)},
+          updated_at = unixepoch()
+      WHERE session_id = ${session.id}
+    `;
+    }
+    async delete(id) {
+        this.initialize();
+        this.agent.sql `
+      DELETE FROM ${CloudflareSessionStore.TABLE_NAME}
+      WHERE session_id = ${id}
+    `;
+    }
+    async list(limit = 100, offset = 0) {
+        this.initialize();
+        const results = this.agent.sql `
+      SELECT data FROM ${CloudflareSessionStore.TABLE_NAME}
+      ORDER BY updated_at DESC
+      LIMIT ${limit}
+      OFFSET ${offset}
+    `;
+        const sessions = [];
+        for (const row of results) {
+            try {
+                const session = JSON.parse(row.data);
+                sessions.push({ session });
+            }
+            catch (error) {
+                console.error("Failed to parse session:", error);
+            }
+        }
+        return sessions;
+    }
+    async count() {
+        this.initialize();
+        const results = this.agent.sql `
+      SELECT COUNT(*) as count FROM ${CloudflareSessionStore.TABLE_NAME}
+    `;
+        const firstResult = results[0];
+        return firstResult?.count ?? 0;
+    }
+}
+//# sourceMappingURL=CloudflareSessionStore.js.map

package/dist/CloudflareSessionStore.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"CloudflareSessionStore.js","sourceRoot":"","sources":["../src/CloudflareSessionStore.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAcH,MAAM,OAAO,sBAAsB;IACzB,MAAM,CAAU,UAAU,GAAG,eAAe,CAAC;IACpC,KAAK,CAAQ;IACtB,WAAW,GAAG,KAAK,CAAC;IAE5B,YAAY,KAAY;QACtB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;IAED,UAAU;QACR,IAAI,IAAI,CAAC,WAAW;YAAE,OAAO;QAE7B,IAAI,CAAC,KAAK,CAAC,GAAG,CAAA;mCACiB,sBAAsB,CAAC,UAAU;;;;;;KAM/D,CAAC;QAEF,IAAI,CAAC,KAAK,CAAC,GAAG,CAAA;uCACqB,sBAAsB,CAAC,UAAU;WAC7D,sBAAsB,CAAC,UAAU;KACvC,CAAC;QAEF,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;IAC1B,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,EAAU;QAClB,IAAI,CAAC,UAAU,EAAE,CAAC;QAElB,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAkB;yBAC3B,sBAAsB,CAAC,UAAU;2BAC/B,EAAE;KACxB,CAAC;QAEF,MAAM,WAAW,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QAE/B,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,IAAI,CAAY,CAAC;YACxD,OAAO,EAAE,OAAO,EAAE,CAAC;QACrB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,2BAA2B,EAAE,GAAG,EAAE,KAAK,CAAC,CAAC;YACvD,OAAO,SAAS,CAAC;QACnB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,OAAgB;QAC3B,IAAI,CAAC,UAAU,EAAE,CAAC;QAElB,IAAI,CAAC,KAAK,CAAC,GAAG,CAAA;oBACE,sBAAsB,CAAC,UAAU;gBACrC,OAAO,CAAC,EAAE,KAAK,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;KACjD,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,OAAgB;QAC3B,IAAI,CAAC,UAAU,EAAE,CAAC;QAElB,IAAI,CAAC,KAAK,CAAC,GAAG,CAAA;eACH,sBAAsB,CAAC,UAAU;mBAC7B,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;;2BAEf,OAAO,CAAC,EAAE;KAChC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,EAAU;QACrB,IAAI,CAAC,UAAU,EAAE,CAAC;QAElB,IAAI,CAAC,KAAK,CAAC,GAAG,CAAA;oBACE,sBAAsB,CAAC,UAAU;2BAC1B,EAAE;KACxB,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,IAAI,CACR,QAAgB,GAAG,EACnB,SAAiB,CAAC;QAElB,IAAI,CAAC,UAAU,EAAE,CAAC;QAElB,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAkB;yBAC3B,sBAAsB,CAAC,UAAU;;cAE5C,KAAK;eACJ,MAAM;KAChB,CAAC;QAEF,MAAM,QAAQ,GAAgC,EAAE,CAAC;QAEjD,KAAK,MAAM,GAAG,IAAI,OAAO,EAAE,CAAC;YAC1B,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAY,CAAC;gBAChD,QAAQ,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;YAC7B,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,0BAA0B,EAAE,KAAK,CAAC,CAAC;YACnD,CAAC;QACH,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,CAAC,UAAU,EAAE,CAAC;QAElB,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAmB;sCACf,sBAAsB,CAAC,UAAU;KAClE,CAAC;QAEF,MAAM,WAAW,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QAC/B,OAAO,WAAW,EAAE,KAAK,IAAI,CAAC,CAAC;IACjC,CAAC"}

package/dist/aria-to-ui-stream.d.ts

@@ -0,0 +1,21 @@
+/**
+ * Universal Stream Mapper: ARIA → AI SDK UI Format
+ *
+ * Converts ARIA stream parts to AI SDK UI message format.
+ */
+import type { AriaStreamPart } from "@projectaria/aria-agents";
+export interface AriaToUIStreamConfig {
+    includeMetadata?: boolean;
+    toolTypePrefix?: string;
+    preserveOriginalType?: boolean;
+    generateToolCallIds?: boolean;
+}
+export declare function ariaToUIStream(ariaStream: AsyncIterable<AriaStreamPart>, config?: AriaToUIStreamConfig): AsyncGenerator<any, void, unknown>;
+export declare function ariaPartToUIPart(part: AriaStreamPart, config?: AriaToUIStreamConfig): any;
+export declare function createARIAStreamResponse(ariaStream: AsyncIterable<AriaStreamPart>, onFinish?: (result: any) => void): Response;
+export declare function ariaHistoryToUIMessages(history: AriaStreamPart[]): any[];
+export declare function isToolCallPart(part: AriaStreamPart): boolean;
+export declare function isToolResultPart(part: AriaStreamPart): boolean;
+export declare function isTextPart(part: AriaStreamPart): boolean;
+export declare function getToolNameFromPart(part: AriaStreamPart): string | null;
+//# sourceMappingURL=aria-to-ui-stream.d.ts.map

package/dist/aria-to-ui-stream.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"aria-to-ui-stream.d.ts","sourceRoot":"","sources":["../src/aria-to-ui-stream.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAMH,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAE/D,MAAM,WAAW,oBAAoB;IACnC,eAAe,CAAC,EAAE,OAAO,CAAC;IAC1B,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,oBAAoB,CAAC,EAAE,OAAO,CAAC;IAC/B,mBAAmB,CAAC,EAAE,OAAO,CAAC;CAC/B;AASD,wBAAuB,cAAc,CACnC,UAAU,EAAE,aAAa,CAAC,cAAc,CAAC,EACzC,MAAM,GAAE,oBAAyB,GAChC,cAAc,CAAC,GAAG,EAAE,IAAI,EAAE,OAAO,CAAC,CAUpC;AAED,wBAAgB,gBAAgB,CAC9B,IAAI,EAAE,cAAc,EACpB,MAAM,GAAE,oBAAyB,GAChC,GAAG,CAkEL;AAMD,wBAAgB,wBAAwB,CACtC,UAAU,EAAE,aAAa,CAAC,cAAc,CAAC,EACzC,QAAQ,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,KAAK,IAAI,GAC/B,QAAQ,CAUV;AAED,wBAAgB,uBAAuB,CACrC,OAAO,EAAE,cAAc,EAAE,GACxB,GAAG,EAAE,CASP;AAED,wBAAgB,cAAc,CAAC,IAAI,EAAE,cAAc,GAAG,OAAO,CAE5D;AAED,wBAAgB,gBAAgB,CAAC,IAAI,EAAE,cAAc,GAAG,OAAO,CAE9D;AAED,wBAAgB,UAAU,CAAC,IAAI,EAAE,cAAc,GAAG,OAAO,CAExD;AAED,wBAAgB,mBAAmB,CAAC,IAAI,EAAE,cAAc,GAAG,MAAM,GAAG,IAAI,CAKvE"}

package/dist/aria-to-ui-stream.js

@@ -0,0 +1,118 @@
+/**
+ * Universal Stream Mapper: ARIA → AI SDK UI Format
+ *
+ * Converts ARIA stream parts to AI SDK UI message format.
+ */
+import { createUIMessageStream, createUIMessageStreamResponse, } from "ai";
+const DEFAULT_CONFIG = {
+    includeMetadata: false,
+    toolTypePrefix: "tool-",
+    preserveOriginalType: false,
+    generateToolCallIds: true,
+};
+export async function* ariaToUIStream(ariaStream, config = {}) {
+    const resolvedConfig = { ...DEFAULT_CONFIG, ...config };
+    for await (const part of ariaStream) {
+        const uiPart = ariaPartToUIPart(part, resolvedConfig);
+        if (uiPart !== null) {
+            yield uiPart;
+        }
+    }
+}
+export function ariaPartToUIPart(part, config = {}) {
+    const resolvedConfig = { ...DEFAULT_CONFIG, ...config };
+    const { toolTypePrefix, generateToolCallIds } = resolvedConfig;
+    const partType = part.type;
+    if (partType === "text-delta") {
+        return {
+            type: "text",
+            text: part.text || "",
+        };
+    }
+    if (partType === "text-start" || partType === "text-end") {
+        return null;
+    }
+    if (partType === "tool-call") {
+        const toolCallId = part.toolCallId ?? (generateToolCallIds ? generateToolCallId() : `tc-${Date.now()}`);
+        return {
+            type: `${toolTypePrefix}${part.toolName}`,
+            toolCallId,
+            state: "input-available",
+            input: part.input ?? {},
+        };
+    }
+    if (partType === "tool-result") {
+        const toolCallId = part.toolCallId ?? (generateToolCallIds ? generateToolCallId() : `tc-${Date.now()}`);
+        const isError = part.output && typeof part.output === "object" && part.output !== null && "error" in part.output;
+        const output = part.output;
+        return {
+            type: `${toolTypePrefix}${part.toolName}`,
+            toolCallId,
+            state: isError ? "error" : "output-available",
+            input: part.input,
+            output: isError ? undefined : part.output,
+            error: isError ? String(output?.error) : undefined,
+        };
+    }
+    if (partType === "tool-error") {
+        const toolCallId = part.toolCallId ?? (generateToolCallIds ? generateToolCallId() : `tc-${Date.now()}`);
+        return {
+            type: `${toolTypePrefix}${part.toolName}`,
+            toolCallId,
+            state: "error",
+            error: part.error,
+        };
+    }
+    if (partType === "error") {
+        return {
+            type: "text",
+            text: `Error: ${part.error instanceof Error ? part.error.message : String(part.error)}`,
+        };
+    }
+    if (partType === "start-step" || partType === "start" || partType === "finish" || partType === "abort") {
+        return null;
+    }
+    return {
+        type: "text",
+        text: `[Unknown stream part: ${partType}]`,
+    };
+}
+function generateToolCallId() {
+    return `tc-${Date.now()}-${Math.random().toString(36).substring(2, 9)}`;
+}
+export function createARIAStreamResponse(ariaStream, onFinish) {
+    const stream = createUIMessageStream({
+        execute: async ({ writer }) => {
+            for await (const part of ariaToUIStream(ariaStream)) {
+                writer.write(part);
+            }
+        },
+    });
+    return createUIMessageStreamResponse(stream);
+}
+export function ariaHistoryToUIMessages(history) {
+    const messages = [];
+    for (const part of history) {
+        const uiPart = ariaPartToUIPart(part);
+        if (uiPart) {
+            messages.push(uiPart);
+        }
+    }
+    return messages;
+}
+export function isToolCallPart(part) {
+    return part.type === "tool-call";
+}
+export function isToolResultPart(part) {
+    return part.type === "tool-result";
+}
+export function isTextPart(part) {
+    return part.type === "text-delta";
+}
+export function getToolNameFromPart(part) {
+    if (part.type === "tool-call" || part.type === "tool-result" || part.type === "tool-error") {
+        return part.toolName;
+    }
+    return null;
+}
+//# sourceMappingURL=aria-to-ui-stream.js.map

package/dist/aria-to-ui-stream.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"aria-to-ui-stream.js","sourceRoot":"","sources":["../src/aria-to-ui-stream.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,qBAAqB,EACrB,6BAA6B,GAC9B,MAAM,IAAI,CAAC;AAUZ,MAAM,cAAc,GAAmC;IACrD,eAAe,EAAE,KAAK;IACtB,cAAc,EAAE,OAAO;IACvB,oBAAoB,EAAE,KAAK;IAC3B,mBAAmB,EAAE,IAAI;CAC1B,CAAC;AAEF,MAAM,CAAC,KAAK,SAAS,CAAC,CAAC,cAAc,CACnC,UAAyC,EACzC,SAA+B,EAAE;IAEjC,MAAM,cAAc,GAAG,EAAE,GAAG,cAAc,EAAE,GAAG,MAAM,EAAE,CAAC;IAExD,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;QACpC,MAAM,MAAM,GAAG,gBAAgB,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;QAEtD,IAAI,MAAM,KAAK,IAAI,EAAE,CAAC;YACpB,MAAM,MAAM,CAAC;QACf,CAAC;IACH,CAAC;AACH,CAAC;AAED,MAAM,UAAU,gBAAgB,CAC9B,IAAoB,EACpB,SAA+B,EAAE;IAEjC,MAAM,cAAc,GAAG,EAAE,GAAG,cAAc,EAAE,GAAG,MAAM,EAAE,CAAC;IACxD,MAAM,EAAE,cAAc,EAAE,mBAAmB,EAAE,GAAG,cAAc,CAAC;IAE/D,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC;IAE3B,IAAI,QAAQ,KAAK,YAAY,EAAE,CAAC;QAC9B,OAAO;YACL,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,IAAI,CAAC,IAAI,IAAI,EAAE;SACtB,CAAC;IACJ,CAAC;IAED,IAAI,QAAQ,KAAK,YAAY,IAAI,QAAQ,KAAK,UAAU,EAAE,CAAC;QACzD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,IAAI,QAAQ,KAAK,WAAW,EAAE,CAAC;QAC7B,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,IAAI,CAAC,mBAAmB,CAAC,CAAC,CAAC,kBAAkB,EAAE,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QACxG,OAAO;YACL,IAAI,EAAE,GAAG,cAAc,GAAG,IAAI,CAAC,QAAQ,EAAE;YACzC,UAAU;YACV,KAAK,EAAE,iBAAiB;YACxB,KAAK,EAAE,IAAI,CAAC,KAAK,IAAI,EAAE;SACxB,CAAC;IACJ,CAAC;IAED,IAAI,QAAQ,KAAK,aAAa,EAAE,CAAC;QAC/B,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,IAAI,CAAC,mBAAmB,CAAC,CAAC,CAAC,kBAAkB,EAAE,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QACxG,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,IAAI,OAAO,IAAI,CAAC,MAAM,KAAK,QAAQ,IAAI,IAAI,CAAC,MAAM,KAAK,IAAI,IAAI,OAAO,IAAI,IAAI,CAAC,MAAM,CAAC;QACjH,MAAM,MAAM,GAAG,IAAI,CAAC,MAAyC,CAAC;QAC9D,OAAO;YACL,IAAI,EAAE,GAAG,cAAc,GAAG,IAAI,CAAC,QAAQ,EAAE;YACzC,UAAU;YACV,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,kBAAkB;YAC7C,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM;YACzC,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS;SACnD,CAAC;IACJ,CAAC;IAED,IAAI,QAAQ,KAAK,YAAY,EAAE,CAAC;QAC9B,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,IAAI,CAAC,mBAAmB,CAAC,CAAC,CAAC,kBAAkB,EAAE,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QACxG,OAAO;YACL,IAAI,EAAE,GAAG,cAAc,GAAG,IAAI,CAAC,QAAQ,EAAE;YACzC,UAAU;YACV,KAAK,EAAE,OAAO;YACd,KAAK,EAAE,IAAI,CAAC,KAAK;SAClB,CAAC;IACJ,CAAC;IAED,IAAI,QAAQ,KAAK,OAAO,EAAE,CAAC;QACzB,OAAO;YACL,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,UAAU,IAAI,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE;SACxF,CAAC;IACJ,CAAC;IAED,IAAI,QAAQ,KAAK,YAAY,IAAI,QAAQ,KAAK,OAAO,IAAI,QAAQ,KAAK,QAAQ,IAAI,QAAQ,KAAK,OAAO,EAAE,CAAC;QACvG,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO;QACL,IAAI,EAAE,MAAM;QACZ,IAAI,EAAE,yBAAyB,QAAQ,GAAG;KAC3C,CAAC;AACJ,CAAC;AAED,SAAS,kBAAkB;IACzB,OAAO,MAAM,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;AAC1E,CAAC;AAED,MAAM,UAAU,wBAAwB,CACtC,UAAyC,EACzC,QAAgC;IAEhC,MAAM,MAAM,GAAG,qBAAqB,CAAC;QACnC,OAAO,EAAE,KAAK,EAAE,EAAE,MAAM,EAAmB,EAAE,EAAE;YAC7C,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,cAAc,CAAC,UAAU,CAAC,EAAE,CAAC;gBACpD,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACrB,CAAC;QACH,CAAC;KACF,CAAC,CAAC;IAEH,OAAO,6BAA6B,CAAC,MAAa,CAAC,CAAC;AACtD,CAAC;AAED,MAAM,UAAU,uBAAuB,CACrC,OAAyB;IAEzB,MAAM,QAAQ,GAAU,EAAE,CAAC;IAC3B,KAAK,MAAM,IAAI,IAAI,OAAO,EAAE,CAAC;QAC3B,MAAM,MAAM,GAAG,gBAAgB,CAAC,IAAI,CAAC,CAAC;QACtC,IAAI,MAAM,EAAE,CAAC;YACX,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACxB,CAAC;IACH,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,IAAoB;IACjD,OAAO,IAAI,CAAC,IAAI,KAAK,WAAW,CAAC;AACnC,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,IAAoB;IACnD,OAAO,IAAI,CAAC,IAAI,KAAK,aAAa,CAAC;AACrC,CAAC;AAED,MAAM,UAAU,UAAU,CAAC,IAAoB;IAC7C,OAAO,IAAI,CAAC,IAAI,KAAK,YAAY,CAAC;AACpC,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,IAAoB;IACtD,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,IAAI,IAAI,CAAC,IAAI,KAAK,aAAa,IAAI,IAAI,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;QAC3F,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC"}

package/dist/index.d.ts

@@ -1,32 +1,35 @@
 /**
  * ARIA Cloudflare Agents Integration
  *
- * Memory Primitives for Meaningful Multi-Turn Conversations:
+ * A framework for building structured, multi-turn AI conversations on Cloudflare Workers.
+ * Integrates ARIA's flow orchestration with Cloudflare's Agent system.
  *
- * 1. Session: The Current Conversation Thread
- *    - Represents a single, ongoing interaction between user and agent
- *    - Contains chronological sequence of messages and actions
- *    - Access via agent.getSession()
+ * Key Components:
+ * - AriaCFChatAgent: Base class extending Cloudflare's AIChatAgent
+ * - createAriaCFChatAgent: Factory function for creating agent subclasses
+ * - CloudflareSessionStore: SQL-based session persistence
+ * - ariaToUIStream: Universal stream mapper (ARIA → AI SDK UI format)
  *
- * 2. State: Data Within the Current Conversation
- *    - Temporary data relevant only to the current active conversation
- *    - Examples: shopping cart items, user preferences mentioned in session
- *    - Access via agent.getWorkingMemory(), agent.setWorkingMemory()
+ * Usage:
+ * ```typescript
+ * import { createAriaCFChatAgent } from "@projectaria/cf-agents";
  *
- * 3. Memory: Searchable, Cross-Session Information
- *    - Knowledge base that spans multiple past sessions
- *    - Searchable by key, type, or semantic query
- *    - Access via agent.storeMemory(), agent.searchMemories()
+ * const MyAgent = createAriaCFChatAgent({
+ *   flows: new Map([[myFlow.id, myFlow]]),
+ *   instructions: "You are a helpful assistant.",
+ *   workersAiBinding: env.AI,
+ * });
  *
- * Note: This package extends Cloudflare's AIChatAgent.
- * Users must install "agents" and "ai" packages as peer dependencies:
- *   bun add agents ai
+ * export { MyAgent as MyAgent };
+ * ```
  */
-export { AriaAgent } from "./AriaAgent.js";
-export type { AriaSession, AriaSessionState, AriaFlowState, AriaMessage, AriaMemory, MemorySearchQuery, MemorySearchResult, FlowTransition, AriaAgentConfig, AriaEnv, WorkingMemory, } from "./types.js";
-export { ariaToChatMessage, chatToAriaMessage, batchAriaToChat, batchChatToAria, extractConversationSummary, truncateMessageHistory, } from "./message-converter.js";
-export { ariaSessionToCloudflareSession, cloudflareSessionToAriaSession, convertWorkingMemory, runAriaAgentWithPersistence, extractMemoriesFromAriaAgent, buildAriaAgentPrompt, createPersistentAriaAgent, } from "./aria-agent-integration.js";
-export type { AriaAgentConfigForPersistence, } from "./aria-agent-integration.js";
+export { AriaCFChatAgent, createAriaCFChatAgent } from "./AriaCFChatAgent.js";
+export { CloudflareSessionStore } from "./CloudflareSessionStore.js";
+export { cloudflareToAriaSession, extractTextContent, extractLastUserInput, createUserMessage, createAssistantMessage, createToolResultMessage, truncateMessageHistory, isToolCall, isToolResult, extractToolCalls, extractToolResults, } from "./message-format.js";
+export { ariaToUIStream, ariaPartToUIPart, createARIAStreamResponse, ariaHistoryToUIMessages, } from "./aria-to-ui-stream.js";
+export type { AriaToUIStreamConfig, } from "./aria-to-ui-stream.js";
+export type { AriaCFChatAgentConfig, AriaCFState, AriaCFChatAgentConfigResolved, } from "./types.js";
 export { createAgent } from "@projectaria/aria-agents";
-export type { Agent, AgentConfig, AgentTextRequest, AgentTextResponse, AgentTextStreamResult, AgentObjectRequest, AgentObjectResponse, AgentObjectStreamResult, AgentSession, } from "@projectaria/aria-agents";
+export type { Agent, AgentSession, FlowDefinition, ToolRegistry, SessionStore, } from "@projectaria/aria-agents";
+export type { UIMessage, ToolSet, LanguageModelUsage, } from "ai";
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AAGH,OAAO,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAG3C,YAAY,EAEV,WAAW,EACX,gBAAgB,EAChB,aAAa,EAGb,WAAW,EAGX,UAAU,EACV,iBAAiB,EACjB,kBAAkB,EAGlB,cAAc,EAGd,eAAe,EACf,OAAO,EAGP,aAAa,GACd,MAAM,YAAY,CAAC;AAGpB,OAAO,EACL,iBAAiB,EACjB,iBAAiB,EACjB,eAAe,EACf,eAAe,EACf,0BAA0B,EAC1B,sBAAsB,GACvB,MAAM,wBAAwB,CAAC;AAGhC,OAAO,EACL,8BAA8B,EAC9B,8BAA8B,EAC9B,oBAAoB,EACpB,2BAA2B,EAC3B,4BAA4B,EAC5B,oBAAoB,EACpB,yBAAyB,GAC1B,MAAM,6BAA6B,CAAC;AAErC,YAAY,EACV,6BAA6B,GAC9B,MAAM,6BAA6B,CAAC;AAGrC,OAAO,EAAE,WAAW,EAAE,MAAM,0BAA0B,CAAC;AACvD,YAAY,EACV,KAAK,EACL,WAAW,EACX,gBAAgB,EAChB,iBAAiB,EACjB,qBAAqB,EACrB,kBAAkB,EAClB,mBAAmB,EACnB,uBAAuB,EACvB,YAAY,GACb,MAAM,0BAA0B,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;GAwBG;AAEH,OAAO,EAAE,eAAe,EAAE,qBAAqB,EAAE,MAAM,sBAAsB,CAAC;AAC9E,OAAO,EAAE,sBAAsB,EAAE,MAAM,6BAA6B,CAAC;AAErE,OAAO,EACL,uBAAuB,EACvB,kBAAkB,EAClB,oBAAoB,EACpB,iBAAiB,EACjB,sBAAsB,EACtB,uBAAuB,EACvB,sBAAsB,EACtB,UAAU,EACV,YAAY,EACZ,gBAAgB,EAChB,kBAAkB,GACnB,MAAM,qBAAqB,CAAC;AAE7B,OAAO,EACL,cAAc,EACd,gBAAgB,EAChB,wBAAwB,EACxB,uBAAuB,GACxB,MAAM,wBAAwB,CAAC;AAEhC,YAAY,EACV,oBAAoB,GACrB,MAAM,wBAAwB,CAAC;AAEhC,YAAY,EACV,qBAAqB,EACrB,WAAW,EACX,6BAA6B,GAC9B,MAAM,YAAY,CAAC;AAEpB,OAAO,EAAE,WAAW,EAAE,MAAM,0BAA0B,CAAC;AACvD,YAAY,EACV,KAAK,EACL,YAAY,EACZ,cAAc,EACd,YAAY,EACZ,YAAY,GACb,MAAM,0BAA0B,CAAC;AAElC,YAAY,EACV,SAAS,EACT,OAAO,EACP,kBAAkB,GACnB,MAAM,IAAI,CAAC"}