dao-ai 0.0.13__py3-none-any.whl → 0.0.14__py3-none-any.whl

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/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/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,

{dao_ai-0.0.13.dist-info → dao_ai-0.0.14.dist-info}/METADATA

@@ -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.dist-info → dao_ai-0.0.14.dist-info}/RECORD

@@ -1,9 +1,9 @@
 dao_ai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-dao_ai/agent_as_code.py,sha256=WjBUJzHMAEX7joMaQGgBgnofiKxjJjdQEVsBJ4VSkPA,655
+dao_ai/agent_as_code.py,sha256=rMWMC0nabtABHPD5H-Yy9ej7QNebLnXfvlZPiwrthoE,457
 dao_ai/catalog.py,sha256=sPZpHTD3lPx4EZUtIWeQV7VQM89WJ6YH__wluk1v2lE,4947
 dao_ai/chat_models.py,sha256=uhwwOTeLyHWqoTTgHrs4n5iSyTwe4EQcLKnh3jRxPWI,8626
 dao_ai/cli.py,sha256=Aez2TQW3Q8Ho1IaIkRggt0NevDxAAVPjXkePC5GPJF0,20429
-dao_ai/config.py,sha256=UrETOJK_sTJAaKD91Pkc_alG9wfzfWbMeHxtxdnUZhQ,44881
+dao_ai/config.py,sha256=VyHqkW1UMQJ0fzyme1fV_3fi_6wDmKRQeCrx881fDQ4,45173
 dao_ai/graph.py,sha256=rIm6cLsWwViB3L1dIZp9qc-U-JgFNB5ngEi22Y3iVGQ,7806
 dao_ai/guardrails.py,sha256=-Qh0f_2Db9t4Nbrrx9FM7tnpqShjMoyxepZ0HByItfU,4027
 dao_ai/messages.py,sha256=tRZQTeb5YFKu8cm1xeaCkKhidq-0tdzncNEzVePvits,6806
@@ -32,9 +32,9 @@ dao_ai/tools/mcp.py,sha256=rUeodCfGFXSvW-byTW9gsO6QI-p5SZkAP8ZFCFzREY0,4400
 dao_ai/tools/python.py,sha256=XcQiTMshZyLUTVR5peB3vqsoUoAAy8gol9_pcrhddfI,1831
 dao_ai/tools/time.py,sha256=Y-23qdnNHzwjvnfkWvYsE7PoWS1hfeKy44tA7sCnNac,8759
 dao_ai/tools/unity_catalog.py,sha256=PXfLj2EgyQgaXq4Qq3t25AmTC4KyVCF_-sCtg6enens,1404
-dao_ai/tools/vector_search.py,sha256=kLveW-JGwc4IbJ7fKclJFmDXhP3h5XBrjljgab2tRD4,2559
-dao_ai-0.0.13.dist-info/METADATA,sha256=rSKZRiJlzOUFYJXyo-ofNYVWWXjr_-dnrmbX34fb_8s,41338
-dao_ai-0.0.13.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-dao_ai-0.0.13.dist-info/entry_points.txt,sha256=Xa-UFyc6gWGwMqMJOt06ZOog2vAfygV_DSwg1AiP46g,43
-dao_ai-0.0.13.dist-info/licenses/LICENSE,sha256=YZt3W32LtPYruuvHE9lGk2bw6ZPMMJD8yLrjgHybyz4,1069
-dao_ai-0.0.13.dist-info/RECORD,,
+dao_ai/tools/vector_search.py,sha256=EDYQs51zIPaAP0ma1D81wJT77GQ-v-cjb2XrFVWfWdg,2621
+dao_ai-0.0.14.dist-info/METADATA,sha256=jQdUAv02pr4wXZdMUJ9L_TksHGTCu4X_vrBKwvzIp4A,41338
+dao_ai-0.0.14.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+dao_ai-0.0.14.dist-info/entry_points.txt,sha256=Xa-UFyc6gWGwMqMJOt06ZOog2vAfygV_DSwg1AiP46g,43
+dao_ai-0.0.14.dist-info/licenses/LICENSE,sha256=YZt3W32LtPYruuvHE9lGk2bw6ZPMMJD8yLrjgHybyz4,1069
+dao_ai-0.0.14.dist-info/RECORD,,