Fix: on python 3.14, adjust minimal dep versions to those that support 3.14. Update polars to the version that added UInt128 (which recently committed code uses). (#154)

For duckdb this is a minor version upgrade, so the new dep applies to all python versions. The type hint change in the code is needed by the new duckdb version's type hints.

Fixes agentune/aoa#411

Author: danarmak

agentune/core/database.py

@@ -314,7 +314,10 @@ def _mandatory_config_dict(self) -> dict[str, str]:
             'python_enable_replacements': 'False',
         }
 
-    def to_config_dict(self) -> dict[str, str]:
+    def to_config_dict(self) -> dict[str, str | bool | int | float | list[str]]:
+        """The verbose return type avoids a type error when passing the dict to duckdb.connect().
+        In practice, we return a dict containing only string values.
+        """
         set_fields = { k: str(v) for k, v
                        in cattrs.Converter().unstructure_attrs_asdict(self).items()
                        if v is not None and k != 'config' }

poetry.lock

@@ -45,18 +45,30 @@ packages = [{include = "agentune"}]
 
 [tool.poetry.dependencies]
 python = "^3.12.6"
-duckdb = ">=1.4.1,<1.5.0"
+duckdb = ">=1.4.2,<1.5.0"
 pyarrow = [
     { version = ">=19.0.1", python = "<3.14" },
     { version = ">=22.0.0", python = ">=3.14" }
 ]
-numpy = "^2.2.4"
+numpy = [
+    { version = "^2.2.4", python = "< 3.14" },
+    { version = "^2.3.4", python = ">= 3.14" }
+]
 janus = "^2.0.0"
-polars = "^1.27.1"
+polars = "^1.34.0"
 attrs = "^25.3.0"
-scikit-learn = "^1.6.1"
-scipy = "^1.15.3"
-scipy-stubs = "^1.15.3.0"
+scikit-learn = [
+    { version = "^1.6.1", python = "< 3.14" },
+    { version = "^1.7.2", python = ">= 3.14" }
+]
+scipy = [
+    { version = "^1.15.3", python = "< 3.14" },
+    { version = "^1.16.1", python = ">= 3.14" }
+]
+scipy-stubs = [
+    { version = "^1.15.3", python = "< 3.14" },
+    { version = "^1.16.1", python = ">= 3.14" }
+]
 cattrs = "^25.3.0"
 httpx = {extras = ["http2"], version = "^0.28.1"}
 frozendict = "^2.4.6"
@@ -87,7 +99,7 @@ addopts = [
 testpaths = [ "tests" ]
 pythonpath = ["."]
 tmp_path_retention_policy = "none"
-asyncio_mode = "auto"
+asyncio_mode =  "auto"
 markers = [
     "integration: marks tests as integration tests (deselected by default)"
 ]