graphiti/core/utils/utils.py

import logging

from core.edges import EpisodicEdge
from core.nodes import EntityNode, EpisodicNode

logger = logging.getLogger(__name__)


def build_episodic_edges(
    entity_nodes: list[EntityNode], episode: EpisodicNode
) -> list[EpisodicEdge]:
    edges: list[EpisodicEdge] = []

    for node in entity_nodes:
        edges.append(
            EpisodicEdge(
                source_node_uuid=episode,
                target_node_uuid=node,
                created_at=episode.created_at,
            )
        )

    return edges