请注意: 本教程的部署方案需配合 CLIProxyAPI 使用。在开始之前,请确保你已有一个正在运行的 CLIProxyAPI 实例。

CLIProxyAPI 自 v6.3.x 版本起,开始支持通过 WebSocket 方式接入 AI Provider,并首个支持了 AIStudio。

然而,这种方式需要一个始终开启的浏览器来运行 AIStudioBuild 上的 WebSocket 通信程序,这总归有些不便。如果选择将其部署在 VPS 上,又会面临 VPS 内存要求较高的问题。

为了解决这个问题,我花了点时间尝试多种无头浏览器方案。最终,我选择使用 Docker 在 HuggingFace 上进行部署,此举能充分利用 HuggingFace 免费实例的大内存优势,实现零成本部署。

第一步:配置 AIStudioBuild 应用

需要根据你的 CLIProxyAPI 的设置,配置好 AIStudioBuild 上的 WebSocket 通信程序:打开官方提供的示例程序,复制该程序后修改图中红框处的两个地方。其中,如果 CLIProxyAPI 中设置了 wsauthtrue,那么就需要设置 JWT_TOKENCLIProxyAPI 中的拟用于鉴权的 api-keys 值;设置 WEBSOCKET_PROXY_URLCLIProxyAPI 所在的地址,例如:wss://mycap.example.com/v1/ws。设置完成后保存,并记录这个应用的链接备用。

多账户使用时,需要多操作一个步骤,将该应用访问权限设置为 Public

安全警告: 设置为 Public 后,请务必妥善保管你的链接。切勿将此链接公开分享,以免导致授权信息泄露。

Cookie 可以通过两种方式获取,两种方式选一种即可,推荐使用指纹浏览器方式获取:

第一种方式:使用 AdsPower 指纹浏览器,登录 https://aistudio.google.com/ ,退出后编辑浏览器环境,复制 Cookie 内容,具体如下图所示:

第二种方式:通过 Chrome 等普通浏览器的隐私模式,登录 https://aistudio.google.com/ ,在浏览器的开发者工具中复制 Cookie 内容,具体位置如下图所示:

第三步:部署 HuggingFace Space

打开 https://huggingface.co/spaces/hkfires/AIStudioBuildWS ,复制该 Space。在 CAMOUFOX_INSTANCE_URL 处填入第一步准备的程序的链接,在 USER_COOKIE_1 处填入第二步准备的 Cookie,点击 Duplicate Space。

等待 HuggingFace 构建完成,出现如下日志,即部署成功:

如果有多个账户,参考 USER_COOKIE_1,在 HuggingFace Space 的设置中依次增加 USER_COOKIE_2USER_COOKIE_3 等环境变量即可。

重要提醒: Cookie 属于敏感信息,请务必使用 “Secrets” (而不是 “Variables”) 来存储,以防止 Cookie 外泄。

部署成功后,我们应该在 CLIProxyAPI 的中看到类似如下的日志。至此,整个部署全部完成。

其他

当然,如果不想使用 HuggingFace,项目也支持本地部署:https://github.com/hkfires/AIStudioBuildWS/

参考项目

https://github.com/cliouo/aistudio-build-proxy-all