AirbyteLib: Treat error trace as logs (airbytehq#34771)

ollionorg · Feb 26, 2024 · 2f94c28 · 2f94c28
1 parent dcb5c21
commit 2f94c28
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/airbyte-lib/airbyte_lib/source.py b/airbyte-lib/airbyte_lib/source.py
@@ -19,6 +19,7 @@
     DestinationSyncMode,
     Status,
     SyncMode,
+    TraceType,
     Type,
 )
 
@@ -395,6 +396,8 @@ def _execute(self, args: list[str]) -> Iterator[AirbyteMessage]:
                     yield message
                     if message.type == Type.LOG:
                         self._add_to_logs(message.log.message)
+                    if message.type == Type.TRACE and message.trace.type == TraceType.ERROR:
+                        self._add_to_logs(message.trace.error.message)
                 except Exception:
                     self._add_to_logs(line)
         except Exception as e: