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

{dao_ai-0.0.24 → dao_ai-0.0.26}/.gitignore

@@ -273,3 +273,6 @@ tags
 .databricks
 
 mlruns/
+
+# Generated bundle configurations (created from databricks.yaml.template at deployment time)
+databricks.yaml

{dao_ai-0.0.24 → dao_ai-0.0.26}/Makefile

@@ -64,7 +64,7 @@ schema: depends
 	@$(PYTHON) -c "from dao_ai.config import AppConfig; import json; print(json.dumps(AppConfig.model_json_schema(), indent=2))"
 
 test: 
-	$(PYTEST) $(TEST_DIR)
+	$(PYTEST) -ra --tb=short $(TEST_DIR)
 
 help:
 	$(info TOP_DIR: $(TOP_DIR))

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dao-ai
-Version: 0.0.24
+Version: 0.0.26
 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
@@ -29,6 +29,7 @@ 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: gepa>=0.0.17
 Requires-Dist: grandalf>=0.8
 Requires-Dist: langchain-mcp-adapters>=0.1.10
 Requires-Dist: langchain-tavily>=0.2.11
@@ -40,7 +41,7 @@ Requires-Dist: langgraph>=0.6.10
 Requires-Dist: langmem>=0.0.29
 Requires-Dist: loguru>=0.7.3
 Requires-Dist: mcp>=1.17.0
-Requires-Dist: mlflow>=3.4.0
+Requires-Dist: mlflow>=3.5.0
 Requires-Dist: nest-asyncio>=1.6.0
 Requires-Dist: openevals>=0.0.19
 Requires-Dist: openpyxl>=3.1.5
@@ -51,6 +52,7 @@ Requires-Dist: pyyaml>=6.0.2
 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
 Provides-Extra: databricks
 Requires-Dist: databricks-connect>=15.0.0; extra == 'databricks'

{dao_ai-0.0.24 → dao_ai-0.0.26}/config/examples/mcp_with_uc_connection.yaml

@@ -29,7 +29,7 @@ resources:
   
   connections:
     github_connection: &github_connection       # UC Connection for GitHub
-      name: github_u2m_connection               # UC Connection name
+      name: github_pat_connection_nfleming               # UC Connection name
 
 tools:
   # Example using UC Connection for GitHub MCP with explicit URL
@@ -39,34 +39,16 @@ tools:
       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)
+      #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
+      url: https://adb-984752964297111.11.azuredatabricks.net/api/2.0/mcp/vector-search/retail_consumer_goods/hardware_store
       client_id: *client_id
       client_secret: *client_secret
       workspace_host: *workspace_host
@@ -76,7 +58,7 @@ tools:
     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
+      url:  https://adb-984752964297111.11.azuredatabricks.net/api/2.0/mcp/functions/retail_consumer_goods/hardware_store
       client_id: *client_id
       client_secret: *client_secret
       workspace_host: *workspace_host
@@ -91,6 +73,18 @@ tools:
       client_secret: *client_secret
       workspace_host: *workspace_host
 
+
+prompts:
+  mcp_agent_prompt: &mcp_agent_prompt
+    schema: *retail_schema
+    name: mcp_agent_prompt
+    description: "MCP agent prompt"
+    default_template: |
+      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.
+
 agents:
   mcp_agent: &mcp_agent                          # MCP agent configuration
     name: mcp_agent                              # Agent identifier
@@ -100,18 +94,14 @@ agents:
       - *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.
+    prompt: *mcp_agent_prompt
       
 app:
-  name: mcp_agent                                # Application name  
+  name: mcp_with_uc_connection                                # 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
+    name: mcp_with_uc_connection                 # Model name in MLflow registry
   agents:                                        # List of agents included in the system
     - *mcp_agent                                 # MCP agent
   orchestration:                                 # Agent orchestration configuration