Commit
·
9a0d22f
1
Parent(s):
608d44d
log statement improved
Browse files
kg_builder/src/knowledge_graph_builder.py
CHANGED
|
@@ -31,7 +31,7 @@ def extract_and_store_graph(
|
|
| 31 |
:param rels: TODO
|
| 32 |
"""
|
| 33 |
|
| 34 |
-
logger.info("Extract graph data using OpenAI functions ...")
|
| 35 |
extract_chain = get_extraction_chain(data_source_name, nodes, rels)
|
| 36 |
data = extract_chain.invoke(document.page_content)['function']
|
| 37 |
# Construct a graph document
|
|
|
|
| 31 |
:param rels: TODO
|
| 32 |
"""
|
| 33 |
|
| 34 |
+
logger.info(f"Extract graph data (Data Source: {data_source_name}) using OpenAI functions ...")
|
| 35 |
extract_chain = get_extraction_chain(data_source_name, nodes, rels)
|
| 36 |
data = extract_chain.invoke(document.page_content)['function']
|
| 37 |
# Construct a graph document
|