from google.protobuf.internal import containers as _containers
from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
from google.protobuf import descriptor as _descriptor
from google.protobuf import message as _message
from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union

DESCRIPTOR: _descriptor.FileDescriptor

class PredictedMovement(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
    __slots__ = ()
    PREDICTED_MOVEMENT_UNSPECIFIED: _ClassVar[PredictedMovement]
    PREDICTED_MOVEMENT_NEUTRAL: _ClassVar[PredictedMovement]
    PREDICTED_MOVEMENT_INCREASE: _ClassVar[PredictedMovement]
    PREDICTED_MOVEMENT_DECREASE: _ClassVar[PredictedMovement]
PREDICTED_MOVEMENT_UNSPECIFIED: PredictedMovement
PREDICTED_MOVEMENT_NEUTRAL: PredictedMovement
PREDICTED_MOVEMENT_INCREASE: PredictedMovement
PREDICTED_MOVEMENT_DECREASE: PredictedMovement

class Entity(_message.Message):
    __slots__ = ("entity_id", "entity_name")
    ENTITY_ID_FIELD_NUMBER: _ClassVar[int]
    ENTITY_NAME_FIELD_NUMBER: _ClassVar[int]
    entity_id: str
    entity_name: str
    def __init__(self, entity_id: _Optional[str] = ..., entity_name: _Optional[str] = ...) -> None: ...

class Relationship(_message.Message):
    __slots__ = ("relationship_id", "start_date", "end_date", "source_text", "predicted_movement")
    RELATIONSHIP_ID_FIELD_NUMBER: _ClassVar[int]
    START_DATE_FIELD_NUMBER: _ClassVar[int]
    END_DATE_FIELD_NUMBER: _ClassVar[int]
    SOURCE_TEXT_FIELD_NUMBER: _ClassVar[int]
    PREDICTED_MOVEMENT_FIELD_NUMBER: _ClassVar[int]
    relationship_id: str
    start_date: str
    end_date: str
    source_text: str
    predicted_movement: PredictedMovement
    def __init__(self, relationship_id: _Optional[str] = ..., start_date: _Optional[str] = ..., end_date: _Optional[str] = ..., source_text: _Optional[str] = ..., predicted_movement: _Optional[_Union[PredictedMovement, str]] = ...) -> None: ...

class EntityRelationship(_message.Message):
    __slots__ = ("from_entity", "relationship", "to_entity")
    FROM_ENTITY_FIELD_NUMBER: _ClassVar[int]
    RELATIONSHIP_FIELD_NUMBER: _ClassVar[int]
    TO_ENTITY_FIELD_NUMBER: _ClassVar[int]
    from_entity: Entity
    relationship: Relationship
    to_entity: Entity
    def __init__(self, from_entity: _Optional[_Union[Entity, _Mapping]] = ..., relationship: _Optional[_Union[Relationship, _Mapping]] = ..., to_entity: _Optional[_Union[Entity, _Mapping]] = ...) -> None: ...

class EntityKnowledgeGraph(_message.Message):
    __slots__ = ("entity_relationships",)
    ENTITY_RELATIONSHIPS_FIELD_NUMBER: _ClassVar[int]
    entity_relationships: _containers.RepeatedCompositeFieldContainer[EntityRelationship]
    def __init__(self, entity_relationships: _Optional[_Iterable[_Union[EntityRelationship, _Mapping]]] = ...) -> None: ...