merge

vizro-ai/src/vizro_ai/_vizro_ai.py

@@ -3,8 +3,8 @@
 
 import pandas as pd
 import plotly.graph_objects as go
-from langchain_openai import ChatOpenAI
 from langchain_core.messages import HumanMessage
+from langchain_openai import ChatOpenAI
 
 from vizro_ai.chains._llm_models import _get_llm_model
 from vizro_ai.components import GetCodeExplanation, GetDebugger

vizro-ai/src/vizro_ai/dashboard/graph/code_generation.py

@@ -1,5 +1,6 @@
 """Code generation graph for dashboard generation."""
 
+import inspect
 import logging
 import operator
 import re
@@ -225,9 +226,9 @@ def _create_and_compile_graph():
 if __name__ == "__main__":
     user_input = """
                 I need a page with a table showing the population per continent \n
-                I also want a page with two \ncards on it. One should be a card saying: 
-                `This was the jolly data dashboard, it was created in Vizro which is amazing` \n, 
-                and the second card should link to `https://vizro.readthedocs.io/`. The title of 
+                I also want a page with two \ncards on it. One should be a card saying:
+                `This was the jolly data dashboard, it was created in Vizro which is amazing` \n,
+                and the second card should link to `https://vizro.readthedocs.io/`. The title of
                 the dashboard should be: `My wonderful \njolly dashboard showing a lot of info`.\n
                 The layout of this page should use `grid=[[0,1]]`
                 """

vizro-ai/src/vizro_ai/dashboard/nodes/core_builder/model.py

@@ -8,8 +8,8 @@
 from typing import Dict
 from vizro.models import VizroBaseModel
 
-from langchain_core.prompts import ChatPromptTemplate
 from langchain_core.messages import HumanMessage
+from langchain_core.prompts import ChatPromptTemplate
 
 
 class ProxyVizroBaseModel(BaseModel):