nathan/nexus-mcp
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix: update source_system fields to use Literal types for stricter validation

Browse Source
This commit is contained in:
Nathan Castaldi 2026-04-15 14:36:17 -04:00
parent a5ad2bca11
commit db3b49624c
1 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
nexus-mcp/lib/schemas.py
View File

@ -14,8 +14,8 @@ This pattern prevents fragile dict access and enables compile-time safety.
""" """
from __future__ import annotations from __future__ import annotations
from pydantic import BaseModel, Field, field_validator from pydantic import BaseModel, ConfigDict, Field, field_validator
from typing import Optional from typing import Literal, Optional
from datetime import datetime from datetime import datetime
from enum import Enum from enum import Enum
@ -62,7 +62,9 @@ class CanonicalUser(BaseModel):
phone: Optional[str] = None phone: Optional[str] = None
# Source tracking # Source tracking
source_system: str = Field(description="System this data came from (AD, Entra, Workday)") source_system: Literal["ActiveDirectory", "Entra", "Workday"] = Field(
description="System this data came from"
)
source_id: Optional[str] = Field(default=None, description="Native ID in source system") source_id: Optional[str] = Field(default=None, description="Native ID in source system")
@field_validator('email') @field_validator('email')
@ -77,6 +79,9 @@ class CanonicalUser(BaseModel):
"""Normalize username to lowercase.""" """Normalize username to lowercase."""
return v.lower().strip() if v else None return v.lower().strip() if v else None
model_config = ConfigDict(
extra="forbid" # Stops silent adapter drift
)
# ── Device/Asset Domain ─────────────────────────────────────────────────────── # ── Device/Asset Domain ───────────────────────────────────────────────────────
@ -132,7 +137,7 @@ class CanonicalDevice(BaseModel):
mac_address: Optional[str] = None mac_address: Optional[str] = None
# Source tracking # Source tracking
source_system: str = Field(description="System this data came from (Intune, Lansweeper, Helix)") source_system: Literal["Intune", "Lansweeper", "Helix"] = Field(description="System this data came from")
source_id: Optional[str] = Field(default=None, description="Native ID in source system") source_id: Optional[str] = Field(default=None, description="Native ID in source system")
@field_validator('hostname') @field_validator('hostname')
@ -202,7 +207,7 @@ class CanonicalIncident(BaseModel):
resolved_date: Optional[datetime] = None resolved_date: Optional[datetime] = None
# Source tracking # Source tracking
source_system: str = "helix" source_system: Literal["helix"] = Field(description="System this data came from")
source_id: Optional[str] = None source_id: Optional[str] = None
@ -251,7 +256,7 @@ class CanonicalShipment(BaseModel):
service_type: Optional[str] = None service_type: Optional[str] = None
# Source tracking # Source tracking
source_system: str = "fedex" source_system: Literal["fedex"] = Field(description="System this data came from")
source_id: Optional[str] = None source_id: Optional[str] = None