From 0f51c7d935e78d914064091d10cd74a4d6a0fada Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Thu, 7 Nov 2024 16:22:17 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 vizro-ai/examples/dashboard_ui/actions.py      | 11 +++++++----
 vizro-ai/examples/dashboard_ui/app.py          |  6 +++++-
 vizro-ai/src/vizro_ai/_llm_models.py           |  2 +-
 vizro-ai/src/vizro_ai/plot/_response_models.py |  6 +++---
 4 files changed, 16 insertions(+), 9 deletions(-)

diff --git a/vizro-ai/examples/dashboard_ui/actions.py b/vizro-ai/examples/dashboard_ui/actions.py
index ff5421d32..15f275eb8 100644
--- a/vizro-ai/examples/dashboard_ui/actions.py
+++ b/vizro-ai/examples/dashboard_ui/actions.py
@@ -28,7 +28,12 @@
 logger = logging.getLogger(__name__)
 logger.setLevel(logging.INFO)  # TODO: remove manual setting and make centrally controlled
 
-SUPPORTED_VENDORS = {"OpenAI": ChatOpenAI, "Anthropic": ChatAnthropic, "Mistral": ChatMistralAI, "xAI (free API credits available)": ChatOpenAI}
+SUPPORTED_VENDORS = {
+    "OpenAI": ChatOpenAI,
+    "Anthropic": ChatAnthropic,
+    "Mistral": ChatMistralAI,
+    "xAI (free API credits available)": ChatOpenAI,
+}
 
 SUPPORTED_MODELS = {
     "OpenAI": [
@@ -64,9 +69,7 @@ def get_vizro_ai_plot(user_prompt, df, model, api_key, api_base, vendor_input):
     if vendor_input == "Mistral":
         llm = vendor(model=model, mistral_api_key=api_key, mistral_api_url=api_base, temperature=DEFAULT_TEMPERATURE)
     if vendor_input == "xAI (free API credits available)":
-        llm = vendor(
-            model=model, openai_api_key=api_key, openai_api_base=api_base, temperature=DEFAULT_TEMPERATURE
-        )
+        llm = vendor(model=model, openai_api_key=api_key, openai_api_base=api_base, temperature=DEFAULT_TEMPERATURE)
 
     vizro_ai = VizroAI(model=llm)
     ai_outputs = vizro_ai.plot(df, user_prompt, max_debug_retry=DEFAULT_RETRY, return_elements=True)
diff --git a/vizro-ai/examples/dashboard_ui/app.py b/vizro-ai/examples/dashboard_ui/app.py
index be7aad14a..eeadeffeb 100644
--- a/vizro-ai/examples/dashboard_ui/app.py
+++ b/vizro-ai/examples/dashboard_ui/app.py
@@ -181,7 +181,11 @@
                 MyDropdown(
                     options=SUPPORTED_MODELS["OpenAI"], value="gpt-4o-mini", multi=False, id="model-dropdown-id"
                 ),
-                OffCanvas(id="settings", options=["OpenAI", "Anthropic", "Mistral", "xAI (free API credits available)"], value="OpenAI"),
+                OffCanvas(
+                    id="settings",
+                    options=["OpenAI", "Anthropic", "Mistral", "xAI (free API credits available)"],
+                    value="OpenAI",
+                ),
                 UserPromptTextArea(id="text-area-id"),
                 # Modal(id="modal"),
             ],
diff --git a/vizro-ai/src/vizro_ai/_llm_models.py b/vizro-ai/src/vizro_ai/_llm_models.py
index 62f32f2cf..6b572d512 100644
--- a/vizro-ai/src/vizro_ai/_llm_models.py
+++ b/vizro-ai/src/vizro_ai/_llm_models.py
@@ -34,7 +34,7 @@
     "OpenAI": ChatOpenAI,
     "Anthropic": ChatAnthropic,
     "Mistral": ChatMistralAI,
-    "xAI": ChatOpenAI, # xAI API is compatible with OpenAI
+    "xAI": ChatOpenAI,  # xAI API is compatible with OpenAI
 }
 
 DEFAULT_MODEL = "gpt-4o-mini"
diff --git a/vizro-ai/src/vizro_ai/plot/_response_models.py b/vizro-ai/src/vizro_ai/plot/_response_models.py
index da07fbea9..49b3e98e1 100644
--- a/vizro-ai/src/vizro_ai/plot/_response_models.py
+++ b/vizro-ai/src/vizro_ai/plot/_response_models.py
@@ -96,9 +96,9 @@ def _check_chart_code(cls, v):
         # Remove markdown code block if present
         code = v
         if code.startswith("```python\n") and code.endswith("```"):
-            code = code[len("```python\n"):-3].strip()
+            code = code[len("```python\n") : -3].strip()
         elif code.startswith("```\n") and code.endswith("```"):
-            code = code[len("```\n"):-3].strip()
+            code = code[len("```\n") : -3].strip()
 
         # TODO: add more checks: ends with return, has return, no second function def, only one indented line
         if f"def {CUSTOM_CHART_NAME}(" not in code:
@@ -110,7 +110,7 @@ def _check_chart_code(cls, v):
                 """The chart code must accept a single argument `data_frame`,
 and it should be the first argument of the chart."""
             )
-        return code 
+        return code
 
     def _get_imports(self, vizro: bool = False):
         imports = list(dict.fromkeys(self.imports + self._additional_vizro_imports))  # remove duplicates