Notebook to query Synapse Serverless SQL

Here is what worked in the end for me, maybe this helps someone else.

def query_synapse(db, query_string):
    """
        Function to query synapse 
    """
    # Synapse serverless SQL endpoint
    server = "<yourserver>.sql.azuresynapse.net"
    database = "<yourdb>"
    driver = "ODBC Driver 18 for SQL Server"

    connection_string = f"""
        Driver={driver};
        Server={server};
        Database={database};
    """
    token = notebookutils.credentials.getToken('https://database.windows.net/').encode("UTF-16-LE")
    token_struct = struct.pack(f"<I{len(token)}s", len(token), token)
    conn = pyodbc.connect(connection_string, attrs_before={1256:token_struct})
        
    cursor = conn.cursor()
    cursor.execute(query_string)

    # Get column names
    columns = [column[0] for column in cursor.description]

    # Fetch data
    rows = cursor.fetchall()

    # Convert to DataFrame
    df = pd.DataFrame.from_records(rows, columns=columns)

    return df