"""
Agent 运行时上下文
用于在工具调用链中传递 user_id 等上下文信息
"""

from contextvars import ContextVar
from typing import Optional

_current_user_id: ContextVar[Optional[str]] = ContextVar("current_user_id", default=None)


def set_current_user(user_id: str):
    """设置当前用户ID（线程/协程安全）"""
    _current_user_id.set(user_id)


def get_current_user() -> str:
    """获取当前用户ID"""
    return _current_user_id.get() or "default"


def clear_current_user():
    """清除当前用户上下文"""
    _current_user_id.set(None)