scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-11 08:38:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 280

fix(get_model_cost_map): harden source-checkout detection and log resolved path

Browse source 
Address PR feedback:
- Witness the repo root with pyproject.toml so moving get_model_cost_map.py
  cannot silently break detection (parents[2] alone was depth-fragile).
- Emit verbose_logger.debug at each branch so the resolved file is
  observable when diagnosing pricing issues.
This commit is contained in:
Chesars 2026-05-04 21:54:49 -03:00
parent c99c255be4
commit 19afe2b20c
1 changed files with 12 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
litellm/litellm_core_utils/get_model_cost_map.py
View file

@ -37,12 +37,20 @@ class GetModelCostMap:
@staticmethod
def load_local_model_cost_map() -> dict:
"""Load the local model cost map: repo-root JSON in a source checkout, bundled package copy in an installed wheel."""
repo_root_json = (
Path(__file__).resolve().parents[2] / "model_prices_and_context_window.json"
)
if repo_root_json.is_file():
candidate_root = Path(__file__).resolve().parents[2]
repo_root_json = candidate_root / "model_prices_and_context_window.json"
# Require pyproject.toml as a witness so we don't misidentify an arbitrary
# parent directory as the repo root if this file is ever moved.
if repo_root_json.is_file() and (candidate_root / "pyproject.toml").is_file():
verbose_logger.debug(
"LiteLLM: loading local model cost map from repo root %s",
repo_root_json,
)
return json.loads(repo_root_json.read_text(encoding="utf-8"))
verbose_logger.debug(
"LiteLLM: loading local model cost map from bundled package resource"
)
return json.loads(
files("litellm")
.joinpath("model_prices_and_context_window_backup.json")