promptflow_tracing_decorator_quickstart_with_openai_calls.py

1import openai
2from promptflow.tracing import start_trace, trace
3
4# 1. Start the trace (this will provide a local UI URL to view traces)
5start_trace()
6
7# 2. Use the @trace decorator to instrument your functions
8@trace
9def call_openai(prompt):
10    client = openai.OpenAI()
11    response = client.chat.completions.create(
12        model="gpt-3.5-turbo",
13        messages=[{"role": "user", "content": prompt}]
14    )
15    return response.choices[0].message.content
16
17@trace
18def my_ai_app(question):
19    # Tracing will capture the hierarchy of these calls
20    answer = call_openai(question)
21    return answer
22
23if __name__ == "__main__":
24    # 3. Run your code; the execution details will be captured automatically
25    result = my_ai_app("What is promptflow-tracing?")
26    print(result)