fix: duckdb skip create function if no numpy (#43)

sqlframe/base/util.py

@@ -265,6 +265,15 @@ def verify_openai_installed():
         )
 
 
+def verify_numpy_installed():
+    try:
+        import numpy  # noqa
+    except ImportError:
+        raise ImportError(
+            """Numpy is required for this functionality. `pip install "sqlframe[pandas]"` (also include your engine if needed) to install pandas/numpy."""
+        )
+
+
 def quote_preserving_alias_or_name(col: t.Union[exp.Column, exp.Alias]) -> str:
     from sqlframe.base.session import _BaseSession
 

sqlframe/duckdb/session.py

@@ -4,7 +4,7 @@
 from functools import cached_property
 
 from sqlframe.base.session import _BaseSession
-from sqlframe.base.util import soundex
+from sqlframe.base.util import soundex, verify_numpy_installed
 from sqlframe.duckdb.catalog import DuckDBCatalog
 from sqlframe.duckdb.dataframe import DuckDBDataFrame
 from sqlframe.duckdb.readwriter import (
@@ -41,7 +41,13 @@ def __init__(self, conn: t.Optional[DuckDBPyConnection] = None, *args, **kwargs)
 
         if not hasattr(self, "_conn"):
             conn = conn or duckdb.connect()
-            conn.create_function("SOUNDEX", lambda x: soundex(x), return_type=VARCHAR)
+            try:
+                # Creating a function requires numpy to be installed so if they don't have it, we'll just skip it
+                verify_numpy_installed()
+                conn.create_function("SOUNDEX", lambda x: soundex(x), return_type=VARCHAR)
+            except ImportError:
+                pass
+
             super().__init__(conn, *args, **kwargs)
 
     @classmethod