fix: await session state load before syncing tool cache

Author: Tarquinen

lib/hooks.ts

@@ -14,11 +14,12 @@ export function createChatMessageTransformHandler(
     logger: Logger,
     config: PluginConfig
 ) {
-    return async(
+    return async (
         input: {},
         output: { messages: WithParts[] }
     ) => {
-        checkSession(client, state, logger, output.messages);
+        await checkSession(client, state, logger, output.messages)
+
         if (state.isSubAgent) {
             return
         }

lib/state/state.ts

@@ -4,12 +4,12 @@ import { loadSessionState } from "./persistence"
 import { getLastUserMessage } from "../messages/utils"
 import { isSubAgentSession } from "../utils"
 
-export const checkSession = (
+export const checkSession = async (
     client: any,
     state: SessionState,
     logger: Logger,
     messages: WithParts[]
-) => {
+): Promise<void> => {
 
     const lastUserMessage = getLastUserMessage(messages)
     if (!lastUserMessage) {
@@ -20,14 +20,11 @@ export const checkSession = (
 
     if (state.sessionId === null || state.sessionId !== lastSessionId) {
         logger.info(`Session changed: ${state.sessionId} -> ${lastSessionId}`)
-        ensureSessionInitialized(
-            client,
-            state,
-            lastSessionId,
-            logger
-        ).catch((err) => {
+        try {
+            await ensureSessionInitialized(client, state, lastSessionId, logger)
+        } catch (err: any) {
             logger.error("Failed to initialize session state", { error: err.message })
-        } )
+        }
     }
 }