def __init__(self, embeddings, context):
        """
        Creates a new GraphContext.

        Args:
            embeddings: embeddings instance
            context: number of records to use as context
        """

        self.embeddings = embeddings
        self.context = context

        Creates a new GraphContext.

        Args:
            embeddings: embeddings instance
            context: number of records to use as context
        
txtairag/rag.py