36 lines
852 B
Python
36 lines
852 B
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from contextvars import ContextVar, Token
|
||
|
|
from dataclasses import dataclass
|
||
|
|
from datetime import datetime
|
||
|
|
from typing import Any, Dict, Optional
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass(slots=True)
|
||
|
|
class MCPRequestContext:
|
||
|
|
user: Dict[str, Any]
|
||
|
|
bot_id: str
|
||
|
|
credential_id: int
|
||
|
|
authenticated_at: datetime
|
||
|
|
|
||
|
|
|
||
|
|
current_mcp_request: ContextVar[Optional[MCPRequestContext]] = ContextVar(
|
||
|
|
"current_mcp_request",
|
||
|
|
default=None,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def set_current_mcp_request(request: MCPRequestContext) -> Token:
|
||
|
|
return current_mcp_request.set(request)
|
||
|
|
|
||
|
|
|
||
|
|
def reset_current_mcp_request(token: Token) -> None:
|
||
|
|
current_mcp_request.reset(token)
|
||
|
|
|
||
|
|
|
||
|
|
def require_mcp_request() -> MCPRequestContext:
|
||
|
|
request = current_mcp_request.get()
|
||
|
|
if request is None:
|
||
|
|
raise RuntimeError("MCP request context is unavailable")
|
||
|
|
return request
|