No column-level lineage for T-SQL MERGE statement

[tsitsimis]

Describe the bug
Column-level lineage does not identify tables within sub-queries used in the USING part of a MERGE statement (T-SQL)

SQL

MERGE INTO s1.table_target AS T 
USING (
    SELECT
        src.key_col    AS key_col
        ,src.col1       AS col1
    FROM  s1.table_source AS  src
) AS S 

    ON T.key_col  = S.key_col
WHEN NOT MATCHED 
THEN INSERT (       
    col1
)
VALUES (
    S.col1  
)

To Reproduce
Web UI (Web User Interface):

Expected behavior
The graph should also show lineage from S (alias) to s1.table_target

[reata]

What's the sqllineage version you're using? The latest v1.5.1 release works fine for this case.

[tsitsimis]

It works! Thank you!

[tsitsimis]

Re-opening since while it works for ansi dialect, it does not for tsql and I'm wondering if this is expected. (sqllineage v1.5.1)

result = LineageRunner(query, dialect="tsql")
print(result.get_column_lineage())

[reata]

Not expected. We need to fix this.