Aether/_deprecated_py_src/api/public/claude.py

"""
Claude API 端点

- /v1/messages - Claude Messages API
- /v1/messages/count_tokens - Token Count API

注意: /v1/models 端点由 models.py 统一处理，根据请求头返回对应格式
"""

from typing import Any

from fastapi import APIRouter, Depends, Request
from sqlalchemy.orm import Session

from src.api.base.pipeline import get_pipeline
from src.database import get_db

router = APIRouter(tags=["Claude API"])
pipeline = get_pipeline()


async def _run_claude_route_shell(adapter: Any, http_request: Request, db: Session) -> Any:
    return await pipeline.run(
        adapter=adapter,
        http_request=http_request,
        db=db,
        mode=adapter.mode,
    )


@router.post("/v1/messages")
async def create_message(
    http_request: Request,
    db: Session = Depends(get_db),
) -> Any:
    """
    Claude Messages API

    兼容 Anthropic Claude Messages API 格式的代理接口。
    根据认证头自动在标准 API 和 Claude Code CLI 模式之间切换:
    - x-api-key -> Chat 模式
    - Authorization: Bearer -> CLI 模式

    **请求格式**:
    ```json
    {
        "model": "claude-sonnet-4-20250514",
        "max_tokens": 1024,
        "messages": [{"role": "user", "content": "Hello"}]
    }
    ```
    """
    from src.api.handlers.claude import build_claude_adapter

    adapter = build_claude_adapter(http_request)
    return await _run_claude_route_shell(adapter, http_request, db)


@router.post("/v1/messages/count_tokens")
async def count_tokens(
    http_request: Request,
    db: Session = Depends(get_db),
) -> Any:
    """
    Claude Token Count API

    计算消息的 Token 数量，用于预估请求成本。

    **认证方式**: x-api-key 请求头
    """
    from src.api.handlers.claude import ClaudeTokenCountAdapter

    adapter = ClaudeTokenCountAdapter()
    return await _run_claude_route_shell(adapter, http_request, db)