From df2561f174fd1a20f14a2ede895ee5ce2311f931 Mon Sep 17 00:00:00 2001
From: Lingyi Zhang <lingyi_zhang@mckinsey.com>
Date: Wed, 3 Jul 2024 23:18:50 -0400
Subject: [PATCH] update string

---
 vizro-ai/examples/example_dashboard_code.ipynb  | 17 +++++++++++++++--
 .../vizro_ai/dashboard/graph/code_generation.py |  4 ++--
 2 files changed, 17 insertions(+), 4 deletions(-)

diff --git a/vizro-ai/examples/example_dashboard_code.ipynb b/vizro-ai/examples/example_dashboard_code.ipynb
index 1792c6a65..f9929fef6 100644
--- a/vizro-ai/examples/example_dashboard_code.ipynb
+++ b/vizro-ai/examples/example_dashboard_code.ipynb
@@ -19,6 +19,17 @@
     "load_dotenv()"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "6d19de01-b997-48b5-a3ae-45c3ca9ce965",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from langchain.globals import set_debug, set_verbose\n",
+    "set_verbose(True)"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
@@ -61,7 +72,9 @@
    "cell_type": "code",
    "execution_count": null,
    "id": "07c18cfb-9172-45b1-be7e-0085ee1fd66a",
-   "metadata": {},
+   "metadata": {
+    "scrolled": true
+   },
    "outputs": [],
    "source": [
     "df_stocks = px.data.stocks()\n",
@@ -95,7 +108,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "res = vizro_ai.dashboard([df, df_stocks], user_question_2_data, debug=True)\n",
+    "res = vizro_ai.dashboard([df, df_stocks], user_question_2_data, verbose=True) # debug mode from langgraph\n",
     "res"
    ]
   },
diff --git a/vizro-ai/src/vizro_ai/dashboard/graph/code_generation.py b/vizro-ai/src/vizro_ai/dashboard/graph/code_generation.py
index b2f9401cc..c47468733 100644
--- a/vizro-ai/src/vizro_ai/dashboard/graph/code_generation.py
+++ b/vizro-ai/src/vizro_ai/dashboard/graph/code_generation.py
@@ -33,10 +33,10 @@
 
 
 DfMetadata = Dict[str, Dict[str, Union[Dict[str, str], str]]]
-"Cleaned dataframe names and their metadata."
+"""Cleaned dataframe names and their metadata."""
 
 Messages = List[BaseMessage]
-"List of messages."
+"""List of messages."""
 
 
 class GraphState(BaseModel):