From 2483161663e07fb2e83f8c1662eed8d981b14acc Mon Sep 17 00:00:00 2001
From: Rob Colbert <rob@digitaltelepresence.com>
Date: Mon, 18 May 2026 15:00:40 -0400
Subject: [PATCH] context window fuel gauge auto update fix

---
 gadget-code/frontend/src/lib/socket.ts | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/gadget-code/frontend/src/lib/socket.ts b/gadget-code/frontend/src/lib/socket.ts
index f1eb4f9..0158d9e 100644
--- a/gadget-code/frontend/src/lib/socket.ts
+++ b/gadget-code/frontend/src/lib/socket.ts
@@ -1,6 +1,6 @@
 import { createContext } from "react";
 import { io, Socket } from "socket.io-client";
-import type { ChatSession } from "./api";
+import type { ChatSession, IWorkOrderCompleteStats } from "./api";
 
 /**
  * Web Worker for heartbeat timing — avoids browser tab throttling.
@@ -30,6 +30,7 @@ export interface ServerToClientEvents {
     turnId: string,
     success: boolean,
     message?: string,
+    stats?: IWorkOrderCompleteStats,
   ) => void;
   log: (
     timestamp: string,
@@ -90,6 +91,7 @@ export interface SocketEvents {
     turnId: string,
     success: boolean,
     message?: string,
+    stats?: IWorkOrderCompleteStats,
   ) => void;
   "agent:thinking": (data: { agentId: string; thinking: string }) => void;
   "agent:response": (data: { agentId: string; chunk: string }) => void;
@@ -228,8 +230,8 @@ class SocketClient {
 
     this.socket.on(
       "workOrderComplete",
-      (turnId: string, success: boolean, message?: string) => {
-        this.emit("workOrderComplete", turnId, success, message);
+      (turnId: string, success: boolean, message?: string, stats?: IWorkOrderCompleteStats) => {
+        this.emit("workOrderComplete", turnId, success, message, stats);
       },
     );