update string

mckinsey · Jul 4, 2024 · df2561f · df2561f
1 parent d1b02a1
commit df2561f
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 4 deletions.
diff --git 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
@@ -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):