@onyx.dev/onyx-database 1.1.0 → 2.0.0

package/dist/schema/cli/schema.cjs

@@ -11,6 +11,8 @@ var process__default = /*#__PURE__*/_interopDefault(process);
 
 // src/config/defaults.ts
 var DEFAULT_BASE_URL = "https://api.onyx.dev";
+var DEFAULT_AI_BASE_URL = "https://ai.onyx.dev";
+var DEFAULT_AI_MODEL = "onyx";
 var sanitizeBaseUrl = (u) => u.replace(/\/+$/, "");
 
 // src/errors/config-error.ts
@@ -69,6 +71,8 @@ function readEnv(targetId) {
   if (targetId && envId !== targetId) return {};
   const res = dropUndefined({
     baseUrl: pick("ONYX_DATABASE_BASE_URL"),
+    aiBaseUrl: pick("ONYX_AI_BASE_URL"),
+    defaultModel: pick("ONYX_DEFAULT_MODEL"),
     databaseId: envId,
     apiKey: pick("ONYX_DATABASE_API_KEY"),
     apiSecret: pick("ONYX_DATABASE_API_SECRET")
@@ -205,6 +209,8 @@ async function resolveConfig(input) {
   }
   const merged = {
     baseUrl: DEFAULT_BASE_URL,
+    aiBaseUrl: DEFAULT_AI_BASE_URL,
+    defaultModel: DEFAULT_AI_MODEL,
     ...dropUndefined(home),
     ...dropUndefined(project),
     ...dropUndefined(cfgPath),
@@ -213,6 +219,8 @@ async function resolveConfig(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 defaultModel = typeof merged.defaultModel === "string" && merged.defaultModel.trim() ? merged.defaultModel.trim() : DEFAULT_AI_MODEL;
   const databaseId = merged.databaseId ?? "";
   const apiKey = merged.apiKey ?? "";
   const apiSecret = merged.apiSecret ?? "";
@@ -248,6 +256,8 @@ async function resolveConfig(input) {
   }
   const resolved = {
     baseUrl,
+    aiBaseUrl,
+    defaultModel,
     databaseId,
     apiKey,
     apiSecret,
@@ -258,6 +268,8 @@ async function resolveConfig(input) {
   };
   const source = {
     databaseId: input?.databaseId ? "explicit config" : env.databaseId ? "env" : cfgPath.databaseId ? "env ONYX_CONFIG_PATH" : project.databaseId ? "project file" : home.databaseId ? "home profile" : "unknown",
+    aiBaseUrl: input?.aiBaseUrl ? "explicit config" : env.aiBaseUrl ? "env" : cfgPath.aiBaseUrl ? "env ONYX_CONFIG_PATH" : project.aiBaseUrl ? "project file" : home.aiBaseUrl ? "home profile" : "default",
+    defaultModel: input?.defaultModel ? "explicit config" : env.defaultModel ? "env" : cfgPath.defaultModel ? "env ONYX_CONFIG_PATH" : project.defaultModel ? "project file" : home.defaultModel ? "home profile" : "default",
     apiKey: input?.apiKey ? "explicit config" : env.apiKey ? "env" : cfgPath.apiKey ? "env ONYX_CONFIG_PATH" : project.apiKey ? "project file" : home.apiKey ? "home profile" : "unknown",
     apiSecret: input?.apiSecret ? "explicit config" : env.apiSecret ? "env" : cfgPath.apiSecret ? "env ONYX_CONFIG_PATH" : project.apiSecret ? "project file" : home.apiSecret ? "home profile" : "unknown"
   };
@@ -277,6 +289,8 @@ async function resolveConfigWithSource(input) {
   const base = await resolveConfig(input);
   const sources = {
     baseUrl: input?.baseUrl ? "explicit config" : env.baseUrl ? "env" : cfgPath.baseUrl ? "env ONYX_CONFIG_PATH" : project.baseUrl ? "project file" : home.baseUrl ? "home profile" : "default",
+    aiBaseUrl: input?.aiBaseUrl ? "explicit config" : env.aiBaseUrl ? "env" : cfgPath.aiBaseUrl ? "env ONYX_CONFIG_PATH" : project.aiBaseUrl ? "project file" : home.aiBaseUrl ? "home profile" : "default",
+    defaultModel: input?.defaultModel ? "explicit config" : env.defaultModel ? "env" : cfgPath.defaultModel ? "env ONYX_CONFIG_PATH" : project.defaultModel ? "project file" : home.defaultModel ? "home profile" : "default",
     databaseId: input?.databaseId ? "explicit config" : env.databaseId ? "env" : cfgPath.databaseId ? "env ONYX_CONFIG_PATH" : project.databaseId ? "project file" : home.databaseId ? "home profile" : "unknown",
     apiKey: input?.apiKey ? "explicit config" : env.apiKey ? "env" : cfgPath.apiKey ? "env ONYX_CONFIG_PATH" : project.apiKey ? "project file" : home.apiKey ? "home profile" : "unknown",
     apiSecret: input?.apiSecret ? "explicit config" : env.apiSecret ? "env" : cfgPath.apiSecret ? "env ONYX_CONFIG_PATH" : project.apiSecret ? "project file" : home.apiSecret ? "home profile" : "unknown",
@@ -1399,38 +1413,69 @@ var OnyxDatabaseImpl = class {
   cfgPromise;
   resolved = null;
   http = null;
+  aiHttp = null;
   streams = /* @__PURE__ */ new Set();
   requestLoggingEnabled;
   responseLoggingEnabled;
   defaultPartition;
+  ai;
   constructor(config, resolveConfigWithCache) {
     this.requestLoggingEnabled = !!config?.requestLoggingEnabled;
     this.responseLoggingEnabled = !!config?.responseLoggingEnabled;
     this.defaultPartition = config?.partition;
     this.cfgPromise = resolveConfigWithCache(config);
+    this.ai = this.createAiFacade();
   }
-  async ensureClient() {
+  async resolveConfig() {
     if (!this.resolved) {
       this.resolved = await this.cfgPromise;
     }
+    return this.resolved;
+  }
+  async ensureClient() {
+    const cfg = await this.resolveConfig();
     if (!this.http) {
       this.http = new HttpClient({
-        baseUrl: this.resolved.baseUrl,
-        apiKey: this.resolved.apiKey,
-        apiSecret: this.resolved.apiSecret,
-        fetchImpl: this.resolved.fetch,
+        baseUrl: cfg.baseUrl,
+        apiKey: cfg.apiKey,
+        apiSecret: cfg.apiSecret,
+        fetchImpl: cfg.fetch,
         requestLoggingEnabled: this.requestLoggingEnabled,
         responseLoggingEnabled: this.responseLoggingEnabled,
-        retryEnabled: this.resolved.retryEnabled,
-        maxRetries: this.resolved.maxRetries,
-        retryInitialDelayMs: this.resolved.retryInitialDelayMs
+        retryEnabled: cfg.retryEnabled,
+        maxRetries: cfg.maxRetries,
+        retryInitialDelayMs: cfg.retryInitialDelayMs
       });
     }
     return {
       http: this.http,
-      fetchImpl: this.resolved.fetch,
-      baseUrl: this.resolved.baseUrl,
-      databaseId: this.resolved.databaseId
+      fetchImpl: cfg.fetch,
+      baseUrl: cfg.baseUrl,
+      databaseId: cfg.databaseId,
+      defaultModel: cfg.defaultModel ?? DEFAULT_AI_MODEL
+    };
+  }
+  async ensureAiClient() {
+    const cfg = await this.resolveConfig();
+    if (!this.aiHttp) {
+      this.aiHttp = new HttpClient({
+        baseUrl: cfg.aiBaseUrl,
+        apiKey: cfg.apiKey,
+        apiSecret: cfg.apiSecret,
+        fetchImpl: cfg.fetch,
+        requestLoggingEnabled: this.requestLoggingEnabled,
+        responseLoggingEnabled: this.responseLoggingEnabled,
+        retryEnabled: cfg.retryEnabled,
+        maxRetries: cfg.maxRetries,
+        retryInitialDelayMs: cfg.retryInitialDelayMs
+      });
+    }
+    return {
+      http: this.aiHttp,
+      fetchImpl: cfg.fetch,
+      aiBaseUrl: cfg.aiBaseUrl,
+      databaseId: cfg.databaseId,
+      defaultModel: cfg.defaultModel ?? DEFAULT_AI_MODEL
     };
   }
   registerStream(handle) {
@@ -1445,7 +1490,75 @@ var OnyxDatabaseImpl = class {
       }
     };
   }
+  createAiFacade() {
+    const chat = ((contentOrRequest, options) => {
+      if (typeof contentOrRequest === "string") {
+        return this.chatWithContent(contentOrRequest, options);
+      }
+      return this.getAiChatClient().create(contentOrRequest, options);
+    });
+    return {
+      chat,
+      chatClient: () => this.getAiChatClient(),
+      getModels: () => this.getModels(),
+      getModel: (modelId) => this.getModel(modelId),
+      requestScriptApproval: (input) => this.requestScriptApproval(input)
+    };
+  }
+  getRequestLoggingEnabled() {
+    return this.requestLoggingEnabled;
+  }
+  getResponseLoggingEnabled() {
+    return this.responseLoggingEnabled;
+  }
   /** -------- IOnyxDatabase -------- */
+  getAiChatClient() {
+    return new AiChatClientImpl(
+      () => this.ensureAiClient(),
+      (handle) => this.registerStream(handle),
+      () => this.requestLoggingEnabled,
+      () => this.responseLoggingEnabled
+    );
+  }
+  async chatWithContent(content, options) {
+    const { defaultModel } = await this.ensureAiClient();
+    const stream = options?.stream ?? false;
+    const request = {
+      model: options?.model ?? defaultModel,
+      messages: [{ role: options?.role ?? "user", content }],
+      stream
+    };
+    if (options && "temperature" in options) {
+      request.temperature = options.temperature ?? null;
+    }
+    const result = await this.getAiChatClient().create(request, options);
+    if (stream) return result;
+    if (options?.raw) return result;
+    const first = result.choices?.[0]?.message?.content;
+    if (typeof first === "string" && first.trim().length > 0) {
+      return first;
+    }
+    throw new Error("Chat completion response is missing message content");
+  }
+  chat(content, options) {
+    if (typeof content === "string") {
+      return this.chatWithContent(content, options);
+    }
+    return this.getAiChatClient();
+  }
+  async getModels() {
+    const { http } = await this.ensureAiClient();
+    return http.request("GET", "/v1/models");
+  }
+  async getModel(modelId) {
+    const { http } = await this.ensureAiClient();
+    const path2 = `/v1/models/${encodeURIComponent(modelId)}`;
+    return http.request("GET", path2);
+  }
+  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);
   }
@@ -1995,6 +2108,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 path2 = `/v1/chat/completions${params.size ? `?${params.toString()}` : ""}`;
+    if (!body.stream) {
+      return http.request("POST", path2, body);
+    }
+    const url = `${aiBaseUrl}${path2}`;
+    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) {