from pydantic import BaseModel
from typing import List, Optional

class KnowledgeEntity(BaseModel):
    entity_id: str
    canonical_name: str
    franchise_id: Optional[str] = None
    is_default_variant: bool = False
    parent_entity_id: Optional[str] = None

class VisualTrait(BaseModel):
    trait_id: str
    confidence: float
    frequency: int
    source_count: int

class CanonicalTrait(BaseModel):
    trait_id: str
    category_id: Optional[str] = None
    trait_type: str