jalr/tabbyAPI-ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: logging (#39)

Browse source 
* add logging

* simplify the logger

* formatting

* final touches

* fix format

* Model: Add log to metrics

Signed-off-by: kingbri <bdashore3@proton.me>

---------

Authored-by: AlpinDale <52078762+AlpinDale@users.noreply.github.com>
This commit is contained in:
AlpinDale 2023-12-23 04:33:31 +00:00 • committed by GitHub
parent f5314fcdad
commit 6a5bbd217c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
11 changed files with 170 additions and 74 deletions
Download patch file Download diff file Expand all files Collapse all files

39
tests/wheel_test.py
View file

@ -2,45 +2,52 @@
from importlib.metadata import version
from importlib.util import find_spec
from logger import init_logger
logger = init_logger(__name__)
successful_packages = []
errored_packages = []
if find_spec("flash_attn") is not None:
print(f"Flash attention on version {version('flash_attn')} successfully imported")
logger.info(
f"Flash attention on version {version('flash_attn')} " "successfully imported"
)
successful_packages.append("flash_attn")
else:
print("Flash attention 2 is not found in your environment.")
logger.error("Flash attention 2 is not found in your environment.")
errored_packages.append("flash_attn")
if find_spec("exllamav2") is not None:
print(f"Exllamav2 on version {version('exllamav2')} successfully imported")
logger.info(f"Exllamav2 on version {version('exllamav2')} " "successfully imported")
successful_packages.append("exllamav2")
else:
print("Exllamav2 is not found in your environment.")
logger.error("Exllamav2 is not found in your environment.")
errored_packages.append("exllamav2")
if find_spec("torch") is not None:
print(f"Torch on version {version('torch')} successfully imported")
logger.info(f"Torch on version {version('torch')} successfully imported")
successful_packages.append("torch")
else:
print("Torch is not found in your environment.")
logger.error("Torch is not found in your environment.")
errored_packages.append("torch")
if find_spec("jinja2") is not None:
print(f"Jinja2 on version {version('jinja2')} successfully imported")
logger.info(f"Jinja2 on version {version('jinja2')} successfully imported")
successful_packages.append("jinja2")
else:
print("Jinja2 is not found in your environment.")
logger.error("Jinja2 is not found in your environment.")
errored_packages.append("jinja2")
print(
f"\nSuccessful imports: {', '.join(successful_packages)}",
f"\nErrored imports: {''.join(errored_packages)}",
)
logger.info(f"\nSuccessful imports: {', '.join(successful_packages)}")
logger.error(f"Errored imports: {''.join(errored_packages)}")
if len(errored_packages) > 0:
print(
"\nIf packages are installed, but not found on this test, please "
"check the wheel versions for the correct python version and CUDA "
"version (if applicable)."
logger.warning(
"If all packages are installed, but not found "
"on this test, please check the wheel versions for the "
"correct python version and CUDA version (if "
"applicable)."
)
else:
logger.info("All wheels are installed correctly.")