# pip install promptlayer
from openai import OpenAI
from promptlayer import PromptLayer
import time
pl_client = PromptLayer()
client = OpenAI()
messages = [{"role": "user", "content": "Hello!"}]
start = time.time()
completion = client.chat.completions.create(model="gpt-4o", messages=messages)
end = time.time()
pl_client.log_request(
provider="openai", model="gpt-4o",
input={"type": "chat", "messages": [{"role": m["role"], "content": [{"type": "text", "text": m["content"]}]} for m in messages]},
output={"type": "chat", "messages": [{"role": "assistant", "content": [{"type": "text", "text": completion.choices[0].message.content}]}]},
request_start_time=start, request_end_time=end
)