import os

import kfp
from kfp.v2 import dsl
from kfp.v2.dsl import component


@component
def do_a_thing(input: str):
    import os

    print(f"--->>> {input} {os.getenv('MY_ENV_VAR')}")


@dsl.pipeline(name="A name", description="A description")
def my_pipeline(start_with_this: str):
    the_thing = do_a_thing(start_with_this).set_env_variable(
        "MY_ENV_VAR", os.getenv("ENV_VAR_FROM_BUILD")
    )


def compile_pipeline():
    print(f"--->>> {os.getenv('ENV_VAR_FROM_BUILD')}")
    kfp.compiler.Compiler(mode=kfp.dsl.PipelineExecutionMode.V2_COMPATIBLE).compile(
        pipeline_func=my_pipeline, package_path="my_pipeline.yaml"
    )


if __name__ == "__main__":
    compile_pipeline()