Update UsageManager.kt

SimiaCryptus · Mar 29, 2024 · 3bffbec · 3bffbec
1 parent d4218ad
commit 3bffbec
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/core/src/main/kotlin/com/simiacryptus/skyenet/core/platform/file/UsageManager.kt b/core/src/main/kotlin/com/simiacryptus/skyenet/core/platform/file/UsageManager.kt
@@ -144,7 +144,7 @@ open class UsageManager(val root: File) : UsageInterface {
         synchronized(txLogFile) {
           txLogFileWriter.write("$session,${apiKey},${model.modelName},${tokens.prompt_tokens},input\n")
           txLogFileWriter.write("$session,${apiKey},${model.modelName},${tokens.completion_tokens},output\n")
-          txLogFileWriter.write("$session,${apiKey},${model.modelName},${tokens.completion_tokens},cost\n")
+          txLogFileWriter.write("$session,${apiKey},${model.modelName},${tokens.cost},cost\n")
           txLogFileWriter.flush()
         }
       }