@axiom-lattice/core 2.1.14 → 2.1.15

package/dist/index.js

@@ -37,17 +37,20 @@ __export(index_exports, {
   AgentType: () => import_protocols.AgentType,
   ChunkBuffer: () => ChunkBuffer,
   ChunkBufferLatticeManager: () => ChunkBufferLatticeManager,
+  ConsoleLoggerClient: () => ConsoleLoggerClient,
   DefaultScheduleClient: () => DefaultScheduleClient,
   EmbeddingsLatticeManager: () => EmbeddingsLatticeManager,
   GraphBuildOptions: () => import_protocols.GraphBuildOptions,
   InMemoryAssistantStore: () => InMemoryAssistantStore,
   InMemoryChunkBuffer: () => InMemoryChunkBuffer,
   InMemoryThreadStore: () => InMemoryThreadStore,
+  LoggerLatticeManager: () => LoggerLatticeManager,
   MemoryLatticeManager: () => MemoryLatticeManager,
   MemoryQueueClient: () => MemoryQueueClient,
   MemoryScheduleStorage: () => MemoryScheduleStorage,
   MemoryType: () => import_protocols2.MemoryType,
   ModelLatticeManager: () => ModelLatticeManager,
+  PinoLoggerClient: () => PinoLoggerClient,
   PostgresDatabase: () => PostgresDatabase,
   Protocols: () => Protocols,
   QueueLatticeManager: () => QueueLatticeManager,
@@ -71,6 +74,7 @@ __export(index_exports, {
   getChunkBuffer: () => getChunkBuffer,
   getEmbeddingsClient: () => getEmbeddingsClient,
   getEmbeddingsLattice: () => getEmbeddingsLattice,
+  getLoggerLattice: () => getLoggerLattice,
   getModelLattice: () => getModelLattice,
   getNextCronTime: () => getNextCronTime,
   getQueueLattice: () => getQueueLattice,
@@ -83,6 +87,7 @@ __export(index_exports, {
   getVectorStoreLattice: () => getVectorStoreLattice,
   hasChunkBuffer: () => hasChunkBuffer,
   isValidCronExpression: () => isValidCronExpression,
+  loggerLatticeManager: () => loggerLatticeManager,
   modelLatticeManager: () => modelLatticeManager,
   parseCronExpression: () => parseCronExpression,
   queueLatticeManager: () => queueLatticeManager,
@@ -91,6 +96,7 @@ __export(index_exports, {
   registerCheckpointSaver: () => registerCheckpointSaver,
   registerChunkBuffer: () => registerChunkBuffer,
   registerEmbeddingsLattice: () => registerEmbeddingsLattice,
+  registerLoggerLattice: () => registerLoggerLattice,
   registerModelLattice: () => registerModelLattice,
   registerQueueLattice: () => registerQueueLattice,
   registerScheduleLattice: () => registerScheduleLattice,
@@ -5292,6 +5298,321 @@ var registerVectorStoreLattice = (key, vectorStore) => vectorStoreLatticeManager
 var getVectorStoreLattice = (key) => vectorStoreLatticeManager.getVectorStoreLattice(key);
 var getVectorStoreClient = (key) => vectorStoreLatticeManager.getVectorStoreClient(key);
 
+// src/logger_lattice/LoggerLatticeManager.ts
+var import_protocols8 = require("@axiom-lattice/protocols");
+
+// src/logger_lattice/PinoLoggerClient.ts
+var import_pino = __toESM(require("pino"));
+var import_pino_pretty = require("pino-pretty");
+var import_pino_roll = require("pino-roll");
+var PinoLoggerClient = class _PinoLoggerClient {
+  constructor(config) {
+    this.config = config;
+    this.context = config.context || {};
+    const loggerConfig = {
+      // Custom timestamp format
+      timestamp: () => `,"@timestamp":"${(/* @__PURE__ */ new Date()).toISOString()}"`,
+      // Base metadata
+      base: {
+        "@version": "1",
+        app_name: "lattice",
+        service_name: config.serviceName || "lattice-service",
+        thread_name: "main",
+        logger_name: config.loggerName || config.name || "lattice-logger"
+      },
+      formatters: {
+        level: (label, number) => {
+          return {
+            level: label.toUpperCase(),
+            level_value: number * 1e3
+          };
+        }
+      }
+    };
+    const useFileLogging = this.shouldUseFileLogging(config);
+    const fileOptions = this.getFileOptions(config);
+    if (useFileLogging && fileOptions) {
+      try {
+        this.pinoLogger = (0, import_pino.default)(
+          loggerConfig,
+          import_pino.default.transport({
+            target: "pino-roll",
+            options: {
+              file: fileOptions.file || "./logs/app",
+              frequency: fileOptions.frequency || "daily",
+              mkdir: fileOptions.mkdir !== false,
+              // Default to true
+              size: fileOptions.size,
+              maxFiles: fileOptions.maxFiles
+            }
+          })
+        );
+      } catch (error) {
+        console.error(
+          "Failed to initialize pino-roll logger, falling back to console",
+          error
+        );
+        this.pinoLogger = (0, import_pino.default)({
+          ...loggerConfig,
+          transport: {
+            target: "pino-pretty",
+            options: {
+              colorize: true
+            }
+          }
+        });
+      }
+    } else {
+      this.pinoLogger = (0, import_pino.default)({
+        ...loggerConfig,
+        transport: {
+          target: "pino-pretty",
+          options: {
+            colorize: true
+          }
+        }
+      });
+    }
+  }
+  /**
+   * Determine if file logging should be used
+   */
+  shouldUseFileLogging(config) {
+    const hasFileConfig = config.file !== void 0;
+    const isProduction = process.env.NODE_ENV === "production";
+    return hasFileConfig || isProduction;
+  }
+  /**
+   * Get file options from config
+   */
+  getFileOptions(config) {
+    if (!config.file) {
+      if (process.env.NODE_ENV === "production") {
+        return {
+          file: "./logs/app",
+          frequency: "daily",
+          mkdir: true
+        };
+      }
+      return null;
+    }
+    if (typeof config.file === "string") {
+      return {
+        file: config.file,
+        frequency: "daily",
+        mkdir: true
+      };
+    }
+    return config.file;
+  }
+  /**
+   * Get contextual logger with merged context
+   */
+  getContextualLogger(additionalContext) {
+    const contextObj = {
+      "x-user-id": this.context["x-user-id"] || "",
+      "x-tenant-id": this.context["x-tenant-id"] || "",
+      "x-request-id": this.context["x-request-id"] || "",
+      "x-thread-id": this.context["x-thread-id"] || "",
+      service_name: this.config.serviceName || "lattice-service",
+      logger_name: this.config.loggerName || this.config.name || "lattice-logger",
+      ...additionalContext
+    };
+    return this.pinoLogger.child(contextObj);
+  }
+  info(msg, obj) {
+    this.getContextualLogger(obj).info(msg);
+  }
+  error(msg, obj) {
+    this.getContextualLogger(obj).error(msg);
+  }
+  warn(msg, obj) {
+    this.getContextualLogger(obj).warn(msg);
+  }
+  debug(msg, obj) {
+    this.getContextualLogger(obj).debug(msg);
+  }
+  updateContext(context) {
+    this.context = {
+      ...this.context,
+      ...context
+    };
+  }
+  child(options) {
+    return new _PinoLoggerClient({
+      ...this.config,
+      ...options,
+      context: {
+        ...this.context,
+        ...options.context
+      }
+    });
+  }
+};
+
+// src/logger_lattice/ConsoleLoggerClient.ts
+var ConsoleLoggerClient = class _ConsoleLoggerClient {
+  constructor(config) {
+    this.config = config;
+    this.context = config.context || {};
+  }
+  formatMessage(level, msg, obj) {
+    const timestamp = (/* @__PURE__ */ new Date()).toISOString();
+    const contextStr = Object.keys(this.context).length > 0 ? ` [${JSON.stringify(this.context)}]` : "";
+    const objStr = obj ? ` ${JSON.stringify(obj)}` : "";
+    return `[${timestamp}] [${level}]${contextStr} ${msg}${objStr}`;
+  }
+  info(msg, obj) {
+    console.log(this.formatMessage("INFO", msg, obj));
+  }
+  error(msg, obj) {
+    console.error(this.formatMessage("ERROR", msg, obj));
+  }
+  warn(msg, obj) {
+    console.warn(this.formatMessage("WARN", msg, obj));
+  }
+  debug(msg, obj) {
+    console.debug(this.formatMessage("DEBUG", msg, obj));
+  }
+  updateContext(context) {
+    this.context = {
+      ...this.context,
+      ...context
+    };
+  }
+  child(options) {
+    const childClient = new _ConsoleLoggerClient({
+      ...this.config,
+      ...options,
+      context: {
+        ...this.context,
+        ...options.context
+      }
+    });
+    return childClient;
+  }
+};
+
+// src/logger_lattice/LoggerLatticeManager.ts
+var LoggerLatticeManager = class _LoggerLatticeManager extends BaseLatticeManager {
+  /**
+   * Get LoggerLatticeManager singleton instance
+   */
+  static getInstance() {
+    if (!_LoggerLatticeManager._instance) {
+      _LoggerLatticeManager._instance = new _LoggerLatticeManager();
+    }
+    return _LoggerLatticeManager._instance;
+  }
+  /**
+   * Get Lattice type prefix
+   */
+  getLatticeType() {
+    return "loggers";
+  }
+  /**
+   * Register logger Lattice
+   * @param key Lattice key name
+   * @param config Logger configuration
+   * @param client Optional logger client. If not provided, will create based on config type.
+   */
+  registerLattice(key, config, client) {
+    let loggerClient;
+    if (client) {
+      loggerClient = client;
+    } else {
+      if (config.type === import_protocols8.LoggerType.PINO) {
+        loggerClient = new PinoLoggerClient(config);
+      } else if (config.type === import_protocols8.LoggerType.CONSOLE) {
+        loggerClient = new ConsoleLoggerClient(config);
+      } else if (config.type === import_protocols8.LoggerType.CUSTOM) {
+        throw new Error(
+          `Custom logger client must be provided. Please pass the client to registerLattice.`
+        );
+      } else {
+        loggerClient = new PinoLoggerClient(config);
+      }
+    }
+    const loggerLattice = {
+      key,
+      config,
+      client: loggerClient,
+      // Logger operations
+      info: (msg, obj) => {
+        loggerClient.info(msg, obj);
+      },
+      error: (msg, obj) => {
+        loggerClient.error(msg, obj);
+      },
+      warn: (msg, obj) => {
+        loggerClient.warn(msg, obj);
+      },
+      debug: (msg, obj) => {
+        loggerClient.debug(msg, obj);
+      },
+      updateContext: loggerClient.updateContext ? (context) => {
+        loggerClient.updateContext(context);
+      } : void 0,
+      child: loggerClient.child ? (options) => {
+        return loggerClient.child(options);
+      } : void 0
+    };
+    this.register(key, loggerLattice);
+  }
+  /**
+   * Get LoggerLattice
+   * @param key Lattice key name
+   */
+  getLoggerLattice(key) {
+    const loggerLattice = this.get(key);
+    if (!loggerLattice) {
+      throw new Error(`LoggerLattice ${key} not found`);
+    }
+    return loggerLattice;
+  }
+  /**
+   * Get all Lattices
+   */
+  getAllLattices() {
+    return this.getAll();
+  }
+  /**
+   * Check if Lattice exists
+   * @param key Lattice key name
+   */
+  hasLattice(key) {
+    return this.has(key);
+  }
+  /**
+   * Remove Lattice
+   * @param key Lattice key name
+   */
+  removeLattice(key) {
+    return this.remove(key);
+  }
+  /**
+   * Clear all Lattices
+   */
+  clearLattices() {
+    this.clear();
+  }
+  /**
+   * Get Lattice count
+   */
+  getLatticeCount() {
+    return this.count();
+  }
+  /**
+   * Get Lattice key list
+   */
+  getLatticeKeys() {
+    return this.keys();
+  }
+};
+var loggerLatticeManager = LoggerLatticeManager.getInstance();
+var registerLoggerLattice = (key, config, client) => loggerLatticeManager.registerLattice(key, config, client);
+var getLoggerLattice = (key) => loggerLatticeManager.getLoggerLattice(key);
+
 // src/index.ts
 var Protocols = __toESM(require("@axiom-lattice/protocols"));
 // Annotate the CommonJS export names for ESM import in node:
@@ -5303,17 +5624,20 @@ var Protocols = __toESM(require("@axiom-lattice/protocols"));
   AgentType,
   ChunkBuffer,
   ChunkBufferLatticeManager,
+  ConsoleLoggerClient,
   DefaultScheduleClient,
   EmbeddingsLatticeManager,
   GraphBuildOptions,
   InMemoryAssistantStore,
   InMemoryChunkBuffer,
   InMemoryThreadStore,
+  LoggerLatticeManager,
   MemoryLatticeManager,
   MemoryQueueClient,
   MemoryScheduleStorage,
   MemoryType,
   ModelLatticeManager,
+  PinoLoggerClient,
   PostgresDatabase,
   Protocols,
   QueueLatticeManager,
@@ -5337,6 +5661,7 @@ var Protocols = __toESM(require("@axiom-lattice/protocols"));
   getChunkBuffer,
   getEmbeddingsClient,
   getEmbeddingsLattice,
+  getLoggerLattice,
   getModelLattice,
   getNextCronTime,
   getQueueLattice,
@@ -5349,6 +5674,7 @@ var Protocols = __toESM(require("@axiom-lattice/protocols"));
   getVectorStoreLattice,
   hasChunkBuffer,
   isValidCronExpression,
+  loggerLatticeManager,
   modelLatticeManager,
   parseCronExpression,
   queueLatticeManager,
@@ -5357,6 +5683,7 @@ var Protocols = __toESM(require("@axiom-lattice/protocols"));
   registerCheckpointSaver,
   registerChunkBuffer,
   registerEmbeddingsLattice,
+  registerLoggerLattice,
   registerModelLattice,
   registerQueueLattice,
   registerScheduleLattice,