Fix failing test

ag2ai · Jan 21, 2025 · a0f9e97 · a0f9e97
1 parent a658d95
commit a0f9e97
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/autogen/token_count_utils.py b/autogen/token_count_utils.py
@@ -11,14 +11,16 @@
 
 import tiktoken
 
-from .agentchat.contrib.img_utils import num_tokens_from_gpt_image
+# from .agentchat.contrib.img_utils import num_tokens_from_gpt_image
 from .import_utils import optional_import_block
 
 # if PIL is not imported, we will redefine num_tokens_from_gpt_image to return 0 tokens for images
 # Otherwise, it would raise an ImportError
 with optional_import_block() as result:
     import PIL  # noqa: F401
 
+    from .agentchat.contrib.img_utils import num_tokens_from_gpt_image
+
 pil_imported = result.is_successful
 if not pil_imported: