tokencostauto 0.1.25__py3-none-any.whl → 0.1.26__py3-none-any.whl

tokencostauto/costs.py

@@ -79,7 +79,7 @@ def count_message_tokens(messages: List[Dict[str, str]], model: str) -> int:
     try:
         encoding = tiktoken.encoding_for_model(model)
     except KeyError:
-        logger.warning("Model not found. Using cl100k_base encoding.")
+        logger.info("Model not found. Using cl100k_base encoding.")
         encoding = tiktoken.get_encoding("cl100k_base")
     if model in {
         "gpt-3.5-turbo-0613",
@@ -100,17 +100,22 @@ def count_message_tokens(messages: List[Dict[str, str]], model: str) -> int:
         tokens_per_message = 4
         tokens_per_name = -1  # if there's a name, the role is omitted
     elif "gpt-3.5-turbo" in model:
-        logger.warning(
+        logger.info(
             "gpt-3.5-turbo may update over time. Returning num tokens assuming gpt-3.5-turbo-0613."
         )
         return count_message_tokens(messages, model="gpt-3.5-turbo-0613")
     elif "gpt-4o" in model:
-        logger.warning(
+        logger.info(
             "Warning: gpt-4o may update over time. Returning num tokens assuming gpt-4o-2024-05-13."
         )
         return count_message_tokens(messages, model="gpt-4o-2024-05-13")
+    elif "gpt-4.1" in model:
+        logger.info(
+            "Warning: gpt-4.1 may update over time. Returning num tokens assuming gpt-4o-2024-05-13."
+        )
+        return count_message_tokens(messages, model="gpt-4o-2024-05-13")
     elif "gpt-4" in model:
-        logger.warning(
+        logger.info(
             "gpt-4 may update over time. Returning num tokens assuming gpt-4-0613."
         )
         return count_message_tokens(messages, model="gpt-4-0613")
@@ -154,7 +159,7 @@ def count_string_tokens(prompt: str, model: str) -> int:
     try:
         encoding = tiktoken.encoding_for_model(model)
     except KeyError:
-        logger.warning("Warning: model not found. Using cl100k_base encoding.")
+        logger.info("Warning: model not found. Using cl100k_base encoding.")
         encoding = tiktoken.get_encoding("cl100k_base")
 
     return len(encoding.encode(prompt))

{tokencostauto-0.1.25.dist-info → tokencostauto-0.1.26.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tokencostauto
-Version: 0.1.25
+Version: 0.1.26
 Summary: To calculate token and translated USD cost of string and message calls to OpenAI, for example when used by AI agents
 Author-email: Trisha Pan <trishaepan@gmail.com>, Alex Reibman <areibman@gmail.com>, Pratyush Shukla <ps4534@nyu.edu>, Thiago MadPin <madpin@gmail.com>
 Project-URL: Homepage, https://github.com/madpin/tokencostaudo

tokencostauto-0.1.26.dist-info/RECORD

@@ -0,0 +1,9 @@
+tokencostauto/__init__.py,sha256=-4d_ryFH62SgNXPXA8vGPFZoAKtOBjnsg37EB_RkZG8,289
+tokencostauto/constants.py,sha256=_82MlTkTrdrwzyRosQD7d3JdgNP9KAUM-cZo8DE00P0,3395
+tokencostauto/costs.py,sha256=tXsgrTypq-dCHaHtoXcg2XepezWsAvZpl9gEsv_53iE,10679
+tokencostauto/model_prices.json,sha256=pjU4wUl3QQHwEvJFywqGIFYGcIb_Q3IxL4NSEL87uNg,468528
+tokencostauto-0.1.26.dist-info/licenses/LICENSE,sha256=4PLv_CD6Ughnsvg_nM2XeTqGwVK6lQVR77kVWbPq-0U,1065
+tokencostauto-0.1.26.dist-info/METADATA,sha256=gMp2jWQc0d2vCamN5pQwQiJ3eqAOEqzsW3Ju5vrCjHo,204076
+tokencostauto-0.1.26.dist-info/WHEEL,sha256=DnLRTWE75wApRYVsjgc6wsVswC54sMSJhAEd4xhDpBk,91
+tokencostauto-0.1.26.dist-info/top_level.txt,sha256=szZQTUJRotfIaeZCDsOgvofIkLt2ak88RP13oI51-TU,14
+tokencostauto-0.1.26.dist-info/RECORD,,

tokencostauto-0.1.25.dist-info/RECORD

@@ -1,9 +0,0 @@
-tokencostauto/__init__.py,sha256=-4d_ryFH62SgNXPXA8vGPFZoAKtOBjnsg37EB_RkZG8,289
-tokencostauto/constants.py,sha256=_82MlTkTrdrwzyRosQD7d3JdgNP9KAUM-cZo8DE00P0,3395
-tokencostauto/costs.py,sha256=vCtP8QEnIYpbDs0ls4DzK6I6SR1AJiX3hmbooyQC1aM,10459
-tokencostauto/model_prices.json,sha256=pjU4wUl3QQHwEvJFywqGIFYGcIb_Q3IxL4NSEL87uNg,468528
-tokencostauto-0.1.25.dist-info/licenses/LICENSE,sha256=4PLv_CD6Ughnsvg_nM2XeTqGwVK6lQVR77kVWbPq-0U,1065
-tokencostauto-0.1.25.dist-info/METADATA,sha256=BCVJyt2HGDmvRnuztA1LURMlkajYoCuRoxjL5anN_Mc,204076
-tokencostauto-0.1.25.dist-info/WHEEL,sha256=DnLRTWE75wApRYVsjgc6wsVswC54sMSJhAEd4xhDpBk,91
-tokencostauto-0.1.25.dist-info/top_level.txt,sha256=szZQTUJRotfIaeZCDsOgvofIkLt2ak88RP13oI51-TU,14
-tokencostauto-0.1.25.dist-info/RECORD,,