@iqai/adk 0.5.2 → 0.5.3

package/CHANGELOG.md

@@ -1,5 +1,14 @@
 # @iqai/adk
 
+## 0.5.3
+
+### Patch Changes
+
+- 1ec769a: fix: improve type safety across cli and adk package
+- 9ba699c: fix: state persistence
+- 4fbb724: Fix: state management
+- edfe628: Add artifact parsing and rewind functionality
+
 ## 0.5.2
 
 ### Patch Changes

package/dist/index.d.mts

@@ -101,6 +101,10 @@ declare class EventActions {
      * a compaction of events within the specified timestamp range.
      */
     compaction?: EventCompaction;
+    /**
+     * The invocation id to rewind to. This is only set for rewind event.
+     */
+    rewindBeforeInvocationId?: string;
     /**
      * Constructor for EventActions
      */
@@ -112,6 +116,7 @@ declare class EventActions {
         escalate?: boolean;
         requestedAuthConfigs?: Record<string, any>;
         compaction?: EventCompaction;
+        rewindBeforeInvocationId?: string;
     });
 }
 
@@ -4142,6 +4147,11 @@ interface EnhancedRunner<T = string, M extends boolean = false> {
         newMessage: FullMessage;
         runConfig?: RunConfig;
     }): AsyncIterable<Event>;
+    rewind(params: {
+        userId: string;
+        sessionId: string;
+        rewindBeforeInvocationId: string;
+    }): any;
     __outputSchema?: ZodSchema;
 }
 /**
@@ -4835,6 +4845,23 @@ declare namespace index$3 {
   export { index$3_BaseSessionService as BaseSessionService, index$3_DatabaseSessionService as DatabaseSessionService, type index$3_GetSessionConfig as GetSessionConfig, index$3_InMemorySessionService as InMemorySessionService, type index$3_ListSessionsResponse as ListSessionsResponse, type index$3_Session as Session, index$3_State as State, index$3_VertexAiSessionService as VertexAiSessionService, index$3_createDatabaseSessionService as createDatabaseSessionService, index$3_createMysqlSessionService as createMysqlSessionService, index$3_createPostgresSessionService as createPostgresSessionService, index$3_createSqliteSessionService as createSqliteSessionService };
 }
 
+interface ParsedArtifactUri {
+    appName: string;
+    userId: string;
+    sessionId?: string;
+    filename: string;
+    version: number;
+}
+declare function parseArtifactUri(uri: string): ParsedArtifactUri | null;
+declare function getArtifactUri(args: {
+    appName: string;
+    userId: string;
+    filename: string;
+    version: number;
+    sessionId?: string;
+}): string;
+declare function isArtifactRef(artifact: Part): boolean;
+
 declare class GcsArtifactService implements BaseArtifactService {
     private readonly bucketName;
     private readonly storageClient;
@@ -5709,6 +5736,13 @@ declare class Runner<T extends BaseAgent = BaseAgent> {
      * Gets the configured summarizer or creates a default LLM-based one.
      */
     private _getOrCreateSummarizer;
+    rewind(args: {
+        userId: string;
+        sessionId: string;
+        rewindBeforeInvocationId: string;
+    }): Promise<void>;
+    private _computeStateDeltaForRewind;
+    private _computeArtifactDeltaForRewind;
 }
 /**
  * An in-memory Runner for testing and development.
@@ -5798,4 +5832,4 @@ declare const traceLlmCall: (invocationContext: InvocationContext, eventId: stri
 
 declare const VERSION = "0.1.0";
 
-export { AF_FUNCTION_CALL_ID_PREFIX, type AfterAgentCallback, type AfterModelCallback, type AfterToolCallback, LlmAgent as Agent, AgentBuilder, type AgentBuilderConfig, type AgentBuilderWithSchema, AgentEvaluator, AgentTool, type AgentToolConfig, type AgentType, index$5 as Agents, AiSdkLlm, AnthropicLlm, ApiKeyCredential, ApiKeyScheme, AuthConfig, AuthCredential, AuthCredentialType, AuthHandler, AuthScheme, AuthSchemeType, AuthTool, type AuthToolArguments, AutoFlow, BaseAgent, type BaseAgentType, BaseCodeExecutor, type BaseCodeExecutorConfig, BaseLLMConnection, BaseLlm, BaseLlmFlow, BaseLlmRequestProcessor, BaseLlmResponseProcessor, type BaseMemoryService, BasePlanner, BaseSessionService, BaseTool, BasicAuthCredential, BearerTokenCredential, type BeforeAgentCallback, type BeforeModelCallback, type BeforeToolCallback, type BuildFunctionDeclarationOptions, type BuiltAgent, BuiltInCodeExecutor, BuiltInPlanner, CallbackContext, type CodeExecutionInput, type CodeExecutionResult, CodeExecutionUtils, CodeExecutorContext, type CreateToolConfig, type CreateToolConfigWithSchema, type CreateToolConfigWithoutSchema, DatabaseSessionService, EnhancedAuthConfig, type EnhancedRunner, type EvalCase, type EvalCaseResult, type EvalMetric, type EvalMetricResult, type EvalMetricResultPerInvocation, EvalResult, type EvalSet, type EvalSetResult, EvalStatus, type EvaluateConfig, index as Evaluation, type EvaluationResult, Evaluator, Event, EventActions, type EventCompaction, index$2 as Events, type EventsCompactionConfig, type EventsSummarizer, ExitLoopTool, type File, FileOperationsTool, FinalResponseMatchV2Evaluator, index$1 as Flows, type FullMessage, FunctionTool, GcsArtifactService, type GetSessionConfig, GetUserChoiceTool, GoogleLlm, GoogleSearch, HttpRequestTool, HttpScheme, InMemoryArtifactService, InMemoryMemoryService, InMemoryRunner, InMemorySessionService, type InstructionProvider, type IntermediateData, type Interval, type Invocation, InvocationContext, type JudgeModelOptions, LLMRegistry, LangGraphAgent, type LangGraphAgentConfig, type LangGraphNode, type ListSessionsResponse, LlmAgent, type LlmAgentConfig, LlmCallsLimitExceededError, LlmEventSummarizer, type LlmModel, type LlmModelConfig, LlmRequest, LlmResponse, LoadArtifactsTool, LoadMemoryTool, LocalEvalService, LoopAgent, type LoopAgentConfig, McpAbi, McpAtp, McpBamm, McpCoinGecko, McpCoinGeckoPro, type McpConfig, McpDiscord, McpError, McpErrorType, McpFilesystem, McpFraxlend, McpGeneric, McpIqWiki, McpMemory, McpNearAgent, McpNearIntents, McpOdos, McpSamplingHandler, type McpSamplingRequest, type McpSamplingResponse, type McpServerConfig, McpTelegram, McpToolset, type McpTransportType, McpUpbit, index$4 as Memory, type MessagePart, type MetricInfo, type MetricValueInfo, index$6 as Models, type MultiAgentResponse, OAuth2Credential, OAuth2Scheme, type OAuthFlow, type OAuthFlows, OpenAiLlm, OpenIdConnectScheme, ParallelAgent, type ParallelAgentConfig, type PerInvocationResult, PlanReActPlanner, PrebuiltMetrics, REQUEST_EUC_FUNCTION_CALL_NAME, ReadonlyContext, RougeEvaluator, RunConfig, Runner, type RunnerAskReturn, SafetyEvaluatorV1, type SamplingHandler, type SearchMemoryResponse, SequentialAgent, type SequentialAgentConfig, type Session, type SessionInput, type SessionOptions, index$3 as Sessions, type SingleAfterModelCallback, type SingleAfterToolCallback, type SingleAgentCallback, type SingleBeforeModelCallback, type SingleBeforeToolCallback, SingleFlow, State, StreamingMode, type TelemetryConfig, TelemetryService, type ThinkingConfig, type ToolConfig, ToolContext, type ToolUnion, index$7 as Tools, TrajectoryEvaluator, TransferToAgentTool, UserInteractionTool, VERSION, VertexAiRagMemoryService, VertexAiSessionService, _findFunctionCallEventIfLastEventIsFunctionResponse, adkToMcpToolType, requestProcessor$2 as agentTransferRequestProcessor, requestProcessor$6 as basicRequestProcessor, buildFunctionDeclaration, requestProcessor as codeExecutionRequestProcessor, responseProcessor as codeExecutionResponseProcessor, requestProcessor$3 as contentRequestProcessor, convertMcpToolToBaseTool, createAuthToolArguments, createBranchContextForSubAgent, createDatabaseSessionService, createFunctionTool, createMysqlSessionService, createPostgresSessionService, createSamplingHandler, createSqliteSessionService, createTool, generateAuthEvent, generateClientFunctionCallId, getLongRunningFunctionCalls, getMcpTools, handleFunctionCallsAsync, handleFunctionCallsLive, requestProcessor$5 as identityRequestProcessor, initializeTelemetry, injectSessionState, requestProcessor$4 as instructionsRequestProcessor, isEnhancedAuthConfig, jsonSchemaToDeclaration, mcpSchemaToParameters, mergeAgentRun, mergeParallelFunctionResponseEvents, newInvocationContextId, requestProcessor$1 as nlPlanningRequestProcessor, responseProcessor$1 as nlPlanningResponseProcessor, normalizeJsonSchema, populateClientFunctionCallId, registerProviders, removeClientFunctionCallId, requestProcessor$7 as requestProcessor, runCompactionForSlidingWindow, shutdownTelemetry, telemetryService, traceLlmCall, traceToolCall, tracer };
+export { AF_FUNCTION_CALL_ID_PREFIX, type AfterAgentCallback, type AfterModelCallback, type AfterToolCallback, LlmAgent as Agent, AgentBuilder, type AgentBuilderConfig, type AgentBuilderWithSchema, AgentEvaluator, AgentTool, type AgentToolConfig, type AgentType, index$5 as Agents, AiSdkLlm, AnthropicLlm, ApiKeyCredential, ApiKeyScheme, AuthConfig, AuthCredential, AuthCredentialType, AuthHandler, AuthScheme, AuthSchemeType, AuthTool, type AuthToolArguments, AutoFlow, BaseAgent, type BaseAgentType, BaseCodeExecutor, type BaseCodeExecutorConfig, BaseLLMConnection, BaseLlm, BaseLlmFlow, BaseLlmRequestProcessor, BaseLlmResponseProcessor, type BaseMemoryService, BasePlanner, BaseSessionService, BaseTool, BasicAuthCredential, BearerTokenCredential, type BeforeAgentCallback, type BeforeModelCallback, type BeforeToolCallback, type BuildFunctionDeclarationOptions, type BuiltAgent, BuiltInCodeExecutor, BuiltInPlanner, CallbackContext, type CodeExecutionInput, type CodeExecutionResult, CodeExecutionUtils, CodeExecutorContext, type CreateToolConfig, type CreateToolConfigWithSchema, type CreateToolConfigWithoutSchema, DatabaseSessionService, EnhancedAuthConfig, type EnhancedRunner, type EvalCase, type EvalCaseResult, type EvalMetric, type EvalMetricResult, type EvalMetricResultPerInvocation, EvalResult, type EvalSet, type EvalSetResult, EvalStatus, type EvaluateConfig, index as Evaluation, type EvaluationResult, Evaluator, Event, EventActions, type EventCompaction, index$2 as Events, type EventsCompactionConfig, type EventsSummarizer, ExitLoopTool, type File, FileOperationsTool, FinalResponseMatchV2Evaluator, index$1 as Flows, type FullMessage, FunctionTool, GcsArtifactService, type GetSessionConfig, GetUserChoiceTool, GoogleLlm, GoogleSearch, HttpRequestTool, HttpScheme, InMemoryArtifactService, InMemoryMemoryService, InMemoryRunner, InMemorySessionService, type InstructionProvider, type IntermediateData, type Interval, type Invocation, InvocationContext, type JudgeModelOptions, LLMRegistry, LangGraphAgent, type LangGraphAgentConfig, type LangGraphNode, type ListSessionsResponse, LlmAgent, type LlmAgentConfig, LlmCallsLimitExceededError, LlmEventSummarizer, type LlmModel, type LlmModelConfig, LlmRequest, LlmResponse, LoadArtifactsTool, LoadMemoryTool, LocalEvalService, LoopAgent, type LoopAgentConfig, McpAbi, McpAtp, McpBamm, McpCoinGecko, McpCoinGeckoPro, type McpConfig, McpDiscord, McpError, McpErrorType, McpFilesystem, McpFraxlend, McpGeneric, McpIqWiki, McpMemory, McpNearAgent, McpNearIntents, McpOdos, McpSamplingHandler, type McpSamplingRequest, type McpSamplingResponse, type McpServerConfig, McpTelegram, McpToolset, type McpTransportType, McpUpbit, index$4 as Memory, type MessagePart, type MetricInfo, type MetricValueInfo, index$6 as Models, type MultiAgentResponse, OAuth2Credential, OAuth2Scheme, type OAuthFlow, type OAuthFlows, OpenAiLlm, OpenIdConnectScheme, ParallelAgent, type ParallelAgentConfig, type ParsedArtifactUri, type PerInvocationResult, PlanReActPlanner, PrebuiltMetrics, REQUEST_EUC_FUNCTION_CALL_NAME, ReadonlyContext, RougeEvaluator, RunConfig, Runner, type RunnerAskReturn, SafetyEvaluatorV1, type SamplingHandler, type SearchMemoryResponse, SequentialAgent, type SequentialAgentConfig, type Session, type SessionInput, type SessionOptions, index$3 as Sessions, type SingleAfterModelCallback, type SingleAfterToolCallback, type SingleAgentCallback, type SingleBeforeModelCallback, type SingleBeforeToolCallback, SingleFlow, State, StreamingMode, type TelemetryConfig, TelemetryService, type ThinkingConfig, type ToolConfig, ToolContext, type ToolUnion, index$7 as Tools, TrajectoryEvaluator, TransferToAgentTool, UserInteractionTool, VERSION, VertexAiRagMemoryService, VertexAiSessionService, _findFunctionCallEventIfLastEventIsFunctionResponse, adkToMcpToolType, requestProcessor$2 as agentTransferRequestProcessor, requestProcessor$6 as basicRequestProcessor, buildFunctionDeclaration, requestProcessor as codeExecutionRequestProcessor, responseProcessor as codeExecutionResponseProcessor, requestProcessor$3 as contentRequestProcessor, convertMcpToolToBaseTool, createAuthToolArguments, createBranchContextForSubAgent, createDatabaseSessionService, createFunctionTool, createMysqlSessionService, createPostgresSessionService, createSamplingHandler, createSqliteSessionService, createTool, generateAuthEvent, generateClientFunctionCallId, getArtifactUri, getLongRunningFunctionCalls, getMcpTools, handleFunctionCallsAsync, handleFunctionCallsLive, requestProcessor$5 as identityRequestProcessor, initializeTelemetry, injectSessionState, requestProcessor$4 as instructionsRequestProcessor, isArtifactRef, isEnhancedAuthConfig, jsonSchemaToDeclaration, mcpSchemaToParameters, mergeAgentRun, mergeParallelFunctionResponseEvents, newInvocationContextId, requestProcessor$1 as nlPlanningRequestProcessor, responseProcessor$1 as nlPlanningResponseProcessor, normalizeJsonSchema, parseArtifactUri, populateClientFunctionCallId, registerProviders, removeClientFunctionCallId, requestProcessor$7 as requestProcessor, runCompactionForSlidingWindow, shutdownTelemetry, telemetryService, traceLlmCall, traceToolCall, tracer };

package/dist/index.d.ts

@@ -101,6 +101,10 @@ declare class EventActions {
      * a compaction of events within the specified timestamp range.
      */
     compaction?: EventCompaction;
+    /**
+     * The invocation id to rewind to. This is only set for rewind event.
+     */
+    rewindBeforeInvocationId?: string;
     /**
      * Constructor for EventActions
      */
@@ -112,6 +116,7 @@ declare class EventActions {
         escalate?: boolean;
         requestedAuthConfigs?: Record<string, any>;
         compaction?: EventCompaction;
+        rewindBeforeInvocationId?: string;
     });
 }
 
@@ -4142,6 +4147,11 @@ interface EnhancedRunner<T = string, M extends boolean = false> {
         newMessage: FullMessage;
         runConfig?: RunConfig;
     }): AsyncIterable<Event>;
+    rewind(params: {
+        userId: string;
+        sessionId: string;
+        rewindBeforeInvocationId: string;
+    }): any;
     __outputSchema?: ZodSchema;
 }
 /**
@@ -4835,6 +4845,23 @@ declare namespace index$3 {
   export { index$3_BaseSessionService as BaseSessionService, index$3_DatabaseSessionService as DatabaseSessionService, type index$3_GetSessionConfig as GetSessionConfig, index$3_InMemorySessionService as InMemorySessionService, type index$3_ListSessionsResponse as ListSessionsResponse, type index$3_Session as Session, index$3_State as State, index$3_VertexAiSessionService as VertexAiSessionService, index$3_createDatabaseSessionService as createDatabaseSessionService, index$3_createMysqlSessionService as createMysqlSessionService, index$3_createPostgresSessionService as createPostgresSessionService, index$3_createSqliteSessionService as createSqliteSessionService };
 }
 
+interface ParsedArtifactUri {
+    appName: string;
+    userId: string;
+    sessionId?: string;
+    filename: string;
+    version: number;
+}
+declare function parseArtifactUri(uri: string): ParsedArtifactUri | null;
+declare function getArtifactUri(args: {
+    appName: string;
+    userId: string;
+    filename: string;
+    version: number;
+    sessionId?: string;
+}): string;
+declare function isArtifactRef(artifact: Part): boolean;
+
 declare class GcsArtifactService implements BaseArtifactService {
     private readonly bucketName;
     private readonly storageClient;
@@ -5709,6 +5736,13 @@ declare class Runner<T extends BaseAgent = BaseAgent> {
      * Gets the configured summarizer or creates a default LLM-based one.
      */
     private _getOrCreateSummarizer;
+    rewind(args: {
+        userId: string;
+        sessionId: string;
+        rewindBeforeInvocationId: string;
+    }): Promise<void>;
+    private _computeStateDeltaForRewind;
+    private _computeArtifactDeltaForRewind;
 }
 /**
  * An in-memory Runner for testing and development.
@@ -5798,4 +5832,4 @@ declare const traceLlmCall: (invocationContext: InvocationContext, eventId: stri
 
 declare const VERSION = "0.1.0";
 
-export { AF_FUNCTION_CALL_ID_PREFIX, type AfterAgentCallback, type AfterModelCallback, type AfterToolCallback, LlmAgent as Agent, AgentBuilder, type AgentBuilderConfig, type AgentBuilderWithSchema, AgentEvaluator, AgentTool, type AgentToolConfig, type AgentType, index$5 as Agents, AiSdkLlm, AnthropicLlm, ApiKeyCredential, ApiKeyScheme, AuthConfig, AuthCredential, AuthCredentialType, AuthHandler, AuthScheme, AuthSchemeType, AuthTool, type AuthToolArguments, AutoFlow, BaseAgent, type BaseAgentType, BaseCodeExecutor, type BaseCodeExecutorConfig, BaseLLMConnection, BaseLlm, BaseLlmFlow, BaseLlmRequestProcessor, BaseLlmResponseProcessor, type BaseMemoryService, BasePlanner, BaseSessionService, BaseTool, BasicAuthCredential, BearerTokenCredential, type BeforeAgentCallback, type BeforeModelCallback, type BeforeToolCallback, type BuildFunctionDeclarationOptions, type BuiltAgent, BuiltInCodeExecutor, BuiltInPlanner, CallbackContext, type CodeExecutionInput, type CodeExecutionResult, CodeExecutionUtils, CodeExecutorContext, type CreateToolConfig, type CreateToolConfigWithSchema, type CreateToolConfigWithoutSchema, DatabaseSessionService, EnhancedAuthConfig, type EnhancedRunner, type EvalCase, type EvalCaseResult, type EvalMetric, type EvalMetricResult, type EvalMetricResultPerInvocation, EvalResult, type EvalSet, type EvalSetResult, EvalStatus, type EvaluateConfig, index as Evaluation, type EvaluationResult, Evaluator, Event, EventActions, type EventCompaction, index$2 as Events, type EventsCompactionConfig, type EventsSummarizer, ExitLoopTool, type File, FileOperationsTool, FinalResponseMatchV2Evaluator, index$1 as Flows, type FullMessage, FunctionTool, GcsArtifactService, type GetSessionConfig, GetUserChoiceTool, GoogleLlm, GoogleSearch, HttpRequestTool, HttpScheme, InMemoryArtifactService, InMemoryMemoryService, InMemoryRunner, InMemorySessionService, type InstructionProvider, type IntermediateData, type Interval, type Invocation, InvocationContext, type JudgeModelOptions, LLMRegistry, LangGraphAgent, type LangGraphAgentConfig, type LangGraphNode, type ListSessionsResponse, LlmAgent, type LlmAgentConfig, LlmCallsLimitExceededError, LlmEventSummarizer, type LlmModel, type LlmModelConfig, LlmRequest, LlmResponse, LoadArtifactsTool, LoadMemoryTool, LocalEvalService, LoopAgent, type LoopAgentConfig, McpAbi, McpAtp, McpBamm, McpCoinGecko, McpCoinGeckoPro, type McpConfig, McpDiscord, McpError, McpErrorType, McpFilesystem, McpFraxlend, McpGeneric, McpIqWiki, McpMemory, McpNearAgent, McpNearIntents, McpOdos, McpSamplingHandler, type McpSamplingRequest, type McpSamplingResponse, type McpServerConfig, McpTelegram, McpToolset, type McpTransportType, McpUpbit, index$4 as Memory, type MessagePart, type MetricInfo, type MetricValueInfo, index$6 as Models, type MultiAgentResponse, OAuth2Credential, OAuth2Scheme, type OAuthFlow, type OAuthFlows, OpenAiLlm, OpenIdConnectScheme, ParallelAgent, type ParallelAgentConfig, type PerInvocationResult, PlanReActPlanner, PrebuiltMetrics, REQUEST_EUC_FUNCTION_CALL_NAME, ReadonlyContext, RougeEvaluator, RunConfig, Runner, type RunnerAskReturn, SafetyEvaluatorV1, type SamplingHandler, type SearchMemoryResponse, SequentialAgent, type SequentialAgentConfig, type Session, type SessionInput, type SessionOptions, index$3 as Sessions, type SingleAfterModelCallback, type SingleAfterToolCallback, type SingleAgentCallback, type SingleBeforeModelCallback, type SingleBeforeToolCallback, SingleFlow, State, StreamingMode, type TelemetryConfig, TelemetryService, type ThinkingConfig, type ToolConfig, ToolContext, type ToolUnion, index$7 as Tools, TrajectoryEvaluator, TransferToAgentTool, UserInteractionTool, VERSION, VertexAiRagMemoryService, VertexAiSessionService, _findFunctionCallEventIfLastEventIsFunctionResponse, adkToMcpToolType, requestProcessor$2 as agentTransferRequestProcessor, requestProcessor$6 as basicRequestProcessor, buildFunctionDeclaration, requestProcessor as codeExecutionRequestProcessor, responseProcessor as codeExecutionResponseProcessor, requestProcessor$3 as contentRequestProcessor, convertMcpToolToBaseTool, createAuthToolArguments, createBranchContextForSubAgent, createDatabaseSessionService, createFunctionTool, createMysqlSessionService, createPostgresSessionService, createSamplingHandler, createSqliteSessionService, createTool, generateAuthEvent, generateClientFunctionCallId, getLongRunningFunctionCalls, getMcpTools, handleFunctionCallsAsync, handleFunctionCallsLive, requestProcessor$5 as identityRequestProcessor, initializeTelemetry, injectSessionState, requestProcessor$4 as instructionsRequestProcessor, isEnhancedAuthConfig, jsonSchemaToDeclaration, mcpSchemaToParameters, mergeAgentRun, mergeParallelFunctionResponseEvents, newInvocationContextId, requestProcessor$1 as nlPlanningRequestProcessor, responseProcessor$1 as nlPlanningResponseProcessor, normalizeJsonSchema, populateClientFunctionCallId, registerProviders, removeClientFunctionCallId, requestProcessor$7 as requestProcessor, runCompactionForSlidingWindow, shutdownTelemetry, telemetryService, traceLlmCall, traceToolCall, tracer };
+export { AF_FUNCTION_CALL_ID_PREFIX, type AfterAgentCallback, type AfterModelCallback, type AfterToolCallback, LlmAgent as Agent, AgentBuilder, type AgentBuilderConfig, type AgentBuilderWithSchema, AgentEvaluator, AgentTool, type AgentToolConfig, type AgentType, index$5 as Agents, AiSdkLlm, AnthropicLlm, ApiKeyCredential, ApiKeyScheme, AuthConfig, AuthCredential, AuthCredentialType, AuthHandler, AuthScheme, AuthSchemeType, AuthTool, type AuthToolArguments, AutoFlow, BaseAgent, type BaseAgentType, BaseCodeExecutor, type BaseCodeExecutorConfig, BaseLLMConnection, BaseLlm, BaseLlmFlow, BaseLlmRequestProcessor, BaseLlmResponseProcessor, type BaseMemoryService, BasePlanner, BaseSessionService, BaseTool, BasicAuthCredential, BearerTokenCredential, type BeforeAgentCallback, type BeforeModelCallback, type BeforeToolCallback, type BuildFunctionDeclarationOptions, type BuiltAgent, BuiltInCodeExecutor, BuiltInPlanner, CallbackContext, type CodeExecutionInput, type CodeExecutionResult, CodeExecutionUtils, CodeExecutorContext, type CreateToolConfig, type CreateToolConfigWithSchema, type CreateToolConfigWithoutSchema, DatabaseSessionService, EnhancedAuthConfig, type EnhancedRunner, type EvalCase, type EvalCaseResult, type EvalMetric, type EvalMetricResult, type EvalMetricResultPerInvocation, EvalResult, type EvalSet, type EvalSetResult, EvalStatus, type EvaluateConfig, index as Evaluation, type EvaluationResult, Evaluator, Event, EventActions, type EventCompaction, index$2 as Events, type EventsCompactionConfig, type EventsSummarizer, ExitLoopTool, type File, FileOperationsTool, FinalResponseMatchV2Evaluator, index$1 as Flows, type FullMessage, FunctionTool, GcsArtifactService, type GetSessionConfig, GetUserChoiceTool, GoogleLlm, GoogleSearch, HttpRequestTool, HttpScheme, InMemoryArtifactService, InMemoryMemoryService, InMemoryRunner, InMemorySessionService, type InstructionProvider, type IntermediateData, type Interval, type Invocation, InvocationContext, type JudgeModelOptions, LLMRegistry, LangGraphAgent, type LangGraphAgentConfig, type LangGraphNode, type ListSessionsResponse, LlmAgent, type LlmAgentConfig, LlmCallsLimitExceededError, LlmEventSummarizer, type LlmModel, type LlmModelConfig, LlmRequest, LlmResponse, LoadArtifactsTool, LoadMemoryTool, LocalEvalService, LoopAgent, type LoopAgentConfig, McpAbi, McpAtp, McpBamm, McpCoinGecko, McpCoinGeckoPro, type McpConfig, McpDiscord, McpError, McpErrorType, McpFilesystem, McpFraxlend, McpGeneric, McpIqWiki, McpMemory, McpNearAgent, McpNearIntents, McpOdos, McpSamplingHandler, type McpSamplingRequest, type McpSamplingResponse, type McpServerConfig, McpTelegram, McpToolset, type McpTransportType, McpUpbit, index$4 as Memory, type MessagePart, type MetricInfo, type MetricValueInfo, index$6 as Models, type MultiAgentResponse, OAuth2Credential, OAuth2Scheme, type OAuthFlow, type OAuthFlows, OpenAiLlm, OpenIdConnectScheme, ParallelAgent, type ParallelAgentConfig, type ParsedArtifactUri, type PerInvocationResult, PlanReActPlanner, PrebuiltMetrics, REQUEST_EUC_FUNCTION_CALL_NAME, ReadonlyContext, RougeEvaluator, RunConfig, Runner, type RunnerAskReturn, SafetyEvaluatorV1, type SamplingHandler, type SearchMemoryResponse, SequentialAgent, type SequentialAgentConfig, type Session, type SessionInput, type SessionOptions, index$3 as Sessions, type SingleAfterModelCallback, type SingleAfterToolCallback, type SingleAgentCallback, type SingleBeforeModelCallback, type SingleBeforeToolCallback, SingleFlow, State, StreamingMode, type TelemetryConfig, TelemetryService, type ThinkingConfig, type ToolConfig, ToolContext, type ToolUnion, index$7 as Tools, TrajectoryEvaluator, TransferToAgentTool, UserInteractionTool, VERSION, VertexAiRagMemoryService, VertexAiSessionService, _findFunctionCallEventIfLastEventIsFunctionResponse, adkToMcpToolType, requestProcessor$2 as agentTransferRequestProcessor, requestProcessor$6 as basicRequestProcessor, buildFunctionDeclaration, requestProcessor as codeExecutionRequestProcessor, responseProcessor as codeExecutionResponseProcessor, requestProcessor$3 as contentRequestProcessor, convertMcpToolToBaseTool, createAuthToolArguments, createBranchContextForSubAgent, createDatabaseSessionService, createFunctionTool, createMysqlSessionService, createPostgresSessionService, createSamplingHandler, createSqliteSessionService, createTool, generateAuthEvent, generateClientFunctionCallId, getArtifactUri, getLongRunningFunctionCalls, getMcpTools, handleFunctionCallsAsync, handleFunctionCallsLive, requestProcessor$5 as identityRequestProcessor, initializeTelemetry, injectSessionState, requestProcessor$4 as instructionsRequestProcessor, isArtifactRef, isEnhancedAuthConfig, jsonSchemaToDeclaration, mcpSchemaToParameters, mergeAgentRun, mergeParallelFunctionResponseEvents, newInvocationContextId, requestProcessor$1 as nlPlanningRequestProcessor, responseProcessor$1 as nlPlanningResponseProcessor, normalizeJsonSchema, parseArtifactUri, populateClientFunctionCallId, registerProviders, removeClientFunctionCallId, requestProcessor$7 as requestProcessor, runCompactionForSlidingWindow, shutdownTelemetry, telemetryService, traceLlmCall, traceToolCall, tracer };

package/dist/index.js

@@ -3394,6 +3394,10 @@ var EventActions = (_class10 = class {
    * a compaction of events within the specified timestamp range.
    */
   
+  /**
+   * The invocation id to rewind to. This is only set for rewind event.
+   */
+  
   /**
    * Constructor for EventActions
    */
@@ -3405,6 +3409,7 @@ var EventActions = (_class10 = class {
     this.escalate = options.escalate;
     this.requestedAuthConfigs = options.requestedAuthConfigs;
     this.compaction = options.compaction;
+    this.rewindBeforeInvocationId = options.rewindBeforeInvocationId;
   }
 }, _class10);
 
@@ -8468,8 +8473,30 @@ function rearrangeEventsForLatestFunctionResponse(events) {
   return resultEvents;
 }
 function getContents(currentBranch, events, agentName = "") {
+  const invocationIdToIndex = /* @__PURE__ */ new Map();
+  for (let idx = 0; idx < events.length; idx++) {
+    if (events[idx].invocationId) {
+      invocationIdToIndex.set(events[idx].invocationId, idx);
+    }
+  }
+  const rewindFilteredEvents = [];
+  let i = events.length - 1;
+  while (i >= 0) {
+    const event = events[i];
+    if (_optionalChain([event, 'access', _245 => _245.actions, 'optionalAccess', _246 => _246.rewindBeforeInvocationId])) {
+      const rewindInvocationId = event.actions.rewindBeforeInvocationId;
+      const rewindIndex = invocationIdToIndex.get(rewindInvocationId);
+      if (rewindIndex !== void 0 && rewindIndex < i) {
+        i = rewindIndex;
+      }
+    } else {
+      rewindFilteredEvents.push(event);
+    }
+    i--;
+  }
+  rewindFilteredEvents.reverse();
   const filteredEvents = [];
-  for (const event of events) {
+  for (const event of rewindFilteredEvents) {
     if (!event.content || !event.content.role || !event.content.parts || event.content.parts.length === 0) {
       continue;
     }
@@ -8558,7 +8585,7 @@ function mergeFunctionResponseEvents(functionResponseEvents) {
   const partIndicesInMergedEvent = {};
   for (let idx = 0; idx < partsInMergedEvent.length; idx++) {
     const part = partsInMergedEvent[idx];
-    if (_optionalChain([part, 'access', _245 => _245.functionResponse, 'optionalAccess', _246 => _246.id])) {
+    if (_optionalChain([part, 'access', _247 => _247.functionResponse, 'optionalAccess', _248 => _248.id])) {
       partIndicesInMergedEvent[part.functionResponse.id] = idx;
     }
   }
@@ -8567,7 +8594,7 @@ function mergeFunctionResponseEvents(functionResponseEvents) {
       throw new Error("There should be at least one function_response part.");
     }
     for (const part of event.content.parts) {
-      if (_optionalChain([part, 'access', _247 => _247.functionResponse, 'optionalAccess', _248 => _248.id])) {
+      if (_optionalChain([part, 'access', _249 => _249.functionResponse, 'optionalAccess', _250 => _250.id])) {
         const functionCallId = part.functionResponse.id;
         if (functionCallId in partIndicesInMergedEvent) {
           partsInMergedEvent[partIndicesInMergedEvent[functionCallId]] = part;
@@ -8607,7 +8634,7 @@ function processCompactionEvents(events) {
   let lastCompactionStartTime = Number.POSITIVE_INFINITY;
   for (let i = events.length - 1; i >= 0; i--) {
     const event = events[i];
-    if (_optionalChain([event, 'access', _249 => _249.actions, 'optionalAccess', _250 => _250.compaction])) {
+    if (_optionalChain([event, 'access', _251 => _251.actions, 'optionalAccess', _252 => _252.compaction])) {
       const compaction = event.actions.compaction;
       const synthesizedEvent = new Event({
         timestamp: compaction.endTimestamp,
@@ -8859,7 +8886,7 @@ var PlanReActPlanner = class extends BasePlanner {
     let firstFcPartIndex = -1;
     for (let i = 0; i < responseParts.length; i++) {
       if (responseParts[i].functionCall) {
-        if (!_optionalChain([responseParts, 'access', _251 => _251[i], 'access', _252 => _252.functionCall, 'optionalAccess', _253 => _253.name])) {
+        if (!_optionalChain([responseParts, 'access', _253 => _253[i], 'access', _254 => _254.functionCall, 'optionalAccess', _255 => _255.name])) {
           continue;
         }
         preservedParts.push(responseParts[i]);
@@ -8898,7 +8925,7 @@ var PlanReActPlanner = class extends BasePlanner {
    * Handles non-function-call parts of the response
    */
   _handleNonFunctionCallParts(responsePart, preservedParts) {
-    if (_optionalChain([responsePart, 'access', _254 => _254.text, 'optionalAccess', _255 => _255.includes, 'call', _256 => _256(FINAL_ANSWER_TAG)])) {
+    if (_optionalChain([responsePart, 'access', _256 => _256.text, 'optionalAccess', _257 => _257.includes, 'call', _258 => _258(FINAL_ANSWER_TAG)])) {
       const [reasoningText, finalAnswerText] = this._splitByLastPattern(
         responsePart.text,
         FINAL_ANSWER_TAG
@@ -9137,7 +9164,7 @@ var OutputSchemaResponseProcessor = (_class26 = class extends BaseLlmResponsePro
   stripCodeFences(raw) {
     const fencePattern = /```(?:json)?\s*([\s\S]*?)```/i;
     const fenceMatch = raw.match(fencePattern);
-    if (_optionalChain([fenceMatch, 'optionalAccess', _257 => _257[1]])) {
+    if (_optionalChain([fenceMatch, 'optionalAccess', _259 => _259[1]])) {
       return fenceMatch[1].trim();
     }
     const lines = raw.split(/\r?\n/).map((l) => l.trim());
@@ -9174,7 +9201,7 @@ var SharedMemoryRequestProcessor = class extends BaseLlmRequestProcessor {
     const memoryService = invocationContext.memoryService;
     if (!memoryService) return;
     const lastUserEvent = invocationContext.session.events.findLast(
-      (e) => e.author === "user" && _optionalChain([e, 'access', _258 => _258.content, 'optionalAccess', _259 => _259.parts, 'optionalAccess', _260 => _260.length])
+      (e) => e.author === "user" && _optionalChain([e, 'access', _260 => _260.content, 'optionalAccess', _261 => _261.parts, 'optionalAccess', _262 => _262.length])
     );
     if (!lastUserEvent) return;
     const query = (_nullishCoalesce(lastUserEvent.content.parts, () => ( []))).map((p) => p.text || "").join(" ");
@@ -9185,7 +9212,7 @@ var SharedMemoryRequestProcessor = class extends BaseLlmRequestProcessor {
     });
     const sessionTexts = new Set(
       (llmRequest.contents || []).flatMap(
-        (c) => _optionalChain([c, 'access', _261 => _261.parts, 'optionalAccess', _262 => _262.map, 'call', _263 => _263((p) => p.text)]) || []
+        (c) => _optionalChain([c, 'access', _263 => _263.parts, 'optionalAccess', _264 => _264.map, 'call', _265 => _265((p) => p.text)]) || []
       )
     );
     for (const memory of results.memories) {
@@ -9606,7 +9633,7 @@ var LlmAgent = (_class27 = class _LlmAgent extends BaseAgent {
    * This matches the Python implementation's _llm_flow property
    */
   get llmFlow() {
-    if (this.disallowTransferToParent && this.disallowTransferToPeers && !_optionalChain([this, 'access', _264 => _264.subAgents, 'optionalAccess', _265 => _265.length])) {
+    if (this.disallowTransferToParent && this.disallowTransferToPeers && !_optionalChain([this, 'access', _266 => _266.subAgents, 'optionalAccess', _267 => _267.length])) {
       return new SingleFlow();
     }
     return new AutoFlow();
@@ -9622,7 +9649,7 @@ var LlmAgent = (_class27 = class _LlmAgent extends BaseAgent {
       );
       return;
     }
-    if (this.outputKey && event.isFinalResponse() && _optionalChain([event, 'access', _266 => _266.content, 'optionalAccess', _267 => _267.parts])) {
+    if (this.outputKey && event.isFinalResponse() && _optionalChain([event, 'access', _268 => _268.content, 'optionalAccess', _269 => _269.parts])) {
       let result = event.content.parts.map((part) => part.text || "").join("");
       if (this.outputSchema) {
         if (!result.trim()) {
@@ -9842,7 +9869,7 @@ var LoopAgent = class extends BaseAgent {
       for (const subAgent of this.subAgents) {
         for await (const event of subAgent.runAsync(ctx)) {
           yield event;
-          if (_optionalChain([event, 'access', _268 => _268.actions, 'optionalAccess', _269 => _269.escalate])) {
+          if (_optionalChain([event, 'access', _270 => _270.actions, 'optionalAccess', _271 => _271.escalate])) {
             return;
           }
         }
@@ -10154,17 +10181,17 @@ var RunConfig = class {
    */
   
   constructor(config) {
-    this.speechConfig = _optionalChain([config, 'optionalAccess', _270 => _270.speechConfig]);
-    this.responseModalities = _optionalChain([config, 'optionalAccess', _271 => _271.responseModalities]);
-    this.saveInputBlobsAsArtifacts = _optionalChain([config, 'optionalAccess', _272 => _272.saveInputBlobsAsArtifacts]) || false;
-    this.supportCFC = _optionalChain([config, 'optionalAccess', _273 => _273.supportCFC]) || false;
-    this.streamingMode = _optionalChain([config, 'optionalAccess', _274 => _274.streamingMode]) || "NONE" /* NONE */;
-    this.outputAudioTranscription = _optionalChain([config, 'optionalAccess', _275 => _275.outputAudioTranscription]);
-    this.inputAudioTranscription = _optionalChain([config, 'optionalAccess', _276 => _276.inputAudioTranscription]);
-    this.realtimeInputConfig = _optionalChain([config, 'optionalAccess', _277 => _277.realtimeInputConfig]);
-    this.enableAffectiveDialog = _optionalChain([config, 'optionalAccess', _278 => _278.enableAffectiveDialog]);
-    this.proactivity = _optionalChain([config, 'optionalAccess', _279 => _279.proactivity]);
-    this.maxLlmCalls = _nullishCoalesce(_optionalChain([config, 'optionalAccess', _280 => _280.maxLlmCalls]), () => ( 500));
+    this.speechConfig = _optionalChain([config, 'optionalAccess', _272 => _272.speechConfig]);
+    this.responseModalities = _optionalChain([config, 'optionalAccess', _273 => _273.responseModalities]);
+    this.saveInputBlobsAsArtifacts = _optionalChain([config, 'optionalAccess', _274 => _274.saveInputBlobsAsArtifacts]) || false;
+    this.supportCFC = _optionalChain([config, 'optionalAccess', _275 => _275.supportCFC]) || false;
+    this.streamingMode = _optionalChain([config, 'optionalAccess', _276 => _276.streamingMode]) || "NONE" /* NONE */;
+    this.outputAudioTranscription = _optionalChain([config, 'optionalAccess', _277 => _277.outputAudioTranscription]);
+    this.inputAudioTranscription = _optionalChain([config, 'optionalAccess', _278 => _278.inputAudioTranscription]);
+    this.realtimeInputConfig = _optionalChain([config, 'optionalAccess', _279 => _279.realtimeInputConfig]);
+    this.enableAffectiveDialog = _optionalChain([config, 'optionalAccess', _280 => _280.enableAffectiveDialog]);
+    this.proactivity = _optionalChain([config, 'optionalAccess', _281 => _281.proactivity]);
+    this.maxLlmCalls = _nullishCoalesce(_optionalChain([config, 'optionalAccess', _282 => _282.maxLlmCalls]), () => ( 500));
     this.validateMaxLlmCalls();
   }
   /**
@@ -10184,6 +10211,48 @@ var RunConfig = class {
   }
 };
 
+// src/artifacts/artifact-util.ts
+var SESSION_SCOPED_ARTIFACT_URI_RE = /^artifact:\/\/apps\/([^/]+)\/users\/([^/]+)\/sessions\/([^/]+)\/artifacts\/([^/]+)\/versions\/(\d+)$/;
+var USER_SCOPED_ARTIFACT_URI_RE = /^artifact:\/\/apps\/([^/]+)\/users\/([^/]+)\/artifacts\/([^/]+)\/versions\/(\d+)$/;
+function parseArtifactUri(uri) {
+  if (!uri || !uri.startsWith("artifact://")) {
+    return null;
+  }
+  let match = SESSION_SCOPED_ARTIFACT_URI_RE.exec(uri);
+  if (match) {
+    return {
+      appName: match[1],
+      userId: match[2],
+      sessionId: match[3],
+      filename: match[4],
+      version: Number.parseInt(match[5], 10)
+    };
+  }
+  match = USER_SCOPED_ARTIFACT_URI_RE.exec(uri);
+  if (match) {
+    return {
+      appName: match[1],
+      userId: match[2],
+      sessionId: void 0,
+      filename: match[3],
+      version: Number.parseInt(match[4], 10)
+    };
+  }
+  return null;
+}
+function getArtifactUri(args) {
+  const { appName, userId, filename, version, sessionId } = args;
+  if (sessionId) {
+    return `artifact://apps/${appName}/users/${userId}/sessions/${sessionId}/artifacts/${filename}/versions/${version}`;
+  }
+  return `artifact://apps/${appName}/users/${userId}/artifacts/${filename}/versions/${version}`;
+}
+function isArtifactRef(artifact) {
+  return Boolean(
+    _optionalChain([artifact, 'access', _283 => _283.fileData, 'optionalAccess', _284 => _284.fileUri]) && artifact.fileData.fileUri.startsWith("artifact://")
+  );
+}
+
 // src/artifacts/in-memory-artifact-service.ts
 var InMemoryArtifactService = (_class29 = class {constructor() { _class29.prototype.__init50.call(this); }
   __init50() {this.artifacts = /* @__PURE__ */ new Map()}
@@ -10224,7 +10293,29 @@ var InMemoryArtifactService = (_class29 = class {constructor() { _class29.protot
     if (targetVersion < 0 || targetVersion >= versions.length) {
       return null;
     }
-    return versions[targetVersion];
+    const artifactEntry = versions[targetVersion];
+    if (!artifactEntry) {
+      return null;
+    }
+    if (isArtifactRef(artifactEntry)) {
+      const parsedUri = parseArtifactUri(_optionalChain([artifactEntry, 'access', _285 => _285.fileData, 'optionalAccess', _286 => _286.fileUri]) || "");
+      if (!parsedUri) {
+        throw new Error(
+          `Invalid artifact reference URI: ${_optionalChain([artifactEntry, 'access', _287 => _287.fileData, 'optionalAccess', _288 => _288.fileUri])}`
+        );
+      }
+      return await this.loadArtifact({
+        appName: parsedUri.appName,
+        userId: parsedUri.userId,
+        sessionId: parsedUri.sessionId || sessionId,
+        filename: parsedUri.filename,
+        version: parsedUri.version
+      });
+    }
+    if (!artifactEntry.text && (!_optionalChain([artifactEntry, 'access', _289 => _289.inlineData, 'optionalAccess', _290 => _290.data]) || artifactEntry.inlineData.data.length === 0) && !artifactEntry.fileData) {
+      return null;
+    }
+    return artifactEntry;
   }
   async listArtifactKeys(args) {
     const { appName, userId, sessionId } = args;
@@ -10308,7 +10399,7 @@ var InMemoryMemoryService = (_class30 = class {
     }
     const userSessions = this._sessionEvents.get(userKey);
     const filteredEvents = session.events.filter(
-      (event) => _optionalChain([event, 'access', _281 => _281.content, 'optionalAccess', _282 => _282.parts])
+      (event) => _optionalChain([event, 'access', _291 => _291.content, 'optionalAccess', _292 => _292.parts])
     );
     userSessions.set(session.id, filteredEvents);
   }
@@ -10409,11 +10500,16 @@ var BaseSessionService = class {
       return;
     }
     for (const key in event.actions.stateDelta) {
-      if (Object.prototype.hasOwnProperty.call(event.actions.stateDelta, key)) {
+      if (Object.hasOwn(event.actions.stateDelta, key)) {
         if (key.startsWith("temp_")) {
           continue;
         }
-        session.state[key] = event.actions.stateDelta[key];
+        const value = event.actions.stateDelta[key];
+        if (value === null || value === void 0) {
+          delete session.state[key];
+        } else {
+          session.state[key] = value;
+        }
       }
     }
   }
@@ -10447,7 +10543,7 @@ var InMemorySessionService = (_class31 = class extends BaseSessionService {const
     return this.createSessionImpl(appName, userId, state, sessionId);
   }
   createSessionImpl(appName, userId, state, sessionId) {
-    const finalSessionId = _optionalChain([sessionId, 'optionalAccess', _283 => _283.trim, 'call', _284 => _284()]) || _crypto.randomUUID.call(void 0, );
+    const finalSessionId = _optionalChain([sessionId, 'optionalAccess', _293 => _293.trim, 'call', _294 => _294()]) || _crypto.randomUUID.call(void 0, );
     const session = {
       appName,
       userId,
@@ -10604,7 +10700,7 @@ var InMemorySessionService = (_class31 = class extends BaseSessionService {const
       warning(`sessionId ${sessionId} not in sessions[appName][userId]`);
       return event;
     }
-    if (_optionalChain([event, 'access', _285 => _285.actions, 'optionalAccess', _286 => _286.stateDelta])) {
+    if (_optionalChain([event, 'access', _295 => _295.actions, 'optionalAccess', _296 => _296.stateDelta])) {
       for (const key in event.actions.stateDelta) {
         const value = event.actions.stateDelta[key];
         if (key.startsWith(State.APP_PREFIX)) {
@@ -10638,14 +10734,14 @@ function _findFunctionCallEventIfLastEventIsFunctionResponse(session) {
     return null;
   }
   const lastEvent = events[events.length - 1];
-  if (_optionalChain([lastEvent, 'access', _287 => _287.content, 'optionalAccess', _288 => _288.parts, 'optionalAccess', _289 => _289.some, 'call', _290 => _290((part) => part.functionResponse)])) {
-    const functionCallId = _optionalChain([lastEvent, 'access', _291 => _291.content, 'access', _292 => _292.parts, 'access', _293 => _293.find, 'call', _294 => _294(
+  if (_optionalChain([lastEvent, 'access', _297 => _297.content, 'optionalAccess', _298 => _298.parts, 'optionalAccess', _299 => _299.some, 'call', _300 => _300((part) => part.functionResponse)])) {
+    const functionCallId = _optionalChain([lastEvent, 'access', _301 => _301.content, 'access', _302 => _302.parts, 'access', _303 => _303.find, 'call', _304 => _304(
       (part) => part.functionResponse
-    ), 'optionalAccess', _295 => _295.functionResponse, 'optionalAccess', _296 => _296.id]);
+    ), 'optionalAccess', _305 => _305.functionResponse, 'optionalAccess', _306 => _306.id]);
     if (!functionCallId) return null;
     for (let i = events.length - 2; i >= 0; i--) {
       const event = events[i];
-      const functionCalls = _optionalChain([event, 'access', _297 => _297.getFunctionCalls, 'optionalCall', _298 => _298()]) || [];
+      const functionCalls = _optionalChain([event, 'access', _307 => _307.getFunctionCalls, 'optionalCall', _308 => _308()]) || [];
       for (const functionCall of functionCalls) {
         if (functionCall.id === functionCallId) {
           return event;
@@ -10858,15 +10954,15 @@ var Runner = (_class32 = class {
    */
   _findAgentToRun(session, rootAgent) {
     const event = _findFunctionCallEventIfLastEventIsFunctionResponse(session);
-    if (_optionalChain([event, 'optionalAccess', _299 => _299.author])) {
+    if (_optionalChain([event, 'optionalAccess', _309 => _309.author])) {
       return rootAgent.findAgent(event.author);
     }
-    const nonUserEvents = _optionalChain([session, 'access', _300 => _300.events, 'optionalAccess', _301 => _301.filter, 'call', _302 => _302((e) => e.author !== "user"), 'access', _303 => _303.reverse, 'call', _304 => _304()]) || [];
+    const nonUserEvents = _optionalChain([session, 'access', _310 => _310.events, 'optionalAccess', _311 => _311.filter, 'call', _312 => _312((e) => e.author !== "user"), 'access', _313 => _313.reverse, 'call', _314 => _314()]) || [];
     for (const event2 of nonUserEvents) {
       if (event2.author === rootAgent.name) {
         return rootAgent;
       }
-      const agent = _optionalChain([rootAgent, 'access', _305 => _305.findSubAgent, 'optionalCall', _306 => _306(event2.author)]);
+      const agent = _optionalChain([rootAgent, 'access', _315 => _315.findSubAgent, 'optionalCall', _316 => _316(event2.author)]);
       if (!agent) {
         this.logger.debug(
           `Event from an unknown agent: ${event2.author}, event id: ${event2.id}`
@@ -10944,7 +11040,7 @@ var Runner = (_class32 = class {
    * Gets the configured summarizer or creates a default LLM-based one.
    */
   _getOrCreateSummarizer() {
-    if (_optionalChain([this, 'access', _307 => _307.eventsCompactionConfig, 'optionalAccess', _308 => _308.summarizer])) {
+    if (_optionalChain([this, 'access', _317 => _317.eventsCompactionConfig, 'optionalAccess', _318 => _318.summarizer])) {
       return this.eventsCompactionConfig.summarizer;
     }
     if (this.agent instanceof LlmAgent) {
@@ -10961,6 +11057,142 @@ var Runner = (_class32 = class {
     }
     return void 0;
   }
+  async rewind(args) {
+    const { userId, sessionId, rewindBeforeInvocationId } = args;
+    const session = await this.sessionService.getSession(
+      this.appName,
+      userId,
+      sessionId
+    );
+    if (!session) {
+      throw new Error(`Session not found: ${sessionId}`);
+    }
+    let rewindEventIndex = -1;
+    for (let i = 0; i < session.events.length; i++) {
+      if (session.events[i].invocationId === rewindBeforeInvocationId) {
+        rewindEventIndex = i;
+        break;
+      }
+    }
+    if (rewindEventIndex === -1) {
+      throw new Error(`Invocation ID not found: ${rewindBeforeInvocationId}`);
+    }
+    const stateDelta = await this._computeStateDeltaForRewind(
+      session,
+      rewindEventIndex
+    );
+    const artifactDelta = await this._computeArtifactDeltaForRewind(
+      session,
+      rewindEventIndex
+    );
+    const rewindEvent = new Event({
+      invocationId: newInvocationContextId(),
+      author: "user",
+      actions: new EventActions({
+        rewindBeforeInvocationId,
+        stateDelta,
+        artifactDelta
+      })
+    });
+    this.logger.info(
+      "Rewinding session to invocation:",
+      rewindBeforeInvocationId
+    );
+    await this.sessionService.appendEvent(session, rewindEvent);
+  }
+  async _computeStateDeltaForRewind(session, rewindEventIndex) {
+    const stateAtRewindPoint = {};
+    for (let i = 0; i < rewindEventIndex; i++) {
+      const event = session.events[i];
+      if (_optionalChain([event, 'access', _319 => _319.actions, 'optionalAccess', _320 => _320.stateDelta])) {
+        for (const [k, v] of Object.entries(event.actions.stateDelta)) {
+          if (k.startsWith("app:") || k.startsWith("user:")) {
+            continue;
+          }
+          if (v === null || v === void 0) {
+            delete stateAtRewindPoint[k];
+          } else {
+            stateAtRewindPoint[k] = v;
+          }
+        }
+      }
+    }
+    const currentState = session.state;
+    const rewindStateDelta = {};
+    for (const [key, valueAtRewind] of Object.entries(stateAtRewindPoint)) {
+      if (!(key in currentState) || currentState[key] !== valueAtRewind) {
+        rewindStateDelta[key] = valueAtRewind;
+      }
+    }
+    for (const key of Object.keys(currentState)) {
+      if (key.startsWith("app:") || key.startsWith("user:")) {
+        continue;
+      }
+      if (!(key in stateAtRewindPoint)) {
+        rewindStateDelta[key] = null;
+      }
+    }
+    return rewindStateDelta;
+  }
+  async _computeArtifactDeltaForRewind(session, rewindEventIndex) {
+    if (!this.artifactService) {
+      return {};
+    }
+    const versionsAtRewindPoint = {};
+    for (let i = 0; i < rewindEventIndex; i++) {
+      const event = session.events[i];
+      if (_optionalChain([event, 'access', _321 => _321.actions, 'optionalAccess', _322 => _322.artifactDelta])) {
+        Object.assign(versionsAtRewindPoint, event.actions.artifactDelta);
+      }
+    }
+    const currentVersions = {};
+    for (const event of session.events) {
+      if (_optionalChain([event, 'access', _323 => _323.actions, 'optionalAccess', _324 => _324.artifactDelta])) {
+        Object.assign(currentVersions, event.actions.artifactDelta);
+      }
+    }
+    const rewindArtifactDelta = {};
+    for (const [filename, vn] of Object.entries(currentVersions)) {
+      if (filename.startsWith("user:")) {
+        continue;
+      }
+      const vt = versionsAtRewindPoint[filename];
+      if (vt === vn) {
+        continue;
+      }
+      let artifact;
+      if (vt === void 0 || vt === null) {
+        artifact = {
+          inlineData: {
+            mimeType: "application/octet-stream",
+            data: ""
+          }
+        };
+      } else {
+        const artifactUri = getArtifactUri({
+          appName: this.appName,
+          userId: session.userId,
+          sessionId: session.id,
+          filename,
+          version: vt
+        });
+        artifact = {
+          fileData: {
+            fileUri: artifactUri
+          }
+        };
+      }
+      const newVersion = await this.artifactService.saveArtifact({
+        appName: this.appName,
+        userId: session.userId,
+        sessionId: session.id,
+        filename,
+        artifact
+      });
+      rewindArtifactDelta[filename] = newVersion;
+    }
+    return rewindArtifactDelta;
+  }
 }, _class32);
 var InMemoryRunner = class extends Runner {
   /**
@@ -11600,7 +11832,7 @@ var AgentBuilder = (_class33 = class _AgentBuilder {
     const outputSchema = this.config.outputSchema;
     const agentType = this.agentType;
     const isMulti = agentType === "parallel" || agentType === "sequential";
-    const subAgentNames = _optionalChain([this, 'access', _309 => _309.config, 'access', _310 => _310.subAgents, 'optionalAccess', _311 => _311.map, 'call', _312 => _312((a) => a.name)]) || [];
+    const subAgentNames = _optionalChain([this, 'access', _325 => _325.config, 'access', _326 => _326.subAgents, 'optionalAccess', _327 => _327.map, 'call', _328 => _328((a) => a.name)]) || [];
     const runConfig = this.runConfig;
     return {
       __outputSchema: outputSchema,
@@ -11609,7 +11841,7 @@ var AgentBuilder = (_class33 = class _AgentBuilder {
         let combinedResponse = "";
         const perAgentBuffers = {};
         const authors = /* @__PURE__ */ new Set();
-        if (!_optionalChain([sessionOptions, 'optionalAccess', _313 => _313.userId])) {
+        if (!_optionalChain([sessionOptions, 'optionalAccess', _329 => _329.userId])) {
           throw new Error("Session configuration is required");
         }
         for await (const event of baseRunner.runAsync({
@@ -11618,7 +11850,7 @@ var AgentBuilder = (_class33 = class _AgentBuilder {
           newMessage,
           runConfig
         })) {
-          if (_optionalChain([event, 'access', _314 => _314.content, 'optionalAccess', _315 => _315.parts]) && Array.isArray(event.content.parts)) {
+          if (_optionalChain([event, 'access', _330 => _330.content, 'optionalAccess', _331 => _331.parts]) && Array.isArray(event.content.parts)) {
             const content = event.content.parts.map(
               (part) => (part && typeof part === "object" && "text" in part ? part.text : "") || ""
             ).join("");
@@ -11664,6 +11896,9 @@ var AgentBuilder = (_class33 = class _AgentBuilder {
           ...params,
           runConfig: _nullishCoalesce(params.runConfig, () => ( runConfig))
         });
+      },
+      rewind(params) {
+        return baseRunner.rewind(params);
       }
     };
   }
@@ -11905,7 +12140,7 @@ var VertexAiSessionService = class extends BaseSessionService {
         path: `operations/${operationId}`,
         request_dict: {}
       });
-      if (_optionalChain([lroResponse, 'optionalAccess', _316 => _316.done])) {
+      if (_optionalChain([lroResponse, 'optionalAccess', _332 => _332.done])) {
         break;
       }
       await new Promise((resolve) => setTimeout(resolve, 1e3));
@@ -12276,12 +12511,12 @@ var DatabaseSessionService = (_class34 = class extends BaseSessionService {
   }
   async createSession(appName, userId, state, sessionId) {
     await this.ensureInitialized();
-    const id = _optionalChain([sessionId, 'optionalAccess', _317 => _317.trim, 'call', _318 => _318()]) || this.generateSessionId();
+    const id = _optionalChain([sessionId, 'optionalAccess', _333 => _333.trim, 'call', _334 => _334()]) || this.generateSessionId();
     return await this.db.transaction().execute(async (trx) => {
       const appState = await trx.selectFrom("app_states").selectAll().where("app_name", "=", appName).executeTakeFirst();
       const userState = await trx.selectFrom("user_states").selectAll().where("app_name", "=", appName).where("user_id", "=", userId).executeTakeFirst();
-      let currentAppState = this.parseJsonSafely(_optionalChain([appState, 'optionalAccess', _319 => _319.state]), {});
-      let currentUserState = this.parseJsonSafely(_optionalChain([userState, 'optionalAccess', _320 => _320.state]), {});
+      let currentAppState = this.parseJsonSafely(_optionalChain([appState, 'optionalAccess', _335 => _335.state]), {});
+      let currentUserState = this.parseJsonSafely(_optionalChain([userState, 'optionalAccess', _336 => _336.state]), {});
       if (!appState) {
         await trx.insertInto("app_states").values({
           app_name: appName,
@@ -12340,21 +12575,21 @@ var DatabaseSessionService = (_class34 = class extends BaseSessionService {
         return void 0;
       }
       let eventQuery = trx.selectFrom("events").selectAll().where("session_id", "=", sessionId).orderBy("timestamp", "desc");
-      if (_optionalChain([config, 'optionalAccess', _321 => _321.afterTimestamp])) {
+      if (_optionalChain([config, 'optionalAccess', _337 => _337.afterTimestamp])) {
         eventQuery = eventQuery.where(
           "timestamp",
           ">=",
           new Date(config.afterTimestamp * 1e3)
         );
       }
-      if (_optionalChain([config, 'optionalAccess', _322 => _322.numRecentEvents])) {
+      if (_optionalChain([config, 'optionalAccess', _338 => _338.numRecentEvents])) {
         eventQuery = eventQuery.limit(config.numRecentEvents);
       }
       const storageEvents = await eventQuery.execute();
       const appState = await trx.selectFrom("app_states").selectAll().where("app_name", "=", appName).executeTakeFirst();
       const userState = await trx.selectFrom("user_states").selectAll().where("app_name", "=", appName).where("user_id", "=", userId).executeTakeFirst();
-      const currentAppState = this.parseJsonSafely(_optionalChain([appState, 'optionalAccess', _323 => _323.state]), {});
-      const currentUserState = this.parseJsonSafely(_optionalChain([userState, 'optionalAccess', _324 => _324.state]), {});
+      const currentAppState = this.parseJsonSafely(_optionalChain([appState, 'optionalAccess', _339 => _339.state]), {});
+      const currentUserState = this.parseJsonSafely(_optionalChain([userState, 'optionalAccess', _340 => _340.state]), {});
       const sessionState = this.parseJsonSafely(storageSession.state, {});
       const mergedState = this.mergeState(
         currentAppState,
@@ -12412,13 +12647,13 @@ var DatabaseSessionService = (_class34 = class extends BaseSessionService {
       }
       const appState = await trx.selectFrom("app_states").selectAll().where("app_name", "=", session.appName).executeTakeFirst();
       const userState = await trx.selectFrom("user_states").selectAll().where("app_name", "=", session.appName).where("user_id", "=", session.userId).executeTakeFirst();
-      let currentAppState = this.parseJsonSafely(_optionalChain([appState, 'optionalAccess', _325 => _325.state]), {});
-      let currentUserState = this.parseJsonSafely(_optionalChain([userState, 'optionalAccess', _326 => _326.state]), {});
+      let currentAppState = this.parseJsonSafely(_optionalChain([appState, 'optionalAccess', _341 => _341.state]), {});
+      let currentUserState = this.parseJsonSafely(_optionalChain([userState, 'optionalAccess', _342 => _342.state]), {});
       let sessionState = this.parseJsonSafely(storageSession.state, {});
       let appStateDelta = {};
       let userStateDelta = {};
       let sessionStateDelta = {};
-      if (_optionalChain([event, 'access', _327 => _327.actions, 'optionalAccess', _328 => _328.stateDelta])) {
+      if (_optionalChain([event, 'access', _343 => _343.actions, 'optionalAccess', _344 => _344.stateDelta])) {
         const deltas = this.extractStateDelta(event.actions.stateDelta);
         appStateDelta = deltas.appStateDelta;
         userStateDelta = deltas.userStateDelta;
@@ -12564,7 +12799,7 @@ var DatabaseSessionService = (_class34 = class extends BaseSessionService {
    * Overrides the base class method to work with plain object state.
    */
   updateSessionState(session, event) {
-    if (!_optionalChain([event, 'access', _329 => _329.actions, 'optionalAccess', _330 => _330.stateDelta])) {
+    if (!_optionalChain([event, 'access', _345 => _345.actions, 'optionalAccess', _346 => _346.stateDelta])) {
       return;
     }
     for (const [key, value] of Object.entries(event.actions.stateDelta)) {
@@ -12734,7 +12969,7 @@ var GcsArtifactService = class {
       };
       return part;
     } catch (error) {
-      if (_optionalChain([error, 'optionalAccess', _331 => _331.code]) === 404) {
+      if (_optionalChain([error, 'optionalAccess', _347 => _347.code]) === 404) {
         return null;
       }
       throw error;
@@ -12985,13 +13220,13 @@ var VertexAiEvalFacade = class _VertexAiEvalFacade {
     };
   }
   _getText(content) {
-    if (_optionalChain([content, 'optionalAccess', _332 => _332.parts])) {
+    if (_optionalChain([content, 'optionalAccess', _348 => _348.parts])) {
       return content.parts.map((p) => p.text || "").filter((text) => text.length > 0).join("\n");
     }
     return "";
   }
   _getScore(evalResult) {
-    if (_optionalChain([evalResult, 'optionalAccess', _333 => _333.summaryMetrics, 'optionalAccess', _334 => _334[0], 'optionalAccess', _335 => _335.meanScore]) !== void 0 && typeof evalResult.summaryMetrics[0].meanScore === "number" && !Number.isNaN(evalResult.summaryMetrics[0].meanScore)) {
+    if (_optionalChain([evalResult, 'optionalAccess', _349 => _349.summaryMetrics, 'optionalAccess', _350 => _350[0], 'optionalAccess', _351 => _351.meanScore]) !== void 0 && typeof evalResult.summaryMetrics[0].meanScore === "number" && !Number.isNaN(evalResult.summaryMetrics[0].meanScore)) {
       return evalResult.summaryMetrics[0].meanScore;
     }
     return void 0;
@@ -13141,7 +13376,7 @@ var ResponseEvaluator = class extends Evaluator {
     return fmeasure;
   }
   extractText(content) {
-    if (_optionalChain([content, 'optionalAccess', _336 => _336.parts])) {
+    if (_optionalChain([content, 'optionalAccess', _352 => _352.parts])) {
       return content.parts.map((p) => p.text || "").filter((text) => text.length > 0).join(" ");
     }
     return "";
@@ -13174,7 +13409,7 @@ var TrajectoryEvaluator = class extends Evaluator {
     for (let i = 0; i < actualInvocations.length; i++) {
       const actual = actualInvocations[i];
       const expected = expectedInvocations[i];
-      if (!_optionalChain([actual, 'access', _337 => _337.intermediateData, 'optionalAccess', _338 => _338.toolUses]) || !_optionalChain([expected, 'access', _339 => _339.intermediateData, 'optionalAccess', _340 => _340.toolUses])) {
+      if (!_optionalChain([actual, 'access', _353 => _353.intermediateData, 'optionalAccess', _354 => _354.toolUses]) || !_optionalChain([expected, 'access', _355 => _355.intermediateData, 'optionalAccess', _356 => _356.toolUses])) {
         perInvocationResults.push({
           actualInvocation: actual,
           expectedInvocation: expected,
@@ -13262,7 +13497,7 @@ var SafetyEvaluatorV1 = class extends Evaluator {
 
 // src/evaluation/llm-as-judge-utils.ts
 function getTextFromContent(content) {
-  if (_optionalChain([content, 'optionalAccess', _341 => _341.parts])) {
+  if (_optionalChain([content, 'optionalAccess', _357 => _357.parts])) {
     return content.parts.map((part) => part.text).filter(Boolean).join("\n");
   }
   return "";
@@ -13274,9 +13509,9 @@ function getEvalStatus(score, threshold) {
 // src/evaluation/llm-as-judge.ts
 var LlmAsJudge = class {
   async sampleJudge(prompt, numSamples, critiqueParser, judgeModelOptions) {
-    const modelName = _optionalChain([judgeModelOptions, 'optionalAccess', _342 => _342.judgeModel]) || "gemini-2.5-flash";
+    const modelName = _optionalChain([judgeModelOptions, 'optionalAccess', _358 => _358.judgeModel]) || "gemini-2.5-flash";
     const model = LLMRegistry.getModelOrCreate(modelName);
-    const config = _optionalChain([judgeModelOptions, 'optionalAccess', _343 => _343.judgeModelConfig]) || {};
+    const config = _optionalChain([judgeModelOptions, 'optionalAccess', _359 => _359.judgeModelConfig]) || {};
     const samples = [];
     for (let i = 0; i < numSamples; i++) {
       try {
@@ -13342,7 +13577,7 @@ function parseCritique(response) {
   const labelMatchIsResponseValid = response.match(
     /"is_the_agent_response_valid":\s*\[*[\n\s]*"*([^"^\]^\s]*)"*[\n\s]*\]*\s*[,\n\}]/
   );
-  if (_optionalChain([labelMatchIsResponseValid, 'optionalAccess', _344 => _344[1]])) {
+  if (_optionalChain([labelMatchIsResponseValid, 'optionalAccess', _360 => _360[1]])) {
     const label = labelMatchIsResponseValid[1].toLowerCase();
     return label === "valid" ? "valid" /* VALID */ : "invalid" /* INVALID */;
   }
@@ -13387,7 +13622,7 @@ var FinalResponseMatchV2Evaluator = class extends Evaluator {
         "{prompt}",
         prompt
       ).replace("{response}", response).replace("{golden_response}", goldenResponse);
-      const numSamples = _nullishCoalesce(_optionalChain([this, 'access', _345 => _345.metric, 'access', _346 => _346.judgeModelOptions, 'optionalAccess', _347 => _347.numSamples]), () => ( DEFAULT_NUM_SAMPLES));
+      const numSamples = _nullishCoalesce(_optionalChain([this, 'access', _361 => _361.metric, 'access', _362 => _362.judgeModelOptions, 'optionalAccess', _363 => _363.numSamples]), () => ( DEFAULT_NUM_SAMPLES));
       const labels = await this.llmAsJudge.sampleJudge(
         formattedPrompt,
         numSamples,
@@ -13427,7 +13662,7 @@ var MetricEvaluatorRegistry = (_class35 = class {constructor() { _class35.protot
     const metricName = metricInfo.metricName;
     if (this.registry.has(metricName)) {
       console.info(
-        `Updating Evaluator class for ${metricName} from ${_optionalChain([this, 'access', _348 => _348.registry, 'access', _349 => _349.get, 'call', _350 => _350(metricName), 'optionalAccess', _351 => _351.evaluator, 'access', _352 => _352.name])} to ${evaluator.name}`
+        `Updating Evaluator class for ${metricName} from ${_optionalChain([this, 'access', _364 => _364.registry, 'access', _365 => _365.get, 'call', _366 => _366(metricName), 'optionalAccess', _367 => _367.evaluator, 'access', _368 => _368.name])} to ${evaluator.name}`
       );
     }
     this.registry.set(metricName, {
@@ -13541,10 +13776,10 @@ var LocalEvalService = class extends BaseEvalService {
       for (const evalMetric of evaluateConfig.evalMetrics) {
         const evaluator = DEFAULT_METRIC_EVALUATOR_REGISTRY.getEvaluator(evalMetric);
         const actual = results.filter(
-          (r) => !_optionalChain([r, 'access', _353 => _353.invocationId, 'optionalAccess', _354 => _354.includes, 'call', _355 => _355("expected")])
+          (r) => !_optionalChain([r, 'access', _369 => _369.invocationId, 'optionalAccess', _370 => _370.includes, 'call', _371 => _371("expected")])
         );
         const expected = results.filter(
-          (r) => _optionalChain([r, 'access', _356 => _356.invocationId, 'optionalAccess', _357 => _357.includes, 'call', _358 => _358("expected")])
+          (r) => _optionalChain([r, 'access', _372 => _372.invocationId, 'optionalAccess', _373 => _373.includes, 'call', _374 => _374("expected")])
         );
         const result = await evaluator.evaluateInvocations(actual, expected);
         evalResult.evalCaseResults.push({
@@ -13936,13 +14171,13 @@ ${failures.join(
     console.log("\n\n");
   }
   static _convertContentToText(content) {
-    if (_optionalChain([content, 'optionalAccess', _359 => _359.parts])) {
+    if (_optionalChain([content, 'optionalAccess', _375 => _375.parts])) {
       return content.parts.map((p) => p.text || "").filter((text) => text.length > 0).join("\n");
     }
     return "";
   }
   static _convertToolCallsToText(intermediateData) {
-    if (_optionalChain([intermediateData, 'optionalAccess', _360 => _360.toolUses])) {
+    if (_optionalChain([intermediateData, 'optionalAccess', _376 => _376.toolUses])) {
       return intermediateData.toolUses.map((t) => JSON.stringify(t)).join("\n");
     }
     return "";
@@ -13997,7 +14232,7 @@ ${failures.join(
     for (const [metricName, evalMetricResultsWithInvocations] of Object.entries(
       evalMetricResults
     )) {
-      const threshold = _optionalChain([evalMetricResultsWithInvocations, 'access', _361 => _361[0], 'optionalAccess', _362 => _362.evalMetricResult, 'access', _363 => _363.threshold]) || 0;
+      const threshold = _optionalChain([evalMetricResultsWithInvocations, 'access', _377 => _377[0], 'optionalAccess', _378 => _378.evalMetricResult, 'access', _379 => _379.threshold]) || 0;
       const scores = evalMetricResultsWithInvocations.map((m) => m.evalMetricResult.score).filter((s) => s !== void 0);
       let overallScore;
       let overallEvalStatus;
@@ -14086,7 +14321,7 @@ var RougeEvaluator = class extends Evaluator {
   }
 };
 function getTextFromContent2(content) {
-  if (_optionalChain([content, 'optionalAccess', _364 => _364.parts])) {
+  if (_optionalChain([content, 'optionalAccess', _380 => _380.parts])) {
     return content.parts.map((part) => part.text).filter(Boolean).join("\n");
   }
   return "";
@@ -14284,4 +14519,7 @@ var VERSION = "0.1.0";
 
 
 
-exports.AF_FUNCTION_CALL_ID_PREFIX = AF_FUNCTION_CALL_ID_PREFIX; exports.Agent = LlmAgent; exports.AgentBuilder = AgentBuilder; exports.AgentEvaluator = AgentEvaluator; exports.AgentTool = AgentTool; exports.Agents = agents_exports; exports.AiSdkLlm = AiSdkLlm; exports.AnthropicLlm = AnthropicLlm; exports.ApiKeyCredential = ApiKeyCredential; exports.ApiKeyScheme = ApiKeyScheme; exports.AuthConfig = AuthConfig; exports.AuthCredential = AuthCredential; exports.AuthCredentialType = AuthCredentialType; exports.AuthHandler = AuthHandler; exports.AuthScheme = AuthScheme; exports.AuthSchemeType = AuthSchemeType; exports.AuthTool = AuthTool; exports.AutoFlow = AutoFlow; exports.BaseAgent = BaseAgent; exports.BaseCodeExecutor = BaseCodeExecutor; exports.BaseLLMConnection = BaseLLMConnection; exports.BaseLlm = BaseLlm; exports.BaseLlmFlow = BaseLlmFlow; exports.BaseLlmRequestProcessor = BaseLlmRequestProcessor; exports.BaseLlmResponseProcessor = BaseLlmResponseProcessor; exports.BasePlanner = BasePlanner; exports.BaseSessionService = BaseSessionService; exports.BaseTool = BaseTool; exports.BasicAuthCredential = BasicAuthCredential; exports.BearerTokenCredential = BearerTokenCredential; exports.BuiltInCodeExecutor = BuiltInCodeExecutor; exports.BuiltInPlanner = BuiltInPlanner; exports.CallbackContext = CallbackContext; exports.CodeExecutionUtils = CodeExecutionUtils; exports.CodeExecutorContext = CodeExecutorContext; exports.DatabaseSessionService = DatabaseSessionService; exports.EnhancedAuthConfig = EnhancedAuthConfig; exports.EvalResult = EvalResult; exports.EvalStatus = EvalStatus; exports.Evaluation = evaluation_exports; exports.Evaluator = Evaluator; exports.Event = Event; exports.EventActions = EventActions; exports.Events = events_exports; exports.ExitLoopTool = ExitLoopTool; exports.FileOperationsTool = FileOperationsTool; exports.FinalResponseMatchV2Evaluator = FinalResponseMatchV2Evaluator; exports.Flows = flows_exports; exports.FunctionTool = FunctionTool; exports.GcsArtifactService = GcsArtifactService; exports.GetUserChoiceTool = GetUserChoiceTool; exports.GoogleLlm = GoogleLlm; exports.GoogleSearch = GoogleSearch; exports.HttpRequestTool = HttpRequestTool; exports.HttpScheme = HttpScheme; exports.InMemoryArtifactService = InMemoryArtifactService; exports.InMemoryMemoryService = InMemoryMemoryService; exports.InMemoryRunner = InMemoryRunner; exports.InMemorySessionService = InMemorySessionService; exports.InvocationContext = InvocationContext; exports.LLMRegistry = LLMRegistry; exports.LangGraphAgent = LangGraphAgent; exports.LlmAgent = LlmAgent; exports.LlmCallsLimitExceededError = LlmCallsLimitExceededError; exports.LlmEventSummarizer = LlmEventSummarizer; exports.LlmRequest = LlmRequest; exports.LlmResponse = LlmResponse; exports.LoadArtifactsTool = LoadArtifactsTool; exports.LoadMemoryTool = LoadMemoryTool; exports.LocalEvalService = LocalEvalService; exports.LoopAgent = LoopAgent; exports.McpAbi = McpAbi; exports.McpAtp = McpAtp; exports.McpBamm = McpBamm; exports.McpCoinGecko = McpCoinGecko; exports.McpCoinGeckoPro = McpCoinGeckoPro; exports.McpDiscord = McpDiscord; exports.McpError = McpError; exports.McpErrorType = McpErrorType; exports.McpFilesystem = McpFilesystem; exports.McpFraxlend = McpFraxlend; exports.McpGeneric = McpGeneric; exports.McpIqWiki = McpIqWiki; exports.McpMemory = McpMemory; exports.McpNearAgent = McpNearAgent; exports.McpNearIntents = McpNearIntents; exports.McpOdos = McpOdos; exports.McpSamplingHandler = McpSamplingHandler; exports.McpTelegram = McpTelegram; exports.McpToolset = McpToolset; exports.McpUpbit = McpUpbit; exports.Memory = memory_exports; exports.Models = models_exports; exports.OAuth2Credential = OAuth2Credential; exports.OAuth2Scheme = OAuth2Scheme; exports.OpenAiLlm = OpenAiLlm; exports.OpenIdConnectScheme = OpenIdConnectScheme; exports.ParallelAgent = ParallelAgent; exports.PlanReActPlanner = PlanReActPlanner; exports.PrebuiltMetrics = PrebuiltMetrics; exports.REQUEST_EUC_FUNCTION_CALL_NAME = REQUEST_EUC_FUNCTION_CALL_NAME; exports.ReadonlyContext = ReadonlyContext; exports.RougeEvaluator = RougeEvaluator; exports.RunConfig = RunConfig; exports.Runner = Runner; exports.SafetyEvaluatorV1 = SafetyEvaluatorV1; exports.SequentialAgent = SequentialAgent; exports.Sessions = sessions_exports; exports.SingleFlow = SingleFlow; exports.State = State; exports.StreamingMode = StreamingMode; exports.TelemetryService = TelemetryService; exports.ToolContext = ToolContext; exports.Tools = tools_exports; exports.TrajectoryEvaluator = TrajectoryEvaluator; exports.TransferToAgentTool = TransferToAgentTool; exports.UserInteractionTool = UserInteractionTool; exports.VERSION = VERSION; exports.VertexAiRagMemoryService = VertexAiRagMemoryService; exports.VertexAiSessionService = VertexAiSessionService; exports._findFunctionCallEventIfLastEventIsFunctionResponse = _findFunctionCallEventIfLastEventIsFunctionResponse; exports.adkToMcpToolType = adkToMcpToolType; exports.agentTransferRequestProcessor = requestProcessor8; exports.basicRequestProcessor = requestProcessor2; exports.buildFunctionDeclaration = buildFunctionDeclaration; exports.codeExecutionRequestProcessor = requestProcessor3; exports.codeExecutionResponseProcessor = responseProcessor; exports.contentRequestProcessor = requestProcessor4; exports.convertMcpToolToBaseTool = convertMcpToolToBaseTool; exports.createAuthToolArguments = createAuthToolArguments; exports.createBranchContextForSubAgent = createBranchContextForSubAgent; exports.createDatabaseSessionService = createDatabaseSessionService; exports.createFunctionTool = createFunctionTool; exports.createMysqlSessionService = createMysqlSessionService; exports.createPostgresSessionService = createPostgresSessionService; exports.createSamplingHandler = createSamplingHandler; exports.createSqliteSessionService = createSqliteSessionService; exports.createTool = createTool; exports.generateAuthEvent = generateAuthEvent; exports.generateClientFunctionCallId = generateClientFunctionCallId; exports.getLongRunningFunctionCalls = getLongRunningFunctionCalls; exports.getMcpTools = getMcpTools; exports.handleFunctionCallsAsync = handleFunctionCallsAsync; exports.handleFunctionCallsLive = handleFunctionCallsLive; exports.identityRequestProcessor = requestProcessor5; exports.initializeTelemetry = initializeTelemetry; exports.injectSessionState = injectSessionState; exports.instructionsRequestProcessor = requestProcessor6; exports.isEnhancedAuthConfig = isEnhancedAuthConfig; exports.jsonSchemaToDeclaration = jsonSchemaToDeclaration; exports.mcpSchemaToParameters = mcpSchemaToParameters; exports.mergeAgentRun = mergeAgentRun; exports.mergeParallelFunctionResponseEvents = mergeParallelFunctionResponseEvents; exports.newInvocationContextId = newInvocationContextId; exports.nlPlanningRequestProcessor = requestProcessor7; exports.nlPlanningResponseProcessor = responseProcessor2; exports.normalizeJsonSchema = normalizeJsonSchema; exports.populateClientFunctionCallId = populateClientFunctionCallId; exports.registerProviders = registerProviders; exports.removeClientFunctionCallId = removeClientFunctionCallId; exports.requestProcessor = requestProcessor; exports.runCompactionForSlidingWindow = runCompactionForSlidingWindow; exports.shutdownTelemetry = shutdownTelemetry; exports.telemetryService = telemetryService; exports.traceLlmCall = traceLlmCall; exports.traceToolCall = traceToolCall; exports.tracer = tracer;
+
+
+
+exports.AF_FUNCTION_CALL_ID_PREFIX = AF_FUNCTION_CALL_ID_PREFIX; exports.Agent = LlmAgent; exports.AgentBuilder = AgentBuilder; exports.AgentEvaluator = AgentEvaluator; exports.AgentTool = AgentTool; exports.Agents = agents_exports; exports.AiSdkLlm = AiSdkLlm; exports.AnthropicLlm = AnthropicLlm; exports.ApiKeyCredential = ApiKeyCredential; exports.ApiKeyScheme = ApiKeyScheme; exports.AuthConfig = AuthConfig; exports.AuthCredential = AuthCredential; exports.AuthCredentialType = AuthCredentialType; exports.AuthHandler = AuthHandler; exports.AuthScheme = AuthScheme; exports.AuthSchemeType = AuthSchemeType; exports.AuthTool = AuthTool; exports.AutoFlow = AutoFlow; exports.BaseAgent = BaseAgent; exports.BaseCodeExecutor = BaseCodeExecutor; exports.BaseLLMConnection = BaseLLMConnection; exports.BaseLlm = BaseLlm; exports.BaseLlmFlow = BaseLlmFlow; exports.BaseLlmRequestProcessor = BaseLlmRequestProcessor; exports.BaseLlmResponseProcessor = BaseLlmResponseProcessor; exports.BasePlanner = BasePlanner; exports.BaseSessionService = BaseSessionService; exports.BaseTool = BaseTool; exports.BasicAuthCredential = BasicAuthCredential; exports.BearerTokenCredential = BearerTokenCredential; exports.BuiltInCodeExecutor = BuiltInCodeExecutor; exports.BuiltInPlanner = BuiltInPlanner; exports.CallbackContext = CallbackContext; exports.CodeExecutionUtils = CodeExecutionUtils; exports.CodeExecutorContext = CodeExecutorContext; exports.DatabaseSessionService = DatabaseSessionService; exports.EnhancedAuthConfig = EnhancedAuthConfig; exports.EvalResult = EvalResult; exports.EvalStatus = EvalStatus; exports.Evaluation = evaluation_exports; exports.Evaluator = Evaluator; exports.Event = Event; exports.EventActions = EventActions; exports.Events = events_exports; exports.ExitLoopTool = ExitLoopTool; exports.FileOperationsTool = FileOperationsTool; exports.FinalResponseMatchV2Evaluator = FinalResponseMatchV2Evaluator; exports.Flows = flows_exports; exports.FunctionTool = FunctionTool; exports.GcsArtifactService = GcsArtifactService; exports.GetUserChoiceTool = GetUserChoiceTool; exports.GoogleLlm = GoogleLlm; exports.GoogleSearch = GoogleSearch; exports.HttpRequestTool = HttpRequestTool; exports.HttpScheme = HttpScheme; exports.InMemoryArtifactService = InMemoryArtifactService; exports.InMemoryMemoryService = InMemoryMemoryService; exports.InMemoryRunner = InMemoryRunner; exports.InMemorySessionService = InMemorySessionService; exports.InvocationContext = InvocationContext; exports.LLMRegistry = LLMRegistry; exports.LangGraphAgent = LangGraphAgent; exports.LlmAgent = LlmAgent; exports.LlmCallsLimitExceededError = LlmCallsLimitExceededError; exports.LlmEventSummarizer = LlmEventSummarizer; exports.LlmRequest = LlmRequest; exports.LlmResponse = LlmResponse; exports.LoadArtifactsTool = LoadArtifactsTool; exports.LoadMemoryTool = LoadMemoryTool; exports.LocalEvalService = LocalEvalService; exports.LoopAgent = LoopAgent; exports.McpAbi = McpAbi; exports.McpAtp = McpAtp; exports.McpBamm = McpBamm; exports.McpCoinGecko = McpCoinGecko; exports.McpCoinGeckoPro = McpCoinGeckoPro; exports.McpDiscord = McpDiscord; exports.McpError = McpError; exports.McpErrorType = McpErrorType; exports.McpFilesystem = McpFilesystem; exports.McpFraxlend = McpFraxlend; exports.McpGeneric = McpGeneric; exports.McpIqWiki = McpIqWiki; exports.McpMemory = McpMemory; exports.McpNearAgent = McpNearAgent; exports.McpNearIntents = McpNearIntents; exports.McpOdos = McpOdos; exports.McpSamplingHandler = McpSamplingHandler; exports.McpTelegram = McpTelegram; exports.McpToolset = McpToolset; exports.McpUpbit = McpUpbit; exports.Memory = memory_exports; exports.Models = models_exports; exports.OAuth2Credential = OAuth2Credential; exports.OAuth2Scheme = OAuth2Scheme; exports.OpenAiLlm = OpenAiLlm; exports.OpenIdConnectScheme = OpenIdConnectScheme; exports.ParallelAgent = ParallelAgent; exports.PlanReActPlanner = PlanReActPlanner; exports.PrebuiltMetrics = PrebuiltMetrics; exports.REQUEST_EUC_FUNCTION_CALL_NAME = REQUEST_EUC_FUNCTION_CALL_NAME; exports.ReadonlyContext = ReadonlyContext; exports.RougeEvaluator = RougeEvaluator; exports.RunConfig = RunConfig; exports.Runner = Runner; exports.SafetyEvaluatorV1 = SafetyEvaluatorV1; exports.SequentialAgent = SequentialAgent; exports.Sessions = sessions_exports; exports.SingleFlow = SingleFlow; exports.State = State; exports.StreamingMode = StreamingMode; exports.TelemetryService = TelemetryService; exports.ToolContext = ToolContext; exports.Tools = tools_exports; exports.TrajectoryEvaluator = TrajectoryEvaluator; exports.TransferToAgentTool = TransferToAgentTool; exports.UserInteractionTool = UserInteractionTool; exports.VERSION = VERSION; exports.VertexAiRagMemoryService = VertexAiRagMemoryService; exports.VertexAiSessionService = VertexAiSessionService; exports._findFunctionCallEventIfLastEventIsFunctionResponse = _findFunctionCallEventIfLastEventIsFunctionResponse; exports.adkToMcpToolType = adkToMcpToolType; exports.agentTransferRequestProcessor = requestProcessor8; exports.basicRequestProcessor = requestProcessor2; exports.buildFunctionDeclaration = buildFunctionDeclaration; exports.codeExecutionRequestProcessor = requestProcessor3; exports.codeExecutionResponseProcessor = responseProcessor; exports.contentRequestProcessor = requestProcessor4; exports.convertMcpToolToBaseTool = convertMcpToolToBaseTool; exports.createAuthToolArguments = createAuthToolArguments; exports.createBranchContextForSubAgent = createBranchContextForSubAgent; exports.createDatabaseSessionService = createDatabaseSessionService; exports.createFunctionTool = createFunctionTool; exports.createMysqlSessionService = createMysqlSessionService; exports.createPostgresSessionService = createPostgresSessionService; exports.createSamplingHandler = createSamplingHandler; exports.createSqliteSessionService = createSqliteSessionService; exports.createTool = createTool; exports.generateAuthEvent = generateAuthEvent; exports.generateClientFunctionCallId = generateClientFunctionCallId; exports.getArtifactUri = getArtifactUri; exports.getLongRunningFunctionCalls = getLongRunningFunctionCalls; exports.getMcpTools = getMcpTools; exports.handleFunctionCallsAsync = handleFunctionCallsAsync; exports.handleFunctionCallsLive = handleFunctionCallsLive; exports.identityRequestProcessor = requestProcessor5; exports.initializeTelemetry = initializeTelemetry; exports.injectSessionState = injectSessionState; exports.instructionsRequestProcessor = requestProcessor6; exports.isArtifactRef = isArtifactRef; exports.isEnhancedAuthConfig = isEnhancedAuthConfig; exports.jsonSchemaToDeclaration = jsonSchemaToDeclaration; exports.mcpSchemaToParameters = mcpSchemaToParameters; exports.mergeAgentRun = mergeAgentRun; exports.mergeParallelFunctionResponseEvents = mergeParallelFunctionResponseEvents; exports.newInvocationContextId = newInvocationContextId; exports.nlPlanningRequestProcessor = requestProcessor7; exports.nlPlanningResponseProcessor = responseProcessor2; exports.normalizeJsonSchema = normalizeJsonSchema; exports.parseArtifactUri = parseArtifactUri; exports.populateClientFunctionCallId = populateClientFunctionCallId; exports.registerProviders = registerProviders; exports.removeClientFunctionCallId = removeClientFunctionCallId; exports.requestProcessor = requestProcessor; exports.runCompactionForSlidingWindow = runCompactionForSlidingWindow; exports.shutdownTelemetry = shutdownTelemetry; exports.telemetryService = telemetryService; exports.traceLlmCall = traceLlmCall; exports.traceToolCall = traceToolCall; exports.tracer = tracer;

package/dist/index.mjs

@@ -3394,6 +3394,10 @@ var EventActions = class {
    * a compaction of events within the specified timestamp range.
    */
   compaction;
+  /**
+   * The invocation id to rewind to. This is only set for rewind event.
+   */
+  rewindBeforeInvocationId;
   /**
    * Constructor for EventActions
    */
@@ -3405,6 +3409,7 @@ var EventActions = class {
     this.escalate = options.escalate;
     this.requestedAuthConfigs = options.requestedAuthConfigs;
     this.compaction = options.compaction;
+    this.rewindBeforeInvocationId = options.rewindBeforeInvocationId;
   }
 };
 
@@ -8468,8 +8473,30 @@ function rearrangeEventsForLatestFunctionResponse(events) {
   return resultEvents;
 }
 function getContents(currentBranch, events, agentName = "") {
+  const invocationIdToIndex = /* @__PURE__ */ new Map();
+  for (let idx = 0; idx < events.length; idx++) {
+    if (events[idx].invocationId) {
+      invocationIdToIndex.set(events[idx].invocationId, idx);
+    }
+  }
+  const rewindFilteredEvents = [];
+  let i = events.length - 1;
+  while (i >= 0) {
+    const event = events[i];
+    if (event.actions?.rewindBeforeInvocationId) {
+      const rewindInvocationId = event.actions.rewindBeforeInvocationId;
+      const rewindIndex = invocationIdToIndex.get(rewindInvocationId);
+      if (rewindIndex !== void 0 && rewindIndex < i) {
+        i = rewindIndex;
+      }
+    } else {
+      rewindFilteredEvents.push(event);
+    }
+    i--;
+  }
+  rewindFilteredEvents.reverse();
   const filteredEvents = [];
-  for (const event of events) {
+  for (const event of rewindFilteredEvents) {
     if (!event.content || !event.content.role || !event.content.parts || event.content.parts.length === 0) {
       continue;
     }
@@ -10184,6 +10211,48 @@ var RunConfig = class {
   }
 };
 
+// src/artifacts/artifact-util.ts
+var SESSION_SCOPED_ARTIFACT_URI_RE = /^artifact:\/\/apps\/([^/]+)\/users\/([^/]+)\/sessions\/([^/]+)\/artifacts\/([^/]+)\/versions\/(\d+)$/;
+var USER_SCOPED_ARTIFACT_URI_RE = /^artifact:\/\/apps\/([^/]+)\/users\/([^/]+)\/artifacts\/([^/]+)\/versions\/(\d+)$/;
+function parseArtifactUri(uri) {
+  if (!uri || !uri.startsWith("artifact://")) {
+    return null;
+  }
+  let match = SESSION_SCOPED_ARTIFACT_URI_RE.exec(uri);
+  if (match) {
+    return {
+      appName: match[1],
+      userId: match[2],
+      sessionId: match[3],
+      filename: match[4],
+      version: Number.parseInt(match[5], 10)
+    };
+  }
+  match = USER_SCOPED_ARTIFACT_URI_RE.exec(uri);
+  if (match) {
+    return {
+      appName: match[1],
+      userId: match[2],
+      sessionId: void 0,
+      filename: match[3],
+      version: Number.parseInt(match[4], 10)
+    };
+  }
+  return null;
+}
+function getArtifactUri(args) {
+  const { appName, userId, filename, version, sessionId } = args;
+  if (sessionId) {
+    return `artifact://apps/${appName}/users/${userId}/sessions/${sessionId}/artifacts/${filename}/versions/${version}`;
+  }
+  return `artifact://apps/${appName}/users/${userId}/artifacts/${filename}/versions/${version}`;
+}
+function isArtifactRef(artifact) {
+  return Boolean(
+    artifact.fileData?.fileUri && artifact.fileData.fileUri.startsWith("artifact://")
+  );
+}
+
 // src/artifacts/in-memory-artifact-service.ts
 var InMemoryArtifactService = class {
   artifacts = /* @__PURE__ */ new Map();
@@ -10224,7 +10293,29 @@ var InMemoryArtifactService = class {
     if (targetVersion < 0 || targetVersion >= versions.length) {
       return null;
     }
-    return versions[targetVersion];
+    const artifactEntry = versions[targetVersion];
+    if (!artifactEntry) {
+      return null;
+    }
+    if (isArtifactRef(artifactEntry)) {
+      const parsedUri = parseArtifactUri(artifactEntry.fileData?.fileUri || "");
+      if (!parsedUri) {
+        throw new Error(
+          `Invalid artifact reference URI: ${artifactEntry.fileData?.fileUri}`
+        );
+      }
+      return await this.loadArtifact({
+        appName: parsedUri.appName,
+        userId: parsedUri.userId,
+        sessionId: parsedUri.sessionId || sessionId,
+        filename: parsedUri.filename,
+        version: parsedUri.version
+      });
+    }
+    if (!artifactEntry.text && (!artifactEntry.inlineData?.data || artifactEntry.inlineData.data.length === 0) && !artifactEntry.fileData) {
+      return null;
+    }
+    return artifactEntry;
   }
   async listArtifactKeys(args) {
     const { appName, userId, sessionId } = args;
@@ -10409,11 +10500,16 @@ var BaseSessionService = class {
       return;
     }
     for (const key in event.actions.stateDelta) {
-      if (Object.prototype.hasOwnProperty.call(event.actions.stateDelta, key)) {
+      if (Object.hasOwn(event.actions.stateDelta, key)) {
         if (key.startsWith("temp_")) {
           continue;
         }
-        session.state[key] = event.actions.stateDelta[key];
+        const value = event.actions.stateDelta[key];
+        if (value === null || value === void 0) {
+          delete session.state[key];
+        } else {
+          session.state[key] = value;
+        }
       }
     }
   }
@@ -10961,6 +11057,142 @@ var Runner = class {
     }
     return void 0;
   }
+  async rewind(args) {
+    const { userId, sessionId, rewindBeforeInvocationId } = args;
+    const session = await this.sessionService.getSession(
+      this.appName,
+      userId,
+      sessionId
+    );
+    if (!session) {
+      throw new Error(`Session not found: ${sessionId}`);
+    }
+    let rewindEventIndex = -1;
+    for (let i = 0; i < session.events.length; i++) {
+      if (session.events[i].invocationId === rewindBeforeInvocationId) {
+        rewindEventIndex = i;
+        break;
+      }
+    }
+    if (rewindEventIndex === -1) {
+      throw new Error(`Invocation ID not found: ${rewindBeforeInvocationId}`);
+    }
+    const stateDelta = await this._computeStateDeltaForRewind(
+      session,
+      rewindEventIndex
+    );
+    const artifactDelta = await this._computeArtifactDeltaForRewind(
+      session,
+      rewindEventIndex
+    );
+    const rewindEvent = new Event({
+      invocationId: newInvocationContextId(),
+      author: "user",
+      actions: new EventActions({
+        rewindBeforeInvocationId,
+        stateDelta,
+        artifactDelta
+      })
+    });
+    this.logger.info(
+      "Rewinding session to invocation:",
+      rewindBeforeInvocationId
+    );
+    await this.sessionService.appendEvent(session, rewindEvent);
+  }
+  async _computeStateDeltaForRewind(session, rewindEventIndex) {
+    const stateAtRewindPoint = {};
+    for (let i = 0; i < rewindEventIndex; i++) {
+      const event = session.events[i];
+      if (event.actions?.stateDelta) {
+        for (const [k, v] of Object.entries(event.actions.stateDelta)) {
+          if (k.startsWith("app:") || k.startsWith("user:")) {
+            continue;
+          }
+          if (v === null || v === void 0) {
+            delete stateAtRewindPoint[k];
+          } else {
+            stateAtRewindPoint[k] = v;
+          }
+        }
+      }
+    }
+    const currentState = session.state;
+    const rewindStateDelta = {};
+    for (const [key, valueAtRewind] of Object.entries(stateAtRewindPoint)) {
+      if (!(key in currentState) || currentState[key] !== valueAtRewind) {
+        rewindStateDelta[key] = valueAtRewind;
+      }
+    }
+    for (const key of Object.keys(currentState)) {
+      if (key.startsWith("app:") || key.startsWith("user:")) {
+        continue;
+      }
+      if (!(key in stateAtRewindPoint)) {
+        rewindStateDelta[key] = null;
+      }
+    }
+    return rewindStateDelta;
+  }
+  async _computeArtifactDeltaForRewind(session, rewindEventIndex) {
+    if (!this.artifactService) {
+      return {};
+    }
+    const versionsAtRewindPoint = {};
+    for (let i = 0; i < rewindEventIndex; i++) {
+      const event = session.events[i];
+      if (event.actions?.artifactDelta) {
+        Object.assign(versionsAtRewindPoint, event.actions.artifactDelta);
+      }
+    }
+    const currentVersions = {};
+    for (const event of session.events) {
+      if (event.actions?.artifactDelta) {
+        Object.assign(currentVersions, event.actions.artifactDelta);
+      }
+    }
+    const rewindArtifactDelta = {};
+    for (const [filename, vn] of Object.entries(currentVersions)) {
+      if (filename.startsWith("user:")) {
+        continue;
+      }
+      const vt = versionsAtRewindPoint[filename];
+      if (vt === vn) {
+        continue;
+      }
+      let artifact;
+      if (vt === void 0 || vt === null) {
+        artifact = {
+          inlineData: {
+            mimeType: "application/octet-stream",
+            data: ""
+          }
+        };
+      } else {
+        const artifactUri = getArtifactUri({
+          appName: this.appName,
+          userId: session.userId,
+          sessionId: session.id,
+          filename,
+          version: vt
+        });
+        artifact = {
+          fileData: {
+            fileUri: artifactUri
+          }
+        };
+      }
+      const newVersion = await this.artifactService.saveArtifact({
+        appName: this.appName,
+        userId: session.userId,
+        sessionId: session.id,
+        filename,
+        artifact
+      });
+      rewindArtifactDelta[filename] = newVersion;
+    }
+    return rewindArtifactDelta;
+  }
 };
 var InMemoryRunner = class extends Runner {
   /**
@@ -11664,6 +11896,9 @@ var AgentBuilder = class _AgentBuilder {
           ...params,
           runConfig: params.runConfig ?? runConfig
         });
+      },
+      rewind(params) {
+        return baseRunner.rewind(params);
       }
     };
   }
@@ -14257,6 +14492,7 @@ export {
   createTool,
   generateAuthEvent,
   generateClientFunctionCallId,
+  getArtifactUri,
   getLongRunningFunctionCalls,
   getMcpTools,
   handleFunctionCallsAsync,
@@ -14265,6 +14501,7 @@ export {
   initializeTelemetry,
   injectSessionState,
   requestProcessor6 as instructionsRequestProcessor,
+  isArtifactRef,
   isEnhancedAuthConfig,
   jsonSchemaToDeclaration,
   mcpSchemaToParameters,
@@ -14274,6 +14511,7 @@ export {
   requestProcessor7 as nlPlanningRequestProcessor,
   responseProcessor2 as nlPlanningResponseProcessor,
   normalizeJsonSchema,
+  parseArtifactUri,
   populateClientFunctionCallId,
   registerProviders,
   removeClientFunctionCallId,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@iqai/adk",
-  "version": "0.5.2",
+  "version": "0.5.3",
   "description": "Agent Development Kit for TypeScript with multi-provider LLM support",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",