fix: include package data; fix test coverage (#2)

* fix: package data included

* chore: uv lock update

* fix: coverage

* fix: fix files in coverage

* fix: coverage report

Author: ds-jakubkolodziejczyk

.github/workflows/ci.yml

@@ -115,7 +115,7 @@ jobs:
       - name: Run Tests With Coverage
         run: |
           # run with coverage to not execute tests twice
-          uv run coverage run -m pytest -v -p no:warnings --junitxml=report.xml
+          uv run -m pytest -v -p no:warnings --junitxml=report.xml --cov=genai_monitor
           uv run coverage report
           uv run coverage xml
 

pyproject.toml

@@ -5,6 +5,7 @@ description = "Observability for Generative AI"
 readme = "README.md"
 requires-python = ">=3.10"
 dependencies = [
+    "setuptools>=61.0",
     "sqlalchemy>=2.0.35",
     "sqlalchemy-mixins>=2.0.5",
     "loguru>=0.7.2",
@@ -32,6 +33,12 @@ litellm = [
     "litellm>=1.60.2",
 ]
 
+[tool.setuptools]
+include-package-data = true
+
+[tool.setuptools.package-data]
+"genai_monitor.registration" = ["data/*.json"]
+
 [tool.uv]
 dev-dependencies = [
     # Development tools
@@ -54,6 +61,7 @@ dev-dependencies = [
     "litellm>=1.60.2",
 ]
 
+
 [tool.ruff]
 exclude = [".venv"]
 extend-include = ["*.ipynb"]

src/genai_monitor/auto/__init__.py

@@ -8,7 +8,7 @@
 from genai_monitor.utils.auto_mode_configuration import load_config
 from genai_monitor.utils.user_registration import register_user
 
-config = load_config(os.getenv("GENAI_EVAL_DB_URL", "sqlite:///genai_eval.db"))
+config = load_config(os.getenv("GENAI_MONITOR_DB_URL", "sqlite:///genai_monitor.db"))
 logger.debug(f"Loaded configuration from database: {config}")
 
 container = get_container()

src/genai_monitor/static/constants.py

@@ -7,7 +7,7 @@
 EMPTY_MODEL_HASH: Literal["<EMPTY>"] = "<EMPTY>"
 
 DEFAULT_GENAI_MONITOR_CONFIG_PATH = get_absolute_path(
-    relative_path=".genai_eval_config.yaml", relative_to=os.path.abspath(__file__)
+    relative_path=".genai_monitor_config.yaml", relative_to=os.path.abspath(__file__)
 )
 DEFAULT_PERSISTENCY_PATH: str = ".binaries/"
 DEFAULT_DB_VERSION: str = "1.0.0"

src/genai_monitor/utils/auto_mode_configuration.py

@@ -36,7 +36,7 @@ def _get_database_version() -> str:
     Returns:
         str: The current database version.
     """
-    return os.getenv("GENAI_EVAL_DB_VERSION", DEFAULT_DB_VERSION)
+    return os.getenv("GENAI_MONITOR_DB_VERSION", DEFAULT_DB_VERSION)
 
 
 # pylint: disable=W0102