From 0a0bfc02a077c546dc3a402e35a50b9da4677f72 Mon Sep 17 00:00:00 2001
From: Stephen Hu <stephenhu@seismic.com>
Date: Thu, 7 Aug 2025 11:07:29 +0800
Subject: [PATCH] Refactor:naive_merge_with_images close useless images (#9296)

### What problem does this PR solve?

naive_merge_with_images close useless images

### Type of change

- [x] Refactoring
---
 rag/nlp/__init__.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/rag/nlp/__init__.py b/rag/nlp/__init__.py
index e76624113..ddfc387bb 100644
--- a/rag/nlp/__init__.py
+++ b/rag/nlp/__init__.py
@@ -611,6 +611,10 @@ def naive_merge_with_images(texts, images, chunk_token_num=128, delimiter="\n。
                 if re.match(f"^{dels}$", sub_sec):
                     continue
                 add_chunk(sub_sec, image)
+        
+        for img in images:
+            if isinstance(img, Image.Image):
+                img.close()
 
     return cks, result_images