@economic/agents 0.0.1-alpha.22 → 0.0.1-alpha.24

package/README.md

@@ -3,7 +3,7 @@
 Base class and utilities for building LLM chat agents on Cloudflare's Agents SDK with lazy skill loading, optional message compaction, and built-in audit logging.
 
 ```bash
-npm install @economic/agents ai @cloudflare/ai-chat
+npm install @economic/agents ai @cloudflare/ai-chat agents
 ```
 
 ---

package/dist/index.d.mts

@@ -1,5 +1,6 @@
 import { AIChatAgent as AIChatAgent$1, OnChatMessageOptions } from "@cloudflare/ai-chat";
 import { LanguageModel, ToolSet, UIMessage, generateText, streamText } from "ai";
+import { Connection, ConnectionContext } from "agents";
 
 //#region src/features/skills/index.d.ts
 /**
@@ -79,9 +80,9 @@ declare function buildLLMParams(config: BuildLLMParamsConfig): Promise<LLMParams
  */
 declare abstract class AIChatAgent<Env extends Cloudflare.Env = Cloudflare.Env> extends AIChatAgent$1<Env> {
   /**
-   * Composed user identifier extracted from `options.body.userId` during
-   * `buildLLMParams`. Expected format: `{agreementNumber}_{userId}`, e.g. `148583_matt`.
-   * Undefined if the client did not include `userId` in the request body.
+   * Composed user identifier extracted from the query params on the initial connection.
+   * Expected format: `{agreementNumber}_{userId}`, e.g. `148583_matt`.
+   * Undefined if the client did not include `userId`.
    */
   protected _userId: string | undefined;
   /**
@@ -100,9 +101,14 @@ declare abstract class AIChatAgent<Env extends Cloudflare.Env = Cloudflare.Env>
   /**
    * Resolves the D1 database binding and userId required for all D1 writes.
    * Returns null and silently no-ops if AGENT_DB is not bound.
-   * Returns null and logs an error if userId is missing from the request body.
+   * Returns null and logs an error if userId is missing.
    */
   private resolveD1Context;
+  onConnect(connection: Connection, ctx: ConnectionContext): Promise<void>;
+  /**
+   * Returns all conversations for the current user.
+   */
+  getConversations(): Promise<Record<string, unknown>[] | undefined>;
   /**
    * Writes an audit event to D1 if `AGENT_DB` is bound on the environment,
    * otherwise silently does nothing.
@@ -147,7 +153,7 @@ declare abstract class AIChatAgent<Env extends Cloudflare.Env = Cloudflare.Env>
    * return streamText(params).toUIMessageStreamResponse();
    * ```
    */
-  protected buildLLMParams(config: Omit<BuildLLMParamsConfig, "messages" | "activeSkills" | "fastModel">): ReturnType<typeof buildLLMParams>;
+  protected buildLLMParams<TBody = Record<string, unknown>>(config: Omit<BuildLLMParamsConfig, "messages" | "activeSkills" | "fastModel">): ReturnType<typeof buildLLMParams>;
   /**
    * Skill names persisted from previous turns, read from DO SQLite.
    * Returns an empty array if no skills have been loaded yet.
@@ -172,49 +178,6 @@ declare abstract class AIChatAgent<Env extends Cloudflare.Env = Cloudflare.Env>
   }): Promise<void>;
 }
 //#endregion
-//#region src/decorators/guard.d.ts
-/**
- * Function run before a guarded handler (e.g. `onChatMessage`). Receives the
- * custom request body from chat options (`options.body`, same shape as
- * `OnChatMessageOptions` from `@cloudflare/ai-chat`).
- *
- * Return a {@link Response} to short-circuit and skip the decorated method.
- * Return nothing (or `undefined`) to allow the method to run.
- *
- * Typical uses include auth, rate limits, or feature flags — all logic lives here;
- * the `guard` decorator only forwards `body` and handles the return shape.
- */
-type GuardFn<TBody = Record<string, unknown>> = (body: TBody) => Response | void | Promise<Response | void>;
-/**
- * Method decorator (TypeScript 5+ stage-3) that runs `guardFn` with the second
- * argument's `body` (the chat request body). If `guardFn` returns a
- * {@link Response}, that value is returned and the original method is not called.
- *
- * Intended for `onChatMessage(onFinish, options?)` on subclasses of
- * `AIChatAgent`; `options` is read as `{ body?: TBody }`.
- *
- * @param guardFn - Called with `options?.body` (cast to `TBody`) before the method body.
- *
- * @example
- * ```ts
- * interface RequestBody { token: string }
- *
- * const requireToken: GuardFn<RequestBody> = async (body) => {
- *   if (!await isValidToken(body?.token)) {
- *     return new Response("Unauthorized", { status: 401 });
- *   }
- * };
- *
- * class MyAgent extends AIChatAgent<Env> {
- *   @guard(requireToken)
- *   async onChatMessage(onFinish, options) {
- *     // ...
- *   }
- * }
- * ```
- */
-declare function guard<TBody = Record<string, unknown>>(guardFn: GuardFn<TBody>): (target: (...args: any[]) => Promise<Response>, _context: ClassMethodDecoratorContext) => (this: unknown, ...args: unknown[]) => Promise<Response>;
-//#endregion
 //#region src/types.d.ts
 /**
  * The context object available throughout an agent's lifetime — passed via
@@ -223,12 +186,12 @@ declare function guard<TBody = Record<string, unknown>>(guardFn: GuardFn<TBody>)
  *
  * Define your own body shape and compose:
  * ```typescript
- * interface MyBody { userId: string; userTier: "free" | "pro" }
- * type MyContext = AgentContext<MyBody>;
+ * interface MyBody { userTier: "free" | "pro" }
+ * type MyContext = AgentToolContext<MyBody>;
  * ```
  */
-type AgentContext<TBody = Record<string, unknown>> = TBody & {
+type AgentToolContext<TBody = Record<string, unknown>> = TBody & {
   log: (message: string, payload?: Record<string, unknown>) => void | Promise<void>;
 };
 //#endregion
-export { AIChatAgent, type AgentContext, type BuildLLMParamsConfig, type GuardFn, type Skill, buildLLMParams, guard };
+export { AIChatAgent, type AgentToolContext, type BuildLLMParamsConfig, type Skill, buildLLMParams };