agents 0.12.2 → 0.12.4

package/dist/experimental/memory/session/index.d.ts

@@ -1,4 +1,25 @@
-import { A as SessionOptions, C as AgentSessionProvider, D as StoredCompaction, E as SessionProvider, O as SessionMessage, S as isSearchProvider, T as SearchResult, _ as R2SkillProvider, b as AgentSearchProvider, f as ContextBlock, g as isWritableProvider, h as WritableContextProvider, k as SessionMessagePart, m as ContextProvider, p as ContextConfig, r as CompactResult, v as SkillProvider, w as SqlProvider, x as SearchProvider, y as isSkillProvider } from "../../../compaction-helpers-D92Ipstp.js";
+import {
+  A as SessionOptions,
+  C as AgentSessionProvider,
+  D as StoredCompaction,
+  E as SessionProvider,
+  O as SessionMessage,
+  S as isSearchProvider,
+  T as SearchResult,
+  _ as R2SkillProvider,
+  b as AgentSearchProvider,
+  f as ContextBlock,
+  g as isWritableProvider,
+  h as WritableContextProvider,
+  k as SessionMessagePart,
+  m as ContextProvider,
+  p as ContextConfig,
+  r as CompactResult,
+  v as SkillProvider,
+  w as SqlProvider,
+  x as SearchProvider,
+  y as isSkillProvider
+} from "../../../compaction-helpers-bYvP1o2S.js";
 import { ToolSet } from "ai";
 
 //#region src/experimental/memory/session/session.d.ts
@@ -42,7 +63,9 @@ declare class Session {
    * Register a compaction function. Called by `compact()` to compress
    * message history into a summary overlay.
    */
-  onCompaction(fn: (messages: SessionMessage[]) => Promise<CompactResult | null>): this;
+  onCompaction(
+    fn: (messages: SessionMessage[]) => Promise<CompactResult | null>
+  ): this;
   /**
    * Auto-compact when estimated token count exceeds the threshold.
    * Checked after each `appendMessage`. Requires `onCompaction()`.
@@ -68,11 +91,18 @@ declare class Session {
   private _broadcast;
   private _emitStatus;
   private _emitError;
-  appendMessage(message: SessionMessage, parentId?: string | null): Promise<void>;
+  appendMessage(
+    message: SessionMessage,
+    parentId?: string | null
+  ): Promise<void>;
   updateMessage(message: SessionMessage): void;
   deleteMessages(messageIds: string[]): void;
   clearMessages(): void;
-  addCompaction(summary: string, fromMessageId: string, toMessageId: string): StoredCompaction;
+  addCompaction(
+    summary: string,
+    fromMessageId: string,
+    toMessageId: string
+  ): StoredCompaction;
   getCompactions(): StoredCompaction[];
   /**
    * Run the registered compaction function and store the result as an overlay.
@@ -95,7 +125,10 @@ declare class Session {
    * AgentContextProvider. Requires the session to have been created
    * via `Session.create(agent)` (not the direct constructor).
    */
-  addContext(label: string, options?: SessionContextOptions): Promise<ContextBlock>;
+  addContext(
+    label: string,
+    options?: SessionContextOptions
+  ): Promise<ContextBlock>;
   /**
    * Remove a dynamically registered context block.
    * Used during extension unload cleanup.
@@ -116,9 +149,12 @@ declare class Session {
   getLoadedSkillKeys(): Set<string>;
   freezeSystemPrompt(): Promise<string>;
   refreshSystemPrompt(): Promise<string>;
-  search(query: string, options?: {
-    limit?: number;
-  }): Array<{
+  search(
+    query: string,
+    options?: {
+      limit?: number;
+    }
+  ): Array<{
     id: string;
     role: string;
     content: string;
@@ -178,7 +214,9 @@ declare class SessionManager {
    * Register a compaction function propagated to all sessions.
    * Called by `Session.compact()` to compress message history.
    */
-  onCompaction(fn: (messages: SessionMessage[]) => Promise<CompactResult | null>): this;
+  onCompaction(
+    fn: (messages: SessionMessage[]) => Promise<CompactResult | null>
+  ): this;
   /**
    * Auto-compact when estimated token count exceeds the threshold.
    * Propagated to all sessions. Requires `onCompaction()`.
@@ -205,18 +243,33 @@ declare class SessionManager {
   private _createHistoryProvider;
   /** Get or create the Session instance for a session ID. */
   getSession(sessionId: string): Session;
-  create(name: string, opts?: {
-    parentSessionId?: string;
-    model?: string;
-    source?: string;
-  }): SessionInfo;
+  create(
+    name: string,
+    opts?: {
+      parentSessionId?: string;
+      model?: string;
+      source?: string;
+    }
+  ): SessionInfo;
   get(sessionId: string): SessionInfo | null;
   list(): SessionInfo[];
   delete(sessionId: string): void;
   rename(sessionId: string, name: string): void;
-  append(sessionId: string, message: SessionMessage, parentId?: string): Promise<string>;
-  upsert(sessionId: string, message: SessionMessage, parentId?: string): Promise<string>;
-  appendAll(sessionId: string, messages: SessionMessage[], parentId?: string): Promise<string | null>;
+  append(
+    sessionId: string,
+    message: SessionMessage,
+    parentId?: string
+  ): Promise<string>;
+  upsert(
+    sessionId: string,
+    message: SessionMessage,
+    parentId?: string
+  ): Promise<string>;
+  appendAll(
+    sessionId: string,
+    messages: SessionMessage[],
+    parentId?: string
+  ): Promise<string | null>;
   getHistory(sessionId: string, leafId?: string): SessionMessage[];
   getMessageCount(sessionId: string): number;
   clearMessages(sessionId: string): void;
@@ -226,14 +279,35 @@ declare class SessionManager {
    * Fork a session at a specific message, creating a new session
    * with the history up to that point copied over.
    */
-  fork(sessionId: string, atMessageId: string, newName: string): Promise<SessionInfo>;
-  addCompaction(sessionId: string, summary: string, fromId: string, toId: string): StoredCompaction;
+  fork(
+    sessionId: string,
+    atMessageId: string,
+    newName: string
+  ): Promise<SessionInfo>;
+  addCompaction(
+    sessionId: string,
+    summary: string,
+    fromId: string,
+    toId: string
+  ): StoredCompaction;
   getCompactions(sessionId: string): StoredCompaction[];
-  compactAndSplit(sessionId: string, summary: string, newName?: string): Promise<SessionInfo>;
-  addUsage(sessionId: string, inputTokens: number, outputTokens: number, cost: number): void;
-  search(query: string, options?: {
-    limit?: number;
-  }): {
+  compactAndSplit(
+    sessionId: string,
+    summary: string,
+    newName?: string
+  ): Promise<SessionInfo>;
+  addUsage(
+    sessionId: string,
+    inputTokens: number,
+    outputTokens: number,
+    cost: number
+  ): void;
+  search(
+    query: string,
+    options?: {
+      limit?: number;
+    }
+  ): {
     id: string;
     role: string;
     content: string;
@@ -255,5 +329,31 @@ declare class AgentContextProvider implements WritableContextProvider {
   set(content: string): Promise<void>;
 }
 //#endregion
-export { AgentContextProvider, AgentSearchProvider, AgentSessionProvider, type ContextBlock, type ContextConfig, type ContextProvider, R2SkillProvider, type SearchProvider, type SearchResult, Session, type SessionContextOptions, type SessionInfo, SessionManager, type SessionManagerOptions, type SessionMessage, type SessionMessagePart, type SessionOptions, type SessionProvider, type SkillProvider, type SqlProvider, type StoredCompaction, type WritableContextProvider, isSearchProvider, isSkillProvider, isWritableProvider };
-//# sourceMappingURL=index.d.ts.map
+export {
+  AgentContextProvider,
+  AgentSearchProvider,
+  AgentSessionProvider,
+  type ContextBlock,
+  type ContextConfig,
+  type ContextProvider,
+  R2SkillProvider,
+  type SearchProvider,
+  type SearchResult,
+  Session,
+  type SessionContextOptions,
+  type SessionInfo,
+  SessionManager,
+  type SessionManagerOptions,
+  type SessionMessage,
+  type SessionMessagePart,
+  type SessionOptions,
+  type SessionProvider,
+  type SkillProvider,
+  type SqlProvider,
+  type StoredCompaction,
+  type WritableContextProvider,
+  isSearchProvider,
+  isSkillProvider,
+  isWritableProvider
+};
+//# sourceMappingURL=index.d.ts.map

package/dist/experimental/memory/session/index.js

@@ -1,5 +1,5 @@
 import "../../../types.js";
-import { m as estimateStringTokens, p as estimateMessageTokens, t as COMPACTION_PREFIX } from "../../../compaction-helpers-CSaqCmdE.js";
+import { m as estimateStringTokens, p as estimateMessageTokens, t as COMPACTION_PREFIX } from "../../../compaction-helpers-DvcZnvQ1.js";
 import { z } from "zod";
 //#region src/experimental/memory/session/search.ts
 /**
@@ -131,17 +131,22 @@ function isSkillProvider(provider) {
 *
 * Descriptions are pulled from R2 custom metadata (`description` key).
 * If a prefix is provided, it is prepended on storage operations and
-* stripped from keys in metadata.
+* stripped from keys in metadata. `keys`, when provided, is matched against
+* these prefix-relative keys.
 *
 * @example
 * ```ts
-* const skills = new R2SkillProvider(env.SKILLS_BUCKET, { prefix: "skills/" });
+* const skills = new R2SkillProvider(env.SKILLS_BUCKET, {
+*   prefix: "skills/",
+*   keys: ["code-review", "debugging"]
+* });
 * ```
 */
 var R2SkillProvider = class {
 	constructor(bucket, options) {
 		this.bucket = bucket;
 		this.prefix = options?.prefix ?? "";
+		this.keys = options?.keys?.length ? new Set(options.keys) : null;
 	}
 	async get() {
 		const entries = [];
@@ -155,6 +160,7 @@ var R2SkillProvider = class {
 			});
 			for (const obj of listed.objects) {
 				const key = obj.key.slice(this.prefix.length);
+				if (!this.allowsKey(key)) continue;
 				const desc = obj.customMetadata?.description;
 				entries.push(`- ${key}${desc ? `: ${desc}` : ""}`);
 			}
@@ -164,6 +170,7 @@ var R2SkillProvider = class {
 		return entries.length > 0 ? entries.join("\n") : null;
 	}
 	async load(key) {
+		if (!this.allowsKey(key)) return null;
 		const obj = await this.bucket.get(this.prefix + key);
 		if (!obj) return null;
 		return obj.text();
@@ -171,6 +178,9 @@ var R2SkillProvider = class {
 	async set(key, content, description) {
 		await this.bucket.put(this.prefix + key, content, { customMetadata: description ? { description } : void 0 });
 	}
+	allowsKey(key) {
+		return this.keys === null || this.keys.has(key);
+	}
 };
 //#endregion
 //#region src/experimental/memory/session/context.ts