Sub-pipelines (list mode)

import vectorshift

from vectorshift import Pipeline
from vectorshift.pipeline import InputNode, OutputNode, PipelineNode, SplitTextNode

vectorshift.api_key = 'your api key '


sub_pipeline = Pipeline.fetch(name="your sub pipeline")

print(sub_pipeline)

input_node = InputNode(node_name="input_0")

split_text_node = SplitTextNode(
    node_name="split_text_node", text=input_node.text, delimiter="newline"
)

pipeline_node = PipelineNode(
    pipeline_id=sub_pipeline.id,
    node_name="sub_pipeline",
    input_0=split_text_node.processed_text,
    execution_mode="batch",
)

output_node = OutputNode(node_name="output_0", value=pipeline_node.output_0)

PIPELINE_NAME = "batched-pipeline"
try:
    main_pipeline = Pipeline.fetch(name=PIPELINE_NAME)
    print(
        f"Pipeline fetched: id={main_pipeline.id}, branch_id={main_pipeline.branch_id}"
    )
except Exception as e:
    print(f"Error fetching pipeline: {e}")
    main_pipeline = Pipeline.new(
        name=PIPELINE_NAME,
        nodes=[input_node, split_text_node, pipeline_node, output_node],
    )
    print(
        f"Pipeline created: id={main_pipeline.id}, branch_id={main_pipeline.branch_id}"
    )