dao-ai 0.0.22__tar.gz → 0.0.24__tar.gz

{dao_ai-0.0.22 → dao_ai-0.0.24}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dao-ai
-Version: 0.0.22
+Version: 0.0.24
 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
@@ -24,27 +24,28 @@ Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Classifier: Topic :: System :: Distributed Computing
 Requires-Python: >=3.12
-Requires-Dist: databricks-agents>=1.6.1
-Requires-Dist: databricks-langchain>=0.8.0
+Requires-Dist: databricks-agents>=1.7.0
+Requires-Dist: databricks-langchain>=0.8.1
+Requires-Dist: databricks-mcp>=0.3.0
 Requires-Dist: databricks-sdk[openai]>=0.67.0
 Requires-Dist: duckduckgo-search>=8.0.2
 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: langgraph-checkpoint-postgres>=2.0.23
+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.7
+Requires-Dist: langgraph>=0.6.10
 Requires-Dist: langmem>=0.0.29
 Requires-Dist: loguru>=0.7.3
-Requires-Dist: mcp>=1.14.1
+Requires-Dist: mcp>=1.17.0
 Requires-Dist: mlflow>=3.4.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: pydantic>=2.11.3
+Requires-Dist: pydantic>=2.12.0
 Requires-Dist: python-dotenv>=1.1.0
 Requires-Dist: pyyaml>=6.0.2
 Requires-Dist: rich>=14.0.0
@@ -653,7 +654,7 @@ test:
 #### 4. MCP (Model Context Protocol) Tools (`type: mcp`)
    MCP tools allow interaction with external services that implement the Model Context Protocol, supporting both HTTP and stdio transports.
 
-   **Configuration Example:**
+   **Configuration Example (Direct URL):**
    ```yaml
    tools:
      weather_tool_mcp:
@@ -664,8 +665,30 @@ test:
          transport: streamable_http
          url: http://localhost:8000/mcp
    ```
+   
+   **Configuration Example (Unity Catalog Connection):**
+   MCP tools can also use Unity Catalog Connections for secure, governed access with on-behalf-of-user capabilities. The connection provides OAuth authentication, while the URL specifies the endpoint:
+   ```yaml
+   resources:
+     connections:
+       github_connection:
+         name: github_u2m_connection  # UC Connection name
+   
+   tools:
+     github_mcp:
+       name: github_mcp
+       function:
+         type: mcp
+         name: github_mcp
+         transport: streamable_http
+         url: https://workspace.databricks.com/api/2.0/mcp/external/github_u2m_connection  # MCP endpoint URL
+         connection: *github_connection  # UC Connection provides OAuth authentication
+   ```
+   
    **Development:**
-   Ensure the MCP service is running and accessible at the specified URL or command. The framework will handle the MCP protocol communication automatically.
+   - **For direct URL connections**: Ensure the MCP service is running and accessible at the specified URL or command. Provide OAuth credentials (client_id, client_secret) or PAT for authentication.
+   - **For UC Connection**: URL is required to specify the endpoint. The connection provides OAuth authentication via the workspace client. Ensure the connection is configured in Unity Catalog with appropriate MCP scopes (`mcp.genie`, `mcp.functions`, `mcp.vectorsearch`, `mcp.external`).
+   - The framework will handle the MCP protocol communication automatically, including session management and authentication.
 
 ### Configuring New Agents
 

{dao_ai-0.0.22 → dao_ai-0.0.24}/README.md

@@ -581,7 +581,7 @@ test:
 #### 4. MCP (Model Context Protocol) Tools (`type: mcp`)
    MCP tools allow interaction with external services that implement the Model Context Protocol, supporting both HTTP and stdio transports.
 
-   **Configuration Example:**
+   **Configuration Example (Direct URL):**
    ```yaml
    tools:
      weather_tool_mcp:
@@ -592,8 +592,30 @@ test:
          transport: streamable_http
          url: http://localhost:8000/mcp
    ```
+   
+   **Configuration Example (Unity Catalog Connection):**
+   MCP tools can also use Unity Catalog Connections for secure, governed access with on-behalf-of-user capabilities. The connection provides OAuth authentication, while the URL specifies the endpoint:
+   ```yaml
+   resources:
+     connections:
+       github_connection:
+         name: github_u2m_connection  # UC Connection name
+   
+   tools:
+     github_mcp:
+       name: github_mcp
+       function:
+         type: mcp
+         name: github_mcp
+         transport: streamable_http
+         url: https://workspace.databricks.com/api/2.0/mcp/external/github_u2m_connection  # MCP endpoint URL
+         connection: *github_connection  # UC Connection provides OAuth authentication
+   ```
+   
    **Development:**
-   Ensure the MCP service is running and accessible at the specified URL or command. The framework will handle the MCP protocol communication automatically.
+   - **For direct URL connections**: Ensure the MCP service is running and accessible at the specified URL or command. Provide OAuth credentials (client_id, client_secret) or PAT for authentication.
+   - **For UC Connection**: URL is required to specify the endpoint. The connection provides OAuth authentication via the workspace client. Ensure the connection is configured in Unity Catalog with appropriate MCP scopes (`mcp.genie`, `mcp.functions`, `mcp.vectorsearch`, `mcp.external`).
+   - The framework will handle the MCP protocol communication automatically, including session management and authentication.
 
 ### Configuring New Agents
 

dao_ai-0.0.24/config/examples/mcp_with_uc_connection.yaml

@@ -0,0 +1,119 @@
+# yaml-language-server: $schema=../../schemas/model_config_schema.json
+
+variables:
+  client_id: &client_id                          # Service principal client ID
+    options:
+      - env: RETAIL_AI_DATABRICKS_CLIENT_ID
+      - scope: retail_ai
+        secret: RETAIL_AI_DATABRICKS_CLIENT_ID
+  client_secret: &client_secret                  # Service principal secret
+    options:
+      - env: RETAIL_AI_DATABRICKS_CLIENT_SECRET
+      - scope: retail_ai
+        secret: RETAIL_AI_DATABRICKS_CLIENT_SECRET
+  workspace_host: &workspace_host                # Databricks workspace URL
+    options:
+      - env: RETAIL_AI_DATABRICKS_HOST
+      - scope: retail_ai
+        secret: RETAIL_AI_DATABRICKS_HOST
+
+schemas:
+  retail_schema: &retail_schema                  # Unity Catalog schema configuration
+    catalog_name: nfleming                       # Unity Catalog name
+    schema_name: retail_ai                       # Schema within the catalog
+
+resources:
+  llms:
+    default_llm: &default_llm                    # Default LLM configuration
+      name: databricks-claude-3-7-sonnet        # Databricks serving endpoint name
+  
+  connections:
+    github_connection: &github_connection       # UC Connection for GitHub
+      name: github_u2m_connection               # UC Connection name
+
+tools:
+  # Example using UC Connection for GitHub MCP with explicit URL
+  github_mcp_uc: &github_mcp_uc
+    name: github_mcp                                # Tool name for agent reference
+    function:
+      type: mcp                                     # Tool type: MCP server
+      name: github_mcp                              # MCP function name
+      transport: streamable_http                    # Transport protocol
+      url: https://adb-984752964297111.11.azuredatabricks.net/api/2.0/mcp/external/github_u2m_connection  # MCP endpoint URL (optional with connection)
+      connection: *github_connection                # Reference to UC Connection (provides OAuth auth)
+
+  # Example using UC Connection with auto-constructed URL
+  # URL pattern: {workspace_host}/api/2.0/mcp/external/{connection_name}
+  github_mcp_auto: &github_mcp_auto
+    name: github_mcp_auto                           # Tool name for agent reference
+    function:
+      type: mcp                                     # Tool type: MCP server
+      name: github_mcp_auto                         # MCP function name
+      transport: streamable_http                    # Transport protocol
+      connection: *github_connection                # URL will be auto-constructed from workspace host + connection name
+
+  # Example using direct URL (existing approach)
+  weather_tool_mcp: &weather_tool_mcp
+    name: weather                                   # Tool name for agent reference
+    function:
+      type: mcp                                     # Tool type: MCP server
+      name: weather                                 # MCP function name
+      transport: streamable_http                    # Transport protocol
+      url: http://localhost:8000/mcp                # MCP server URL
+      
+  vector_search_mcp: &vector_search_mcp          # Vector search MCP tool configuration
+    name: vector_search_mcp                                  
+    function:
+      type: mcp                                  # MCP function type
+      name: vector_search_mcp                              
+      url: https://adb-984752964297111.11.azuredatabricks.net/api/2.0/mcp/vector-search/nfleming/retail_ai
+      client_id: *client_id
+      client_secret: *client_secret
+      workspace_host: *workspace_host
+
+  unity_catalog_mcp: &unity_catalog_mcp          # Unity Catalog MCP tool configuration
+    name: unity_catalog_mcp                                  
+    function:
+      type: mcp                                  # MCP function type
+      name: unity_catalog_mcp                              
+      url:  https://adb-984752964297111.11.azuredatabricks.net/api/2.0/mcp/functions/nfleming/retail_ai
+      client_id: *client_id
+      client_secret: *client_secret
+      workspace_host: *workspace_host
+
+  genie_mcp: &genie_mcp                          # Genie MCP tool configuration
+    name: genie_mcp                                  
+    function:
+      type: mcp                                  # MCP function type
+      name: genie_mcp                              
+      url:  https://adb-984752964297111.11.azuredatabricks.net/api/2.0/mcp/genie/01f01c91f1f414d59daaefd2b7ec82ea
+      client_id: *client_id
+      client_secret: *client_secret
+      workspace_host: *workspace_host
+
+agents:
+  mcp_agent: &mcp_agent                          # MCP agent configuration
+    name: mcp_agent                              # Agent identifier
+    model: *default_llm                          # Reference to LLM configuration
+    tools:                                       # Tools available to this agent
+      - *github_mcp_uc
+      - *vector_search_mcp
+      - *unity_catalog_mcp
+      - *genie_mcp
+    prompt: |                                    # System prompt defining agent behavior
+      You are a multi-purpose agent that can perform vector search, interact with Unity Catalog,
+      interact with Genie rooms, and interact with GitHub. Use the tools provided to answer 
+      questions and perform tasks. You many need to use the tools in combination to achieve 
+      complex tasks.
+      
+app:
+  name: mcp_agent                                # Application name  
+  log_level: DEBUG                               # Logging level for the application
+  registered_model:                              # MLflow registered model configuration
+    schema: *retail_schema                       # Schema where model will be registered
+    name: mcp_agent                              # Model name in MLflow registry
+  agents:                                        # List of agents included in the system
+    - *mcp_agent                                 # MCP agent
+  orchestration:                                 # Agent orchestration configuration
+    swarm:                                       # Supervisor orchestration pattern
+      model: *default_llm                        # LLM for routing decisions