@onyx.dev/onyx-database 1.1.0 → 1.2.0

package/dist/edge.cjs

@@ -2,6 +2,7 @@
 
 // src/config/defaults.ts
 var DEFAULT_BASE_URL = "https://api.onyx.dev";
+var DEFAULT_AI_BASE_URL = "https://ai.onyx.dev";
 var sanitizeBaseUrl = (u) => u.replace(/\/+$/, "");
 
 // src/errors/config-error.ts
@@ -53,6 +54,7 @@ function readEnv(targetId) {
   if (targetId && envId !== targetId) return {};
   const res = dropUndefined({
     baseUrl: pick("ONYX_DATABASE_BASE_URL"),
+    aiBaseUrl: pick("ONYX_AI_BASE_URL"),
     databaseId: envId,
     apiKey: pick("ONYX_DATABASE_API_KEY"),
     apiSecret: pick("ONYX_DATABASE_API_SECRET")
@@ -71,11 +73,13 @@ async function resolveConfig(input) {
   const env = readEnv(input?.databaseId);
   const merged = {
     baseUrl: DEFAULT_BASE_URL,
+    aiBaseUrl: DEFAULT_AI_BASE_URL,
     ...dropUndefined(env),
     ...dropUndefined(input)
   };
   dbg("merged (pre-validate):", mask(merged));
   const baseUrl = sanitizeBaseUrl(merged.baseUrl ?? DEFAULT_BASE_URL);
+  const aiBaseUrl = sanitizeBaseUrl(merged.aiBaseUrl ?? DEFAULT_AI_BASE_URL);
   const databaseId = merged.databaseId ?? "";
   const apiKey = merged.apiKey ?? "";
   const apiSecret = merged.apiSecret ?? "";
@@ -100,6 +104,7 @@ async function resolveConfig(input) {
   }
   const resolved = {
     baseUrl,
+    aiBaseUrl,
     databaseId,
     apiKey,
     apiSecret,
@@ -1120,6 +1125,7 @@ var OnyxDatabaseImpl = class {
   cfgPromise;
   resolved = null;
   http = null;
+  aiHttp = null;
   streams = /* @__PURE__ */ new Set();
   requestLoggingEnabled;
   responseLoggingEnabled;
@@ -1154,6 +1160,30 @@ var OnyxDatabaseImpl = class {
       databaseId: this.resolved.databaseId
     };
   }
+  async ensureAiClient() {
+    if (!this.resolved) {
+      this.resolved = await this.cfgPromise;
+    }
+    if (!this.aiHttp) {
+      this.aiHttp = new HttpClient({
+        baseUrl: this.resolved.aiBaseUrl,
+        apiKey: this.resolved.apiKey,
+        apiSecret: this.resolved.apiSecret,
+        fetchImpl: this.resolved.fetch,
+        requestLoggingEnabled: this.requestLoggingEnabled,
+        responseLoggingEnabled: this.responseLoggingEnabled,
+        retryEnabled: this.resolved.retryEnabled,
+        maxRetries: this.resolved.maxRetries,
+        retryInitialDelayMs: this.resolved.retryInitialDelayMs
+      });
+    }
+    return {
+      http: this.aiHttp,
+      fetchImpl: this.resolved.fetch,
+      aiBaseUrl: this.resolved.aiBaseUrl,
+      databaseId: this.resolved.databaseId
+    };
+  }
   registerStream(handle) {
     this.streams.add(handle);
     return {
@@ -1166,7 +1196,34 @@ var OnyxDatabaseImpl = class {
       }
     };
   }
+  getRequestLoggingEnabled() {
+    return this.requestLoggingEnabled;
+  }
+  getResponseLoggingEnabled() {
+    return this.responseLoggingEnabled;
+  }
   /** -------- IOnyxDatabase -------- */
+  chat() {
+    return new AiChatClientImpl(
+      () => this.ensureAiClient(),
+      (handle) => this.registerStream(handle),
+      () => this.requestLoggingEnabled,
+      () => this.responseLoggingEnabled
+    );
+  }
+  async getModels() {
+    const { http } = await this.ensureAiClient();
+    return http.request("GET", "/v1/models");
+  }
+  async getModel(modelId) {
+    const { http } = await this.ensureAiClient();
+    const path = `/v1/models/${encodeURIComponent(modelId)}`;
+    return http.request("GET", path);
+  }
+  async requestScriptApproval(input) {
+    const { http } = await this.ensureAiClient();
+    return http.request("POST", "/api/script-approvals", input);
+  }
   from(table) {
     return new QueryBuilderImpl(this, String(table), this.defaultPartition);
   }
@@ -1716,6 +1773,153 @@ var CascadeBuilderImpl = class {
     return this.db.delete(table, primaryKey, opts);
   }
 };
+var AiChatClientImpl = class {
+  constructor(resolveAiClient, registerStream, requestLoggingEnabled, responseLoggingEnabled) {
+    this.resolveAiClient = resolveAiClient;
+    this.registerStream = registerStream;
+    this.requestLoggingEnabled = requestLoggingEnabled;
+    this.responseLoggingEnabled = responseLoggingEnabled;
+  }
+  normalizeEventData(rawEvent) {
+    const lines = rawEvent.split("\n");
+    const dataLines = [];
+    for (const line of lines) {
+      const trimmed = line.trim();
+      if (!trimmed || trimmed.startsWith(":")) continue;
+      if (trimmed.startsWith("data:")) {
+        dataLines.push(trimmed.slice(5).trim());
+      } else {
+        dataLines.push(trimmed);
+      }
+    }
+    if (!dataLines.length) return null;
+    const joined = dataLines.join("\n").trim();
+    return joined.length > 0 ? joined : null;
+  }
+  logRequest(url, body, headers) {
+    if (!this.requestLoggingEnabled()) return;
+    console.log(`POST ${url}`);
+    if (body != null) {
+      const serialized = typeof body === "string" ? body : JSON.stringify(body);
+      console.log(serialized);
+    }
+    const headerLog = { ...headers };
+    if (headerLog["x-onyx-secret"]) headerLog["x-onyx-secret"] = "[REDACTED]";
+    console.log("Headers:", headerLog);
+  }
+  logResponse(status, statusText, raw) {
+    if (!this.responseLoggingEnabled()) return;
+    const statusLine = `${status} ${statusText}`.trim();
+    console.log(statusLine);
+    if (raw && raw.trim().length > 0) {
+      console.log(raw);
+    }
+  }
+  toOnyxError(status, statusText, raw) {
+    let parsed = raw;
+    try {
+      parsed = parseJsonAllowNaN(raw);
+    } catch {
+    }
+    const message = typeof parsed === "object" && parsed !== null && "error" in parsed && typeof parsed.error === "object" && typeof parsed.error?.message === "string" ? String(parsed.error.message) : `${status} ${statusText}`;
+    return new OnyxHttpError(message, status, statusText, parsed, raw);
+  }
+  async create(request, options) {
+    const body = { ...request, stream: !!request.stream };
+    const { http, fetchImpl, aiBaseUrl, databaseId } = await this.resolveAiClient();
+    const params = new URLSearchParams();
+    const scopedDb = options?.databaseId ?? databaseId;
+    if (scopedDb) params.append("databaseId", scopedDb);
+    const path = `/v1/chat/completions${params.size ? `?${params.toString()}` : ""}`;
+    if (!body.stream) {
+      return http.request("POST", path, body);
+    }
+    const url = `${aiBaseUrl}${path}`;
+    const headers = http.headers({ Accept: "text/event-stream" });
+    this.logRequest(url, body, headers);
+    const res = await fetchImpl(url, {
+      method: "POST",
+      headers,
+      body: JSON.stringify(body)
+    });
+    if (!res.ok) {
+      const raw = await res.text();
+      this.logResponse(res.status, res.statusText, raw);
+      throw this.toOnyxError(res.status, res.statusText, raw);
+    }
+    this.logResponse(res.status, res.statusText);
+    const bodyStream = res.body;
+    const reader = bodyStream && typeof bodyStream.getReader === "function" ? bodyStream.getReader() : null;
+    if (!reader) {
+      throw new OnyxHttpError(
+        "Streaming response body is not readable",
+        res.status,
+        res.statusText,
+        null,
+        ""
+      );
+    }
+    let canceled = false;
+    const handle = {
+      cancel: () => {
+        if (canceled) return;
+        canceled = true;
+        try {
+          reader.cancel?.();
+        } catch {
+        }
+      }
+    };
+    const registered = this.registerStream(handle);
+    const normalizeEvent = (rawEvent) => this.normalizeEventData(rawEvent);
+    const stream = {
+      async *[Symbol.asyncIterator]() {
+        const decoder = new TextDecoder("utf-8");
+        let buffer = "";
+        try {
+          while (!canceled) {
+            const { done, value } = await reader.read();
+            if (done) break;
+            if (value) {
+              buffer += decoder.decode(value, { stream: true });
+            }
+            let splitIndex = buffer.indexOf("\n\n");
+            while (splitIndex !== -1) {
+              const rawEvent = buffer.slice(0, splitIndex);
+              buffer = buffer.slice(splitIndex + 2);
+              const data = normalizeEvent(rawEvent);
+              if (data === "[DONE]") return;
+              if (data) {
+                try {
+                  yield parseJsonAllowNaN(data);
+                } catch {
+                }
+              }
+              splitIndex = buffer.indexOf("\n\n");
+            }
+          }
+          buffer += decoder.decode();
+          const remaining = buffer.trim();
+          if (remaining && remaining !== "[DONE]") {
+            const data = normalizeEvent(remaining);
+            if (data && data !== "[DONE]") {
+              try {
+                yield parseJsonAllowNaN(data);
+              } catch {
+              }
+            }
+          }
+        } finally {
+          registered.cancel();
+        }
+      },
+      cancel() {
+        registered.cancel();
+      }
+    };
+    return stream;
+  }
+};
 function createOnyxFacade(resolveConfig2) {
   let cachedCfg = null;
   function resolveConfigWithCache(config) {