sqlsaber 0.31.0__tar.gz → 0.32.0__tar.gz

sqlsaber-0.32.0/AGENTS.md

@@ -0,0 +1,44 @@
+# Repository Guidelines
+
+## Project Structure & Module Organization
+- Source: `src/sqlsaber/`
+  - `cli/`: CLI entry (`saber`, `sqlsaber`), REPL, prompts.
+  - `agents/`: agent implementations (pydantic‑ai).
+  - `tools/`: SQL, introspection, registry.
+  - `database/`: connection, resolver, schema utilities.
+  - `memory/`, `conversation/`: state and persistence.
+  - `config/`: settings, API keys, OAuth, DB configs.
+- Tests: `tests/` mirror modules (`test_cli/`, `test_tools/`, …).
+- Docs & assets: `docs/`, `sqlsaber.gif`, `sqlsaber.svg`.
+
+## Build, Test, and Development Commands
+- Install (editable): `uv sync`
+- Lint: `uv run ruff check .`
+- Format: `uv run ruff format .`
+- Tests (all): `uv run pytest -q`
+- Tests (targeted): `uv run pytest tests/test_tools -q`
+- Run CLI (dev): `uv run saber` or `uv run python -m sqlsaber`
+
+
+Note: Prefer `uv run ruff ...` over `uvx ruff ...` to avoid hitting user-level uv caches that may be restricted in sandboxed environments.
+
+## Coding Style & Naming Conventions
+- Python 3.12+, 4‑space indent, use modern type hints.
+- Ruff is the linter/formatter; code must be clean and formatted.
+- Naming: modules/files `snake_case.py`; classes `PascalCase`; functions/vars `snake_case`; constants `UPPER_SNAKE`.
+- Keep public CLI surfaces in `cli/`; factor reusable logic into modules under `sqlsaber/`.
+
+## Testing Guidelines
+- Framework: `pytest` with `pytest-asyncio`.
+- Place tests under `tests/`, name files `test_*.py` and mirror package layout.
+- Include tests for new behavior and bug fixes; prefer async tests for async code.
+- Use fixtures from `tests/conftest.py` where possible.
+
+## Commit & Pull Request Guidelines
+- Commits: short, imperative; prefer conventional prefixes (e.g., `feat:`, `fix:`, `docs:`). Reference issues/PRs when relevant.
+- PRs must: describe the change and rationale, include tests, pass `ruff` and `pytest`, and update docs/README for user‑visible changes.
+- For CLI UX changes, include before/after samples (command + output snippet).
+
+## Security & Configuration Tips
+- Never commit secrets. Configure via CLI: `saber db add` and `saber models set` (credentials stored via system keyring).
+- Queries run read‑only by default; avoid introducing mutating behavior in tools without explicit safeguards.

{sqlsaber-0.31.0 → sqlsaber-0.32.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sqlsaber
-Version: 0.31.0
+Version: 0.32.0
 Summary: SQLsaber - Open-source agentic SQL assistant
 License-File: LICENSE
 Requires-Python: >=3.12

{sqlsaber-0.31.0 → sqlsaber-0.32.0}/docs/src/content/docs/changelog.md

@@ -9,6 +9,12 @@ All notable changes to SQLsaber will be documented here.
 
 ---
 
+### v0.32.0 - 2025-10-14
+
+#### Added
+
+- Optional MLflow autologging
+  - Set `MLFLOW_URI` and `MLFLOW_EXP` before running `sqlsaber` to forward telemetry via `mlflow.pydantic_ai.autolog()` with no runtime requirement when the MLflow package is absent.
 
 ### v0.31.0 - 2025-10-13
 

{sqlsaber-0.31.0 → sqlsaber-0.32.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "sqlsaber"
-version = "0.31.0"
+version = "0.32.0"
 description = "SQLsaber - Open-source agentic SQL assistant"
 readme = "README.md"
 requires-python = ">=3.12"

{sqlsaber-0.31.0 → sqlsaber-0.32.0}/src/sqlsaber/cli/commands.py

@@ -1,6 +1,7 @@
 """CLI command definitions and handlers."""
 
 import asyncio
+import os
 import sys
 from typing import Annotated
 
@@ -43,6 +44,52 @@ app.command(create_threads_app(), name="threads")
 console = create_console()
 config_manager = DatabaseConfigManager()
 
+_MLFLOW_CONFIGURED = False
+
+
+def _maybe_configure_mlflow(log) -> bool:
+    """Enable mlflow autologging when environment variables are present."""
+    global _MLFLOW_CONFIGURED
+    if _MLFLOW_CONFIGURED:
+        return True
+
+    tracking_uri = os.getenv("MLFLOW_URI")
+    experiment = os.getenv("MLFLOW_EXP")
+    if not tracking_uri and not experiment:
+        return False
+
+    try:
+        import mlflow
+    except ModuleNotFoundError:
+        log.warning(
+            "mlflow.setup.skipped",
+            reason="mlflow package not installed",
+            uri=tracking_uri,
+            experiment=experiment,
+        )
+        return False
+
+    try:
+        mlflow.pydantic_ai.autolog()
+    except Exception:
+        log.warning("mlflow.autolog.failed", exc_info=True)
+    try:
+        if tracking_uri:
+            mlflow.set_tracking_uri(tracking_uri)
+        if experiment:
+            mlflow.set_experiment(experiment)
+    except Exception:
+        log.warning("mlflow.setup.failed", exc_info=True)
+        return False
+
+    _MLFLOW_CONFIGURED = True
+    log.info(
+        "mlflow.setup.enabled",
+        uri=tracking_uri,
+        experiment=experiment,
+    )
+    return True
+
 
 @app.meta.default
 def meta_handler(
@@ -175,6 +222,7 @@ def query(
 
         # Create pydantic-ai agent instance with database name for memory context
         sqlsaber_agent = SQLSaberAgent(db_conn, db_name, thinking_enabled=thinking)
+        _maybe_configure_mlflow(log)
 
         try:
             if actual_query:
@@ -188,6 +236,7 @@ def query(
                 run = await streaming_handler.execute_streaming_query(
                     actual_query, sqlsaber_agent
                 )
+
                 # Persist non-interactive run as a thread snapshot so it can be resumed later
                 try:
                     if run is not None:

{sqlsaber-0.31.0 → sqlsaber-0.32.0}/tests/test_cli/test_commands.py

@@ -1,9 +1,11 @@
 """Tests for CLI commands."""
 
-from unittest.mock import patch
+import sys
+from unittest.mock import MagicMock, patch
 
 import pytest
 
+from sqlsaber.cli import commands
 from sqlsaber.cli.commands import app
 from sqlsaber.config.database import DatabaseConfig
 
@@ -61,3 +63,47 @@ class TestCLICommands:
         assert "memory" in captured.out
         assert "models" in captured.out
         assert "auth" in captured.out
+
+    def test_maybe_configure_mlflow_no_env(self, monkeypatch):
+        """MLflow stays disabled when env vars are absent."""
+        monkeypatch.delenv("MLFLOW_URI", raising=False)
+        monkeypatch.delenv("MLFLOW_EXP", raising=False)
+        commands._MLFLOW_CONFIGURED = False
+        logger = MagicMock()
+
+        assert not commands._maybe_configure_mlflow(logger)
+        logger.warning.assert_not_called()
+
+    def test_maybe_configure_mlflow_with_env(self, monkeypatch):
+        """MLflow autolog is configured when env vars and package are present."""
+        monkeypatch.setenv("MLFLOW_URI", "http://localhost:5000")
+        monkeypatch.setenv("MLFLOW_EXP", "sqlsaber-bench")
+        commands._MLFLOW_CONFIGURED = False
+        autolog_called = MagicMock()
+        set_uri_called = MagicMock()
+        set_exp_called = MagicMock()
+
+        class FakeMlflow:
+            def __init__(self):
+                self.pydantic_ai = MagicMock(autolog=autolog_called)
+
+            def set_tracking_uri(self, uri):
+                set_uri_called(uri)
+
+            def set_experiment(self, exp):
+                set_exp_called(exp)
+
+        monkeypatch.setitem(sys.modules, "mlflow", FakeMlflow())
+        logger = MagicMock()
+
+        assert commands._maybe_configure_mlflow(logger)
+        autolog_called.assert_called_once()
+        set_uri_called.assert_called_once_with("http://localhost:5000")
+        set_exp_called.assert_called_once_with("sqlsaber-bench")
+        logger.info.assert_called()
+
+        # Ensure subsequent calls short-circuit
+        logger.reset_mock()
+        assert commands._maybe_configure_mlflow(logger)
+        logger.info.assert_not_called()
+        commands._MLFLOW_CONFIGURED = False

{sqlsaber-0.31.0 → sqlsaber-0.32.0}/uv.lock

@@ -1922,7 +1922,7 @@ wheels = [
 
 [[package]]
 name = "sqlsaber"
-version = "0.31.0"
+version = "0.32.0"
 source = { editable = "." }
 dependencies = [
     { name = "aiomysql" },

sqlsaber-0.31.0/AGENT.md

@@ -1,23 +0,0 @@
-# SQLSaber Agent Guide
-
-## Commands
-
-- **Run Python**: `uv run python`
-- **Run tests**: `uv run python -m pytest`
-- **Run single test**: `uv run python -m pytest tests/test_path/test_file.py::test_function`
-- **Lint**: `uv run ruff check --fix`
-- **Format**: `uv run ruff format`
-
-## Architecture
-
-- **CLI App**: Agentic SQL assistant for natural language to SQL
-- **Core modules**: `agents/` (AI logic), `cli/` (commands), `database/` (connections)
-- **Database support**: PostgreSQL, SQLite, MySQL via asyncpg/aiosqlite/aiomysql
-
-## Code Style
-
-- **Imports**: stdlib → 3rd party → local, use relative imports within modules
-- **Naming**: snake_case functions/vars, PascalCase classes, UPPER_SNAKE constants, `_private` methods
-- **Types**: Always use modern type hints (3.12+), async functions for I/O
-- **Errors**: Use try/finally for cleanup
-- **Docstrings**: Triple-quoted with Args/Returns sections