@projectaria/cf-agents 0.1.1 → 0.1.3

package/dist/index.js

@@ -1,33 +1,31 @@
 /**
  * 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 };
+ * ```
  */
-// Main exports
-export { AriaAgent } from "./AriaAgent.js";
-// Message conversion utilities
-export { ariaToChatMessage, chatToAriaMessage, batchAriaToChat, batchChatToAria, extractConversationSummary, truncateMessageHistory, } from "./message-converter.js";
-// ARIA Agent integration - use createAgent() with persistence
-export { ariaSessionToCloudflareSession, cloudflareSessionToAriaSession, convertWorkingMemory, runAriaAgentWithPersistence, extractMemoriesFromAriaAgent, buildAriaAgentPrompt, createPersistentAriaAgent, } from "./aria-agent-integration.js";
-// Re-export ARIA's createAgent for convenience
+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 { createAgent } from "@projectaria/aria-agents";
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AAEH,eAAe;AACf,OAAO,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AA4B3C,+BAA+B;AAC/B,OAAO,EACL,iBAAiB,EACjB,iBAAiB,EACjB,eAAe,EACf,eAAe,EACf,0BAA0B,EAC1B,sBAAsB,GACvB,MAAM,wBAAwB,CAAC;AAEhC,8DAA8D;AAC9D,OAAO,EACL,8BAA8B,EAC9B,8BAA8B,EAC9B,oBAAoB,EACpB,2BAA2B,EAC3B,4BAA4B,EAC5B,oBAAoB,EACpB,yBAAyB,GAC1B,MAAM,6BAA6B,CAAC;AAMrC,+CAA+C;AAC/C,OAAO,EAAE,WAAW,EAAE,MAAM,0BAA0B,CAAC"}
+{"version":3,"file":"index.js","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;AAYhC,OAAO,EAAE,WAAW,EAAE,MAAM,0BAA0B,CAAC"}

package/dist/message-format.d.ts

@@ -0,0 +1,39 @@
+/**
+ * Message Format Conversion: Cloudflare ↔ ARIA
+ */
+import type { UIMessage } from "ai";
+import type { AgentSession } from "@projectaria/aria-agents";
+import type { AriaCFState } from "./types";
+export declare function cloudflareToAriaSession(messages: UIMessage[], state: AriaCFState): AgentSession;
+export declare function extractTextContent(message: UIMessage): string;
+export declare function extractLastUserMessage(messages: UIMessage[]): UIMessage | null;
+export declare function extractLastUserInput(messages: UIMessage[]): string;
+export declare function createUserMessage(text: string): UIMessage;
+export declare function createAssistantMessage(text: string): UIMessage;
+export declare function createToolResultMessage(toolName: string, toolCallId: string, output: unknown): UIMessage;
+export declare function truncateMessageHistory(messages: UIMessage[], maxMessages: number): UIMessage[];
+export declare function isToolCall(message: UIMessage): boolean;
+export declare function isToolResult(message: UIMessage): boolean;
+export declare function extractToolCalls(message: UIMessage): Array<{
+    toolName: string;
+    toolCallId: string;
+    input: unknown;
+}>;
+export declare function extractToolResults(message: UIMessage): Array<{
+    toolName: string;
+    toolCallId: string;
+    output: unknown;
+}>;
+export declare function countMessages(messages: UIMessage[]): number;
+export declare function countUserMessages(messages: UIMessage[]): number;
+export declare function hasToolCalls(messages: UIMessage[]): boolean;
+export declare function getConversationSummary(messages: UIMessage[]): {
+    totalMessages: number;
+    userMessages: number;
+    assistantMessages: number;
+    toolCalls: number;
+    hasTools: boolean;
+    lastMessageRole: string | null;
+    lastMessageText: string;
+};
+//# sourceMappingURL=message-format.d.ts.map

package/dist/message-format.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"message-format.d.ts","sourceRoot":"","sources":["../src/message-format.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AACpC,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AAC7D,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,SAAS,CAAC;AAE3C,wBAAgB,uBAAuB,CACrC,QAAQ,EAAE,SAAS,EAAE,EACrB,KAAK,EAAE,WAAW,GACjB,YAAY,CAad;AAMD,wBAAgB,kBAAkB,CAAC,OAAO,EAAE,SAAS,GAAG,MAAM,CAK7D;AAED,wBAAgB,sBAAsB,CAAC,QAAQ,EAAE,SAAS,EAAE,GAAG,SAAS,GAAG,IAAI,CAO9E;AAED,wBAAgB,oBAAoB,CAAC,QAAQ,EAAE,SAAS,EAAE,GAAG,MAAM,CAGlE;AAED,wBAAgB,iBAAiB,CAAC,IAAI,EAAE,MAAM,GAAG,SAAS,CAOzD;AAED,wBAAgB,sBAAsB,CAAC,IAAI,EAAE,MAAM,GAAG,SAAS,CAO9D;AAED,wBAAgB,uBAAuB,CACrC,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,MAAM,EAClB,MAAM,EAAE,OAAO,GACd,SAAS,CAeX;AAED,wBAAgB,sBAAsB,CACpC,QAAQ,EAAE,SAAS,EAAE,EACrB,WAAW,EAAE,MAAM,GAClB,SAAS,EAAE,CAKb;AAED,wBAAgB,UAAU,CAAC,OAAO,EAAE,SAAS,GAAG,OAAO,CAItD;AAED,wBAAgB,YAAY,CAAC,OAAO,EAAE,SAAS,GAAG,OAAO,CAExD;AAED,wBAAgB,gBAAgB,CAC9B,OAAO,EAAE,SAAS,GACjB,KAAK,CAAC;IAAE,QAAQ,EAAE,MAAM,CAAC;IAAC,UAAU,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,OAAO,CAAA;CAAE,CAAC,CAUjE;AAED,wBAAgB,kBAAkB,CAChC,OAAO,EAAE,SAAS,GACjB,KAAK,CAAC;IAAE,QAAQ,EAAE,MAAM,CAAC;IAAC,UAAU,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,OAAO,CAAA;CAAE,CAAC,CAQlE;AAED,wBAAgB,aAAa,CAAC,QAAQ,EAAE,SAAS,EAAE,GAAG,MAAM,CAE3D;AAED,wBAAgB,iBAAiB,CAAC,QAAQ,EAAE,SAAS,EAAE,GAAG,MAAM,CAE/D;AAED,wBAAgB,YAAY,CAAC,QAAQ,EAAE,SAAS,EAAE,GAAG,OAAO,CAE3D;AAED,wBAAgB,sBAAsB,CAAC,QAAQ,EAAE,SAAS,EAAE,GAAG;IAC7D,aAAa,EAAE,MAAM,CAAC;IACtB,YAAY,EAAE,MAAM,CAAC;IACrB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,EAAE,OAAO,CAAC;IAClB,eAAe,EAAE,MAAM,GAAG,IAAI,CAAC;IAC/B,eAAe,EAAE,MAAM,CAAC;CACzB,CAeA"}

package/dist/message-format.js

@@ -0,0 +1,124 @@
+/**
+ * Message Format Conversion: Cloudflare ↔ ARIA
+ */
+export function cloudflareToAriaSession(messages, state) {
+    const now = new Date().toISOString();
+    const sessionId = state.ariaSessionId ?? generateSessionId();
+    return {
+        sessionId,
+        conversationId: sessionId,
+        history: messages,
+        workingMemory: state.workingMemory,
+        context: new Map(),
+        createdAt: now,
+        updatedAt: now,
+    };
+}
+function generateSessionId() {
+    return `aria-${Date.now()}-${Math.random().toString(36).substring(2, 9)}`;
+}
+export function extractTextContent(message) {
+    return message.parts
+        .filter((p) => p.type === "text")
+        .map((p) => p.text)
+        .join("");
+}
+export function extractLastUserMessage(messages) {
+    for (let i = messages.length - 1; i >= 0; i--) {
+        if (messages[i].role === "user") {
+            return messages[i];
+        }
+    }
+    return null;
+}
+export function extractLastUserInput(messages) {
+    const lastUser = extractLastUserMessage(messages);
+    return lastUser ? extractTextContent(lastUser) : "";
+}
+export function createUserMessage(text) {
+    return {
+        id: `msg-${Date.now()}-${Math.random().toString(36).substring(2, 9)}`,
+        role: "user",
+        parts: [{ type: "text", text }],
+        metadata: { createdAt: new Date().toISOString() },
+    };
+}
+export function createAssistantMessage(text) {
+    return {
+        id: `msg-${Date.now()}-${Math.random().toString(36).substring(2, 9)}`,
+        role: "assistant",
+        parts: [{ type: "text", text }],
+        metadata: { createdAt: new Date().toISOString() },
+    };
+}
+export function createToolResultMessage(toolName, toolCallId, output) {
+    return {
+        id: `msg-${Date.now()}-${Math.random().toString(36).substring(2, 9)}`,
+        role: "assistant",
+        parts: [
+            {
+                type: `tool-${toolName}`,
+                toolCallId,
+                state: "output-available",
+                input: {},
+                output,
+            },
+        ],
+        metadata: { createdAt: new Date().toISOString() },
+    };
+}
+export function truncateMessageHistory(messages, maxMessages) {
+    if (messages.length <= maxMessages) {
+        return messages;
+    }
+    return messages.slice(-maxMessages);
+}
+export function isToolCall(message) {
+    return message.parts.some((p) => p.type === "tool-call" || (typeof p.type === "string" && p.type.startsWith("tool-")));
+}
+export function isToolResult(message) {
+    return message.parts.some((p) => p.type === "tool-result" || (typeof p.type === "string" && p.type.startsWith("tool-") && "output" in p));
+}
+export function extractToolCalls(message) {
+    return message.parts
+        .filter((p) => p.type === "tool-call" || (typeof p.type === "string" && p.type.startsWith("tool-")))
+        .map((p) => ({
+        toolName: p.toolName || (typeof p.type === "string" && p.type.startsWith("tool-") ? p.type.slice(5) : "unknown"),
+        toolCallId: p.toolCallId ?? `tc-${Date.now()}-${Math.random().toString(36).substring(2, 9)}`,
+        input: p.input ?? {},
+    }));
+}
+export function extractToolResults(message) {
+    return message.parts
+        .filter((p) => p.type === "tool-result" || (typeof p.type === "string" && p.type.startsWith("tool-") && "output" in p))
+        .map((p) => ({
+        toolName: p.toolName || (typeof p.type === "string" && p.type.startsWith("tool-") ? p.type.slice(5) : "unknown"),
+        toolCallId: p.toolCallId ?? `tc-${Date.now()}-${Math.random().toString(36).substring(2, 9)}`,
+        output: p.output,
+    }));
+}
+export function countMessages(messages) {
+    return messages.length;
+}
+export function countUserMessages(messages) {
+    return messages.filter((m) => m.role === "user").length;
+}
+export function hasToolCalls(messages) {
+    return messages.some(isToolCall);
+}
+export function getConversationSummary(messages) {
+    const userMsgs = messages.filter((m) => m.role === "user");
+    const assistantMsgs = messages.filter((m) => m.role === "assistant");
+    const toolMsgs = messages.filter(isToolCall);
+    const lastMessage = messages[messages.length - 1];
+    return {
+        totalMessages: messages.length,
+        userMessages: userMsgs.length,
+        assistantMessages: assistantMsgs.length,
+        toolCalls: toolMsgs.length,
+        hasTools: toolMsgs.length > 0,
+        lastMessageRole: lastMessage?.role ?? null,
+        lastMessageText: lastMessage ? extractTextContent(lastMessage).slice(0, 100) : "",
+    };
+}
+//# sourceMappingURL=message-format.js.map

package/dist/message-format.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"message-format.js","sourceRoot":"","sources":["../src/message-format.ts"],"names":[],"mappings":"AAAA;;GAEG;AAMH,MAAM,UAAU,uBAAuB,CACrC,QAAqB,EACrB,KAAkB;IAElB,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IACrC,MAAM,SAAS,GAAG,KAAK,CAAC,aAAa,IAAI,iBAAiB,EAAE,CAAC;IAE7D,OAAO;QACL,SAAS;QACT,cAAc,EAAE,SAAS;QACzB,OAAO,EAAE,QAAQ;QACjB,aAAa,EAAE,KAAK,CAAC,aAAa;QAClC,OAAO,EAAE,IAAI,GAAG,EAAE;QAClB,SAAS,EAAE,GAAG;QACd,SAAS,EAAE,GAAG;KACf,CAAC;AACJ,CAAC;AAED,SAAS,iBAAiB;IACxB,OAAO,QAAQ,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;AAC5E,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,OAAkB;IACnD,OAAO,OAAO,CAAC,KAAK;SACjB,MAAM,CAAC,CAAC,CAAC,EAAuC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;SACrE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;SAClB,IAAI,CAAC,EAAE,CAAC,CAAC;AACd,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,QAAqB;IAC1D,KAAK,IAAI,CAAC,GAAG,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC9C,IAAI,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAChC,OAAO,QAAQ,CAAC,CAAC,CAAC,CAAC;QACrB,CAAC;IACH,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,QAAqB;IACxD,MAAM,QAAQ,GAAG,sBAAsB,CAAC,QAAQ,CAAC,CAAC;IAClD,OAAO,QAAQ,CAAC,CAAC,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;AACtD,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,IAAY;IAC5C,OAAO;QACL,EAAE,EAAE,OAAO,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE;QACrE,IAAI,EAAE,MAAM;QACZ,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;QAC/B,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE;KAClD,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,IAAY;IACjD,OAAO;QACL,EAAE,EAAE,OAAO,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE;QACrE,IAAI,EAAE,WAAW;QACjB,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;QAC/B,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE;KAClD,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,uBAAuB,CACrC,QAAgB,EAChB,UAAkB,EAClB,MAAe;IAEf,OAAO;QACL,EAAE,EAAE,OAAO,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE;QACrE,IAAI,EAAE,WAAW;QACjB,KAAK,EAAE;YACL;gBACE,IAAI,EAAE,QAAQ,QAAQ,EAAE;gBACxB,UAAU;gBACV,KAAK,EAAE,kBAA2B;gBAClC,KAAK,EAAE,EAAE;gBACT,MAAM;aACP;SACF;QACD,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE;KAClD,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,sBAAsB,CACpC,QAAqB,EACrB,WAAmB;IAEnB,IAAI,QAAQ,CAAC,MAAM,IAAI,WAAW,EAAE,CAAC;QACnC,OAAO,QAAQ,CAAC;IAClB,CAAC;IACD,OAAO,QAAQ,CAAC,KAAK,CAAC,CAAC,WAAW,CAAC,CAAC;AACtC,CAAC;AAED,MAAM,UAAU,UAAU,CAAC,OAAkB;IAC3C,OAAO,OAAO,CAAC,KAAK,CAAC,IAAI,CACvB,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,KAAK,QAAQ,IAAI,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAC5F,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,OAAkB;IAC7C,OAAO,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,aAAa,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,KAAK,QAAQ,IAAI,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,QAAQ,IAAI,CAAC,CAAC,CAAC,CAAC;AAC5I,CAAC;AAED,MAAM,UAAU,gBAAgB,CAC9B,OAAkB;IAElB,OAAO,OAAO,CAAC,KAAK;SACjB,MAAM,CAAC,CAAC,CAAC,EAAY,EAAE,CACtB,CAAC,CAAC,IAAI,KAAK,WAAW,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,KAAK,QAAQ,IAAI,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CACrF;SACA,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QACX,QAAQ,EAAE,CAAC,CAAC,QAAQ,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,KAAK,QAAQ,IAAI,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAChH,UAAU,EAAE,CAAC,CAAC,UAAU,IAAI,MAAM,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE;QAC5F,KAAK,EAAE,CAAC,CAAC,KAAK,IAAI,EAAE;KACrB,CAAC,CAAC,CAAC;AACR,CAAC;AAED,MAAM,UAAU,kBAAkB,CAChC,OAAkB;IAElB,OAAO,OAAO,CAAC,KAAK;SACjB,MAAM,CAAC,CAAC,CAAC,EAAY,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,aAAa,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,KAAK,QAAQ,IAAI,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,QAAQ,IAAI,CAAC,CAAC,CAAC;SAChI,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QACX,QAAQ,EAAE,CAAC,CAAC,QAAQ,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,KAAK,QAAQ,IAAI,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAChH,UAAU,EAAE,CAAC,CAAC,UAAU,IAAI,MAAM,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE;QAC5F,MAAM,EAAE,CAAC,CAAC,MAAM;KACjB,CAAC,CAAC,CAAC;AACR,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,QAAqB;IACjD,OAAO,QAAQ,CAAC,MAAM,CAAC;AACzB,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,QAAqB;IACrD,OAAO,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,MAAM,CAAC;AAC1D,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,QAAqB;IAChD,OAAO,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;AACnC,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,QAAqB;IAS1D,MAAM,QAAQ,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;IAC3D,MAAM,aAAa,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC;IACrE,MAAM,QAAQ,GAAG,QAAQ,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;IAC7C,MAAM,WAAW,GAAG,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAElD,OAAO;QACL,aAAa,EAAE,QAAQ,CAAC,MAAM;QAC9B,YAAY,EAAE,QAAQ,CAAC,MAAM;QAC7B,iBAAiB,EAAE,aAAa,CAAC,MAAM;QACvC,SAAS,EAAE,QAAQ,CAAC,MAAM;QAC1B,QAAQ,EAAE,QAAQ,CAAC,MAAM,GAAG,CAAC;QAC7B,eAAe,EAAE,WAAW,EAAE,IAAI,IAAI,IAAI;QAC1C,eAAe,EAAE,WAAW,CAAC,CAAC,CAAC,kBAAkB,CAAC,WAAW,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE;KAClF,CAAC;AACJ,CAAC"}

package/dist/types.d.ts

@@ -1,158 +1,71 @@
 /**
- * Core types for ARIA Cloudflare Agents integration
- *
- * Memory Primitives:
- * - Session: Current conversation thread with chronological messages
- * - State: Temporary data within the current conversation
- * - Memory: Searchable, cross-session information
+ * Core type definitions for @projectaria/cf-agents
  */
-import type { WorkingMemory } from "@projectaria/aria-agents";
-export type { UIMessage } from "ai";
-/**
- * Session represents a single, ongoing interaction between a user and the agent.
- * Contains the chronological sequence of messages and actions during that specific interaction.
- */
-export interface AriaSession {
-    /** Unique session identifier (agent name) */
-    id: string;
-    /** Chronological message history */
-    messages: AriaMessage[];
-    /** Current active flow */
-    activeFlow?: AriaFlowState;
-    /** Message history limit (default: 50) */
-    historyLimit: number;
-    /** Session timestamp */
-    createdAt: number;
-    updatedAt: number;
+import type { FlowDefinition, ToolRegistry } from "@projectaria/aria-agents";
+import type { LanguageModel } from "ai";
+export interface AriaCFState {
+    ariaSessionId: string | null;
+    workingMemory: Record<string, unknown>;
+    agentConfig: {
+        id: string;
+        name: string;
+        modelId: string;
+    } | null;
+    lastActivity: string | null;
+    activeFlow: {
+        flowId: string;
+        nodeId: string;
+    } | null;
 }
-/**
- * State (session.state): Data within the current conversation
- * Used to manage information relevant only to the current active conversation.
- * Examples: shopping cart items, user preferences mentioned in this session.
- */
-export interface AriaSessionState {
-    /** Working memory - temporary data for current conversation */
-    workingMemory: WorkingMemory;
-    /** Current flow being executed */
-    activeFlowId?: string;
-    /** Current node within the flow */
-    activeNodeId?: string;
-    /** Message history limit */
-    historyLimit: number;
-    /** Session metadata */
-    metadata?: Record<string, unknown>;
+export declare const ARIA_CF_INITIAL_STATE: AriaCFState;
+export interface AriaCFChatAgentConfig {
+    flows: Map<string, FlowDefinition>;
+    model: LanguageModel;
+    instructions?: string;
+    guidelines?: string[];
+    tools?: ToolRegistry;
+    historyLimit?: number;
+    enablePersistence?: boolean;
 }
-/**
- * Flow state tracking
- */
-export interface AriaFlowState {
-    flowId: string;
-    currentNodeId: string;
-    enteredAt: number;
+export type AriaCFChatAgentConfigResolved = Required<AriaCFChatAgentConfig>;
+export interface AriaSessionData {
+    sessionId: string;
+    history: import("ai").UIMessage[];
+    workingMemory: Record<string, unknown>;
+    context: Map<string, unknown>;
+    activeFlow: {
+        flowId: string;
+        nodeId: string;
+    } | null;
+    createdAt: string;
+    updatedAt: string;
 }
-/**
- * Message format compatible with both ARIA and AI SDK
- */
-export interface AriaMessage {
-    role: "system" | "user" | "assistant" | "tool";
-    content: string;
-    timestamp: number;
-    /** Optional tool call information */
-    toolCall?: {
-        toolName: string;
-        toolCallId: string;
-        input?: Record<string, unknown>;
-        output?: unknown;
-    };
-    /** Optional metadata */
-    metadata?: Record<string, unknown>;
+export interface SchedulingToolInput {
+    when: number | string | Date;
+    callback: string;
+    data?: Record<string, unknown>;
 }
-/**
- * Flow transition record for tracking execution history
- */
-export interface FlowTransition {
-    id: string;
-    flowId: string;
-    fromNodeId: string | null;
-    toNodeId: string;
-    transitionedAt: number;
-    metadata?: Record<string, unknown>;
+export interface SchedulingToolResult {
+    taskId: string;
+    scheduledAt: string;
+    callback: string;
 }
-/**
- * Memory represents searchable, cross-session information.
- * Acts as a knowledge base the agent can search beyond the immediate conversation.
- */
-export interface AriaMemory {
-    /** Unique memory identifier */
-    id: string;
-    /** Session this memory belongs to (null if cross-session) */
-    sessionId: string | null;
-    /** Memory content/key for retrieval */
+export interface KVStorageToolInput {
+    action: "get" | "set" | "delete" | "list";
     key: string;
-    /** Memory value */
-    value: unknown;
-    /** Memory type for categorization */
-    type: "preference" | "fact" | "context" | "entity" | "custom";
-    /** Timestamp when memory was created */
-    createdAt: number;
-    /** Timestamp when memory was last accessed */
-    lastAccessedAt: number;
-    /** Access frequency for relevance scoring */
-    accessCount: number;
-    /** Optional embedding vector for semantic search */
-    embedding?: number[];
-    /** Additional metadata */
-    metadata?: Record<string, unknown>;
+    value?: unknown;
+    namespace?: string;
 }
-/**
- * Memory search query
- */
-export interface MemorySearchQuery {
-    /** Search query text */
-    query?: string;
-    /** Memory types to search */
-    types?: Array<"preference" | "fact" | "context" | "entity" | "custom">;
-    /** Session filter (null = all sessions) */
-    sessionId?: string | null;
-    /** Maximum results */
-    limit?: number;
-    /** Minimum relevance score (0-1) */
-    minScore?: number;
-}
-/**
- * Memory search result
- */
-export interface MemorySearchResult {
-    memory: AriaMemory;
-    score: number;
-    relevance: string;
+export interface SQLToolInput {
+    query: string;
+    params?: unknown[];
 }
-/**
- * Configuration for ARIA Agent
- */
-export interface AriaAgentConfig {
-    /** Maximum messages in session history */
-    historyLimit?: number;
-    /** Whether to enable cross-session memory */
-    enableMemory?: boolean;
-    /** Memory retention period (milliseconds) */
-    memoryRetentionMs?: number;
-    /** Whether to track flow transitions */
-    trackTransitions?: boolean;
-    /** Custom session ID generator */
-    generateSessionId?: () => string;
+export type { Agent, AgentSession, FlowDefinition, ToolRegistry, } from "@projectaria/aria-agents";
+export type { UIMessage, ToolSet, LanguageModelUsage, } from "ai";
+export interface AriaToUIStreamConfig {
+    includeMetadata?: boolean;
+    toolTypePrefix?: string;
+    preserveOriginalType?: boolean;
+    generateToolCallIds?: boolean;
 }
-/**
- * Agent environment bindings
- */
-export type AriaEnv = Cloudflare.Env & {
-    /** AI model API key */
-    OPENAI_API_KEY?: string;
-    /** Optional: Anthropic API key */
-    ANTHROPIC_API_KEY?: string;
-};
-/**
- * Re-export WorkingMemory from ARIA agents
- */
-export type { WorkingMemory } from "@projectaria/aria-agents";
 //# sourceMappingURL=types.d.ts.map

package/dist/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAGH,OAAO,KAAK,EAA+B,aAAa,EAAE,MAAM,0BAA0B,CAAC;AAG3F,YAAY,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAEpC;;;GAGG;AACH,MAAM,WAAW,WAAW;IAC1B,6CAA6C;IAC7C,EAAE,EAAE,MAAM,CAAC;IAEX,oCAAoC;IACpC,QAAQ,EAAE,WAAW,EAAE,CAAC;IAExB,0BAA0B;IAC1B,UAAU,CAAC,EAAE,aAAa,CAAC;IAE3B,0CAA0C;IAC1C,YAAY,EAAE,MAAM,CAAC;IAErB,wBAAwB;IACxB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED;;;;GAIG;AACH,MAAM,WAAW,gBAAgB;IAC/B,+DAA+D;IAC/D,aAAa,EAAE,aAAa,CAAC;IAE7B,kCAAkC;IAClC,YAAY,CAAC,EAAE,MAAM,CAAC;IAEtB,mCAAmC;IACnC,YAAY,CAAC,EAAE,MAAM,CAAC;IAEtB,4BAA4B;IAC5B,YAAY,EAAE,MAAM,CAAC;IAErB,uBAAuB;IACvB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC;AAED;;GAEG;AACH,MAAM,WAAW,aAAa;IAC5B,MAAM,EAAE,MAAM,CAAC;IACf,aAAa,EAAE,MAAM,CAAC;IACtB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED;;GAEG;AACH,MAAM,WAAW,WAAW;IAC1B,IAAI,EAAE,QAAQ,GAAG,MAAM,GAAG,WAAW,GAAG,MAAM,CAAC;IAC/C,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;IAElB,qCAAqC;IACrC,QAAQ,CAAC,EAAE;QACT,QAAQ,EAAE,MAAM,CAAC;QACjB,UAAU,EAAE,MAAM,CAAC;QACnB,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAChC,MAAM,CAAC,EAAE,OAAO,CAAC;KAClB,CAAC;IAEF,wBAAwB;IACxB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC;AAED;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B,EAAE,EAAE,MAAM,CAAC;IACX,MAAM,EAAE,MAAM,CAAC;IACf,UAAU,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,QAAQ,EAAE,MAAM,CAAC;IACjB,cAAc,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC;AAED;;;GAGG;AACH,MAAM,WAAW,UAAU;IACzB,+BAA+B;IAC/B,EAAE,EAAE,MAAM,CAAC;IAEX,6DAA6D;IAC7D,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;IAEzB,uCAAuC;IACvC,GAAG,EAAE,MAAM,CAAC;IAEZ,mBAAmB;IACnB,KAAK,EAAE,OAAO,CAAC;IAEf,qCAAqC;IACrC,IAAI,EAAE,YAAY,GAAG,MAAM,GAAG,SAAS,GAAG,QAAQ,GAAG,QAAQ,CAAC;IAE9D,wCAAwC;IACxC,SAAS,EAAE,MAAM,CAAC;IAElB,8CAA8C;IAC9C,cAAc,EAAE,MAAM,CAAC;IAEvB,6CAA6C;IAC7C,WAAW,EAAE,MAAM,CAAC;IAEpB,oDAAoD;IACpD,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;IAErB,0BAA0B;IAC1B,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC;AAED;;GAEG;AACH,MAAM,WAAW,iBAAiB;IAChC,wBAAwB;IACxB,KAAK,CAAC,EAAE,MAAM,CAAC;IAEf,6BAA6B;IAC7B,KAAK,CAAC,EAAE,KAAK,CAAC,YAAY,GAAG,MAAM,GAAG,SAAS,GAAG,QAAQ,GAAG,QAAQ,CAAC,CAAC;IAEvE,2CAA2C;IAC3C,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAE1B,sBAAsB;IACtB,KAAK,CAAC,EAAE,MAAM,CAAC;IAEf,oCAAoC;IACpC,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED;;GAEG;AACH,MAAM,WAAW,kBAAkB;IACjC,MAAM,EAAE,UAAU,CAAC;IACnB,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,MAAM,CAAC;CACnB;AAED;;GAEG;AACH,MAAM,WAAW,eAAe;IAC9B,0CAA0C;IAC1C,YAAY,CAAC,EAAE,MAAM,CAAC;IAEtB,6CAA6C;IAC7C,YAAY,CAAC,EAAE,OAAO,CAAC;IAEvB,6CAA6C;IAC7C,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAE3B,wCAAwC;IACxC,gBAAgB,CAAC,EAAE,OAAO,CAAC;IAE3B,kCAAkC;IAClC,iBAAiB,CAAC,EAAE,MAAM,MAAM,CAAC;CAClC;AAED;;GAEG;AACH,MAAM,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,GAAG;IACrC,uBAAuB;IACvB,cAAc,CAAC,EAAE,MAAM,CAAC;IAExB,kCAAkC;IAClC,iBAAiB,CAAC,EAAE,MAAM,CAAC;CAC5B,CAAC;AAEF;;GAEG;AACH,YAAY,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC"}
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,EAGV,cAAc,EACd,YAAY,EACb,MAAM,0BAA0B,CAAC;AAClC,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,IAAI,CAAC;AAExC,MAAM,WAAW,WAAW;IAC1B,aAAa,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,aAAa,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACvC,WAAW,EAAE;QACX,EAAE,EAAE,MAAM,CAAC;QACX,IAAI,EAAE,MAAM,CAAC;QACb,OAAO,EAAE,MAAM,CAAC;KACjB,GAAG,IAAI,CAAC;IACT,YAAY,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,UAAU,EAAE;QACV,MAAM,EAAE,MAAM,CAAC;QACf,MAAM,EAAE,MAAM,CAAC;KAChB,GAAG,IAAI,CAAC;CACV;AAED,eAAO,MAAM,qBAAqB,EAAE,WAMnC,CAAC;AAEF,MAAM,WAAW,qBAAqB;IACpC,KAAK,EAAE,GAAG,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;IACnC,KAAK,EAAE,aAAa,CAAC;IACrB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC;IACtB,KAAK,CAAC,EAAE,YAAY,CAAC;IACrB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,iBAAiB,CAAC,EAAE,OAAO,CAAC;CAC7B;AAED,MAAM,MAAM,6BAA6B,GAAG,QAAQ,CAAC,qBAAqB,CAAC,CAAC;AAE5E,MAAM,WAAW,eAAe;IAC9B,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,OAAO,IAAI,EAAE,SAAS,EAAE,CAAC;IAClC,aAAa,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACvC,OAAO,EAAE,GAAG,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAC9B,UAAU,EAAE;QACV,MAAM,EAAE,MAAM,CAAC;QACf,MAAM,EAAE,MAAM,CAAC;KAChB,GAAG,IAAI,CAAC;IACT,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,mBAAmB;IAClC,IAAI,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAAC;IAC7B,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAChC;AAED,MAAM,WAAW,oBAAoB;IACnC,MAAM,EAAE,MAAM,CAAC;IACf,WAAW,EAAE,MAAM,CAAC;IACpB,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,kBAAkB;IACjC,MAAM,EAAE,KAAK,GAAG,KAAK,GAAG,QAAQ,GAAG,MAAM,CAAC;IAC1C,GAAG,EAAE,MAAM,CAAC;IACZ,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,YAAY;IAC3B,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,CAAC,EAAE,OAAO,EAAE,CAAC;CACpB;AAED,YAAY,EACV,KAAK,EACL,YAAY,EACZ,cAAc,EACd,YAAY,GACb,MAAM,0BAA0B,CAAC;AAElC,YAAY,EACV,SAAS,EACT,OAAO,EACP,kBAAkB,GACnB,MAAM,IAAI,CAAC;AAEZ,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"}

package/dist/types.js

@@ -1,10 +1,11 @@
 /**
- * Core types for ARIA Cloudflare Agents integration
- *
- * Memory Primitives:
- * - Session: Current conversation thread with chronological messages
- * - State: Temporary data within the current conversation
- * - Memory: Searchable, cross-session information
+ * Core type definitions for @projectaria/cf-agents
  */
-export {};
+export const ARIA_CF_INITIAL_STATE = {
+    ariaSessionId: null,
+    workingMemory: {},
+    agentConfig: null,
+    lastActivity: null,
+    activeFlow: null,
+};
 //# sourceMappingURL=types.js.map

package/dist/types.js.map

@@ -1 +1 @@
-{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG"}
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA;;GAEG;AAyBH,MAAM,CAAC,MAAM,qBAAqB,GAAgB;IAChD,aAAa,EAAE,IAAI;IACnB,aAAa,EAAE,EAAE;IACjB,WAAW,EAAE,IAAI;IACjB,YAAY,EAAE,IAAI;IAClB,UAAU,EAAE,IAAI;CACjB,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@projectaria/cf-agents",
-  "version": "0.1.1",
+  "version": "0.1.3",
   "description": "ARIA Cloudflare Agents integration with proper memory primitives for meaningful multi-turn conversations",
   "type": "module",
   "main": "./dist/index.js",
@@ -28,7 +28,8 @@
   },
   "dependencies": {
     "ai": "^6.0.0",
-    "nanoid": "^3.3.7"
+    "nanoid": "^3.3.7",
+    "zod": "^4.3.3"
   },
   "devDependencies": {
     "@cloudflare/ai-chat": "^0.0.3",