Refactoring

1 year ago · a60863b8c4
1 changed files with 2 additions and 2 deletions
--- a/dao/src/main/java/org/thingsboard/server/dao/ai/AiModelServiceImpl.java
+++ b/dao/src/main/java/org/thingsboard/server/dao/ai/AiModelServiceImpl.java
@ -146,8 +146,8 @@ class AiModelServiceImpl extends CachedVersionedEntityService<AiModelCacheKey, A
        deleteByTenantIdAndIdInternal(tenantId, id.getId());
    }

-    private boolean deleteByTenantIdAndIdInternal(TenantId tenantId, UUID aiModelId) {
-        AiModel aiModel = aiModelDao.findById(tenantId, aiModelId);
+    private boolean deleteByTenantIdAndIdInternal(TenantId tenantId, UUID modelId) {
+        AiModel aiModel = findAiModelById(tenantId, new AiModelId(modelId)).orElse(null);
        if (aiModel == null) {
            return false;
        }