fix type

run-llama · Dec 20, 2024 · c7d2166 · c7d2166
1 parent 7d81f9d
commit c7d2166
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/llama-index-core/llama_index/core/schema.py b/llama-index-core/llama_index/core/schema.py
@@ -969,7 +969,8 @@ def custom_model_dump(
     ) -> Dict[str, Any]:
         """For full backward compatibility with the text field, we customize the model serializer."""
         data = super().custom_model_dump(handler, info)
-        if "text" not in info.exclude:
+        exclude_set = set(info.exclude or [])
+        if "text" not in exclude_set:
             data["text"] = self.text
         return data