sqlsaber-viz 0.1.1__tar.gz → 0.2.0__tar.gz

sqlsaber_viz-0.2.0/CHANGELOG.md

@@ -0,0 +1,16 @@
+# Changelog
+
+## [0.2.0](https://github.com/SarthakJariwala/sqlsaber/compare/sqlsaber-viz-v0.1.1...sqlsaber-viz-v0.2.0) (2026-02-06)
+
+
+### Features
+
+* allow overriding subagent model configuration via cli ([#118](https://github.com/SarthakJariwala/sqlsaber/issues/118)) ([ac31f8d](https://github.com/SarthakJariwala/sqlsaber/commit/ac31f8d71d328ab956f1f399a71e38ff5f769f7e))
+* visualization model override support via api and prompt improvements ([35a506b](https://github.com/SarthakJariwala/sqlsaber/commit/35a506b23337dd64f8f20606e7e547755eae6f21))
+
+## [0.1.1](https://github.com/SarthakJariwala/sqlsaber/compare/sqlsaber-viz-v0.1.0...sqlsaber-viz-v0.1.1) (2026-02-05)
+
+
+### Features
+
+* first plugins release ([bd1a5b5](https://github.com/SarthakJariwala/sqlsaber/commit/bd1a5b5079d6eaff2a52bdf4b8812d54ea8c2783))

{sqlsaber_viz-0.1.1 → sqlsaber_viz-0.2.0}/PKG-INFO

@@ -1,10 +1,10 @@
 Metadata-Version: 2.4
 Name: sqlsaber-viz
-Version: 0.1.1
+Version: 0.2.0
 Summary: SQLsaber terminal visualization plugin
 Requires-Python: >=3.12
 Requires-Dist: plotext>=5.3.0
-Requires-Dist: sqlsaber>=0.54.0
+Requires-Dist: sqlsaber>=0.55.1
 Description-Content-Type: text/markdown
 
 # SQLSaber Visualization Plugin

{sqlsaber_viz-0.1.1 → sqlsaber_viz-0.2.0}/pyproject.toml

@@ -1,10 +1,10 @@
 [project]
 name = "sqlsaber-viz"
-version = "0.1.1"
+version = "0.2.0"
 description = "SQLsaber terminal visualization plugin"
 readme = "README.md"
 requires-python = ">=3.12"
-dependencies = ["sqlsaber>=0.54.0", "plotext>=5.3.0"]
+dependencies = ["sqlsaber>=0.55.1", "plotext>=5.3.0"]
 
 [project.entry-points."sqlsaber.tools"]
 viz = "sqlsaber_viz:register_tools"

{sqlsaber_viz-0.1.1 → sqlsaber_viz-0.2.0}/src/sqlsaber_viz/prompts.py

@@ -3,12 +3,12 @@
 VIZ_SYSTEM_PROMPT = """You are a visualization spec generator. Given a user's request and data summary, generate a valid JSON visualization spec.
 
 ## Workflow
-1. Decide the appropriate chart type based on the request and data
+1. Decide the appropriate chart type based on the request and data. To see all available chart types, call `get_available_chart_types`
 2. Call `get_vizspec_template` with the chart type and file to get the correct spec structure
 3. Fill in the template with actual column names from the provided data summary
 4. Return ONLY the final JSON spec (no explanations, no markdown code blocks)
 
-## Chart Type Selection
+## Example Chart Type Selection
 - Comparing categories → bar
 - Comparing categories across series → bar with encoding.series
 - Trend over time → line
@@ -22,10 +22,9 @@ VIZ_SYSTEM_PROMPT = """You are a visualization spec generator. Given a user's re
 - {"filter": {"field": "col", "op": "!=", "value": null}} - Filter rows
 
 ## Rules
-1. ALWAYS call `get_vizspec_template` first to get the correct structure
-2. Use ONLY columns that exist in the provided data summary
-3. Match field types: category columns for x in bar charts, numeric columns for y
-4. Add limit transform for bar charts to avoid overcrowding (10-20 bars max)
-5. Sort bar charts by y value descending for better readability
-6. Title should describe what the chart shows
+- Use ONLY columns that exist in the provided data summary
+- Match field types: category columns for x in bar charts, numeric columns for y
+- Add limit transform for bar charts to avoid overcrowding (10-20 bars max)
+- Sort bar charts by y value descending for better readability
+- Title should describe what the chart shows
 """

{sqlsaber_viz-0.1.1 → sqlsaber_viz-0.2.0}/src/sqlsaber_viz/renderers/plotext_renderer.py

@@ -205,7 +205,9 @@ class PlotextRenderer:
         plt.box(labels, data)
         return None
 
-    def _render_histogram(self, chart: HistogramChart, rows: list[dict], plt) -> str | None:
+    def _render_histogram(
+        self, chart: HistogramChart, rows: list[dict], plt
+    ) -> str | None:
         field = chart.histogram.field
         bins = chart.histogram.bins
 

{sqlsaber_viz-0.1.1 → sqlsaber_viz-0.2.0}/src/sqlsaber_viz/spec_agent.py

@@ -24,7 +24,11 @@ class SpecAgent:
         self.agent = self._build_agent()
 
     def _build_agent(self):
-        model_name = self._model_name_override or self.config.model.name
+        model_name = (
+            self._model_name_override
+            or self.config.model.get_subagent_model("viz")
+            or self.config.model.name
+        )
         model_name_only = (
             model_name.split(":", 1)[1] if ":" in model_name else model_name
         )

{sqlsaber_viz-0.1.1 → sqlsaber_viz-0.2.0}/src/sqlsaber_viz/tools.py

@@ -39,6 +39,18 @@ class VizTool(Tool):
         self._last_rows: list[dict] | None = None
         self._last_file: str | None = None
         self._replay_messages: list | None = None
+        self._viz_model_name: str | None = None
+        self._viz_api_key: str | None = None
+
+    def set_viz_model(self, model_name: str | None, api_key: str | None = None) -> None:
+        """Set the model used by the internal SpecAgent for viz generation.
+
+        Args:
+            model_name: Model override (format: 'provider:model'). None to use default.
+            api_key: Optional API key for the model provider.
+        """
+        self._viz_model_name = model_name
+        self._viz_api_key = api_key
 
     def set_replay_messages(self, messages: list) -> None:
         """Set message history for replay scenarios (e.g., threads show)."""
@@ -87,7 +99,9 @@ class VizTool(Tool):
         self._last_rows = rows
         self._last_file = file
 
-        agent = _get_spec_agent_cls()()
+        agent = _get_spec_agent_cls()(
+            model_name=self._viz_model_name, api_key=self._viz_api_key
+        )
 
         try:
             spec = await asyncio.wait_for(

{sqlsaber_viz-0.1.1 → sqlsaber_viz-0.2.0}/tests/test_data_loader.py

@@ -23,4 +23,3 @@ def test_extract_data_summary_types() -> None:
 
 def test_infer_column_type_nulls() -> None:
     assert infer_column_type([None, None]) == "null"
-

{sqlsaber_viz-0.1.1 → sqlsaber_viz-0.2.0}/tests/test_transforms.py

@@ -1,6 +1,6 @@
 """Transform pipeline tests."""
 
-from sqlsaber_viz.spec import FilterConfig, FilterTransform, LimitTransform, SortItem, SortTransform
+from sqlsaber_viz.spec import FilterConfig, LimitTransform, SortItem, SortTransform
 from sqlsaber_viz.transforms import apply_filter, apply_sort, apply_transforms
 
 

{sqlsaber_viz-0.1.1 → sqlsaber_viz-0.2.0}/tests/test_viz_tool.py

@@ -7,7 +7,6 @@ from types import SimpleNamespace
 import pytest
 from rich.console import Console
 
-import sqlsaber.tools  # Ensure plugin discovery runs before importing viz tool.
 import sqlsaber_viz.tools as tools
 from sqlsaber_viz.spec import VizSpec
 from sqlsaber_viz.tools import VizTool
@@ -24,7 +23,7 @@ def _make_ctx(payload: dict, tool_call_id: str) -> SimpleNamespace:
 
 
 class DummyAgent:
-    def __init__(self):
+    def __init__(self, model_name: str | None = None, api_key: str | None = None):
         pass
 
     async def generate_spec(
@@ -52,7 +51,9 @@ class DummyAgent:
 
 
 @pytest.mark.asyncio
-async def test_viz_tool_execute_adds_bar_defaults(monkeypatch: pytest.MonkeyPatch) -> None:
+async def test_viz_tool_execute_adds_bar_defaults(
+    monkeypatch: pytest.MonkeyPatch,
+) -> None:
     monkeypatch.setattr(tools, "_get_spec_agent_cls", lambda: DummyAgent)
 
     payload = {

{sqlsaber_viz-0.1.1 → sqlsaber_viz-0.2.0}/uv.lock

@@ -2866,7 +2866,7 @@ wheels = [
 
 [[package]]
 name = "sqlsaber"
-version = "0.54.0"
+version = "0.55.1"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "aiomysql" },
@@ -2886,14 +2886,14 @@ dependencies = [
     { name = "structlog" },
     { name = "tabulate" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/ca/1e/14cc18ebb1a221b7863350922beea8b4d5e06ee5d540c188d4ca6e8b1a54/sqlsaber-0.54.0.tar.gz", hash = "sha256:8c4744591b14e374f53679d49448904d265a90686e13adc75cd6ade44abdcc07", size = 8133371, upload-time = "2026-02-04T02:26:30.972Z" }
+sdist = { url = "https://files.pythonhosted.org/packages/a8/17/374b0751a61ad7c333770848e045ebbaaeddcfbba160faa8569ef64f9e92/sqlsaber-0.55.1.tar.gz", hash = "sha256:4de35461aa184de3a8eb6c3fab8460e6645c4449ba9d2f03eb5d3922727c6433", size = 8328110, upload-time = "2026-02-05T01:01:44.721Z" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/3c/3a/5b15e2b589b3c866db2966cd80239b99be40a3441db6de4fce691fe32308/sqlsaber-0.54.0-py3-none-any.whl", hash = "sha256:72455830f968ce9e91f581ff28469c5d3589d8d817bbac5b1c8ffc13905c0b50", size = 137833, upload-time = "2026-02-04T02:26:28.661Z" },
+    { url = "https://files.pythonhosted.org/packages/5e/a7/565f6c39819cf022cd8e4bb887f3b76ffff22c32450dd8f97b424d1b37cb/sqlsaber-0.55.1-py3-none-any.whl", hash = "sha256:43abeb89cf2b78ce12b419316eeea55b60805d7e91298bf00b2b39c289dc02af", size = 138128, upload-time = "2026-02-05T01:01:46.735Z" },
 ]
 
 [[package]]
 name = "sqlsaber-viz"
-version = "0.1.1"
+version = "0.2.0"
 source = { editable = "." }
 dependencies = [
     { name = "plotext" },
@@ -2909,7 +2909,7 @@ dev = [
 [package.metadata]
 requires-dist = [
     { name = "plotext", specifier = ">=5.3.0" },
-    { name = "sqlsaber", specifier = ">=0.54.0" },
+    { name = "sqlsaber", specifier = ">=0.55.1" },
 ]
 
 [package.metadata.requires-dev]

sqlsaber_viz-0.1.1/CHANGELOG.md

@@ -1,8 +0,0 @@
-# Changelog
-
-## [0.1.1](https://github.com/SarthakJariwala/sqlsaber/compare/sqlsaber-viz-v0.1.0...sqlsaber-viz-v0.1.1) (2026-02-05)
-
-
-### Features
-
-* first plugins release ([bd1a5b5](https://github.com/SarthakJariwala/sqlsaber/commit/bd1a5b5079d6eaff2a52bdf4b8812d54ea8c2783))