promptflow_sdk_flow_init_test_and_bulk_run.py

1import promptflow as pf
2from promptflow.entities import Flow
3
4# 1. Initialize a flow from a local directory
5# Note: Ensure you have a valid flow directory with 'flow.dag.yaml'
6flow_path = "path/to/your/flow_directory"
7
8# 2. Test the flow with sample inputs
9# This executes the flow locally using the provided data
10inputs = {"question": "What is the capital of France?"}
11result = pf.PFClient().test(flow=flow_path, inputs=inputs)
12print(f"Flow test result: {result}")
13
14# 3. Create a run with a data file (bulk test)
15# The data file can be a .jsonl file containing multiple test cases
16data_path = "path/to/your/data.jsonl"
17run = pf.PFClient().run(
18    flow=flow_path,
19    data=data_path,
20    column_mapping={"question": "${data.question}"}
21)
22print(f"Run created: {run}")
23
24# 4. Stream the run logs to the console
25pf.PFClient().runs.stream(run.name)
26
27# 5. Get the run results as a dataframe
28details = pf.PFClient().runs.get_details(run.name)
29print(details.head())