@ragable/sdk 0.0.2 → 0.3.0

package/dist/index.js

@@ -22,11 +22,27 @@ var __publicField = (obj, key, value) => __defNormalProp(obj, typeof key !== "sy
 // src/index.ts
 var index_exports = {};
 __export(index_exports, {
+  AgentsClient: () => AgentsClient,
   Ragable: () => Ragable,
+  RagableBrowser: () => RagableBrowser,
+  RagableBrowserAgentsClient: () => RagableBrowserAgentsClient,
+  RagableBrowserAuthClient: () => RagableBrowserAuthClient,
+  RagableBrowserDatabaseClient: () => RagableBrowserDatabaseClient,
   RagableError: () => RagableError,
-  createClient: () => createClient
+  RagableRequestClient: () => RagableRequestClient,
+  ShiftClient: () => ShiftClient,
+  createBrowserClient: () => createBrowserClient,
+  createClient: () => createClient,
+  createRagPipeline: () => createRagPipeline,
+  extractErrorMessage: () => extractErrorMessage,
+  formatRetrievalContext: () => formatRetrievalContext,
+  normalizeBrowserApiBase: () => normalizeBrowserApiBase,
+  parseSseDataLine: () => parseSseDataLine,
+  readSseStream: () => readSseStream
 });
 module.exports = __toCommonJS(index_exports);
+
+// src/request-client.ts
 var RagableError = class extends Error {
   constructor(message, status, body) {
     super(message);
@@ -37,6 +53,20 @@ var RagableError = class extends Error {
     this.body = body;
   }
 };
+function extractErrorMessage(payload, fallback) {
+  if (payload && typeof payload === "object") {
+    if ("error" in payload && typeof payload.error === "string") {
+      return payload.error;
+    }
+    if ("message" in payload && typeof payload.message === "string") {
+      return payload.message;
+    }
+  }
+  if (typeof payload === "string" && payload.length > 0) {
+    return payload;
+  }
+  return fallback || "Request failed";
+}
 var RagableRequestClient = class {
   constructor(options) {
     __publicField(this, "apiKey");
@@ -48,7 +78,24 @@ var RagableRequestClient = class {
     this.fetchImpl = options.fetch ?? fetch;
     this.defaultHeaders = options.headers;
   }
+  toUrl(path) {
+    const normalizedBase = this.baseUrl.replace(/\/+$/, "");
+    const normalizedPath = path.startsWith("/") ? path : `/${path}`;
+    return `${normalizedBase}${normalizedPath}`;
+  }
   async request(path, options = {}) {
+    const response = await this.rawFetch(path, options);
+    const payload = await this.parseResponseBody(response);
+    if (!response.ok) {
+      const message = extractErrorMessage(payload, response.statusText);
+      throw new RagableError(message, response.status, payload);
+    }
+    return payload;
+  }
+  /**
+   * Low-level fetch with API key and JSON body encoding. Caller handles status and body.
+   */
+  async rawFetch(path, options = {}) {
     const headers = new Headers(this.defaultHeaders);
     headers.set("Authorization", `Bearer ${this.apiKey}`);
     let body = options.body;
@@ -56,22 +103,11 @@ var RagableRequestClient = class {
       headers.set("Content-Type", "application/json");
       body = JSON.stringify(body);
     }
-    const response = await this.fetchImpl(this.toUrl(path), {
+    return this.fetchImpl(this.toUrl(path), {
       ...options,
       headers,
       body
     });
-    const payload = await this.parseResponseBody(response);
-    if (!response.ok) {
-      const message = extractErrorMessage(payload, response.statusText);
-      throw new RagableError(message, response.status, payload);
-    }
-    return payload;
-  }
-  toUrl(path) {
-    const normalizedBase = this.baseUrl.replace(/\/+$/, "");
-    const normalizedPath = path.startsWith("/") ? path : `/${path}`;
-    return `${normalizedBase}${normalizedPath}`;
   }
   async parseResponseBody(response) {
     if (response.status === 204) {
@@ -84,6 +120,11 @@ var RagableRequestClient = class {
     return response.text();
   }
 };
+function isBodyInit(value) {
+  return typeof value === "string" || value instanceof Blob || value instanceof FormData || value instanceof URLSearchParams || value instanceof ArrayBuffer || ArrayBuffer.isView(value);
+}
+
+// src/shift.ts
 var ShiftClient = class {
   constructor(client) {
     this.client = client;
@@ -186,34 +227,6 @@ var ShiftClient = class {
     );
   }
 };
-var Ragable = class {
-  constructor(options) {
-    __publicField(this, "shift");
-    __publicField(this, "infrastructure");
-    const client = new RagableRequestClient(options);
-    this.shift = new ShiftClient(client);
-    this.infrastructure = {
-      shift: this.shift
-    };
-  }
-};
-function createClient(options) {
-  return new Ragable(options);
-}
-function extractErrorMessage(payload, fallback) {
-  if (payload && typeof payload === "object") {
-    if ("error" in payload && typeof payload.error === "string") {
-      return payload.error;
-    }
-    if ("message" in payload && typeof payload.message === "string") {
-      return payload.message;
-    }
-  }
-  if (typeof payload === "string" && payload.length > 0) {
-    return payload;
-  }
-  return fallback || "Request failed";
-}
 function resolveUploadFileName(params) {
   if (params.fileName) {
     return params.fileName;
@@ -244,18 +257,398 @@ function normalizeUploadFile(file, contentType) {
 function isNamedBlob(value) {
   return value instanceof Blob && "name" in value && typeof value.name === "string";
 }
-function isBodyInit(value) {
-  return typeof value === "string" || value instanceof Blob || value instanceof FormData || value instanceof URLSearchParams || value instanceof ArrayBuffer || ArrayBuffer.isView(value);
-}
 function toArrayBuffer(value) {
   const copy = new Uint8Array(value.byteLength);
   copy.set(value);
   return copy.buffer;
 }
+
+// src/sse.ts
+async function parseMaybeJsonBody(response) {
+  const contentType = response.headers.get("content-type") ?? "";
+  if (contentType.includes("application/json")) {
+    try {
+      return await response.json();
+    } catch {
+      return null;
+    }
+  }
+  try {
+    return await response.text();
+  } catch {
+    return null;
+  }
+}
+function parseSseDataLine(line) {
+  const dataPrefix = "data: ";
+  if (!line.startsWith(dataPrefix)) {
+    return null;
+  }
+  const json = line.slice(dataPrefix.length).trim();
+  if (json.length === 0 || json === "[DONE]") {
+    return null;
+  }
+  try {
+    return JSON.parse(json);
+  } catch {
+    return null;
+  }
+}
+async function* readSseStream(body) {
+  const reader = body.getReader();
+  const decoder = new TextDecoder();
+  let buffer = "";
+  try {
+    while (true) {
+      const { done, value } = await reader.read();
+      if (done) {
+        break;
+      }
+      buffer += decoder.decode(value, { stream: true });
+      let boundary = buffer.indexOf("\n\n");
+      while (boundary !== -1) {
+        const block = buffer.slice(0, boundary);
+        buffer = buffer.slice(boundary + 2);
+        for (const line of block.split("\n")) {
+          const evt = parseSseDataLine(line);
+          if (evt) {
+            yield evt;
+          }
+        }
+        boundary = buffer.indexOf("\n\n");
+      }
+    }
+    if (buffer.trim().length > 0) {
+      for (const line of buffer.split("\n")) {
+        const evt = parseSseDataLine(line);
+        if (evt) {
+          yield evt;
+        }
+      }
+    }
+  } finally {
+    reader.releaseLock();
+  }
+}
+
+// src/agents.ts
+var AgentsClient = class {
+  constructor(client) {
+    this.client = client;
+  }
+  async list() {
+    return this.client.request("/v1/agents");
+  }
+  async get(agentId) {
+    return this.client.request(`/v1/agents/${agentId}`);
+  }
+  async chat(agentId, params) {
+    return this.client.request(`/v1/agents/${agentId}/chat`, {
+      method: "POST",
+      body: {
+        message: params.message,
+        ...params.history !== void 0 ? { history: params.history } : {}
+      }
+    });
+  }
+  /**
+   * Stream agent execution as SSE (`data: {json}` lines). Yields parsed JSON objects.
+   */
+  async *chatStream(agentId, params) {
+    const response = await this.client.rawFetch(
+      `/v1/agents/${agentId}/chat/stream`,
+      {
+        method: "POST",
+        body: {
+          message: params.message,
+          ...params.history !== void 0 ? { history: params.history } : {}
+        }
+      }
+    );
+    if (!response.ok) {
+      const payload = await parseMaybeJsonBody(response);
+      const message = extractErrorMessage(payload, response.statusText);
+      throw new RagableError(message, response.status, payload);
+    }
+    const body = response.body;
+    if (!body) {
+      return;
+    }
+    yield* readSseStream(body);
+  }
+};
+
+// src/browser.ts
+function normalizeBrowserApiBase(baseUrl) {
+  return (baseUrl ?? "http://localhost:8080/api").replace(/\/+$/, "");
+}
+function requireAuthGroupId(options) {
+  const id = options.authGroupId?.trim();
+  if (!id) {
+    throw new Error(
+      "authGroupId is required for auth and database methods on the browser client"
+    );
+  }
+  return id;
+}
+async function requireAccessToken(options) {
+  const getter = options.getAccessToken;
+  if (!getter) {
+    throw new Error(
+      "getAccessToken is required for this call (return the end-user access JWT)"
+    );
+  }
+  const token = await getter();
+  if (!token?.trim()) {
+    throw new Error("getAccessToken returned no token");
+  }
+  return token.trim();
+}
+async function parseJsonOrThrow(response) {
+  const payload = await parseMaybeJsonBody(response);
+  if (!response.ok) {
+    const message = extractErrorMessage(payload, response.statusText);
+    throw new RagableError(message, response.status, payload);
+  }
+  return payload;
+}
+var RagableBrowserAuthClient = class {
+  constructor(options) {
+    this.options = options;
+  }
+  get fetchImpl() {
+    return this.options.fetch ?? fetch;
+  }
+  toUrl(path) {
+    return `${normalizeBrowserApiBase(this.options.baseUrl)}${path.startsWith("/") ? path : `/${path}`}`;
+  }
+  baseHeaders(json) {
+    const h = new Headers(this.options.headers);
+    if (json) {
+      h.set("Content-Type", "application/json");
+    }
+    return h;
+  }
+  authPrefix() {
+    const gid = requireAuthGroupId(this.options);
+    return `/auth-groups/${gid}/auth`;
+  }
+  async register(body) {
+    const response = await this.fetchImpl(`${this.toUrl(this.authPrefix())}/register`, {
+      method: "POST",
+      headers: this.baseHeaders(true),
+      body: JSON.stringify(body)
+    });
+    return parseJsonOrThrow(response);
+  }
+  async login(body) {
+    const response = await this.fetchImpl(`${this.toUrl(this.authPrefix())}/login`, {
+      method: "POST",
+      headers: this.baseHeaders(true),
+      body: JSON.stringify(body)
+    });
+    return parseJsonOrThrow(response);
+  }
+  async refresh(body) {
+    const response = await this.fetchImpl(`${this.toUrl(this.authPrefix())}/refresh`, {
+      method: "POST",
+      headers: this.baseHeaders(true),
+      body: JSON.stringify(body)
+    });
+    return parseJsonOrThrow(response);
+  }
+  async getMe() {
+    const token = await requireAccessToken(this.options);
+    const headers = this.baseHeaders(false);
+    headers.set("Authorization", `Bearer ${token}`);
+    const response = await this.fetchImpl(`${this.toUrl(this.authPrefix())}/me`, {
+      method: "GET",
+      headers
+    });
+    return parseJsonOrThrow(response);
+  }
+  async updateMe(body) {
+    const token = await requireAccessToken(this.options);
+    const headers = this.baseHeaders(true);
+    headers.set("Authorization", `Bearer ${token}`);
+    const response = await this.fetchImpl(`${this.toUrl(this.authPrefix())}/me`, {
+      method: "PATCH",
+      headers,
+      body: JSON.stringify(body)
+    });
+    return parseJsonOrThrow(response);
+  }
+};
+var RagableBrowserDatabaseClient = class {
+  constructor(options) {
+    this.options = options;
+  }
+  get fetchImpl() {
+    return this.options.fetch ?? fetch;
+  }
+  toUrl(path) {
+    return `${normalizeBrowserApiBase(this.options.baseUrl)}${path.startsWith("/") ? path : `/${path}`}`;
+  }
+  async query(params) {
+    const gid = requireAuthGroupId(this.options);
+    const token = await requireAccessToken(this.options);
+    const headers = this.baseHeaders();
+    headers.set("Authorization", `Bearer ${token}`);
+    headers.set("Content-Type", "application/json");
+    const response = await this.fetchImpl(
+      this.toUrl(`/auth-groups/${gid}/data/query`),
+      {
+        method: "POST",
+        headers,
+        body: JSON.stringify({
+          databaseInstanceId: params.databaseInstanceId,
+          sql: params.sql,
+          ...params.params !== void 0 ? { params: params.params } : {},
+          ...params.timeoutMs !== void 0 ? { timeoutMs: params.timeoutMs } : {},
+          ...params.rowLimit !== void 0 ? { rowLimit: params.rowLimit } : {}
+        })
+      }
+    );
+    return parseJsonOrThrow(response);
+  }
+  baseHeaders() {
+    return new Headers(this.options.headers);
+  }
+};
+var RagableBrowserAgentsClient = class {
+  constructor(options) {
+    this.options = options;
+  }
+  get fetchImpl() {
+    return this.options.fetch ?? fetch;
+  }
+  toUrl(path) {
+    return `${normalizeBrowserApiBase(this.options.baseUrl)}${path.startsWith("/") ? path : `/${path}`}`;
+  }
+  /**
+   * Stream agent execution as SSE (`POST /public/organizations/:orgId/agents/:agentId/chat/stream`).
+   */
+  async *chatStream(agentId, params) {
+    const orgId = this.options.organizationId;
+    const body = {
+      message: params.message,
+      ...params.history !== void 0 ? { history: params.history } : {},
+      ...params.triggerSubtype !== void 0 ? { triggerSubtype: params.triggerSubtype } : {},
+      ...params.triggerNodeId !== void 0 ? { triggerNodeId: params.triggerNodeId } : {}
+    };
+    const headers = new Headers(this.options.headers);
+    headers.set("Content-Type", "application/json");
+    const response = await this.fetchImpl(
+      this.toUrl(`/public/organizations/${orgId}/agents/${agentId}/chat/stream`),
+      {
+        method: "POST",
+        headers,
+        body: JSON.stringify(body)
+      }
+    );
+    if (!response.ok) {
+      const payload = await parseMaybeJsonBody(response);
+      const message = extractErrorMessage(payload, response.statusText);
+      throw new RagableError(message, response.status, payload);
+    }
+    const streamBody = response.body;
+    if (!streamBody) {
+      return;
+    }
+    yield* readSseStream(streamBody);
+  }
+};
+var RagableBrowser = class {
+  constructor(options) {
+    __publicField(this, "agents");
+    __publicField(this, "auth");
+    __publicField(this, "database");
+    this.agents = new RagableBrowserAgentsClient(options);
+    this.auth = new RagableBrowserAuthClient(options);
+    this.database = new RagableBrowserDatabaseClient(options);
+  }
+};
+function createBrowserClient(options) {
+  return new RagableBrowser(options);
+}
+
+// src/rag.ts
+function formatRetrievalContext(results, options = {}) {
+  const header = options.header ?? "Relevant passages:\n";
+  const separator = options.separator ?? "\n\n";
+  const minScore = options.minScore;
+  let filtered = results;
+  if (typeof minScore === "number") {
+    filtered = results.filter((r) => r.score >= minScore);
+  }
+  const blocks = filtered.map((r, i) => {
+    const scoreSuffix = options.includeScores === true ? ` (score: ${r.score.toFixed(4)})` : "";
+    return `[${i + 1}]${scoreSuffix}
+${r.content.trim()}`;
+  });
+  let text = (blocks.length > 0 ? header : "") + blocks.join(separator);
+  if (typeof options.maxChars === "number" && text.length > options.maxChars) {
+    text = text.slice(0, options.maxChars).trimEnd() + "\n\u2026";
+  }
+  return text;
+}
+function createRagPipeline(client, options) {
+  const { indexId } = options;
+  return {
+    indexId,
+    ingestText(params) {
+      return client.shift.documents.create(indexId, params);
+    },
+    ingestFile(params) {
+      return client.shift.documents.upload(indexId, params);
+    },
+    search(params) {
+      return client.shift.search(indexId, params);
+    },
+    async retrieve(params) {
+      const { format: formatOpts, ...searchParams } = params;
+      const { results } = await client.shift.search(indexId, searchParams);
+      const context = formatRetrievalContext(results, formatOpts ?? {});
+      return { results, context };
+    }
+  };
+}
+
+// src/index.ts
+var Ragable = class {
+  constructor(options) {
+    __publicField(this, "shift");
+    __publicField(this, "agents");
+    __publicField(this, "infrastructure");
+    const client = new RagableRequestClient(options);
+    this.shift = new ShiftClient(client);
+    this.agents = new AgentsClient(client);
+    this.infrastructure = {
+      shift: this.shift
+    };
+  }
+};
+function createClient(options) {
+  return new Ragable(options);
+}
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  AgentsClient,
   Ragable,
+  RagableBrowser,
+  RagableBrowserAgentsClient,
+  RagableBrowserAuthClient,
+  RagableBrowserDatabaseClient,
   RagableError,
-  createClient
+  RagableRequestClient,
+  ShiftClient,
+  createBrowserClient,
+  createClient,
+  createRagPipeline,
+  extractErrorMessage,
+  formatRetrievalContext,
+  normalizeBrowserApiBase,
+  parseSseDataLine,
+  readSseStream
 });
 //# sourceMappingURL=index.js.map