- added IntoValues type alias

- Using IntoExprColumn on StarExpression
- fixed lhs type for LambdaExpression, and value type for ConstantExpression

Author: OutSquareCapital

_duckdb-stubs/__init__.pyi

@@ -12,7 +12,7 @@ if typing.TYPE_CHECKING:
     import pyarrow.lib
     from builtins import list as lst
     from collections.abc import Callable, Iterable, Sequence, Mapping
-    from ._typing import ParquetFieldIdsType, IntoExpr, IntoExprColumn, PythonLiteral
+    from ._typing import ParquetFieldIdsType, IntoExpr, IntoExprColumn, PythonLiteral, IntoValues
     from duckdb import sqltypes, func
 
 __all__: lst[str] = [
@@ -458,7 +458,7 @@ class DuckDBPyConnection:
     ) -> sqltypes.DuckDBPyType: ...
     def unregister(self, view_name: str) -> DuckDBPyConnection: ...
     def unregister_filesystem(self, name: str) -> None: ...
-    def values(self, *args: lst[PythonLiteral] | tuple[Expression, ...] | Expression) -> DuckDBPyRelation: ...
+    def values(self, *args: IntoValues) -> DuckDBPyRelation: ...
     def view(self, view_name: str) -> DuckDBPyRelation: ...
     @property
     def description(self) -> lst[tuple[str, sqltypes.DuckDBPyType, None, None, None, None, None]]: ...
@@ -1009,12 +1009,12 @@ class token_type:
 def CaseExpression(condition: IntoExpr, value: IntoExpr) -> Expression: ...
 def CoalesceOperator(*args: IntoExpr) -> Expression: ...
 def ColumnExpression(*args: str) -> Expression: ...
-def ConstantExpression(value: typing.Any) -> Expression: ...
+def ConstantExpression(value: PythonLiteral) -> Expression: ...
 def DefaultExpression() -> Expression: ...
 def FunctionExpression(function_name: str, *args: IntoExpr) -> Expression: ...
-def LambdaExpression(lhs: typing.Any, rhs: IntoExpr) -> Expression: ...
+def LambdaExpression(lhs: IntoExprColumn | tuple[IntoExprColumn, ...], rhs: IntoExpr) -> Expression: ...
 def SQLExpression(expression: str) -> Expression: ...
-def StarExpression(*, exclude: Iterable[str | Expression] | None = None) -> Expression: ...
+def StarExpression(*, exclude: Iterable[IntoExprColumn] | None = None) -> Expression: ...
 def aggregate(
     df: pandas.DataFrame,
     aggr_expr: str | Iterable[IntoExpr],
@@ -1433,9 +1433,7 @@ def union_type(
 ) -> sqltypes.DuckDBPyType: ...
 def unregister(view_name: str, *, connection: DuckDBPyConnection | None = None) -> DuckDBPyConnection: ...
 def unregister_filesystem(name: str, *, connection: DuckDBPyConnection | None = None) -> None: ...
-def values(
-    *args: lst[PythonLiteral] | tuple[Expression, ...] | Expression, connection: DuckDBPyConnection | None = None
-) -> DuckDBPyRelation: ...
+def values(*args: IntoValues, connection: DuckDBPyConnection | None = None) -> DuckDBPyRelation: ...
 def view(view_name: str, *, connection: DuckDBPyConnection | None = None) -> DuckDBPyRelation: ...
 def write_csv(
     df: pandas.DataFrame,

_duckdb-stubs/_typing.pyi

@@ -53,7 +53,8 @@ PythonLiteral: TypeAlias = (
 """Python objects that can be converted to a `ConstantExpression`."""
 # the field_ids argument to to_parquet and write_parquet has a recursive structure
 ParquetFieldIdsType: TypeAlias = Mapping[str, int | ParquetFieldIdsType]
-
+IntoValues: TypeAlias = list[PythonLiteral] | tuple[Expression, ...] | Expression
+"""Types that can be converted to a table of values."""
 IntoExprColumn: TypeAlias = Expression | str
 """Types that are, or can be used as a `ColumnExpression`."""
 IntoExpr: TypeAlias = IntoExprColumn | PythonLiteral