From f2139b048e368a12241bef482e6f3496a8d4c692 Mon Sep 17 00:00:00 2001 From: Lorenzo Drudi Date: Sat, 27 Apr 2024 12:05:44 +0200 Subject: [PATCH] feat: improve library import. --- openai_cost_logger/__init__.py | 4 ++++ setup.py | 9 ++------- 2 files changed, 6 insertions(+), 7 deletions(-) diff --git a/openai_cost_logger/__init__.py b/openai_cost_logger/__init__.py index 7a69ef6..e511bd8 100644 --- a/openai_cost_logger/__init__.py +++ b/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') \ No newline at end of file diff --git a/setup.py b/setup.py index 5e0ede3..797e609 100644 --- a/setup.py +++ b/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'],