py/v0.2.0 (#73)

* v0.2.0

* chore: update maintainer

* fix(pkg-py): handle possibly lazy data frames

* chore: add tests to `make py-check`

* ci(pkg-py): Check tests before release

Author: gadenbuie

.github/workflows/py-release.yml

@@ -33,8 +33,9 @@ jobs:
       - name: 📦 Install the project
         run: uv sync --python ${{ env.PYTHON_VERSION }} --all-extras --all-groups
 
-      #   - name: 🧪 Check tests
-      #     run: make py-check-tests
+      - name: 🧪 Check tests
+        run: make py-check-tests
+
       - name: 📝 Check types
         working-directory: ./pkg-py
         run: make py-check-types

Makefile

@@ -111,11 +111,10 @@ r-docs-preview: ## [r] Build R docs
 
 .PHONY: py-setup
 py-setup:  ## [py] Setup python environment
-	uv sync --all-extras --all-groups
+	uv sync --all-extras --all-groups --upgrade
 
 .PHONY: py-check
-# py-check:  py-check-format py-check-types py-check-tests ## [py] Run python checks
-py-check:  py-check-format py-check-types ## [py] Run python checks
+py-check:  py-check-format py-check-types py-check-tests ## [py] Run python checks
 
 .PHONY: py-check-tox
 py-check-tox:  ## [py] Run python 3.9 - 3.12 checks with tox

pkg-py/CHANGELOG.md

@@ -5,7 +5,7 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.1.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
-## [UNRELEASED]
+## [0.2.0] - 2025-09-02
 
 * `querychat.init()` now accepts a `client` argument, replacing the previous `create_chat_callback` argument. (#60)
 

pkg-py/src/querychat/querychat.py

@@ -7,7 +7,14 @@
 import warnings
 from dataclasses import dataclass
 from pathlib import Path
-from typing import TYPE_CHECKING, Any, Callable, Optional, Protocol, Union
+from typing import (
+    TYPE_CHECKING,
+    Any,
+    Callable,
+    Optional,
+    Protocol,
+    Union,
+)
 
 import chatlas
 import chevron
@@ -217,8 +224,16 @@ def df_to_html(df: IntoFrame, maxrows: int = 5) -> str:
         HTML string representation of the table
 
     """
-    ndf = nw.from_native(df)
-    df_short = nw.from_native(df).head(maxrows)
+    df_short: nw.DataFrame[Any]
+
+    if isinstance(df, nw.LazyFrame):
+        ndf_eager = df.collect()
+        df_short = df.head(maxrows).collect()
+    elif isinstance(df, nw.DataFrame):
+        ndf_eager = df
+        df_short = df.head(maxrows)
+    else:
+        raise TypeError("df must be a Narwhals DataFrame or LazyFrame")
 
     # Generate HTML table
     table_html = df_short.to_pandas().to_html(
@@ -227,9 +242,9 @@ def df_to_html(df: IntoFrame, maxrows: int = 5) -> str:
     )
 
     # Add note about truncated rows if needed
-    if len(df_short) != len(ndf):
+    if len(df_short) != len(ndf_eager):
         rows_notice = (
-            f"\n\n(Showing only the first {maxrows} rows out of {len(ndf)}.)\n"
+            f"\n\n(Showing only the first {maxrows} rows out of {len(ndf_eager)}.)\n"
         )
     else:
         rows_notice = ""
@@ -400,11 +415,15 @@ def init(
     data_source_obj: DataSource
     if isinstance(data_source, sqlalchemy.Engine):
         data_source_obj = SQLAlchemySource(data_source, table_name)
-    else:
+    elif isinstance(data_source, (nw.DataFrame, nw.LazyFrame)):
         data_source_obj = DataFrameSource(
-            nw.from_native(data_source).to_pandas(),
+            nw.to_native(data_source),
             table_name,
         )
+    else:
+        raise TypeError(
+            "`data_source` must be a Narwhals DataFrame or LazyFrame, or a SQLAlchemy Engine",
+        )
     # Process greeting
     if greeting is None:
         print(

pyproject.toml

@@ -4,12 +4,20 @@ build-backend = "hatchling.build"
 
 [project]
 name = "querychat"
-version = "0.1.0"
+version = "0.2.0"
 description = "Chat with your data using natural language"
 readme = "pkg-py/README.md"
 requires-python = ">=3.9"
 license = { file = "pkg-py/LICENSE" }
-authors = [{ name = "Posit", email = "[email protected]" }]
+authors = [
+  { name = "Joe Cheng", email = "[email protected]" },
+  { name = "Garrick Aden-Buie", email = "[email protected]" },
+  { name = "Dan Chen", email = "[email protected]" },
+  { name = "Barret Schloerke", email = "[email protected]" }
+]
+maintainers = [
+  { name = "Garrick Aden-Buie", email = "[email protected]" }
+]
 dependencies = [
     "duckdb",
     "pandas",