dao-ai 0.0.6__tar.gz → 0.0.8__tar.gz

dao_ai-0.0.8/.python-version

@@ -0,0 +1 @@
+3.12

{dao_ai-0.0.6 → dao_ai-0.0.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dao-ai
-Version: 0.0.6
+Version: 0.0.8
 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
@@ -23,28 +23,27 @@ Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Classifier: Topic :: System :: Distributed Computing
-Requires-Python: >=3.11
+Requires-Python: >=3.12
 Requires-Dist: databricks-agents>=1.2.0
 Requires-Dist: databricks-langchain>=0.4.2
 Requires-Dist: databricks-sdk[openai]>=0.55.0
 Requires-Dist: databricks-vectorsearch>=0.56
 Requires-Dist: duckduckgo-search>=8.0.2
 Requires-Dist: grandalf>=0.8
-Requires-Dist: langchain-mcp-adapters>=0.1.4
-Requires-Dist: langchain>=0.3.24
-Requires-Dist: langgraph-checkpoint-postgres>=2.0.21
-Requires-Dist: langgraph-prebuilt<=0.2.3
-Requires-Dist: langgraph-supervisor>=0.0.27
-Requires-Dist: langgraph-swarm>=0.0.11
-Requires-Dist: langgraph<=0.4.10
-Requires-Dist: langmem>=0.0.27
+Requires-Dist: langchain-mcp-adapters>=0.1.9
+Requires-Dist: langchain>=0.3.27
+Requires-Dist: langgraph-checkpoint-postgres>=2.0.23
+Requires-Dist: langgraph-supervisor>=0.0.29
+Requires-Dist: langgraph-swarm>=0.0.14
+Requires-Dist: langgraph>=0.6.5
+Requires-Dist: langmem>=0.0.29
 Requires-Dist: loguru>=0.7.3
 Requires-Dist: mcp>=1.9.1
 Requires-Dist: mlflow>=3.1.1
 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.6
+Requires-Dist: psycopg[binary,pool]>=3.2.9
 Requires-Dist: pydantic>=2.11.3
 Requires-Dist: python-dotenv>=1.1.0
 Requires-Dist: pyyaml>=6.0.2

{dao_ai-0.0.6 → dao_ai-0.0.8}/config/dais2025/model_config_dais.yaml

@@ -61,7 +61,7 @@ resources:
     store_vector_store: &store_vector_store
       embedding_model: *embedding_model
       endpoint:
-        name: rct-store-ops-dais
+        name: dbdemos_vs_endpoint
         type: STANDARD
       index:
         schema: *dais_schema

{dao_ai-0.0.6 → dao_ai-0.0.8}/config/hardware_store/supervisor_postgres.yaml

@@ -14,10 +14,13 @@ schemas:
   retail_schema: &retail_schema
     catalog_name: retail_consumer_goods                    # Unity Catalog name
     schema_name: hardware_store                    # Schema within the catalog
-    # permissions:                              # Access permissions for the schema
-    #   - principals: [users]                   # Grant access to all users
-    #     privileges:
-    #       - ALL_PRIVILEGES                    # Full permissions for demo purposes
+    permissions:                              # Access permissions for the schema
+       - principals: [users]                   # Grant access to all users
+         privileges:
+           - USE_CATALOG                    # Full permissions for demo purposes
+           - USE_SCHEMA                    # Full permissions for demo purposes
+           - SELECT                    # Full permissions for demo purposes
+           - EXECUTE                    # Full permissions for demo purposes
 
 # =============================================================================
 # RESOURCES CONFIGURATION
@@ -33,7 +36,7 @@ resources:
   llms:
     # Primary LLM for general tasks
     default_llm: &default_llm
-      name: databricks-claude-3-7-sonnet  # Databricks serving endpoint name
+      name: databricks-claude-3-7-sonnet # Databricks serving endpoint name
       temperature: 0.1                              # Low temperature for consistent responses
       max_tokens: 8192                              # Maximum tokens per response
 
@@ -462,8 +465,8 @@ agents:
     model: *tool_calling_llm                       # Reference to LLM configuration
     tools:                                          # Tools available to this agent
       - *find_product_details_by_description_tool
-    guardrails:                                      # Quality checks (currently disabled)
-      - *llm_judge_guardrail
+    guardrails: []                                    # Quality checks (currently disabled)
+     # - *llm_judge_guardrail
     prompt: |                                       # System prompt defining agent behavior
       ### User Information
       - **User Id**: {user_id}
@@ -912,13 +915,13 @@ agents:
 # and agent orchestration patterns for the multi-agent system
 
 app:
-  name: hardware_store_postgres                      # Application name  
+  name: hardware_store_postgres_beta                      # Application name  
   description: "Multi-agent system for retail customer service and product assistance"
   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_postgres                  # Model name in MLflow registry
-  endpoint_name: hardware_store_postgres           # Model serving endpoint name
+    name: hardware_store_postgres_beta                  # Model name in MLflow registry
+  endpoint_name: hardware_store_postgres_beta           # Model serving endpoint name
   environment_vars: 
     PGHOST: "{{secrets/retail_ai/PGHOST}}"  # Databricks host URL
     RETAIL_AI_DATABRICKS_CLIENT_ID: "{{secrets/retail_ai/RETAIL_AI_DATABRICKS_CLIENT_ID}}"
@@ -926,7 +929,7 @@ app:
     RETAIL_AI_DATABRICKS_HOST: "{{secrets/retail_ai/RETAIL_AI_DATABRICKS_HOST}}"
   tags:                                             # Tags for resource organization
     business: rcg                                   # Business unit identifier
-    streaming: true                                 # Indicates streaming capabilities
+    streaming: True                                 # Indicates streaming capabilities
   permissions:                                      # Model serving permissions
     - principals: [users]                           # Grant access to all users
       entitlements:
@@ -943,6 +946,9 @@ app:
     memory: *memory                            # Memory store for conversation context
     supervisor:                                     # Supervisor orchestration pattern
       model: *tool_calling_llm                      # LLM for routing decisions
+    # swarm:
+    #   model: *tool_calling_llm
+    #   default_agent: *general
 
   message_hooks: 
     - dao_ai.hooks.null_hook
@@ -957,13 +963,10 @@ app:
           user_id: my_user_id
           store_num: 87887
   #chat_history: 
-    #max_tokens: 256
+  #  #max_tokens: 256
   #  max_messages_before_summary: 5
-  #  max_summary_tokens: 128
   #  model: *fast_llm                          # Model for summarization tasks
-    # swarm:
-    #   model: *tool_calling_llm
-    #   default_agent: *general
+
 
 # =============================================================================
 # UNITY CATALOG FUNCTIONS DEPLOYMENT
@@ -1033,6 +1036,12 @@ evaluation:
       thread_id: "1"
       user_id: nate_fleming
       store_num: 87887
+  guidelines:
+    - name: MyRelevanceGuideline
+      guidelines:
+        - Ensure responses are relevant to the user's query
+        - Prioritize information from the user's context
+        - Avoid unnecessary jargon or technical details
 
 # =============================================================================
 # DATASETS CONFIGURATION

{dao_ai-0.0.6 → dao_ai-0.0.8}/databricks.yaml

@@ -1,6 +1,5 @@
 # yaml-language-server: $schema=./schemas/bundle_config_schema.json
 
-
 bundle:
   name: dao-ai-bundle
 
@@ -31,11 +30,11 @@ resources:
             spark_version: 16.2.x-scala2.12
             num_workers: 1
       environments:
-        - environment_key: default
+        - environment_key: dao-ai-env
           spec:
-            client: "3"
+            client: "4"
             dependencies:
-              - "-r ${workspace.file_path}/requirements.txt"
+              - -r ${workspace.file_path}/requirements.txt
       tasks:
         - task_key: ingest-and-transform
           notebook_task:
@@ -50,7 +49,7 @@ resources:
             notebook_path: ./notebooks/02_provision_vector_search.py
             base_parameters:
               config-path: ${var.config_path}
-          environment_key: default
+          environment_key: dao-ai-env
         - task_key: generate-evaluation-data
           depends_on:
             - task_key: provision-vector-search
@@ -58,7 +57,7 @@ resources:
             notebook_path: ./notebooks/03_generate_evaluation_data.py
             base_parameters:
               config-path: ${var.config_path}
-          environment_key: default
+          environment_key: dao-ai-env
         - task_key: unity-catalog-tools
           depends_on:
             - task_key: provision-vector-search
@@ -66,7 +65,7 @@ resources:
             notebook_path: ./notebooks/04_unity_catalog_tools.py
             base_parameters:
               config-path: ${var.config_path}
-          environment_key: default
+          environment_key: dao-ai-env
         - task_key: deploy-agents
           depends_on:
             - task_key: unity-catalog-tools
@@ -74,7 +73,7 @@ resources:
             notebook_path: ./notebooks/05_agent_as_code_driver.py
             base_parameters:
               config-path: ${var.config_path}
-          environment_key: default
+          environment_key: dao-ai-env
         - task_key: run-evaluation
           depends_on:
             - task_key: deploy-agents
@@ -83,7 +82,7 @@ resources:
             notebook_path: ./notebooks/06_run_evaluation.py
             base_parameters:
               config-path: ${var.config_path}
-          environment_key: default
+          environment_key: dao-ai-env
           
 targets:
   dev:

dao_ai-0.0.8/environment.yaml

@@ -0,0 +1,52 @@
+environment_version: '3'
+dependencies:
+  - --index-url https://pypi.org/simple
+  # Core framework dependencies
+  - langchain==0.3.27
+  - langchain-core==0.3.72
+  - langchain-community==0.3.26
+  - langchain-anthropic==0.3.18
+  - langchain-openai==0.3.24
+  - langgraph==0.6.5
+  - langgraph-checkpoint==2.1.1
+  - langgraph-checkpoint-postgres==2.0.23
+  - langgraph-prebuilt==0.6.3
+  - langgraph-supervisor==0.0.29
+  - langmem==0.0.29
+  - langsmith==0.4.1
+  # Databricks dependencies
+  - databricks-agents==1.2.0
+  - databricks-ai-bridge==0.5.1
+  - databricks-langchain==0.5.1
+  - databricks-sdk==0.61.0
+  - databricks-vectorsearch==0.56
+  - unitycatalog-ai==0.3.1
+  - unitycatalog-langchain==0.2.0
+  # MLflow and AI model dependencies
+  - mlflow==3.1.1
+  - anthropic==0.60.0
+  - openai==1.90.0
+  - tiktoken==0.9.0
+  # Data processing dependencies
+  - pandas==2.3.0
+  - numpy==1.26.4
+  - pyarrow==20.0.0
+  - sqlparse==0.5.3
+  - pydantic==2.11.7
+  - pydantic-settings==2.10.0
+  # Database dependencies
+  - psycopg==3.2.9
+  - psycopg-binary==3.2.9
+  - psycopg-pool==3.2.6
+  - sqlalchemy==2.0.41
+  # Web framework dependencies
+  - fastapi==0.115.13
+  - uvicorn==0.34.3
+  - starlette==0.46.2
+  # Utilities
+  - loguru==0.7.3
+  - python-dotenv==1.1.0
+  - pyyaml==6.0.2
+  - requests==2.32.4
+  - tenacity==9.1.2
+  - typing-extensions==4.14.0

{dao_ai-0.0.6 → dao_ai-0.0.8}/notebooks/06_run_evaluation.py

@@ -225,14 +225,20 @@ eval_df: pd.DataFrame = spark.read.table(payload_table).toPandas()
 
 evaluation_table_name: str = config.evaluation.table.full_name
 
-scorers_list = [Safety(), clarity, response_completeness, tool_call_efficiency]
+scorers = [Safety(), clarity, response_completeness, tool_call_efficiency]
+custom_scorers = []
+
+if config.evaluation.guidelines:
+    custom_scorers = [Guidelines(name=guideline.name, guidelines=guideline.guidelines) for guideline in config.evaluation.guidelines]
+
+scorers += custom_scorers
 
 with mlflow.start_run(run_id=model_version.run_id):
   eval_results = mlflow.genai.evaluate(
       data=eval_df,
       predict_fn=predict_fn,
       model_id=model_version.model_id,
-      scorers=scorers_list,
+      scorers=scorers,
   )
 
 

{dao_ai-0.0.6 → dao_ai-0.0.8}/notebooks/08_run_examples.py

@@ -1,5 +1,7 @@
 # Databricks notebook source
 # MAGIC %pip install --quiet -r ../requirements.txt
+# MAGIC %pip uninstall -y databricks-connect pyspark pyspark-connect
+# MAGIC %pip install databricks-connect
 # MAGIC %restart_python
 
 # COMMAND ----------

{dao_ai-0.0.6 → dao_ai-0.0.8}/pyproject.toml

@@ -4,11 +4,11 @@ build-backend = "hatchling.build"
 
 [project]
 name = "dao-ai"
-version = "0.0.6"
+version = "0.0.8"
 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" }
-requires-python = ">=3.11"
+requires-python = ">=3.12"
 authors = [
     { name = "Nate Fleming", email = "nate.fleming@databricks.com" },
     { name = "Nate Fleming", email = "nate.fleming@gmail.com" },
@@ -48,21 +48,20 @@ dependencies = [
     "databricks-vectorsearch>=0.56",
     "duckduckgo-search>=8.0.2",
     "grandalf>=0.8",
-    "langchain>=0.3.24",
-    "langchain-mcp-adapters>=0.1.4",
-    "langgraph<=0.4.10",
-    "langgraph-checkpoint-postgres>=2.0.21",
-    "langgraph-prebuilt<=0.2.3",
-    "langgraph-supervisor>=0.0.27",
-    "langgraph-swarm>=0.0.11",
-    "langmem>=0.0.27",
+    "langchain>=0.3.27",
+    "langchain-mcp-adapters>=0.1.9",
+    "langgraph>=0.6.5",
+    "langgraph-checkpoint-postgres>=2.0.23",
+    "langgraph-supervisor>=0.0.29",
+    "langgraph-swarm>=0.0.14",
+    "langmem>=0.0.29",
     "loguru>=0.7.3",
     "mcp>=1.9.1",
     "mlflow>=3.1.1",
     "nest-asyncio>=1.6.0",
     "openevals>=0.0.19",
     "openpyxl>=3.1.5",
-    "psycopg[binary,pool]>=3.2.6",
+    "psycopg[binary,pool]>=3.2.9",
     "pydantic>=2.11.3",
     "python-dotenv>=1.1.0",
     "pyyaml>=6.0.2",

{dao_ai-0.0.6 → dao_ai-0.0.8}/requirements.txt

@@ -4,7 +4,7 @@ aiohttp-retry==2.9.1
 aiosignal==1.3.2
 alembic==1.16.2
 annotated-types==0.7.0
-anthropic==0.54.0
+anthropic==0.60.0
 anyio==4.9.0
 attrs==25.3.0
 blinker==1.9.0
@@ -20,7 +20,7 @@ contourpy==1.3.2
 cycler==0.12.1
 databricks-agents==1.2.0
 databricks-ai-bridge==0.5.1
-databricks-connect==16.1.6
+databricks-connect==16.3.4
 databricks-langchain==0.5.1
 databricks-sdk==0.61.0
 databricks-vectorsearch==0.56
@@ -65,21 +65,21 @@ joblib==1.5.1
 jsonpatch==1.33
 jsonpointer==3.0.0
 kiwisolver==1.4.8
-langchain==0.3.26
-langchain-anthropic==0.3.15
+langchain==0.3.27
+langchain-anthropic==0.3.18
 langchain-community==0.3.26
-langchain-core==0.3.66
-langchain-mcp-adapters==0.1.7
+langchain-core==0.3.72
+langchain-mcp-adapters==0.1.9
 langchain-openai==0.3.24
-langchain-text-splitters==0.3.8
-langgraph==0.4.8
-langgraph-checkpoint==2.1.0
-langgraph-checkpoint-postgres==2.0.21
-langgraph-prebuilt==0.2.2
-langgraph-sdk==0.1.70
-langgraph-supervisor==0.0.27
-langgraph-swarm==0.0.11
-langmem==0.0.27
+langchain-text-splitters==0.3.9
+langgraph==0.6.5
+langgraph-checkpoint==2.1.1
+langgraph-checkpoint-postgres==2.0.23
+langgraph-prebuilt==0.6.3
+langgraph-sdk==0.2.0
+langgraph-supervisor==0.0.29
+langgraph-swarm==0.0.14
+langmem==0.0.29
 langsmith==0.4.1
 loguru==0.7.3
 lxml==5.4.0

{dao_ai-0.0.6 → dao_ai-0.0.8}/schemas/model_config_schema.json

@@ -466,16 +466,9 @@
           "title": "Max Messages Before Summary"
         },
         "max_summary_tokens": {
-          "anyOf": [
-            {
-              "type": "integer"
-            },
-            {
-              "type": "null"
-            }
-          ],
-          "default": null,
-          "title": "Max Summary Tokens"
+          "default": 255,
+          "title": "Max Summary Tokens",
+          "type": "integer"
         }
       },
       "required": [
@@ -1097,6 +1090,13 @@
           "additionalProperties": true,
           "title": "Custom Inputs",
           "type": "object"
+        },
+        "guidelines": {
+          "items": {
+            "$ref": "#/$defs/GuidelineModel"
+          },
+          "title": "Guidelines",
+          "type": "array"
         }
       },
       "required": [
@@ -1264,6 +1264,28 @@
       "title": "GuardrailModel",
       "type": "object"
     },
+    "GuidelineModel": {
+      "additionalProperties": false,
+      "properties": {
+        "name": {
+          "title": "Name",
+          "type": "string"
+        },
+        "guidelines": {
+          "items": {
+            "type": "string"
+          },
+          "title": "Guidelines",
+          "type": "array"
+        }
+      },
+      "required": [
+        "name",
+        "guidelines"
+      ],
+      "title": "GuidelineModel",
+      "type": "object"
+    },
     "HumanInTheLoopModel": {
       "additionalProperties": false,
       "properties": {

{dao_ai-0.0.6 → dao_ai-0.0.8}/src/dao_ai/cli.py

@@ -3,6 +3,7 @@ import json
 import os
 import subprocess
 import sys
+import traceback
 from argparse import ArgumentParser, Namespace
 from pathlib import Path
 from typing import Optional, Sequence
@@ -388,7 +389,9 @@ def handle_chat_command(options: Namespace) -> None:
 
                 except Exception as e:
                     print(f"\n❌ Error during streaming: {e}")
+                    print(f"Stack trace:\n{traceback.format_exc()}")
                     logger.error(f"Streaming error: {e}")
+                    logger.error(f"Stack trace: {traceback.format_exc()}")
 
             except EOFError:
                 # Handle Ctrl-D

{dao_ai-0.0.6 → dao_ai-0.0.8}/src/dao_ai/config.py

@@ -28,6 +28,7 @@ from langchain_core.language_models import LanguageModelLike
 from langchain_core.runnables.base import RunnableLike
 from langchain_openai import ChatOpenAI
 from langgraph.checkpoint.base import BaseCheckpointSaver
+from langgraph.graph.state import CompiledStateGraph
 from langgraph.store.base import BaseStore
 from loguru import logger
 from mlflow.models import ModelConfig
@@ -41,10 +42,9 @@ from mlflow.models.resources import (
     DatabricksUCConnection,
     DatabricksVectorSearchIndex,
 )
+from mlflow.pyfunc import ChatModel
 from pydantic import BaseModel, ConfigDict, Field, field_serializer, model_validator
 
-from dao_ai.chat_models import ChatDatabricksFiltered
-
 
 class HasValue(ABC):
     @abstractmethod
@@ -275,9 +275,14 @@ class LLMModel(BaseModel, IsDatabricksResource):
         # chat_client: LanguageModelLike = self.as_open_ai_client()
 
         # Create ChatDatabricksWrapper instance directly
+        from dao_ai.chat_models import ChatDatabricksFiltered
+
         chat_client: LanguageModelLike = ChatDatabricksFiltered(
             model=self.name, temperature=self.temperature, max_tokens=self.max_tokens
         )
+        # chat_client: LanguageModelLike = ChatDatabricks(
+        #     model=self.name, temperature=self.temperature, max_tokens=self.max_tokens
+        # )
 
         fallbacks: Sequence[LanguageModelLike] = []
         for fallback in self.fallbacks:
@@ -1001,7 +1006,15 @@ class ChatHistoryModel(BaseModel):
     max_tokens: int = 256
     max_tokens_before_summary: Optional[int] = None
     max_messages_before_summary: Optional[int] = None
-    max_summary_tokens: Optional[int] = None
+    max_summary_tokens: int = 255
+
+    @model_validator(mode="after")
+    def validate_max_summary_tokens(self) -> "ChatHistoryModel":
+        if self.max_summary_tokens >= self.max_tokens:
+            raise ValueError(
+                f"max_summary_tokens ({self.max_summary_tokens}) must be less than max_tokens ({self.max_tokens})"
+            )
+        return self
 
 
 class AppModel(BaseModel):
@@ -1057,12 +1070,19 @@ class AppModel(BaseModel):
         return self
 
 
+class GuidelineModel(BaseModel):
+    model_config = ConfigDict(use_enum_values=True, extra="forbid")
+    name: str
+    guidelines: list[str]
+
+
 class EvaluationModel(BaseModel):
     model_config = ConfigDict(use_enum_values=True, extra="forbid")
     model: LLMModel
     table: TableModel
     num_evals: int
     custom_inputs: dict[str, Any] = Field(default_factory=dict)
+    guidelines: list[GuidelineModel] = Field(default_factory=list)
 
 
 class DatasetFormat(str, Enum):
@@ -1302,3 +1322,16 @@ class AppConfig(BaseModel):
         return [
             guardrail for guardrail in self.guardrails.values() if predicate(guardrail)
         ]
+
+    def as_graph(self) -> CompiledStateGraph:
+        from dao_ai.graph import create_dao_ai_graph
+
+        graph: CompiledStateGraph = create_dao_ai_graph(config=self)
+        return graph
+
+    def as_chat_model(self) -> ChatModel:
+        from dao_ai.models import create_agent
+
+        graph: CompiledStateGraph = self.as_graph()
+        app: ChatModel = create_agent(graph)
+        return app