feat: add custom encoder

GreptimeTeam · Dec 26, 2023 · 66eddf7 · 66eddf7
1 parent 4c649de
commit 66eddf7
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/greptimeai/utils/json_encoder.py b/src/greptimeai/utils/json_encoder.py
@@ -0,0 +1,9 @@
+import json
+from io import BufferedReader
+
+
+class CustomEncoder(json.JSONEncoder):
+    def default(self, obj):
+        if isinstance(obj, BufferedReader):
+            return obj.name
+        return super().default(obj)