Spaces:
Running
Running
File size: 666 Bytes
56a3465 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
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
|