@antipopp/agno-client 0.6.1 → 0.8.0

package/dist/index.d.mts

@@ -13,6 +13,7 @@ declare class AgnoClient extends EventEmitter {
     private eventProcessor;
     private state;
     private pendingUISpecs;
+    private runCompletedSuccessfully;
     constructor(config: AgnoClientConfig);
     /**
      * Get current messages
@@ -48,6 +49,13 @@ declare class AgnoClient extends EventEmitter {
      * Handle error
      */
     private handleError;
+    /**
+     * Refresh messages from the session API after run completion.
+     * Replaces streamed messages with authoritative session data.
+     * Preserves client-side properties like ui_component that aren't stored on the server.
+     * @private
+     */
+    private refreshSessionMessages;
     /**
      * Load a session
      */

package/dist/index.d.ts

@@ -13,6 +13,7 @@ declare class AgnoClient extends EventEmitter {
     private eventProcessor;
     private state;
     private pendingUISpecs;
+    private runCompletedSuccessfully;
     constructor(config: AgnoClientConfig);
     /**
      * Get current messages
@@ -48,6 +49,13 @@ declare class AgnoClient extends EventEmitter {
      * Handle error
      */
     private handleError;
+    /**
+     * Refresh messages from the session API after run completion.
+     * Replaces streamed messages with authoritative session data.
+     * Preserves client-side properties like ui_component that aren't stored on the server.
+     * @private
+     */
+    private refreshSessionMessages;
     /**
      * Load a session
      */

package/dist/index.js

@@ -238,6 +238,18 @@ var ConfigManager = class {
   setUserId(userId) {
     this.updateField("userId", userId);
   }
+  /**
+   * Get custom headers
+   */
+  getHeaders() {
+    return this.config.headers;
+  }
+  /**
+   * Set custom headers
+   */
+  setHeaders(headers) {
+    this.updateField("headers", headers);
+  }
   /**
    * Get current entity ID (agent or team based on mode)
    */
@@ -260,6 +272,31 @@ var ConfigManager = class {
       return `${endpoint}/agents/${encodedEntityId}/runs`;
     }
   }
+  /**
+   * Build request headers by merging global headers, per-request headers, and auth token.
+   * Merge order (lowest to highest precedence):
+   * 1. Global headers from config
+   * 2. Per-request headers (overrides global)
+   * 3. Authorization header from authToken (overrides all)
+   *
+   * @param perRequestHeaders - Optional headers for this specific request
+   * @returns Merged headers object ready for fetch
+   */
+  buildRequestHeaders(perRequestHeaders) {
+    const headers = {};
+    const globalHeaders = this.getHeaders();
+    if (globalHeaders) {
+      Object.assign(headers, globalHeaders);
+    }
+    if (perRequestHeaders) {
+      Object.assign(headers, perRequestHeaders);
+    }
+    const authToken = this.getAuthToken();
+    if (authToken) {
+      headers["Authorization"] = `Bearer ${authToken}`;
+    }
+    return headers;
+  }
 };
 
 // src/managers/session-manager.ts
@@ -267,15 +304,11 @@ var SessionManager = class {
   /**
    * Fetch all sessions for an entity
    */
-  async fetchSessions(endpoint, entityType, entityId, dbId, authToken) {
+  async fetchSessions(endpoint, entityType, entityId, dbId, headers) {
     const url = new URL(`${endpoint}/sessions`);
     url.searchParams.set("type", entityType);
     url.searchParams.set("component_id", entityId);
     url.searchParams.set("db_id", dbId);
-    const headers = {};
-    if (authToken) {
-      headers["Authorization"] = `Bearer ${authToken}`;
-    }
     const response = await fetch(url.toString(), { headers });
     if (!response.ok) {
       if (response.status === 404) {
@@ -290,15 +323,14 @@ var SessionManager = class {
    * Fetch a specific session's runs
    * Returns an array of RunSchema directly (not wrapped in { data, meta })
    */
-  async fetchSession(endpoint, entityType, sessionId, dbId, authToken) {
+  async fetchSession(endpoint, entityType, sessionId, dbId, headers, userId) {
     const url = new URL(`${endpoint}/sessions/${sessionId}/runs`);
     url.searchParams.set("type", entityType);
     if (dbId) {
       url.searchParams.set("db_id", dbId);
     }
-    const headers = {};
-    if (authToken) {
-      headers["Authorization"] = `Bearer ${authToken}`;
+    if (userId) {
+      url.searchParams.set("user_id", userId);
     }
     const response = await fetch(url.toString(), { headers });
     if (!response.ok) {
@@ -309,15 +341,11 @@ var SessionManager = class {
   /**
    * Delete a session
    */
-  async deleteSession(endpoint, sessionId, dbId, authToken) {
+  async deleteSession(endpoint, sessionId, dbId, headers) {
     const url = new URL(`${endpoint}/sessions/${sessionId}`);
     if (dbId) {
       url.searchParams.set("db_id", dbId);
     }
-    const headers = {};
-    if (authToken) {
-      headers["Authorization"] = `Bearer ${authToken}`;
-    }
     const response = await fetch(url.toString(), {
       method: "DELETE",
       headers
@@ -826,9 +854,10 @@ function toSafeISOString(timestamp) {
   return new Date(ts).toISOString();
 }
 var AgnoClient = class extends import_eventemitter3.default {
-  // toolCallId -> UIComponentSpec
   constructor(config) {
     super();
+    // toolCallId -> UIComponentSpec
+    this.runCompletedSuccessfully = false;
     this.messageStore = new MessageStore();
     this.configManager = new ConfigManager(config);
     this.sessionManager = new SessionManager();
@@ -836,6 +865,7 @@ var AgnoClient = class extends import_eventemitter3.default {
     this.pendingUISpecs = /* @__PURE__ */ new Map();
     this.state = {
       isStreaming: false,
+      isRefreshing: false,
       isEndpointActive: false,
       agents: [],
       teams: [],
@@ -887,6 +917,7 @@ var AgnoClient = class extends import_eventemitter3.default {
     if (this.state.isStreaming) {
       throw new Error("Already streaming a message");
     }
+    this.runCompletedSuccessfully = false;
     const runUrl = this.configManager.getRunUrl();
     if (!runUrl) {
       throw new Error("No agent or team selected");
@@ -928,11 +959,7 @@ var AgnoClient = class extends import_eventemitter3.default {
       if (userId) {
         formData.append("user_id", userId);
       }
-      const headers = { ...options?.headers };
-      const authToken = this.configManager.getAuthToken();
-      if (authToken) {
-        headers["Authorization"] = `Bearer ${authToken}`;
-      }
+      const headers = this.configManager.buildRequestHeaders(options?.headers);
       await streamResponse({
         apiUrl: runUrl,
         headers,
@@ -949,11 +976,15 @@ var AgnoClient = class extends import_eventemitter3.default {
         onError: (error) => {
           this.handleError(error, newSessionId);
         },
-        onComplete: () => {
+        onComplete: async () => {
           this.state.isStreaming = false;
           this.emit("stream:end");
           this.emit("message:complete", this.messageStore.getMessages());
           this.emit("state:change", this.getState());
+          if (this.runCompletedSuccessfully) {
+            this.runCompletedSuccessfully = false;
+            await this.refreshSessionMessages();
+          }
         }
       });
     } catch (error) {
@@ -1017,6 +1048,9 @@ var AgnoClient = class extends import_eventemitter3.default {
       return updated || lastMessage;
     });
     this.applyPendingUISpecs();
+    if (event === import_agno_types2.RunEvent.RunCompleted || event === import_agno_types2.RunEvent.TeamRunCompleted) {
+      this.runCompletedSuccessfully = true;
+    }
     this.emit("message:update", this.messageStore.getMessages());
   }
   /**
@@ -1038,6 +1072,70 @@ var AgnoClient = class extends import_eventemitter3.default {
     this.emit("stream:end");
     this.emit("state:change", this.getState());
   }
+  /**
+   * Refresh messages from the session API after run completion.
+   * Replaces streamed messages with authoritative session data.
+   * Preserves client-side properties like ui_component that aren't stored on the server.
+   * @private
+   */
+  async refreshSessionMessages() {
+    const sessionId = this.configManager.getSessionId();
+    if (!sessionId) {
+      Logger.debug("[AgnoClient] Cannot refresh: no session ID");
+      return;
+    }
+    this.state.isRefreshing = true;
+    this.emit("state:change", this.getState());
+    try {
+      const existingUIComponents = /* @__PURE__ */ new Map();
+      for (const message of this.messageStore.getMessages()) {
+        if (message.tool_calls) {
+          for (const toolCall of message.tool_calls) {
+            if (toolCall.ui_component) {
+              existingUIComponents.set(toolCall.tool_call_id, toolCall.ui_component);
+            }
+          }
+        }
+      }
+      const config = this.configManager.getConfig();
+      const entityType = this.configManager.getMode();
+      const dbId = this.configManager.getDbId() || "";
+      const userId = this.configManager.getUserId();
+      const headers = this.configManager.buildRequestHeaders();
+      const response = await this.sessionManager.fetchSession(
+        config.endpoint,
+        entityType,
+        sessionId,
+        dbId,
+        headers,
+        userId
+      );
+      const messages = this.sessionManager.convertSessionToMessages(response);
+      if (existingUIComponents.size > 0) {
+        for (const message of messages) {
+          if (message.tool_calls) {
+            for (let i = 0; i < message.tool_calls.length; i++) {
+              const toolCall = message.tool_calls[i];
+              const uiComponent = existingUIComponents.get(toolCall.tool_call_id);
+              if (uiComponent) {
+                message.tool_calls[i].ui_component = uiComponent;
+              }
+            }
+          }
+        }
+      }
+      this.messageStore.setMessages(messages);
+      Logger.debug("[AgnoClient] Session refreshed:", `${messages.length} messages`);
+      this.emit("message:refreshed", messages);
+      this.emit("message:update", messages);
+    } catch (error) {
+      Logger.error("[AgnoClient] Failed to refresh session:", error);
+      this.emit("message:error", `Session refresh failed: ${error instanceof Error ? error.message : String(error)}`);
+    } finally {
+      this.state.isRefreshing = false;
+      this.emit("state:change", this.getState());
+    }
+  }
   /**
    * Load a session
    */
@@ -1046,13 +1144,16 @@ var AgnoClient = class extends import_eventemitter3.default {
     const config = this.configManager.getConfig();
     const entityType = this.configManager.getMode();
     const dbId = this.configManager.getDbId() || "";
-    Logger.debug("[AgnoClient] Loading session with:", { entityType, dbId });
+    const userId = this.configManager.getUserId();
+    Logger.debug("[AgnoClient] Loading session with:", { entityType, dbId, userId });
+    const headers = this.configManager.buildRequestHeaders();
     const response = await this.sessionManager.fetchSession(
       config.endpoint,
       entityType,
       sessionId,
       dbId,
-      config.authToken
+      headers,
+      userId
     );
     const messages = this.sessionManager.convertSessionToMessages(response);
     Logger.debug("[AgnoClient] Setting messages to store:", `${messages.length} messages`);
@@ -1076,12 +1177,13 @@ var AgnoClient = class extends import_eventemitter3.default {
     if (!entityId) {
       throw new Error("Entity ID must be configured");
     }
+    const headers = this.configManager.buildRequestHeaders();
     const sessions = await this.sessionManager.fetchSessions(
       config.endpoint,
       entityType,
       entityId,
       dbId,
-      config.authToken
+      headers
     );
     this.state.sessions = sessions;
     this.emit("state:change", this.getState());
@@ -1093,11 +1195,12 @@ var AgnoClient = class extends import_eventemitter3.default {
   async deleteSession(sessionId) {
     const config = this.configManager.getConfig();
     const dbId = this.configManager.getDbId() || "";
+    const headers = this.configManager.buildRequestHeaders();
     await this.sessionManager.deleteSession(
       config.endpoint,
       sessionId,
       dbId,
-      config.authToken
+      headers
     );
     this.state.sessions = this.state.sessions.filter(
       (s) => s.session_id !== sessionId
@@ -1248,11 +1351,7 @@ var AgnoClient = class extends import_eventemitter3.default {
     if (userId) {
       formData.append("user_id", userId);
     }
-    const headers = { ...options?.headers };
-    const authToken = this.configManager.getAuthToken();
-    if (authToken) {
-      headers["Authorization"] = `Bearer ${authToken}`;
-    }
+    const headers = this.configManager.buildRequestHeaders(options?.headers);
     try {
       await streamResponse({
         apiUrl: continueUrl,
@@ -1264,13 +1363,17 @@ var AgnoClient = class extends import_eventemitter3.default {
         onError: (error) => {
           this.handleError(error, currentSessionId);
         },
-        onComplete: () => {
+        onComplete: async () => {
           this.state.isStreaming = false;
           this.state.pausedRunId = void 0;
           this.state.toolsAwaitingExecution = void 0;
           this.emit("stream:end");
           this.emit("message:complete", this.messageStore.getMessages());
           this.emit("state:change", this.getState());
+          if (this.runCompletedSuccessfully) {
+            this.runCompletedSuccessfully = false;
+            await this.refreshSessionMessages();
+          }
         }
       });
     } catch (error) {
@@ -1285,7 +1388,8 @@ var AgnoClient = class extends import_eventemitter3.default {
    */
   async checkStatus() {
     try {
-      const response = await fetch(`${this.configManager.getEndpoint()}/health`);
+      const headers = this.configManager.buildRequestHeaders();
+      const response = await fetch(`${this.configManager.getEndpoint()}/health`, { headers });
       const isActive = response.ok;
       this.state.isEndpointActive = isActive;
       this.emit("state:change", this.getState());
@@ -1300,12 +1404,8 @@ var AgnoClient = class extends import_eventemitter3.default {
    * Fetch agents from endpoint
    */
   async fetchAgents() {
-    const config = this.configManager.getConfig();
-    const headers = {};
-    if (config.authToken) {
-      headers["Authorization"] = `Bearer ${config.authToken}`;
-    }
-    const response = await fetch(`${config.endpoint}/agents`, { headers });
+    const headers = this.configManager.buildRequestHeaders();
+    const response = await fetch(`${this.configManager.getEndpoint()}/agents`, { headers });
     if (!response.ok) {
       throw new Error("Failed to fetch agents");
     }
@@ -1318,12 +1418,8 @@ var AgnoClient = class extends import_eventemitter3.default {
    * Fetch teams from endpoint
    */
   async fetchTeams() {
-    const config = this.configManager.getConfig();
-    const headers = {};
-    if (config.authToken) {
-      headers["Authorization"] = `Bearer ${config.authToken}`;
-    }
-    const response = await fetch(`${config.endpoint}/teams`, { headers });
+    const headers = this.configManager.buildRequestHeaders();
+    const response = await fetch(`${this.configManager.getEndpoint()}/teams`, { headers });
     if (!response.ok) {
       throw new Error("Failed to fetch teams");
     }

package/dist/index.mjs

@@ -200,6 +200,18 @@ var ConfigManager = class {
   setUserId(userId) {
     this.updateField("userId", userId);
   }
+  /**
+   * Get custom headers
+   */
+  getHeaders() {
+    return this.config.headers;
+  }
+  /**
+   * Set custom headers
+   */
+  setHeaders(headers) {
+    this.updateField("headers", headers);
+  }
   /**
    * Get current entity ID (agent or team based on mode)
    */
@@ -222,6 +234,31 @@ var ConfigManager = class {
       return `${endpoint}/agents/${encodedEntityId}/runs`;
     }
   }
+  /**
+   * Build request headers by merging global headers, per-request headers, and auth token.
+   * Merge order (lowest to highest precedence):
+   * 1. Global headers from config
+   * 2. Per-request headers (overrides global)
+   * 3. Authorization header from authToken (overrides all)
+   *
+   * @param perRequestHeaders - Optional headers for this specific request
+   * @returns Merged headers object ready for fetch
+   */
+  buildRequestHeaders(perRequestHeaders) {
+    const headers = {};
+    const globalHeaders = this.getHeaders();
+    if (globalHeaders) {
+      Object.assign(headers, globalHeaders);
+    }
+    if (perRequestHeaders) {
+      Object.assign(headers, perRequestHeaders);
+    }
+    const authToken = this.getAuthToken();
+    if (authToken) {
+      headers["Authorization"] = `Bearer ${authToken}`;
+    }
+    return headers;
+  }
 };
 
 // src/managers/session-manager.ts
@@ -229,15 +266,11 @@ var SessionManager = class {
   /**
    * Fetch all sessions for an entity
    */
-  async fetchSessions(endpoint, entityType, entityId, dbId, authToken) {
+  async fetchSessions(endpoint, entityType, entityId, dbId, headers) {
     const url = new URL(`${endpoint}/sessions`);
     url.searchParams.set("type", entityType);
     url.searchParams.set("component_id", entityId);
     url.searchParams.set("db_id", dbId);
-    const headers = {};
-    if (authToken) {
-      headers["Authorization"] = `Bearer ${authToken}`;
-    }
     const response = await fetch(url.toString(), { headers });
     if (!response.ok) {
       if (response.status === 404) {
@@ -252,15 +285,14 @@ var SessionManager = class {
    * Fetch a specific session's runs
    * Returns an array of RunSchema directly (not wrapped in { data, meta })
    */
-  async fetchSession(endpoint, entityType, sessionId, dbId, authToken) {
+  async fetchSession(endpoint, entityType, sessionId, dbId, headers, userId) {
     const url = new URL(`${endpoint}/sessions/${sessionId}/runs`);
     url.searchParams.set("type", entityType);
     if (dbId) {
       url.searchParams.set("db_id", dbId);
     }
-    const headers = {};
-    if (authToken) {
-      headers["Authorization"] = `Bearer ${authToken}`;
+    if (userId) {
+      url.searchParams.set("user_id", userId);
     }
     const response = await fetch(url.toString(), { headers });
     if (!response.ok) {
@@ -271,15 +303,11 @@ var SessionManager = class {
   /**
    * Delete a session
    */
-  async deleteSession(endpoint, sessionId, dbId, authToken) {
+  async deleteSession(endpoint, sessionId, dbId, headers) {
     const url = new URL(`${endpoint}/sessions/${sessionId}`);
     if (dbId) {
       url.searchParams.set("db_id", dbId);
     }
-    const headers = {};
-    if (authToken) {
-      headers["Authorization"] = `Bearer ${authToken}`;
-    }
     const response = await fetch(url.toString(), {
       method: "DELETE",
       headers
@@ -788,9 +816,10 @@ function toSafeISOString(timestamp) {
   return new Date(ts).toISOString();
 }
 var AgnoClient = class extends EventEmitter {
-  // toolCallId -> UIComponentSpec
   constructor(config) {
     super();
+    // toolCallId -> UIComponentSpec
+    this.runCompletedSuccessfully = false;
     this.messageStore = new MessageStore();
     this.configManager = new ConfigManager(config);
     this.sessionManager = new SessionManager();
@@ -798,6 +827,7 @@ var AgnoClient = class extends EventEmitter {
     this.pendingUISpecs = /* @__PURE__ */ new Map();
     this.state = {
       isStreaming: false,
+      isRefreshing: false,
       isEndpointActive: false,
       agents: [],
       teams: [],
@@ -849,6 +879,7 @@ var AgnoClient = class extends EventEmitter {
     if (this.state.isStreaming) {
       throw new Error("Already streaming a message");
     }
+    this.runCompletedSuccessfully = false;
     const runUrl = this.configManager.getRunUrl();
     if (!runUrl) {
       throw new Error("No agent or team selected");
@@ -890,11 +921,7 @@ var AgnoClient = class extends EventEmitter {
       if (userId) {
         formData.append("user_id", userId);
       }
-      const headers = { ...options?.headers };
-      const authToken = this.configManager.getAuthToken();
-      if (authToken) {
-        headers["Authorization"] = `Bearer ${authToken}`;
-      }
+      const headers = this.configManager.buildRequestHeaders(options?.headers);
       await streamResponse({
         apiUrl: runUrl,
         headers,
@@ -911,11 +938,15 @@ var AgnoClient = class extends EventEmitter {
         onError: (error) => {
           this.handleError(error, newSessionId);
         },
-        onComplete: () => {
+        onComplete: async () => {
           this.state.isStreaming = false;
           this.emit("stream:end");
           this.emit("message:complete", this.messageStore.getMessages());
           this.emit("state:change", this.getState());
+          if (this.runCompletedSuccessfully) {
+            this.runCompletedSuccessfully = false;
+            await this.refreshSessionMessages();
+          }
         }
       });
     } catch (error) {
@@ -979,6 +1010,9 @@ var AgnoClient = class extends EventEmitter {
       return updated || lastMessage;
     });
     this.applyPendingUISpecs();
+    if (event === RunEvent.RunCompleted || event === RunEvent.TeamRunCompleted) {
+      this.runCompletedSuccessfully = true;
+    }
     this.emit("message:update", this.messageStore.getMessages());
   }
   /**
@@ -1000,6 +1034,70 @@ var AgnoClient = class extends EventEmitter {
     this.emit("stream:end");
     this.emit("state:change", this.getState());
   }
+  /**
+   * Refresh messages from the session API after run completion.
+   * Replaces streamed messages with authoritative session data.
+   * Preserves client-side properties like ui_component that aren't stored on the server.
+   * @private
+   */
+  async refreshSessionMessages() {
+    const sessionId = this.configManager.getSessionId();
+    if (!sessionId) {
+      Logger.debug("[AgnoClient] Cannot refresh: no session ID");
+      return;
+    }
+    this.state.isRefreshing = true;
+    this.emit("state:change", this.getState());
+    try {
+      const existingUIComponents = /* @__PURE__ */ new Map();
+      for (const message of this.messageStore.getMessages()) {
+        if (message.tool_calls) {
+          for (const toolCall of message.tool_calls) {
+            if (toolCall.ui_component) {
+              existingUIComponents.set(toolCall.tool_call_id, toolCall.ui_component);
+            }
+          }
+        }
+      }
+      const config = this.configManager.getConfig();
+      const entityType = this.configManager.getMode();
+      const dbId = this.configManager.getDbId() || "";
+      const userId = this.configManager.getUserId();
+      const headers = this.configManager.buildRequestHeaders();
+      const response = await this.sessionManager.fetchSession(
+        config.endpoint,
+        entityType,
+        sessionId,
+        dbId,
+        headers,
+        userId
+      );
+      const messages = this.sessionManager.convertSessionToMessages(response);
+      if (existingUIComponents.size > 0) {
+        for (const message of messages) {
+          if (message.tool_calls) {
+            for (let i = 0; i < message.tool_calls.length; i++) {
+              const toolCall = message.tool_calls[i];
+              const uiComponent = existingUIComponents.get(toolCall.tool_call_id);
+              if (uiComponent) {
+                message.tool_calls[i].ui_component = uiComponent;
+              }
+            }
+          }
+        }
+      }
+      this.messageStore.setMessages(messages);
+      Logger.debug("[AgnoClient] Session refreshed:", `${messages.length} messages`);
+      this.emit("message:refreshed", messages);
+      this.emit("message:update", messages);
+    } catch (error) {
+      Logger.error("[AgnoClient] Failed to refresh session:", error);
+      this.emit("message:error", `Session refresh failed: ${error instanceof Error ? error.message : String(error)}`);
+    } finally {
+      this.state.isRefreshing = false;
+      this.emit("state:change", this.getState());
+    }
+  }
   /**
    * Load a session
    */
@@ -1008,13 +1106,16 @@ var AgnoClient = class extends EventEmitter {
     const config = this.configManager.getConfig();
     const entityType = this.configManager.getMode();
     const dbId = this.configManager.getDbId() || "";
-    Logger.debug("[AgnoClient] Loading session with:", { entityType, dbId });
+    const userId = this.configManager.getUserId();
+    Logger.debug("[AgnoClient] Loading session with:", { entityType, dbId, userId });
+    const headers = this.configManager.buildRequestHeaders();
     const response = await this.sessionManager.fetchSession(
       config.endpoint,
       entityType,
       sessionId,
       dbId,
-      config.authToken
+      headers,
+      userId
     );
     const messages = this.sessionManager.convertSessionToMessages(response);
     Logger.debug("[AgnoClient] Setting messages to store:", `${messages.length} messages`);
@@ -1038,12 +1139,13 @@ var AgnoClient = class extends EventEmitter {
     if (!entityId) {
       throw new Error("Entity ID must be configured");
     }
+    const headers = this.configManager.buildRequestHeaders();
     const sessions = await this.sessionManager.fetchSessions(
       config.endpoint,
       entityType,
       entityId,
       dbId,
-      config.authToken
+      headers
     );
     this.state.sessions = sessions;
     this.emit("state:change", this.getState());
@@ -1055,11 +1157,12 @@ var AgnoClient = class extends EventEmitter {
   async deleteSession(sessionId) {
     const config = this.configManager.getConfig();
     const dbId = this.configManager.getDbId() || "";
+    const headers = this.configManager.buildRequestHeaders();
     await this.sessionManager.deleteSession(
       config.endpoint,
       sessionId,
       dbId,
-      config.authToken
+      headers
     );
     this.state.sessions = this.state.sessions.filter(
       (s) => s.session_id !== sessionId
@@ -1210,11 +1313,7 @@ var AgnoClient = class extends EventEmitter {
     if (userId) {
       formData.append("user_id", userId);
     }
-    const headers = { ...options?.headers };
-    const authToken = this.configManager.getAuthToken();
-    if (authToken) {
-      headers["Authorization"] = `Bearer ${authToken}`;
-    }
+    const headers = this.configManager.buildRequestHeaders(options?.headers);
     try {
       await streamResponse({
         apiUrl: continueUrl,
@@ -1226,13 +1325,17 @@ var AgnoClient = class extends EventEmitter {
         onError: (error) => {
           this.handleError(error, currentSessionId);
         },
-        onComplete: () => {
+        onComplete: async () => {
           this.state.isStreaming = false;
           this.state.pausedRunId = void 0;
           this.state.toolsAwaitingExecution = void 0;
           this.emit("stream:end");
           this.emit("message:complete", this.messageStore.getMessages());
           this.emit("state:change", this.getState());
+          if (this.runCompletedSuccessfully) {
+            this.runCompletedSuccessfully = false;
+            await this.refreshSessionMessages();
+          }
         }
       });
     } catch (error) {
@@ -1247,7 +1350,8 @@ var AgnoClient = class extends EventEmitter {
    */
   async checkStatus() {
     try {
-      const response = await fetch(`${this.configManager.getEndpoint()}/health`);
+      const headers = this.configManager.buildRequestHeaders();
+      const response = await fetch(`${this.configManager.getEndpoint()}/health`, { headers });
       const isActive = response.ok;
       this.state.isEndpointActive = isActive;
       this.emit("state:change", this.getState());
@@ -1262,12 +1366,8 @@ var AgnoClient = class extends EventEmitter {
    * Fetch agents from endpoint
    */
   async fetchAgents() {
-    const config = this.configManager.getConfig();
-    const headers = {};
-    if (config.authToken) {
-      headers["Authorization"] = `Bearer ${config.authToken}`;
-    }
-    const response = await fetch(`${config.endpoint}/agents`, { headers });
+    const headers = this.configManager.buildRequestHeaders();
+    const response = await fetch(`${this.configManager.getEndpoint()}/agents`, { headers });
     if (!response.ok) {
       throw new Error("Failed to fetch agents");
     }
@@ -1280,12 +1380,8 @@ var AgnoClient = class extends EventEmitter {
    * Fetch teams from endpoint
    */
   async fetchTeams() {
-    const config = this.configManager.getConfig();
-    const headers = {};
-    if (config.authToken) {
-      headers["Authorization"] = `Bearer ${config.authToken}`;
-    }
-    const response = await fetch(`${config.endpoint}/teams`, { headers });
+    const headers = this.configManager.buildRequestHeaders();
+    const response = await fetch(`${this.configManager.getEndpoint()}/teams`, { headers });
     if (!response.ok) {
       throw new Error("Failed to fetch teams");
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@antipopp/agno-client",
-  "version": "0.6.1",
+  "version": "0.8.0",
   "description": "Core client library for Agno agents with streaming support and HITL frontend tool execution",
   "author": "antipopp",
   "license": "MIT",
@@ -34,7 +34,7 @@
   ],
   "dependencies": {
     "eventemitter3": "^5.0.1",
-    "@antipopp/agno-types": "0.6.1"
+    "@antipopp/agno-types": "0.8.0"
   },
   "devDependencies": {
     "tsup": "^8.0.1",