feat: add resource model metadata to search path

This commit is contained in:
Paul Gauthier (aider) 2024-10-22 10:34:29 -07:00
parent 1b530f9200
commit d234d5e3ec

View file

@ -8,6 +8,7 @@ import traceback
from pathlib import Path
import git
import importlib_resources
from dotenv import load_dotenv
from prompt_toolkit.enums import EditingMode
@ -321,6 +322,10 @@ def register_litellm_models(git_root, model_metadata_fname, io, verbose=False):
".aider.model.metadata.json", git_root, model_metadata_fname
)
# Add the resource file path
resource_metadata = importlib_resources.files("aider.resources").joinpath("model-metadata.json")
model_metatdata_files.append(str(resource_metadata))
try:
model_metadata_files_loaded = models.register_litellm_models(model_metatdata_files)
if len(model_metadata_files_loaded) > 0 and verbose: