Raise error message when DuckDB backend not found, closes #464

src/python/txtai/database/duckdb.py

@@ -6,7 +6,13 @@
 
 from tempfile import TemporaryDirectory
 
-import duckdb
+# Conditional import
+try:
+ import duckdb
+
+ DUCKDB = True
+except ImportError:
+ DUCKDB = False
 
 from .filedb import FileDB
 
@@ -16,6 +22,12 @@ class DuckDB(FileDB):
  Database instance backed by DuckDB.
  """
 
+ def __init__(self, config):
+ super().__init__(config)
+
+ if not DUCKDB:
+ raise ImportError('DuckDB is not available - install "database" extra to enable')
+
  def connect(self, path=":memory:"):
  # Create connection and start a transaction
  # pylint: disable=I1101

src/python/txtai/database/encoder/image.py

@@ -4,8 +4,6 @@
 
 from io import BytesIO
 
-from .base import Encoder
-
 # Conditional import
 try:
  from PIL import Image
@@ -14,6 +12,8 @@
 except ImportError:
  PIL = False
 
+from .base import Encoder
+
 
 class ImageEncoder(Encoder):
  """

test/python/testoptional.py

@@ -23,6 +23,7 @@ def setUpClass(cls):
  modules = [
  "annoy",
  "croniter",
+ "duckdb",
  "fastapi",
  "fasttext",
  "hnswlib",
@@ -106,8 +107,12 @@ def testDatabase(self):
  Test missing database dependencies
  """
 
+ from txtai.database import DuckDB
  from txtai.database import ImageEncoder
 
+ with self.assertRaises(ImportError):
+ DuckDB({})
+
  with self.assertRaises(ImportError):
  ImageEncoder()