Function log_token_usage

experiments/token_cost/token_usage_tracker.py:104–134

记录一次 token 使用（累加到每日总计，不实时写入文件）

(
    task_type: str,
    model: str,
    input_tokens: int,
    output_tokens: int,
    user_id: Optional[str] = None,
    date: Optional[str] = None,
    extra: Optional[Dict] = None
)

def log_token_usage(
    task_type: str,
    model: str,
    input_tokens: int,
    output_tokens: int,
    user_id: Optional[str] = None,
    date: Optional[str] = None,
    extra: Optional[Dict] = None
) -> None:
    """记录一次 token 使用（累加到每日总计，不实时写入文件）"""
    record_date = date or datetime.now().strftime("%Y-%m-%d")

    if record_date not in _daily_token_totals:
        _daily_token_totals[record_date] = {
            "date": record_date,
            "embedding_input": 0,
            "embedding_output": 0,
            "llm_input": 0,
            "llm_output": 0,
            "call_count": 0,
        }

    total = _daily_token_totals[record_date]
    total["call_count"] += 1

    if "embedding" in task_type.lower():
        total["embedding_input"] += input_tokens
        total["embedding_output"] += output_tokens
    else:
        total["llm_input"] += input_tokens
        total["llm_output"] += output_tokens


def flush_token_logs() -> None: