Handle list of custom type tool when deserialising

chapter-gtm · Aug 15, 2024 · 4eaecaa · 4eaecaa
1 parent 05d20da
commit 4eaecaa
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/app/db/models/custom_types.py b/src/app/db/models/custom_types.py
@@ -100,4 +100,11 @@ def process_result_value(self, value, dialect):
             obj = Tool.from_dict(value)
             obj.certainty = Scale(obj.certainty) if obj.certainty else Scale.LOW
             return obj
+        elif value and isinstance(value, list):
+            objs = []
+            for item in value:
+                obj = Tool.from_dict(item)
+                obj.certainty = Scale(obj.certainty) if obj.certainty else Scale.LOW
+                objs.append(obj)
+            return objs
         return None