fix: add logs

parea-ai · Feb 1, 2024 · 275c4fc · 275c4fc
1 parent 49e9631
commit 275c4fc
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/src/utils/wrap_openai.ts b/src/utils/wrap_openai.ts
@@ -49,7 +49,9 @@ function wrapMethod(method: Function, idxArgs: number = 0) {
 
         try {
           response = await method.apply(this, args);
-          console.log('response in try', response);
+          if (idxArgs > 0) {
+            await args[0].logger.info(`response in try: ${response}`);
+          }
           traceInsert(traceId, {
             output: getOutput(response),
             input_tokens: response.usage.prompt_tokens,
@@ -72,8 +74,10 @@ function wrapMethod(method: Function, idxArgs: number = 0) {
             latency: (endTimestamp.getTime() - startTimestamp.getTime()) / 1000,
             status: status,
           });
-          console.log('traceLog in finally', traceLog);
-          console.log('pareaLogger in finally', pareaLogger);
+          if (idxArgs > 0) {
+            await args[0].logger.info(`'traceLog in finally: ${traceLog}`);
+            await args[0].logger.info(`pareaLogger in finally: ${pareaLogger}`);
+          }
           await pareaLogger.recordLog(traceLog);
         }