# This file was auto-generated by Fern from our API Definition.

import typing

import pydantic
from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
from .local_memory_entity import LocalMemoryEntity
from .local_memory_relation import LocalMemoryRelation


class LocalMemoryData(UniversalBaseModel):
    """
    Complete Local Memory sandbox data structure.

    Represents a knowledge graph with:
    - Entities: nodes containing information
    - Relations: edges connecting entities
    """

    entities: typing.Optional[typing.List[LocalMemoryEntity]] = pydantic.Field(default=None)
    """
    List of entities in the knowledge graph
    """

    relations: typing.Optional[typing.List[LocalMemoryRelation]] = pydantic.Field(default=None)
    """
    List of relations between entities
    """

    if IS_PYDANTIC_V2:
        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
    else:

        class Config:
            frozen = True
            smart_union = True
            extra = pydantic.Extra.allow
