File size: 661 Bytes
ada3f28 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | from langchain_experimental.graph_transformers.diffbot import DiffbotGraphTransformer
from langchain_community.graphs import Neo4jGraph
from langchain.docstore.document import Document
from typing import List
import os
import logging
import uuid
from src.llm import get_combined_chunks, get_llm
logging.basicConfig(format='%(asctime)s - %(message)s',level='INFO')
def get_graph_from_diffbot(graph,chunkId_chunkDoc_list:List):
combined_chunk_document_list = get_combined_chunks(chunkId_chunkDoc_list)
llm,model_name = get_llm('diffbot')
graph_documents = llm.convert_to_graph_documents(combined_chunk_document_list)
return graph_documents
|