heymenn commited on
Commit
04fe71e
·
verified ·
1 Parent(s): 2da135a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -10,6 +10,8 @@ from kig_core.schemas import PlannerState, KeyIssue, GraphConfig
10
  from kig_core.planner import build_graph
11
  from kig_core.utils import key_issues_to_dataframe, dataframe_to_excel_bytes
12
  from kig_core.graph_client import neo4j_client # Import the initialized client instance
 
 
13
 
14
  # Configure logging for Streamlit app
15
  logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
 
10
  from kig_core.planner import build_graph
11
  from kig_core.utils import key_issues_to_dataframe, dataframe_to_excel_bytes
12
  from kig_core.graph_client import neo4j_client # Import the initialized client instance
13
+ from langchain_core.messages import HumanMessage, AIMessage, SystemMessage
14
+
15
 
16
  # Configure logging for Streamlit app
17
  logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')