@antipopp/agno-client 0.8.0 → 0.10.0

package/dist/index.mjs

@@ -1,85 +1,9 @@
+// src/index.ts
+import { RunEvent as RunEvent2 } from "@antipopp/agno-types";
+
 // src/client.ts
-import EventEmitter from "eventemitter3";
 import { RunEvent } from "@antipopp/agno-types";
-
-// src/stores/message-store.ts
-var MessageStore = class {
-  constructor() {
-    this.messages = [];
-  }
-  /**
-   * Get all messages
-   */
-  getMessages() {
-    return [...this.messages];
-  }
-  /**
-   * Set messages (replaces all)
-   */
-  setMessages(messages) {
-    this.messages = [...messages];
-  }
-  /**
-   * Add a message
-   */
-  addMessage(message) {
-    this.messages = [...this.messages, message];
-  }
-  /**
-   * Update the last message
-   */
-  updateLastMessage(updater) {
-    if (this.messages.length === 0)
-      return void 0;
-    const lastMessage = this.messages[this.messages.length - 1];
-    const updatedMessage = updater(lastMessage);
-    this.messages = [
-      ...this.messages.slice(0, -1),
-      updatedMessage
-    ];
-    return updatedMessage;
-  }
-  /**
-   * Update a specific message by index
-   */
-  updateMessage(index, updater) {
-    if (index < 0 || index >= this.messages.length)
-      return void 0;
-    const message = this.messages[index];
-    const updatedMessage = updater(message);
-    this.messages = [
-      ...this.messages.slice(0, index),
-      updatedMessage,
-      ...this.messages.slice(index + 1)
-    ];
-    return updatedMessage;
-  }
-  /**
-   * Remove last N messages
-   */
-  removeLastMessages(count) {
-    this.messages = this.messages.slice(0, -count);
-  }
-  /**
-   * Clear all messages
-   */
-  clear() {
-    this.messages = [];
-  }
-  /**
-   * Get the last message
-   */
-  getLastMessage() {
-    return this.messages.length > 0 ? this.messages[this.messages.length - 1] : void 0;
-  }
-  /**
-   * Check if last message has streaming error
-   */
-  hasLastMessageError() {
-    const lastMessage = this.getLastMessage();
-    return lastMessage?.streamingError === true;
-  }
-};
+import EventEmitter from "eventemitter3";
 
 // src/managers/config-manager.ts
 var ConfigManager = class {
@@ -212,6 +136,18 @@ var ConfigManager = class {
   setHeaders(headers) {
     this.updateField("headers", headers);
   }
+  /**
+   * Get global query parameters
+   */
+  getParams() {
+    return this.config.params;
+  }
+  /**
+   * Set global query parameters
+   */
+  setParams(params) {
+    this.updateField("params", params);
+  }
   /**
    * Get current entity ID (agent or team based on mode)
    */
@@ -225,14 +161,14 @@ var ConfigManager = class {
     const mode = this.getMode();
     const endpoint = this.getEndpoint();
     const entityId = this.getCurrentEntityId();
-    if (!entityId)
+    if (!entityId) {
       return null;
+    }
     const encodedEntityId = encodeURIComponent(entityId);
     if (mode === "team") {
       return `${endpoint}/teams/${encodedEntityId}/runs`;
-    } else {
-      return `${endpoint}/agents/${encodedEntityId}/runs`;
     }
+    return `${endpoint}/agents/${encodedEntityId}/runs`;
   }
   /**
    * Build request headers by merging global headers, per-request headers, and auth token.
@@ -255,10 +191,30 @@ var ConfigManager = class {
     }
     const authToken = this.getAuthToken();
     if (authToken) {
-      headers["Authorization"] = `Bearer ${authToken}`;
+      headers.Authorization = `Bearer ${authToken}`;
     }
     return headers;
   }
+  /**
+   * Build query string by merging global params and per-request params.
+   * Merge order (lowest to highest precedence):
+   * 1. Global params from config
+   * 2. Per-request params (overrides global)
+   *
+   * @param perRequestParams - Optional query parameters for this specific request
+   * @returns URLSearchParams object ready to append to URLs
+   */
+  buildQueryString(perRequestParams) {
+    const params = {};
+    const globalParams = this.getParams();
+    if (globalParams) {
+      Object.assign(params, globalParams);
+    }
+    if (perRequestParams) {
+      Object.assign(params, perRequestParams);
+    }
+    return new URLSearchParams(params);
+  }
 };
 
 // src/managers/session-manager.ts
@@ -266,11 +222,16 @@ var SessionManager = class {
   /**
    * Fetch all sessions for an entity
    */
-  async fetchSessions(endpoint, entityType, entityId, dbId, headers) {
+  async fetchSessions(endpoint, entityType, entityId, dbId, headers, params) {
     const url = new URL(`${endpoint}/sessions`);
     url.searchParams.set("type", entityType);
     url.searchParams.set("component_id", entityId);
     url.searchParams.set("db_id", dbId);
+    if (params) {
+      params.forEach((value, key) => {
+        url.searchParams.set(key, value);
+      });
+    }
     const response = await fetch(url.toString(), { headers });
     if (!response.ok) {
       if (response.status === 404) {
@@ -285,7 +246,7 @@ 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, userId) {
+  async fetchSession(endpoint, entityType, sessionId, dbId, headers, userId, params) {
     const url = new URL(`${endpoint}/sessions/${sessionId}/runs`);
     url.searchParams.set("type", entityType);
     if (dbId) {
@@ -294,6 +255,11 @@ var SessionManager = class {
     if (userId) {
       url.searchParams.set("user_id", userId);
     }
+    if (params) {
+      params.forEach((value, key) => {
+        url.searchParams.set(key, value);
+      });
+    }
     const response = await fetch(url.toString(), { headers });
     if (!response.ok) {
       throw new Error(`Failed to fetch session: ${response.statusText}`);
@@ -303,11 +269,16 @@ var SessionManager = class {
   /**
    * Delete a session
    */
-  async deleteSession(endpoint, sessionId, dbId, headers) {
+  async deleteSession(endpoint, sessionId, dbId, headers, params) {
     const url = new URL(`${endpoint}/sessions/${sessionId}`);
     if (dbId) {
       url.searchParams.set("db_id", dbId);
     }
+    if (params) {
+      params.forEach((value, key) => {
+        url.searchParams.set(key, value);
+      });
+    }
     const response = await fetch(url.toString(), {
       method: "DELETE",
       headers
@@ -366,7 +337,9 @@ var SessionManager = class {
               tool_name: reasoningMsg.tool_name ?? "",
               tool_args: reasoningMsg.tool_args ?? {},
               tool_call_error: reasoningMsg.tool_call_error ?? false,
-              metrics: reasoningMsg.metrics ?? { time: 0 },
+              metrics: reasoningMsg.metrics ?? {
+                time: 0
+              },
               created_at: reasoningMsg.created_at ?? timestamp
             });
           }
@@ -400,6 +373,164 @@ var SessionManager = class {
   }
 };
 
+// src/parsers/stream-parser.ts
+function isLegacyFormat(data) {
+  return typeof data === "object" && data !== null && "event" in data && !("data" in data) && typeof data.event === "string";
+}
+function convertNewFormatToLegacy(newFormatData) {
+  const { event, data } = newFormatData;
+  let parsedData;
+  if (typeof data === "string") {
+    try {
+      parsedData = JSON.parse(data);
+    } catch {
+      parsedData = {};
+    }
+  } else {
+    parsedData = data;
+  }
+  return {
+    event,
+    ...parsedData
+  };
+}
+function processChunk(chunk, onChunk) {
+  onChunk(chunk);
+}
+function parseBuffer(buffer, onChunk) {
+  let currentIndex = 0;
+  let jsonStartIndex = buffer.indexOf("{", currentIndex);
+  while (jsonStartIndex !== -1 && jsonStartIndex < buffer.length) {
+    let braceCount = 0;
+    let inString = false;
+    let escapeNext = false;
+    let jsonEndIndex = -1;
+    let i = jsonStartIndex;
+    for (; i < buffer.length; i++) {
+      const char = buffer[i];
+      if (inString) {
+        if (escapeNext) {
+          escapeNext = false;
+        } else if (char === "\\") {
+          escapeNext = true;
+        } else if (char === '"') {
+          inString = false;
+        }
+      } else if (char === '"') {
+        inString = true;
+      } else if (char === "{") {
+        braceCount++;
+      } else if (char === "}") {
+        braceCount--;
+        if (braceCount === 0) {
+          jsonEndIndex = i;
+          break;
+        }
+      }
+    }
+    if (jsonEndIndex !== -1) {
+      const jsonString = buffer.slice(jsonStartIndex, jsonEndIndex + 1);
+      try {
+        const parsed = JSON.parse(jsonString);
+        if (isLegacyFormat(parsed)) {
+          processChunk(parsed, onChunk);
+        } else {
+          const legacyChunk = convertNewFormatToLegacy(parsed);
+          processChunk(legacyChunk, onChunk);
+        }
+      } catch (error) {
+        if (typeof process !== "undefined" && process.env?.NODE_ENV === "development") {
+          console.error("Failed to parse JSON chunk:", {
+            error,
+            chunk: jsonString.substring(0, 100) + (jsonString.length > 100 ? "..." : ""),
+            position: jsonStartIndex
+          });
+        }
+        if (jsonString.length > 1e4) {
+          throw new Error(
+            `Failed to parse large JSON chunk at position ${jsonStartIndex}`
+          );
+        }
+        jsonStartIndex = buffer.indexOf("{", jsonStartIndex + 1);
+        continue;
+      }
+      currentIndex = jsonEndIndex + 1;
+      buffer = buffer.slice(currentIndex).trim();
+      currentIndex = 0;
+      jsonStartIndex = buffer.indexOf("{", currentIndex);
+    } else {
+      break;
+    }
+  }
+  return buffer;
+}
+async function streamResponse(options) {
+  const {
+    apiUrl,
+    headers = {},
+    params,
+    requestBody,
+    onChunk,
+    onError,
+    onComplete,
+    signal
+  } = options;
+  let buffer = "";
+  const finalUrl = params?.toString() ? `${apiUrl}?${params.toString()}` : apiUrl;
+  try {
+    const response = await fetch(finalUrl, {
+      method: "POST",
+      headers: {
+        ...!(requestBody instanceof FormData) && {
+          "Content-Type": "application/json"
+        },
+        ...headers
+      },
+      body: requestBody instanceof FormData ? requestBody : JSON.stringify(requestBody),
+      signal
+    });
+    if (!response.ok) {
+      let errorMessage = `HTTP ${response.status}: ${response.statusText}`;
+      const contentType = response.headers.get("content-type");
+      if (contentType?.includes("application/json")) {
+        try {
+          const errorData = await response.json();
+          errorMessage = errorData.detail || errorData.message || errorMessage;
+        } catch {
+        }
+      }
+      throw new Error(errorMessage);
+    }
+    if (!response.body) {
+      throw new Error("No response body");
+    }
+    const reader = response.body.getReader();
+    const decoder = new TextDecoder();
+    const processStream = async () => {
+      while (true) {
+        const { done, value } = await reader.read();
+        if (done) {
+          buffer = parseBuffer(buffer, onChunk);
+          onComplete();
+          return;
+        }
+        buffer += decoder.decode(value, { stream: true });
+        buffer = parseBuffer(buffer, onChunk);
+      }
+    };
+    await processStream();
+  } catch (error) {
+    if (error instanceof Error && error.name === "AbortError") {
+      return;
+    }
+    if (typeof error === "object" && error !== null && "detail" in error) {
+      onError(new Error(String(error.detail)));
+    } else {
+      onError(new Error(String(error)));
+    }
+  }
+}
+
 // src/processors/event-processor.ts
 import { RunEvent as RunEventEnum } from "@antipopp/agno-types";
 
@@ -428,9 +559,8 @@ function processToolCall(toolCall, prevToolCalls = []) {
       ...toolCall
     };
     return updatedToolCalls;
-  } else {
-    return [...prevToolCalls, toolCall];
   }
+  return [...prevToolCalls, toolCall];
 }
 function processChunkToolCalls(chunk, existingToolCalls = []) {
   let updatedToolCalls = [...existingToolCalls];
@@ -518,7 +648,7 @@ var EventProcessor = class {
         }
         break;
       case RunEventEnum.ReasoningStep:
-      case RunEventEnum.TeamReasoningStep:
+      case RunEventEnum.TeamReasoningStep: {
         const existingSteps = lastMessage.extra_data?.reasoning_steps ?? [];
         const incomingSteps = chunk.extra_data?.reasoning_steps ?? [];
         updatedMessage.extra_data = {
@@ -526,6 +656,7 @@ var EventProcessor = class {
           reasoning_steps: [...existingSteps, ...incomingSteps]
         };
         break;
+      }
       case RunEventEnum.ReasoningCompleted:
       case RunEventEnum.TeamReasoningCompleted:
         if (chunk.extra_data?.reasoning_steps) {
@@ -536,7 +667,7 @@ var EventProcessor = class {
         }
         break;
       case RunEventEnum.RunCompleted:
-      case RunEventEnum.TeamRunCompleted:
+      case RunEventEnum.TeamRunCompleted: {
         let updatedContent;
         if (typeof chunk.content === "string") {
           updatedContent = chunk.content;
@@ -561,6 +692,7 @@ var EventProcessor = class {
           references: chunk.extra_data?.references ?? lastMessage.extra_data?.references
         };
         break;
+      }
       case RunEventEnum.UpdatingMemory:
       case RunEventEnum.TeamMemoryUpdateStarted:
       case RunEventEnum.TeamMemoryUpdateCompleted:
@@ -583,164 +715,95 @@ var EventProcessor = class {
   }
 };
 
-// src/parsers/stream-parser.ts
-function isLegacyFormat(data) {
-  return typeof data === "object" && data !== null && "event" in data && !("data" in data) && typeof data.event === "string";
-}
-function convertNewFormatToLegacy(newFormatData) {
-  const { event, data } = newFormatData;
-  let parsedData;
-  if (typeof data === "string") {
-    try {
-      parsedData = JSON.parse(data);
-    } catch {
-      parsedData = {};
-    }
-  } else {
-    parsedData = data;
+// src/stores/message-store.ts
+var MessageStore = class {
+  constructor() {
+    this.messages = [];
   }
-  return {
-    event,
-    ...parsedData
-  };
-}
-function processChunk(chunk, onChunk) {
-  onChunk(chunk);
-}
-function parseBuffer(buffer, onChunk) {
-  let currentIndex = 0;
-  let jsonStartIndex = buffer.indexOf("{", currentIndex);
-  while (jsonStartIndex !== -1 && jsonStartIndex < buffer.length) {
-    let braceCount = 0;
-    let inString = false;
-    let escapeNext = false;
-    let jsonEndIndex = -1;
-    let i = jsonStartIndex;
-    for (; i < buffer.length; i++) {
-      const char = buffer[i];
-      if (inString) {
-        if (escapeNext) {
-          escapeNext = false;
-        } else if (char === "\\") {
-          escapeNext = true;
-        } else if (char === '"') {
-          inString = false;
-        }
-      } else {
-        if (char === '"') {
-          inString = true;
-        } else if (char === "{") {
-          braceCount++;
-        } else if (char === "}") {
-          braceCount--;
-          if (braceCount === 0) {
-            jsonEndIndex = i;
-            break;
-          }
-        }
-      }
-    }
-    if (jsonEndIndex !== -1) {
-      const jsonString = buffer.slice(jsonStartIndex, jsonEndIndex + 1);
-      try {
-        const parsed = JSON.parse(jsonString);
-        if (isLegacyFormat(parsed)) {
-          processChunk(parsed, onChunk);
-        } else {
-          const legacyChunk = convertNewFormatToLegacy(parsed);
-          processChunk(legacyChunk, onChunk);
-        }
-      } catch (error) {
-        if (typeof process !== "undefined" && process.env?.NODE_ENV === "development") {
-          console.error("Failed to parse JSON chunk:", {
-            error,
-            chunk: jsonString.substring(0, 100) + (jsonString.length > 100 ? "..." : ""),
-            position: jsonStartIndex
-          });
-        }
-        if (jsonString.length > 1e4) {
-          throw new Error(`Failed to parse large JSON chunk at position ${jsonStartIndex}`);
-        }
-        jsonStartIndex = buffer.indexOf("{", jsonStartIndex + 1);
-        continue;
-      }
-      currentIndex = jsonEndIndex + 1;
-      buffer = buffer.slice(currentIndex).trim();
-      currentIndex = 0;
-      jsonStartIndex = buffer.indexOf("{", currentIndex);
-    } else {
-      break;
-    }
+  /**
+   * Get all messages
+   */
+  getMessages() {
+    return [...this.messages];
   }
-  return buffer;
-}
-async function streamResponse(options) {
-  const {
-    apiUrl,
-    headers = {},
-    requestBody,
-    onChunk,
-    onError,
-    onComplete,
-    signal
-  } = options;
-  let buffer = "";
-  try {
-    const response = await fetch(apiUrl, {
-      method: "POST",
-      headers: {
-        ...!(requestBody instanceof FormData) && {
-          "Content-Type": "application/json"
-        },
-        ...headers
-      },
-      body: requestBody instanceof FormData ? requestBody : JSON.stringify(requestBody),
-      signal
-    });
-    if (!response.ok) {
-      let errorMessage = `HTTP ${response.status}: ${response.statusText}`;
-      const contentType = response.headers.get("content-type");
-      if (contentType?.includes("application/json")) {
-        try {
-          const errorData = await response.json();
-          errorMessage = errorData.detail || errorData.message || errorMessage;
-        } catch {
-        }
-      }
-      throw new Error(errorMessage);
-    }
-    if (!response.body) {
-      throw new Error("No response body");
+  /**
+   * Set messages (replaces all)
+   */
+  setMessages(messages) {
+    this.messages = [...messages];
+  }
+  /**
+   * Add a message
+   */
+  addMessage(message) {
+    this.messages = [...this.messages, message];
+  }
+  /**
+   * Update the last message
+   */
+  updateLastMessage(updater) {
+    if (this.messages.length === 0) {
+      return void 0;
     }
-    const reader = response.body.getReader();
-    const decoder = new TextDecoder();
-    const processStream = async () => {
-      while (true) {
-        const { done, value } = await reader.read();
-        if (done) {
-          buffer = parseBuffer(buffer, onChunk);
-          onComplete();
-          return;
-        }
-        buffer += decoder.decode(value, { stream: true });
-        buffer = parseBuffer(buffer, onChunk);
-      }
-    };
-    await processStream();
-  } catch (error) {
-    if (error instanceof Error && error.name === "AbortError") {
-      return;
+    const lastMessage = this.messages.at(-1);
+    if (!lastMessage) {
+      return void 0;
     }
-    if (typeof error === "object" && error !== null && "detail" in error) {
-      onError(new Error(String(error.detail)));
-    } else {
-      onError(new Error(String(error)));
+    const updatedMessage = updater(lastMessage);
+    this.messages = [...this.messages.slice(0, -1), updatedMessage];
+    return updatedMessage;
+  }
+  /**
+   * Update a specific message by index
+   */
+  updateMessage(index, updater) {
+    if (index < 0 || index >= this.messages.length) {
+      return void 0;
     }
+    const message = this.messages[index];
+    const updatedMessage = updater(message);
+    this.messages = [
+      ...this.messages.slice(0, index),
+      updatedMessage,
+      ...this.messages.slice(index + 1)
+    ];
+    return updatedMessage;
   }
-}
+  /**
+   * Remove last N messages
+   */
+  removeLastMessages(count) {
+    this.messages = this.messages.slice(0, -count);
+  }
+  /**
+   * Clear all messages
+   */
+  clear() {
+    this.messages = [];
+  }
+  /**
+   * Get the last message
+   */
+  getLastMessage() {
+    return this.messages.length > 0 ? this.messages.at(-1) : void 0;
+  }
+  /**
+   * Check if last message has streaming error
+   */
+  hasLastMessageError() {
+    const lastMessage = this.getLastMessage();
+    return lastMessage?.streamingError === true;
+  }
+};
 
 // src/utils/logger.ts
-var SENSITIVE_KEYS = ["authToken", "Authorization", "token", "password", "apiKey"];
+var SENSITIVE_KEYS = [
+  "authToken",
+  "Authorization",
+  "token",
+  "password",
+  "apiKey"
+];
 function sanitizeObject(obj) {
   if (obj === null || obj === void 0) {
     return obj;
@@ -828,6 +891,7 @@ var AgnoClient = class extends EventEmitter {
     this.state = {
       isStreaming: false,
       isRefreshing: false,
+      isCancelling: false,
       isEndpointActive: false,
       agents: [],
       teams: [],
@@ -872,6 +936,49 @@ var AgnoClient = class extends EventEmitter {
     this.emit("message:update", this.messageStore.getMessages());
     this.emit("state:change", this.getState());
   }
+  /**
+   * Cancel an active or paused run
+   */
+  async cancelRun() {
+    if (!(this.state.isStreaming || this.state.isPaused)) {
+      throw new Error("No active or paused run to cancel");
+    }
+    const runUrl = this.configManager.getRunUrl();
+    if (!runUrl) {
+      throw new Error("No agent or team selected");
+    }
+    const runId = this.state.pausedRunId || this.currentRunId;
+    if (!runId) {
+      throw new Error("No run ID available to cancel");
+    }
+    const cancelUrl = `${runUrl}/${runId}/cancel`;
+    const headers = this.configManager.buildRequestHeaders();
+    this.state.isCancelling = true;
+    this.emit("state:change", this.getState());
+    try {
+      const response = await fetch(cancelUrl, {
+        method: "POST",
+        headers
+      });
+      if (!response.ok) {
+        throw new Error("Failed to cancel run");
+      }
+      this.state.isStreaming = false;
+      this.state.isPaused = false;
+      this.state.isCancelling = false;
+      this.state.pausedRunId = void 0;
+      this.state.toolsAwaitingExecution = void 0;
+      this.currentRunId = void 0;
+      this.emit("run:cancelled", { runId });
+      this.emit("state:change", this.getState());
+    } catch (error) {
+      this.state.isCancelling = false;
+      this.emit("state:change", this.getState());
+      throw new Error(
+        `Error cancelling run: ${error instanceof Error ? error.message : String(error)}`
+      );
+    }
+  }
   /**
    * Send a message to the agent/team (streaming)
    */
@@ -922,17 +1029,21 @@ var AgnoClient = class extends EventEmitter {
         formData.append("user_id", userId);
       }
       const headers = this.configManager.buildRequestHeaders(options?.headers);
+      const params = this.configManager.buildQueryString(options?.params);
       await streamResponse({
         apiUrl: runUrl,
         headers,
+        params,
         requestBody: formData,
         onChunk: (chunk) => {
-          this.handleChunk(chunk, newSessionId, formData.get("message"));
-          if (chunk.event === RunEvent.RunStarted || chunk.event === RunEvent.TeamRunStarted || chunk.event === RunEvent.ReasoningStarted || chunk.event === RunEvent.TeamReasoningStarted) {
-            if (chunk.session_id) {
-              newSessionId = chunk.session_id;
-              this.configManager.setSessionId(chunk.session_id);
-            }
+          this.handleChunk(
+            chunk,
+            newSessionId,
+            formData.get("message")
+          );
+          if ((chunk.event === RunEvent.RunStarted || chunk.event === RunEvent.TeamRunStarted || chunk.event === RunEvent.ReasoningStarted || chunk.event === RunEvent.TeamReasoningStarted) && chunk.session_id) {
+            newSessionId = chunk.session_id;
+            this.configManager.setSessionId(chunk.session_id);
           }
         },
         onError: (error) => {
@@ -940,6 +1051,7 @@ var AgnoClient = class extends EventEmitter {
         },
         onComplete: async () => {
           this.state.isStreaming = false;
+          this.currentRunId = void 0;
           this.emit("stream:end");
           this.emit("message:complete", this.messageStore.getMessages());
           this.emit("state:change", this.getState());
@@ -961,20 +1073,21 @@ var AgnoClient = class extends EventEmitter {
    */
   handleChunk(chunk, currentSessionId, messageContent) {
     const event = chunk.event;
-    if (event === RunEvent.RunStarted || event === RunEvent.TeamRunStarted || event === RunEvent.ReasoningStarted || event === RunEvent.TeamReasoningStarted) {
-      if (chunk.session_id && (!currentSessionId || currentSessionId !== chunk.session_id)) {
-        const sessionData = {
-          session_id: chunk.session_id,
-          session_name: messageContent,
-          created_at: toSafeISOString(chunk.created_at)
-        };
-        const sessionExists = this.state.sessions.some(
-          (s) => s.session_id === chunk.session_id
-        );
-        if (!sessionExists) {
-          this.state.sessions = [sessionData, ...this.state.sessions];
-          this.emit("session:created", sessionData);
-        }
+    if ((event === RunEvent.RunStarted || event === RunEvent.TeamRunStarted) && chunk.run_id) {
+      this.currentRunId = chunk.run_id;
+    }
+    if ((event === RunEvent.RunStarted || event === RunEvent.TeamRunStarted || event === RunEvent.ReasoningStarted || event === RunEvent.TeamReasoningStarted) && chunk.session_id && (!currentSessionId || currentSessionId !== chunk.session_id)) {
+      const sessionData = {
+        session_id: chunk.session_id,
+        session_name: messageContent,
+        created_at: toSafeISOString(chunk.created_at)
+      };
+      const sessionExists = this.state.sessions.some(
+        (s) => s.session_id === chunk.session_id
+      );
+      if (!sessionExists) {
+        this.state.sessions = [sessionData, ...this.state.sessions];
+        this.emit("session:created", sessionData);
       }
     }
     if (event === RunEvent.RunPaused) {
@@ -1054,7 +1167,10 @@ var AgnoClient = class extends EventEmitter {
         if (message.tool_calls) {
           for (const toolCall of message.tool_calls) {
             if (toolCall.ui_component) {
-              existingUIComponents.set(toolCall.tool_call_id, toolCall.ui_component);
+              existingUIComponents.set(
+                toolCall.tool_call_id,
+                toolCall.ui_component
+              );
             }
           }
         }
@@ -1064,13 +1180,15 @@ var AgnoClient = class extends EventEmitter {
       const dbId = this.configManager.getDbId() || "";
       const userId = this.configManager.getUserId();
       const headers = this.configManager.buildRequestHeaders();
+      const params = this.configManager.buildQueryString();
       const response = await this.sessionManager.fetchSession(
         config.endpoint,
         entityType,
         sessionId,
         dbId,
         headers,
-        userId
+        userId,
+        params
       );
       const messages = this.sessionManager.convertSessionToMessages(response);
       if (existingUIComponents.size > 0) {
@@ -1078,7 +1196,9 @@ var AgnoClient = class extends EventEmitter {
           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);
+              const uiComponent = existingUIComponents.get(
+                toolCall.tool_call_id
+              );
               if (uiComponent) {
                 message.tool_calls[i].ui_component = uiComponent;
               }
@@ -1087,12 +1207,18 @@ var AgnoClient = class extends EventEmitter {
         }
       }
       this.messageStore.setMessages(messages);
-      Logger.debug("[AgnoClient] Session refreshed:", `${messages.length} 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)}`);
+      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());
@@ -1101,24 +1227,33 @@ var AgnoClient = class extends EventEmitter {
   /**
    * Load a session
    */
-  async loadSession(sessionId) {
+  async loadSession(sessionId, options) {
     Logger.debug("[AgnoClient] loadSession called with sessionId:", sessionId);
     const config = this.configManager.getConfig();
     const entityType = this.configManager.getMode();
     const dbId = this.configManager.getDbId() || "";
     const userId = this.configManager.getUserId();
-    Logger.debug("[AgnoClient] Loading session with:", { entityType, dbId, userId });
+    Logger.debug("[AgnoClient] Loading session with:", {
+      entityType,
+      dbId,
+      userId
+    });
     const headers = this.configManager.buildRequestHeaders();
+    const params = this.configManager.buildQueryString(options?.params);
     const response = await this.sessionManager.fetchSession(
       config.endpoint,
       entityType,
       sessionId,
       dbId,
       headers,
-      userId
+      userId,
+      params
     );
     const messages = this.sessionManager.convertSessionToMessages(response);
-    Logger.debug("[AgnoClient] Setting messages to store:", `${messages.length} messages`);
+    Logger.debug(
+      "[AgnoClient] Setting messages to store:",
+      `${messages.length} messages`
+    );
     this.messageStore.setMessages(messages);
     this.configManager.setSessionId(sessionId);
     Logger.debug("[AgnoClient] Emitting events...");
@@ -1131,7 +1266,7 @@ var AgnoClient = class extends EventEmitter {
   /**
    * Fetch all sessions
    */
-  async fetchSessions() {
+  async fetchSessions(options) {
     const config = this.configManager.getConfig();
     const entityType = this.configManager.getMode();
     const entityId = this.configManager.getCurrentEntityId();
@@ -1140,12 +1275,14 @@ var AgnoClient = class extends EventEmitter {
       throw new Error("Entity ID must be configured");
     }
     const headers = this.configManager.buildRequestHeaders();
+    const params = this.configManager.buildQueryString(options?.params);
     const sessions = await this.sessionManager.fetchSessions(
       config.endpoint,
       entityType,
       entityId,
       dbId,
-      headers
+      headers,
+      params
     );
     this.state.sessions = sessions;
     this.emit("state:change", this.getState());
@@ -1154,15 +1291,17 @@ var AgnoClient = class extends EventEmitter {
   /**
    * Delete a session
    */
-  async deleteSession(sessionId) {
+  async deleteSession(sessionId, options) {
     const config = this.configManager.getConfig();
     const dbId = this.configManager.getDbId() || "";
     const headers = this.configManager.buildRequestHeaders();
+    const params = this.configManager.buildQueryString(options?.params);
     await this.sessionManager.deleteSession(
       config.endpoint,
       sessionId,
       dbId,
-      headers
+      headers,
+      params
     );
     this.state.sessions = this.state.sessions.filter(
       (s) => s.session_id !== sessionId
@@ -1183,7 +1322,9 @@ var AgnoClient = class extends EventEmitter {
     }
     const existingToolCalls = lastMessage.tool_calls || [];
     const existingIds = new Set(existingToolCalls.map((t) => t.tool_call_id));
-    const newToolCalls = toolCalls.filter((t) => !existingIds.has(t.tool_call_id));
+    const newToolCalls = toolCalls.filter(
+      (t) => !existingIds.has(t.tool_call_id)
+    );
     if (newToolCalls.length > 0) {
       this.messageStore.updateLastMessage((msg) => ({
         ...msg,
@@ -1230,8 +1371,9 @@ var AgnoClient = class extends EventEmitter {
    * Batches all updates to emit only one message:update event
    */
   applyPendingUISpecs() {
-    if (this.pendingUISpecs.size === 0)
+    if (this.pendingUISpecs.size === 0) {
       return;
+    }
     const messages = this.messageStore.getMessages();
     const updatedMessages = [];
     for (let i = messages.length - 1; i >= 0; i--) {
@@ -1276,7 +1418,7 @@ var AgnoClient = class extends EventEmitter {
    * Teams do not support the continue endpoint.
    *
    * @param tools - Array of tool calls with execution results
-   * @param options - Optional request headers
+   * @param options - Optional request headers and query parameters
    * @throws Error if no paused run exists
    * @throws Error if called with team mode (teams don't support HITL)
    */
@@ -1286,7 +1428,7 @@ var AgnoClient = class extends EventEmitter {
         "HITL (Human-in-the-Loop) frontend tool execution is not supported for teams. Only agents support the continue endpoint."
       );
     }
-    if (!this.state.isPaused || !this.state.pausedRunId) {
+    if (!(this.state.isPaused && this.state.pausedRunId)) {
       throw new Error("No paused run to continue");
     }
     const runUrl = this.configManager.getRunUrl();
@@ -1314,10 +1456,12 @@ var AgnoClient = class extends EventEmitter {
       formData.append("user_id", userId);
     }
     const headers = this.configManager.buildRequestHeaders(options?.headers);
+    const params = this.configManager.buildQueryString(options?.params);
     try {
       await streamResponse({
         apiUrl: continueUrl,
         headers,
+        params,
         requestBody: formData,
         onChunk: (chunk) => {
           this.handleChunk(chunk, currentSessionId, "");
@@ -1348,10 +1492,17 @@ var AgnoClient = class extends EventEmitter {
   /**
    * Check endpoint status
    */
-  async checkStatus() {
+  async checkStatus(options) {
     try {
       const headers = this.configManager.buildRequestHeaders();
-      const response = await fetch(`${this.configManager.getEndpoint()}/health`, { headers });
+      const params = this.configManager.buildQueryString(options?.params);
+      const url = new URL(`${this.configManager.getEndpoint()}/health`);
+      if (params.toString()) {
+        params.forEach((value, key) => {
+          url.searchParams.set(key, value);
+        });
+      }
+      const response = await fetch(url.toString(), { headers });
       const isActive = response.ok;
       this.state.isEndpointActive = isActive;
       this.emit("state:change", this.getState());
@@ -1365,9 +1516,16 @@ var AgnoClient = class extends EventEmitter {
   /**
    * Fetch agents from endpoint
    */
-  async fetchAgents() {
+  async fetchAgents(options) {
     const headers = this.configManager.buildRequestHeaders();
-    const response = await fetch(`${this.configManager.getEndpoint()}/agents`, { headers });
+    const params = this.configManager.buildQueryString(options?.params);
+    const url = new URL(`${this.configManager.getEndpoint()}/agents`);
+    if (params.toString()) {
+      params.forEach((value, key) => {
+        url.searchParams.set(key, value);
+      });
+    }
+    const response = await fetch(url.toString(), { headers });
     if (!response.ok) {
       throw new Error("Failed to fetch agents");
     }
@@ -1379,9 +1537,16 @@ var AgnoClient = class extends EventEmitter {
   /**
    * Fetch teams from endpoint
    */
-  async fetchTeams() {
+  async fetchTeams(options) {
     const headers = this.configManager.buildRequestHeaders();
-    const response = await fetch(`${this.configManager.getEndpoint()}/teams`, { headers });
+    const params = this.configManager.buildQueryString(options?.params);
+    const url = new URL(`${this.configManager.getEndpoint()}/teams`);
+    if (params.toString()) {
+      params.forEach((value, key) => {
+        url.searchParams.set(key, value);
+      });
+    }
+    const response = await fetch(url.toString(), { headers });
     if (!response.ok) {
       throw new Error("Failed to fetch teams");
     }
@@ -1394,19 +1559,19 @@ var AgnoClient = class extends EventEmitter {
    * Initialize client (check status and fetch agents/teams)
    * Automatically selects the first available agent or team if none is configured
    */
-  async initialize() {
-    const isActive = await this.checkStatus();
+  async initialize(options) {
+    const isActive = await this.checkStatus(options);
     if (!isActive) {
       return { agents: [], teams: [] };
     }
     const [agents, teams] = await Promise.all([
-      this.fetchAgents(),
-      this.fetchTeams()
+      this.fetchAgents(options),
+      this.fetchTeams(options)
     ]);
     const currentConfig = this.configManager.getConfig();
     const hasAgentConfigured = currentConfig.agentId;
     const hasTeamConfigured = currentConfig.teamId;
-    if (!hasAgentConfigured && !hasTeamConfigured) {
+    if (!(hasAgentConfigured || hasTeamConfigured)) {
       if (agents.length > 0) {
         const firstAgent = agents[0];
         this.configManager.updateConfig({
@@ -1427,10 +1592,28 @@ var AgnoClient = class extends EventEmitter {
     }
     return { agents, teams };
   }
+  /**
+   * Dispose of the client and clean up all resources.
+   * Call this method when the client is no longer needed to prevent memory leaks.
+   * After calling dispose(), the client instance should not be reused.
+   */
+  dispose() {
+    this.removeAllListeners();
+    this.messageStore.clear();
+    this.pendingUISpecs.clear();
+    this.eventProcessor.reset();
+    this.state.isStreaming = false;
+    this.state.isRefreshing = false;
+    this.state.isEndpointActive = false;
+    this.state.agents = [];
+    this.state.teams = [];
+    this.state.sessions = [];
+    this.state.isPaused = false;
+    this.state.pausedRunId = void 0;
+    this.state.toolsAwaitingExecution = void 0;
+    this.state.errorMessage = void 0;
+  }
 };
-
-// src/index.ts
-import { RunEvent as RunEvent2 } from "@antipopp/agno-types";
 export {
   AgnoClient,
   Logger,