@antipopp/agno-client 0.7.0 → 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

@@ -323,12 +323,15 @@ 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, headers) {
+  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);
     }
+    if (userId) {
+      url.searchParams.set("user_id", userId);
+    }
     const response = await fetch(url.toString(), { headers });
     if (!response.ok) {
       throw new Error(`Failed to fetch session: ${response.statusText}`);
@@ -851,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();
@@ -861,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: [],
@@ -912,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");
@@ -970,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) {
@@ -1038,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());
   }
   /**
@@ -1059,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
    */
@@ -1067,14 +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,
-      headers
+      headers,
+      userId
     );
     const messages = this.sessionManager.convertSessionToMessages(response);
     Logger.debug("[AgnoClient] Setting messages to store:", `${messages.length} messages`);
@@ -1284,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) {

package/dist/index.mjs

@@ -285,12 +285,15 @@ 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, headers) {
+  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);
     }
+    if (userId) {
+      url.searchParams.set("user_id", userId);
+    }
     const response = await fetch(url.toString(), { headers });
     if (!response.ok) {
       throw new Error(`Failed to fetch session: ${response.statusText}`);
@@ -813,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();
@@ -823,6 +827,7 @@ var AgnoClient = class extends EventEmitter {
     this.pendingUISpecs = /* @__PURE__ */ new Map();
     this.state = {
       isStreaming: false,
+      isRefreshing: false,
       isEndpointActive: false,
       agents: [],
       teams: [],
@@ -874,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");
@@ -932,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) {
@@ -1000,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());
   }
   /**
@@ -1021,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
    */
@@ -1029,14 +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,
-      headers
+      headers,
+      userId
     );
     const messages = this.sessionManager.convertSessionToMessages(response);
     Logger.debug("[AgnoClient] Setting messages to store:", `${messages.length} messages`);
@@ -1246,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) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@antipopp/agno-client",
-  "version": "0.7.0",
+  "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.7.0"
+    "@antipopp/agno-types": "0.8.0"
   },
   "devDependencies": {
     "tsup": "^8.0.1",