models.py

from dataclasses import dataclass, field
from typing import Any
from enum import Enum

class RequestState(Enum):
    RECEIVED = "RECEIVED"
    NORMALIZED = "NORMALIZED"
    POLICY_CHECKING = "POLICY_CHECKING"
    ALLOWED = "ALLOWED"
    DENIED = "DENIED"
    EXECUTING = "EXECUTING"
    SUCCEEDED = "SUCCEEDED"
    FAILED = "FAILED"

class DecisionType(Enum):
    ALLOW = "ALLOW"
    DENY = "DENY"

@dataclass
class ToolCallRequest:
    tool_name: str
    args: dict[str, Any]
    user_id: str | None = None
    session_id: str | None = None

@dataclass
class AuthorizeRequest:
    request_id: str
    tool_name: str
    normalized_args: dict[str, Any]

@dataclass
class PolicyDecision:
    request_id: str
    decision: DecisionType
    reason: str
    match_rules: list[str] = field(default_factory=list)

@dataclass
class StructuredError:
    code: str
    message: str

@dataclass
class AuditEvent:
    event_id: str
    request_id: str
    event_type: str
    event_payload: dict[str, Any]
    source_module: str
    timestamp: str

@dataclass
class ToolExecutionResult:
    request_id: str
    success: bool
    output: Any | None = None
    error: StructuredError | None = None

@dataclass
class GatewayResult:
    request_id: str
    state: RequestState
    decision: PolicyDecision | None = None
    execution_result: ToolExecutionResult | None = None
    error: StructuredError |None = None