@onyx.dev/onyx-database 1.2.0 → 2.0.1

package/dist/gen/cli/generate.cjs

@@ -10,6 +10,7 @@ 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 +70,7 @@ function readEnv(targetId) {
   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")
@@ -206,6 +208,7 @@ 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),
@@ -215,6 +218,7 @@ 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 ?? "";
@@ -251,6 +255,7 @@ async function resolveConfig(input) {
   const resolved = {
     baseUrl,
     aiBaseUrl,
+    defaultModel,
     databaseId,
     apiKey,
     apiSecret,
@@ -262,6 +267,7 @@ 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"
   };
@@ -1374,58 +1380,64 @@ var OnyxDatabaseImpl = class {
   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() {
-    if (!this.resolved) {
-      this.resolved = await this.cfgPromise;
-    }
+    const cfg = await this.resolveConfig();
     if (!this.aiHttp) {
       this.aiHttp = new HttpClient({
-        baseUrl: this.resolved.aiBaseUrl,
-        apiKey: this.resolved.apiKey,
-        apiSecret: this.resolved.apiSecret,
-        fetchImpl: this.resolved.fetch,
+        baseUrl: cfg.aiBaseUrl,
+        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.aiHttp,
-      fetchImpl: this.resolved.fetch,
-      aiBaseUrl: this.resolved.aiBaseUrl,
-      databaseId: this.resolved.databaseId
+      fetchImpl: cfg.fetch,
+      aiBaseUrl: cfg.aiBaseUrl,
+      databaseId: cfg.databaseId,
+      defaultModel: cfg.defaultModel ?? DEFAULT_AI_MODEL
     };
   }
   registerStream(handle) {
@@ -1440,6 +1452,21 @@ 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;
   }
@@ -1447,7 +1474,7 @@ var OnyxDatabaseImpl = class {
     return this.responseLoggingEnabled;
   }
   /** -------- IOnyxDatabase -------- */
-  chat() {
+  getAiChatClient() {
     return new AiChatClientImpl(
       () => this.ensureAiClient(),
       (handle) => this.registerStream(handle),
@@ -1455,6 +1482,32 @@ var OnyxDatabaseImpl = class {
       () => 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");