open-webui/backend/utils/models.py
2024-05-26 01:15:48 -07:00

11 lines
238 B
Python

from apps.webui.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