adrienbrdne commited on
Commit
d29ee3e
·
verified ·
1 Parent(s): c8b6c15

Update ki_gen/data_retriever.py

Browse files
Files changed (1) hide show
  1. ki_gen/data_retriever.py +1 -0
ki_gen/data_retriever.py CHANGED
@@ -21,6 +21,7 @@ from langchain_core.prompts import ChatPromptTemplate
21
  from langchain_core.pydantic_v1 import Field
22
  from pydantic import BaseModel
23
 
 
24
 
25
  from langgraph.graph import StateGraph
26
 
 
21
  from langchain_core.pydantic_v1 import Field
22
  from pydantic import BaseModel
23
 
24
+ from neo4j import GraphDatabase
25
 
26
  from langgraph.graph import StateGraph
27