| :param credentials: model credentials | :param credentials: model credentials | ||||
| :return: model schema | :return: model schema | ||||
| """ | """ | ||||
| return self._get_customizable_model_schema(model, credentials) | |||||
| def _get_customizable_model_schema(self, model: str, credentials: dict) -> Optional[AIModelEntity]: | |||||
| """ | |||||
| Get customizable model schema and fill in the template | |||||
| """ | |||||
| # get customizable model schema | |||||
| schema = self.get_customizable_model_schema(model, credentials) | schema = self.get_customizable_model_schema(model, credentials) | ||||
| if not schema: | if not schema: | ||||
| return None | return None | ||||