@economic/agents 0.0.1-alpha.21 → 0.0.1-alpha.23

package/dist/chunk-CBBoxR_p.mjs

@@ -0,0 +1,26 @@
+import { createRequire } from "node:module";
+//#region \0rolldown/runtime.js
+var __create = Object.create;
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __commonJSMin = (cb, mod) => () => (mod || cb((mod = { exports: {} }).exports, mod), mod.exports);
+var __copyProps = (to, from, except, desc) => {
+	if (from && typeof from === "object" || typeof from === "function") for (var keys = __getOwnPropNames(from), i = 0, n = keys.length, key; i < n; i++) {
+		key = keys[i];
+		if (!__hasOwnProp.call(to, key) && key !== except) __defProp(to, key, {
+			get: ((k) => from[k]).bind(null, key),
+			enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable
+		});
+	}
+	return to;
+};
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", {
+	value: mod,
+	enumerable: true
+}) : target, mod));
+var __require = /* @__PURE__ */ createRequire(import.meta.url);
+//#endregion
+export { __require as n, __toESM as r, __commonJSMin as t };

package/dist/index.d.mts

@@ -1,5 +1,7 @@
 import { AIChatAgent as AIChatAgent$1, OnChatMessageOptions } from "@cloudflare/ai-chat";
 import { LanguageModel, ToolSet, UIMessage, generateText, streamText } from "ai";
+import { AsyncLocalStorage } from "node:async_hooks";
+import { DurableObject } from "cloudflare:workers";
 
 //#region src/features/skills/index.d.ts
 /**
@@ -65,6 +67,73 @@ type BuildLLMParamsConfig = Omit<LLMParams, "messages" | "experimental_context"
  */
 declare function buildLLMParams(config: BuildLLMParamsConfig): Promise<LLMParams>;
 //#endregion
+//#region ../../node_modules/partyserver/dist/index.d.ts
+//#region src/types.d.ts
+type ImmutablePrimitive = undefined | null | boolean | string | number;
+type Immutable<T> = T extends ImmutablePrimitive ? T : T extends Array<infer U> ? ImmutableArray<U> : T extends Map<infer K, infer V> ? ImmutableMap<K, V> : T extends Set<infer M> ? ImmutableSet<M> : ImmutableObject<T>;
+type ImmutableArray<T> = ReadonlyArray<Immutable<T>>;
+type ImmutableMap<K, V> = ReadonlyMap<Immutable<K>, Immutable<V>>;
+type ImmutableSet<T> = ReadonlySet<Immutable<T>>;
+type ImmutableObject<T> = { readonly [K in keyof T]: Immutable<T[K]> };
+type ConnectionState<T> = ImmutableObject<T> | null;
+type ConnectionSetStateFn<T> = (prevState: ConnectionState<T>) => T;
+type ConnectionContext = {
+  request: Request;
+};
+/** A WebSocket connected to the Server */
+type Connection<TState = unknown> = WebSocket & {
+  /** Connection identifier */id: string;
+  /**
+   * Arbitrary state associated with this connection.
+   * Read-only — use {@link Connection.setState} to update.
+   *
+   * This property is configurable, meaning it can be redefined via
+   * `Object.defineProperty` by downstream consumers (e.g. the Cloudflare
+   * Agents SDK) to namespace or wrap internal state storage.
+   */
+  state: ConnectionState<TState>;
+  /**
+   * Update the state associated with this connection.
+   *
+   * Accepts either a new state value or an updater function that receives
+   * the previous state and returns the next state.
+   *
+   * This property is configurable, meaning it can be redefined via
+   * `Object.defineProperty` by downstream consumers. If you redefine
+   * `state` and `setState`, you are responsible for calling
+   * `serializeAttachment` / `deserializeAttachment` yourself if you need
+   * the state to survive hibernation.
+   */
+  setState(state: TState | ConnectionSetStateFn<TState> | null): ConnectionState<TState>;
+  /**
+   * @deprecated use {@link Connection.setState} instead.
+   *
+   * Low-level method to persist data in the connection's attachment storage.
+   * This property is configurable and can be redefined by downstream
+   * consumers that need to wrap or namespace the underlying storage.
+   */
+  serializeAttachment<T = unknown>(attachment: T): void;
+  /**
+   * @deprecated use {@link Connection.state} instead.
+   *
+   * Low-level method to read data from the connection's attachment storage.
+   * This property is configurable and can be redefined by downstream
+   * consumers that need to wrap or namespace the underlying storage.
+   */
+  deserializeAttachment<T = unknown>(): T | null;
+  /**
+   * Tags assigned to this connection via {@link Server.getConnectionTags}.
+   * Always includes the connection id as the first tag.
+   */
+  tags: readonly string[];
+  /**
+   * @deprecated Use `this.name` on the Server instead.
+   * The server name. Populated from `Server.name` after initialization.
+   */
+  server: string;
+}; //#endregion
+//#region src/index.d.ts
+//#endregion
 //#region src/agents/AIChatAgent.d.ts
 /**
  * Base class for Cloudflare Agents SDK chat agents with lazy skill loading
@@ -79,9 +148,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 +169,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 +221,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 +246,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 | undefined) => 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: unknown[]) => 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 +254,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 };