BLDX-1076: Add GENERIC_OPENLINEAGE connector type and make emit() accept connector_type

Add GENERIC_OPENLINEAGE = ("generic-openlineage", ELT) to AtlanConnectorType
so PyAtlan can target the marketplace generic-openlineage (GOLC) endpoint at
/events/openlineage/generic-openlineage/api/v1/lineage.

Make OpenLineageEvent.emit() / emit_async() accept an optional connector_type
kwarg (defaulting to SPARK for backwards compatibility), mirroring the
existing pattern on emit_raw()/emit_raw_async(). Applied to both the current
and pyatlan_v9 copies of event.py.

Unit tests cover:
- GENERIC_OPENLINEAGE enum value
- Routing to /events/openlineage/generic-openlineage/api/v1/lineage
- emit() honors the connector_type kwarg
- emit() default is still SPARK

Author: mothership-atlan

pyatlan/model/enums.py

@@ -272,6 +272,7 @@ def get_connector_name(
     SYNAPSE = ("synapse", AtlanConnectionCategory.WAREHOUSE)
     AIRFLOW = ("airflow", AtlanConnectionCategory.ELT)
     OPENLINEAGE = ("openlineage", AtlanConnectionCategory.ELT)
+    GENERIC_OPENLINEAGE = ("generic-openlineage", AtlanConnectionCategory.ELT)
     DATAFLOW = ("dataflow", AtlanConnectionCategory.ELT)
     QLIKSENSE = ("qlik-sense", AtlanConnectionCategory.BI)
     KAFKA = ("kafka", AtlanConnectionCategory.EVENT_BUS)

pyatlan/model/open_lineage/event.py

@@ -108,26 +108,36 @@ def creator(
             event_time=datetime.now(tz=utc).isoformat(),  # type:ignore[call-arg]
         )
 
-    def emit(self, client: AtlanClient) -> None:
+    def emit(
+        self,
+        client: AtlanClient,
+        connector_type: AtlanConnectorType = AtlanConnectorType.SPARK,
+    ) -> None:
         """
         Send the OpenLineage event to Atlan to be processed.
 
         :param client: connectivity to an Atlan tenant
+        :param connector_type: connector type for the OpenLineage event
         :raises AtlanError: on any API communication issues
         """
         return client.open_lineage.send(
-            request=self, connector_type=AtlanConnectorType.SPARK
+            request=self, connector_type=connector_type
         )
 
-    async def emit_async(self, client: AsyncAtlanClient) -> None:
+    async def emit_async(
+        self,
+        client: AsyncAtlanClient,
+        connector_type: AtlanConnectorType = AtlanConnectorType.SPARK,
+    ) -> None:
         """
         Send the OpenLineage event to Atlan to be processed (async version).
 
         :param client: async connectivity to an Atlan tenant
+        :param connector_type: connector type for the OpenLineage event
         :raises AtlanError: on any API communication issues
         """
         return await client.open_lineage.send(
-            request=self, connector_type=AtlanConnectorType.SPARK
+            request=self, connector_type=connector_type
         )
 
     @classmethod

pyatlan_v9/model/open_lineage/event.py

@@ -118,26 +118,36 @@ def creator(
             event_time=datetime.now(tz=utc).isoformat(),
         )
 
-    def emit(self, client: AtlanClient) -> None:
+    def emit(
+        self,
+        client: AtlanClient,
+        connector_type: AtlanConnectorType = AtlanConnectorType.SPARK,
+    ) -> None:
         """
         Send the OpenLineage event to Atlan to be processed.
 
         :param client: connectivity to an Atlan tenant
+        :param connector_type: connector type for the OpenLineage event
         :raises AtlanError: on any API communication issues
         """
         return client.open_lineage.send(
-            request=self, connector_type=AtlanConnectorType.SPARK
+            request=self, connector_type=connector_type
         )
 
-    async def emit_async(self, client: "AsyncAtlanClient") -> None:
+    async def emit_async(
+        self,
+        client: "AsyncAtlanClient",
+        connector_type: AtlanConnectorType = AtlanConnectorType.SPARK,
+    ) -> None:
         """
         Asynchronously send the OpenLineage event to Atlan to be processed.
 
         :param client: async connectivity to an Atlan tenant
+        :param connector_type: connector type for the OpenLineage event
         :raises AtlanError: on any API communication issues
         """
         return await client.open_lineage.send(
-            request=self, connector_type=AtlanConnectorType.SPARK
+            request=self, connector_type=connector_type
         )
 
     @classmethod

tests/unit/model/open_lineage/open_lineage_test.py

@@ -152,6 +152,54 @@ def test_ol_client_send(
     mock_api_caller.reset_mock()
 
 
+def test_ol_generic_openlineage_enum_value():
+    assert AtlanConnectorType.GENERIC_OPENLINEAGE.value == "generic-openlineage"
+
+
+def test_ol_client_send_generic_openlineage_routes_to_endpoint(mock_api_caller):
+    mock_api_caller._call_api.return_value = "Event received"
+    OpenLineageClient(client=mock_api_caller).send(
+        request=OpenLineageEvent(),
+        connector_type=AtlanConnectorType.GENERIC_OPENLINEAGE,
+    )
+
+    assert mock_api_caller._call_api.call_count == 1
+    api_arg = mock_api_caller._call_api.call_args.kwargs["api"]
+    assert api_arg.path == "generic-openlineage/api/v1/lineage"
+    mock_api_caller.reset_mock()
+
+
+def test_ol_event_emit_accepts_connector_type(mock_api_caller):
+    mock_api_caller._call_api.return_value = "Event received"
+    client = AtlanClient()
+    with patch.object(
+        client.open_lineage, "send", return_value=None
+    ) as mock_send:
+        OpenLineageEvent().emit(
+            client=client,
+            connector_type=AtlanConnectorType.GENERIC_OPENLINEAGE,
+        )
+        mock_send.assert_called_once()
+        assert (
+            mock_send.call_args.kwargs["connector_type"]
+            == AtlanConnectorType.GENERIC_OPENLINEAGE
+        )
+
+
+def test_ol_event_emit_defaults_to_spark(mock_api_caller):
+    mock_api_caller._call_api.return_value = "Event received"
+    client = AtlanClient()
+    with patch.object(
+        client.open_lineage, "send", return_value=None
+    ) as mock_send:
+        OpenLineageEvent().emit(client=client)
+        mock_send.assert_called_once()
+        assert (
+            mock_send.call_args.kwargs["connector_type"]
+            == AtlanConnectorType.SPARK
+        )
+
+
 def test_ol_client_send_when_ol_not_configured(client, mock_session):
     expected_error = (
         "ATLAN-PYTHON-400-064 Requested OpenLineage "