@123toto/ai-app-assistant-server 0.2.0 → 0.2.1

package/README.md

@@ -44,6 +44,7 @@ import { createManagedAiAppAssistantServer } from "@123toto/ai-app-assistant-ser
 
 const aiAppAssistant = createManagedAiAppAssistantServer({
   configuration: {
+    enabled: process.env.AI_APP_ASSISTANT_LLM_ENABLED === "true",
     storage: { type: "redis", client: redis },
     encryptionKey: process.env.AI_APP_ASSISTANT_SECRET_ENCRYPTION_KEY!,
     defaultConfiguration: {
@@ -63,6 +64,9 @@ const aiAppAssistant = createManagedAiAppAssistantServer({
 await aiAppAssistant.initialize();
 ```
 
+`enabled: false` disables every provider operation while keeping the host
+application and the read-only administration endpoints available.
+
 Documents may also be loaded after application bootstrap with `aiAppAssistant.setDocuments(documents)`.
 
 ## Framework connectors

package/dist/express.d.cts

@@ -1,4 +1,4 @@
-import { M as ManagedAiAppAssistantServer } from './managed-server-CthTKEQ6.cjs';
+import { M as ManagedAiAppAssistantServer } from './managed-server-CQb-RK-i.cjs';
 import './ai-sdk-qU3KgNu-.cjs';
 import 'ai';
 import '@123toto/ai-app-assistant-contracts';

package/dist/express.d.ts

@@ -1,4 +1,4 @@
-import { M as ManagedAiAppAssistantServer } from './managed-server-BGQ-1jkx.js';
+import { M as ManagedAiAppAssistantServer } from './managed-server-CGmvVYsF.js';
 import './ai-sdk-qU3KgNu-.js';
 import 'ai';
 import '@123toto/ai-app-assistant-contracts';

package/dist/index.cjs

@@ -1609,12 +1609,17 @@ var AiAppAssistantConfigurationManager = class {
     this.#repository = options.repository;
     this.#quotaStore = options.quotaStore ?? createMemoryAiAppAssistantQuotaStore();
   }
+  /** Returns the current state of the host-controlled global kill switch. */
+  isEnabled() {
+    return typeof this.#options.enabled === "function" ? this.#options.enabled() : this.#options.enabled ?? true;
+  }
   /** Returns the built-in provider catalogue; no credentials are exposed. */
   listProviders() {
     return listAiProviders();
   }
   /** Discovers models with an explicit key or the currently configured secret. */
   async listModels(input) {
+    this.assertEnabled();
     const apiKey = await this.resolveApiKey(input.provider, input.apiKey);
     if (this.#options.listModels) {
       return this.#options.listModels({ ...input, ...apiKey ? { apiKey } : {} });
@@ -1640,7 +1645,10 @@ var AiAppAssistantConfigurationManager = class {
         let connected = event.connectionValidated;
         if (event.reloadRequired) {
           this.#recentConnectionValidation = void 0;
-          if (connected) {
+          if (!this.isEnabled()) {
+            connected = false;
+            this.#runtimeConnection = { status: "disabled", checkedAt: this.now() };
+          } else if (connected) {
             const configuration = await this.getRuntimeConfiguration();
             this.#runtimeConnection = configuration ? {
               status: "connected",
@@ -1670,6 +1678,7 @@ var AiAppAssistantConfigurationManager = class {
   }
   /** Tests credentials and briefly caches a successful result for the next save. */
   async testConnection(input) {
+    this.assertEnabled();
     const apiKey = await this.resolveApiKey(input.provider, input.apiKey);
     const connection = this.#options.testConnection ? await this.#options.testConnection({ ...input, ...apiKey ? { apiKey } : {} }) : await testAiSdkConnection({
       model: `${input.provider}:${input.model}`,
@@ -1698,6 +1707,10 @@ var AiAppAssistantConfigurationManager = class {
   /** Checks the effective stored/deployment connection used by live questions. */
   async validateRuntimeConnection() {
     this.#lastReconnectAttempt = Date.now();
+    if (!this.isEnabled()) {
+      this.#runtimeConnection = { status: "disabled", checkedAt: this.now() };
+      return false;
+    }
     const configuration = await this.getRuntimeConfiguration();
     if (!configuration || configuration.provider !== "ollama" && !configuration.apiKey) {
       this.#runtimeConnection = {
@@ -1740,6 +1753,7 @@ var AiAppAssistantConfigurationManager = class {
     const initialActive = this.effectiveConfiguration(initial);
     const initialApiKey = input.apiKey ?? (initialActive?.provider === input.provider ? initialActive.apiKey : void 0) ?? this.resolveDefaultApiKey(input.provider);
     const initialConnectionChanged = connectionChanged(initialActive, input);
+    if (!this.isEnabled() && initialConnectionChanged) throw disabled();
     let connection = initialConnectionChanged ? await this.validateConnectionForSave(input, initialApiKey) : this.lastKnownConnection(input.provider, input.model);
     if (initialConnectionChanged && !connection.success) {
       return { saved: false, connection, reloadRequired: false };
@@ -1968,6 +1982,7 @@ var AiAppAssistantConfigurationManager = class {
   }
   /** Combines configuration, provider health and application access rules. */
   async canUse(identity) {
+    if (!this.isEnabled()) return false;
     const configuration = await this.getRuntimeConfiguration();
     if (!configuration || configuration.provider !== "ollama" && !configuration.apiKey) return false;
     if (!await this.ensureRuntimeConnection()) return false;
@@ -1986,6 +2001,7 @@ var AiAppAssistantConfigurationManager = class {
   }
   /** Enforces access and quota immediately before any model call. */
   async assertCanAsk(identity) {
+    this.assertEnabled();
     if (!await this.canUse(identity)) {
       throw forbidden("AI assistant access is not enabled for this user");
     }
@@ -2001,6 +2017,7 @@ var AiAppAssistantConfigurationManager = class {
   }
   /** Retries a failed provider lazily, with a shared backoff across requests. */
   async ensureRuntimeConnection() {
+    if (!this.isEnabled()) return false;
     if (this.#runtimeConnection.status === "connected") return true;
     const intervalMs = Math.max(1e3, this.#options.reconnectIntervalMs ?? 3e4);
     if (Date.now() - this.#lastReconnectAttempt < intervalMs) return false;
@@ -2108,6 +2125,9 @@ var AiAppAssistantConfigurationManager = class {
   now() {
     return (this.#options.now?.() ?? /* @__PURE__ */ new Date()).toISOString();
   }
+  assertEnabled() {
+    if (!this.isEnabled()) throw disabled();
+  }
   async publishAndEmit(event) {
     await this.#options.synchronizer?.publish(event);
     await this.emit(event);
@@ -2231,6 +2251,13 @@ function configurationChanges(previous, input, allowModelChangesByOthers) {
 function forbidden(message) {
   return new AiAppAssistantManagementError(403, "forbidden", message);
 }
+function disabled() {
+  return new AiAppAssistantManagementError(
+    503,
+    "assistant_disabled",
+    "AI assistant is disabled"
+  );
+}
 
 // src/telemetry.ts
 function createMemoryAiAppAssistantTelemetryStore(options) {
@@ -2400,6 +2427,11 @@ function createManagedAiAppAssistantRuntime(options) {
     return reloadQueue;
   });
   const rebuild = async (event) => {
+    if (!options.configuration.isEnabled()) {
+      assistant = void 0;
+      activeGenerator = void 0;
+      return;
+    }
     const connected = event.connectionValidated || await options.configuration.validateRuntimeConnection();
     const configuration = await options.configuration.getRuntimeConfiguration();
     if (!connected || !configuration) {