utils/kg/construct_kg.py

from langchain_community.graphs import Neo4jGraph
from langchain_experimental.graph_transformers import LLMGraphTransformer
from langchain_openai import ChatOpenAI
from langchain_core.documents import Document

def get_graph(text,allowed_nodes=None,prompt=None):
    
    llm = ChatOpenAI(temperature=0, model_name="gpt-4o-2024-08-06")

    if allowed_nodes:
        llm_transformer =   LLMGraphTransformer(llm=llm,allowed_nodes=allowed_nodes)
    else:
        llm_transformer = LLMGraphTransformer(llm=llm)
    documents = [Document(page_content=text)]

    graph_documents = llm_transformer.convert_to_graph_documents(documents)
    
    return graph_documents