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

dao_ai/config.py

@@ -351,12 +351,82 @@ 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 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 +518,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")
@@ -1107,21 +1140,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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dao-ai
-Version: 0.0.12
+Version: 0.0.13
 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.12.dist-info → dao_ai-0.0.13.dist-info}/RECORD

@@ -3,7 +3,7 @@ dao_ai/agent_as_code.py,sha256=WjBUJzHMAEX7joMaQGgBgnofiKxjJjdQEVsBJ4VSkPA,655
 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=UrETOJK_sTJAaKD91Pkc_alG9wfzfWbMeHxtxdnUZhQ,44881
 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
@@ -33,8 +33,8 @@ 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-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,,