Correct fraction handling

Author: pamelafox

src/openai_messages_token_helper/images_helper.py

@@ -17,12 +17,12 @@ def get_image_dims(image_uri: str) -> tuple[int, int]:
         raise ValueError("Image must be a base64 string.")
 
 
-def count_tokens_for_image(image_uri: str, detail: str = "auto", model: str = None) -> int:
+def count_tokens_for_image(image_uri: str, detail: str = "auto", model: str | None = None) -> int:
     # From https://github.com/openai/openai-cookbook/pull/881/files
     # Based on https://platform.openai.com/docs/guides/vision
-    multiplier = 1
+    multiplier = Fraction(1, 1)
     if model == "gpt-4o-mini":
-        multiplier = 33 + Fraction(1, 3)
+        multiplier = Fraction(100, 3)
     COST_PER_TILE = 85 * multiplier
     LOW_DETAIL_COST = COST_PER_TILE
     HIGH_DETAIL_COST_PER_TILE = COST_PER_TILE * 2

tests/verify_openai.py

@@ -4,8 +4,8 @@
 import azure.identity
 import openai
 from dotenv import load_dotenv
-from image_messages import IMAGE_MESSAGE_COUNTS
-from messages import MESSAGE_COUNTS
+from image_messages import IMAGE_MESSAGE_COUNTS  # type: ignore[import-not-found]
+from messages import MESSAGE_COUNTS  # type: ignore[import-not-found]
 
 # Setup the OpenAI client to use either Azure OpenAI or OpenAI API
 load_dotenv()