diff --git a/routstr/upstream/ppqai.py b/routstr/upstream/ppqai.py index f8a9c965..688c018a 100644 --- a/routstr/upstream/ppqai.py +++ b/routstr/upstream/ppqai.py @@ -131,7 +131,9 @@ class PPQAIUpstreamProvider(BaseUpstreamProvider): ( model for model in or_models - if model.id == ppqai_model.id + if (model.id == ppqai_model.id) + or (model.id.split("/")[-1] == ppqai_model.id) + or (model.id == ppqai_model.id.split("/")[-1]) ), None, )