Merge pull request #2773 from davidbrai/fix-edit-command-llm-log

fix: log llm completition in edit command
continuedev · Nov 1, 2024 · fc39a60 · fc39a60
2 parents 2c44b50 + 3094180
commit fc39a60
Showing 1 changed file with 13 additions and 8 deletions.
diff --git a/core/llm/index.ts b/core/llm/index.ts
@@ -501,15 +501,20 @@ export abstract class BaseLLM implements ILLM {
     }
 
     let completion = "";
-    for await (const chunk of this._streamComplete(prompt, completionOptions)) {
-      completion += chunk;
-      yield chunk;
-    }
-
-    this._logTokensGenerated(completionOptions.model, prompt, completion);
+    try {
+      for await (const chunk of this._streamComplete(
+        prompt,
+        completionOptions,
+      )) {
+        completion += chunk;
+        yield chunk;
+      }
+    } finally {
+      this._logTokensGenerated(completionOptions.model, prompt, completion);
 
-    if (log && this.writeLog) {
-      await this.writeLog(`Completion:\n\n${completion}\n\n`);
+      if (log && this.writeLog) {
+        await this.writeLog(`Completion:\n\n${completion}\n\n`);
+      }
     }
 
     return {