mirror of
https://github.com/theroyallab/tabbyAPI.git
synced 2026-07-16 16:50:26 +00:00
84 lines
2.4 KiB
Python
84 lines
2.4 KiB
Python
"""Construct a model of all optional dependencies"""
|
|
|
|
import importlib.util
|
|
from importlib.metadata import version as package_version
|
|
from common.logger import xlogger
|
|
from packaging import version
|
|
from pydantic import BaseModel, computed_field
|
|
|
|
|
|
# Declare the exported parts of this module
|
|
__all__ = ["dependencies"]
|
|
|
|
|
|
class DependenciesModel(BaseModel):
|
|
"""Model of which optional dependencies are installed."""
|
|
|
|
torch: bool
|
|
exllamav3: bool
|
|
infinity_emb: bool
|
|
sentence_transformers: bool
|
|
|
|
@computed_field
|
|
@property
|
|
def extras(self) -> bool:
|
|
return self.infinity_emb and self.sentence_transformers
|
|
|
|
@computed_field
|
|
@property
|
|
def inference(self) -> bool:
|
|
return self.torch and self.exllamav3
|
|
|
|
|
|
def is_installed(package_name: str) -> bool:
|
|
"""Utility function to check if a package is installed."""
|
|
|
|
spec = importlib.util.find_spec(package_name)
|
|
return spec is not None
|
|
|
|
|
|
def get_installed_deps() -> DependenciesModel:
|
|
"""Check if optional dependencies are installed by looping over the fields."""
|
|
|
|
fields = DependenciesModel.model_fields
|
|
|
|
installed_deps = {}
|
|
|
|
for field_name in fields.keys():
|
|
installed_deps[field_name] = is_installed(field_name)
|
|
|
|
return DependenciesModel(**installed_deps)
|
|
|
|
|
|
def check_package_version(package_name: str, required_version_str: str):
|
|
"""
|
|
Fetches and verifies a given package version.
|
|
|
|
This assumes that the required package is installed.
|
|
"""
|
|
|
|
required_version = version.parse(required_version_str)
|
|
|
|
# Prefer the version reported by the package itself. Installed metadata
|
|
# can be stale when running a source checkout from sys.path.
|
|
try:
|
|
version_str = importlib.import_module(f"{package_name}.version").__version__
|
|
except (ImportError, AttributeError):
|
|
version_str = package_version(package_name)
|
|
|
|
current_version = version.parse(version_str.split("+")[0])
|
|
|
|
unsupported_message = (
|
|
f"ERROR: TabbyAPI requires {package_name} {required_version} "
|
|
f"or greater. Your current version is {current_version}. "
|
|
"Please update your dependencies."
|
|
)
|
|
|
|
if current_version < required_version:
|
|
raise RuntimeError(unsupported_message)
|
|
else:
|
|
xlogger.info(f"{package_name} version: {current_version}")
|
|
|
|
|
|
dependencies = get_installed_deps()
|