lint

vizro-ai/src/vizro_ai/dashboard/_pydantic_output.py

@@ -106,6 +106,7 @@ def _get_pydantic_model(
             if "google" in llm_model.__class__.__module__.lower():
                 return _handle_google_llm_response(llm_model, response_model, prompt, message_content)
 
+            # For other models, use standard structured output
             pydantic_llm = prompt | llm_model.with_structured_output(response_model)
             return pydantic_llm.invoke(message_content)
 

vizro-ai/src/vizro_ai/plot/_response_models.py

@@ -4,6 +4,7 @@
     from pydantic.v1 import BaseModel, Field, PrivateAttr, create_model, validator
 except ImportError:  # pragma: no cov
     from pydantic import BaseModel, Field, PrivateAttr, create_model, validator
+import logging
 from typing import Optional, Union
 
 import autoflake
@@ -175,7 +176,11 @@ class ChartPlan(BaseChartPlan):
         Insights to what the chart explains or tries to show.
         Ideally concise and between 30 and 60 words.""",
     )
-    code_explanation: str = Field(..., description="Explanation of the code steps used for `chart_code` field.")
+    code_explanation: str = Field(
+        ...,
+        description="""
+        Explanation of the code steps used for `chart_code` field.""",
+    )
 
 
 class ChartPlanFactory: