@minded-ai/mindedjs 1.0.141 → 1.0.142-beta-2

package/src/internalTools/timer.ts

@@ -1,17 +1,17 @@
 import * as mindedConnection from '../platform/mindedConnection';
 import { mindedConnectionSocketMessageType, TimerResetResponse, TimerCancelResponse } from '../platform/mindedConnectionTypes';
 
-// Store timer handlers by timer name
+// Store timer handlers by timer name (one handler per timer name)
 export const timerHandlers = new Map<
   string,
-  Array<{
+  {
     handler: (params: { sessionId: string; payload: Record<string, any> }) => void | Promise<void>;
-  }>
+  }
 >();
 
 /**
  * Reset or set a timer for a specific session.
- * When the timer expires, all registered timer handlers will be called.
+ * When the timer expires, the registered timer handler will be called.
  *
  * @param params - The timer configuration
  * @param params.sessionId - The session ID to associate the timer with
@@ -85,6 +85,9 @@ export async function cancelTimer(sessionId: string, timerName: string): Promise
  * Register a handler for timer trigger events for a specific timer name.
  * The handler will be called when the specified timer expires for any session.
  *
+ * Note: Only one handler per timer name is supported. Calling this function multiple times
+ * with the same timer name will replace the previous handler.
+ *
  * @param params - The configuration for the timer handler
  * @param params.timerName - The name of the timer to handle
  * @param params.handler - Function to call when the timer triggers
@@ -116,23 +119,17 @@ export function onTimer({
   timerName: string;
   handler: (params: { sessionId: string; payload: Record<string, any> }) => void | Promise<void>;
 }): void {
-  // Initialize handlers array for this timer name if it doesn't exist
-  if (!timerHandlers.has(timerName)) {
-    timerHandlers.set(timerName, []);
-  }
-
-  // Add the handler to the array for this timer name
-  const handlers = timerHandlers.get(timerName)!;
-  handlers.push({ handler });
+  // Set the handler for this timer name (replaces any existing handler)
+  timerHandlers.set(timerName, { handler });
 }
 
 mindedConnection.on(mindedConnectionSocketMessageType.TIMER_TRIGGER, async (timerTriggerMessage) => {
-  // Get handlers for the specific timer name
-  const handlers = timerHandlers.get(timerTriggerMessage.timerName) || [];
+  // Get handler for the specific timer name
+  const handlerEntry = timerHandlers.get(timerTriggerMessage.timerName);
 
-  // Call all handlers registered for this timer name
-  for (const { handler } of handlers) {
-    await handler({
+  // Call the handler if it exists
+  if (handlerEntry) {
+    await handlerEntry.handler({
       sessionId: timerTriggerMessage.sessionId,
       payload: timerTriggerMessage.eventArgs,
     });

package/src/nodes/addPromptNode.ts

@@ -141,12 +141,16 @@ export const addPromptNode = async ({ graph, node, llm, tools, emit, agent }: Ad
       };
     }
 
+    // Model text response
     state.goto = null;
-    state.messages.push(result);
-    let stateUpdate: any = { goto: null, messages: state.messages };
+    //update state so AI_MESSAGE will get the latest state
+    const stateForAIMessage = structuredClone(state);
+    stateForAIMessage.goto = null;
+    stateForAIMessage.messages.push(result);
+    let stateUpdate: any = { goto: null, messages: [result] };
     if (result.getType() === 'ai') {
       logger.info({ msg: `[Model] Response`, content: result.content });
-      const results = await emit(AgentEvents.AI_MESSAGE, { message: result.content as string, state });
+      const results = await emit(AgentEvents.AI_MESSAGE, { message: result.content as string, state: stateForAIMessage });
       const handlerResult = results.find((r) => r !== undefined);
       if (handlerResult && handlerResult.state) {
         stateUpdate = { ...stateUpdate, ...handlerResult.state };