mirror of
https://github.com/snailyp/gemini-balance.git
synced 2026-05-12 02:19:59 +08:00
83ce50975a
新增与 OpenAI 规范兼容的 API 端点： - `/openai/v1/models` - `/openai/v1/chat/completions` (支持流式传输、重试和密钥切换) - `/openai/v1/embeddings` - `/openai/v1/images/generations` 包含： - 在 `app/router/openai_compatiable_routes.py` 中新增路由。 - `OpenAICompatiableService` 用于处理请求逻辑、日志记录和错误管理。 - 更新 `OpenaiApiClient` 以支持新方法和代理使用。 - 修改 `app/domain/openai_models.py` 以实现兼容性。 - 为新 API 添加专用日志记录器 (`openai_compatible`)。 - 为新路由 (`/openai`, `/api/version/check`) 添加认证中间件豁免。 增强配置编辑器 UI： - 在 `app/static/js/config_editor.js` 和 `app/templates/config_editor.html` 中添加批量代理删除功能。
76 lines
2.5 KiB
Python
76 lines
2.5 KiB
Python
"""
|
|
中间件配置模块，负责设置和配置应用程序的中间件
|
|
"""
|
|
|
|
from fastapi import FastAPI, Request
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from fastapi.responses import RedirectResponse
|
|
from starlette.middleware.base import BaseHTTPMiddleware
|
|
|
|
# from app.middleware.request_logging_middleware import RequestLoggingMiddleware
|
|
from app.core.constants import API_VERSION
|
|
from app.core.security import verify_auth_token
|
|
from app.log.logger import get_middleware_logger
|
|
|
|
logger = get_middleware_logger()
|
|
|
|
|
|
class AuthMiddleware(BaseHTTPMiddleware):
|
|
"""
|
|
认证中间件，处理未经身份验证的请求
|
|
"""
|
|
|
|
async def dispatch(self, request: Request, call_next):
|
|
# 允许特定路径绕过身份验证
|
|
if (
|
|
request.url.path not in ["/", "/auth"]
|
|
and not request.url.path.startswith("/static")
|
|
and not request.url.path.startswith("/gemini")
|
|
and not request.url.path.startswith("/v1")
|
|
and not request.url.path.startswith(f"/{API_VERSION}")
|
|
and not request.url.path.startswith("/health")
|
|
and not request.url.path.startswith("/hf")
|
|
and not request.url.path.startswith("/openai")
|
|
and not request.url.path.startswith("/api/version/check")
|
|
):
|
|
|
|
auth_token = request.cookies.get("auth_token")
|
|
if not auth_token or not verify_auth_token(auth_token):
|
|
logger.warning(f"Unauthorized access attempt to {request.url.path}")
|
|
return RedirectResponse(url="/")
|
|
logger.debug("Request authenticated successfully")
|
|
|
|
response = await call_next(request)
|
|
return response
|
|
|
|
|
|
def setup_middlewares(app: FastAPI) -> None:
|
|
"""
|
|
设置应用程序的中间件
|
|
|
|
Args:
|
|
app: FastAPI应用程序实例
|
|
"""
|
|
# 添加认证中间件
|
|
app.add_middleware(AuthMiddleware)
|
|
|
|
# 添加请求日志中间件（可选，默认注释掉）
|
|
# app.add_middleware(RequestLoggingMiddleware)
|
|
|
|
# 配置CORS中间件
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=["*"], # 生产环境建议配置具体的域名
|
|
allow_credentials=True,
|
|
allow_methods=[
|
|
"GET",
|
|
"POST",
|
|
"PUT",
|
|
"DELETE",
|
|
"OPTIONS",
|
|
], # 明确指定允许的HTTP方法
|
|
allow_headers=["*"], # 生产环境建议配置具体的请求头
|
|
expose_headers=["*"], # 允许前端访问的响应头
|
|
max_age=600, # 预检请求缓存时间(秒)
|
|
)
|