@agentforge-io/core 2.3.0 → 2.3.1

package/dist/services/agent.service.js

@@ -379,6 +379,11 @@ class AgentService {
         const filter = params.overrides?.extraToolsFilter;
         const fromConnectors = filter && resolvedExtras ? filter(resolvedExtras) : resolvedExtras;
         const extraTools = mergeExtraTools(params.overrides?.extraTools, fromConnectors);
+        // Hoisted accumulators so the post-loop persistence (after the
+        // try) can see the final list. Defined here, populated inside
+        // the for-await loop below.
+        const toolCallStartByUseId = new Map();
+        const accumulatedToolCalls = [];
         try {
             // Team orchestrators route through OrchestratorService.stream()
             // so the synthetic `delegate_to_*` tools the orchestrator was
@@ -403,11 +408,56 @@ class AgentService {
                     messageId: 'streaming',
                     agent: { timezone: agent.timezone },
                 }, { ...(params.overrides ?? {}), extraTools });
+            // Accumulate tool_use / tool_result chunks during streaming so
+            // we can persist them on the assistant message row (line ~700).
+            // Without this, `getHistory` returns the assistant's text but
+            // loses the tool calls — which means clients that render
+            // proposal cards (Recording Assist) or generic tool rows from
+            // history have nothing to rehydrate. FIFO matching mirrors the
+            // runner's heuristic so the shape stays consistent whether the
+            // turn streamed or used `run()`.
             for await (const chunk of stream) {
                 if (chunk.type === 'text_delta')
                     fullContent += chunk.delta;
                 if (chunk.type === 'usage')
                     finalUsage = chunk.usage;
+                if (chunk.type === 'tool_use_start') {
+                    toolCallStartByUseId.set(chunk.toolUseId, {
+                        name: chunk.toolName,
+                        start: Date.now(),
+                    });
+                }
+                if (chunk.type === 'tool_result') {
+                    // Match by toolUseId when present; otherwise FIFO on toolName
+                    // (matches AgentRunner.run's heuristic). Older runners that
+                    // don't emit `toolUseId` on result chunks still produce a
+                    // usable record.
+                    let entry = null;
+                    const useId = chunk.toolUseId;
+                    if (useId && toolCallStartByUseId.has(useId)) {
+                        const v = toolCallStartByUseId.get(useId);
+                        entry = { id: useId, name: v.name, start: v.start };
+                        toolCallStartByUseId.delete(useId);
+                    }
+                    else {
+                        const fifo = Array.from(toolCallStartByUseId.entries()).find(([, v]) => v.name === chunk.toolName);
+                        if (fifo) {
+                            entry = { id: fifo[0], name: fifo[1].name, start: fifo[1].start };
+                            toolCallStartByUseId.delete(fifo[0]);
+                        }
+                    }
+                    if (entry) {
+                        accumulatedToolCalls.push({
+                            toolName: entry.name,
+                            toolUseId: entry.id,
+                            input: {},
+                            output: typeof chunk.result === 'string'
+                                ? chunk.result
+                                : JSON.stringify(chunk.result),
+                            durationMs: Date.now() - entry.start,
+                        });
+                    }
+                }
                 yield chunk;
             }
         }
@@ -507,6 +557,12 @@ class AgentService {
                 role: 'assistant',
                 content: fullContent,
                 usage: finalUsage,
+                // Persist the accumulated tool calls so `getHistory` can
+                // surface them on reload — without this, proposal cards
+                // (Recording Assist) and other tool-result-driven UI
+                // disappear after refresh. Empty arrays drop to undefined
+                // so we don't pollute the column with `[]`.
+                toolCalls: accumulatedToolCalls.length > 0 ? accumulatedToolCalls : undefined,
             });
             const now = new Date();
             await this.dispatchUsage({

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@agentforge-io/core",
-  "version": "2.3.0",
+  "version": "2.3.1",
   "description": "Framework-free AI runtime SDK. Owns: agent loop (pluggable LLM provider — Anthropic by default, LangChain-backed providers as drop-ins), conversations, tools, streaming, agent-job queue, SdkHooks. Identity, billing, infra (email/uploads/secrets) live in the host's modules — not here.",
   "license": "MIT",
   "main": "dist/index.js",