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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dao-ai
-Version: 0.0.31
+Version: 0.0.33
 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
@@ -25,29 +25,29 @@ Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 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-agents>=1.8.2
+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-mcp-adapters>=0.2.1
 Requires-Dist: langchain-tavily>=0.2.11
-Requires-Dist: langchain>=0.3.27
-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: langmem>=0.0.29
+Requires-Dist: langchain>=1.1.3
+Requires-Dist: langgraph-checkpoint-postgres>=3.0.2
+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.30
 Requires-Dist: loguru>=0.7.3
-Requires-Dist: mcp>=1.17.0
-Requires-Dist: mlflow>=3.5.1
+Requires-Dist: mcp>=1.23.3
+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
-Requires-Dist: psycopg[binary,pool]>=3.2.9
+Requires-Dist: psycopg[binary,pool]>=3.3.2
 Requires-Dist: pydantic>=2.12.0
 Requires-Dist: python-dotenv>=1.1.0
 Requires-Dist: pyyaml>=6.0.2
@@ -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.33}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "dao-ai"
-version = "0.0.31"
+version = "0.0.33"
 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" }
@@ -43,29 +43,29 @@ classifiers = [
     "Topic :: System :: Distributed Computing",
 ]
 dependencies = [
-    "databricks-agents>=1.7.0",
-    "databricks-langchain>=0.8.1",
+    "databricks-agents>=1.8.2",
+    "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-mcp-adapters>=0.1.10",
+    "langchain>=1.1.3",
+    "langchain-mcp-adapters>=0.2.1",
     "langchain-tavily>=0.2.11",
-    "langgraph>=0.6.10",
-    "langgraph-checkpoint-postgres>=2.0.25",
-    "langgraph-supervisor>=0.0.29",
-    "langgraph-swarm>=0.0.14",
-    "langmem>=0.0.29",
+    "langgraph>=1.0.4",
+    "langgraph-checkpoint-postgres>=3.0.2",
+    "langgraph-supervisor>=0.0.31",
+    "langgraph-swarm>=0.1.0",
+    "langmem>=0.0.30",
     "loguru>=0.7.3",
-    "mcp>=1.17.0",
-    "mlflow>=3.5.1",
+    "mcp>=1.23.3",
+    "mlflow>=3.7.0",
     "nest-asyncio>=1.6.0",
     "openevals>=0.0.19",
     "openpyxl>=3.1.5",
-    "psycopg[binary,pool]>=3.2.9",
+    "psycopg[binary,pool]>=3.3.2",
     "pydantic>=2.12.0",
     "python-dotenv>=1.1.0",
     "pyyaml>=6.0.2",
@@ -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.33}/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
@@ -21,19 +22,21 @@ coloredlogs==15.0.1
 contourpy==1.3.2
 cryptography==45.0.7
 cycler==0.12.1
-databricks-agents==1.7.0
+databricks-agents==1.8.2
 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.1
+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-checkpoint-postgres==3.0.2
+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
@@ -102,11 +110,11 @@ markdown-it-py==3.0.0
 markupsafe==3.0.2
 marshmallow==3.26.1
 matplotlib==3.10.3
-mcp==1.17.0
+mcp==1.23.3
 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
@@ -134,8 +142,8 @@ pre-commit==4.2.0
 primp==0.15.0
 propcache==0.3.2
 protobuf==5.29.5
-psycopg==3.2.9
-psycopg-binary==3.2.9
+psycopg==3.3.2
+psycopg-binary==3.3.2
 psycopg-pool==3.2.6
 py4j==0.10.9.7
 pyarrow==20.0.0
@@ -144,8 +152,9 @@ 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
+pyjwt==2.10.1
 pyparsing==3.2.3
 pytest==8.4.1
 python-dateutil==2.9.0.post0
@@ -155,7 +164,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 +177,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 +195,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.33}/schemas/model_config_schema.json

@@ -132,6 +132,17 @@
           "title": "Name",
           "type": "string"
         },
+        "service_principal": {
+          "anyOf": [
+            {
+              "$ref": "#/$defs/ServicePrincipalModel"
+            },
+            {
+              "type": "null"
+            }
+          ],
+          "default": null
+        },
         "description": {
           "anyOf": [
             {
@@ -434,6 +445,19 @@
           },
           "title": "Pip Requirements",
           "type": "array"
+        },
+        "python_version": {
+          "anyOf": [
+            {
+              "type": "string"
+            },
+            {
+              "type": "null"
+            }
+          ],
+          "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).",
+          "title": "Python Version"
         }
       },
       "required": [
@@ -671,6 +695,7 @@
     },
     "DatabaseModel": {
       "additionalProperties": false,
+      "description": "Configuration for a Databricks Lakebase (PostgreSQL) database instance.\n\nAuthentication Model:\n--------------------\nThis model uses TWO separate authentication contexts:\n\n1. **Workspace API Authentication** (inherited from IsDatabricksResource):\n   - Uses ambient/default authentication (environment variables, notebook context, app service principal)\n   - Used for: discovering database instance, getting host DNS, checking instance status\n   - Controlled by: DATABRICKS_HOST, DATABRICKS_TOKEN env vars, or SDK default config\n\n2. **Database Connection Authentication** (configured via service_principal, client_id/client_secret, OR user):\n   - Used for: connecting to the PostgreSQL database as a specific identity\n   - Service Principal: Set service_principal with workspace_host to connect as a service principal\n   - OAuth M2M: Set client_id, client_secret, workspace_host to connect as a service principal\n   - User Auth: Set user (and optionally password) to connect as a user identity\n\nExample Service Principal Configuration:\n```yaml\ndatabases:\n  my_lakebase:\n    name: my-database\n    service_principal:\n      client_id:\n        env: SERVICE_PRINCIPAL_CLIENT_ID\n      client_secret:\n        scope: my-scope\n        secret: sp-client-secret\n    workspace_host:\n      env: DATABRICKS_HOST\n```\n\nExample OAuth M2M Configuration (alternative):\n```yaml\ndatabases:\n  my_lakebase:\n    name: my-database\n    client_id:\n      env: SERVICE_PRINCIPAL_CLIENT_ID\n    client_secret:\n      scope: my-scope\n      secret: sp-client-secret\n    workspace_host:\n      env: DATABRICKS_HOST\n```\n\nExample User Configuration:\n```yaml\ndatabases:\n  my_lakebase:\n    name: my-database\n    user: my-user@databricks.com\n```",
       "properties": {
         "on_behalf_of_user": {
           "anyOf": [
@@ -941,6 +966,17 @@
           "default": null,
           "title": "Password"
         },
+        "service_principal": {
+          "anyOf": [
+            {
+              "$ref": "#/$defs/ServicePrincipalModel"
+            },
+            {
+              "type": "null"
+            }
+          ],
+          "default": null
+        },
         "client_id": {
           "anyOf": [
             {
@@ -1858,6 +1894,17 @@
           "default": null,
           "title": "Pat"
         },
+        "service_principal": {
+          "anyOf": [
+            {
+              "$ref": "#/$defs/ServicePrincipalModel"
+            },
+            {
+              "type": "null"
+            }
+          ],
+          "default": null
+        },
         "client_id": {
           "anyOf": [
             {
@@ -2713,6 +2760,74 @@
       "title": "SecretVariableModel",
       "type": "object"
     },
+    "ServicePrincipalModel": {
+      "properties": {
+        "client_id": {
+          "anyOf": [
+            {
+              "$ref": "#/$defs/CompositeVariableModel"
+            },
+            {
+              "$ref": "#/$defs/EnvironmentVariableModel"
+            },
+            {
+              "$ref": "#/$defs/SecretVariableModel"
+            },
+            {
+              "$ref": "#/$defs/PrimitiveVariableModel"
+            },
+            {
+              "type": "string"
+            },
+            {
+              "type": "integer"
+            },
+            {
+              "type": "number"
+            },
+            {
+              "type": "boolean"
+            }
+          ],
+          "title": "Client Id"
+        },
+        "client_secret": {
+          "anyOf": [
+            {
+              "$ref": "#/$defs/CompositeVariableModel"
+            },
+            {
+              "$ref": "#/$defs/EnvironmentVariableModel"
+            },
+            {
+              "$ref": "#/$defs/SecretVariableModel"
+            },
+            {
+              "$ref": "#/$defs/PrimitiveVariableModel"
+            },
+            {
+              "type": "string"
+            },
+            {
+              "type": "integer"
+            },
+            {
+              "type": "number"
+            },
+            {
+              "type": "boolean"
+            }
+          ],
+          "title": "Client Secret"
+        }
+      },
+      "required": [
+        "client_id",
+        "client_secret"
+      ],
+      "title": "ServicePrincipalModel",
+      "type": "object"
+    },
     "StorageType": {
       "enum": [
         "postgres",
@@ -3436,6 +3551,13 @@
       "title": "Variables",
       "type": "object"
     },
+    "service_principals": {
+      "additionalProperties": {
+        "$ref": "#/$defs/ServicePrincipalModel"
+      },
+      "title": "Service Principals",
+      "type": "object"
+    },
     "schemas": {
       "additionalProperties": {
         "$ref": "#/$defs/SchemaModel"