Update helpers.py

helpers.py

@@ -146,15 +146,18 @@ key = os.getenv("GOOGLE_API_KEY")
 #         new_metadata = None
 #     return new_metadata
 
-def extract_metadata(response_string):
+def extract_metadata(response):
+    if not isinstance(response, str):
+        response = str(response)  # Chuyển sang string nếu cần
+
     # Tìm tất cả các dictionary trong chuỗi đầu vào
-    matches = re.findall(r'\{.*?\}', response_string, re.DOTALL)
+    matches = re.findall(r'\{.*?\}', response, re.DOTALL)
     if not matches:
-        return None
-    
+        return None  # Trả về None nếu không tìm thấy dict nào
+
     smallest_dict = None
     min_length = float("inf")
-    
+
     for match in matches:
         try:
             parsed_dict = ast.literal_eval(match)  # Chuyển đổi string thành dictionary
@@ -165,9 +168,15 @@ def extract_metadata(response_string):
                     min_length = dict_length
         except Exception:
             continue  # Bỏ qua nếu không phải dictionary hợp lệ
-    
+
     return smallest_dict
 
+# 🛠 **Ví dụ sử dụng**
+response = None  # Hoặc một object không phải string
+smallest_dict = extract_smallest_dict(response)
+print(smallest_dict)  # Output: None
+
+
 # # Example usage:
 # input_str = "Some random text before and then {'a': 'abc', 'b': 'bcd'} and some random text after."
 # metadata = extract_metadata(input_str)