dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-06-01 18:25:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

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

12
aider/models.py
View file

@ -451,6 +451,18 @@ class Model(ModelSettings):
def get_model_info(self, model):
if not litellm._lazy_module:
# Try and do this quickly, without triggering the litellm import
try:
# 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:
# If fetching from GitHub fails, fall back to local resource
try:
with importlib.resources.open_text(
"litellm", "model_prices_and_context_window_backup.json"