wrenai 0.7.0__tar.gz

wrenai-0.7.0/.claude/CLAUDE.md

@@ -0,0 +1,76 @@
+# CLAUDE.md — wren package
+
+Python SDK and CLI for Wren Engine. Wraps `wren-core-py` (PyO3 bindings) + Ibis connectors into a single installable package with YAML-based MDL project management, named connection profiles, and optional semantic memory.
+
+## Build & Development
+
+```bash
+cd wren
+just install          # build wren-core-py wheel + uv sync
+just install-all      # with all optional extras (including memory)
+just install-extra <extra>   # e.g. just install-extra postgres
+just install-memory   # install memory extra (lancedb + sentence-transformers)
+just dev              # run `wren` CLI
+just test             # pytest tests/
+just test-memory      # memory-specific tests
+just lint             # ruff format --check + ruff check
+just format           # ruff auto-fix (also aliased as `just fmt`)
+just build            # uv build (produces wheel)
+```
+
+Uses `uv` (not Poetry). `pyproject.toml` uses `hatchling` as build backend.
+
+## CLI Command Groups
+
+- `wren query` / `wren dry-plan` / `wren validate` — Core query operations
+- `wren context init|build|validate|show` — YAML MDL project management
+- `wren profile add|list|show|remove|activate` — Named connection profiles
+- `wren docs connection-info` — Generate connection field docs
+- `wren utils parse-type` — SQL type normalization
+- `wren memory index|fetch|store|recall` — Semantic memory (when `wren[memory]` installed)
+
+## Key Design Points
+
+- **WrenEngine** is the main entry point. It accepts a base64-encoded MDL JSON string, a `DataSource`, and a connection dict.
+- **Query flow**: `_plan()` → wren-core `SessionContext.transform_sql()` → `_transpile()` via sqlglot → connector `.query()`.
+- **Manifest extraction**: `_plan()` tries to extract a minimal sub-manifest scoped to the query's referenced tables before calling wren-core — this reduces planning overhead. Falls back to the full manifest on error.
+- **`get_session_context` is `@cache`-decorated** — same `(manifest_str, function_path, properties, data_source)` tuple reuses the same SessionContext. Avoid mutating session state.
+- **Write dialect mapping**: `canner` → `trino`; file sources (`local_file`, `s3_file`, `minio_file`, `gcs_file`) → `duckdb`. All others use `data_source.name` directly.
+- **WrenEngine is a context manager** (`__enter__` / `__exit__` call `close()`).
+- **Profile-based workflow**: When no explicit `--connection-*` flags are given, the CLI auto-discovers the active profile from `~/.wren/profiles.yml`. Profiles store datasource type + connection fields.
+- **YAML MDL project**: `wren context build` compiles YAML model/view/relationship files from a project directory into `target/mdl.json`. `_require_mdl()` auto-discovers this target file.
+- **Config system**: `~/.wren/config.json` with `strict_mode` (reject queries referencing non-MDL tables) and `denied_functions` (block specific SQL functions).
+- **Field registry** (`model/field_registry.py`): Single source of truth for per-datasource connection fields, derived from Pydantic models. Used by CLI interactive prompts, MCP web UI forms, and documentation generation.
+
+## Connectors
+
+`connector/factory.py` dispatches on `DataSource` to return the right connector. Each connector wraps an Ibis backend and exposes `.query(sql, limit)` and `.dry_run(sql)`. Base class in `connector/base.py`; Ibis-backed connectors share `connector/ibis.py`.
+
+- **Dedicated modules**: `postgres.py`, `mysql.py`, `mssql.py`, `bigquery.py`, `duckdb.py`, `oracle.py` (native oracledb, not Ibis), `redshift.py`, `spark.py`, `databricks.py`, `canner.py`
+- **Shared Ibis module** (`ibis.py`): trino, clickhouse, snowflake, athena
+- **File connectors**: `local_file`, `s3_file`, `minio_file`, `gcs_file` all map to duckdb
+- **doris** maps to mysql connector (MySQL-compatible protocol)
+- **canner** maps to postgres connector
+
+## Memory Module (Optional)
+
+`wren/src/wren/memory/` — LanceDB-backed semantic memory for schema and query retrieval. Install via `wren[memory]`.
+
+- **`WrenMemory`** — Main API: `index_manifest()`, `get_context()`, `store_query()`, `recall_queries()`, `describe_schema()`, `schema_is_current()`, `status()`, `reset()`
+- Uses sentence-transformers for embedding MDL schema items and NL↔SQL query pairs
+- **Seed queries** (`seed_queries.py`): On index, generates canonical NL-SQL pairs from the MDL manifest to bootstrap the query history
+- CLI: `wren memory index|fetch|store|recall` subcommands (auto-registered when extras installed)
+- Backing store: LanceDB (local or remote via opendal)
+
+## Optional Extras
+
+Install per data-source extras: `postgres`, `mysql`, `bigquery`, `snowflake`, `clickhouse`, `trino`, `mssql`, `databricks`, `redshift`, `spark`, `athena`, `oracle`, `memory`, `all`, `dev`.
+
+On macOS, `mysql` extra needs:
+```bash
+PKG_CONFIG_PATH="$(brew --prefix mysql-client)/lib/pkgconfig" just install-extra mysql
+```
+
+## Dependency on wren-core-py
+
+`wren-core-py` wheel is built locally from `../wren-core-py/` and installed via `--find-links`. Run `just build-core` (or `just install`) to rebuild after Rust changes.

wrenai-0.7.0/.gitignore

@@ -0,0 +1,73 @@
+# ---- IDE & editors ----
+.idea/
+.vscode/
+*.iml
+*.ipr
+*.iws
+*.swp
+*~
+.run/
+
+# ---- OS ----
+.DS_Store
+__MACOSX/
+
+# ---- Environment files (templates kept) ----
+**/.env
+**/.env.*
+!**/.env.example
+!**/.env.*.example
+
+# ---- Secrets / credentials ----
+*.pem
+
+# ---- Python ----
+__pycache__/
+*.py[cod]
+*$py.class
+*.so
+*.egg
+*.egg-info/
+.eggs/
+build/
+dist/
+.python-version
+.tox/
+.coverage
+.coverage.*
+htmlcov/
+.mypy_cache/
+.ruff_cache/
+.pytest_cache/
+.hypothesis/
+venv/
+.venv/
+
+# ---- Rust ----
+target/
+**/*.rs.bk
+*.pdb
+
+# ---- WebAssembly (wasm-pack) ----
+pkg/
+
+# ---- Node / TypeScript (skills, sdks, wasm consumers) ----
+node_modules/
+.pnp
+.pnp.js
+npm-debug.log*
+yarn-debug.log*
+yarn-error.log*
+*.tsbuildinfo
+
+# ---- SQLite (memory layer caches) ----
+*.sqlite
+*.sqlite3
+
+# ---- Local caches & temporary files ----
+local_cache/
+.tmp/
+.tmp
+
+# ---- Claude Code worktrees (local dev only) ----
+**/.claude/worktrees/

wrenai-0.7.0/CHANGELOG.md

@@ -0,0 +1,76 @@
+# Changelog
+
+## [0.7.0](https://github.com/Canner/WrenAI/compare/wren-v0.6.0...wren-v0.7.0) (2026-05-22)
+
+
+### ⚠ BREAKING CHANGES
+
+* **wren:** rename PyPI package from wren-engine to wrenai ([#2315](https://github.com/Canner/WrenAI/issues/2315))
+
+### Features
+
+* **wasm:** full Cube support — validate, translate, PyO3, CLI, WASM, docs ([#2282](https://github.com/Canner/WrenAI/issues/2282)) ([026111e](https://github.com/Canner/WrenAI/commit/026111e54ec31e7165f9fd79c5c998070e66626c))
+
+
+### Documentation
+
+* **wren:** rebrand README to Wren AI and expand rename migration note ([#2316](https://github.com/Canner/WrenAI/issues/2316)) ([f3a00eb](https://github.com/Canner/WrenAI/commit/f3a00eb971a27186824954eff64b32f7e290db3c))
+
+
+### Miscellaneous Chores
+
+* **wren:** rename PyPI package from wren-engine to wrenai ([#2315](https://github.com/Canner/WrenAI/issues/2315)) ([20cffa9](https://github.com/Canner/WrenAI/commit/20cffa904f2d47c048c9247a77687b3fdfe24416))
+
+## [0.6.0](https://github.com/Canner/WrenAI/compare/wren-v0.5.0...wren-v0.6.0) (2026-05-13)
+
+
+### Features
+
+* **context:** bind a connection profile to a project ([#2251](https://github.com/Canner/WrenAI/issues/2251)) ([41fbe41](https://github.com/Canner/WrenAI/commit/41fbe411fd1f1bb7a4080fbcedc7c886678276d1))
+
+## [0.5.0](https://github.com/Canner/WrenAI/compare/wren-v0.4.0...wren-v0.5.0) (2026-05-05)
+
+
+### Features
+
+* **core:** import wren-engine into core/ ([cc9b67f](https://github.com/Canner/WrenAI/commit/cc9b67f593bf94c7418e0abb0ed46aa4a21613c3))
+* **core:** import wren-engine into core/ ([#2209](https://github.com/Canner/WrenAI/issues/2209)) ([8b8a1a3](https://github.com/Canner/WrenAI/commit/8b8a1a3c5bf2a43d56ea1587782a0d5d853803b2))
+
+## [0.4.0](https://github.com/Canner/wren-engine/compare/wren-v0.3.0...wren-v0.4.0) (2026-04-30)
+
+
+### Features
+
+* **wren:** .env-driven profile secrets, auto connection validation, and wren-install-guide skill ([#1588](https://github.com/Canner/wren-engine/issues/1588)) ([38ceaf1](https://github.com/Canner/wren-engine/commit/38ceaf1f73a91bf123e609aa6e402d5b971d3340))
+
+## [0.3.0](https://github.com/Canner/wren-engine/compare/wren-v0.2.0...wren-v0.3.0) (2026-04-20)
+
+
+### Features
+
+* add MDL layout versioning and dialect field on Model and View ([#1556](https://github.com/Canner/wren-engine/issues/1556)) ([0384931](https://github.com/Canner/wren-engine/commit/03849312d5934c606c0e43d0bd41d091892b4454))
+* add wren-core-wasm module with browser WASM support ([#1568](https://github.com/Canner/wren-engine/issues/1568)) ([4f9201b](https://github.com/Canner/wren-engine/commit/4f9201b7f98ce34fba9069b10da7a52b2c338b8b))
+* **wren-core:** add refSql model support ([#1555](https://github.com/Canner/wren-engine/issues/1555)) ([bdaddf2](https://github.com/Canner/wren-engine/commit/bdaddf25bb9c10287b7f2062e727c26ac0e76303))
+* **wren:** add `wren docs connection-info` CLI command ([#1507](https://github.com/Canner/wren-engine/issues/1507)) ([87efbfd](https://github.com/Canner/wren-engine/commit/87efbfdb91aece1c46999a3cc1bae72b77de778b))
+* **wren:** add LanceDB-backed memory layer for schema and query retrieval ([#1494](https://github.com/Canner/wren-engine/issues/1494)) ([dfdff01](https://github.com/Canner/wren-engine/commit/dfdff010371f649abfa999b9fd08a729b81b15f2))
+* **wren:** add memory list, forget, dump & load commands ([#1531](https://github.com/Canner/wren-engine/issues/1531)) ([de424e8](https://github.com/Canner/wren-engine/commit/de424e8b08d4125456895935e99dfd79ddd21f3f))
+* **wren:** add profile management for named connection profiles ([#1509](https://github.com/Canner/wren-engine/issues/1509)) ([b086576](https://github.com/Canner/wren-engine/commit/b086576fbc0ead4ce46af915403a5c7268eac525))
+* **wren:** add standalone wren Python SDK package ([#1471](https://github.com/Canner/wren-engine/issues/1471)) ([41f3f21](https://github.com/Canner/wren-engine/commit/41f3f21f97aba6418b087c1d6437183fdd30f2b3))
+* **wren:** CLI 0.2.0 — context management, profiles, strict mode & memory ([#1522](https://github.com/Canner/wren-engine/issues/1522)) ([fbec650](https://github.com/Canner/wren-engine/commit/fbec650d4e44a62a3ed7fa3a943c74af83d63402))
+* **wren:** CTE-based SQL planning with per-model expansion ([#1479](https://github.com/Canner/wren-engine/issues/1479)) ([fca2b11](https://github.com/Canner/wren-engine/commit/fca2b11bb4d2d45d883803f75605f0b84571188f))
+* **wren:** extend standalone CLI with MySQL support and auto-discovery ([#1476](https://github.com/Canner/wren-engine/issues/1476)) ([6a78c12](https://github.com/Canner/wren-engine/commit/6a78c1210e0bd36e34984f85ae7e240a70b5ef0f))
+* **wren:** generate AGENTS.md during `wren context init` ([#1526](https://github.com/Canner/wren-engine/issues/1526)) ([40bf46f](https://github.com/Canner/wren-engine/commit/40bf46f636d38f39f01fc005583d71c1679a2507))
+* **wren:** preserve SELECT * in CTE rewriter ([#1536](https://github.com/Canner/wren-engine/issues/1536)) ([ed03388](https://github.com/Canner/wren-engine/commit/ed03388f7534a4f85b85a4fd3b6fc8a36179425e))
+
+
+### Bug Fixes
+
+* **oracle:** replace ibis[oracle] with native oracledb cursor connector ([#1495](https://github.com/Canner/wren-engine/issues/1495)) ([2e3c1de](https://github.com/Canner/wren-engine/commit/2e3c1def9645dcab3e0105bfa7053e740d162f83))
+* **wren:** address CodeRabbit review feedback ([9c46f55](https://github.com/Canner/wren-engine/commit/9c46f554d0e25ce1000be11a496718289092e81d))
+* **wren:** fix CLI 0.2.0 docs — description placement, install extras, CLI flags ([#1523](https://github.com/Canner/wren-engine/issues/1523)) ([536988e](https://github.com/Canner/wren-engine/commit/536988edeb1055620e0204f25243cbc70ee25f8a))
+* **wren:** suppress model-loading noise and improve memory CLI error message ([#1529](https://github.com/Canner/wren-engine/issues/1529)) ([bd20444](https://github.com/Canner/wren-engine/commit/bd204445d81805e3b54cb9f441177f06c5e45eb8))
+
+
+### Dependencies
+
+* **wren:** bump transitive deps for security patches ([53c16c4](https://github.com/Canner/wren-engine/commit/53c16c4a36eb7dbe1bb467c4cb05b0b64b6ec902))

wrenai-0.7.0/PKG-INFO

@@ -0,0 +1,364 @@
+Metadata-Version: 2.4
+Name: wrenai
+Version: 0.7.0
+Summary: Wren Engine CLI and Python SDK — semantic SQL layer for 20+ data sources
+Project-URL: Homepage, https://getwren.ai
+Project-URL: Repository, https://github.com/Canner/WrenAI
+Project-URL: Issues, https://github.com/Canner/WrenAI/issues
+Author-email: Wren AI <contact@getwren.ai>
+License: Apache-2.0
+Keywords: analytics,cli,data-modeling,database,datafusion,mdl,python,sdk,semantic,semantic-layer,sql,wren,wrenai
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: >=3.11
+Requires-Dist: boto3>=1.26
+Requires-Dist: duckdb>=1.0
+Requires-Dist: ibis-framework>=10
+Requires-Dist: loguru>=0.7
+Requires-Dist: opendal>=0.45
+Requires-Dist: pandas>=2
+Requires-Dist: pyarrow-hotfix>=0.6
+Requires-Dist: pyarrow>=14
+Requires-Dist: pyasn1>=0.6.3
+Requires-Dist: pydantic>=2
+Requires-Dist: pyopenssl>=26.0.0
+Requires-Dist: python-dotenv>=1.0
+Requires-Dist: pyyaml>=6.0
+Requires-Dist: requests>=2.33.0
+Requires-Dist: sqlglot>=27
+Requires-Dist: typer>=0.12
+Requires-Dist: wren-core-py>=0.1
+Provides-Extra: all
+Requires-Dist: clickhouse-connect>=0.8; extra == 'all'
+Requires-Dist: databricks-sdk; extra == 'all'
+Requires-Dist: databricks-sql-connector; extra == 'all'
+Requires-Dist: google-auth; extra == 'all'
+Requires-Dist: ibis-framework[bigquery]; extra == 'all'
+Requires-Dist: inquirerpy>=0.3.4; extra == 'all'
+Requires-Dist: jinja2>=3.1; extra == 'all'
+Requires-Dist: lancedb>=0.6; extra == 'all'
+Requires-Dist: mysqlclient>=2.2; extra == 'all'
+Requires-Dist: oracledb>=2; extra == 'all'
+Requires-Dist: psycopg[binary]>=3; extra == 'all'
+Requires-Dist: pyathena[pandas]>=3; extra == 'all'
+Requires-Dist: pyodbc<6,>=5; extra == 'all'
+Requires-Dist: pyspark>=3.5; extra == 'all'
+Requires-Dist: python-multipart>=0.0.9; extra == 'all'
+Requires-Dist: redshift-connector; extra == 'all'
+Requires-Dist: sentence-transformers>=2.2; extra == 'all'
+Requires-Dist: snowflake-connector-python[pandas]>=3.10; extra == 'all'
+Requires-Dist: starlette>=0.37; extra == 'all'
+Requires-Dist: trino<1,>=0.333; extra == 'all'
+Requires-Dist: uvicorn>=0.29; extra == 'all'
+Provides-Extra: athena
+Requires-Dist: pyathena[pandas]>=3; extra == 'athena'
+Provides-Extra: bigquery
+Requires-Dist: google-auth; extra == 'bigquery'
+Requires-Dist: ibis-framework[bigquery]; extra == 'bigquery'
+Provides-Extra: clickhouse
+Requires-Dist: clickhouse-connect>=0.8; extra == 'clickhouse'
+Provides-Extra: databricks
+Requires-Dist: databricks-sdk; extra == 'databricks'
+Requires-Dist: databricks-sql-connector; extra == 'databricks'
+Provides-Extra: dev
+Requires-Dist: httpx>=0.27; extra == 'dev'
+Requires-Dist: orjson>=3; extra == 'dev'
+Requires-Dist: pytest>=8; extra == 'dev'
+Requires-Dist: ruff>=0.4; extra == 'dev'
+Requires-Dist: testcontainers[mysql,postgres]>=4; extra == 'dev'
+Provides-Extra: interactive
+Requires-Dist: inquirerpy>=0.3.4; extra == 'interactive'
+Provides-Extra: main
+Requires-Dist: inquirerpy>=0.3.4; extra == 'main'
+Requires-Dist: jinja2>=3.1; extra == 'main'
+Requires-Dist: python-multipart>=0.0.9; extra == 'main'
+Requires-Dist: starlette>=0.37; extra == 'main'
+Requires-Dist: uvicorn>=0.29; extra == 'main'
+Provides-Extra: memory
+Requires-Dist: lancedb>=0.6; extra == 'memory'
+Requires-Dist: sentence-transformers>=2.2; extra == 'memory'
+Provides-Extra: mssql
+Requires-Dist: pyodbc<6,>=5; extra == 'mssql'
+Provides-Extra: mysql
+Requires-Dist: mysqlclient>=2.2; extra == 'mysql'
+Provides-Extra: oracle
+Requires-Dist: oracledb>=2; extra == 'oracle'
+Provides-Extra: postgres
+Requires-Dist: psycopg[binary]>=3; extra == 'postgres'
+Provides-Extra: redshift
+Requires-Dist: redshift-connector; extra == 'redshift'
+Provides-Extra: snowflake
+Requires-Dist: snowflake-connector-python[pandas]>=3.10; extra == 'snowflake'
+Provides-Extra: spark
+Requires-Dist: pyspark>=3.5; extra == 'spark'
+Provides-Extra: trino
+Requires-Dist: trino<1,>=0.333; extra == 'trino'
+Provides-Extra: ui
+Requires-Dist: jinja2>=3.1; extra == 'ui'
+Requires-Dist: python-multipart>=0.0.9; extra == 'ui'
+Requires-Dist: starlette>=0.37; extra == 'ui'
+Requires-Dist: uvicorn>=0.29; extra == 'ui'
+Description-Content-Type: text/markdown
+
+# wrenai
+
+[![PyPI version](https://img.shields.io/pypi/v/wrenai.svg)](https://pypi.org/project/wrenai/)
+[![Python](https://img.shields.io/pypi/pyversions/wrenai.svg)](https://pypi.org/project/wrenai/)
+[![License](https://img.shields.io/pypi/l/wrenai.svg)](https://github.com/Canner/WrenAI/blob/main/LICENSE)
+
+Wren AI CLI and Python SDK — semantic SQL layer for 20+ data sources.
+
+Translate natural SQL queries through an [MDL (Modeling Definition Language)](https://docs.getwren.ai/) semantic layer and execute them against your database. Powered by [Apache DataFusion](https://datafusion.apache.org/).
+
+## Installation
+
+```bash
+pip install wrenai              # Core (DuckDB included)
+pip install wrenai[postgres]    # PostgreSQL
+pip install wrenai[mysql]       # MySQL
+pip install wrenai[bigquery]    # BigQuery
+pip install wrenai[snowflake]   # Snowflake
+pip install wrenai[clickhouse]  # ClickHouse
+pip install wrenai[trino]       # Trino
+pip install wrenai[mssql]       # SQL Server
+pip install wrenai[databricks]  # Databricks
+pip install wrenai[redshift]    # Redshift
+pip install wrenai[spark]       # Spark
+pip install wrenai[athena]      # Athena
+pip install wrenai[oracle]      # Oracle
+pip install 'wrenai[memory]'    # Schema & query memory (LanceDB)
+pip install 'wrenai[ui]'        # Browser-based profile form (starlette + uvicorn)
+pip install 'wrenai[main]'      # memory + interactive prompts + ui
+pip install 'wrenai[all]'       # All connectors + main
+```
+
+Requires Python 3.11+.
+
+## Quick start
+
+**1. Initialize a project** — scaffolds a YAML-based MDL project:
+
+```bash
+mkdir my-project && cd my-project
+wren context init
+```
+
+This creates `wren_project.yml`, `models/`, and `views/`. Edit `wren_project.yml` to set your `data_source` and add models under `models/`:
+
+```yaml
+# wren_project.yml
+schema_version: 2
+name: my_project
+catalog: wren
+schema: public
+data_source: postgres
+```
+
+```yaml
+# models/orders/metadata.yml
+name: orders
+table_reference:
+  schema: mydb
+  table: orders
+columns:
+  - name: order_id
+    type: integer
+  - name: customer_id
+    type: integer
+  - name: total
+    type: double
+  - name: status
+    type: varchar
+primary_key: order_id
+```
+
+> **Already have an MDL JSON?** Import it directly:
+> `wren context init --from-mdl path/to/mdl.json`
+
+**2. Configure a connection profile:**
+
+```bash
+# Browser form (recommended, requires wrenai[ui])
+wren profile add my-db --ui
+
+# Interactive terminal prompts
+wren profile add my-db --interactive
+
+# Import from an existing connection file
+wren profile add my-db --from-file connection_info.json
+```
+
+**3. Build the manifest:**
+
+```bash
+wren context build
+```
+
+This compiles YAML files into `target/mdl.json`. The CLI auto-discovers this file when you run queries from within the project directory.
+
+**4. Run queries:**
+
+```bash
+wren --sql 'SELECT order_id FROM "orders" LIMIT 10'
+```
+
+`wren` walks up from the current directory to find `wren_project.yml` and uses `target/mdl.json`. You can also pass `--mdl path/to/mdl.json` explicitly.
+
+For the full CLI reference and per-datasource connection field reference, see [`docs/cli.md`](docs/cli.md) and [`docs/connections.md`](docs/connections.md).
+
+**4a. (Optional) Aggregation queries with cubes** — define cubes under `cubes/`,
+then query them with a structured input instead of writing `GROUP BY` SQL by hand:
+
+```bash
+wren cube list
+wren cube describe order_metrics
+wren cube query --cube order_metrics --measures revenue --time-dimension "created_at:month"
+```
+
+The translator produces `DATE_TRUNC` / `GROUP BY` / `WHERE` clauses for you and
+runs them through the same engine path as `wren --sql`. See the
+[Cube guide](../../docs/core/guides/modeling/cube.md) for full YAML structure
+and the [CLI reference](../../docs/core/reference/cli.md#wren-cube--pre-aggregation-queries) for all
+flags.
+
+**5. (Optional) Configure security policy** — create `~/.wren/config.json`:
+
+```json
+{
+  "strict_mode": true,
+  "denied_functions": ["pg_read_file", "dblink", "lo_import"]
+}
+```
+
+| Key | Default | Description |
+|-----|---------|-------------|
+| `strict_mode` | `false` | When `true`, every table in a query must be defined in the MDL. Queries referencing undeclared tables are rejected before execution. |
+| `denied_functions` | `[]` | List of function names (case-insensitive) that are forbidden in queries. |
+
+**6. (Optional) Index schema for semantic search** (requires `wrenai[memory]`):
+
+```bash
+wren memory index                              # index MDL schema
+wren memory fetch -q "customer order price"    # fetch relevant schema context
+wren memory store --nl "top customers" --sql "SELECT ..."  # store NL→SQL pair
+wren memory recall -q "best customers"         # retrieve similar past queries
+```
+
+---
+
+## Connection profiles
+
+Profiles let you store named connection configurations in `~/.wren/profiles.yml` and switch between them easily — useful when working across multiple databases or environments.
+
+```bash
+# Add a profile (browser form, interactive prompts, or file import)
+wren profile add prod --ui                        # opens http://localhost:<port>
+wren profile add staging --interactive            # terminal prompts
+wren profile add local --from-file conn.json      # import existing file
+
+# List and switch profiles
+wren profile list                                 # * marks the active profile
+wren profile switch prod
+
+# Inspect a profile (sensitive fields masked)
+wren profile debug prod
+
+# Remove a profile
+wren profile rm old-profile --force
+```
+
+The `--ui` flag opens a browser-based form that auto-derives fields from each datasource's schema — including file upload for BigQuery credentials, variant selection for Databricks/Redshift, and sensible defaults for all 20+ supported sources. Requires `pip install 'wrenai[ui]'`.
+
+Once a profile is active, `wren` uses it automatically:
+
+```bash
+wren profile switch prod
+wren --sql 'SELECT COUNT(*) FROM "orders"'        # connects using prod profile
+```
+
+---
+
+## Python SDK
+
+```python
+import base64, orjson
+from wren import WrenEngine, DataSource
+
+manifest = { ... }  # your MDL dict
+manifest_str = base64.b64encode(orjson.dumps(manifest)).decode()
+
+with WrenEngine(manifest_str, DataSource.mysql, {"host": "...", ...}) as engine:
+    result = engine.query('SELECT * FROM "orders" LIMIT 10')
+    print(result.to_pandas())
+```
+
+---
+
+## Development
+
+```bash
+just install-dev    # Install with dev dependencies
+just lint           # Ruff format check + lint
+just format         # Auto-fix
+```
+
+| Command | What it runs | Docker needed |
+|---------|-------------|---------------|
+| `just test-unit` | Unit tests (engine, CTE rewriter, field registry, profiles) | No |
+| `just test-duckdb` | DuckDB connector tests | No |
+| `just test-postgres` | PostgreSQL connector tests | Yes |
+| `just test-mysql` | MySQL connector tests | Yes |
+| `just test` | All tests | Yes |
+
+Profile web tests (`test_profile_web.py`) require `wrenai[ui]`:
+
+```bash
+uv sync --extra dev --extra ui --find-links ../wren-core-py/target/wheels/
+uv run pytest tests/test_profile_web.py -v
+```
+
+## Publishing
+
+```bash
+./scripts/publish.sh            # Build + publish to PyPI
+./scripts/publish.sh --test     # Build + publish to TestPyPI
+./scripts/publish.sh --build    # Build only
+```
+
+## Package rename: `wren-engine` → `wrenai`
+
+Starting with the 0.7.0 release, this PyPI distribution is renamed from
+[`wren-engine`](https://pypi.org/project/wren-engine/) to
+[`wrenai`](https://pypi.org/project/wrenai/) to align with the **Wren AI**
+brand. The legacy `wren-engine` project on PyPI is frozen at 0.6.x and
+will not receive further updates.
+
+### What stays the same
+
+- The Python import path: `import wren` (and submodules under `wren.*`)
+- The `wren` CLI entrypoint and every subcommand (`wren query`,
+  `wren context`, `wren profile`, `wren memory`, …)
+- All extras (`postgres`, `mysql`, `bigquery`, …, `memory`, `ui`, `main`,
+  `all`)
+- Configuration files under `~/.wren/` (profiles, memory, config)
+
+Only the name you type after `pip install` is different.
+
+### Migration
+
+```bash
+pip uninstall wren-engine
+pip install wrenai                  # or: pip install "wrenai[<extras>]"
+wren --version                      # should print: wrenai X.Y.Z
+```
+
+If your project pinned `wren-engine` in a `requirements.txt`,
+`pyproject.toml`, or lockfile, replace it with `wrenai` and re-lock.
+
+## License
+
+Apache-2.0