dao-ai 0.0.9__tar.gz → 0.0.12__tar.gz

{dao_ai-0.0.9 → dao_ai-0.0.12}/Makefile

@@ -18,7 +18,7 @@ UV := uv
 SYNC := $(UV) sync 
 BUILD := $(UV) build 
 PYTHON := $(UV) run python 
-EXPORT := $(UV) pip freeze --exclude-editable
+EXPORT := $(UV) pip freeze --exclude-editable | grep -v -E "(databricks-vectorsearch|pyspark|databricks-connect)" 
 PUBLISH := $(UV) run twine upload
 PYTEST := $(UV) run pytest -v -s
 RUFF_CHECK := $(UV) run ruff check --fix --ignore E501 

{dao_ai-0.0.9 → dao_ai-0.0.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dao-ai
-Version: 0.0.9
+Version: 0.0.12
 Summary: DAO AI: A modular, multi-agent orchestration framework for complex AI workflows. Supports agent handoff, tool integration, and dynamic configuration via YAML.
 Project-URL: Homepage, https://github.com/natefleming/dao-ai
 Project-URL: Documentation, https://natefleming.github.io/dao-ai
@@ -27,7 +27,6 @@ Requires-Python: >=3.12
 Requires-Dist: databricks-agents>=1.2.0
 Requires-Dist: databricks-langchain>=0.4.2
 Requires-Dist: databricks-sdk[openai]>=0.55.0
-Requires-Dist: databricks-vectorsearch>=0.56
 Requires-Dist: duckduckgo-search>=8.0.2
 Requires-Dist: grandalf>=0.8
 Requires-Dist: langchain-mcp-adapters>=0.1.9
@@ -51,6 +50,10 @@ Requires-Dist: rich>=14.0.0
 Requires-Dist: scipy<=1.15
 Requires-Dist: sqlparse>=0.5.3
 Requires-Dist: unitycatalog-ai[databricks]>=0.3.0
+Provides-Extra: databricks
+Requires-Dist: databricks-connect>=15.0.0; extra == 'databricks'
+Requires-Dist: databricks-vectorsearch>=0.56; extra == 'databricks'
+Requires-Dist: pyspark>=3.5.0; extra == 'databricks'
 Provides-Extra: dev
 Requires-Dist: mypy>=1.0.0; extra == 'dev'
 Requires-Dist: pre-commit>=3.0.0; extra == 'dev'

{dao_ai-0.0.9 → dao_ai-0.0.12}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "dao-ai"
-version = "0.0.9"
+version = "0.0.12"
 description = "DAO AI: A modular, multi-agent orchestration framework for complex AI workflows. Supports agent handoff, tool integration, and dynamic configuration via YAML."
 readme = "README.md"
 license = { text = "MIT" }
@@ -45,7 +45,6 @@ dependencies = [
     "databricks-agents>=1.2.0",
     "databricks-langchain>=0.4.2",
     "databricks-sdk[openai]>=0.55.0",
-    "databricks-vectorsearch>=0.56",
     "duckduckgo-search>=8.0.2",
     "grandalf>=0.8",
     "langchain>=0.3.27",
@@ -98,6 +97,11 @@ test = [
     "pytest-cov>=4.0.0",
     "pytest-mock>=3.10.0",
 ]
+databricks = [
+    "databricks-vectorsearch>=0.56",
+    "pyspark>=3.5.0",
+    "databricks-connect>=15.0.0",
+]
 
 [dependency-groups]
 dev = [
@@ -116,6 +120,11 @@ test = [
     "pytest-cov>=4.0.0",
     "pytest-mock>=3.10.0",
 ]
+databricks = [
+    "databricks-vectorsearch>=0.56",
+    "pyspark>=3.5.0",
+    "databricks-connect>=15.0.0",
+]
 all = [
     { include-group = "dev" },
     { include-group = "docs" },

{dao_ai-0.0.9 → dao_ai-0.0.12}/requirements.txt

@@ -20,10 +20,8 @@ contourpy==1.3.2
 cycler==0.12.1
 databricks-agents==1.2.0
 databricks-ai-bridge==0.5.1
-databricks-connect==16.3.4
 databricks-langchain==0.5.1
 databricks-sdk==0.61.0
-databricks-vectorsearch==0.56
 dataclasses-json==0.6.7
 deprecation==2.1.0
 distlib==0.3.9

{dao_ai-0.0.9 → dao_ai-0.0.12}/src/dao_ai/config.py

@@ -1,4 +1,5 @@
 import atexit
+import importlib
 import os
 import sys
 from abc import ABC, abstractmethod
@@ -14,7 +15,7 @@ from typing import (
     TypeAlias,
     Union,
 )
-import importlib
+
 from databricks.sdk import WorkspaceClient
 from databricks.sdk.credentials_provider import (
     CredentialsStrategy,

{dao_ai-0.0.9 → dao_ai-0.0.12}/src/dao_ai/utils.py

@@ -92,7 +92,7 @@ def load_function(function_name: str) -> Callable[..., Any]:
         >>> version = func("my_model")
     """
     logger.debug(f"Loading function: {function_name}")
-        
+
     try:
         # Split the FQN into module path and function name
         module_path, func_name = function_name.rsplit(".", 1)