tokenator 0.1.6__tar.gz → 0.1.8__tar.gz

{tokenator-0.1.6 → tokenator-0.1.8}/PKG-INFO

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

{tokenator-0.1.6 → tokenator-0.1.8}/pyproject.toml

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

{tokenator-0.1.6 → tokenator-0.1.8}/src/tokenator/__init__.py

@@ -1,13 +1,14 @@
 """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__)
 

{tokenator-0.1.6 → tokenator-0.1.8}/src/tokenator/client_openai.py

@@ -147,3 +147,5 @@ def tokenator_openai(
         return AsyncOpenAIWrapper(client=client, db_path=db_path)
         
     raise ValueError("Client must be an instance of OpenAI or AsyncOpenAI")
+
+__all__ = ["tokenator_openai"]