Add wrapper to allow iterating over formats() in Python SDK

Author: jakubno

python/e2b_code_interpreter/models.py

@@ -79,6 +79,12 @@ def __init__(self, is_main_result: bool, data: [MIMEType, str]):
         self.javascript = data.pop("application/javascript", None)
         self.extra = data
 
+    # Allows to iterate over formats()
+    def __getitem__(self, item):
+        if item in self.raw:
+            return self.raw[item]
+        return getattr(self, item)
+
     def formats(self) -> Iterable[str]:
         """
         Returns all available formats of the result.