Skip to content

Commit

Permalink
Update model_repository.py
Browse files Browse the repository at this point in the history
  • Loading branch information
cvarrei authored and samuelrince committed Jul 17, 2024
1 parent 19af8cb commit 1333f34
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions ecologits/model_repository.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,7 @@ def __init__(self, models: list[Model]) -> None:

def find_model(self, provider: str, model_name: str) -> Optional[Model]:
for model in self.__models:
# To handle specific LiteLLM calling (e.g., mistral/mistral-small)
if model.provider == provider and model.name in model_name:
return model
return None
Expand Down

0 comments on commit 1333f34

Please sign in to comment.