dao-ai 0.0.13__tar.gz → 0.0.14__tar.gz

{dao_ai-0.0.13 → dao_ai-0.0.14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dao-ai
-Version: 0.0.13
+Version: 0.0.14
 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
@@ -38,7 +38,7 @@ 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: mlflow>=3.3.2
 Requires-Dist: nest-asyncio>=1.6.0
 Requires-Dist: openevals>=0.0.19
 Requires-Dist: openpyxl>=3.1.5

{dao_ai-0.0.13 → dao_ai-0.0.14}/notebooks/01_ingest_and_transform.py

@@ -1,5 +1,5 @@
 # Databricks notebook source
-# MAGIC %pip install --quiet -r ../requirements.txt
+# MAGIC %pip install --quiet --upgrade -r ../requirements.txt
 # MAGIC %restart_python
 
 # COMMAND ----------

{dao_ai-0.0.13 → dao_ai-0.0.14}/notebooks/02_provision_vector_search.py

@@ -1,7 +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 %pip install --quiet --upgrade -r ../requirements.txt
+# MAGIC %pip uninstall --quiet -y databricks-connect pyspark pyspark-connect
+# MAGIC %pip install --quiet databricks-connect
 # MAGIC %restart_python
 
 # COMMAND ----------

{dao_ai-0.0.13 → dao_ai-0.0.14}/notebooks/03_generate_evaluation_data.py

@@ -1,7 +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 %pip install --quiet --upgrade -r ../requirements.txt
+# MAGIC %pip uninstall --quiet -y databricks-connect pyspark pyspark-connect
+# MAGIC %pip install --quiet databricks-connect
 # MAGIC %restart_python
 
 # COMMAND ----------
@@ -65,26 +65,43 @@ for _, vector_store in config.resources.vector_stores.items():
     spark.table(vector_store.source_table.full_name)
     .withColumn("id", F.col(vector_store.primary_key))
     .withColumn("content", F.col(vector_store.embedding_source_column))
-    .withColumn("doc_uri", F.lit("source"))
+    .withColumn("doc_uri", doc_uri)
   )
   parsed_docs_pdf: pd.DataFrame = parsed_docs_df.toPandas()
 
   display(parsed_docs_pdf)
 
-  agent_description = f"""
-  The agent is a RAG chatbot that answers questions about retail hardware and gives recommendations for purchases. 
-  """
-  question_guidelines = f"""
-  # User personas
-  - An employee or client asking about products and inventory
-
-
-  # Example questions
-  - What grills do you have in stock?
-  - Can you recommend a accessories for my Toro lawn mower?
-
-  # Additional Guidelines
-  - Questions should be succinct, and human-like
+  agent_description: str = evaluation.agent_description
+  if not agent_description:
+      agent_description = """
+  A general-purpose chatbot AI agent is designed to engage in natural conversations 
+  across diverse topics and tasks, drawing from broad knowledge to answer questions, 
+  assist with writing, solve problems, and provide explanations while maintaining 
+  context throughout interactions. It aims to be a versatile, adaptable assistant 
+  that can help with the wide spectrum of things people encounter in daily life, 
+  adjusting its communication style and level of detail based on user needs.
+      """
+
+  question_guidelines: str = evaluation.question_guidelines
+  if not question_guidelines:
+      question_guidelines = f"""
+# User personas
+- A curious individual seeking information or explanations
+- A student looking for homework help or learning assistance  
+- A professional needing quick research or writing support
+- A creative person brainstorming ideas or seeking inspiration
+
+# Example questions
+- Can you explain how photosynthesis works?
+- Help me write a professional email to my boss
+- What are some good books similar to Harry Potter?
+- How do I fix a leaky faucet?
+
+# Additional Guidelines  
+- Questions should be conversational and natural
+- Users may ask follow-up questions to dig deeper into topics
+- Requests can range from simple facts to complex multi-step tasks
+- Tone can vary from casual chat to formal assistance
   """
 
   evals_pdf: pd.DataFrame = generate_evals_df(

{dao_ai-0.0.13 → dao_ai-0.0.14}/notebooks/04_unity_catalog_tools.py

@@ -1,7 +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 %pip install --quiet --upgrade -r ../requirements.txt
+# MAGIC %pip uninstall --quiet -y databricks-connect pyspark pyspark-connect
+# MAGIC %pip install --quiet databricks-connect
 # MAGIC %restart_python
 
 # COMMAND ----------

{dao_ai-0.0.13 → dao_ai-0.0.14}/notebooks/05_agent_as_code_driver.py

@@ -1,7 +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 %pip install --quiet --upgrade -r ../requirements.txt
+# MAGIC %pip uninstall --quiet -y databricks-connect pyspark pyspark-connect
+# MAGIC %pip install --quiet databricks-connect
 # MAGIC %restart_python
 
 # COMMAND ----------

{dao_ai-0.0.13 → dao_ai-0.0.14}/notebooks/06_run_evaluation.py

@@ -1,8 +1,8 @@
 # Databricks notebook source
 # DBTITLE 1,Install Required Python Packages and Restart Kernel
-# MAGIC %pip install --quiet -r ../requirements.txt
-# MAGIC %pip uninstall -y databricks-connect pyspark pyspark-connect
-# MAGIC %pip install databricks-connect
+# MAGIC %pip install --quiet --upgrade -r ../requirements.txt
+# MAGIC %pip uninstall --quiet -y databricks-connect pyspark pyspark-connect
+# MAGIC %pip install --quiet databricks-connect
 # MAGIC %restart_python
 
 # COMMAND ----------
@@ -127,11 +127,6 @@ from mlflow.genai.scorers import scorer, Safety, Guidelines
 from mlflow.entities import Feedback, Trace
 
 
-clarity = Guidelines(
-    name="clarity",
-    guidelines=["The response must be clear, coherent, and concise"],
-)
-
 @scorer
 def response_completeness(outputs: dict[str, Any]) -> Feedback:
 
@@ -225,7 +220,7 @@ eval_df: pd.DataFrame = spark.read.table(payload_table).toPandas()
 
 evaluation_table_name: str = config.evaluation.table.full_name
 
-scorers = [Safety(), clarity, response_completeness, tool_call_efficiency]
+scorers = [Safety(), response_completeness, tool_call_efficiency]
 custom_scorers = []
 
 if config.evaluation.guidelines:

{dao_ai-0.0.13 → dao_ai-0.0.14}/notebooks/08_run_examples.py

@@ -1,7 +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 %pip install --quiet --upgrade -r ../requirements.txt
+# MAGIC %pip uninstall --quiet -y databricks-connect pyspark pyspark-connect
+# MAGIC %pip install --quiet databricks-connect
 # MAGIC %restart_python
 
 # COMMAND ----------
@@ -50,15 +50,7 @@ mlflow.langchain.autolog()
 
 config: AppConfig = AppConfig.from_file(path=config_path)
 
-log_level: str = config.app.log_level
-
-logger.remove()
-logger.add(sys.stderr, level=log_level)
-
-graph: CompiledStateGraph = create_dao_ai_graph(config=config)
-
-app: ChatModel = create_agent(graph)
-
+app: ChatModel = config.as_chat_model()
 
 # COMMAND ----------
 

{dao_ai-0.0.13 → dao_ai-0.0.14}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "dao-ai"
-version = "0.0.13"
+version = "0.0.14"
 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" }
@@ -56,7 +56,7 @@ dependencies = [
     "langmem>=0.0.29",
     "loguru>=0.7.3",
     "mcp>=1.9.1",
-    "mlflow>=3.1.1",
+    "mlflow>=3.3.2",
     "nest-asyncio>=1.6.0",
     "openevals>=0.0.19",
     "openpyxl>=3.1.5",

{dao_ai-0.0.13 → dao_ai-0.0.14}/requirements.txt

@@ -12,11 +12,13 @@ boto3==1.40.1
 botocore==1.40.1
 cachetools==5.5.2
 certifi==2025.6.15
+cffi==1.17.1
 cfgv==3.4.0
 charset-normalizer==3.4.2
 click==8.2.1
 cloudpickle==3.1.1
 contourpy==1.3.2
+cryptography==45.0.7
 cycler==0.12.1
 databricks-agents==1.2.0
 databricks-ai-bridge==0.5.1
@@ -88,8 +90,9 @@ marshmallow==3.26.1
 matplotlib==3.10.3
 mcp==1.9.4
 mdurl==0.1.2
-mlflow==3.1.1
-mlflow-skinny==3.1.1
+mlflow==3.3.2
+mlflow-skinny==3.3.2
+mlflow-tracing==3.3.2
 multidict==6.5.0
 mypy==1.16.1
 mypy-extensions==1.1.0
@@ -121,6 +124,7 @@ py4j==0.10.9.7
 pyarrow==20.0.0
 pyasn1==0.6.1
 pyasn1-modules==0.4.2
+pycparser==2.22
 pydantic==2.11.7
 pydantic-core==2.33.2
 pydantic-settings==2.10.0

{dao_ai-0.0.13 → dao_ai-0.0.14}/schemas/model_config_schema.json

@@ -1086,6 +1086,30 @@
           "title": "Num Evals",
           "type": "integer"
         },
+        "agent_description": {
+          "anyOf": [
+            {
+              "type": "string"
+            },
+            {
+              "type": "null"
+            }
+          ],
+          "default": null,
+          "title": "Agent Description"
+        },
+        "question_guidelines": {
+          "anyOf": [
+            {
+              "type": "string"
+            },
+            {
+              "type": "null"
+            }
+          ],
+          "default": null,
+          "title": "Question Guidelines"
+        },
         "custom_inputs": {
           "additionalProperties": true,
           "title": "Custom Inputs",

{dao_ai-0.0.13 → dao_ai-0.0.14}/src/dao_ai/agent_as_code.py

@@ -1,14 +1,11 @@
 import sys
 
 import mlflow
-from langgraph.graph.state import CompiledStateGraph
 from loguru import logger
 from mlflow.models import ModelConfig
 from mlflow.pyfunc import ChatModel
 
 from dao_ai.config import AppConfig
-from dao_ai.graph import create_dao_ai_graph
-from dao_ai.models import create_agent
 
 mlflow.langchain.autolog()
 
@@ -20,8 +17,6 @@ log_level: str = config.app.log_level
 logger.remove()
 logger.add(sys.stderr, level=log_level)
 
-graph: CompiledStateGraph = create_dao_ai_graph(config=config)
-
-app: ChatModel = create_agent(graph)
+app: ChatModel = config.as_chat_model()
 
 mlflow.models.set_model(app)

{dao_ai-0.0.13 → dao_ai-0.0.14}/src/dao_ai/config.py

@@ -409,6 +409,9 @@ class VolumePathModel(BaseModel, HasFullName):
             return f"/Volumes/{catalog_name}/{schema_name}/{volume_name}{path}"
         return self.path
 
+    def as_path(self) -> Path:
+        return Path(self.full_name)
+
     def create(self, w: WorkspaceClient | None = None) -> None:
         from dao_ai.providers.databricks import DatabricksProvider
 
@@ -1126,6 +1129,8 @@ class EvaluationModel(BaseModel):
     model: LLMModel
     table: TableModel
     num_evals: int
+    agent_description: Optional[str] = None
+    question_guidelines: Optional[str] = None
     custom_inputs: dict[str, Any] = Field(default_factory=dict)
     guidelines: list[GuidelineModel] = Field(default_factory=list)
 
@@ -1229,6 +1234,10 @@ class AppConfig(BaseModel):
     def initialize(self) -> None:
         from dao_ai.hooks.core import create_hooks
 
+        if self.app and self.app.log_level:
+            logger.remove()
+            logger.add(sys.stderr, level=self.app.log_level)
+
         logger.debug("Calling initialization hooks...")
         initialization_functions: Sequence[Callable[..., Any]] = create_hooks(
             self.app.initialization_hooks

{dao_ai-0.0.13 → dao_ai-0.0.14}/src/dao_ai/tools/vector_search.py

@@ -51,7 +51,9 @@ def create_vector_search_tool(
 
     vector_search_tool: BaseTool = VectorSearchRetrieverTool(
         name=name,
+        tool_name=name,
         description=description,
+        tool_description=description,
         index_name=index_name,
         columns=columns,
         **search_parameters,