mirror of
https://github.com/fawney19/Aether.git
synced 2026-01-12 04:28:28 +08:00
35e29d46bd
- 新增 src/services/billing/ 模块，包含计费计算器、模板和使用量映射 - 将 ChatAdapterBase 和 CliAdapterBase 中的计费逻辑重构为调用 billing 模块 - 为每个 adapter 添加 BILLING_TEMPLATE 类属性，指定计费模板 - 支持 Claude/OpenAI/Gemini 三种计费模板，支持阶梯计费和缓存 TTL 定价 - 新增 tests/services/billing/ 单元测试
106 lines
3.6 KiB
Python
106 lines
3.6 KiB
Python
"""
|
||
OpenAI CLI Adapter - 基于通用 CLI Adapter 基类的简化实现
|
||
|
||
继承 CliAdapterBase，只需配置 FORMAT_ID 和 HANDLER_CLASS。
|
||
"""
|
||
|
||
from typing import Any, AsyncIterator, Dict, Optional, Tuple, Type, Union
|
||
|
||
import httpx
|
||
from fastapi import Request
|
||
|
||
from src.api.handlers.base.cli_adapter_base import CliAdapterBase, register_cli_adapter
|
||
from src.api.handlers.base.cli_handler_base import CliMessageHandlerBase
|
||
from src.api.handlers.openai.adapter import OpenAIChatAdapter
|
||
from src.config.settings import config
|
||
|
||
|
||
@register_cli_adapter
|
||
class OpenAICliAdapter(CliAdapterBase):
|
||
"""
|
||
OpenAI CLI API 适配器
|
||
|
||
处理 /v1/responses 端点的请求。
|
||
"""
|
||
|
||
FORMAT_ID = "OPENAI_CLI"
|
||
BILLING_TEMPLATE = "openai" # 使用 OpenAI 计费模板
|
||
name = "openai.cli"
|
||
|
||
@property
|
||
def HANDLER_CLASS(self) -> Type[CliMessageHandlerBase]:
|
||
"""延迟导入 Handler 类避免循环依赖"""
|
||
from src.api.handlers.openai_cli.handler import OpenAICliMessageHandler
|
||
|
||
return OpenAICliMessageHandler
|
||
|
||
def __init__(self, allowed_api_formats: Optional[list[str]] = None):
|
||
super().__init__(allowed_api_formats or ["OPENAI_CLI"])
|
||
|
||
def extract_api_key(self, request: Request) -> Optional[str]:
|
||
"""从请求中提取 API 密钥 (Authorization: Bearer)"""
|
||
authorization = request.headers.get("authorization")
|
||
if authorization and authorization.startswith("Bearer "):
|
||
return authorization.replace("Bearer ", "")
|
||
return None
|
||
|
||
# =========================================================================
|
||
# 模型列表查询
|
||
# =========================================================================
|
||
|
||
@classmethod
|
||
async def fetch_models(
|
||
cls,
|
||
client: httpx.AsyncClient,
|
||
base_url: str,
|
||
api_key: str,
|
||
extra_headers: Optional[Dict[str, str]] = None,
|
||
) -> Tuple[list, Optional[str]]:
|
||
"""查询 OpenAI 兼容 API 支持的模型列表（带 CLI User-Agent）"""
|
||
# 复用 OpenAIChatAdapter 的实现，添加 CLI User-Agent
|
||
cli_headers = {"User-Agent": config.internal_user_agent_openai_cli}
|
||
if extra_headers:
|
||
cli_headers.update(extra_headers)
|
||
models, error = await OpenAIChatAdapter.fetch_models(
|
||
client, base_url, api_key, cli_headers
|
||
)
|
||
# 更新 api_format 为 CLI 格式
|
||
for m in models:
|
||
m["api_format"] = cls.FORMAT_ID
|
||
return models, error
|
||
|
||
@classmethod
|
||
def build_endpoint_url(cls, base_url: str, request_data: Dict[str, Any], model_name: Optional[str] = None) -> str:
|
||
"""构建OpenAI CLI API端点URL"""
|
||
base_url = base_url.rstrip("/")
|
||
if base_url.endswith("/v1"):
|
||
return f"{base_url}/chat/completions"
|
||
else:
|
||
return f"{base_url}/v1/chat/completions"
|
||
|
||
@classmethod
|
||
def build_base_headers(cls, api_key: str) -> Dict[str, str]:
|
||
"""构建OpenAI CLI API认证头"""
|
||
return {
|
||
"Authorization": f"Bearer {api_key}",
|
||
"Content-Type": "application/json",
|
||
}
|
||
|
||
@classmethod
|
||
def get_protected_header_keys(cls) -> tuple:
|
||
"""返回OpenAI CLI API的保护头部key"""
|
||
return ("authorization", "content-type")
|
||
|
||
@classmethod
|
||
def build_request_body(cls, request_data: Dict[str, Any]) -> Dict[str, Any]:
|
||
"""构建OpenAI CLI API请求体"""
|
||
return request_data.copy()
|
||
|
||
@classmethod
|
||
def get_cli_user_agent(cls) -> Optional[str]:
|
||
"""获取OpenAI CLI User-Agent"""
|
||
return config.internal_user_agent_openai_cli
|
||
|
||
|
||
__all__ = ["OpenAICliAdapter"]
|