From a60863b8c46921534e6b4c05621b161d71a0afbd Mon Sep 17 00:00:00 2001
From: Yevhenii <uman980124@gmail.com>
Date: Tue, 16 Sep 2025 18:52:21 +0300
Subject: [PATCH] Refactoring

---
 .../org/thingsboard/server/dao/ai/AiModelServiceImpl.java     | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/dao/src/main/java/org/thingsboard/server/dao/ai/AiModelServiceImpl.java b/dao/src/main/java/org/thingsboard/server/dao/ai/AiModelServiceImpl.java
index 971eba8921..7cdf5b027d 100644
--- 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;
         }