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

feat: Fetch model info from GitHub first, then fall back to local resource

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-08-25 07:52:48 -07:00
parent 63c6bffb46
commit bd30ae936c
1 changed files with 17 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

22
aider/models.py
View file

@ -452,15 +452,27 @@ class Model(ModelSettings):
if not litellm._lazy_module:
# Try and do this quickly, without triggering the litellm import
try:
with importlib.resources.open_text(
"litellm", "model_prices_and_context_window_backup.json"
) as f:
content = json.load(f)
# First, attempt to fetch from GitHub
import requests
url = "https://raw.githubusercontent.com/BerriAI/litellm/main/model_prices_and_context_window.json"
response = requests.get(url, timeout=5)
if response.status_code == 200:
content = response.json()
info = content.get(model)
if info:
return info
except Exception:
pass # If there's any error, fall back to the slow way
# If fetching from GitHub fails, fall back to local resource
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: