Add FastAPI backend with agent system
This commit is contained in:
24
backend/app/agents/context.py
Normal file
24
backend/app/agents/context.py
Normal file
@@ -0,0 +1,24 @@
|
||||
"""
|
||||
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)
|
||||
Reference in New Issue
Block a user