README - token count fixes

Author: peterbanda

README.md

@@ -367,6 +367,7 @@ For this to work you need to use `OpenAIServiceStreamedFactory` from `openai-sca
 
 An example how to count message tokens:
 ```scala
+import io.cequence.openaiscala.service.OpenAICountTokensHelper
 import io.cequence.openaiscala.domain.{AssistantMessage, BaseMessage, FunctionSpec, ModelId, SystemMessage, UserMessage}
 
 class MyCompletionService extends OpenAICountTokensHelper {
@@ -381,17 +382,14 @@ class MyCompletionService extends OpenAICountTokensHelper {
       UserMessage("Where was it played?"),
     )
 
-    val tokens = countMessageTokens(model, messages)
+    val tokenCount = countMessageTokens(model, messages)
   }
 }
 ```
 
 An example how to count message tokens when a function is involved:
 ```scala
 import io.cequence.openaiscala.service.OpenAICountTokensHelper
-import io.cequence.openaiscala.domain.{ChatRole, FunMessageSpec, FunctionSpec}
-
-// TODO: simpler example
 import io.cequence.openaiscala.domain.{BaseMessage, FunctionSpec, ModelId, SystemMessage, UserMessage}
 
 class MyCompletionService extends OpenAICountTokensHelper {
@@ -420,7 +418,7 @@ class MyCompletionService extends OpenAICountTokensHelper {
       )
     )
 
-    val tokens = countFunMessageTokens(model, messages, Seq(function), Some(function.name))
+    val tokenCount = countFunMessageTokens(model, messages, Seq(function), Some(function.name))
   }
 }
 ```

openai-count-tokens/README.md

@@ -43,13 +43,11 @@ class MyCompletionService extends OpenAICountTokensHelper {
       UserMessage("Where was it played?"),
     )
 
-    val tokens = countMessageTokens(model, messages)
+    val tokenCount = countMessageTokens(model, messages)
   }
 }
 ```
 
-
-
 ## Development and testing
 
 Test with sbt