@onyx.dev/onyx-database 2.1.0 → 2.3.0

package/dist/gen/cli/generate.cjs

@@ -231,7 +231,6 @@ async function resolveConfig(input) {
   const maxRetries = retryConfig.maxRetries ?? 3;
   const retryInitialDelayMs = retryConfig.initialDelayMs ?? 300;
   const missing = [];
-  if (!databaseId) missing.push("databaseId");
   if (!apiKey) missing.push("apiKey");
   if (!apiSecret) missing.push("apiSecret");
   if (missing.length) {
@@ -1521,6 +1520,27 @@ var OnyxDatabaseImpl = class {
     const { http } = await this.ensureAiClient();
     return http.request("POST", "/api/script-approvals", input);
   }
+  async predict(publishedModelId, inputs) {
+    const { http, databaseId } = await this.ensureClient();
+    const path = `/data/${encodeURIComponent(
+      databaseId
+    )}/model-builder/published-model/${encodeURIComponent(publishedModelId)}/predict`;
+    return http.request(
+      "POST",
+      path,
+      serializeDates({ inputs })
+    );
+  }
+  async predictFromScript(publishedModelId, scriptId, scriptParameters = {}) {
+    const { http, databaseId } = await this.ensureClient();
+    const path = `/data/${encodeURIComponent(
+      databaseId
+    )}/model-builder/published-model/${encodeURIComponent(publishedModelId)}/predict/script`;
+    return http.request("POST", path, {
+      scriptId,
+      scriptParameters
+    });
+  }
   from(table) {
     return new QueryBuilderImpl(this, String(table), this.defaultPartition);
   }
@@ -2353,7 +2373,7 @@ function normalizeIntrospection(raw) {
 async function fetchSchemaFromApi(config) {
   const db = onyx.init({
     baseUrl: config.baseUrl,
-    databaseId: config.databaseId,
+    ...config.databaseId ? { databaseId: config.databaseId } : {},
     apiKey: config.apiKey,
     apiSecret: config.apiSecret,
     fetch: config.fetch
@@ -2376,12 +2396,11 @@ async function generateTypes(options) {
   if (!schemaInput) {
     if (opts.source === "file") throw new Error("Failed to read schema from file");
     const cfg = await resolveConfig({});
-    if (!cfg.databaseId) {
-      throw new Error("Missing databaseId. Set ONYX_DATABASE_ID or pass to onyx.init().");
-    }
-    if (!opts.quiet)
-      process__default.default.stderr.write(`[onyx-gen] fetching schema from API for db ${cfg.databaseId}
+    if (!opts.quiet) {
+      const target = cfg.databaseId ? `db ${cfg.databaseId}` : "default database context";
+      process__default.default.stderr.write(`[onyx-gen] fetching schema from API for ${target}
 `);
+    }
     schemaInput = await fetchSchemaFromApi(cfg);
   }
   const schema = normalizeIntrospection(schemaInput);