tokenator 0.1.5__tar.gz → 0.1.7__tar.gz

{tokenator-0.1.5 → tokenator-0.1.7}/PKG-INFO

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

{tokenator-0.1.5 → tokenator-0.1.7}/pyproject.toml

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

{tokenator-0.1.5 → tokenator-0.1.7}/src/tokenator/__init__.py

@@ -1,18 +1,18 @@
 """Tokenator - Track and analyze your OpenAI API token usage and costs."""
 
 import logging
-from .client_openai import OpenAIWrapper
+from .client_openai import tokenator_openai
+from .client_anthropic import tokenator_anthropic
 from . import usage
-from .utils import get_default_db_path, is_colab
+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", "is_colab"]
+__all__ = ["tokenator_openai", "tokenator_anthropic", "usage", "get_default_db_path"]
 
 logger = logging.getLogger(__name__)
 
 try:
-    # Always run migrations, even in Colab
-    from .migrations import check_and_run_migrations
     check_and_run_migrations()
 except Exception as e:
     logger.warning(f"Failed to run migrations, but continuing anyway: {e}") 

{tokenator-0.1.5 → tokenator-0.1.7}/src/tokenator/utils.py

@@ -47,4 +47,10 @@ def get_default_db_path() -> str:
         # Fallback to current directory if we can't create the default path
         fallback_path = os.path.join(os.getcwd(), "tokenator_usage.db")
         logger.warning(f"Could not create default db path, falling back to {fallback_path}. Error: {e}")
-        return fallback_path 
+        return fallback_path 
+
+__all__ = [
+    "get_default_db_path",
+    "is_colab",
+    # ... other exports ...
+]