mirror of
https://github.com/fawney19/Aether.git
synced 2026-09-03 01:40:21 +08:00
1d9c77522a
- 删除全部 Python 源码 (src/) 及 Alembic 迁移脚本，归档至 _deprecated_py_src/ - 重构 Rust gateway ai_pipeline: 拆分 planner/finalize 模块，新增 contracts/adaptation 层 - 重组 handlers 模块为 admin/public/proxy/internal/shared 子模块结构 - 新增 executor 模块，引入 Rust 原生数据库迁移 (aether-data/migrations) - 简化 CI/Docker 构建流程，移除 base image 二级构建，统一为单一 app image - 移除 Python 相关基础设施文件 (entrypoint.sh, gunicorn_conf.py, Dockerfile.base)
75 lines
1.9 KiB
Python
75 lines
1.9 KiB
Python
"""
|
|
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)
|