clean

Author: abhishekg999

src/tracer/Tracer.ts

@@ -54,7 +54,7 @@ export abstract class Tracer {
     this.apiClient = new JudgmentApiClient(
       this.configuration.apiUrl,
       this.configuration.apiKey,
-      this.configuration.organizationId,
+      this.configuration.organizationId
     );
 
     this._initialized = false;
@@ -65,7 +65,7 @@ export abstract class Tracer {
   private async resolveProjectId(): Promise<string> {
     try {
       Logger.info(
-        `Resolving project ID for project: ${this.configuration.projectName}`,
+        `Resolving project ID for project: ${this.configuration.projectName}`
       );
 
       const response = await this.apiClient.projectsResolve({
@@ -76,7 +76,7 @@ export abstract class Tracer {
 
       if (!resolvedProjectId) {
         throw new Error(
-          `Project ID not found for project: ${this.configuration.projectName}`,
+          `Project ID not found for project: ${this.configuration.projectName}`
         );
       }
 
@@ -86,15 +86,15 @@ export abstract class Tracer {
       return this.projectId;
     } catch (error) {
       throw new Error(
-        `Failed to resolve project ID: ${error instanceof Error ? error.message : String(error)}`,
+        `Failed to resolve project ID: ${error instanceof Error ? error.message : String(error)}`
       );
     }
   }
 
   public static getExporter(
     apiKey: string,
     organizationId: string,
-    projectId: string,
+    projectId: string
   ): JudgmentSpanExporter {
     const endpoint = JUDGMENT_API_URL?.endsWith("/")
       ? `${JUDGMENT_API_URL}otel/v1/traces`
@@ -120,7 +120,7 @@ export abstract class Tracer {
           this.configuration.projectName +
           ", please create it first at https://app.judgmentlabs.ai/org/" +
           (this.configuration.organizationId || "unknown") +
-          "/projects. Skipping Judgment export.",
+          "/projects. Skipping Judgment export."
       );
       return new NoOpSpanExporter();
     }
@@ -135,7 +135,7 @@ export abstract class Tracer {
     if (kind !== null) {
       currentSpan.setAttribute(
         OpenTelemetryKeys.AttributeKeys.JUDGMENT_SPAN_KIND,
-        kind,
+        kind
       );
     }
   }
@@ -186,7 +186,7 @@ export abstract class Tracer {
   public asyncEvaluate(
     scorer: BaseScorer,
     example: ExampleModel,
-    model?: string,
+    model?: string
   ): void {
     if (!this._initialized) {
       Logger.warn("Tracer not initialized, skipping asyncEvaluate");
@@ -212,15 +212,15 @@ export abstract class Tracer {
     const spanId = spanContext.spanId;
 
     Logger.info(
-      `asyncEvaluate: project=${this.configuration.projectName}, traceId=${traceId}, spanId=${spanId}, scorer=${scorer.name}`,
+      `asyncEvaluate: project=${this.configuration.projectName}, traceId=${traceId}, spanId=${spanId}, scorer=${scorer.name}`
     );
 
     const evaluationRun = this.createEvaluationRun(
       scorer,
       example,
       model,
       traceId,
-      spanId,
+      spanId
     );
     this.enqueueEvaluation(evaluationRun);
   }
@@ -250,24 +250,24 @@ export abstract class Tracer {
     const spanId = spanContext.spanId;
 
     Logger.info(
-      `asyncTraceEvaluate: project=${this.configuration.projectName}, traceId=${traceId}, spanId=${spanId}, scorer=${scorer.name}`,
+      `asyncTraceEvaluate: project=${this.configuration.projectName}, traceId=${traceId}, spanId=${spanId}, scorer=${scorer.name}`
     );
 
     try {
       const traceEvaluationRun = this.createTraceEvaluationRun(
         scorer,
         model,
         traceId,
-        spanId,
+        spanId
       );
       const traceEvalJson = this.serializer(traceEvaluationRun);
       currentSpan.setAttribute(
         OpenTelemetryKeys.AttributeKeys.PENDING_TRACE_EVAL,
-        traceEvalJson,
+        traceEvalJson
       );
     } catch (error) {
       Logger.error(
-        `Failed to serialize trace evaluation: ${error instanceof Error ? error.message : String(error)}`,
+        `Failed to serialize trace evaluation: ${error instanceof Error ? error.message : String(error)}`
       );
     }
   }
@@ -276,7 +276,7 @@ export abstract class Tracer {
     scorer: BaseScorer,
     model: string | undefined,
     traceId: string,
-    spanId: string,
+    spanId: string
   ): Record<string, unknown> {
     const evalName = `async_trace_evaluate_${spanId || Date.now()}`;
     const modelName = model || JUDGMENT_DEFAULT_GPT_MODEL;
@@ -312,7 +312,7 @@ export abstract class Tracer {
     example: ExampleModel,
     model: string | undefined,
     traceId: string,
-    spanId: string,
+    spanId: string
   ): ExampleEvaluationRun {
     const runId = `async_evaluate_${spanId || Date.now()}`;
     const modelName = model || JUDGMENT_DEFAULT_GPT_MODEL;
@@ -334,7 +334,7 @@ export abstract class Tracer {
   }
 
   private async enqueueEvaluation(
-    evaluationRun: ExampleEvaluationRun,
+    evaluationRun: ExampleEvaluationRun
   ): Promise<void> {
     if (!this.apiClient) {
       Logger.warn("API client not available, skipping evaluation enqueue");
@@ -346,14 +346,14 @@ export abstract class Tracer {
       Logger.info(`Enqueuing evaluation run: ${evaluationRun.eval_name}`);
     } catch (error) {
       Logger.error(
-        `Failed to enqueue evaluation run: ${error instanceof Error ? error.message : String(error)}`,
+        `Failed to enqueue evaluation run: ${error instanceof Error ? error.message : String(error)}`
       );
     }
   }
 
   public observe<TArgs extends any[], TResult>(
     func: (...args: TArgs) => TResult,
-    spanKind: SpanKind = "span",
+    spanKind: SpanKind = "span"
   ): (...args: TArgs) => TResult {
     return (...args: TArgs) => {
       const currentSpan = trace.getActiveSpan();
@@ -371,13 +371,11 @@ export abstract class Tracer {
           { kind: OpenTelemetrySpanKind.INTERNAL },
           (span) => {
             try {
-              // Set span kind attribute
               span.setAttribute(
                 OpenTelemetryKeys.AttributeKeys.JUDGMENT_SPAN_KIND,
-                spanKind,
+                spanKind
               );
 
-              // Automatically set input attributes from function arguments
               const argNames = parseFunctionArgs(func);
               if (argNames.length === args.length) {
                 const inputObj: Record<string, unknown> = {};
@@ -386,7 +384,7 @@ export abstract class Tracer {
                 });
                 span.setAttribute(
                   OpenTelemetryKeys.AttributeKeys.JUDGMENT_INPUT,
-                  this.serializer(inputObj),
+                  this.serializer(inputObj)
                 );
               }
 
@@ -395,10 +393,9 @@ export abstract class Tracer {
               if (result instanceof Promise) {
                 return result
                   .then((res) => {
-                    // Set output attribute before ending span
                     span.setAttribute(
                       OpenTelemetryKeys.AttributeKeys.JUDGMENT_OUTPUT,
-                      this.serializer(res),
+                      this.serializer(res)
                     );
                     span.end();
                     return res;
@@ -409,10 +406,9 @@ export abstract class Tracer {
                     throw err;
                   }) as TResult;
               } else {
-                // Set output attribute before ending span
                 span.setAttribute(
                   OpenTelemetryKeys.AttributeKeys.JUDGMENT_OUTPUT,
-                  this.serializer(result),
+                  this.serializer(result)
                 );
                 span.end();
                 return result;
@@ -422,21 +418,11 @@ export abstract class Tracer {
               span.end();
               throw err;
             }
-          },
+          }
         );
       });
     };
   }
 
-  // Helper method for observing methods with proper this binding
-  public observeMethod<TArgs extends any[], TResult>(
-    method: (...args: TArgs) => TResult,
-    spanKind: SpanKind = "span",
-  ): (...args: TArgs) => TResult {
-    // Bind the method to its instance if it has one
-    const boundMethod = method.bind ? method.bind(this) : method;
-    return this.observe(boundMethod, spanKind);
-  }
-
   public async shutdown(): Promise<void> {}
 }