dao-ai 0.0.31__tar.gz → 0.0.32__tar.gz

{dao_ai-0.0.31 → dao_ai-0.0.32}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dao-ai
-Version: 0.0.31
+Version: 0.0.32
 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
@@ -26,24 +26,24 @@ Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Classifier: Topic :: System :: Distributed Computing
 Requires-Python: >=3.11
 Requires-Dist: databricks-agents>=1.7.0
-Requires-Dist: databricks-langchain>=0.8.1
+Requires-Dist: databricks-langchain>=0.11.0
 Requires-Dist: databricks-mcp>=0.3.0
 Requires-Dist: databricks-sdk[openai]>=0.67.0
-Requires-Dist: duckduckgo-search>=8.0.2
+Requires-Dist: ddgs>=9.9.3
 Requires-Dist: flashrank>=0.2.8
 Requires-Dist: gepa>=0.0.17
 Requires-Dist: grandalf>=0.8
 Requires-Dist: langchain-mcp-adapters>=0.1.10
 Requires-Dist: langchain-tavily>=0.2.11
-Requires-Dist: langchain>=0.3.27
+Requires-Dist: langchain>=1.1.3
 Requires-Dist: langgraph-checkpoint-postgres>=2.0.25
-Requires-Dist: langgraph-supervisor>=0.0.29
-Requires-Dist: langgraph-swarm>=0.0.14
-Requires-Dist: langgraph>=0.6.10
+Requires-Dist: langgraph-supervisor>=0.0.31
+Requires-Dist: langgraph-swarm>=0.1.0
+Requires-Dist: langgraph>=1.0.4
 Requires-Dist: langmem>=0.0.29
 Requires-Dist: loguru>=0.7.3
 Requires-Dist: mcp>=1.17.0
-Requires-Dist: mlflow>=3.5.1
+Requires-Dist: mlflow>=3.7.0
 Requires-Dist: nest-asyncio>=1.6.0
 Requires-Dist: openevals>=0.0.19
 Requires-Dist: openpyxl>=3.1.5
@@ -55,7 +55,7 @@ Requires-Dist: rich>=14.0.0
 Requires-Dist: scipy<=1.15
 Requires-Dist: sqlparse>=0.5.3
 Requires-Dist: tomli>=2.3.0
-Requires-Dist: unitycatalog-ai[databricks]>=0.3.0
+Requires-Dist: unitycatalog-ai[databricks]>=0.3.2
 Provides-Extra: databricks
 Requires-Dist: databricks-connect>=15.0.0; extra == 'databricks'
 Requires-Dist: databricks-vectorsearch>=0.63; extra == 'databricks'

{dao_ai-0.0.31 → dao_ai-0.0.32}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "dao-ai"
-version = "0.0.31"
+version = "0.0.32"
 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" }
@@ -44,24 +44,24 @@ classifiers = [
 ]
 dependencies = [
     "databricks-agents>=1.7.0",
-    "databricks-langchain>=0.8.1",
+    "databricks-langchain>=0.11.0",
     "databricks-mcp>=0.3.0",
     "databricks-sdk[openai]>=0.67.0",
-    "duckduckgo-search>=8.0.2",
+    "ddgs>=9.9.3",
     "flashrank>=0.2.8",
     "gepa>=0.0.17",
     "grandalf>=0.8",
-    "langchain>=0.3.27",
+    "langchain>=1.1.3",
     "langchain-mcp-adapters>=0.1.10",
     "langchain-tavily>=0.2.11",
-    "langgraph>=0.6.10",
+    "langgraph>=1.0.4",
     "langgraph-checkpoint-postgres>=2.0.25",
-    "langgraph-supervisor>=0.0.29",
-    "langgraph-swarm>=0.0.14",
+    "langgraph-supervisor>=0.0.31",
+    "langgraph-swarm>=0.1.0",
     "langmem>=0.0.29",
     "loguru>=0.7.3",
     "mcp>=1.17.0",
-    "mlflow>=3.5.1",
+    "mlflow>=3.7.0",
     "nest-asyncio>=1.6.0",
     "openevals>=0.0.19",
     "openpyxl>=3.1.5",
@@ -73,7 +73,7 @@ dependencies = [
     "scipy<=1.15",
     "sqlparse>=0.5.3",
     "tomli>=2.3.0",
-    "unitycatalog-ai[databricks]>=0.3.0",
+    "unitycatalog-ai[databricks]>=0.3.2",
 ]
 
 [project.scripts]

{dao_ai-0.0.31 → dao_ai-0.0.32}/requirements.txt

@@ -4,12 +4,13 @@ aiohttp-retry==2.9.1
 aiosignal==1.3.2
 alembic==1.16.2
 annotated-types==0.7.0
-anthropic==0.60.0
+anthropic==0.75.0
 anyio==4.9.0
 attrs==25.3.0
 blinker==1.9.0
 boto3==1.40.1
 botocore==1.40.1
+brotli==1.2.0
 cachetools==5.5.2
 certifi==2025.6.15
 cffi==1.17.1
@@ -23,17 +24,19 @@ cryptography==45.0.7
 cycler==0.12.1
 databricks-agents==1.7.0
 databricks-ai-bridge==0.8.0
-databricks-langchain==0.8.1
+databricks-langchain==0.11.0
 databricks-mcp==0.3.0
 databricks-sdk==0.67.0
 dataclasses-json==0.6.7
+ddgs==9.9.3
 deprecation==2.1.0
 distlib==0.3.9
 distro==1.9.0
 docker==7.1.0
-duckduckgo-search==8.0.4
+docstring-parser==0.17.0
 dydantic==0.0.8
 et-xmlfile==2.0.0
+fake-useragent==2.2.0
 fastapi==0.115.13
 filelock==3.18.0
 flashrank==0.2.10
@@ -57,12 +60,16 @@ grpcio==1.73.0
 grpcio-status==1.71.0
 gunicorn==23.0.0
 h11==0.16.0
+h2==4.3.0
 hf-xet==1.1.10
+hpack==4.1.0
 httpcore==1.0.9
 httpx==0.28.1
 httpx-sse==0.4.0
+huey==2.5.5
 huggingface-hub==0.35.0
 humanfriendly==10.0
+hyperframe==6.1.0
 identify==2.6.12
 idna==3.10
 importlib-metadata==8.7.0
@@ -77,22 +84,23 @@ jsonpointer==3.0.0
 jsonschema==4.25.1
 jsonschema-specifications==2025.9.1
 kiwisolver==1.4.8
-langchain==0.3.27
-langchain-anthropic==0.3.18
-langchain-community==0.3.26
-langchain-core==0.3.72
-langchain-mcp-adapters==0.1.10
-langchain-openai==0.3.24
-langchain-tavily==0.2.11
-langchain-text-splitters==0.3.9
-langgraph==0.6.10
+langchain==1.1.3
+langchain-anthropic==1.2.0
+langchain-classic==1.0.0
+langchain-community==0.4.1
+langchain-core==1.1.2
+langchain-mcp-adapters==0.2.0
+langchain-openai==1.1.1
+langchain-tavily==0.2.13
+langchain-text-splitters==1.0.0
+langgraph==1.0.4
 langgraph-checkpoint==2.1.2
 langgraph-checkpoint-postgres==2.0.25
-langgraph-prebuilt==0.6.3
+langgraph-prebuilt==1.0.5
 langgraph-sdk==0.2.9
-langgraph-supervisor==0.0.29
-langgraph-swarm==0.0.14
-langmem==0.0.29
+langgraph-supervisor==0.0.31
+langgraph-swarm==0.1.0
+langmem==0.0.30
 langsmith==0.4.1
 litellm==1.75.9
 loguru==0.7.3
@@ -104,9 +112,9 @@ marshmallow==3.26.1
 matplotlib==3.10.3
 mcp==1.17.0
 mdurl==0.1.2
-mlflow==3.5.1
-mlflow-skinny==3.5.1
-mlflow-tracing==3.5.1
+mlflow==3.7.0
+mlflow-skinny==3.7.0
+mlflow-tracing==3.7.0
 mpmath==1.3.0
 multidict==6.5.0
 mypy==1.16.1
@@ -115,7 +123,7 @@ nest-asyncio==1.6.0
 nodeenv==1.9.1
 numpy==1.26.4
 onnxruntime==1.23.2
-openai==1.108.2
+openai==2.9.0
 openevals==0.1.0
 openpyxl==3.1.5
 opentelemetry-api==1.34.1
@@ -144,7 +152,7 @@ pyasn1-modules==0.4.2
 pycparser==2.22
 pydantic==2.12.0
 pydantic-core==2.41.1
-pydantic-settings==2.10.0
+pydantic-settings==2.12.0
 pygments==2.19.2
 pyparsing==3.2.3
 pytest==8.4.1
@@ -155,7 +163,7 @@ pytz==2025.2
 pyyaml==6.0.2
 referencing==0.36.2
 regex==2024.11.6
-requests==2.32.4
+requests==2.32.5
 requests-toolbelt==1.0.0
 rich==14.0.0
 rpds-py==0.27.1
@@ -168,6 +176,7 @@ setuptools==80.9.0
 six==1.17.0
 smmap==5.0.2
 sniffio==1.3.1
+socksio==1.0.0
 sqlalchemy==2.0.41
 sqlparse==0.5.3
 sse-starlette==2.3.6
@@ -185,10 +194,11 @@ typing-extensions==4.15.0
 typing-inspect==0.9.0
 typing-inspection==0.4.2
 tzdata==2025.2
-unitycatalog-ai==0.3.1
+unitycatalog-ai==0.3.2
 unitycatalog-client==0.3.0
-unitycatalog-langchain==0.2.0
+unitycatalog-langchain==0.3.0
 urllib3==2.5.0
+uuid-utils==0.12.0
 uvicorn==0.34.3
 virtualenv==20.31.2
 werkzeug==3.1.1

{dao_ai-0.0.31 → dao_ai-0.0.32}/src/dao_ai/config.py

@@ -725,6 +725,46 @@ class WarehouseModel(BaseModel, IsDatabricksResource):
 
 
 class DatabaseModel(BaseModel, IsDatabricksResource):
+    """
+    Configuration for a Databricks Lakebase (PostgreSQL) database instance.
+
+    Authentication Model:
+    --------------------
+    This model uses TWO separate authentication contexts:
+
+    1. **Workspace API Authentication** (inherited from IsDatabricksResource):
+       - Uses ambient/default authentication (environment variables, notebook context, app service principal)
+       - Used for: discovering database instance, getting host DNS, checking instance status
+       - Controlled by: DATABRICKS_HOST, DATABRICKS_TOKEN env vars, or SDK default config
+
+    2. **Database Connection Authentication** (configured via client_id/client_secret OR user):
+       - Used for: connecting to the PostgreSQL database as a specific identity
+       - OAuth M2M: Set client_id, client_secret, workspace_host to connect as a service principal
+       - User Auth: Set user (and optionally password) to connect as a user identity
+
+    Example OAuth M2M Configuration:
+    ```yaml
+    databases:
+      my_lakebase:
+        name: my-database
+        client_id:
+          env: SERVICE_PRINCIPAL_CLIENT_ID
+        client_secret:
+          scope: my-scope
+          secret: sp-client-secret
+        workspace_host:
+          env: DATABRICKS_HOST
+    ```
+
+    Example User Configuration:
+    ```yaml
+    databases:
+      my_lakebase:
+        name: my-database
+        user: my-user@databricks.com
+    ```
+    """
+
     model_config = ConfigDict(use_enum_values=True, extra="forbid")
     name: str
     instance_name: Optional[str] = None
@@ -883,7 +923,7 @@ class DatabaseModel(BaseModel, IsDatabricksResource):
     def create(self, w: WorkspaceClient | None = None) -> None:
         from dao_ai.providers.databricks import DatabricksProvider
 
-        provider: DatabricksProvider = DatabricksProvider()
+        provider: DatabricksProvider = DatabricksProvider(w=w)
         provider.create_lakebase(self)
         provider.create_lakebase_instance_role(self)
 
@@ -1613,6 +1653,12 @@ class AppModel(BaseModel):
     chat_history: Optional[ChatHistoryModel] = None
     code_paths: list[str] = Field(default_factory=list)
     pip_requirements: list[str] = Field(default_factory=list)
+    python_version: Optional[str] = Field(
+        default="3.12",
+        description="Python version for Model Serving deployment. Defaults to 3.12 "
+        "which is supported by Databricks Model Serving. This allows deploying from "
+        "environments with different Python versions (e.g., Databricks Apps with 3.11).",
+    )
 
     @model_validator(mode="after")
     def validate_agents_not_empty(self):

{dao_ai-0.0.31 → dao_ai-0.0.32}/src/dao_ai/prompts.py

@@ -1,10 +1,10 @@
 from typing import Any, Callable, Optional, Sequence
 
-from langchain.prompts import PromptTemplate
 from langchain_core.messages import (
     BaseMessage,
     SystemMessage,
 )
+from langchain_core.prompts import PromptTemplate
 from langchain_core.runnables import RunnableConfig
 from loguru import logger