HAGMI is a AI-Studio to Gemini Mimicking Interface
这是一个将AI Studio转换为Gemini API的项目
cp config.yaml.example config.yaml配置文件中已包含各配置项说明。
在服务启动时将尝试使用 config.yaml 中配置的登录凭证恢复登录状态，若未登录或登录状态已过期，将尝试使用帐号密码进行登录。
注：目前无法处理ReCaptcha或者2FA，如果无法使用自动登录，请选择方法2。
可以使用 scripts/create_state.py 脚本进行登录保存登录状态文件。脚本接受的参数如下
usage: create_state.py [-h] [--email [email protected]] [--password PASSWORD] [--remote http://127.0.0.1:8000]
登录AI Studio 并保存登录状态
options:
-h, --help show this help message and exit
--email [email protected]
Google 帐号自动填充 (default: None)
--password PASSWORD Google 帐号密码自动填充(可选) (default: None)
--remote http://127.0.0.1:8000
推送到远程HAGMI实例 (default: None)
若脚本提供了 email 与/或 password参数，脚本将尝试自动填充帐号密码进行登录，并由用户处理可能的额外验证(ReCaptcha或者2FA)。
如果 email 和 password参数未提供，则由用户完成整个登录流程。
在登录完成后，脚本将尝试从 AI Studio 获取帐号信息，若脚本未能获取帐号信息，将提示用户输入帐号名称，最后把保存登录状态至states/{email}.json。
如果提供了 remote 参数，脚本将尝试推送登录状态文件到远程 HAGMI 实例。
项目使用 uv 作为包管理， 安装方法参考 uv的安装文档
在第一次启动之前需要通过命令安装camoufox和playwright依赖
uv run playwright install firefox --with-deps
uv run camoufox fetch安装完依赖后，后续可以通过命令直接启动项目
uv run python app.pydocker-compose build
docker-compose up -dcurl "http://localhost:8000/v1beta/models/gemini-2.5-pro:streamGenerateContent" \
-H "x-goog-api-key: YOUR_API_KEY" \
-H 'Content-Type: application/json' \
-X POST \
-d '{
"contents": [
{
"parts": [
{
"text": "Hi."
}
]
},
{
"role": "model",
"parts": [
{
"text": "<think>"
}
]
}
],
"generationConfig": {
"thinkingConfig": {
"thinkingBudget": 0
}
}
}'sequenceDiagram
box Gemini
participant Client
participant FastAPI
participant Adapter
end
box AI Studio
participant AIStudio
participant Playwright
participant gRPC
end
Client->>FastAPI: 请求
activate FastAPI
FastAPI->>Adapter: 将GenAI格式请求转换成AI Studio 的 Prompt History
Adapter-->>FastAPI: prompt_history
FastAPI->>Playwright: 使用Promprt History 创建Playwright任务
loop 拦截并修改 AI Studio 的 gPRC 请求
activate AIStudio
Playwright->>AIStudio: 浏览器操作
AIStudio-->>AIStudio: BotGuard
AIStudio-xgRPC: 被拦截的gRPC请求
Playwright-->>AIStudio: 仿造的 gRPC 返回
end
deactivate AIStudio
Playwright->>FastAPI: 拦截生成内容 gRPC 请求(with BotGuard验证参数)
FastAPI->>gRPC: 使用 aiohttp 向 AI Studio 发送 gRPC 请求.
activate gRPC
loop 流式返回
gRPC-->>Adapter: 流式解析 gRPC 返回
Adapter-->>FastAPI: gRPC 流式返回
FastAPI->>Adapter: 将gRPC事件转换成GenAI格式
Adapter-->>FastAPI: GenAI格式事件
FastAPI-->>Client: GenAI格式事件
end
deactivate gRPC
deactivate FastAPI
- 概念验证
- 多模态输入输出
- 生成参数配置
- Function Calling
- Structured Output
- Gounding
- Url Context
- Code Execution
- AI Studio代理配置
- 预构建Docker镜像
- 详细配置文档
- 注入隐藏模型
- 添加OpenAI API支持
- 将 context <200k 的请求 offload 到 GenAI API
- Imagen 系列模型支持
- TTS 系列模型支持
- Gemini Flash Live 系列模型支持
- Gemini Flash Native Audio 系列模型支持
- Veo 系列模型支持
通过本项目使用AI Studio可能违反Google的服务条款，请自行承担风险。