@iqai/adk 0.1.11 → 0.1.12

package/CHANGELOG.md

@@ -1,5 +1,11 @@
 # @iqai/adk
 
+## 0.1.12
+
+### Patch Changes
+
+- 1b9f517: - Add support for withMemory and withArtifactService methods to agent builder instead of passing it from withSession
+
 ## 0.1.11
 
 ### Patch Changes

package/dist/index.d.mts

@@ -3901,14 +3901,11 @@ interface AgentBuilderConfig {
     rootNode?: string;
 }
 /**
- * Session configuration for the AgentBuilder
+ * Session configuration options
  */
-interface SessionConfig {
-    service: BaseSessionService;
-    userId: string;
-    appName: string;
-    memoryService?: BaseMemoryService;
-    artifactService?: BaseArtifactService;
+interface SessionOptions {
+    userId?: string;
+    appName?: string;
 }
 /**
  * Message part interface for flexible message input
@@ -3965,6 +3962,15 @@ type AgentType = "llm" | "sequential" | "parallel" | "loop" | "langgraph";
  *   .withInstruction("You are a research assistant")
  *   .build();
  *
+ * // With memory and artifact services
+ * const { runner } = await AgentBuilder
+ *   .create("persistent-agent")
+ *   .withModel("gemini-2.5-flash")
+ *   .withMemory(new RedisMemoryService())
+ *   .withArtifactService(new S3ArtifactService())
+ *   .withSession(new DatabaseSessionService(), { userId: "user123", appName: "myapp" })
+ *   .build();
+ *
  * // Multi-agent workflow
  * const { runner } = await AgentBuilder
  *   .create("workflow")
@@ -3975,6 +3981,8 @@ type AgentType = "llm" | "sequential" | "parallel" | "loop" | "langgraph";
 declare class AgentBuilder {
     private config;
     private sessionConfig?;
+    private memoryService?;
+    private artifactService?;
     private agentType;
     /**
      * Private constructor - use static create() method
@@ -4051,21 +4059,29 @@ declare class AgentBuilder {
     /**
      * Configure session management with optional smart defaults
      * @param service Session service to use
-     * @param userId User identifier (optional, defaults to agent-based ID)
-     * @param appName Application name (optional, defaults to agent-based name)
-     * @param memoryService Optional memory service
-     * @param artifactService Optional artifact service
+     * @param options Session configuration options (userId and appName)
+     * @returns This builder instance for chaining
+     */
+    withSession(service: BaseSessionService, options?: SessionOptions): this;
+    /**
+     * Configure memory service for the agent
+     * @param memoryService Memory service to use for conversation history and context
+     * @returns This builder instance for chaining
+     */
+    withMemory(memoryService: BaseMemoryService): this;
+    /**
+     * Configure artifact service for the agent
+     * @param artifactService Artifact service to use for managing generated artifacts
      * @returns This builder instance for chaining
      */
-    withSession(service: BaseSessionService, userId?: string, appName?: string, memoryService?: BaseMemoryService, artifactService?: BaseArtifactService): this;
+    withArtifactService(artifactService: BaseArtifactService): this;
     /**
      * Configure with an in-memory session with custom IDs
      * Note: In-memory sessions are created automatically by default, use this only if you need custom appName/userId
-     * @param appName Application name (optional, defaults to agent-based name)
-     * @param userId User identifier (optional, defaults to agent-based ID)
+     * @param options Session configuration options (userId and appName)
      * @returns This builder instance for chaining
      */
-    withQuickSession(appName?: string, userId?: string): this;
+    withQuickSession(options?: SessionOptions): this;
     /**
      * Build the agent and optionally create runner and session
      * @returns Built agent with optional runner and session
@@ -4140,7 +4156,7 @@ declare const index$4_RunConfig: typeof RunConfig;
 type index$4_SequentialAgent = SequentialAgent;
 declare const index$4_SequentialAgent: typeof SequentialAgent;
 type index$4_SequentialAgentConfig = SequentialAgentConfig;
-type index$4_SessionConfig = SessionConfig;
+type index$4_SessionOptions = SessionOptions;
 type index$4_SingleAgentCallback = SingleAgentCallback;
 type index$4_StreamingMode = StreamingMode;
 declare const index$4_StreamingMode: typeof StreamingMode;
@@ -4149,7 +4165,7 @@ declare const index$4_createBranchContextForSubAgent: typeof createBranchContext
 declare const index$4_mergeAgentRun: typeof mergeAgentRun;
 declare const index$4_newInvocationContextId: typeof newInvocationContextId;
 declare namespace index$4 {
-  export { type index$4_AfterAgentCallback as AfterAgentCallback, LlmAgent as Agent, index$4_AgentBuilder as AgentBuilder, type index$4_AgentBuilderConfig as AgentBuilderConfig, type index$4_AgentType as AgentType, index$4_BaseAgent as BaseAgent, type index$4_BeforeAgentCallback as BeforeAgentCallback, type index$4_BuiltAgent as BuiltAgent, index$4_CallbackContext as CallbackContext, type index$4_EnhancedRunner as EnhancedRunner, type index$4_FullMessage as FullMessage, type index$4_InstructionProvider as InstructionProvider, index$4_InvocationContext as InvocationContext, index$4_LangGraphAgent as LangGraphAgent, type index$4_LangGraphAgentConfig as LangGraphAgentConfig, type index$4_LangGraphNode as LangGraphNode, index$4_LlmAgent as LlmAgent, type index$4_LlmAgentConfig as LlmAgentConfig, index$4_LlmCallsLimitExceededError as LlmCallsLimitExceededError, index$4_LoopAgent as LoopAgent, type index$4_LoopAgentConfig as LoopAgentConfig, type index$4_MessagePart as MessagePart, index$4_ParallelAgent as ParallelAgent, type index$4_ParallelAgentConfig as ParallelAgentConfig, index$4_ReadonlyContext as ReadonlyContext, index$4_RunConfig as RunConfig, index$4_SequentialAgent as SequentialAgent, type index$4_SequentialAgentConfig as SequentialAgentConfig, type index$4_SessionConfig as SessionConfig, type index$4_SingleAgentCallback as SingleAgentCallback, index$4_StreamingMode as StreamingMode, type index$4_ToolUnion as ToolUnion, index$4_createBranchContextForSubAgent as createBranchContextForSubAgent, index$4_mergeAgentRun as mergeAgentRun, index$4_newInvocationContextId as newInvocationContextId };
+  export { type index$4_AfterAgentCallback as AfterAgentCallback, LlmAgent as Agent, index$4_AgentBuilder as AgentBuilder, type index$4_AgentBuilderConfig as AgentBuilderConfig, type index$4_AgentType as AgentType, index$4_BaseAgent as BaseAgent, type index$4_BeforeAgentCallback as BeforeAgentCallback, type index$4_BuiltAgent as BuiltAgent, index$4_CallbackContext as CallbackContext, type index$4_EnhancedRunner as EnhancedRunner, type index$4_FullMessage as FullMessage, type index$4_InstructionProvider as InstructionProvider, index$4_InvocationContext as InvocationContext, index$4_LangGraphAgent as LangGraphAgent, type index$4_LangGraphAgentConfig as LangGraphAgentConfig, type index$4_LangGraphNode as LangGraphNode, index$4_LlmAgent as LlmAgent, type index$4_LlmAgentConfig as LlmAgentConfig, index$4_LlmCallsLimitExceededError as LlmCallsLimitExceededError, index$4_LoopAgent as LoopAgent, type index$4_LoopAgentConfig as LoopAgentConfig, type index$4_MessagePart as MessagePart, index$4_ParallelAgent as ParallelAgent, type index$4_ParallelAgentConfig as ParallelAgentConfig, index$4_ReadonlyContext as ReadonlyContext, index$4_RunConfig as RunConfig, index$4_SequentialAgent as SequentialAgent, type index$4_SequentialAgentConfig as SequentialAgentConfig, type index$4_SessionOptions as SessionOptions, type index$4_SingleAgentCallback as SingleAgentCallback, index$4_StreamingMode as StreamingMode, type index$4_ToolUnion as ToolUnion, index$4_createBranchContextForSubAgent as createBranchContextForSubAgent, index$4_mergeAgentRun as mergeAgentRun, index$4_newInvocationContextId as newInvocationContextId };
 }
 
 /**
@@ -5091,4 +5107,4 @@ declare const traceLlmCall: (invocationContext: InvocationContext, eventId: stri
 
 declare const VERSION = "0.1.0";
 
-export { AF_FUNCTION_CALL_ID_PREFIX, type AfterAgentCallback, LlmAgent as Agent, AgentBuilder, type AgentBuilderConfig, type AgentType, index$4 as Agents, AiSdkLlm, AnthropicLlm, ApiKeyCredential, ApiKeyScheme, AuthConfig, AuthCredential, AuthCredentialType, AuthHandler, AuthScheme, AuthSchemeType, AuthTool, type AuthToolArguments, AutoFlow, BaseAgent, BaseCodeExecutor, type BaseCodeExecutorConfig, BaseLLMConnection, BaseLlm, BaseLlmFlow, BaseLlmRequestProcessor, BaseLlmResponseProcessor, type BaseMemoryService, BasePlanner, BaseSessionService, BaseTool, BasicAuthCredential, BearerTokenCredential, type BeforeAgentCallback, type BuildFunctionDeclarationOptions, type BuiltAgent, BuiltInCodeExecutor, BuiltInPlanner, CallbackContext, type CodeExecutionInput, type CodeExecutionResult, CodeExecutionUtils, CodeExecutorContext, type CreateToolConfig, type CreateToolConfigWithSchema, type CreateToolConfigWithoutSchema, DatabaseSessionService, EnhancedAuthConfig, type EnhancedRunner, Event, EventActions, index$1 as Events, ExitLoopTool, type File, FileOperationsTool, index as Flows, type FullMessage, type FunctionDeclaration, FunctionTool, GcsArtifactService, type GetSessionConfig, GetUserChoiceTool, GoogleLlm, GoogleSearch, HttpRequestTool, HttpScheme, InMemoryArtifactService, InMemoryMemoryService, InMemoryRunner, InMemorySessionService, type InstructionProvider, InvocationContext, type JSONSchema, LLMRegistry, LangGraphAgent, type LangGraphAgentConfig, type LangGraphNode, type ListSessionsResponse, LlmAgent, type LlmAgentConfig, LlmCallsLimitExceededError, LlmRequest, LlmResponse, LoadArtifactsTool, LoadMemoryTool, LoopAgent, type LoopAgentConfig, McpAbi, McpAtp, McpBamm, McpCoinGecko, type McpConfig, McpDiscord, McpError, McpErrorType, McpFilesystem, McpFraxlend, McpGeneric, McpIqWiki, McpMemory, McpNearAgent, McpNearIntentSwaps, McpOdos, McpSamplingHandler, type McpSamplingRequest, type McpSamplingResponse, type McpServerConfig, McpTelegram, McpToolset, type McpTransportType, index$3 as Memory, type MessagePart, index$5 as Models, OAuth2Credential, OAuth2Scheme, type OAuthFlow, type OAuthFlows, OpenAiLlm, OpenIdConnectScheme, ParallelAgent, type ParallelAgentConfig, PlanReActPlanner, REQUEST_EUC_FUNCTION_CALL_NAME, ReadonlyContext, RunConfig, Runner, type SamplingHandler, type SearchMemoryResponse, SequentialAgent, type SequentialAgentConfig, type Session, type SessionConfig, index$2 as Sessions, type SingleAgentCallback, SingleFlow, State, StreamingMode, type TelemetryConfig, TelemetryService, type ThinkingConfig, type ToolConfig, ToolContext, type ToolUnion, index$6 as Tools, TransferToAgentTool, UserInteractionTool, VERSION, VertexAiSessionService, _findFunctionCallEventIfLastEventIsFunctionResponse, adkToMcpToolType, requestProcessor$2 as agentTransferRequestProcessor, requestProcessor$6 as basicRequestProcessor, buildFunctionDeclaration, requestProcessor as codeExecutionRequestProcessor, responseProcessor as codeExecutionResponseProcessor, requestProcessor$3 as contentRequestProcessor, 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, shutdownTelemetry, telemetryService, traceLlmCall, traceToolCall, tracer };
+export { AF_FUNCTION_CALL_ID_PREFIX, type AfterAgentCallback, LlmAgent as Agent, AgentBuilder, type AgentBuilderConfig, type AgentType, index$4 as Agents, AiSdkLlm, AnthropicLlm, ApiKeyCredential, ApiKeyScheme, AuthConfig, AuthCredential, AuthCredentialType, AuthHandler, AuthScheme, AuthSchemeType, AuthTool, type AuthToolArguments, AutoFlow, BaseAgent, BaseCodeExecutor, type BaseCodeExecutorConfig, BaseLLMConnection, BaseLlm, BaseLlmFlow, BaseLlmRequestProcessor, BaseLlmResponseProcessor, type BaseMemoryService, BasePlanner, BaseSessionService, BaseTool, BasicAuthCredential, BearerTokenCredential, type BeforeAgentCallback, type BuildFunctionDeclarationOptions, type BuiltAgent, BuiltInCodeExecutor, BuiltInPlanner, CallbackContext, type CodeExecutionInput, type CodeExecutionResult, CodeExecutionUtils, CodeExecutorContext, type CreateToolConfig, type CreateToolConfigWithSchema, type CreateToolConfigWithoutSchema, DatabaseSessionService, EnhancedAuthConfig, type EnhancedRunner, Event, EventActions, index$1 as Events, ExitLoopTool, type File, FileOperationsTool, index as Flows, type FullMessage, type FunctionDeclaration, FunctionTool, GcsArtifactService, type GetSessionConfig, GetUserChoiceTool, GoogleLlm, GoogleSearch, HttpRequestTool, HttpScheme, InMemoryArtifactService, InMemoryMemoryService, InMemoryRunner, InMemorySessionService, type InstructionProvider, InvocationContext, type JSONSchema, LLMRegistry, LangGraphAgent, type LangGraphAgentConfig, type LangGraphNode, type ListSessionsResponse, LlmAgent, type LlmAgentConfig, LlmCallsLimitExceededError, LlmRequest, LlmResponse, LoadArtifactsTool, LoadMemoryTool, LoopAgent, type LoopAgentConfig, McpAbi, McpAtp, McpBamm, McpCoinGecko, type McpConfig, McpDiscord, McpError, McpErrorType, McpFilesystem, McpFraxlend, McpGeneric, McpIqWiki, McpMemory, McpNearAgent, McpNearIntentSwaps, McpOdos, McpSamplingHandler, type McpSamplingRequest, type McpSamplingResponse, type McpServerConfig, McpTelegram, McpToolset, type McpTransportType, index$3 as Memory, type MessagePart, index$5 as Models, OAuth2Credential, OAuth2Scheme, type OAuthFlow, type OAuthFlows, OpenAiLlm, OpenIdConnectScheme, ParallelAgent, type ParallelAgentConfig, PlanReActPlanner, REQUEST_EUC_FUNCTION_CALL_NAME, ReadonlyContext, RunConfig, Runner, type SamplingHandler, type SearchMemoryResponse, SequentialAgent, type SequentialAgentConfig, type Session, type SessionOptions, index$2 as Sessions, type SingleAgentCallback, SingleFlow, State, StreamingMode, type TelemetryConfig, TelemetryService, type ThinkingConfig, type ToolConfig, ToolContext, type ToolUnion, index$6 as Tools, TransferToAgentTool, UserInteractionTool, VERSION, VertexAiSessionService, _findFunctionCallEventIfLastEventIsFunctionResponse, adkToMcpToolType, requestProcessor$2 as agentTransferRequestProcessor, requestProcessor$6 as basicRequestProcessor, buildFunctionDeclaration, requestProcessor as codeExecutionRequestProcessor, responseProcessor as codeExecutionResponseProcessor, requestProcessor$3 as contentRequestProcessor, 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, shutdownTelemetry, telemetryService, traceLlmCall, traceToolCall, tracer };

package/dist/index.d.ts

@@ -3901,14 +3901,11 @@ interface AgentBuilderConfig {
     rootNode?: string;
 }
 /**
- * Session configuration for the AgentBuilder
+ * Session configuration options
  */
-interface SessionConfig {
-    service: BaseSessionService;
-    userId: string;
-    appName: string;
-    memoryService?: BaseMemoryService;
-    artifactService?: BaseArtifactService;
+interface SessionOptions {
+    userId?: string;
+    appName?: string;
 }
 /**
  * Message part interface for flexible message input
@@ -3965,6 +3962,15 @@ type AgentType = "llm" | "sequential" | "parallel" | "loop" | "langgraph";
  *   .withInstruction("You are a research assistant")
  *   .build();
  *
+ * // With memory and artifact services
+ * const { runner } = await AgentBuilder
+ *   .create("persistent-agent")
+ *   .withModel("gemini-2.5-flash")
+ *   .withMemory(new RedisMemoryService())
+ *   .withArtifactService(new S3ArtifactService())
+ *   .withSession(new DatabaseSessionService(), { userId: "user123", appName: "myapp" })
+ *   .build();
+ *
  * // Multi-agent workflow
  * const { runner } = await AgentBuilder
  *   .create("workflow")
@@ -3975,6 +3981,8 @@ type AgentType = "llm" | "sequential" | "parallel" | "loop" | "langgraph";
 declare class AgentBuilder {
     private config;
     private sessionConfig?;
+    private memoryService?;
+    private artifactService?;
     private agentType;
     /**
      * Private constructor - use static create() method
@@ -4051,21 +4059,29 @@ declare class AgentBuilder {
     /**
      * Configure session management with optional smart defaults
      * @param service Session service to use
-     * @param userId User identifier (optional, defaults to agent-based ID)
-     * @param appName Application name (optional, defaults to agent-based name)
-     * @param memoryService Optional memory service
-     * @param artifactService Optional artifact service
+     * @param options Session configuration options (userId and appName)
+     * @returns This builder instance for chaining
+     */
+    withSession(service: BaseSessionService, options?: SessionOptions): this;
+    /**
+     * Configure memory service for the agent
+     * @param memoryService Memory service to use for conversation history and context
+     * @returns This builder instance for chaining
+     */
+    withMemory(memoryService: BaseMemoryService): this;
+    /**
+     * Configure artifact service for the agent
+     * @param artifactService Artifact service to use for managing generated artifacts
      * @returns This builder instance for chaining
      */
-    withSession(service: BaseSessionService, userId?: string, appName?: string, memoryService?: BaseMemoryService, artifactService?: BaseArtifactService): this;
+    withArtifactService(artifactService: BaseArtifactService): this;
     /**
      * Configure with an in-memory session with custom IDs
      * Note: In-memory sessions are created automatically by default, use this only if you need custom appName/userId
-     * @param appName Application name (optional, defaults to agent-based name)
-     * @param userId User identifier (optional, defaults to agent-based ID)
+     * @param options Session configuration options (userId and appName)
      * @returns This builder instance for chaining
      */
-    withQuickSession(appName?: string, userId?: string): this;
+    withQuickSession(options?: SessionOptions): this;
     /**
      * Build the agent and optionally create runner and session
      * @returns Built agent with optional runner and session
@@ -4140,7 +4156,7 @@ declare const index$4_RunConfig: typeof RunConfig;
 type index$4_SequentialAgent = SequentialAgent;
 declare const index$4_SequentialAgent: typeof SequentialAgent;
 type index$4_SequentialAgentConfig = SequentialAgentConfig;
-type index$4_SessionConfig = SessionConfig;
+type index$4_SessionOptions = SessionOptions;
 type index$4_SingleAgentCallback = SingleAgentCallback;
 type index$4_StreamingMode = StreamingMode;
 declare const index$4_StreamingMode: typeof StreamingMode;
@@ -4149,7 +4165,7 @@ declare const index$4_createBranchContextForSubAgent: typeof createBranchContext
 declare const index$4_mergeAgentRun: typeof mergeAgentRun;
 declare const index$4_newInvocationContextId: typeof newInvocationContextId;
 declare namespace index$4 {
-  export { type index$4_AfterAgentCallback as AfterAgentCallback, LlmAgent as Agent, index$4_AgentBuilder as AgentBuilder, type index$4_AgentBuilderConfig as AgentBuilderConfig, type index$4_AgentType as AgentType, index$4_BaseAgent as BaseAgent, type index$4_BeforeAgentCallback as BeforeAgentCallback, type index$4_BuiltAgent as BuiltAgent, index$4_CallbackContext as CallbackContext, type index$4_EnhancedRunner as EnhancedRunner, type index$4_FullMessage as FullMessage, type index$4_InstructionProvider as InstructionProvider, index$4_InvocationContext as InvocationContext, index$4_LangGraphAgent as LangGraphAgent, type index$4_LangGraphAgentConfig as LangGraphAgentConfig, type index$4_LangGraphNode as LangGraphNode, index$4_LlmAgent as LlmAgent, type index$4_LlmAgentConfig as LlmAgentConfig, index$4_LlmCallsLimitExceededError as LlmCallsLimitExceededError, index$4_LoopAgent as LoopAgent, type index$4_LoopAgentConfig as LoopAgentConfig, type index$4_MessagePart as MessagePart, index$4_ParallelAgent as ParallelAgent, type index$4_ParallelAgentConfig as ParallelAgentConfig, index$4_ReadonlyContext as ReadonlyContext, index$4_RunConfig as RunConfig, index$4_SequentialAgent as SequentialAgent, type index$4_SequentialAgentConfig as SequentialAgentConfig, type index$4_SessionConfig as SessionConfig, type index$4_SingleAgentCallback as SingleAgentCallback, index$4_StreamingMode as StreamingMode, type index$4_ToolUnion as ToolUnion, index$4_createBranchContextForSubAgent as createBranchContextForSubAgent, index$4_mergeAgentRun as mergeAgentRun, index$4_newInvocationContextId as newInvocationContextId };
+  export { type index$4_AfterAgentCallback as AfterAgentCallback, LlmAgent as Agent, index$4_AgentBuilder as AgentBuilder, type index$4_AgentBuilderConfig as AgentBuilderConfig, type index$4_AgentType as AgentType, index$4_BaseAgent as BaseAgent, type index$4_BeforeAgentCallback as BeforeAgentCallback, type index$4_BuiltAgent as BuiltAgent, index$4_CallbackContext as CallbackContext, type index$4_EnhancedRunner as EnhancedRunner, type index$4_FullMessage as FullMessage, type index$4_InstructionProvider as InstructionProvider, index$4_InvocationContext as InvocationContext, index$4_LangGraphAgent as LangGraphAgent, type index$4_LangGraphAgentConfig as LangGraphAgentConfig, type index$4_LangGraphNode as LangGraphNode, index$4_LlmAgent as LlmAgent, type index$4_LlmAgentConfig as LlmAgentConfig, index$4_LlmCallsLimitExceededError as LlmCallsLimitExceededError, index$4_LoopAgent as LoopAgent, type index$4_LoopAgentConfig as LoopAgentConfig, type index$4_MessagePart as MessagePart, index$4_ParallelAgent as ParallelAgent, type index$4_ParallelAgentConfig as ParallelAgentConfig, index$4_ReadonlyContext as ReadonlyContext, index$4_RunConfig as RunConfig, index$4_SequentialAgent as SequentialAgent, type index$4_SequentialAgentConfig as SequentialAgentConfig, type index$4_SessionOptions as SessionOptions, type index$4_SingleAgentCallback as SingleAgentCallback, index$4_StreamingMode as StreamingMode, type index$4_ToolUnion as ToolUnion, index$4_createBranchContextForSubAgent as createBranchContextForSubAgent, index$4_mergeAgentRun as mergeAgentRun, index$4_newInvocationContextId as newInvocationContextId };
 }
 
 /**
@@ -5091,4 +5107,4 @@ declare const traceLlmCall: (invocationContext: InvocationContext, eventId: stri
 
 declare const VERSION = "0.1.0";
 
-export { AF_FUNCTION_CALL_ID_PREFIX, type AfterAgentCallback, LlmAgent as Agent, AgentBuilder, type AgentBuilderConfig, type AgentType, index$4 as Agents, AiSdkLlm, AnthropicLlm, ApiKeyCredential, ApiKeyScheme, AuthConfig, AuthCredential, AuthCredentialType, AuthHandler, AuthScheme, AuthSchemeType, AuthTool, type AuthToolArguments, AutoFlow, BaseAgent, BaseCodeExecutor, type BaseCodeExecutorConfig, BaseLLMConnection, BaseLlm, BaseLlmFlow, BaseLlmRequestProcessor, BaseLlmResponseProcessor, type BaseMemoryService, BasePlanner, BaseSessionService, BaseTool, BasicAuthCredential, BearerTokenCredential, type BeforeAgentCallback, type BuildFunctionDeclarationOptions, type BuiltAgent, BuiltInCodeExecutor, BuiltInPlanner, CallbackContext, type CodeExecutionInput, type CodeExecutionResult, CodeExecutionUtils, CodeExecutorContext, type CreateToolConfig, type CreateToolConfigWithSchema, type CreateToolConfigWithoutSchema, DatabaseSessionService, EnhancedAuthConfig, type EnhancedRunner, Event, EventActions, index$1 as Events, ExitLoopTool, type File, FileOperationsTool, index as Flows, type FullMessage, type FunctionDeclaration, FunctionTool, GcsArtifactService, type GetSessionConfig, GetUserChoiceTool, GoogleLlm, GoogleSearch, HttpRequestTool, HttpScheme, InMemoryArtifactService, InMemoryMemoryService, InMemoryRunner, InMemorySessionService, type InstructionProvider, InvocationContext, type JSONSchema, LLMRegistry, LangGraphAgent, type LangGraphAgentConfig, type LangGraphNode, type ListSessionsResponse, LlmAgent, type LlmAgentConfig, LlmCallsLimitExceededError, LlmRequest, LlmResponse, LoadArtifactsTool, LoadMemoryTool, LoopAgent, type LoopAgentConfig, McpAbi, McpAtp, McpBamm, McpCoinGecko, type McpConfig, McpDiscord, McpError, McpErrorType, McpFilesystem, McpFraxlend, McpGeneric, McpIqWiki, McpMemory, McpNearAgent, McpNearIntentSwaps, McpOdos, McpSamplingHandler, type McpSamplingRequest, type McpSamplingResponse, type McpServerConfig, McpTelegram, McpToolset, type McpTransportType, index$3 as Memory, type MessagePart, index$5 as Models, OAuth2Credential, OAuth2Scheme, type OAuthFlow, type OAuthFlows, OpenAiLlm, OpenIdConnectScheme, ParallelAgent, type ParallelAgentConfig, PlanReActPlanner, REQUEST_EUC_FUNCTION_CALL_NAME, ReadonlyContext, RunConfig, Runner, type SamplingHandler, type SearchMemoryResponse, SequentialAgent, type SequentialAgentConfig, type Session, type SessionConfig, index$2 as Sessions, type SingleAgentCallback, SingleFlow, State, StreamingMode, type TelemetryConfig, TelemetryService, type ThinkingConfig, type ToolConfig, ToolContext, type ToolUnion, index$6 as Tools, TransferToAgentTool, UserInteractionTool, VERSION, VertexAiSessionService, _findFunctionCallEventIfLastEventIsFunctionResponse, adkToMcpToolType, requestProcessor$2 as agentTransferRequestProcessor, requestProcessor$6 as basicRequestProcessor, buildFunctionDeclaration, requestProcessor as codeExecutionRequestProcessor, responseProcessor as codeExecutionResponseProcessor, requestProcessor$3 as contentRequestProcessor, 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, shutdownTelemetry, telemetryService, traceLlmCall, traceToolCall, tracer };
+export { AF_FUNCTION_CALL_ID_PREFIX, type AfterAgentCallback, LlmAgent as Agent, AgentBuilder, type AgentBuilderConfig, type AgentType, index$4 as Agents, AiSdkLlm, AnthropicLlm, ApiKeyCredential, ApiKeyScheme, AuthConfig, AuthCredential, AuthCredentialType, AuthHandler, AuthScheme, AuthSchemeType, AuthTool, type AuthToolArguments, AutoFlow, BaseAgent, BaseCodeExecutor, type BaseCodeExecutorConfig, BaseLLMConnection, BaseLlm, BaseLlmFlow, BaseLlmRequestProcessor, BaseLlmResponseProcessor, type BaseMemoryService, BasePlanner, BaseSessionService, BaseTool, BasicAuthCredential, BearerTokenCredential, type BeforeAgentCallback, type BuildFunctionDeclarationOptions, type BuiltAgent, BuiltInCodeExecutor, BuiltInPlanner, CallbackContext, type CodeExecutionInput, type CodeExecutionResult, CodeExecutionUtils, CodeExecutorContext, type CreateToolConfig, type CreateToolConfigWithSchema, type CreateToolConfigWithoutSchema, DatabaseSessionService, EnhancedAuthConfig, type EnhancedRunner, Event, EventActions, index$1 as Events, ExitLoopTool, type File, FileOperationsTool, index as Flows, type FullMessage, type FunctionDeclaration, FunctionTool, GcsArtifactService, type GetSessionConfig, GetUserChoiceTool, GoogleLlm, GoogleSearch, HttpRequestTool, HttpScheme, InMemoryArtifactService, InMemoryMemoryService, InMemoryRunner, InMemorySessionService, type InstructionProvider, InvocationContext, type JSONSchema, LLMRegistry, LangGraphAgent, type LangGraphAgentConfig, type LangGraphNode, type ListSessionsResponse, LlmAgent, type LlmAgentConfig, LlmCallsLimitExceededError, LlmRequest, LlmResponse, LoadArtifactsTool, LoadMemoryTool, LoopAgent, type LoopAgentConfig, McpAbi, McpAtp, McpBamm, McpCoinGecko, type McpConfig, McpDiscord, McpError, McpErrorType, McpFilesystem, McpFraxlend, McpGeneric, McpIqWiki, McpMemory, McpNearAgent, McpNearIntentSwaps, McpOdos, McpSamplingHandler, type McpSamplingRequest, type McpSamplingResponse, type McpServerConfig, McpTelegram, McpToolset, type McpTransportType, index$3 as Memory, type MessagePart, index$5 as Models, OAuth2Credential, OAuth2Scheme, type OAuthFlow, type OAuthFlows, OpenAiLlm, OpenIdConnectScheme, ParallelAgent, type ParallelAgentConfig, PlanReActPlanner, REQUEST_EUC_FUNCTION_CALL_NAME, ReadonlyContext, RunConfig, Runner, type SamplingHandler, type SearchMemoryResponse, SequentialAgent, type SequentialAgentConfig, type Session, type SessionOptions, index$2 as Sessions, type SingleAgentCallback, SingleFlow, State, StreamingMode, type TelemetryConfig, TelemetryService, type ThinkingConfig, type ToolConfig, ToolContext, type ToolUnion, index$6 as Tools, TransferToAgentTool, UserInteractionTool, VERSION, VertexAiSessionService, _findFunctionCallEventIfLastEventIsFunctionResponse, adkToMcpToolType, requestProcessor$2 as agentTransferRequestProcessor, requestProcessor$6 as basicRequestProcessor, buildFunctionDeclaration, requestProcessor as codeExecutionRequestProcessor, responseProcessor as codeExecutionResponseProcessor, requestProcessor$3 as contentRequestProcessor, 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, shutdownTelemetry, telemetryService, traceLlmCall, traceToolCall, tracer };

package/dist/index.js

@@ -9886,6 +9886,8 @@ var InMemoryRunner = class extends Runner {
 var AgentBuilder = (_class31 = class _AgentBuilder {
   
   
+  
+  
   __init54() {this.agentType = "llm"}
   /**
    * Private constructor - use static create() method
@@ -10001,31 +10003,43 @@ var AgentBuilder = (_class31 = class _AgentBuilder {
   /**
    * Configure session management with optional smart defaults
    * @param service Session service to use
-   * @param userId User identifier (optional, defaults to agent-based ID)
-   * @param appName Application name (optional, defaults to agent-based name)
-   * @param memoryService Optional memory service
-   * @param artifactService Optional artifact service
+   * @param options Session configuration options (userId and appName)
    * @returns This builder instance for chaining
    */
-  withSession(service, userId, appName, memoryService, artifactService) {
+  withSession(service, options = {}) {
     this.sessionConfig = {
       service,
-      userId: userId || this.generateDefaultUserId(),
-      appName: appName || this.generateDefaultAppName(),
-      memoryService,
-      artifactService
+      userId: options.userId || this.generateDefaultUserId(),
+      appName: options.appName || this.generateDefaultAppName()
     };
     return this;
   }
+  /**
+   * Configure memory service for the agent
+   * @param memoryService Memory service to use for conversation history and context
+   * @returns This builder instance for chaining
+   */
+  withMemory(memoryService) {
+    this.memoryService = memoryService;
+    return this;
+  }
+  /**
+   * Configure artifact service for the agent
+   * @param artifactService Artifact service to use for managing generated artifacts
+   * @returns This builder instance for chaining
+   */
+  withArtifactService(artifactService) {
+    this.artifactService = artifactService;
+    return this;
+  }
   /**
    * Configure with an in-memory session with custom IDs
    * Note: In-memory sessions are created automatically by default, use this only if you need custom appName/userId
-   * @param appName Application name (optional, defaults to agent-based name)
-   * @param userId User identifier (optional, defaults to agent-based ID)
+   * @param options Session configuration options (userId and appName)
    * @returns This builder instance for chaining
    */
-  withQuickSession(appName, userId) {
-    return this.withSession(new InMemorySessionService(), userId, appName);
+  withQuickSession(options = {}) {
+    return this.withSession(new InMemorySessionService(), options);
   }
   /**
    * Build the agent and optionally create runner and session
@@ -10047,8 +10061,8 @@ var AgentBuilder = (_class31 = class _AgentBuilder {
         appName: this.sessionConfig.appName,
         agent,
         sessionService: this.sessionConfig.service,
-        memoryService: this.sessionConfig.memoryService,
-        artifactService: this.sessionConfig.artifactService
+        memoryService: this.memoryService,
+        artifactService: this.artifactService
       };
       const baseRunner = new Runner(runnerConfig);
       runner = this.createEnhancedRunner(baseRunner, session);

package/dist/index.mjs

@@ -9886,6 +9886,8 @@ var InMemoryRunner = class extends Runner {
 var AgentBuilder = class _AgentBuilder {
   config;
   sessionConfig;
+  memoryService;
+  artifactService;
   agentType = "llm";
   /**
    * Private constructor - use static create() method
@@ -10001,31 +10003,43 @@ var AgentBuilder = class _AgentBuilder {
   /**
    * Configure session management with optional smart defaults
    * @param service Session service to use
-   * @param userId User identifier (optional, defaults to agent-based ID)
-   * @param appName Application name (optional, defaults to agent-based name)
-   * @param memoryService Optional memory service
-   * @param artifactService Optional artifact service
+   * @param options Session configuration options (userId and appName)
    * @returns This builder instance for chaining
    */
-  withSession(service, userId, appName, memoryService, artifactService) {
+  withSession(service, options = {}) {
     this.sessionConfig = {
       service,
-      userId: userId || this.generateDefaultUserId(),
-      appName: appName || this.generateDefaultAppName(),
-      memoryService,
-      artifactService
+      userId: options.userId || this.generateDefaultUserId(),
+      appName: options.appName || this.generateDefaultAppName()
     };
     return this;
   }
+  /**
+   * Configure memory service for the agent
+   * @param memoryService Memory service to use for conversation history and context
+   * @returns This builder instance for chaining
+   */
+  withMemory(memoryService) {
+    this.memoryService = memoryService;
+    return this;
+  }
+  /**
+   * Configure artifact service for the agent
+   * @param artifactService Artifact service to use for managing generated artifacts
+   * @returns This builder instance for chaining
+   */
+  withArtifactService(artifactService) {
+    this.artifactService = artifactService;
+    return this;
+  }
   /**
    * Configure with an in-memory session with custom IDs
    * Note: In-memory sessions are created automatically by default, use this only if you need custom appName/userId
-   * @param appName Application name (optional, defaults to agent-based name)
-   * @param userId User identifier (optional, defaults to agent-based ID)
+   * @param options Session configuration options (userId and appName)
    * @returns This builder instance for chaining
    */
-  withQuickSession(appName, userId) {
-    return this.withSession(new InMemorySessionService(), userId, appName);
+  withQuickSession(options = {}) {
+    return this.withSession(new InMemorySessionService(), options);
   }
   /**
    * Build the agent and optionally create runner and session
@@ -10047,8 +10061,8 @@ var AgentBuilder = class _AgentBuilder {
         appName: this.sessionConfig.appName,
         agent,
         sessionService: this.sessionConfig.service,
-        memoryService: this.sessionConfig.memoryService,
-        artifactService: this.sessionConfig.artifactService
+        memoryService: this.memoryService,
+        artifactService: this.artifactService
       };
       const baseRunner = new Runner(runnerConfig);
       runner = this.createEnhancedRunner(baseRunner, session);

package/package.json

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