vizro-mcp 0.1.3__py3-none-any.whl → 0.1.4__py3-none-any.whl

vizro_mcp/__init__.py

@@ -3,7 +3,7 @@ import sys
 
 from .server import mcp
 
-__version__ = "0.1.3"
+__version__ = "0.1.4"
 
 
 def main():

vizro_mcp/_schemas/schemas.py

@@ -1,6 +1,6 @@
 """Schema defining pydantic models for usage in the MCP server."""
 
-from typing import Annotated, Any, Optional
+from typing import Annotated, Any
 
 import vizro.figures as vf
 import vizro.models as vm
@@ -153,7 +153,7 @@ class ChartPlan(BaseModel):
             imports = [imp for imp in imports if "vizro" not in imp]
         return "\n".join(imports) + "\n"
 
-    def get_chart_code(self, chart_name: Optional[str] = None, vizro: bool = False):
+    def get_chart_code(self, chart_name: str | None = None, vizro: bool = False):
         chart_code = self.chart_code
         if vizro:
             chart_code = chart_code.replace(f"def {self.chart_name}", f"@capture('graph')\ndef {self.chart_name}")

vizro_mcp/_utils/__init__.py

@@ -7,7 +7,6 @@ from .configs import (
 )
 from .prompts import (
     CHART_INSTRUCTIONS,
-    LAYOUT_INSTRUCTIONS,
     get_chart_prompt,
     get_dashboard_instructions,
     get_dashboard_prompt,
@@ -34,7 +33,6 @@ __all__ = [  # noqa: RUF022
     "VizroCodeAndPreviewLink",
     # Constants
     "CHART_INSTRUCTIONS",
-    "LAYOUT_INSTRUCTIONS",
     "GAPMINDER",
     "IRIS",
     "SAMPLE_DASHBOARD_CONFIG",

vizro_mcp/_utils/configs.py

@@ -1,7 +1,7 @@
 """Pre-set configs for the Vizro MCP."""
 
 from dataclasses import dataclass
-from typing import Any, Literal, Optional
+from typing import Any, Literal
 
 
 @dataclass
@@ -10,7 +10,7 @@ class DFMetaData:
     file_path_or_url: str
     file_location_type: Literal["local", "remote"]
     read_function_string: Literal["pd.read_csv", "pd.read_json", "pd.read_html", "pd.read_parquet", "pd.read_excel"]
-    column_names_types: Optional[dict[str, str]] = None
+    column_names_types: dict[str, str] | None = None
 
 
 @dataclass

vizro_mcp/_utils/prompts.py

@@ -1,7 +1,7 @@
 """Prompts for the Vizro MCP."""
 # ruff: noqa: E501 #Ignore line length only in prompts
 
-from typing import Literal, Optional, Protocol
+from typing import Literal, Protocol
 
 import vizro
 import vizro.actions as va
@@ -32,7 +32,7 @@ IMPORTANT:
 - IF the user has no plan (ie no components or pages), use the config at the bottom of this prompt, OTHERWISE:
 - make a plan of what components you would like to use, then request all necessary schemas using the `get_model_json_schema` tool
   - start with `Dashboard`, don't forget `Graph`,
-  - MUST explicitly use `Flex` (layout setting is NOT optional, contrary to what schema says); use `Grid` only if asked for precise placement or no scroll
+  - MUST explicitly set layout (layout setting is NOT optional, contrary to what schema says)
 - assemble your components into a page, then add the page or pages to a dashboard, DO NOT show config or code to the user until you have validated the solution
 - ALWAYS validate the dashboard configuration using the `validate_dashboard_config` tool
 - using `custom_chart` is encouraged for advanced visualizations, no need to call the planner tool in advanced mode
@@ -61,11 +61,12 @@ class HasNameAndDoc(Protocol):
     """Protocol for objects that have a name and a docstring."""
 
     __name__: str
-    __doc__: Optional[str]
+    __doc__: str | None
 
 
 # This dict is used to give the model and overview of what is available in the vizro.models namespace.
 # It helps it to narrow down the choices when asking for a model.
+# Intentionally omitted: Accordion, Action, Table, Layout and VizroBaseModel
 MODEL_GROUPS: dict[str, list[type[HasNameAndDoc]]] = {
     "main": [vm.Dashboard, vm.Page],
     "static components": [
@@ -75,10 +76,10 @@ MODEL_GROUPS: dict[str, list[type[HasNameAndDoc]]] = {
         vm.Container,
         vm.Tabs,
     ],
-    "dynamic components": [vm.Figure, vm.Graph, vm.AgGrid],
+    "dynamic components - ie reactive to controls": [vm.Figure, vm.Graph, vm.AgGrid],
     "layouts": [vm.Grid, vm.Flex],
-    "controls": [vm.Filter, vm.Parameter],
-    "selectors": [
+    "controls - control display of dynamic components": [vm.Filter, vm.Parameter],
+    "selectors - input mechanisms for controls": [
         vm.Dropdown,
         vm.RadioItems,
         vm.Checklist,
@@ -88,37 +89,13 @@ MODEL_GROUPS: dict[str, list[type[HasNameAndDoc]]] = {
         vm.Switch,
     ],
     "navigation": [vm.Navigation, vm.NavBar, vm.NavLink],
-    "additional_info": [vm.Tooltip],
+    "additional_info - info about the component": [vm.Tooltip],
     "actions available for the actions argument of a model": [
-        va.export_data,
-        va.set_control,
+        getattr(va, func) for func in va.__all__ if not hasattr(getattr(va, func), "__deprecated__")
     ],
-    "functions available for vm.Figure(...,figure=...) model": [vf.__dict__[func] for func in vf.__all__],
+    "functions available for vm.Figure(...,figure=...) model": [getattr(vf, func) for func in vf.__all__],
 }
 
-LAYOUT_INSTRUCTIONS = """Some extra information about Layout in Vizro:
-Overall:
-- Use Flex for most dashboards, use Grid only for cases below
-- Great tip: group elements in a `Container` and decide on appropriate layout inside the container and
-  outside the container
-
-Flex (Recommended in ALMOST ALL cases, definitely as default unless asked otherwise!):
-- use wrap=True if direction="row" to avoid horizontal scroll
-- Sizing of components: use min/max widths only to prevent items from shrinking too small
-    - Graph example: vm.Graph(figure=px.violin(..., width=900)).
-    - AgGrid example: vm.AgGrid(figure=dash_ag_grid(tips, style={"width": 900})).
-
-Grid (Use only if asked):
-- Use when:
-  - you need precise alignment across both rows and columns
-  - you want a component (often charts) to fill a concrete part of the page, e.g. when asked to fill one row, or a column
-- ALWAYS opt for `Grid` to make components fill space instead of playing with `width` and `height`
-- NEVER use collapsible containers inside `Grid`
-- IMPORTANT: you must use integers to refer to the elements, starting from 0, every element must be referenced,
-  elements can't overlap and must be rectangular, and you must use the same number of columns and rows for each row
-
-Do NOT forget to call `validate_dashboard_config` after each iteration."""
-
 
 def get_overview_vizro_models() -> dict[str, list[dict[str, str]]]:
     """Get all available models in the vizro.models namespace.
@@ -183,7 +160,7 @@ Create a super simple Vizro dashboard with one page and one chart and one filter
 """
 
 
-def get_dashboard_prompt(file_path_or_url: str, user_context: Optional[str] = None) -> str:
+def get_dashboard_prompt(file_path_or_url: str, user_context: str | None = None) -> str:
     """Get a prompt for creating a Vizro dashboard."""
     USER_INSTRUCTIONS = f"""
 3. Create a Vizro dashboard that follows the user context:
@@ -214,7 +191,7 @@ Create a dashboard based on the following dataset: `{file_path_or_url}`. Proceed
 """
 
 
-def get_chart_prompt(file_path_or_url: str, user_context: Optional[str] = None) -> str:
+def get_chart_prompt(file_path_or_url: str, user_context: str | None = None) -> str:
     """Get a prompt for creating a Vizro chart."""
     FALLBACK_INSTRUCTIONS = """
 - Think what chart could reflect this data-set best, ideally the chart shows some insights about the data-set

vizro_mcp/_utils/utils.py

@@ -7,7 +7,7 @@ import json
 import re
 from dataclasses import dataclass
 from pathlib import Path
-from typing import TYPE_CHECKING, Literal, Optional, Union
+from typing import TYPE_CHECKING, Literal
 from urllib.parse import quote, urlencode
 
 import pandas as pd
@@ -44,7 +44,7 @@ def convert_github_url_to_raw(path_or_url: str) -> str:
 
 
 def load_dataframe_by_format(
-    path_or_url: Union[str, Path], mime_type: Optional[str] = None
+    path_or_url: str | Path, mime_type: str | None = None
 ) -> tuple[pd.DataFrame, Literal["pd.read_csv", "pd.read_json", "pd.read_html", "pd.read_excel", "pd.read_parquet"]]:
     """Load a dataframe based on file format determined by MIME type or file extension."""
     file_path_str_lower = str(path_or_url).lower()

vizro_mcp/server.py

@@ -4,7 +4,7 @@ import mimetypes
 import webbrowser
 from dataclasses import dataclass
 from pathlib import Path
-from typing import Any, Literal, Optional
+from typing import Any, Literal
 
 import vizro
 import vizro.models as vm
@@ -22,7 +22,6 @@ from vizro_mcp._utils import (
     CHART_INSTRUCTIONS,
     GAPMINDER,
     IRIS,
-    LAYOUT_INSTRUCTIONS,
     STOCKS,
     TIPS,
     DFInfo,
@@ -48,7 +47,7 @@ class ValidateResults:
     valid: bool
     message: str
     python_code: str
-    pycafe_url: Optional[str]
+    pycafe_url: str | None
     browser_opened: bool
 
 
@@ -58,8 +57,8 @@ class DataAnalysisResults:
 
     valid: bool
     message: str
-    df_info: Optional[DFInfo]
-    df_metadata: Optional[DFMetaData]
+    df_info: DFInfo | None
+    df_metadata: DFMetaData | None
 
 
 @dataclass
@@ -149,7 +148,8 @@ that model if necessary. Do NOT forget to call `validate_dashboard_config` after
         return ModelJsonSchemaResults(
             model_name=model_name,
             json_schema=model_class.model_json_schema(schema_generator=NoDefsGenerateJsonSchema),
-            additional_info=LAYOUT_INSTRUCTIONS,
+            additional_info="""Grid layout: use integers starting from 0 to reference elements.
+Elements can't overlap, must be rectangular, and rows must have equal column counts.""",
         )
 
     return ModelJsonSchemaResults(
@@ -315,7 +315,7 @@ def create_starter_dashboard():
 @mcp.prompt()
 def create_dashboard(
     file_path_or_url: str = Field(description="The absolute path or URL to the data file you want to use."),
-    context: Optional[str] = Field(default=None, description="(Optional) Describe the dashboard you want to create."),
+    context: str | None = Field(default=None, description="(Optional) Describe the dashboard you want to create."),
 ) -> str:
     """Prompt template for creating an EDA dashboard based on one dataset."""
     return get_dashboard_prompt(file_path_or_url, context)
@@ -372,7 +372,7 @@ def validate_chart_code(
 @mcp.prompt()
 def create_vizro_chart(
     file_path_or_url: str = Field(description="The absolute path or URL to the data file you want to use."),
-    context: Optional[str] = Field(default=None, description="(Optional) Describe the chart you want to create."),
+    context: str | None = Field(default=None, description="(Optional) Describe the chart you want to create."),
 ) -> str:
     """Prompt template for creating a Vizro chart."""
     return get_chart_prompt(file_path_or_url, context)

{vizro_mcp-0.1.3.dist-info → vizro_mcp-0.1.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vizro-mcp
-Version: 0.1.3
+Version: 0.1.4
 Summary: MCP server to help create Vizro dashboards and charts
 Author: Vizro Team
 License-File: LICENSE.txt
@@ -9,6 +9,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Requires-Python: >=3.10
 Requires-Dist: click>=8.1.7
 Requires-Dist: httpx>=0.28.1

vizro_mcp-0.1.4.dist-info/RECORD

@@ -0,0 +1,14 @@
+vizro_mcp/__init__.py,sha256=b07rnAb9ypm8N9O4fLH5LYOeVC48ncU5k_xesNH0zmc,379
+vizro_mcp/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+vizro_mcp/server.py,sha256=DoD1B_-EeHcTJSAog6lyaAi-_AA4u1dXAsHLgq0ku3Y,13689
+vizro_mcp/_schemas/__init__.py,sha256=0lk_CiNjo4QFyX0i-VuSeL7EMM6sD_2p836y-ftM3Gs,196
+vizro_mcp/_schemas/schemas.py,sha256=6XCNSTag7RL0xL3SkRVMlJ2GOW0nbm73drhcKRpNfOU,8795
+vizro_mcp/_utils/__init__.py,sha256=ucPbfXe8kEFT7chz6ed_BeqAw71MnoWlZTYyCp4tPYY,1130
+vizro_mcp/_utils/configs.py,sha256=DH5B-RbtLuAkq6cgVL7oBOg3J-T_Mi4ovU430h5iUus,3614
+vizro_mcp/_utils/prompts.py,sha256=TJ--Tpv_tw2L8Tbt6PtaCA435HhUPpqUibNoEryKu94,10767
+vizro_mcp/_utils/utils.py,sha256=1CAD1U27Uca_9KmR3SolimFqMrWVlWApKOFSWDvpeRY,8173
+vizro_mcp-0.1.4.dist-info/METADATA,sha256=3iHwiGkUAQpUM0zX0X47G5_0F6sSQARAcLa-eetlgXU,12413
+vizro_mcp-0.1.4.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+vizro_mcp-0.1.4.dist-info/entry_points.txt,sha256=iSUzPHvx4Ogsn91tK2C0OHxI44SNCHT1F1zUqbTj5O0,45
+vizro_mcp-0.1.4.dist-info/licenses/LICENSE.txt,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+vizro_mcp-0.1.4.dist-info/RECORD,,

{vizro_mcp-0.1.3.dist-info → vizro_mcp-0.1.4.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: hatchling 1.27.0
+Generator: hatchling 1.28.0
 Root-Is-Purelib: true
 Tag: py3-none-any

vizro_mcp-0.1.3.dist-info/RECORD

@@ -1,14 +0,0 @@
-vizro_mcp/__init__.py,sha256=9_2ffpEe97JqIlmGOJszrMrDv2N8f9VyPpVmZ7s9C4k,379
-vizro_mcp/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-vizro_mcp/server.py,sha256=ydjjwhyyr983_nHlA7uOAvZcdQyJ4x5KiXj4hPIaiWk,13603
-vizro_mcp/_schemas/__init__.py,sha256=0lk_CiNjo4QFyX0i-VuSeL7EMM6sD_2p836y-ftM3Gs,196
-vizro_mcp/_schemas/schemas.py,sha256=gwDJL_9OK_S0NrZTk1rmJCVP6LLqsMIbVh-s-nZmBrw,8808
-vizro_mcp/_utils/__init__.py,sha256=r57TYaklMaV5TYD8HUG6uv-whvMFfyuULs4cqEZXGkg,1182
-vizro_mcp/_utils/configs.py,sha256=28ALgRgKYzRnPunSTjZxHFz2kyQy8yW4ZLhDgCUkE0s,3627
-vizro_mcp/_utils/prompts.py,sha256=c4q6m9RvQIAWXuEpAgP5giDEAprdgpR23GDpyAQ5ekA,11922
-vizro_mcp/_utils/utils.py,sha256=ZzBp6_jHMI6VsaptWC6WFMf1S4QCnrbo2Iq1-O4QH0A,8199
-vizro_mcp-0.1.3.dist-info/METADATA,sha256=9GUwdMqHFWoQYnmTTU0rD0OPujYgU9EONC60ZAdV9Qs,12362
-vizro_mcp-0.1.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-vizro_mcp-0.1.3.dist-info/entry_points.txt,sha256=iSUzPHvx4Ogsn91tK2C0OHxI44SNCHT1F1zUqbTj5O0,45
-vizro_mcp-0.1.3.dist-info/licenses/LICENSE.txt,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-vizro_mcp-0.1.3.dist-info/RECORD,,