feat: improve library import.

openai_cost_logger/__init__.py

@@ -1 +1,5 @@
+from .openai_cost_logger import OpenAICostLogger
+from .constants import MODELS_COST, DEFAULT_LOG_PATH
+from .openai_cost_logger_viz import OpenAICostLoggerViz
+from .openai_cost_logger_utils import OpenAICostLoggerUtils
 print('imported openai_cost_logger')

setup.py

@@ -1,15 +1,10 @@
 from setuptools import setup, find_packages
-import os
 
 # Read the README file (reStructuredText format)
 with open('README.rst') as f:
     long_description = f.read()
 
-# Get the version number from the environment
-# version_number = os.getenv('VERSION_NUMBER')
-# version_number = version_number.strip("v")
-
-version_number = '0.2.1'
+version_number = '0.3.0'
 
 setup(
     name='openai-cost-logger',
@@ -20,7 +15,7 @@
     url='https://github.com/drudilorenzo/openai-cost-tracker',
     long_description=long_description,
     long_description_content_type='text/x-rst',
-    keywords=['openai', 'cost', 'logger', 'tracker'],
+    keywords=['openai', 'cost', 'logger', 'tracker', 'viz', 'llms'],
     license='MIT',
     packages=find_packages(include=['openai_cost_logger', 'openai_cost_logger.*']),
     requires=['openai', 'pandas', 'matplotlib'],