Blame for clients/python/openai.py

"""OpenAI client - chat, embeddings, assistants, tools."""
import os
import requests

BASE = os.getenv("OPENAI_API_BASE", "https://api.openai.com/v1")
KEY = os.getenv("OPENAI_API_KEY")


def _h():
    if not KEY:
        raise RuntimeError("OPENAI_API_KEY not set")
    return {"Authorization": f"Bearer {KEY}"}


def chat(messages, model="gpt-4o", stream=False, tools=None, **kwargs):
    body = {"model": model, "messages": messages, "stream": stream}
    if tools:
        body["tools"] = tools
    body.update(kwargs)
    return requests.post(f"{BASE}/chat/completions", headers=_h(), json=body, timeout=120, stream=stream)


def embeddings(inputs, model="text-embedding-3-small"):
    return requests.post(f"{BASE}/embeddings", headers=_h(), json={"model": model, "input": inputs}, timeout=120).json()


def create_assistant(**kwargs):
    return requests.post(f"{BASE}/assistants", headers=_h(), json=kwargs, timeout=60).json()


def create_run(thread_id, assistant_id, **kwargs):
    return requests.post(f"{BASE}/threads/{thread_id}/runs", headers=_h(), json={"assistant_id": assistant_id, **kwargs}, timeout=60).json()