diff --git a/.DS_Store b/.DS_Store
index 651e36ed..7489d923 100644
Binary files a/.DS_Store and b/.DS_Store differ
diff --git a/lightrag/llm.py b/lightrag/llm.py
index 4df835b0..591b5dc9 100644
--- a/lightrag/llm.py
+++ b/lightrag/llm.py
@@ -2,6 +2,7 @@ import base64
 import copy
 import json
 import os
+import re
 import struct
 from functools import lru_cache
 from typing import List, Dict, Callable, Any, Union, Optional