dao-ai 0.0.12__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

@@ -351,12 +351,85 @@ class IndexModel(BaseModel, HasFullName, IsDatabricksResource):
         )
 
 
+class GenieRoomModel(BaseModel, IsDatabricksResource):
+    model_config = ConfigDict(use_enum_values=True, extra="forbid")
+    name: str
+    description: Optional[str] = None
+    space_id: str
+
+    @property
+    def api_scopes(self) -> Sequence[str]:
+        return [
+            "dashboards.genie",
+        ]
+
+    def as_resource(self) -> DatabricksResource:
+        return DatabricksGenieSpace(
+            genie_space_id=self.space_id, on_behalf_of_user=self.on_behalf_of_user
+        )
+
+
+class VolumeModel(BaseModel, HasFullName):
+    model_config = ConfigDict(use_enum_values=True, extra="forbid")
+    schema_model: Optional[SchemaModel] = Field(default=None, alias="schema")
+    name: str
+
+    @property
+    def full_name(self) -> str:
+        if self.schema_model:
+            return f"{self.schema_model.catalog_name}.{self.schema_model.schema_name}.{self.name}"
+        return self.name
+
+    def create(self, w: WorkspaceClient | None = None) -> None:
+        from dao_ai.providers.base import ServiceProvider
+        from dao_ai.providers.databricks import DatabricksProvider
+
+        provider: ServiceProvider = DatabricksProvider(w=w)
+        provider.create_volume(self)
+
+
+class VolumePathModel(BaseModel, HasFullName):
+    model_config = ConfigDict(use_enum_values=True, extra="forbid")
+    volume: Optional[VolumeModel] = None
+    path: Optional[str] = None
+
+    @model_validator(mode="after")
+    def validate_path_or_volume(self):
+        if not self.volume and not self.path:
+            raise ValueError("Either 'volume' or 'path' must be provided")
+        return self
+
+    @property
+    def full_name(self) -> str:
+        if self.volume and self.volume.schema_model:
+            catalog_name: str = self.volume.schema_model.catalog_name
+            schema_name: str = self.volume.schema_model.schema_name
+            volume_name: str = self.volume.name
+            path = f"/{self.path}" if self.path else ""
+            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
+
+        if self.volume:
+            self.volume.create(w=w)
+
+        provider: DatabricksProvider = DatabricksProvider(w=w)
+        provider.create_path(self)
+
+
 class VectorStoreModel(BaseModel, IsDatabricksResource):
     model_config = ConfigDict(use_enum_values=True, extra="forbid")
     embedding_model: Optional[LLMModel] = None
     index: Optional[IndexModel] = None
     endpoint: Optional[VectorSearchEndpoint] = None
     source_table: TableModel
+    source_path: Optional[VolumePathModel] = None
+    checkpoint_path: Optional[VolumePathModel] = None
     primary_key: Optional[str] = None
     columns: Optional[list[str]] = Field(default_factory=list)
     doc_uri: Optional[str] = None
@@ -448,43 +521,6 @@ class VectorStoreModel(BaseModel, IsDatabricksResource):
         provider.create_vector_store(self)
 
 
-class GenieRoomModel(BaseModel, IsDatabricksResource):
-    model_config = ConfigDict(use_enum_values=True, extra="forbid")
-    name: str
-    description: Optional[str] = None
-    space_id: str
-
-    @property
-    def api_scopes(self) -> Sequence[str]:
-        return [
-            "dashboards.genie",
-        ]
-
-    def as_resource(self) -> DatabricksResource:
-        return DatabricksGenieSpace(
-            genie_space_id=self.space_id, on_behalf_of_user=self.on_behalf_of_user
-        )
-
-
-class VolumeModel(BaseModel, HasFullName):
-    model_config = ConfigDict(use_enum_values=True, extra="forbid")
-    schema_model: Optional[SchemaModel] = Field(default=None, alias="schema")
-    name: str
-
-    @property
-    def full_name(self) -> str:
-        if self.schema_model:
-            return f"{self.schema_model.catalog_name}.{self.schema_model.schema_name}.{self.name}"
-        return self.name
-
-    def create(self, w: WorkspaceClient | None = None) -> None:
-        from dao_ai.providers.base import ServiceProvider
-        from dao_ai.providers.databricks import DatabricksProvider
-
-        provider: ServiceProvider = DatabricksProvider(w=w)
-        provider.create_volume(self)
-
-
 class FunctionModel(BaseModel, HasFullName, IsDatabricksResource):
     model_config = ConfigDict()
     schema_model: Optional[SchemaModel] = Field(default=None, alias="schema")
@@ -1093,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)
 
@@ -1107,21 +1145,6 @@ class DatasetFormat(str, Enum):
     EXCEL = "excel"
 
 
-class VolumePathModel(BaseModel, HasFullName):
-    model_config = ConfigDict(use_enum_values=True, extra="forbid")
-    volume: VolumeModel
-    path: str
-
-    @property
-    def full_name(self) -> str:
-        if self.volume.schema_model:
-            catalog_name: str = self.volume.schema_model.catalog_name
-            schema_name: str = self.volume.schema_model.schema_name
-            volume_name: str = self.volume.name
-            return f"/Volumes/{catalog_name}/{schema_name}/{volume_name}/{self.path}"
-        return f"/Volumes/{volume_name}/{self.path}"
-
-
 class DatasetModel(BaseModel):
     model_config = ConfigDict(use_enum_values=True, extra="forbid")
     table: Optional[TableModel] = None
@@ -1211,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/providers/databricks.py

@@ -54,6 +54,7 @@ from dao_ai.config import (
     UnityCatalogFunctionSqlModel,
     VectorStoreModel,
     VolumeModel,
+    VolumePathModel,
     WarehouseModel,
 )
 from dao_ai.models import get_latest_model_version
@@ -441,6 +442,12 @@ class DatabricksProvider(ServiceProvider):
             )
         return volume_info
 
+    def create_path(self, volume_path: VolumePathModel) -> Path:
+        path: Path = volume_path.full_name
+        logger.info(f"Creating volume path: {path}")
+        self.w.files.create_directory(path)
+        return path
+
     def create_dataset(self, dataset: DatasetModel) -> None:
         current_dir: Path = "file:///" / Path.cwd().relative_to("/")
 

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dao-ai
-Version: 0.0.12
+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.12.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=vT3CYFmTDzxc7zSLIat4B5vt-Q4Su-tl3BvFVbkXQIo,44197
+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
@@ -22,7 +22,7 @@ dao_ai/memory/core.py,sha256=K45iCEFbqJCVxMi4m3vmBJi4c6TQ-UtKGzyugDTkPP0,4141
 dao_ai/memory/postgres.py,sha256=YILzA7xtqawPAOLFaGG_i17zW7cQxXTzTD8yd-ipe8k,12480
 dao_ai/providers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dao_ai/providers/base.py,sha256=-fjKypCOk28h6vioPfMj9YZSw_3Kcbi2nMuAyY7vX9k,1383
-dao_ai/providers/databricks.py,sha256=vV11SptrzIBEQQyiUqrysMfdpKKPrhVPvUVeEOQxx0o,27684
+dao_ai/providers/databricks.py,sha256=XRPOqwF5SeA9rPAOWMg2gSMC7lw31BI5VI_4K0KIOqo,27931
 dao_ai/tools/__init__.py,sha256=ye6MHaJY7tUnJ8336YJiLxuZr55zDPNdOw6gm7j5jlc,1103
 dao_ai/tools/agent.py,sha256=_XMz6HtrybpVthhRyStADechF6vXLFyK97i01XTBhtw,1868
 dao_ai/tools/core.py,sha256=Kei33S8vrmvPOAyrFNekaWmV2jqZ-IPS1QDSvU7RZF0,1984
@@ -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.12.dist-info/METADATA,sha256=20Ysj-78vWn6m64Bw0_OWjjGXNwiFDSV8ORbpcwLr_8,41338
-dao_ai-0.0.12.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-dao_ai-0.0.12.dist-info/entry_points.txt,sha256=Xa-UFyc6gWGwMqMJOt06ZOog2vAfygV_DSwg1AiP46g,43
-dao_ai-0.0.12.dist-info/licenses/LICENSE,sha256=YZt3W32LtPYruuvHE9lGk2bw6ZPMMJD8yLrjgHybyz4,1069
-dao_ai-0.0.12.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,,