chore: controllable logger, tracer observe fix

Author: abhishekg999

src/env.ts

@@ -27,3 +27,4 @@ export const JUDGMENT_ENABLE_EVALUATIONS = getEnvVar(
   "true",
 );
 export const JUDGMENT_NO_COLOR = getEnvVar("JUDGMENT_NO_COLOR");
+export const JUDGMENT_LOG_LEVEL = getEnvVar("JUDGMENT_LOG_LEVEL", "warn");

src/tracer/NodeTracer.ts

@@ -1,5 +1,6 @@
 import { resourceFromAttributes } from "@opentelemetry/resources";
 import { NodeSDK, NodeSDKConfiguration } from "@opentelemetry/sdk-node";
+import { Logger } from "../utils/logger";
 import { VERSION } from "../version";
 import { OpenTelemetryKeys } from "./OpenTelemetryKeys";
 import { Tracer, TracerInitializeOptions } from "./Tracer";
@@ -13,7 +14,7 @@ export class NodeTracer extends Tracer {
   private nodeSDK?: NodeSDK;
 
   public async initialize(
-    options: NodeTracerInitializeOptions = {}
+    options: NodeTracerInitializeOptions = {},
   ): Promise<NodeTracer> {
     if (this._initialized) {
       return this;
@@ -42,7 +43,7 @@ export class NodeTracer extends Tracer {
       return this;
     } catch (error) {
       throw new Error(
-        `Failed to initialize node tracer: ${error instanceof Error ? error.message : String(error)}`
+        `Failed to initialize node tracer: ${error instanceof Error ? error.message : String(error)}`,
       );
     }
   }
@@ -64,20 +65,16 @@ export class NodeTracer extends Tracer {
   }
 
   public static createWithConfiguration(
-    configuration: TracerConfiguration
+    configuration: TracerConfiguration,
   ): NodeTracer {
     return new NodeTracer(configuration);
   }
 
-  public async forceFlush(): Promise<void> {
-    if (this.nodeSDK) {
-      await this.nodeSDK.shutdown();
-    }
-  }
-
   public async shutdown(): Promise<void> {
-    if (this.nodeSDK) {
-      await this.nodeSDK.shutdown();
+    if (!this.nodeSDK) {
+      Logger.warn("Node SDK not initialized, skipping shutdown");
+      return;
     }
+    await this.nodeSDK.shutdown();
   }
 }

src/tracer/Tracer.ts

@@ -55,7 +55,6 @@ export abstract class Tracer {
     );
 
     this._initialized = false;
-    
   }
 
   public abstract initialize(options: TracerInitializeOptions): Promise<Tracer>;
@@ -384,13 +383,14 @@ export abstract class Tracer {
                   OpenTelemetryKeys.AttributeKeys.JUDGMENT_OUTPUT,
                   this.serializer(res),
                 );
-                span.end();
                 return res;
               })
               .catch((err) => {
                 span.recordException(err as Error);
-                span.end();
                 throw err;
+              })
+              .finally(() => {
+                span.end();
               }) as TResult;
           } else {
             span.setAttribute(

src/utils/logger.ts

@@ -1,3 +1,5 @@
+import { JUDGMENT_LOG_LEVEL } from "../env";
+
 export class Logger {
   private static readonly RESET = "\x1b[0m";
   private static readonly RED = "\x1b[31m";
@@ -13,13 +15,27 @@ export class Logger {
   } as const;
 
   private static initialized = false;
-  private static currentLevel: number = Logger.Level.INFO;
+  private static currentLevel: number = Logger.Level.WARNING;
   private static useColor = true;
 
   private static initialize(): void {
     if (!Logger.initialized) {
       const noColor = process.env.JUDGMENT_NO_COLOR;
       Logger.useColor = !noColor && process.stdout.isTTY;
+
+      const logLevel = JUDGMENT_LOG_LEVEL.toLowerCase();
+      if (logLevel) {
+        const levelMap: Record<string, number> = {
+          debug: Logger.Level.DEBUG,
+          info: Logger.Level.INFO,
+          warning: Logger.Level.WARNING,
+          warn: Logger.Level.WARNING,
+          error: Logger.Level.ERROR,
+          critical: Logger.Level.CRITICAL,
+        };
+        Logger.currentLevel = levelMap[logLevel] ?? Logger.Level.WARNING;
+      }
+
       Logger.initialized = true;
     }
   }