25 lines
585 B
Python
25 lines
585 B
Python
|
"""
|
|||
|
|
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)
|