Fix SSE route dependency and align architecture docs
This commit is contained in:
@@ -1,21 +1,28 @@
|
||||
"""Define schema models for compliance."""
|
||||
|
||||
from pydantic import BaseModel
|
||||
from typing import Optional
|
||||
from enum import Enum
|
||||
# Group related schema definitions so validation rules stay consistent.
|
||||
|
||||
|
||||
|
||||
class RiskLevel(str, Enum):
|
||||
"""Define the Risk Level enumeration."""
|
||||
high = "high"
|
||||
medium = "medium"
|
||||
low = "low"
|
||||
|
||||
|
||||
class ComplianceStatus(str, Enum):
|
||||
"""Define the Compliance Status enumeration."""
|
||||
pass_status = "pass"
|
||||
warning = "warning"
|
||||
fail = "fail"
|
||||
|
||||
|
||||
class Regulation(BaseModel):
|
||||
"""Define the Regulation API model."""
|
||||
id: int
|
||||
name: str
|
||||
clause: Optional[str] = None
|
||||
@@ -26,6 +33,7 @@ class Regulation(BaseModel):
|
||||
|
||||
|
||||
class ComplianceSegment(BaseModel):
|
||||
"""Define the Compliance Segment API model."""
|
||||
id: int
|
||||
index: int
|
||||
intent: str
|
||||
@@ -37,6 +45,7 @@ class ComplianceSegment(BaseModel):
|
||||
|
||||
|
||||
class RiskDashboard(BaseModel):
|
||||
"""Define the Risk Dashboard API model."""
|
||||
score: float
|
||||
high_risk_count: int
|
||||
medium_risk_count: int
|
||||
@@ -47,6 +56,7 @@ class RiskDashboard(BaseModel):
|
||||
|
||||
|
||||
class PriorityAction(BaseModel):
|
||||
"""Define the Priority Action API model."""
|
||||
regulation: str
|
||||
issue: str
|
||||
suggestion: str
|
||||
@@ -54,6 +64,7 @@ class PriorityAction(BaseModel):
|
||||
|
||||
|
||||
class ComplianceResult(BaseModel):
|
||||
"""Define the Compliance Result API model."""
|
||||
task_id: str
|
||||
dashboard: RiskDashboard
|
||||
segments: list[ComplianceSegment]
|
||||
@@ -61,9 +72,11 @@ class ComplianceResult(BaseModel):
|
||||
|
||||
|
||||
class ComplianceChatRequest(BaseModel):
|
||||
"""Define the Compliance Chat Request API model."""
|
||||
query: str
|
||||
|
||||
|
||||
class AnalyzeResponse(BaseModel):
|
||||
"""Define the Analyze Response API model."""
|
||||
task_id: str
|
||||
status: str = "processing"
|
||||
Reference in New Issue
Block a user