OpenAI

import cognisafe
from openai import OpenAI

cognisafe.configure(
    api_key="csk_your_key_here",
    project_id="my-app",
)
cognisafe.patch_openai()

client = OpenAI()
# All calls through this client are now logged and scored by Cognisafe.

response = client.chat.completions.create(
    model="gpt-4o",
    messages=[
        {"role": "system", "content": "You are a helpful assistant."},
        {"role": "user", "content": "What is the capital of France?"},
    ],
)

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

# Proxy environment variable
UPSTREAM_URL=https://your-resource.openai.azure.com

import cognisafe
from openai import AzureOpenAI

cognisafe.configure(api_key="csk_...", project_id="my-app")
cognisafe.patch_openai()

client = AzureOpenAI(
    azure_endpoint=cognisafe.get_proxy_url(),  # points to Cognisafe proxy
    api_version="2024-02-01",
)

response = client.chat.completions.create(
    model="gpt-4o-deployment",
    messages=[{"role": "user", "content": "Hello."}],
)

import httpx

# Direct HTTP — set base_url to the proxy
response = httpx.post(
    "http://localhost:8080/v1/chat/completions",
    headers={"Authorization": "Bearer sk-your-openai-key"},
    json={
        "model": "gpt-4o",
        "messages": [{"role": "user", "content": "Hello."}],
    },
)