dao-ai 0.0.28__tar.gz → 0.0.30__tar.gz

dao_ai-0.0.30/.python-version

@@ -0,0 +1 @@
+3.11

{dao_ai-0.0.28 → dao_ai-0.0.30}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dao-ai
-Version: 0.0.28
+Version: 0.0.30
 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
@@ -18,12 +18,13 @@ Classifier: Intended Audience :: Science/Research
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
 Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Classifier: Topic :: System :: Distributed Computing
-Requires-Python: >=3.12
+Requires-Python: >=3.11
 Requires-Dist: databricks-agents>=1.7.0
 Requires-Dist: databricks-langchain>=0.8.1
 Requires-Dist: databricks-mcp>=0.3.0

{dao_ai-0.0.28 → dao_ai-0.0.30}/pyproject.toml

@@ -4,11 +4,11 @@ build-backend = "hatchling.build"
 
 [project]
 name = "dao-ai"
-version = "0.0.28"
+version = "0.0.30"
 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" }
-requires-python = ">=3.12"
+requires-python = ">=3.11"
 authors = [
     { name = "Nate Fleming", email = "nate.fleming@databricks.com" },
     { name = "Nate Fleming", email = "nate.fleming@gmail.com" },
@@ -35,6 +35,7 @@ classifiers = [
     "License :: OSI Approved :: MIT License",
     "Operating System :: OS Independent",
     "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.11",
     "Programming Language :: Python :: 3.12",
     "Programming Language :: Python :: 3.13",
     "Topic :: Scientific/Engineering :: Artificial Intelligence",
@@ -166,7 +167,7 @@ exclude = [
 
 [tool.ruff]
 # Enable the Ruff formatter
-target-version = "py312"  # Specify your Python version
+target-version = "py311"  # Minimum supported Python version
 line-length = 88  # Same as Black's default
 indent-width = 4
 

{dao_ai-0.0.28 → dao_ai-0.0.30}/requirements.txt

@@ -52,6 +52,7 @@ grandalf==0.8
 graphene==3.4.3
 graphql-core==3.2.6
 graphql-relay==3.2.0
+greenlet==3.2.3
 grpcio==1.73.0
 grpcio-status==1.71.0
 gunicorn==23.0.0

{dao_ai-0.0.28 → dao_ai-0.0.30}/src/dao_ai/config.py

@@ -2009,21 +2009,45 @@ class AppConfig(BaseModel):
     def create_agent(
         self,
         w: WorkspaceClient | None = None,
+        vsc: "VectorSearchClient | None" = None,
+        pat: str | None = None,
+        client_id: str | None = None,
+        client_secret: str | None = None,
+        workspace_host: str | None = None,
     ) -> None:
         from dao_ai.providers.base import ServiceProvider
         from dao_ai.providers.databricks import DatabricksProvider
 
-        provider: ServiceProvider = DatabricksProvider(w=w)
+        provider: ServiceProvider = DatabricksProvider(
+            w=w,
+            vsc=vsc,
+            pat=pat,
+            client_id=client_id,
+            client_secret=client_secret,
+            workspace_host=workspace_host,
+        )
         provider.create_agent(self)
 
     def deploy_agent(
         self,
         w: WorkspaceClient | None = None,
+        vsc: "VectorSearchClient | None" = None,
+        pat: str | None = None,
+        client_id: str | None = None,
+        client_secret: str | None = None,
+        workspace_host: str | None = None,
     ) -> None:
         from dao_ai.providers.base import ServiceProvider
         from dao_ai.providers.databricks import DatabricksProvider
 
-        provider: ServiceProvider = DatabricksProvider(w=w)
+        provider: ServiceProvider = DatabricksProvider(
+            w=w,
+            vsc=vsc,
+            pat=pat,
+            client_id=client_id,
+            client_secret=client_secret,
+            workspace_host=workspace_host,
+        )
         provider.deploy_agent(self)
 
     def find_agents(

{dao_ai-0.0.28 → dao_ai-0.0.30}/src/dao_ai/providers/databricks.py

@@ -219,6 +219,13 @@ class DatabricksProvider(ServiceProvider):
         logger.debug("Creating agent...")
         mlflow.set_registry_uri("databricks-uc")
 
+        # Set up experiment for proper tracking
+        experiment: Experiment = self.get_or_create_experiment(config)
+        mlflow.set_experiment(experiment_id=experiment.experiment_id)
+        logger.debug(
+            f"Using experiment: {experiment.name} (ID: {experiment.experiment_id})"
+        )
+
         llms: Sequence[LLMModel] = list(config.resources.llms.values())
         vector_indexes: Sequence[IndexModel] = list(
             config.resources.vector_stores.values()

{dao_ai-0.0.28 → dao_ai-0.0.30}/src/dao_ai/tools/genie.py

@@ -298,9 +298,6 @@ def create_genie_tool(
     if isinstance(genie_room, dict):
         genie_room = GenieRoomModel(**genie_room)
 
-    space_id: AnyVariable = genie_room.space_id or os.environ.get(
-        "DATABRICKS_GENIE_SPACE_ID"
-    )
     space_id: AnyVariable = genie_room.space_id or os.environ.get(
         "DATABRICKS_GENIE_SPACE_ID"
     )

{dao_ai-0.0.28 → dao_ai-0.0.30}/tests/dao_ai/test_databricks.py

@@ -303,6 +303,75 @@ def test_function_model_api_scopes():
     assert function.api_scopes == ["sql.statement-execution"]
 
 
+@pytest.mark.unit
+def test_create_agent_sets_experiment():
+    """Test that create_agent properly sets up MLflow experiment before starting run."""
+    from unittest.mock import MagicMock, patch
+
+    import mlflow
+
+    from dao_ai.config import AppConfig
+    from dao_ai.providers.databricks import DatabricksProvider
+
+    # Create a minimal mock config
+    mock_config = MagicMock(spec=AppConfig)
+    mock_app = MagicMock()
+    mock_app.name = "test_app"
+    mock_app.code_paths = []
+    mock_app.pip_requirements = []
+    mock_app.input_example = None
+    mock_config.app = mock_app
+
+    # Mock resources
+    mock_resources = MagicMock()
+    mock_resources.llms = MagicMock(values=lambda: [])
+    mock_resources.vector_stores = MagicMock(values=lambda: [])
+    mock_resources.warehouses = MagicMock(values=lambda: [])
+    mock_resources.genie_rooms = MagicMock(values=lambda: [])
+    mock_resources.tables = MagicMock(values=lambda: [])
+    mock_resources.functions = MagicMock(values=lambda: [])
+    mock_resources.connections = MagicMock(values=lambda: [])
+    mock_resources.databases = MagicMock(values=lambda: [])
+    mock_resources.volumes = MagicMock(values=lambda: [])
+    mock_config.resources = mock_resources
+
+    # Create mock experiment
+    mock_experiment = MagicMock()
+    mock_experiment.experiment_id = "test_experiment_123"
+    mock_experiment.name = "/Users/test_user/test_app"
+
+    with (
+        patch.object(
+            DatabricksProvider, "get_or_create_experiment", return_value=mock_experiment
+        ) as mock_get_experiment,
+        patch.object(mlflow, "set_experiment") as mock_set_experiment,
+        patch.object(mlflow, "set_registry_uri"),
+        patch.object(mlflow, "start_run") as mock_start_run,
+        patch.object(mlflow, "set_tag"),
+        patch.object(mlflow.pyfunc, "log_model") as mock_log_model,
+        patch.object(mlflow, "register_model"),
+        patch("dao_ai.providers.databricks.MlflowClient"),
+        patch("dao_ai.providers.databricks.is_installed", return_value=True),
+        patch(
+            "dao_ai.providers.databricks.is_lib_provided",
+            return_value=True,
+        ),
+    ):
+        # Set up mock context managers
+        mock_start_run.return_value.__enter__.return_value = MagicMock()
+        mock_log_model.return_value = MagicMock(model_uri="test_uri")
+
+        # Create provider and call create_agent
+        provider = DatabricksProvider()
+        provider.create_agent(config=mock_config)
+
+        # Verify experiment was retrieved/created and set
+        mock_get_experiment.assert_called_once_with(mock_config)
+        mock_set_experiment.assert_called_once_with(
+            experiment_id=mock_experiment.experiment_id
+        )
+
+
 @pytest.mark.unit
 def test_create_agent_sets_framework_tags():
     """Test that create_agent sets framework and framework_version tags."""

dao_ai-0.0.28/.python-version

@@ -1 +0,0 @@
-3.12