sqlsaber-viz 0.1.1__py3-none-any.whl → 0.2.0__py3-none-any.whl

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/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/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/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.dist-info → sqlsaber_viz-0.2.0.dist-info}/METADATA

@@ -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.dist-info → sqlsaber_viz-0.2.0.dist-info}/RECORD

@@ -1,16 +1,16 @@
 sqlsaber_viz/__init__.py,sha256=HdXfKzS8O7f1qD1nQo9XBPIB5Y6NvIroWmlaulqdvDs,394
 sqlsaber_viz/data_loader.py,sha256=mTFGCd6GZP_9KK110gupKumJQYsg0QoHVGXeqBefYbY,4380
-sqlsaber_viz/prompts.py,sha256=UCoIH32qfv8FguL3T0L77l8n-DjSYR7q_FOrZg-4_3U,1508
+sqlsaber_viz/prompts.py,sha256=OP8f22RSNUWSsJm69oVnP4ZSRG2qmuBxXXuW5ggc5n4,1506
 sqlsaber_viz/spec.py,sha256=_zgUBrHPrjztnyECRZycLps1KeEiLn29x9B0fSdqf10,2923
-sqlsaber_viz/spec_agent.py,sha256=LDH5gOd6zw-dno-0yNT7wKR7OnU4ILlH829cdzxOwXc,4694
+sqlsaber_viz/spec_agent.py,sha256=RuU6PUZZaRduzJyncl4Vm3bFfIvAcDH6UwxH9gRGz8w,4789
 sqlsaber_viz/templates.py,sha256=8WDYYVIjEDmKGUo9VPuv_vOOoieqnqGLl39hQpGEWL0,5022
-sqlsaber_viz/tools.py,sha256=d6axy4Ac_TRrJJqqGr9u9k_-5vFHE5XZUVEjzy0jEIM,7756
+sqlsaber_viz/tools.py,sha256=Q36tnDuxWsNmMVv79rkhUpO8rv9QvqWLoYmPNShNf4o,8350
 sqlsaber_viz/transforms.py,sha256=I_PxG4fy1ckdwC_9in3KVEa9cAQ9oLmq113eEGmvLEk,4440
 sqlsaber_viz/renderers/__init__.py,sha256=oIDE7QqaulKVN_yVKXqERuPskS-Pvxa2ILk234JbkIs,174
 sqlsaber_viz/renderers/base.py,sha256=V-8dCvgvE1AMpVlwOPRxnNwqumUbjlZmwMr-KpculyI,312
 sqlsaber_viz/renderers/html_renderer.py,sha256=f48JyfpcYIcnwCb_zmi9HlWflir94OOMuSVgi8PQhPA,367
-sqlsaber_viz/renderers/plotext_renderer.py,sha256=QFZbaFxvfEr32BCOFRq_fxtZYD_dhAJtbhkcr9VGzP8,13371
-sqlsaber_viz-0.1.1.dist-info/METADATA,sha256=LPq6Wg5944bec4J41-3YreZeUM3K7CPzLnoX2n5ri8M,432
-sqlsaber_viz-0.1.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-sqlsaber_viz-0.1.1.dist-info/entry_points.txt,sha256=us67FM9sQiA3XG-UxYbHI9GAKQB5Xpqe_STfQtKEWys,51
-sqlsaber_viz-0.1.1.dist-info/RECORD,,
+sqlsaber_viz/renderers/plotext_renderer.py,sha256=ipLHJw35Xq-mONfshzz30Ns6EHsnhalhxMmmXiSxdP4,13385
+sqlsaber_viz-0.2.0.dist-info/METADATA,sha256=eO-WEqJ091AibjUHQ1HgJopRIB7ILYg8wjRBpDNmr2k,432
+sqlsaber_viz-0.2.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+sqlsaber_viz-0.2.0.dist-info/entry_points.txt,sha256=us67FM9sQiA3XG-UxYbHI9GAKQB5Xpqe_STfQtKEWys,51
+sqlsaber_viz-0.2.0.dist-info/RECORD,,