open-webui/backend/utils/models.py
2024-05-24 22:21:57 -07:00

11 lines
236 B
Python

from apps.web.models.models import Models, ModelModel, ModelForm, ModelResponse
def get_model_id_from_custom_model_id(id: str):
model = Models.get_model_by_id(id)
if model:
return model.id
else:
return id