@aigne/core 1.72.0 → 1.74.0-beta

package/dist/prompt/agent-session.d.cts

@@ -0,0 +1,167 @@
+import { ChatModelInputMessage } from "../agents/chat-model.cjs";
+import { AgentInvokeOptions } from "../agents/agent.cjs";
+import { CompactConfig, CompactContent, Compactor, CompactorInput, DEFAULT_COMPACT_ASYNC, DEFAULT_COMPACT_MODE, DEFAULT_KEEP_RECENT_RATIO, DEFAULT_MAX_TOKENS, DEFAULT_MEMORY_QUERY_LIMIT, DEFAULT_MEMORY_RATIO, DEFAULT_SESSION_MEMORY_ASYNC, DEFAULT_SESSION_MEMORY_MODE, DEFAULT_SESSION_MODE, DEFAULT_USER_MEMORY_ASYNC, DEFAULT_USER_MEMORY_MODE, EntryContent, MemoryExtractor, MemoryExtractorInput, MemoryExtractorOutput, MemoryFact, SessionMemoryConfig, SessionMemoryMetadata, SessionMode, UserMemoryConfig, UserMemoryExtractor, UserMemoryExtractorInput, UserMemoryExtractorOutput } from "./compact/types.cjs";
+import { AFS } from "@aigne/afs";
+
+//#region src/prompt/agent-session.d.ts
+interface AgentSessionOptions {
+  sessionId: string;
+  userId?: string;
+  agentId?: string;
+  afs?: AFS;
+  /**
+       * Session mode
+       * - "auto": Enable history recording, compaction, and memory extraction
+       * - "disabled": Disable all session features (history, compaction, memory)
+       *
+       * **Should be "disabled" for internal utility agents** (extractors, compactors, etc.)
+       * to avoid recursive memory extraction and unnecessary overhead.
+       *
+       * @default DEFAULT_SESSION_MODE ("auto")
+       */
+  mode?: SessionMode;
+  /**
+       * Compaction configuration
+       */
+  compact?: CompactConfig;
+  /**
+       * Session memory configuration
+       */
+  sessionMemory?: SessionMemoryConfig;
+  /**
+       * User memory configuration
+       */
+  userMemory?: UserMemoryConfig;
+}
+declare class AgentSession {
+  readonly sessionId: string;
+  readonly userId?: string;
+  readonly agentId?: string;
+  private afs?;
+  private historyModulePath?;
+  private mode;
+  private compactConfig;
+  private sessionMemoryConfig;
+  private userMemoryConfig;
+  private runtimeState;
+  private initialized?;
+  private compactionPromise?;
+  private sessionMemoryUpdatePromise?;
+  private userMemoryUpdatePromise?;
+  constructor(options: AgentSessionOptions);
+  /**
+       * Check if memory extraction is enabled
+       * Memory extraction requires mode to be "auto" AND AFS history module to be available
+       */
+  private get isMemoryEnabled();
+  setSystemMessages(...messages: ChatModelInputMessage[]): Promise<void>;
+  getMessages(): Promise<ChatModelInputMessage[]>;
+  /**
+       * Format user memory facts into a system message
+       * Applies token budget limit to ensure memory injection fits within constraints
+       */
+  private formatUserMemory;
+  /**
+       * Format session memory facts into a system message
+       * Applies token budget limit to ensure memory injection fits within constraints
+       */
+  private formatSessionMemory;
+  private formatMemoryTemplate;
+  startMessage(input: unknown, message: ChatModelInputMessage, options: AgentInvokeOptions): Promise<void>;
+  endMessage(output: unknown, message: ChatModelInputMessage | undefined, options: AgentInvokeOptions): Promise<void>;
+  /**
+       * Manually trigger compaction
+       */
+  compact(options: AgentInvokeOptions): Promise<void>;
+  /**
+       * Internal method that performs the actual compaction
+       */
+  private doCompact;
+  private compactCurrentEntry;
+  private maybeCompactCurrentEntry;
+  private maybeAutoCompact;
+  /**
+       * Estimate token count for messages
+       * Applies singleMessageLimit to each text block individually
+       * Non-text tokens (images, tool calls) are always counted in full
+       */
+  private estimateMessagesTokens;
+  /**
+       * Split entries into batches based on token limit
+       * Each batch will not exceed the specified maxTokens
+       */
+  private splitIntoBatches;
+  appendCurrentMessages(messages: ChatModelInputMessage | ChatModelInputMessage[], options: AgentInvokeOptions): Promise<void>;
+  /**
+       * Truncate text content to fit within target token limit
+       * @param text The text to truncate
+       * @param currentTokens Current token count of the text
+       * @param targetTokens Target token count after truncation
+       * @returns Truncated text
+       */
+  private truncateText;
+  private truncateLargeMessage;
+  private ensureInitialized;
+  private initialize;
+  /**
+       * Load session memory facts
+       * @returns Array of memory fact entries for the current session
+       */
+  private loadSessionMemory;
+  /**
+       * Load user memory facts
+       * @returns Array of memory fact entries for the current user
+       */
+  private loadUserMemory;
+  /**
+       * Load session history including compact content and history entries
+       * @returns Object containing history compact and history entries
+       */
+  private loadSessionHistory;
+  /**
+       * Manually trigger session memory update
+       */
+  updateSessionMemory(options: AgentInvokeOptions): Promise<void>;
+  private maybeAutoUpdateSessionMemory;
+  private maybeAutoUpdateUserMemory;
+  /**
+       * Internal method that performs the actual session memory update
+       */
+  private doUpdateSessionMemory;
+  /**
+       * Manually trigger user memory update
+       */
+  updateUserMemory(options: AgentInvokeOptions): Promise<void>;
+  /**
+       * Internal method that performs the actual user memory extraction
+       */
+  private doUpdateUserMemory;
+  /**
+       * Find Agent Skill content from a single message
+       * @param msg - Message to search in
+       * @returns The skill content text if found, undefined otherwise
+       */
+  private findSkillContentInMessage;
+  /**
+       * Find the last Agent Skill from a list of messages
+       * @param messages - Messages to search through
+       * @returns The last Agent Skill found, or undefined if none found
+       */
+  private findLastAgentSkillFromMessages;
+  /**
+       * Find the last Agent Skill from a list of history entries
+       * @param entries - History entries to search through
+       * @returns The last Agent Skill found, or undefined if none found
+       */
+  private findLastAgentSkill;
+  private initializeDefaultCompactor;
+  private initializeDefaultSessionMemoryExtractor;
+  private initializeDefaultUserMemoryExtractor;
+  private get maxTokens();
+  private get keepRecentRatio();
+  private get keepRecentTokens();
+  private get singleMessageLimit();
+}
+//#endregion
+export { AgentSession, AgentSessionOptions };
+//# sourceMappingURL=agent-session.d.cts.map

package/dist/prompt/agent-session.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-session.d.cts","names":[],"sources":["../../src/prompt/agent-session.ts"],"mappings":";;;;;;UAiCiB,mBAAA;EAAA,SAAA;EAAA,MAAA;EAAA,OAAA;EAAA,GAAA,GAIT,GAAA;EAAA;;;;;;AA0CR;;;;EA1CQ,IAAA,GAYC,WAAA;EAAA;;;EAAA,OAAA,GAKG,aAAA;EAAA;;;EAAA,aAAA,GAKM,mBAAA;EAAA;;;EAAA,UAAA,GAKH,gBAAA;AAAA;AAAA,cAeF,YAAA;EAAA,SAAA,SAAA;EAAA,SAAA,MAAA;EAAA,SAAA,OAAA;EAAA,QAAA,GAAA;EAAA,QAAA,iBAAA;EAAA,QAAA,IAAA;EAAA,QAAA,aAAA;EAAA,QAAA,mBAAA;EAAA,QAAA,gBAAA;EAAA,QAAA,YAAA;EAAA,QAAA,WAAA;EAAA,QAAA,iBAAA;EAAA,QAAA,0BAAA;EAAA,QAAA,uBAAA;EAAA,YAAA,OAAA,EAiBU,mBAAA;EAAA;;;;EAAA,YAAA,gBAAA;EAAA,kBAAA,GAAA,QAAA,EAwBgB,qBAAA,KAA0B,OAAA;EAAA,YAAA,GAM1C,OAAA,CAAQ,qBAAA;EAAA;;;;EAAA,QAAA,gBAAA;EAAA;;;;EAAA,QAAA,mBAAA;EAAA,QAAA,oBAAA;EAAA,aAAA,KAAA,WAAA,OAAA,EA8KlB,qBAAA,EAAA,OAAA,EACA,kBAAA,GACR,OAAA;EAAA,WAAA,MAAA,WAAA,OAAA,EAkBQ,qBAAA,cAAA,OAAA,EACA,kBAAA,GACR,OAAA;EAAA;;;EAAA,QAAA,OAAA,EAyDoB,kBAAA,GAAqB,OAAA;EAAA;;;EAAA,QAAA,SAAA;EAAA,QAAA,mBAAA;EAAA,QAAA,wBAAA;EAAA,QAAA,gBAAA;EAAA;;;;;EAAA,QAAA,sBAAA;EAAA;;;;EAAA,QAAA,gBAAA;EAAA,sBAAA,QAAA,EA2UhC,qBAAA,GAAwB,qBAAA,IAAA,OAAA,EACzB,kBAAA,GACR,OAAA;EAAA;;;;;;;EAAA,QAAA,YAAA;EAAA,QAAA,oBAAA;EAAA,QAAA,iBAAA;EAAA,QAAA,UAAA;EAAA;;;;EAAA,QAAA,iBAAA;EAAA;;;;EAAA,QAAA,cAAA;EAAA;;;;EAAA,QAAA,kBAAA;EAAA;;;EAAA,oBAAA,OAAA,EA0OgC,kBAAA,GAAqB,OAAA;EAAA,QAAA,4BAAA;EAAA,QAAA,yBAAA;EAAA;;;EAAA,QAAA,qBAAA;EAAA;;;EAAA,iBAAA,OAAA,EA6KxB,kBAAA,GAAqB,OAAA;EAAA;;;EAAA,QAAA,kBAAA;EAAA;;;;;EAAA,QAAA,yBAAA;EAAA;;;;;EAAA,QAAA,8BAAA;EAAA;;;;;EAAA,QAAA,kBAAA;EAAA,QAAA,0BAAA;EAAA,QAAA,uCAAA;EAAA,QAAA,oCAAA;EAAA,YAAA,UAAA;EAAA,YAAA,gBAAA;EAAA,YAAA,iBAAA;EAAA,YAAA,mBAAA;AAAA"}

package/dist/prompt/agent-session.d.mts

@@ -0,0 +1,167 @@
+import { ChatModelInputMessage } from "../agents/chat-model.mjs";
+import { AgentInvokeOptions } from "../agents/agent.mjs";
+import { CompactConfig, CompactContent, Compactor, CompactorInput, DEFAULT_COMPACT_ASYNC, DEFAULT_COMPACT_MODE, DEFAULT_KEEP_RECENT_RATIO, DEFAULT_MAX_TOKENS, DEFAULT_MEMORY_QUERY_LIMIT, DEFAULT_MEMORY_RATIO, DEFAULT_SESSION_MEMORY_ASYNC, DEFAULT_SESSION_MEMORY_MODE, DEFAULT_SESSION_MODE, DEFAULT_USER_MEMORY_ASYNC, DEFAULT_USER_MEMORY_MODE, EntryContent, MemoryExtractor, MemoryExtractorInput, MemoryExtractorOutput, MemoryFact, SessionMemoryConfig, SessionMemoryMetadata, SessionMode, UserMemoryConfig, UserMemoryExtractor, UserMemoryExtractorInput, UserMemoryExtractorOutput } from "./compact/types.mjs";
+import { AFS } from "@aigne/afs";
+
+//#region src/prompt/agent-session.d.ts
+interface AgentSessionOptions {
+  sessionId: string;
+  userId?: string;
+  agentId?: string;
+  afs?: AFS;
+  /**
+       * Session mode
+       * - "auto": Enable history recording, compaction, and memory extraction
+       * - "disabled": Disable all session features (history, compaction, memory)
+       *
+       * **Should be "disabled" for internal utility agents** (extractors, compactors, etc.)
+       * to avoid recursive memory extraction and unnecessary overhead.
+       *
+       * @default DEFAULT_SESSION_MODE ("auto")
+       */
+  mode?: SessionMode;
+  /**
+       * Compaction configuration
+       */
+  compact?: CompactConfig;
+  /**
+       * Session memory configuration
+       */
+  sessionMemory?: SessionMemoryConfig;
+  /**
+       * User memory configuration
+       */
+  userMemory?: UserMemoryConfig;
+}
+declare class AgentSession {
+  readonly sessionId: string;
+  readonly userId?: string;
+  readonly agentId?: string;
+  private afs?;
+  private historyModulePath?;
+  private mode;
+  private compactConfig;
+  private sessionMemoryConfig;
+  private userMemoryConfig;
+  private runtimeState;
+  private initialized?;
+  private compactionPromise?;
+  private sessionMemoryUpdatePromise?;
+  private userMemoryUpdatePromise?;
+  constructor(options: AgentSessionOptions);
+  /**
+       * Check if memory extraction is enabled
+       * Memory extraction requires mode to be "auto" AND AFS history module to be available
+       */
+  private get isMemoryEnabled();
+  setSystemMessages(...messages: ChatModelInputMessage[]): Promise<void>;
+  getMessages(): Promise<ChatModelInputMessage[]>;
+  /**
+       * Format user memory facts into a system message
+       * Applies token budget limit to ensure memory injection fits within constraints
+       */
+  private formatUserMemory;
+  /**
+       * Format session memory facts into a system message
+       * Applies token budget limit to ensure memory injection fits within constraints
+       */
+  private formatSessionMemory;
+  private formatMemoryTemplate;
+  startMessage(input: unknown, message: ChatModelInputMessage, options: AgentInvokeOptions): Promise<void>;
+  endMessage(output: unknown, message: ChatModelInputMessage | undefined, options: AgentInvokeOptions): Promise<void>;
+  /**
+       * Manually trigger compaction
+       */
+  compact(options: AgentInvokeOptions): Promise<void>;
+  /**
+       * Internal method that performs the actual compaction
+       */
+  private doCompact;
+  private compactCurrentEntry;
+  private maybeCompactCurrentEntry;
+  private maybeAutoCompact;
+  /**
+       * Estimate token count for messages
+       * Applies singleMessageLimit to each text block individually
+       * Non-text tokens (images, tool calls) are always counted in full
+       */
+  private estimateMessagesTokens;
+  /**
+       * Split entries into batches based on token limit
+       * Each batch will not exceed the specified maxTokens
+       */
+  private splitIntoBatches;
+  appendCurrentMessages(messages: ChatModelInputMessage | ChatModelInputMessage[], options: AgentInvokeOptions): Promise<void>;
+  /**
+       * Truncate text content to fit within target token limit
+       * @param text The text to truncate
+       * @param currentTokens Current token count of the text
+       * @param targetTokens Target token count after truncation
+       * @returns Truncated text
+       */
+  private truncateText;
+  private truncateLargeMessage;
+  private ensureInitialized;
+  private initialize;
+  /**
+       * Load session memory facts
+       * @returns Array of memory fact entries for the current session
+       */
+  private loadSessionMemory;
+  /**
+       * Load user memory facts
+       * @returns Array of memory fact entries for the current user
+       */
+  private loadUserMemory;
+  /**
+       * Load session history including compact content and history entries
+       * @returns Object containing history compact and history entries
+       */
+  private loadSessionHistory;
+  /**
+       * Manually trigger session memory update
+       */
+  updateSessionMemory(options: AgentInvokeOptions): Promise<void>;
+  private maybeAutoUpdateSessionMemory;
+  private maybeAutoUpdateUserMemory;
+  /**
+       * Internal method that performs the actual session memory update
+       */
+  private doUpdateSessionMemory;
+  /**
+       * Manually trigger user memory update
+       */
+  updateUserMemory(options: AgentInvokeOptions): Promise<void>;
+  /**
+       * Internal method that performs the actual user memory extraction
+       */
+  private doUpdateUserMemory;
+  /**
+       * Find Agent Skill content from a single message
+       * @param msg - Message to search in
+       * @returns The skill content text if found, undefined otherwise
+       */
+  private findSkillContentInMessage;
+  /**
+       * Find the last Agent Skill from a list of messages
+       * @param messages - Messages to search through
+       * @returns The last Agent Skill found, or undefined if none found
+       */
+  private findLastAgentSkillFromMessages;
+  /**
+       * Find the last Agent Skill from a list of history entries
+       * @param entries - History entries to search through
+       * @returns The last Agent Skill found, or undefined if none found
+       */
+  private findLastAgentSkill;
+  private initializeDefaultCompactor;
+  private initializeDefaultSessionMemoryExtractor;
+  private initializeDefaultUserMemoryExtractor;
+  private get maxTokens();
+  private get keepRecentRatio();
+  private get keepRecentTokens();
+  private get singleMessageLimit();
+}
+//#endregion
+export { AgentSession, AgentSessionOptions };
+//# sourceMappingURL=agent-session.d.mts.map

package/dist/prompt/agent-session.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-session.d.mts","names":[],"sources":["../../src/prompt/agent-session.ts"],"mappings":";;;;;;UAiCiB,mBAAA;EAAA,SAAA;EAAA,MAAA;EAAA,OAAA;EAAA,GAAA,GAIT,GAAA;EAAA;;;;;;AA0CR;;;;EA1CQ,IAAA,GAYC,WAAA;EAAA;;;EAAA,OAAA,GAKG,aAAA;EAAA;;;EAAA,aAAA,GAKM,mBAAA;EAAA;;;EAAA,UAAA,GAKH,gBAAA;AAAA;AAAA,cAeF,YAAA;EAAA,SAAA,SAAA;EAAA,SAAA,MAAA;EAAA,SAAA,OAAA;EAAA,QAAA,GAAA;EAAA,QAAA,iBAAA;EAAA,QAAA,IAAA;EAAA,QAAA,aAAA;EAAA,QAAA,mBAAA;EAAA,QAAA,gBAAA;EAAA,QAAA,YAAA;EAAA,QAAA,WAAA;EAAA,QAAA,iBAAA;EAAA,QAAA,0BAAA;EAAA,QAAA,uBAAA;EAAA,YAAA,OAAA,EAiBU,mBAAA;EAAA;;;;EAAA,YAAA,gBAAA;EAAA,kBAAA,GAAA,QAAA,EAwBgB,qBAAA,KAA0B,OAAA;EAAA,YAAA,GAM1C,OAAA,CAAQ,qBAAA;EAAA;;;;EAAA,QAAA,gBAAA;EAAA;;;;EAAA,QAAA,mBAAA;EAAA,QAAA,oBAAA;EAAA,aAAA,KAAA,WAAA,OAAA,EA8KlB,qBAAA,EAAA,OAAA,EACA,kBAAA,GACR,OAAA;EAAA,WAAA,MAAA,WAAA,OAAA,EAkBQ,qBAAA,cAAA,OAAA,EACA,kBAAA,GACR,OAAA;EAAA;;;EAAA,QAAA,OAAA,EAyDoB,kBAAA,GAAqB,OAAA;EAAA;;;EAAA,QAAA,SAAA;EAAA,QAAA,mBAAA;EAAA,QAAA,wBAAA;EAAA,QAAA,gBAAA;EAAA;;;;;EAAA,QAAA,sBAAA;EAAA;;;;EAAA,QAAA,gBAAA;EAAA,sBAAA,QAAA,EA2UhC,qBAAA,GAAwB,qBAAA,IAAA,OAAA,EACzB,kBAAA,GACR,OAAA;EAAA;;;;;;;EAAA,QAAA,YAAA;EAAA,QAAA,oBAAA;EAAA,QAAA,iBAAA;EAAA,QAAA,UAAA;EAAA;;;;EAAA,QAAA,iBAAA;EAAA;;;;EAAA,QAAA,cAAA;EAAA;;;;EAAA,QAAA,kBAAA;EAAA;;;EAAA,oBAAA,OAAA,EA0OgC,kBAAA,GAAqB,OAAA;EAAA,QAAA,4BAAA;EAAA,QAAA,yBAAA;EAAA;;;EAAA,QAAA,qBAAA;EAAA;;;EAAA,iBAAA,OAAA,EA6KxB,kBAAA,GAAqB,OAAA;EAAA;;;EAAA,QAAA,kBAAA;EAAA;;;;;EAAA,QAAA,yBAAA;EAAA;;;;;EAAA,QAAA,8BAAA;EAAA;;;;;EAAA,QAAA,kBAAA;EAAA,QAAA,0BAAA;EAAA,QAAA,uCAAA;EAAA,QAAA,oCAAA;EAAA,YAAA,UAAA;EAAA,YAAA,gBAAA;EAAA,YAAA,iBAAA;EAAA,YAAA,mBAAA;AAAA"}