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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dao-ai
-Version: 0.0.32
+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,7 +25,7 @@ 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-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
@@ -33,21 +33,21 @@ 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>=1.1.3
-Requires-Dist: langgraph-checkpoint-postgres>=2.0.25
+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.29
+Requires-Dist: langmem>=0.0.30
 Requires-Dist: loguru>=0.7.3
-Requires-Dist: mcp>=1.17.0
+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

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

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "dao-ai"
-version = "0.0.32"
+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,7 +43,7 @@ classifiers = [
     "Topic :: System :: Distributed Computing",
 ]
 dependencies = [
-    "databricks-agents>=1.7.0",
+    "databricks-agents>=1.8.2",
     "databricks-langchain>=0.11.0",
     "databricks-mcp>=0.3.0",
     "databricks-sdk[openai]>=0.67.0",
@@ -52,20 +52,20 @@ dependencies = [
     "gepa>=0.0.17",
     "grandalf>=0.8",
     "langchain>=1.1.3",
-    "langchain-mcp-adapters>=0.1.10",
+    "langchain-mcp-adapters>=0.2.1",
     "langchain-tavily>=0.2.11",
     "langgraph>=1.0.4",
-    "langgraph-checkpoint-postgres>=2.0.25",
+    "langgraph-checkpoint-postgres>=3.0.2",
     "langgraph-supervisor>=0.0.31",
     "langgraph-swarm>=0.1.0",
-    "langmem>=0.0.29",
+    "langmem>=0.0.30",
     "loguru>=0.7.3",
-    "mcp>=1.17.0",
+    "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",

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

@@ -22,7 +22,7 @@ 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.11.0
 databricks-mcp==0.3.0
@@ -89,13 +89,13 @@ 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-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-checkpoint-postgres==3.0.2
 langgraph-prebuilt==1.0.5
 langgraph-sdk==0.2.9
 langgraph-supervisor==0.0.31
@@ -110,7 +110,7 @@ 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.7.0
 mlflow-skinny==3.7.0
@@ -142,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
@@ -154,6 +154,7 @@ pydantic==2.12.0
 pydantic-core==2.41.1
 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

{dao_ai-0.0.32 → 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"