# Use HTTP gateway
with TensorZeroGateway.build_http(
    gateway_url="..."
) as client:
    response = client.inference(
        model_name="azure::gpt41mini",
        input={
            "messages": [
                {
                    "role": "user",
                    "content": "Write a haiku about the knowledge graph."
                }
            ],
        },
        credentials={"azure_ad_token": token}
    )
    print(response)

# Use embedded gateway
with TensorZeroGateway.build_embedded(
    clickhouse_url="http://chuser:chpassword@localhost:8123/tensorzero",  # optional: for observability
    config_file="config/tensorzero.toml",  # optional: for custom functions, models, metrics, etc.
) as client:
    response = client.inference(
        model_name="gpt41mini",
        input={
            "messages": [
                {
                    "role": "user",
                    "content": "Write a haiku about the knowledge graph."
                }
            ],
        },
        credentials={"azure_ad_token": token}
    )
    print(response)