diff --git a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
index 2ec0c181..34c6c9c8 100644
--- a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
+++ b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
@@ -902,7 +902,8 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
         String category = generateModifyDTO.getCategory();
         Long originalId = generateModifyDTO.getOriginalId();
         boolean isOverride = generateModifyDTO.getIsOverride();
-        boolean isFromLibrary = generateModifyDTO.getOriginalIdSource().equals("Library");
+        boolean isFromLibrary = !StringUtil.isNullOrEmpty(generateModifyDTO.getOriginalIdSource())
+                && generateModifyDTO.getOriginalIdSource().equals("Library");
 
         // 获取原始路径和可能的generateId
         PathInfo pathInfo = getOriginalPathAndGenerateId(isFromLibrary, originalId);