Spaces:
Running
Running
Update app.py
Browse files
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')
|