Initial commit

src/api/public/gemini.py

@@ -0,0 +1,130 @@
+"""
+Gemini API 专属端点
+
+托管 Gemini API 相关路由:
+- /v1beta/models/{model}:generateContent
+- /v1beta/models/{model}:streamGenerateContent
+
+注意: Gemini API 的 model 在 URL 路径中，而不是请求体中
+
+路径配置来源: src.core.api_format_metadata.APIFormat.GEMINI
+- path_prefix: 本站路径前缀（如 /gemini），通过 router prefix 配置
+- default_path: 标准 API 路径模板
+"""
+
+from fastapi import APIRouter, Depends, Request
+from sqlalchemy.orm import Session
+
+from src.api.base.pipeline import ApiRequestPipeline
+from src.api.handlers.gemini import build_gemini_adapter
+from src.api.handlers.gemini_cli import build_gemini_cli_adapter
+from src.core.api_format_metadata import get_api_format_definition
+from src.core.enums import APIFormat
+from src.database import get_db
+
+# 从配置获取路径前缀
+_gemini_def = get_api_format_definition(APIFormat.GEMINI)
+
+router = APIRouter(tags=["Gemini API"], prefix=_gemini_def.path_prefix)
+pipeline = ApiRequestPipeline()
+
+
+def _is_cli_request(request: Request) -> bool:
+    """
+    判断是否为 CLI 请求
+
+    检查顺序:
+    1. x-app header 包含 "cli"
+    2. user-agent 包含 "GeminiCLI" 或 "gemini-cli"
+    """
+    # 检查 x-app header
+    x_app = request.headers.get("x-app", "")
+    if "cli" in x_app.lower():
+        return True
+
+    # 检查 user-agent
+    user_agent = request.headers.get("user-agent", "")
+    user_agent_lower = user_agent.lower()
+    if "geminicli" in user_agent_lower or "gemini-cli" in user_agent_lower:
+        return True
+
+    return False
+
+
+@router.post("/v1beta/models/{model}:generateContent")
+async def generate_content(
+    model: str,
+    http_request: Request,
+    db: Session = Depends(get_db),
+):
+    """
+    Gemini generateContent 端点
+
+    非流式生成内容请求
+    """
+    # 根据 user-agent 或 x-app header 选择适配器
+    if _is_cli_request(http_request):
+        adapter = build_gemini_cli_adapter()
+    else:
+        adapter = build_gemini_adapter()
+
+    return await pipeline.run(
+        adapter=adapter,
+        http_request=http_request,
+        db=db,
+        mode=adapter.mode,
+        api_format_hint=adapter.allowed_api_formats[0],
+        # 将 model 注入到请求体中，stream 用于内部判断流式模式
+        path_params={"model": model, "stream": False},
+    )
+
+
+@router.post("/v1beta/models/{model}:streamGenerateContent")
+async def stream_generate_content(
+    model: str,
+    http_request: Request,
+    db: Session = Depends(get_db),
+):
+    """
+    Gemini streamGenerateContent 端点
+
+    流式生成内容请求
+
+    注意: Gemini API 通过 URL 端点区分流式/非流式，不需要在请求体中添加 stream 字段
+    """
+    # 根据 user-agent 或 x-app header 选择适配器
+    if _is_cli_request(http_request):
+        adapter = build_gemini_cli_adapter()
+    else:
+        adapter = build_gemini_adapter()
+
+    return await pipeline.run(
+        adapter=adapter,
+        http_request=http_request,
+        db=db,
+        mode=adapter.mode,
+        api_format_hint=adapter.allowed_api_formats[0],
+        # model 注入到请求体，stream 用于内部判断流式模式（不发送到 API）
+        path_params={"model": model, "stream": True},
+    )
+
+
+# 兼容 v1 路径（部分 SDK 可能使用）
+@router.post("/v1/models/{model}:generateContent")
+async def generate_content_v1(
+    model: str,
+    http_request: Request,
+    db: Session = Depends(get_db),
+):
+    """v1 兼容端点"""
+    return await generate_content(model, http_request, db)
+
+
+@router.post("/v1/models/{model}:streamGenerateContent")
+async def stream_generate_content_v1(
+    model: str,
+    http_request: Request,
+    db: Session = Depends(get_db),
+):
+    """v1 兼容端点"""
+    return await stream_generate_content(model, http_request, db)