open-webui/backend/utils/models.py

11 lines
238 B
Python
Raw Normal View History

2024-05-26 08:15:48 +00:00
from apps.webui.models.models import Models, ModelModel, ModelForm, ModelResponse
2024-05-25 05:21:57 +00:00
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