Execute and inspect Lakehouse statements

from thinkplatform.sdk import client_from_profile

with client_from_profile("work") as client:
    statement = client.lakehouse.statements.execute(
        "SELECT 1 AS example", wait_timeout="0s"
    )
    print(statement.statement_id, statement.status.state)

import os
from thinkplatform.sdk import client_from_profile

with client_from_profile("work") as client:
    statement = client.lakehouse.statements.get(
        os.environ["THINKPLATFORM_STATEMENT_ID"]
    )
    print(statement.status.state)
    if statement.status.state == "SUCCEEDED":
        if statement.result is not None:
            print(statement.result.data_array)
            print("Next chunk:", statement.result.next_chunk_index)
    elif statement.status.error is not None:
        print(statement.status.error)