GraphQueryEngine
autogen.agentchat.contrib.graph_rag.graph_query_engine.GraphQueryEngine #
Bases: Protocol
An abstract base class that represents a graph query engine on top of a underlying graph database.
This interface defines the basic methods for graph-based RAG.
init_db #
This method initializes graph database with the input documents or records. Usually, it takes the following steps, 1. connecting to a graph database. 2. extract graph nodes, edges based on input data, graph schema and etc. 3. build indexes etc.
Args: input_doc: a list of input documents that are used to build the graph in database.
Source code in autogen/agentchat/contrib/graph_rag/graph_query_engine.py
add_records #
Add new records to the underlying database and add to the graph if required.
query #
This method transform a string format question into database query and return the result.