Clean up docstring

posit-dev · Dec 17, 2024 · 710e6fe · 710e6fe
1 parent e59a4ad
commit 710e6fe
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/chatlas/_chat.py b/chatlas/_chat.py
@@ -199,9 +199,8 @@ def token_count(
         """
         Get the token count for the given input.
 
-        Get the token count for the given input. This can be useful for
-        understanding how many tokens your input will cost before sending it to
-        the model.
+        This is useful for estimating the number of tokens your input will cost 
+        before sending it to the model.
 
         Parameters
         ----------
@@ -224,11 +223,11 @@ def token_count(
         # Estimate the token count before sending the input
         print(chat.token_count("What is 2 + 2?"))
 
-        # Once input is sent, you can get the actual input and output token counts
+        # Once input is sent, you can get the actual input and output
+        # token counts from the chat object
         chat.chat("What is 2 + 2?", echo="none")
-        print(chat.last_turn().tokens)
+        print(chat.token_usage)
         ```
-
         """
 
         return self.provider.token_count(