OpenAI Agents SDK Integration with Veritrix Tracing

from openai import OpenAI
from veritrix import trace

trace.init("support-agent")

client = OpenAI()

def get_order_status(order_id: str) -> str:
    # Simulate a tool call
    return f"Order {order_id} is currently being processed."

tools = [
    {
        "type": "function",
        "function": {
            "name": "get_order_status",
            "description": "Returns the current status of a customer order.",
            "parameters": {
                "type": "object",
                "properties": {
                    "order_id": {"type": "string", "description": "The order ID to look up."}
                },
                "required": ["order_id"],
            },
        },
    }
]

response = client.chat.completions.create(
    model="gpt-4o",
    messages=[
        {"role": "system", "content": "You are a helpful customer support agent."},
        {"role": "user", "content": "What is the status of order #12345?"},
    ],
    tools=tools,
)

print(response.choices[0].message.content)