From 9576957adf7934e1d3bc5ddf0f8c0deef962f973 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=8E=8B=E7=A5=A5=E5=AE=87?= <625024108@qq.com> Date: Sat, 2 Aug 2025 02:17:48 +0800 Subject: [PATCH] update --- examples/utils/misc/conversation_test_truncate.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/utils/misc/conversation_test_truncate.py b/examples/utils/misc/conversation_test_truncate.py index ae34692b..f660ff45 100644 --- a/examples/utils/misc/conversation_test_truncate.py +++ b/examples/utils/misc/conversation_test_truncate.py @@ -1,6 +1,6 @@ from swarms.structs.conversation import Conversation from dotenv import load_dotenv - +from swarms.utils.litellm_tokenizer import count_tokens # Load environment variables from .env file load_dotenv() @@ -22,7 +22,7 @@ def demonstrate_truncation(): conversation.add("user", short_message) # Display token count - from swarms.utils.litellm_tokenizer import count_tokens + tokens = count_tokens(short_message, conversation.tokenizer_model_name) print(f"Short message token count: {tokens}")