tokenator 0.1.2__tar.gz → 0.1.3__tar.gz

{tokenator-0.1.2 → tokenator-0.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tokenator
-Version: 0.1.2
+Version: 0.1.3
 Summary: Token usage tracking wrapper for LLMs
 License: MIT
 Author: Ujjwal Maheshwari

{tokenator-0.1.2 → tokenator-0.1.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "tokenator"
-version = "0.1.2"
+version = "0.1.3"
 description = "Token usage tracking wrapper for LLMs"
 authors = ["Ujjwal Maheshwari <your.email@example.com>"]
 readme = "README.md"

tokenator-0.1.3/src/tokenator/__init__.py

@@ -0,0 +1,20 @@
+"""Tokenator - Track and analyze your OpenAI API token usage and costs."""
+
+import logging
+from .client_openai import OpenAIWrapper
+from . import usage
+from .utils import get_default_db_path, is_colab
+
+__version__ = "0.1.0"
+__all__ = ["OpenAIWrapper", "usage", "get_default_db_path"]
+
+logger = logging.getLogger(__name__)
+
+try:
+    if not is_colab():
+        from .migrations import check_and_run_migrations
+        check_and_run_migrations()
+    else:
+        logger.info("Running in Colab environment - skipping migrations")
+except Exception as e:
+    logger.warning(f"Failed to run migrations, but continuing anyway: {e}") 

tokenator-0.1.2/src/tokenator/__init__.py

@@ -1,12 +0,0 @@
-"""Tokenator - Track and analyze your OpenAI API token usage and costs."""
-
-from .client_openai import OpenAIWrapper
-from . import usage
-from .utils import get_default_db_path
-from .migrations import check_and_run_migrations
-
-__version__ = "0.1.0"
-__all__ = ["OpenAIWrapper", "usage", "get_default_db_path"]
-
-# Run migrations on import
-check_and_run_migrations()