import tomllib
from dataclasses import dataclass


@dataclass
class AgentIdentity:
    name: str
    personality: dict[str, str]
    values: dict[str, bool]
    quirks: dict[str, str]


@dataclass
class UserProfile:
    name: str
    preferences: dict[str, str]


def load_agent_identity(path: str) -> AgentIdentity:
    with open(path, "rb") as file:
        data = tomllib.load(file)

    return AgentIdentity(
        name=data["name"],
        personality=data["personality"],
        values=data["values"],
        quirks=data["quirks"],
    )


def load_user_profile(path: str) -> UserProfile:
    with open(path, "rb") as file:
        data = tomllib.load(file)

    return UserProfile(
        name=data["name"],
        preferences=data["preferences"],
    )