feat: improve code

Waterball-Software-Academy · Jun 30, 2023 · 0b62dd7 · 0b62dd7
1 parent 90166ed
commit 0b62dd7
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/chatgpt-api/src/main/kotlin/tw/waterballsa/utopia/chatgpt/JTokkit.kt b/chatgpt-api/src/main/kotlin/tw/waterballsa/utopia/chatgpt/JTokkit.kt
@@ -10,14 +10,8 @@ import com.knuddels.jtokkit.api.ModelType
 class JTokkit {
     companion object {
         private val registry = Encodings.newLazyEncodingRegistry()
-        private lateinit var encoding: Encoding
+        private val encoding = registry.getEncodingForModel(ModelType.GPT_3_5_TURBO)
     }
 
-    init {
-        encoding = registry.getEncodingForModel(ModelType.GPT_3_5_TURBO)
-    }
-
-    fun measureNumOfTokens(string: String): Int {
-        return encoding.countTokens(string)
-    }
+    fun measureNumOfTokens(string: String): Int = encoding.countTokens(string)
 }