@cubos/agent-sdk 0.0.1148891 → 0.0.1150186

package/dist/index.js

@@ -88,186 +88,11 @@ function parseRetryAfter(header) {
   return Math.max(0, date - Date.now());
 }
 
-// src/sse.ts
-var MAX_BACKOFF_MS = 30000;
-var DEFAULT_IDLE_TIMEOUT_MS = 30000;
-
-class Fatal extends Error {
-  cause;
-  constructor(cause) {
-    super("fatal stream error");
-    this.cause = cause;
-  }
-}
-
-class SseIdleTimeout extends Error {
-  constructor(url, ms) {
-    super(`Stream ${url} sent nothing for ${ms}ms — reconnecting.`);
-    this.name = "SseIdleTimeout";
-  }
-}
-async function readSse(opts) {
-  const doFetch = opts.fetchImpl ?? globalThis.fetch;
-  const idleMs = opts.idleTimeoutMs ?? DEFAULT_IDLE_TIMEOUT_MS;
-  const cursor = opts.cursor ?? { lastEventId: opts.lastEventId };
-  let attempt = 0;
-  let refreshing = false;
-  while (!opts.signal.aborted) {
-    let madeProgress = false;
-    let retryAfterMs = null;
-    const connection = new AbortController;
-    const unlink = forward(opts.signal, connection);
-    let idleTimer;
-    let wentIdle = false;
-    const armIdle = () => {
-      if (idleMs <= 0)
-        return;
-      clearTimeout(idleTimer);
-      idleTimer = setTimeout(() => {
-        wentIdle = true;
-        connection.abort();
-      }, idleMs);
-    };
-    try {
-      const headers = {
-        ...await opts.headers?.({ forceRefresh: refreshing }),
-        Accept: "text/event-stream"
-      };
-      if (cursor.lastEventId !== undefined)
-        headers["Last-Event-ID"] = cursor.lastEventId;
-      armIdle();
-      const res = await doFetch(opts.url, { headers, signal: connection.signal });
-      if (opts.signal.aborted)
-        return;
-      if (res.ok && res.body) {
-        refreshing = false;
-        opts.onOpen?.();
-      }
-      if (!res.ok || !res.body) {
-        if (res.status === 401 && opts.headers && !refreshing) {
-          refreshing = true;
-          continue;
-        }
-        if (res.status === 429) {
-          retryAfterMs = Math.min(MAX_BACKOFF_MS, parseRetryAfter(res.headers.get("retry-after")) ?? MAX_BACKOFF_MS);
-          await raiseForStatus(res, `Could not open ${opts.url}.`);
-        }
-        if (res.status >= 400 && res.status < 500) {
-          await raiseForStatus(res, `Could not open ${opts.url}.`).catch((err) => {
-            throw new Fatal(err);
-          });
-          return;
-        }
-        throw new Error(`stream open failed with HTTP ${res.status}`);
-      }
-      for await (const frame of frames(res.body, connection.signal, armIdle)) {
-        const parsed = parseFrame(frame, opts.event);
-        if (parsed === null)
-          continue;
-        if (parsed.id !== null)
-          cursor.lastEventId = parsed.id;
-        opts.onEvent(parsed.data, parsed.event);
-        madeProgress = true;
-      }
-      if (wentIdle)
-        throw new SseIdleTimeout(opts.url, idleMs);
-    } catch (err) {
-      if (opts.signal.aborted)
-        return;
-      if (err instanceof Fatal)
-        throw err.cause;
-      opts.onError?.(wentIdle ? new SseIdleTimeout(opts.url, idleMs) : err);
-    } finally {
-      clearTimeout(idleTimer);
-      unlink();
-      connection.abort();
-      opts.onClose?.();
-    }
-    if (opts.signal.aborted)
-      return;
-    if (madeProgress)
-      attempt = 0;
-    const delayMs = retryAfterMs ?? Math.min(MAX_BACKOFF_MS, 1000 * 2 ** attempt);
-    attempt += 1;
-    await sleep(delayMs, opts.signal);
-  }
-}
-async function* frames(body, signal, onRead) {
-  const reader = body.getReader();
-  const decoder = new TextDecoder;
-  let buffer = "";
-  try {
-    while (!signal.aborted) {
-      const { value, done } = await reader.read();
-      if (done)
-        return;
-      onRead();
-      buffer += decoder.decode(value, { stream: true });
-      for (;; ) {
-        const sep = buffer.indexOf(`
-
-`);
-        if (sep === -1)
-          break;
-        yield buffer.slice(0, sep);
-        buffer = buffer.slice(sep + 2);
-      }
-    }
-  } finally {
-    reader.cancel().catch(() => {});
-  }
-}
-function parseFrame(frame, expectedEvent) {
-  let dataLine = null;
-  let id = null;
-  let eventName = "message";
-  for (const line of frame.split(`
-`)) {
-    if (line.startsWith(":"))
-      continue;
-    if (line.startsWith("data:"))
-      dataLine = line.slice(5).trimStart();
-    else if (line.startsWith("event:"))
-      eventName = line.slice(6).trim();
-    else if (line.startsWith("id:"))
-      id = line.slice(3).trim();
-  }
-  const wanted = typeof expectedEvent === "string" ? eventName === expectedEvent : expectedEvent.includes(eventName);
-  if (!wanted || dataLine === null)
-    return null;
-  try {
-    return { data: JSON.parse(dataLine), id, event: eventName };
-  } catch {
-    return null;
-  }
-}
-function sleep(ms, signal) {
-  return new Promise((resolve) => {
-    const onAbort = () => {
-      clearTimeout(timer);
-      resolve();
-    };
-    const timer = setTimeout(() => {
-      signal.removeEventListener("abort", onAbort);
-      resolve();
-    }, ms);
-    signal.addEventListener("abort", onAbort, { once: true });
-  });
-}
-function forward(outer, child) {
-  const onAbort = () => child.abort();
-  if (outer.aborted)
-    child.abort();
-  else
-    outer.addEventListener("abort", onAbort, { once: true });
-  return () => outer.removeEventListener("abort", onAbort);
-}
-
 // src/http.ts
 var DEFAULT_TIMEOUT_MS = 30000;
 var DEFAULT_MAX_RETRIES = 2;
 var MAX_RETRY_WAIT_MS = 20000;
-function sleep2(ms, signal) {
+function sleep(ms, signal) {
   return new Promise((resolve, reject) => {
     if (signal?.aborted) {
       reject(signal.reason ?? new Error("Aborted"));
@@ -371,7 +196,7 @@ function createTransport(baseUrl, auth, fetchImpl = globalThis.fetch, timeoutMs
       if (waitMs === null || waitMs > MAX_RETRY_WAIT_MS)
         return res;
       attempt += 1;
-      await sleep2(waitMs, opts.signal);
+      await sleep(waitMs, opts.signal);
     }
   }
   return {
@@ -451,6 +276,13 @@ function agentsApi(t, tenantSlug) {
       body: { role_slugs: roleSlugs },
       raw: true
     }),
+    listKnowledgeBases: (agentSlug, signal) => t.request("GET", `${agent(agentSlug)}/knowledge-bases`, {
+      signal
+    }),
+    attachKnowledgeBase: (agentSlug, kbSlug) => t.request("PUT", `${agent(agentSlug)}/knowledge-bases/${enc(kbSlug)}`, { raw: true }),
+    detachKnowledgeBase: (agentSlug, kbSlug) => t.request("DELETE", `${agent(agentSlug)}/knowledge-bases/${enc(kbSlug)}`, {
+      raw: true
+    }),
     listAutoCallTools: (agentSlug, signal) => t.request("GET", `${agent(agentSlug)}/auto-call-tools`, {
       signal
     }),
@@ -506,6 +338,181 @@ function channelsApi(t, tenantSlug) {
   };
 }
 
+// src/sse.ts
+var MAX_BACKOFF_MS = 30000;
+var DEFAULT_IDLE_TIMEOUT_MS = 30000;
+
+class Fatal extends Error {
+  cause;
+  constructor(cause) {
+    super("fatal stream error");
+    this.cause = cause;
+  }
+}
+
+class SseIdleTimeout extends Error {
+  constructor(url, ms) {
+    super(`Stream ${url} sent nothing for ${ms}ms — reconnecting.`);
+    this.name = "SseIdleTimeout";
+  }
+}
+async function readSse(opts) {
+  const doFetch = opts.fetchImpl ?? globalThis.fetch;
+  const idleMs = opts.idleTimeoutMs ?? DEFAULT_IDLE_TIMEOUT_MS;
+  const cursor = opts.cursor ?? { lastEventId: opts.lastEventId };
+  let attempt = 0;
+  let refreshing = false;
+  while (!opts.signal.aborted) {
+    let madeProgress = false;
+    let retryAfterMs = null;
+    const connection = new AbortController;
+    const unlink = forward(opts.signal, connection);
+    let idleTimer;
+    let wentIdle = false;
+    const armIdle = () => {
+      if (idleMs <= 0)
+        return;
+      clearTimeout(idleTimer);
+      idleTimer = setTimeout(() => {
+        wentIdle = true;
+        connection.abort();
+      }, idleMs);
+    };
+    try {
+      const headers = {
+        ...await opts.headers?.({ forceRefresh: refreshing }),
+        Accept: "text/event-stream"
+      };
+      if (cursor.lastEventId !== undefined)
+        headers["Last-Event-ID"] = cursor.lastEventId;
+      armIdle();
+      const res = await doFetch(opts.url, { headers, signal: connection.signal });
+      if (opts.signal.aborted)
+        return;
+      if (res.ok && res.body) {
+        refreshing = false;
+        opts.onOpen?.();
+      }
+      if (!res.ok || !res.body) {
+        if (res.status === 401 && opts.headers && !refreshing) {
+          refreshing = true;
+          continue;
+        }
+        if (res.status === 429) {
+          retryAfterMs = Math.min(MAX_BACKOFF_MS, parseRetryAfter(res.headers.get("retry-after")) ?? MAX_BACKOFF_MS);
+          await raiseForStatus(res, `Could not open ${opts.url}.`);
+        }
+        if (res.status >= 400 && res.status < 500) {
+          await raiseForStatus(res, `Could not open ${opts.url}.`).catch((err) => {
+            throw new Fatal(err);
+          });
+          return;
+        }
+        throw new Error(`stream open failed with HTTP ${res.status}`);
+      }
+      for await (const frame of frames(res.body, connection.signal, armIdle)) {
+        const parsed = parseFrame(frame, opts.event);
+        if (parsed === null)
+          continue;
+        if (parsed.id !== null)
+          cursor.lastEventId = parsed.id;
+        opts.onEvent(parsed.data, parsed.event);
+        madeProgress = true;
+      }
+      if (wentIdle)
+        throw new SseIdleTimeout(opts.url, idleMs);
+    } catch (err) {
+      if (opts.signal.aborted)
+        return;
+      if (err instanceof Fatal)
+        throw err.cause;
+      opts.onError?.(wentIdle ? new SseIdleTimeout(opts.url, idleMs) : err);
+    } finally {
+      clearTimeout(idleTimer);
+      unlink();
+      connection.abort();
+      opts.onClose?.();
+    }
+    if (opts.signal.aborted)
+      return;
+    if (madeProgress)
+      attempt = 0;
+    const delayMs = retryAfterMs ?? Math.min(MAX_BACKOFF_MS, 1000 * 2 ** attempt);
+    attempt += 1;
+    await sleep2(delayMs, opts.signal);
+  }
+}
+async function* frames(body, signal, onRead) {
+  const reader = body.getReader();
+  const decoder = new TextDecoder;
+  let buffer = "";
+  try {
+    while (!signal.aborted) {
+      const { value, done } = await reader.read();
+      if (done)
+        return;
+      onRead();
+      buffer += decoder.decode(value, { stream: true });
+      for (;; ) {
+        const sep = buffer.indexOf(`
+
+`);
+        if (sep === -1)
+          break;
+        yield buffer.slice(0, sep);
+        buffer = buffer.slice(sep + 2);
+      }
+    }
+  } finally {
+    reader.cancel().catch(() => {});
+  }
+}
+function parseFrame(frame, expectedEvent) {
+  let dataLine = null;
+  let id = null;
+  let eventName = "message";
+  for (const line of frame.split(`
+`)) {
+    if (line.startsWith(":"))
+      continue;
+    if (line.startsWith("data:"))
+      dataLine = line.slice(5).trimStart();
+    else if (line.startsWith("event:"))
+      eventName = line.slice(6).trim();
+    else if (line.startsWith("id:"))
+      id = line.slice(3).trim();
+  }
+  const wanted = typeof expectedEvent === "string" ? eventName === expectedEvent : expectedEvent.includes(eventName);
+  if (!wanted || dataLine === null)
+    return null;
+  try {
+    return { data: JSON.parse(dataLine), id, event: eventName };
+  } catch {
+    return null;
+  }
+}
+function sleep2(ms, signal) {
+  return new Promise((resolve) => {
+    const onAbort = () => {
+      clearTimeout(timer);
+      resolve();
+    };
+    const timer = setTimeout(() => {
+      signal.removeEventListener("abort", onAbort);
+      resolve();
+    }, ms);
+    signal.addEventListener("abort", onAbort, { once: true });
+  });
+}
+function forward(outer, child) {
+  const onAbort = () => child.abort();
+  if (outer.aborted)
+    child.abort();
+  else
+    outer.addEventListener("abort", onAbort, { once: true });
+  return () => outer.removeEventListener("abort", onAbort);
+}
+
 // src/client-tools.ts
 var DEFAULT_CLAIM_TTL_SECONDS = 70;
 var SUBMIT_ATTEMPTS = 5;
@@ -894,6 +901,13 @@ function sharedProvidersApi(t) {
     })
   };
 }
+function pluginsApi(t) {
+  const base = "/api/v1/plugins";
+  return {
+    list: (signal) => t.request("GET", base, { signal }),
+    setEnabled: (slug, enabled) => t.request("PATCH", `${base}/${enc(slug)}`, { body: { enabled } })
+  };
+}
 
 // src/admin/knowledge-bases.ts
 function knowledgeBasesApi(t, tenantSlug) {
@@ -1093,6 +1107,7 @@ function createAdminClient(options) {
     tenants: tenantsApi(t),
     apiKeys: apiKeysApi(t),
     sharedProviders: sharedProvidersApi(t),
+    plugins: pluginsApi(t),
     tenant(slug) {
       let scope = scopes.get(slug);
       if (!scope) {
@@ -1807,8 +1822,8 @@ class AgentClient {
       const call = toClientToolCall(raw);
       if (call)
         handlers.onClientToolCall?.(call);
-      for (const call2 of toToolCallArguments(raw)) {
-        streamedArgs.set(call2.toolCallId, call2.arguments);
+      for (const call of toToolCallArguments(raw)) {
+        streamedArgs.set(call.toolCallId, call.arguments);
       }
       const activity = toToolActivity(raw);
       if (activity)
@@ -1960,23 +1975,23 @@ function createUserClient(options) {
   return new AgentClient(options);
 }
 export {
-  serveClientTools,
-  readSse,
-  mergeVoiceMessages,
-  mergeToolActivities,
-  createUserClient,
-  createAdminClient,
-  SseIdleTimeout,
-  MemoryConversationCache,
-  DEFAULT_TIMEOUT_MS,
-  DEFAULT_MAX_RETRIES,
-  DEFAULT_IDLE_TIMEOUT_MS,
-  AgentNetworkError,
-  AgentError,
-  AgentConfigError,
+  AgentApiError,
   AgentClient,
-  AgentApiError
+  AgentConfigError,
+  AgentError,
+  AgentNetworkError,
+  DEFAULT_IDLE_TIMEOUT_MS,
+  DEFAULT_MAX_RETRIES,
+  DEFAULT_TIMEOUT_MS,
+  MemoryConversationCache,
+  SseIdleTimeout,
+  createAdminClient,
+  createUserClient,
+  mergeToolActivities,
+  mergeVoiceMessages,
+  readSse,
+  serveClientTools
 };
 
-//# debugId=F147527891A76CE064756E2164756E21
+//# debugId=A22E700F7386AA9764756E2164756E21
 //# sourceMappingURL=index.js.map