@iqai/adk 0.0.12 → 0.0.14

package/CHANGELOG.md

@@ -1,5 +1,18 @@
 # @iqai/adk
 
+## 0.0.14
+
+### Patch Changes
+
+- 25c9c8c: bug-fixes: session-managers, runner
+  new example: pglite
+
+## 0.0.13
+
+### Patch Changes
+
+- 93b982b: Adds better sqlite3 dependency
+
 ## 0.0.12
 
 ### Patch Changes

package/dist/index.d.mts

@@ -3489,9 +3489,10 @@ type index_SqliteSessionService = SqliteSessionService;
 declare const index_SqliteSessionService: typeof SqliteSessionService;
 declare const index_cloneSession: typeof cloneSession;
 declare const index_generateSessionId: typeof generateSessionId;
+declare const index_sessionsSchema: typeof sessionsSchema;
 declare const index_validateSession: typeof validateSession;
 declare namespace index {
-  export { index_InMemorySessionService as InMemorySessionService, type index_ListSessionOptions as ListSessionOptions, index_PgLiteSessionService as PgLiteSessionService, index_PostgresSessionService as PostgresSessionService, type index_Session as Session, type index_SessionService as SessionService, index_SessionState as SessionState, index_SqliteSessionService as SqliteSessionService, index_cloneSession as cloneSession, index_generateSessionId as generateSessionId, index_validateSession as validateSession };
+  export { index_InMemorySessionService as InMemorySessionService, type index_ListSessionOptions as ListSessionOptions, index_PgLiteSessionService as PgLiteSessionService, index_PostgresSessionService as PostgresSessionService, type index_Session as Session, type index_SessionService as SessionService, index_SessionState as SessionState, index_SqliteSessionService as SqliteSessionService, index_cloneSession as cloneSession, index_generateSessionId as generateSessionId, index_sessionsSchema as sessionsSchema, index_validateSession as validateSession };
 }
 
 /**
@@ -3557,4 +3558,4 @@ declare class InMemoryRunner extends Runner {
 
 declare const VERSION = "0.1.0";
 
-export { Agent, type AgentConfig, index$3 as Agents, AnthropicLLM, type AnthropicLLMConfig, AnthropicLLMConnection, ApiKeyCredential, ApiKeyScheme, type AudioTranscriptionConfig, AuthConfig, AuthCredential, AuthCredentialType, AuthHandler, AuthScheme, AuthSchemeType, BaseAgent, BaseLLM, BaseLLMConnection, type BaseMemoryService, BaseTool, BasicAuthCredential, BearerTokenCredential, type BuildFunctionDeclarationOptions, ExitLoopTool, FileOperationsTool, type FunctionCall, type FunctionDeclaration, FunctionTool, GetUserChoiceTool, GoogleLLM, type GoogleLLMConfig, GoogleSearch, HttpRequestTool, HttpScheme, type IToolContext, type ImageContent, InMemoryMemoryService, InMemoryRunner, InMemorySessionService, InvocationContext, type JSONSchema, LLMRegistry, LLMRequest, type LLMRequestConfig, LLMResponse, LangGraphAgent, type LangGraphAgentConfig, type LangGraphNode, type ListSessionOptions, LoadMemoryTool, LoopAgent, type LoopAgentConfig, type McpConfig, McpError, McpErrorType, McpSamplingHandler, type McpSamplingRequest, type McpSamplingResponse, McpToolset, type McpTransportType, index$1 as Memory, type MemoryResult, type Message, type MessageContent, type MessageRole, index$4 as Models, OAuth2Credential, OAuth2Scheme, type OAuthFlow, type OAuthFlows, OpenAILLM, type OpenAILLMConfig, OpenAILLMConnection, OpenIdConnectScheme, ParallelAgent, type ParallelAgentConfig, PersistentMemoryService, PgLiteSessionService, PostgresSessionService, RunConfig, Runner, type SamplingHandler, type SearchMemoryOptions, type SearchMemoryResponse, SequentialAgent, type SequentialAgentConfig, type Session, type SessionService, SessionState, index as Sessions, type SpeechConfig, SqliteSessionService, StreamingMode, type TextContent, type ToolCall, type ToolConfig, ToolContext, index$2 as Tools, TransferToAgentTool, UserInteractionTool, VERSION, adkToMcpToolType, buildFunctionDeclaration, cloneSession, createFunctionTool, createSamplingHandler, generateSessionId, getMcpTools, jsonSchemaToDeclaration, mcpSchemaToParameters, normalizeJsonSchema, registerProviders, validateSession };
+export { Agent, type AgentConfig, index$3 as Agents, AnthropicLLM, type AnthropicLLMConfig, AnthropicLLMConnection, ApiKeyCredential, ApiKeyScheme, type AudioTranscriptionConfig, AuthConfig, AuthCredential, AuthCredentialType, AuthHandler, AuthScheme, AuthSchemeType, BaseAgent, BaseLLM, BaseLLMConnection, type BaseMemoryService, BaseTool, BasicAuthCredential, BearerTokenCredential, type BuildFunctionDeclarationOptions, ExitLoopTool, FileOperationsTool, type FunctionCall, type FunctionDeclaration, FunctionTool, GetUserChoiceTool, GoogleLLM, type GoogleLLMConfig, GoogleSearch, HttpRequestTool, HttpScheme, type IToolContext, type ImageContent, InMemoryMemoryService, InMemoryRunner, InMemorySessionService, InvocationContext, type JSONSchema, LLMRegistry, LLMRequest, type LLMRequestConfig, LLMResponse, LangGraphAgent, type LangGraphAgentConfig, type LangGraphNode, type ListSessionOptions, LoadMemoryTool, LoopAgent, type LoopAgentConfig, type McpConfig, McpError, McpErrorType, McpSamplingHandler, type McpSamplingRequest, type McpSamplingResponse, McpToolset, type McpTransportType, index$1 as Memory, type MemoryResult, type Message, type MessageContent, type MessageRole, index$4 as Models, OAuth2Credential, OAuth2Scheme, type OAuthFlow, type OAuthFlows, OpenAILLM, type OpenAILLMConfig, OpenAILLMConnection, OpenIdConnectScheme, ParallelAgent, type ParallelAgentConfig, PersistentMemoryService, PgLiteSessionService, PostgresSessionService, RunConfig, Runner, type SamplingHandler, type SearchMemoryOptions, type SearchMemoryResponse, SequentialAgent, type SequentialAgentConfig, type Session, type SessionService, SessionState, index as Sessions, type SpeechConfig, SqliteSessionService, StreamingMode, type TextContent, type ToolCall, type ToolConfig, ToolContext, index$2 as Tools, TransferToAgentTool, UserInteractionTool, VERSION, adkToMcpToolType, buildFunctionDeclaration, cloneSession, createFunctionTool, createSamplingHandler, generateSessionId, getMcpTools, jsonSchemaToDeclaration, mcpSchemaToParameters, normalizeJsonSchema, registerProviders, sessionsSchema, validateSession };

package/dist/index.d.ts

@@ -3489,9 +3489,10 @@ type index_SqliteSessionService = SqliteSessionService;
 declare const index_SqliteSessionService: typeof SqliteSessionService;
 declare const index_cloneSession: typeof cloneSession;
 declare const index_generateSessionId: typeof generateSessionId;
+declare const index_sessionsSchema: typeof sessionsSchema;
 declare const index_validateSession: typeof validateSession;
 declare namespace index {
-  export { index_InMemorySessionService as InMemorySessionService, type index_ListSessionOptions as ListSessionOptions, index_PgLiteSessionService as PgLiteSessionService, index_PostgresSessionService as PostgresSessionService, type index_Session as Session, type index_SessionService as SessionService, index_SessionState as SessionState, index_SqliteSessionService as SqliteSessionService, index_cloneSession as cloneSession, index_generateSessionId as generateSessionId, index_validateSession as validateSession };
+  export { index_InMemorySessionService as InMemorySessionService, type index_ListSessionOptions as ListSessionOptions, index_PgLiteSessionService as PgLiteSessionService, index_PostgresSessionService as PostgresSessionService, type index_Session as Session, type index_SessionService as SessionService, index_SessionState as SessionState, index_SqliteSessionService as SqliteSessionService, index_cloneSession as cloneSession, index_generateSessionId as generateSessionId, index_sessionsSchema as sessionsSchema, index_validateSession as validateSession };
 }
 
 /**
@@ -3557,4 +3558,4 @@ declare class InMemoryRunner extends Runner {
 
 declare const VERSION = "0.1.0";
 
-export { Agent, type AgentConfig, index$3 as Agents, AnthropicLLM, type AnthropicLLMConfig, AnthropicLLMConnection, ApiKeyCredential, ApiKeyScheme, type AudioTranscriptionConfig, AuthConfig, AuthCredential, AuthCredentialType, AuthHandler, AuthScheme, AuthSchemeType, BaseAgent, BaseLLM, BaseLLMConnection, type BaseMemoryService, BaseTool, BasicAuthCredential, BearerTokenCredential, type BuildFunctionDeclarationOptions, ExitLoopTool, FileOperationsTool, type FunctionCall, type FunctionDeclaration, FunctionTool, GetUserChoiceTool, GoogleLLM, type GoogleLLMConfig, GoogleSearch, HttpRequestTool, HttpScheme, type IToolContext, type ImageContent, InMemoryMemoryService, InMemoryRunner, InMemorySessionService, InvocationContext, type JSONSchema, LLMRegistry, LLMRequest, type LLMRequestConfig, LLMResponse, LangGraphAgent, type LangGraphAgentConfig, type LangGraphNode, type ListSessionOptions, LoadMemoryTool, LoopAgent, type LoopAgentConfig, type McpConfig, McpError, McpErrorType, McpSamplingHandler, type McpSamplingRequest, type McpSamplingResponse, McpToolset, type McpTransportType, index$1 as Memory, type MemoryResult, type Message, type MessageContent, type MessageRole, index$4 as Models, OAuth2Credential, OAuth2Scheme, type OAuthFlow, type OAuthFlows, OpenAILLM, type OpenAILLMConfig, OpenAILLMConnection, OpenIdConnectScheme, ParallelAgent, type ParallelAgentConfig, PersistentMemoryService, PgLiteSessionService, PostgresSessionService, RunConfig, Runner, type SamplingHandler, type SearchMemoryOptions, type SearchMemoryResponse, SequentialAgent, type SequentialAgentConfig, type Session, type SessionService, SessionState, index as Sessions, type SpeechConfig, SqliteSessionService, StreamingMode, type TextContent, type ToolCall, type ToolConfig, ToolContext, index$2 as Tools, TransferToAgentTool, UserInteractionTool, VERSION, adkToMcpToolType, buildFunctionDeclaration, cloneSession, createFunctionTool, createSamplingHandler, generateSessionId, getMcpTools, jsonSchemaToDeclaration, mcpSchemaToParameters, normalizeJsonSchema, registerProviders, validateSession };
+export { Agent, type AgentConfig, index$3 as Agents, AnthropicLLM, type AnthropicLLMConfig, AnthropicLLMConnection, ApiKeyCredential, ApiKeyScheme, type AudioTranscriptionConfig, AuthConfig, AuthCredential, AuthCredentialType, AuthHandler, AuthScheme, AuthSchemeType, BaseAgent, BaseLLM, BaseLLMConnection, type BaseMemoryService, BaseTool, BasicAuthCredential, BearerTokenCredential, type BuildFunctionDeclarationOptions, ExitLoopTool, FileOperationsTool, type FunctionCall, type FunctionDeclaration, FunctionTool, GetUserChoiceTool, GoogleLLM, type GoogleLLMConfig, GoogleSearch, HttpRequestTool, HttpScheme, type IToolContext, type ImageContent, InMemoryMemoryService, InMemoryRunner, InMemorySessionService, InvocationContext, type JSONSchema, LLMRegistry, LLMRequest, type LLMRequestConfig, LLMResponse, LangGraphAgent, type LangGraphAgentConfig, type LangGraphNode, type ListSessionOptions, LoadMemoryTool, LoopAgent, type LoopAgentConfig, type McpConfig, McpError, McpErrorType, McpSamplingHandler, type McpSamplingRequest, type McpSamplingResponse, McpToolset, type McpTransportType, index$1 as Memory, type MemoryResult, type Message, type MessageContent, type MessageRole, index$4 as Models, OAuth2Credential, OAuth2Scheme, type OAuthFlow, type OAuthFlows, OpenAILLM, type OpenAILLMConfig, OpenAILLMConnection, OpenIdConnectScheme, ParallelAgent, type ParallelAgentConfig, PersistentMemoryService, PgLiteSessionService, PostgresSessionService, RunConfig, Runner, type SamplingHandler, type SearchMemoryOptions, type SearchMemoryResponse, SequentialAgent, type SequentialAgentConfig, type Session, type SessionService, SessionState, index as Sessions, type SpeechConfig, SqliteSessionService, StreamingMode, type TextContent, type ToolCall, type ToolConfig, ToolContext, index$2 as Tools, TransferToAgentTool, UserInteractionTool, VERSION, adkToMcpToolType, buildFunctionDeclaration, cloneSession, createFunctionTool, createSamplingHandler, generateSessionId, getMcpTools, jsonSchemaToDeclaration, mcpSchemaToParameters, normalizeJsonSchema, registerProviders, sessionsSchema, validateSession };

package/dist/index.js

@@ -6245,6 +6245,7 @@ __export(sessions_exports, {
   SqliteSessionService: () => SqliteSessionService,
   cloneSession: () => cloneSession,
   generateSessionId: () => generateSessionId,
+  sessionsSchema: () => sessionsSchema,
   validateSession: () => validateSession
 });
 
@@ -6509,15 +6510,17 @@ var PostgresSessionService = class {
       session.events = [];
     }
     session.events.push(event);
+    if (event.content) {
+      session.messages = session.messages || [];
+      session.messages.push({
+        role: event.author,
+        content: event.content
+      });
+    }
     session.updatedAt = /* @__PURE__ */ new Date();
     await this.updateSession(session);
     return event;
   }
-  // TODO: Consider if table creation/migration logic is needed here or handled externally (e.g., drizzle-kit migrations)
-  // TODO: Implement methods corresponding to Python's append_event, list_events,
-  // get_app_state, update_app_state, get_user_state, update_user_state
-  // if full parity with Python's DatabaseSessionService is desired.
-  // This would require defining corresponding Drizzle schemas for Events, AppState, UserState.
 };
 
 // src/sessions/pglite-session-service.ts
@@ -7096,6 +7099,7 @@ var Runner = class {
       });
     }
     let lastPartialEvent = null;
+    let assistantContent = "";
     try {
       for await (const response of this.agent.runStreaming(invocationContext)) {
         const event = new Event({
@@ -7107,14 +7111,23 @@ var Runner = class {
           partial: response.is_partial,
           raw_response: response.raw_response
         });
-        if (!event.is_partial && event.content) {
-          await this.sessionService.appendEvent(session, event);
-        }
+        await this.sessionService.appendEvent(session, event);
         if (event.is_partial) {
           lastPartialEvent = event;
         }
+        if (response.role === "assistant" && response.content) {
+          assistantContent += response.content;
+        }
         yield event;
       }
+      if (assistantContent.trim()) {
+        session.messages = session.messages || [];
+        session.messages.push({
+          role: "assistant",
+          content: assistantContent
+        });
+        await this.sessionService.updateSession(session);
+      }
     } catch (error) {
       console.error("Error running agent:", error);
       if (lastPartialEvent && session.events && session.events.length === 0) {
@@ -7144,6 +7157,13 @@ var Runner = class {
       author: "user",
       content: typeof newMessage.content === "string" ? newMessage.content : null
     });
+    if (event.author === "user" || event.author === "assistant") {
+      session.messages = session.messages || [];
+      session.messages.push({
+        role: event.author,
+        content: event.content
+      });
+    }
     await this.sessionService.appendEvent(session, event);
   }
   /**
@@ -7156,7 +7176,10 @@ var Runner = class {
   }) {
     return new InvocationContext({
       sessionId: session.id,
-      messages: session.messages || [],
+      messages: [
+        ...session.messages || [],
+        ...newMessage ? [newMessage] : []
+      ],
       config: runConfig,
       userId: session.userId,
       appName: this.appName,
@@ -7259,4 +7282,5 @@ var VERSION = "0.1.0";
 
 
 
-exports.Agent = Agent; exports.Agents = agents_exports; exports.AnthropicLLM = AnthropicLLM; exports.AnthropicLLMConnection = AnthropicLLMConnection; 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.BaseAgent = BaseAgent; exports.BaseLLM = BaseLLM; exports.BaseLLMConnection = BaseLLMConnection; exports.BaseTool = BaseTool; exports.BasicAuthCredential = BasicAuthCredential; exports.BearerTokenCredential = BearerTokenCredential; exports.ExitLoopTool = ExitLoopTool; exports.FileOperationsTool = FileOperationsTool; exports.FunctionTool = FunctionTool; exports.GetUserChoiceTool = GetUserChoiceTool; exports.GoogleLLM = GoogleLLM; exports.GoogleSearch = GoogleSearch; exports.HttpRequestTool = HttpRequestTool; exports.HttpScheme = HttpScheme; exports.InMemoryMemoryService = InMemoryMemoryService; exports.InMemoryRunner = InMemoryRunner; exports.InMemorySessionService = InMemorySessionService; exports.InvocationContext = InvocationContext; exports.LLMRegistry = LLMRegistry; exports.LLMRequest = LLMRequest; exports.LLMResponse = LLMResponse; exports.LangGraphAgent = LangGraphAgent; exports.LoadMemoryTool = LoadMemoryTool; exports.LoopAgent = LoopAgent; exports.McpError = McpError; exports.McpErrorType = McpErrorType; exports.McpSamplingHandler = McpSamplingHandler; exports.McpToolset = McpToolset; exports.Memory = memory_exports; exports.Models = models_exports; exports.OAuth2Credential = OAuth2Credential; exports.OAuth2Scheme = OAuth2Scheme; exports.OpenAILLM = OpenAILLM; exports.OpenAILLMConnection = OpenAILLMConnection; exports.OpenIdConnectScheme = OpenIdConnectScheme; exports.ParallelAgent = ParallelAgent; exports.PersistentMemoryService = PersistentMemoryService; exports.PgLiteSessionService = PgLiteSessionService; exports.PostgresSessionService = PostgresSessionService; exports.RunConfig = RunConfig; exports.Runner = Runner; exports.SequentialAgent = SequentialAgent; exports.SessionState = SessionState; exports.Sessions = sessions_exports; exports.SqliteSessionService = SqliteSessionService; exports.StreamingMode = StreamingMode; exports.ToolContext = ToolContext; exports.Tools = tools_exports; exports.TransferToAgentTool = TransferToAgentTool; exports.UserInteractionTool = UserInteractionTool; exports.VERSION = VERSION; exports.adkToMcpToolType = adkToMcpToolType; exports.buildFunctionDeclaration = buildFunctionDeclaration; exports.cloneSession = cloneSession; exports.createFunctionTool = createFunctionTool; exports.createSamplingHandler = createSamplingHandler; exports.generateSessionId = generateSessionId; exports.getMcpTools = getMcpTools; exports.jsonSchemaToDeclaration = jsonSchemaToDeclaration; exports.mcpSchemaToParameters = mcpSchemaToParameters; exports.normalizeJsonSchema = normalizeJsonSchema; exports.registerProviders = registerProviders; exports.validateSession = validateSession;
+
+exports.Agent = Agent; exports.Agents = agents_exports; exports.AnthropicLLM = AnthropicLLM; exports.AnthropicLLMConnection = AnthropicLLMConnection; 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.BaseAgent = BaseAgent; exports.BaseLLM = BaseLLM; exports.BaseLLMConnection = BaseLLMConnection; exports.BaseTool = BaseTool; exports.BasicAuthCredential = BasicAuthCredential; exports.BearerTokenCredential = BearerTokenCredential; exports.ExitLoopTool = ExitLoopTool; exports.FileOperationsTool = FileOperationsTool; exports.FunctionTool = FunctionTool; exports.GetUserChoiceTool = GetUserChoiceTool; exports.GoogleLLM = GoogleLLM; exports.GoogleSearch = GoogleSearch; exports.HttpRequestTool = HttpRequestTool; exports.HttpScheme = HttpScheme; exports.InMemoryMemoryService = InMemoryMemoryService; exports.InMemoryRunner = InMemoryRunner; exports.InMemorySessionService = InMemorySessionService; exports.InvocationContext = InvocationContext; exports.LLMRegistry = LLMRegistry; exports.LLMRequest = LLMRequest; exports.LLMResponse = LLMResponse; exports.LangGraphAgent = LangGraphAgent; exports.LoadMemoryTool = LoadMemoryTool; exports.LoopAgent = LoopAgent; exports.McpError = McpError; exports.McpErrorType = McpErrorType; exports.McpSamplingHandler = McpSamplingHandler; exports.McpToolset = McpToolset; exports.Memory = memory_exports; exports.Models = models_exports; exports.OAuth2Credential = OAuth2Credential; exports.OAuth2Scheme = OAuth2Scheme; exports.OpenAILLM = OpenAILLM; exports.OpenAILLMConnection = OpenAILLMConnection; exports.OpenIdConnectScheme = OpenIdConnectScheme; exports.ParallelAgent = ParallelAgent; exports.PersistentMemoryService = PersistentMemoryService; exports.PgLiteSessionService = PgLiteSessionService; exports.PostgresSessionService = PostgresSessionService; exports.RunConfig = RunConfig; exports.Runner = Runner; exports.SequentialAgent = SequentialAgent; exports.SessionState = SessionState; exports.Sessions = sessions_exports; exports.SqliteSessionService = SqliteSessionService; exports.StreamingMode = StreamingMode; exports.ToolContext = ToolContext; exports.Tools = tools_exports; exports.TransferToAgentTool = TransferToAgentTool; exports.UserInteractionTool = UserInteractionTool; exports.VERSION = VERSION; exports.adkToMcpToolType = adkToMcpToolType; exports.buildFunctionDeclaration = buildFunctionDeclaration; exports.cloneSession = cloneSession; exports.createFunctionTool = createFunctionTool; exports.createSamplingHandler = createSamplingHandler; exports.generateSessionId = generateSessionId; exports.getMcpTools = getMcpTools; exports.jsonSchemaToDeclaration = jsonSchemaToDeclaration; exports.mcpSchemaToParameters = mcpSchemaToParameters; exports.normalizeJsonSchema = normalizeJsonSchema; exports.registerProviders = registerProviders; exports.sessionsSchema = sessionsSchema; exports.validateSession = validateSession;

package/dist/index.mjs

@@ -6245,6 +6245,7 @@ __export(sessions_exports, {
   SqliteSessionService: () => SqliteSessionService,
   cloneSession: () => cloneSession,
   generateSessionId: () => generateSessionId,
+  sessionsSchema: () => sessionsSchema,
   validateSession: () => validateSession
 });
 
@@ -6509,15 +6510,17 @@ var PostgresSessionService = class {
       session.events = [];
     }
     session.events.push(event);
+    if (event.content) {
+      session.messages = session.messages || [];
+      session.messages.push({
+        role: event.author,
+        content: event.content
+      });
+    }
     session.updatedAt = /* @__PURE__ */ new Date();
     await this.updateSession(session);
     return event;
   }
-  // TODO: Consider if table creation/migration logic is needed here or handled externally (e.g., drizzle-kit migrations)
-  // TODO: Implement methods corresponding to Python's append_event, list_events,
-  // get_app_state, update_app_state, get_user_state, update_user_state
-  // if full parity with Python's DatabaseSessionService is desired.
-  // This would require defining corresponding Drizzle schemas for Events, AppState, UserState.
 };
 
 // src/sessions/pglite-session-service.ts
@@ -7096,6 +7099,7 @@ var Runner = class {
       });
     }
     let lastPartialEvent = null;
+    let assistantContent = "";
     try {
       for await (const response of this.agent.runStreaming(invocationContext)) {
         const event = new Event({
@@ -7107,14 +7111,23 @@ var Runner = class {
           partial: response.is_partial,
           raw_response: response.raw_response
         });
-        if (!event.is_partial && event.content) {
-          await this.sessionService.appendEvent(session, event);
-        }
+        await this.sessionService.appendEvent(session, event);
         if (event.is_partial) {
           lastPartialEvent = event;
         }
+        if (response.role === "assistant" && response.content) {
+          assistantContent += response.content;
+        }
         yield event;
       }
+      if (assistantContent.trim()) {
+        session.messages = session.messages || [];
+        session.messages.push({
+          role: "assistant",
+          content: assistantContent
+        });
+        await this.sessionService.updateSession(session);
+      }
     } catch (error) {
       console.error("Error running agent:", error);
       if (lastPartialEvent && session.events && session.events.length === 0) {
@@ -7144,6 +7157,13 @@ var Runner = class {
       author: "user",
       content: typeof newMessage.content === "string" ? newMessage.content : null
     });
+    if (event.author === "user" || event.author === "assistant") {
+      session.messages = session.messages || [];
+      session.messages.push({
+        role: event.author,
+        content: event.content
+      });
+    }
     await this.sessionService.appendEvent(session, event);
   }
   /**
@@ -7156,7 +7176,10 @@ var Runner = class {
   }) {
     return new InvocationContext({
       sessionId: session.id,
-      messages: session.messages || [],
+      messages: [
+        ...session.messages || [],
+        ...newMessage ? [newMessage] : []
+      ],
       config: runConfig,
       userId: session.userId,
       appName: this.appName,
@@ -7258,5 +7281,6 @@ export {
   mcpSchemaToParameters,
   normalizeJsonSchema,
   registerProviders,
+  sessionsSchema,
   validateSession
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@iqai/adk",
-  "version": "0.0.12",
+  "version": "0.0.14",
   "description": "Agent Development Kit for TypeScript with multi-provider LLM support",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -25,8 +25,8 @@
     "@google-cloud/vertexai": "^0.5.0",
     "@google/genai": "^1.0.0",
     "@modelcontextprotocol/sdk": "^1.11.1",
-    "@types/better-sqlite3": "^7.6.13",
     "axios": "^1.6.2",
+    "better-sqlite3": "^11.10.0",
     "dotenv": "^16.4.7",
     "drizzle-orm": "^0.43.1",
     "openai": "^4.93.0",
@@ -34,6 +34,7 @@
     "zod": "^3.24.4"
   },
   "devDependencies": {
+    "@types/better-sqlite3": "^7.6.13",
     "@types/node": "^20.17.30",
     "tsup": "^8.4.0",
     "typescript": "^5.3.2",