Remove API key from log when using loggingEnabled

takahirom · Nov 7, 2024 · 07a6d6f · 07a6d6f
1 parent 01f7df8
commit 07a6d6f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/...-ai-openai/src/commonMain/kotlin/com/github/takahirom/roborazzi/OpenAiAiAssertionModel.kt b/...-ai-openai/src/commonMain/kotlin/com/github/takahirom/roborazzi/OpenAiAiAssertionModel.kt
@@ -57,7 +57,11 @@ class OpenAiAiAssertionModel(
     }
     if (loggingEnabled) {
       install(Logging) {
-        logger = Logger.SIMPLE
+        logger = object: Logger {
+          override fun log(message: String) {
+            Logger.SIMPLE.log(message.replace(apiKey, "API_KEY"))
+          }
+        }
         level = LogLevel.ALL
       }
     }