The Braintrust AI proxy offers a unified way to access the world's leading AI models through a single API, including models from OpenAI, Anthropic, LLaMa 2, Mistral, and more. The benefits of using the proxy include:
- Code Simplification: Use a consistent API across different AI providers.
- Cost Reduction: The proxy automatically caches results, reusing them when possible.
- Enhanced Observability: Log requests automatically for better tracking and debugging. [Coming soon]
See the full list of supported models here. To read more about why we launched the AI proxy, check out our announcement blog post.
This repository contains the code for the proxy — both the underlying implementation and wrappers that allow you to deploy it on Vercel, Cloudflare, AWS Lambda, or an Express server.
You can communicate with the proxy via the standard OpenAI drivers/API, and simply set the base url to
https://braintrustproxy.com/v1
. Try running the following script in your favorite language, twice.
import { OpenAI } from "openai";
const client = new OpenAI({
baseURL: "https://braintrustproxy.com/v1",
apiKey: process.env.OPENAI_API_KEY, // Can use Braintrust, Anthropic, etc. keys here too
});
async function main() {
const start = performance.now();
const response = await client.chat.completions.create({
model: "gpt-3.5-turbo", // // Can use claude-2, llama-2-13b-chat here too
messages: [{ role: "user", content: "What is a proxy?" }],
seed: 1, // A seed activates the proxy's cache
});
console.log(response.choices[0].message.content);
console.log(`Took ${(performance.now() - start) / 1000}s`);
}
main();
from openai import OpenAI
import os
import time
client = OpenAI(
base_url="https://braintrustproxy.com/v1",
api_key=os.environ["OPENAI_API_KEY"], # Can use Braintrust, Anthropic, etc. keys here too
)
start = time.time()
response = client.chat.completions.create(
model="gpt-3.5-turbo", # Can use claude-2, llama-2-13b-chat here too
messages=[{"role": "user", "content": "What is a proxy?"}],
seed=1, # A seed activates the proxy's cache
)
print(response.choices[0].message.content)
print(f"Took {time.time()-start}s")
time curl -i https://braintrustproxy.com/v1/chat/completions \
-H "Content-Type: application/json" \
-d '{
"model": "gpt-3.5-turbo",
"messages": [
{
"role": "user",
"content": "What is a proxy?"
}
],
"seed": 1
}' \
-H "Authorization: Bearer $OPENAI_API_KEY" # Can use Braintrust, Anthropic, etc. keys here too
You can find the full documentation for using the proxy here.
The proxy is hosted for you, with end-to-end encryption, at https://braintrustproxy.com/v1
. However, you
can also deploy it yourself and customize its behavior.
To see docs for how to deploy on various platforms, see the READMEs in the corresponding folders:
To build the proxy, install pnpm and run:
pnpm install
pnpm build