AI-Gateway:一款整合了OpenAI、Anthropic、LLama2等大语言模型的统一API接口

关于AI-Gateway

AI-Gateway是一款针对大语言模型的统一API接口,该接口可以用在应用程序和托管的大语言模型(LLM)之间,该工具可以允许我们通过一个统一的API接口将API请求转发给OpenAI、Anthropic、Mistral、LLama2、Anyscale、Google Gemini等大语言模型。

功能特性

1、功能非常快,体积非常小;

2、跨多模型和提供商实现负载均衡;

3、支持回退以确保应用程序保持弹性;

4、默认配置下,支持自动重试;

5、可根据需要安装中间件;

6、支持多种大语言模型;

支持的SDK

|-------------------|--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
| 编程语言 | 支持的SDK |
| Node.js / JS / TS | Portkey SDK OpenAI SDK LangchainJS LlamaIndex.TS |
| Python | Portkey SDK OpenAI SDK Langchain LlamaIndex |
| Go | go-openai |
| Java | openai-java |
| Rust | async-openai |
| Ruby | ruby-openai |

工具安装&部署

本地部署

如果你熟悉Node.js和npx的话,可以直接在本地安装和部署你自己的AI-Gateway:

复制代码
npx @portkey-ai/gateway

安装完成后,就可以打开浏览器并访问下列地址使用AI-Gateway了:

复制代码
http://localhost:8787

Cloudflare Worker部署

使用下列命令将该项目源码克隆至本地:

复制代码
git clone https://github.com/portkey-ai/gateway

然后安装NPM依赖:

复制代码
cd gateway

npm i

使用wrangler部署即可:

复制代码
npm run deploy

Node.js部署

使用下列命令将该项目源码克隆至本地:

复制代码
git clone https://github.com/portkey-ai/gateway

然后安装NPM依赖:

复制代码
cd gateway

npm i

npm run build

然后运行Node服务器:

复制代码
node build/start-server.js

Docker部署

广大研究人员也可以直接使用Docker来运行AI-Gateway:

复制代码
docker run -d -p 8787:8787 portkeyai/gateway:latest

Docker-Compose部署

使用下列命令下载Compose文件:

复制代码
wget "https://raw.githubusercontent.com/Portkey-AI/gateway/main/docker-compose.yaml"

然后运行Docker-Compose即可:

复制代码
docker compose up -d

服务当前会运行并监听8787端口。

AI-Gateway配置

AI-Gateway支持通过自定义配置来实现多功能路由策略,例如回退、负载均衡和自动重连等。

我们可以通过x-portkey-config Header在调用OpenAI时应用下列配置信息:

复制代码
// 使用OpenAI JS SDK时

const client = new OpenAI({

  baseURL: "http://127.0.0.1:8787", // The gateway URL

  defaultHeaders: {

    'x-portkey-config': {.. your config here ..},

  }

});

下面给出的样例配置会在回退至Gemini Pro之前尝试向OpenAI发送5次请求:

复制代码
{

  "retry": { "count": 5 },

  "strategy": { "mode": "fallback" },

  "targets": [{

      "provider": "openai",

      "api_key": "sk-***"

    },{

      "provider": "google",

      "api_key": "gt5***",

      "override_params": {"model": "gemini-pro"}

  }]

}

下面的配置将启动负载均衡功能:

复制代码
{

  "strategy": { "mode": "loadbalance" },

  "targets": [{

      "provider": "openai",

      "api_key": "sk-***",

      "weight": "0.5"

    },{

      "provider": "openai",

      "api_key": "sk-***",

      "weight": "0.5"

    }

  ]

}

工具使用

下面的例子中,我们将通过AI-Gateway尝试向OpenAI发送一个Chat completions调用:

复制代码
curl '127.0.0.1:8787/v1/chat/completions' \

  -H 'x-portkey-provider: openai' \

  -H "Authorization: Bearer $OPENAI_KEY" \

  -H 'Content-Type: application/json' \

  -d '{"messages": [{"role": "user","content": "Say this is test."}], "max_tokens": 20, "model": "gpt-4"}'

许可证协议

本项目的开发与发布遵循MIT开源许可证协议。

项目地址

AI-Gateway :【GitHub传送门

参考资料

AI Gateway - Simplify Your LLM Integrations | Portkey

https://hub.docker.com/r/portkeyai/gateway

相关推荐
Debroon1 分钟前
RuleAlign 规则对齐框架:将医生的诊断规则形式化并注入模型,无需额外人工标注的自动对齐方法
人工智能
羊小猪~~8 分钟前
神经网络基础--什么是正向传播??什么是方向传播??
人工智能·pytorch·python·深度学习·神经网络·算法·机器学习
AI小杨9 分钟前
【车道线检测】一、传统车道线检测:基于霍夫变换的车道线检测史诗级详细教程
人工智能·opencv·计算机视觉·霍夫变换·车道线检测
晨曦_子画14 分钟前
编程语言之战:AI 之后的 Kotlin 与 Java
android·java·开发语言·人工智能·kotlin
道可云16 分钟前
道可云人工智能&元宇宙每日资讯|2024国际虚拟现实创新大会将在青岛举办
大数据·人工智能·3d·机器人·ar·vr
牛角上的男孩16 分钟前
Istio Gateway发布服务
云原生·gateway·istio
Yawesh_best23 分钟前
思源笔记轻松连接本地Ollama大语言模型,开启AI写作新体验!
笔记·语言模型·ai写作
人工智能培训咨询叶梓25 分钟前
探索开放资源上指令微调语言模型的现状
人工智能·语言模型·自然语言处理·性能优化·调优·大模型微调·指令微调
zzZ_CMing25 分钟前
大语言模型训练的全过程:预训练、微调、RLHF
人工智能·自然语言处理·aigc
newxtc26 分钟前
【旷视科技-注册/登录安全分析报告】
人工智能·科技·安全·ddddocr