promptflow_tracing_decorator_openai_function_quickstart.py

1import openai
2from promptflow.tracing import start_trace, trace
3
4# Initialize the trace
5start_trace()
6
7@trace
8def call_openai(prompt):
9    client = openai.OpenAI()
10    response = client.chat.completions.create(
11        model="gpt-3.5-turbo",
12        messages=[{"role": "user", "content": prompt}]
13    )
14    return response.choices[0].message.content
15
16if __name__ == "__main__":
17    # This will be captured in the trace
18    result = call_openai("Hello, how are you?")
19    print(result)