dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-30 09:14:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

fix: use importlib.resources.open_text() to read JSON file in get_model_info

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-08-25 07:50:58 -07:00
parent f4adc09277
commit 6d7a6ccee3
1 changed files with 8 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

15
aider/models.py
View file

@ -452,15 +452,16 @@ class Model(ModelSettings):
def get_model_info(self, model):
if not litellm._lazy_module:
# Try and do this quickly, without triggering the litellm import
spec = importlib.util.find_spec("litellm")
if spec:
origin = Path(spec.origin)
fname = origin.parent / "model_prices_and_context_window_backup.json"
if fname.exists():
data = json.loads(fname.read_text())
info = data.get(model)
try:
with importlib.resources.open_text(
"litellm", "model_prices_and_context_window_backup.json"
) as f:
content = json.load(f)
info = content.get(model)
if info:
return info
except Exception:
pass # If there's any error, fall back to the slow way
# Do it the slow way...
try: