querychat 0.2.0__tar.gz → 0.2.1__tar.gz

{querychat-0.2.0 → querychat-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: querychat
-Version: 0.2.0
+Version: 0.2.1
 Summary: Chat with your data using natural language
 Project-URL: Homepage, https://github.com/posit-dev/querychat
 Project-URL: Repository, https://github.com/posit-dev/querychat

{querychat-0.2.0 → querychat-0.2.1}/pkg-py/src/querychat/querychat.py

@@ -18,7 +18,7 @@ from typing import (
 
 import chatlas
 import chevron
-import narwhals as nw
+import narwhals.stable.v1 as nw
 import sqlalchemy
 from shiny import Inputs, Outputs, Session, module, reactive, ui
 
@@ -26,7 +26,7 @@ from ._utils import temp_env_vars
 
 if TYPE_CHECKING:
     import pandas as pd
-    from narwhals.typing import IntoFrame
+    from narwhals.stable.v1.typing import IntoFrame
 
 from .datasource import DataFrameSource, DataSource, SQLAlchemySource
 
@@ -224,16 +224,15 @@ def df_to_html(df: IntoFrame, maxrows: int = 5) -> str:
         HTML string representation of the table
 
     """
-    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)
+    ndf = nw.from_native(df)
+
+    if isinstance(ndf, (nw.LazyFrame, nw.DataFrame)):
+        df_short = ndf.lazy().head(maxrows).collect()
+        nrow_full = ndf.lazy().select(nw.len()).collect().item()
     else:
-        raise TypeError("df must be a Narwhals DataFrame or LazyFrame")
+        raise TypeError(
+            "Must be able to convert `df` into a Narwhals DataFrame or LazyFrame",
+        )
 
     # Generate HTML table
     table_html = df_short.to_pandas().to_html(
@@ -242,9 +241,9 @@ def df_to_html(df: IntoFrame, maxrows: int = 5) -> str:
     )
 
     # Add note about truncated rows if needed
-    if len(df_short) != len(ndf_eager):
+    if len(df_short) != nrow_full:
         rows_notice = (
-            f"\n\n(Showing only the first {maxrows} rows out of {len(ndf_eager)}.)\n"
+            f"\n\n(Showing only the first {maxrows} rows out of {nrow_full}.)\n"
         )
     else:
         rows_notice = ""

{querychat-0.2.0 → querychat-0.2.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "querychat"
-version = "0.2.0"
+version = "0.2.1"
 description = "Chat with your data using natural language"
 readme = "pkg-py/README.md"
 requires-python = ">=3.9"