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

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

@@ -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 → dao_ai-0.0.13}/config/examples/genie.yaml

@@ -5,11 +5,10 @@ schemas:
   retail_schema: &retail_schema
     catalog_name: nfleming                    # Unity Catalog name
     schema_name: retail_ai                    # 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
 
+  quick_serve_retaurant_schema: &quick_serve_retaurant_schema
+    catalog_name: retail_consumer_goods                    # Unity Catalog name
+    schema_name: quick_serve_restaurant                    # Schema within the catalog
 
 resources:
   llms:
@@ -20,6 +19,40 @@ resources:
       max_tokens: 8192                              # Maximum tokens per response
       on_behalf_of_user: False
 
+  tables:
+    product_table: &product_table
+      schema: *quick_serve_retaurant_schema                        # Reference to schema defined above
+      name: products                                # Table name
+    
+    inventory_table: &inventory_table
+      schema: *quick_serve_retaurant_schema
+      name: inventory
+
+    items_description: &items_description
+      schema: *quick_serve_retaurant_schema
+      name: items_description
+
+    fulfil_item_orders: &fulfil_item_orders
+      schema: *quick_serve_retaurant_schema
+      name: fulfil_item_orders
+
+    orders: &orders
+      schema: *quick_serve_retaurant_schema
+      name: orders
+
+  functions:
+
+    lookup_items_by_descriptions: &lookup_items_by_descriptions
+      schema: *quick_serve_retaurant_schema
+      name: lookup_items_by_descriptions
+
+    match_item_by_description_and_price: &match_item_by_description_and_price
+      schema: *quick_serve_retaurant_schema
+      name: match_item_by_description_and_price     
+    
+    match_historical_item_order_by_date: &match_historical_item_order_by_date
+      schema: *quick_serve_retaurant_schema
+      name: match_historical_item_order_by_date     
 
   genie_rooms:
     # Genie space for retail data queries
@@ -37,7 +70,7 @@ tools:
       name: dao_ai.tools.create_genie_tool       # Factory function path
       args:                                         # Arguments passed to factory
         name: my_genie_tool
-        description: My Answers questions about genie
+        description: Answers questions about retail products and inventory
         genie_room: *retail_genie_room              # Reference to Genie room config
 
 
@@ -49,9 +82,8 @@ agents:
     tools:                                          # Tools available to this agent
       - *genie_tool
     prompt: |                                       # System prompt defining agent behavior
-      Answers foobar questions about genie
-    handoff_prompt: |                               # Conditions for routing to this agent
-      Answers all foobar questions
+      Answers questions about retail products and inventory
+
 
 app:
   name: genie_example                      # Application name  

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

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "dao-ai"
-version = "0.0.12"
+version = "0.0.13"
 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" }

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

@@ -2547,6 +2547,28 @@
         "source_table": {
           "$ref": "#/$defs/TableModel"
         },
+        "source_path": {
+          "anyOf": [
+            {
+              "$ref": "#/$defs/VolumePathModel"
+            },
+            {
+              "type": "null"
+            }
+          ],
+          "default": null
+        },
+        "checkpoint_path": {
+          "anyOf": [
+            {
+              "$ref": "#/$defs/VolumePathModel"
+            },
+            {
+              "type": "null"
+            }
+          ],
+          "default": null
+        },
         "primary_key": {
           "anyOf": [
             {
@@ -2626,17 +2648,29 @@
       "additionalProperties": false,
       "properties": {
         "volume": {
-          "$ref": "#/$defs/VolumeModel"
+          "anyOf": [
+            {
+              "$ref": "#/$defs/VolumeModel"
+            },
+            {
+              "type": "null"
+            }
+          ],
+          "default": null
         },
         "path": {
-          "title": "Path",
-          "type": "string"
+          "anyOf": [
+            {
+              "type": "string"
+            },
+            {
+              "type": "null"
+            }
+          ],
+          "default": null,
+          "title": "Path"
         }
       },
-      "required": [
-        "volume",
-        "path"
-      ],
       "title": "VolumePathModel",
       "type": "object"
     },

{dao_ai-0.0.12 → dao_ai-0.0.13}/src/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-0.0.12 → dao_ai-0.0.13}/src/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("/")