fix: state_type incorrect reference (#11)

MeltanoLabs · Feb 28, 2024 · fc42fbd · fc42fbd
1 parent 919385e
commit fc42fbd
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tap_airbyte/tap.py b/tap_airbyte/tap.py
@@ -654,13 +654,14 @@ def sync_all(self) -> None:  # type: ignore
                     self._process_log_message(airbyte_message)
                 elif airbyte_message["type"] == AirbyteMessage.STATE:
                     state_message = airbyte_message["state"]
+                    state_type = state_message["type"]
                     if "data" in state_message:
                         unpacked_state = state_message["data"]
-                    elif "type" == "STREAM":
+                    elif state_type == "STREAM":
                         unpacked_state = state_message["stream"]
-                    elif "type" == "GLOBAL":
+                    elif state_type == "GLOBAL":
                         unpacked_state = state_message["global"]
-                    elif "type" == "LEGACY":
+                    elif state_type == "LEGACY":
                         unpacked_state = state_message["legacy"]
                     self.airbyte_state = unpacked_state
                     with STDOUT_LOCK: