Function write_request_log

app/utils/request_logging.py:135–174

Persist a request log entry without breaking request handling.

(
    *,
    provider: str,
    model: str,
    source_info: RequestSourceInfo,
    success: bool,
    started_at: float,
    status_code: int = 200,
    first_token_time: float = 0.0,
    input_tokens: int = 0,
    output_tokens: int = 0,
    cache_creation_tokens: int = 0,
    cache_read_tokens: int = 0,
    total_tokens: Optional[int] = None,
    error_message: Optional[str] = None,
)

async def write_request_log(
    *,
    provider: str,
    model: str,
    source_info: RequestSourceInfo,
    success: bool,
    started_at: float,
    status_code: int = 200,
    first_token_time: float = 0.0,
    input_tokens: int = 0,
    output_tokens: int = 0,
    cache_creation_tokens: int = 0,
    cache_read_tokens: int = 0,
    total_tokens: Optional[int] = None,
    error_message: Optional[str] = None,
) -> None:
    """Persist a request log entry without breaking request handling."""
    duration = max(0.0, time.perf_counter() - started_at)
    try:
        dao = get_request_log_dao()
        await dao.add_log(
            provider=provider,
            endpoint=source_info.endpoint,
            source=source_info.source,
            protocol=source_info.protocol,
            client_name=source_info.client_name,
            model=model,
            status_code=status_code,
            success=success,
            duration=duration,
            first_token_time=first_token_time,
            input_tokens=input_tokens,
            output_tokens=output_tokens,
            cache_creation_tokens=cache_creation_tokens,
            cache_read_tokens=cache_read_tokens,
            total_tokens=total_tokens,
            error_message=error_message,
        )
    except Exception as exc:
        logger.error(f"写入请求日志失败: {exc}")


def _openai_payload_has_output(payload: Dict[str, Any]) -> bool: