promptflow_sdk_chat_flow_load_and_batch_test.py

1import os
2from promptflow.core import Flow
3from promptflow.connections import AzureOpenAIConnection, OpenAIConnection
4
5# 1. Create a connection (Assuming you have an OpenAI account)
6# If using Azure OpenAI, use AzureOpenAIConnection instead
7conn = OpenAIConnection(
8    name="open_ai_connection", 
9    api_key=os.environ.get("OPENAI_API_KEY")
10)
11
12# 2. Define a simple flow from a yaml file or direct code
13# For this example, we assume a flow directory 'basic-chat' exists with flow.dag.yaml
14# You can initialize a sample flow via: pf flow init --flow ./basic-chat
15flow_path = "basic-chat"
16
17# 3. Test the flow
18f = Flow.load(source=flow_path)
19result = f(
20    question="What is Prompt Flow?",
21    # Add other inputs required by your flow
22)
23
24print(f"Flow execution result: {result}")
25
26# 4. Run a batch test with data
27data_path = "data.jsonl" # Path to your test data
28run = f.test(data=data_path)
29print(f"Batch run completed: {run}")