isyuricunha/dbeaver-proxy-to-mistral

dbeaver-proxy-to-mistral

镜像概述

这是一个轻量级FastAPI代理，旨在让DBeaver CE（及其他OpenAI兼容客户端）能够与Mistral API协同工作。DBeaver的AI助手使用类OpenAI的API格式，但需要特定的JSON字段和SSE事件类型，该代理解决了格式转换问题，确保双方正常通信。

核心功能

• 暴露OpenAI兼容端点：/responses、/models、 legacy /chat/completions（含/v1/*别名）；
• 转换请求格式：将DBeaver使用的POST /responses请求转换为Mistral的POST /chat/completions；
• 转换响应格式：将Mistral响应转换为DBeaver可解析的结构，避免空指针异常；
• 支持流式与非流式响应：兼容Server-Sent Events（SSE）；
• 灵活配置：通过环境变量设置，可作为systemd服务运行。

使用场景

• DBeaver CE用户：利用Mistral API实现SQL生成、查询优化等AI功能；
• OpenAI兼容客户端：需要连接Mistral API的其他工具或应用。

配置说明

环境变量

必填项

• MISTRAL_API_KEY：Mistral API密钥

可选项

• MISTRAL_BASE_URL：Mistral API基础地址，默认https://api.mistral.ai/v1；
• MISTRAL_MODEL：默认使用的模型，默认mistral-large-latest；
• MISTRAL_MODELS：代理对外展示的模型列表（逗号分隔），未设置则使用MISTRAL_MODEL；
• HOST：监听地址，默认0.0.0.0；
• PORT：监听端口，默认60916；
• REQUEST_TIMEOUT_SECONDS：请求超时时间，默认60秒。

注意事项

• GET /models接口无需API密钥即可访问；
• 调用Mistral API的路由（/responses、/chat/completions）若缺少MISTRAL_API_KEY会返回401错误。

Docker部署示例

构建镜像

bash
docker build -t dbeaver-proxy-to-mistral:local .

运行容器

bash
docker run --rm -p 60916:60916 \
  -e MISTRAL_API_KEY="你的Mistral API密钥" \
  -e MISTRAL_MODEL="mistral-large-latest" \
  docker.xuanyuan.run/dbeaver-proxy-to-mistral:local

Docker Compose

若已存在.env文件（含MISTRAL_API_KEY等配置）：

bash
docker compose up --build

DBeaver使用说明

在DBeaver CE中：

1. 设置OpenAI端点为http://<你的服务器地址>:60916/v1/；
2. 填写任意token值（DBeaver要求必填，但代理实际使用服务器环境中的MISTRAL_API_KEY）。