dao-ai 0.0.16__tar.gz → 0.0.18__tar.gz

{dao_ai-0.0.16 → dao_ai-0.0.18}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dao-ai
-Version: 0.0.16
+Version: 0.0.18
 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

dao_ai-0.0.16/config/examples/executive_assistant.yaml → dao_ai-0.0.18/config/examples/deep_research.yaml

@@ -34,8 +34,7 @@ resources:
       max_tokens: 8192
 
     deep_reasoning_llm: &deep_reasoning_llm
-      #name: databricks-claude-sonnet-4
-      name: databricks-meta-llama-3-3-70b-instruct
+      name: databricks-claude-sonnet-4
       temperature: 0.1
       max_tokens: 8192
       fallbacks: 
@@ -149,7 +148,7 @@ agents:
   executive_assistant_genie: &executive_assistant_genie
     name: executive_assistant_genie                    
     description: "Executive Assistant Genie Agent"
-    model: *deep_reasoning_llm                            
+    model: *reasoning_llm                            
     tools:                                        
       - *executive_assistant_genie_tool
       - *tavily_search_tool
@@ -235,6 +234,4 @@ app:
     - *executive_assistant_genie                                      
   orchestration:                                  
     swarm:                                    
-      model: *fast_llm                           
-
-
+      model: *fast_llm                           

dao_ai-0.0.18/config/examples/executive_assistant.yaml

@@ -0,0 +1,239 @@
+# yaml-language-server: $schema=../../schemas/model_config_schema.json
+
+variables:
+  tavily_search_key: &tavily_search_key                                                                                       
+    options:
+      - env: TAVILY_API_KEY                         
+      - scope: retail_consumer_goods                    
+        secret: TAVILY_API_KEY                      
+
+schemas:
+  executive_assistant_schema: &executive_assistant_schema
+    catalog_name: retail_consumer_goods                  
+    schema_name: store_ops                
+
+resources:
+  llms:
+
+    fast_llm: &fast_llm
+      name: databricks-meta-llama-3-1-8b-instruct 
+      temperature: 0.1                              
+      max_tokens: 8192                             
+
+    tool_calling_llm: &tool_calling_llm
+      name: databricks-claude-3-7-sonnet
+      temperature: 0.1
+      max_tokens: 8192
+      fallbacks:                                   
+        - databricks-meta-llama-3-3-70b-instruct
+
+    # LLM for complex reasoning tasks
+    reasoning_llm: &reasoning_llm
+      name: databricks-claude-3-7-sonnet
+      temperature: 0.1
+      max_tokens: 8192
+
+    deep_reasoning_llm: &deep_reasoning_llm
+      name: databricks-claude-sonnet-4
+      temperature: 0.1
+      max_tokens: 8192
+      fallbacks: 
+        - databricks-claude-3-7-sonnet
+
+    judge_llm: &judge_llm
+      name: databricks-claude-3-7-sonnet
+      temperature: 0.5                            
+      max_tokens: 8192
+
+    embedding_model: &embedding_model
+      name: databricks-gte-large-en              
+
+
+  tables:
+    all_tables:
+      schema: *executive_assistant_schema                        
+    #   name: employee_performance                              
+    
+    # products:
+    #   schema: *executive_assistant_schema
+    #   name: products
+
+    # inventory: 
+    #   schema: *executive_assistant_schema
+    #   name: inventory
+
+    # customers: 
+    #   schema: *executive_assistant_schema
+    #   name: customers
+
+    # managers:
+    #   schema: *executive_assistant_schema
+    #   name: managers
+
+    # employee_tasks: 
+    #   schema: *executive_assistant_schema
+    #   name: employee_tasks
+
+    # appointments: 
+    #   schema: *executive_assistant_schema
+    #   name: appointments
+
+    # evaluation: 
+    #   schema: *executive_assistant_schema
+    #   name: evaluation
+
+
+  functions:
+
+    all_functions: 
+      schema: *executive_assistant_schema
+    #   name: find_inventory_by_sku
+
+    # find_inventory_by_upc: &find_inventory_by_upc
+    #   schema: *executive_assistant_schema
+    #   name: find_inventory_by_upc     
+    
+    # find_product_by_sku: &find_product_by_sku
+    #   schema: *executive_assistant_schema
+    #   name: find_product_by_sku     
+
+    # find_product_by_upc: &find_product_by_upc
+    #   schema: *executive_assistant_schema
+    #   name: find_product_by_upc     
+
+    # find_store_by_number: &find_store_by_number
+    #   schema: *executive_assistant_schema
+    #   name: find_store_by_number     
+    
+    # find_store_inventory_by_sku: &find_store_inventory_by_sku
+    #   schema: *executive_assistant_schema
+    #   name: find_store_inventory_by_sku     
+
+    # find_store_inventory_by_upc: &find_store_inventory_by_upc
+    #   schema: *executive_assistant_schema
+    #   name: find_store_inventory_by_upc
+
+  genie_rooms:
+    executive_assistant_genie_room: &executive_assistant_genie_room
+      name: "Executive Assistant Genie Room"                       
+      description: "A room for Genie agents to interact" 
+      space_id: 01f05dd06c421ad6b522bf7a517cf6d2          
+
+
+tools:
+  executive_assistant_genie_tool: &executive_assistant_genie_tool
+    name: executive_assistant_genie_tool
+    function:
+      type: factory                                 
+      name: dao_ai.tools.create_genie_tool     
+      args:                                     
+        name: executive_assistant_genie_tool
+        description: Query the data warehouse for customer KPIs, business metrics, employee performance, inventory data, and other operational insights. Use this tool multiple times as needed to gather comprehensive data for executive-level analysis and recommendations.
+        genie_room: *executive_assistant_genie_room            
+
+
+  tavily_search_tool: &tavily_search_tool
+    name: tavily_search_tool
+    function:
+      type: factory
+      name: langchain_tavily.TavilySearch
+      args:
+        max_results: 5
+        topic: general
+        search_depth: advanced
+        include_domains: ~
+        exclude_domains: ~
+
+agents:
+  executive_assistant_genie: &executive_assistant_genie
+    name: executive_assistant_genie                    
+    description: "Executive Assistant Genie Agent"
+    model: *reasoning_llm                            
+    tools:                                        
+      - *executive_assistant_genie_tool
+      - *tavily_search_tool
+    prompt: |
+      You are an executive-level data analyst and strategic advisor for a retail consumer goods company. Your primary audience is C-suite executives and senior management who need actionable insights about business performance, customer KPIs, and strategic recommendations.
+
+      ## Your Core Responsibilities:
+      
+      ### 1. Customer KPI Analysis & Reporting
+      - Analyze customer acquisition, retention, lifetime value, and churn metrics
+      - Track customer satisfaction scores, NPS, and engagement metrics
+      - Monitor customer segment performance and behavior patterns
+      - Provide trend analysis and year-over-year comparisons
+      - Identify customer pain points and opportunities
+      
+      ### 2. Business Performance Insights
+      - Monitor revenue, profit margins, and growth trajectories
+      - Analyze inventory turnover, stockouts, and supply chain efficiency
+      - Track employee performance and operational KPIs
+      - Evaluate store performance across locations
+      - Assess product performance and category trends
+      
+      ### 3. Strategic Reasoning & Recommendations
+      - Provide context and root cause analysis for performance changes
+      - Identify correlations between different business metrics
+      - Suggest data-driven action plans and strategic initiatives
+      - Recommend resource allocation and investment priorities
+      - Highlight risks and opportunities based on data trends
+      
+      ## How to Operate:
+      
+      ### Tool Usage Strategy
+      - **Use multiple tool calls** when needed to build comprehensive answers
+      - Query different data sources to cross-validate insights
+      - Drill down into specific metrics when initial results show concerning trends
+      - Always verify assumptions with data before making recommendations
+
+      ### Communication Style
+      - **Executive Summary First**: Lead with key findings and recommendations
+      - **Data-Driven**: Support all claims with specific metrics and evidence
+      - **Action-Oriented**: Provide clear, implementable next steps
+      - **Risk-Aware**: Highlight potential challenges and mitigation strategies
+      - **Strategic Context**: Connect findings to broader business objectives
+      
+      ### Response Structure
+      1. **Executive Summary** (2-3 bullet points of key insights)
+      2. **Key Metrics & Analysis** (specific data points with context)
+      3. **Root Cause Analysis** (why trends are occurring)
+      4. **Strategic Recommendations** (actionable next steps)
+      5. **Risk Assessment** (potential challenges and opportunities)
+      
+      ## Guidelines:
+      
+      - Always query the data warehouse using your genie tool to get current, accurate information
+      - Make multiple queries if needed to build a complete picture
+      - Translate technical metrics into business impact language
+      - Provide both short-term tactical and long-term strategic recommendations
+      - When data shows concerning trends, immediately investigate potential causes
+      - Use industry benchmarks and best practices in your recommendations when relevant
+      - Be proactive in identifying trends before they become problems
+      
+      Remember: You are trusted advisor to senior leadership. Your insights directly influence major business decisions, resource allocation, and strategic direction. Ensure your analysis is thorough, your reasoning is sound, and your recommendations are actionable.
+
+
+app:
+  name: executive_assistant_agent                        
+  description: "An executive assistant application"
+  log_level: DEBUG                                  
+  registered_model:                                
+    schema: *executive_assistant_schema                        
+    name: executive_assistant_agent                          
+  endpoint_name: executive_assistant_agent    
+  environment_vars: 
+    TAVILY_API_KEY: "{{secrets/retail_consumer_goods/TAVILY_API_KEY}}" 
+  tags:                                            
+    business: rcg                                 
+    streaming: true                                
+  permissions:                                     
+    - principals: [users]                         
+      entitlements:
+        - CAN_QUERY                               
+  agents:                                        
+    - *executive_assistant_genie                                      
+  orchestration:                                  
+    swarm:                                    
+      model: *fast_llm                           
+
+

{dao_ai-0.0.16 → dao_ai-0.0.18}/config/hardware_store/supervisor.yaml

@@ -898,7 +898,7 @@ agents:
 app:
   name: hardware_store_supervisor                      # Application name  
   description: "Multi-agent system for retail customer service and product assistance"
-  log_level: INFO                                   # Logging level for the application
+  log_level: DEBUG                                   # Logging level for the application
   registered_model:                                 # MLflow registered model configuration
     schema: *retail_schema                          # Schema where model will be registered
     name: hardware_store_supervisor                           # Model name in MLflow registry
@@ -926,7 +926,7 @@ app:
     - dao_ai.hooks.null_hook
     - hardware_store.hooks.require_store_num_hook
   input_example:
-      messages:
+      input:
         - role: user
           content: Can you recommend a lamp to match my oak side tables?
       custom_inputs:

{dao_ai-0.0.16 → dao_ai-0.0.18}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "dao-ai"
-version = "0.0.16"
+version = "0.0.18"
 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" }

{dao_ai-0.0.16 → dao_ai-0.0.18}/schemas/model_config_schema.json

@@ -480,12 +480,35 @@
     "ChatPayload": {
       "additionalProperties": false,
       "properties": {
+        "input": {
+          "anyOf": [
+            {
+              "items": {
+                "$ref": "#/$defs/Message"
+              },
+              "type": "array"
+            },
+            {
+              "type": "null"
+            }
+          ],
+          "default": null,
+          "title": "Input"
+        },
         "messages": {
-          "items": {
-            "$ref": "#/$defs/Message"
-          },
-          "title": "Messages",
-          "type": "array"
+          "anyOf": [
+            {
+              "items": {
+                "$ref": "#/$defs/Message"
+              },
+              "type": "array"
+            },
+            {
+              "type": "null"
+            }
+          ],
+          "default": null,
+          "title": "Messages"
         },
         "custom_inputs": {
           "additionalProperties": true,
@@ -494,7 +517,6 @@
         }
       },
       "required": [
-        "messages",
         "custom_inputs"
       ],
       "title": "ChatPayload",
@@ -1200,13 +1222,18 @@
           "default": null
         },
         "name": {
-          "title": "Name",
-          "type": "string"
+          "anyOf": [
+            {
+              "type": "string"
+            },
+            {
+              "type": "null"
+            }
+          ],
+          "default": null,
+          "title": "Name"
         }
       },
-      "required": [
-        "name"
-      ],
       "title": "FunctionModel",
       "type": "object"
     },
@@ -2339,13 +2366,18 @@
           "default": null
         },
         "name": {
-          "title": "Name",
-          "type": "string"
+          "anyOf": [
+            {
+              "type": "string"
+            },
+            {
+              "type": "null"
+            }
+          ],
+          "default": null,
+          "title": "Name"
         }
       },
-      "required": [
-        "name"
-      ],
       "title": "TableModel",
       "type": "object"
     },

{dao_ai-0.0.16 → dao_ai-0.0.18}/src/dao_ai/agent_as_code.py

@@ -3,7 +3,7 @@ import sys
 import mlflow
 from loguru import logger
 from mlflow.models import ModelConfig
-from mlflow.pyfunc import ChatModel
+from mlflow.pyfunc import ResponsesAgent
 
 from dao_ai.config import AppConfig
 
@@ -17,6 +17,6 @@ log_level: str = config.app.log_level
 logger.remove()
 logger.add(sys.stderr, level=log_level)
 
-app: ChatModel = config.as_chat_model()
+app: ResponsesAgent = config.as_responses_agent()
 
 mlflow.models.set_model(app)