tokenator 0.1.0__py3-none-any.whl → 0.1.1__py3-none-any.whl

tokenator/schemas.py

@@ -2,6 +2,7 @@
 
 import uuid
 from datetime import datetime
+import os
 
 from sqlalchemy import create_engine, Column, Integer, String, DateTime, Float, Index
 from sqlalchemy.orm import sessionmaker, scoped_session, declarative_base
@@ -12,7 +13,12 @@ Base = declarative_base()
 
 def get_engine(db_path: str = None):
     """Create SQLAlchemy engine with the given database path."""
-    db_path = db_path or get_default_db_path()
+    if db_path is None:
+        try:
+            import google.colab  # type: ignore
+            db_path = '/content/tokenator.db'
+        except ImportError:
+            db_path = get_default_db_path()
     return create_engine(f"sqlite:///{db_path}", echo=False)
 
 def get_session(db_path: str = None):

tokenator/utils.py

@@ -23,13 +23,16 @@ def get_default_db_path() -> str:
         xdg_data_home = os.environ.get("XDG_DATA_HOME", "")
         if not xdg_data_home:
             xdg_data_home = os.path.join(str(Path.home()), ".local", "share")
-        return os.path.join(xdg_data_home, "tokenator", "usage.db")
+        db_path = os.path.join(xdg_data_home, "tokenator", "usage.db")
     elif system == "windows":
         # Use %LOCALAPPDATA% on Windows
         local_app_data = os.environ.get("LOCALAPPDATA", "")
         if not local_app_data:
             local_app_data = os.path.join(str(Path.home()), "AppData", "Local")
-        return os.path.join(local_app_data, "tokenator", "usage.db")
+        db_path = os.path.join(local_app_data, "tokenator", "usage.db")
     else:
-        # Fallback for other systems
-        return os.path.join(str(Path.home()), ".tokenator", "usage.db") 
+        db_path = os.path.join(str(Path.home()), ".tokenator", "usage.db")
+    
+    # Create directory if it doesn't exist
+    os.makedirs(os.path.dirname(db_path), exist_ok=True)
+    return db_path 

{tokenator-0.1.0.dist-info → tokenator-0.1.1.dist-info}/METADATA

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

{tokenator-0.1.0.dist-info → tokenator-0.1.1.dist-info}/RECORD

@@ -7,10 +7,10 @@ tokenator/migrations/env.py,sha256=eFTw66gG464JV53740RKU32wqEL8uZFReS_INrvkFrU,1
 tokenator/migrations/script.py.mako,sha256=nJL-tbLQE0Qy4P9S4r4ntNAcikPtoFUlvXe6xvm9ot8,635
 tokenator/migrations.py,sha256=GH84M6AzidTcscMP0nQspBaQ1v6bjx00V4FeN-v5XAo,1354
 tokenator/models.py,sha256=EprE_MMJxDS-YXlcIQLZzfekH7xTYbeOC3bx3B2osVw,1171
-tokenator/schemas.py,sha256=KQrKv_9Euoai3hBZcuiVclU3BlxQkdfEbwYoFV8MKXY,2112
+tokenator/schemas.py,sha256=eVdBWi6_hTETnPw50glq0OvSh3PbP2pLl_aHdf3fi-M,2278
 tokenator/usage.py,sha256=aHjGwzDzaiVznahNk5HqVyk3IxDo5FtFVfOUCeE7DZ4,7833
-tokenator/utils.py,sha256=VF-t6eL16_KokICMfZvO1rIJX9Hm7rexUcaeuLI2iRA,1311
-tokenator-0.1.0.dist-info/LICENSE,sha256=wdG-B6-ODk8RQ4jq5uXSn0w1UWTzCH_MMyvh7AwtGns,1074
-tokenator-0.1.0.dist-info/METADATA,sha256=dlsCQKC_2qVw4OEfj8ISrt5QwIY774PO_H2S3ilj92g,2444
-tokenator-0.1.0.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-tokenator-0.1.0.dist-info/RECORD,,
+tokenator/utils.py,sha256=WAMRY78d2g0PGSQZzPGoEuhJvQO5Iabgx4Wb6ykYp0Q,1407
+tokenator-0.1.1.dist-info/LICENSE,sha256=wdG-B6-ODk8RQ4jq5uXSn0w1UWTzCH_MMyvh7AwtGns,1074
+tokenator-0.1.1.dist-info/METADATA,sha256=ot6LXHhCJ6JrPOU6o18D-rgoarh2ZT0mn4uhPgJbcXU,2444
+tokenator-0.1.1.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+tokenator-0.1.1.dist-info/RECORD,,