refactor: remove unused logging logic 🧹

microsoft · Dec 11, 2024 · 1ef1dd8 · 1ef1dd8
1 parent 680d579
commit 1ef1dd8
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git a/packages/cli/src/run.ts b/packages/cli/src/run.ts
@@ -410,7 +410,6 @@ export async function runScriptInternal(
 
     await aggregateResults(scriptId, outTrace, stats, result)
     await traceAgentMemory(trace)
-    stats.log()
 
     if (outAnnotations && result.annotations?.length) {
         if (isJSONLFilename(outAnnotations))

diff --git a/packages/core/src/usage.ts b/packages/core/src/usage.ts
@@ -254,13 +254,11 @@ export class GenerationStats {
      * @param indent - The indentation used for logging.
      */
     private logTokens(indent: string) {
-        if (!this.resolvedModel) return
-
         const unknowns = new Set<string>()
         const c = this.cost()
         if (this.model && isNaN(c) && isCosteable(this.model))
             unknowns.add(this.model)
-        if (this.model || c) {
+        if (this.resolvedModel || c) {
             const au = this.accumulatedUsage()
             logVerbose(
                 `${indent}${this.label ? `${this.label} (${this.resolvedModel})` : this.resolvedModel}> ${au.total_tokens} tokens (${au.prompt_tokens} -> ${au.completion_tokens}) ${renderCost(c)}`