diff --git a/routstr/proxy.py b/routstr/proxy.py index edceaaa6..38224ea8 100644 --- a/routstr/proxy.py +++ b/routstr/proxy.py @@ -65,7 +65,11 @@ def get_upstreams() -> list[BaseUpstreamProvider]: def get_model_instance(model_id: str) -> Model | None: """Get Model instance by ID from global cache.""" - return _model_instances.get(model_id.lower()) + return next( + (model for alias, model in _model_instances.items() + if alias.lower() in model_id.lower()), + None + ) def get_provider_for_model(model_id: str) -> BaseUpstreamProvider | None: