dao-ai 0.0.17__tar.gz → 0.0.18__tar.gz

{dao_ai-0.0.17 → dao_ai-0.0.18}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dao-ai
-Version: 0.0.17
+Version: 0.0.18
 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.17 → dao_ai-0.0.18}/config/examples/executive_assistant.yaml

@@ -50,68 +50,68 @@ resources:
 
 
   tables:
-    employee_performance:
+    all_tables:
       schema: *executive_assistant_schema                        
-      name: employee_performance                              
+    #   name: employee_performance                              
     
-    products:
-      schema: *executive_assistant_schema
-      name: products
+    # products:
+    #   schema: *executive_assistant_schema
+    #   name: products
 
-    inventory: 
-      schema: *executive_assistant_schema
-      name: inventory
+    # inventory: 
+    #   schema: *executive_assistant_schema
+    #   name: inventory
 
-    customers: 
-      schema: *executive_assistant_schema
-      name: customers
+    # customers: 
+    #   schema: *executive_assistant_schema
+    #   name: customers
 
-    managers:
-      schema: *executive_assistant_schema
-      name: managers
+    # managers:
+    #   schema: *executive_assistant_schema
+    #   name: managers
 
-    employee_tasks: 
-      schema: *executive_assistant_schema
-      name: employee_tasks
+    # employee_tasks: 
+    #   schema: *executive_assistant_schema
+    #   name: employee_tasks
 
-    appointments: 
-      schema: *executive_assistant_schema
-      name: appointments
+    # appointments: 
+    #   schema: *executive_assistant_schema
+    #   name: appointments
 
-    evaluation: 
-      schema: *executive_assistant_schema
-      name: evaluation
+    # evaluation: 
+    #   schema: *executive_assistant_schema
+    #   name: evaluation
 
 
   functions:
 
-    find_inventory_by_sku: &find_inventory_by_sku
+    all_functions: 
       schema: *executive_assistant_schema
-      name: find_inventory_by_sku
+    #   name: find_inventory_by_sku
 
-    find_inventory_by_upc: &find_inventory_by_upc
-      schema: *executive_assistant_schema
-      name: find_inventory_by_upc     
+    # find_inventory_by_upc: &find_inventory_by_upc
+    #   schema: *executive_assistant_schema
+    #   name: find_inventory_by_upc     
     
-    find_product_by_sku: &find_product_by_sku
-      schema: *executive_assistant_schema
-      name: find_product_by_sku     
+    # find_product_by_sku: &find_product_by_sku
+    #   schema: *executive_assistant_schema
+    #   name: find_product_by_sku     
 
-    find_product_by_upc: &find_product_by_upc
-      schema: *executive_assistant_schema
-      name: find_product_by_upc     
+    # find_product_by_upc: &find_product_by_upc
+    #   schema: *executive_assistant_schema
+    #   name: find_product_by_upc     
 
-    find_store_by_number: &find_store_by_number
-      schema: *executive_assistant_schema
-      name: find_store_by_number     
+    # find_store_by_number: &find_store_by_number
+    #   schema: *executive_assistant_schema
+    #   name: find_store_by_number     
     
-    find_store_inventory_by_sku: &find_store_inventory_by_sku
-      schema: *executive_assistant_schema
-      name: find_store_inventory_by_sku     
+    # find_store_inventory_by_sku: &find_store_inventory_by_sku
+    #   schema: *executive_assistant_schema
+    #   name: find_store_inventory_by_sku     
 
-    find_store_inventory_by_upc: &find_store_inventory_by_upc
-      schema: *executive_assistant_schema
-      name: find_store_inventory_by_upc
+    # find_store_inventory_by_upc: &find_store_inventory_by_upc
+    #   schema: *executive_assistant_schema
+    #   name: find_store_inventory_by_upc
 
   genie_rooms:
     executive_assistant_genie_room: &executive_assistant_genie_room

{dao_ai-0.0.17 → dao_ai-0.0.18}/config/hardware_store/supervisor.yaml

@@ -898,7 +898,7 @@ agents:
 app:
   name: hardware_store_supervisor                      # Application name  
   description: "Multi-agent system for retail customer service and product assistance"
-  log_level: INFO                                   # Logging level for the application
+  log_level: DEBUG                                   # Logging level for the application
   registered_model:                                 # MLflow registered model configuration
     schema: *retail_schema                          # Schema where model will be registered
     name: hardware_store_supervisor                           # Model name in MLflow registry
@@ -926,7 +926,7 @@ app:
     - dao_ai.hooks.null_hook
     - hardware_store.hooks.require_store_num_hook
   input_example:
-      messages:
+      input:
         - role: user
           content: Can you recommend a lamp to match my oak side tables?
       custom_inputs:

{dao_ai-0.0.17 → dao_ai-0.0.18}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "dao-ai"
-version = "0.0.17"
+version = "0.0.18"
 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.17 → dao_ai-0.0.18}/schemas/model_config_schema.json

@@ -480,12 +480,35 @@
     "ChatPayload": {
       "additionalProperties": false,
       "properties": {
+        "input": {
+          "anyOf": [
+            {
+              "items": {
+                "$ref": "#/$defs/Message"
+              },
+              "type": "array"
+            },
+            {
+              "type": "null"
+            }
+          ],
+          "default": null,
+          "title": "Input"
+        },
         "messages": {
-          "items": {
-            "$ref": "#/$defs/Message"
-          },
-          "title": "Messages",
-          "type": "array"
+          "anyOf": [
+            {
+              "items": {
+                "$ref": "#/$defs/Message"
+              },
+              "type": "array"
+            },
+            {
+              "type": "null"
+            }
+          ],
+          "default": null,
+          "title": "Messages"
         },
         "custom_inputs": {
           "additionalProperties": true,
@@ -494,7 +517,6 @@
         }
       },
       "required": [
-        "messages",
         "custom_inputs"
       ],
       "title": "ChatPayload",
@@ -1200,13 +1222,18 @@
           "default": null
         },
         "name": {
-          "title": "Name",
-          "type": "string"
+          "anyOf": [
+            {
+              "type": "string"
+            },
+            {
+              "type": "null"
+            }
+          ],
+          "default": null,
+          "title": "Name"
         }
       },
-      "required": [
-        "name"
-      ],
       "title": "FunctionModel",
       "type": "object"
     },
@@ -2339,13 +2366,18 @@
           "default": null
         },
         "name": {
-          "title": "Name",
-          "type": "string"
+          "anyOf": [
+            {
+              "type": "string"
+            },
+            {
+              "type": "null"
+            }
+          ],
+          "default": null,
+          "title": "Name"
         }
       },
-      "required": [
-        "name"
-      ],
       "title": "TableModel",
       "type": "object"
     },

{dao_ai-0.0.17 → dao_ai-0.0.18}/src/dao_ai/agent_as_code.py

@@ -3,7 +3,7 @@ import sys
 import mlflow
 from loguru import logger
 from mlflow.models import ModelConfig
-from mlflow.pyfunc import ChatModel
+from mlflow.pyfunc import ResponsesAgent
 
 from dao_ai.config import AppConfig
 
@@ -17,6 +17,6 @@ log_level: str = config.app.log_level
 logger.remove()
 logger.add(sys.stderr, level=log_level)
 
-app: ChatModel = config.as_chat_model()
+app: ResponsesAgent = config.as_responses_agent()
 
 mlflow.models.set_model(app)

{dao_ai-0.0.17 → dao_ai-0.0.18}/src/dao_ai/config.py

@@ -9,6 +9,7 @@ from pathlib import Path
 from typing import (
     Any,
     Callable,
+    Iterator,
     Literal,
     Optional,
     Sequence,
@@ -21,6 +22,7 @@ from databricks.sdk.credentials_provider import (
     CredentialsStrategy,
     ModelServingUserCredentials,
 )
+from databricks.sdk.service.catalog import FunctionInfo, TableInfo
 from databricks.vector_search.client import VectorSearchClient
 from databricks.vector_search.index import VectorSearchIndex
 from databricks_langchain import (
@@ -44,8 +46,14 @@ from mlflow.models.resources import (
     DatabricksUCConnection,
     DatabricksVectorSearchIndex,
 )
-from mlflow.pyfunc import ChatModel
-from pydantic import BaseModel, ConfigDict, Field, field_serializer, model_validator
+from mlflow.pyfunc import ChatModel, ResponsesAgent
+from pydantic import (
+    BaseModel,
+    ConfigDict,
+    Field,
+    field_serializer,
+    model_validator,
+)
 
 
 class HasValue(ABC):
@@ -69,7 +77,7 @@ class IsDatabricksResource(ABC):
     on_behalf_of_user: Optional[bool] = False
 
     @abstractmethod
-    def as_resource(self) -> DatabricksResource: ...
+    def as_resources(self) -> Sequence[DatabricksResource]: ...
 
     @property
     @abstractmethod
@@ -235,22 +243,68 @@ class SchemaModel(BaseModel, HasFullName):
 class TableModel(BaseModel, HasFullName, IsDatabricksResource):
     model_config = ConfigDict(use_enum_values=True, extra="forbid")
     schema_model: Optional[SchemaModel] = Field(default=None, alias="schema")
-    name: str
+    name: Optional[str] = None
+
+    @model_validator(mode="after")
+    def validate_name_or_schema_required(self) -> "TableModel":
+        if not self.name and not self.schema_model:
+            raise ValueError(
+                "Either 'name' or 'schema_model' must be provided for TableModel"
+            )
+        return self
 
     @property
     def full_name(self) -> str:
         if self.schema_model:
-            return f"{self.schema_model.catalog_name}.{self.schema_model.schema_name}.{self.name}"
+            name: str = ""
+            if self.name:
+                name = f".{self.name}"
+            return f"{self.schema_model.catalog_name}.{self.schema_model.schema_name}{name}"
         return self.name
 
     @property
     def api_scopes(self) -> Sequence[str]:
         return []
 
-    def as_resource(self) -> DatabricksResource:
-        return DatabricksTable(
-            table_name=self.full_name, on_behalf_of_user=self.on_behalf_of_user
-        )
+    def as_resources(self) -> Sequence[DatabricksResource]:
+        resources: list[DatabricksResource] = []
+
+        excluded_suffixes: Sequence[str] = [
+            "_payload",
+            "_assessment_logs",
+            "_request_logs",
+        ]
+        
+        excluded_prefixes: Sequence[str] = [
+            "trace_logs_"
+        ]
+
+        if self.name:
+            resources.append(
+                DatabricksTable(
+                    table_name=self.full_name, on_behalf_of_user=self.on_behalf_of_user
+                )
+            )
+        else:
+            w: WorkspaceClient = self.workspace_client
+            schema_full_name: str = self.schema_model.full_name
+            tables: Iterator[TableInfo] = w.tables.list(
+                catalog_name=self.schema_model.catalog_name,
+                schema_name=self.schema_model.schema_name,
+            )
+            resources.extend(
+                [
+                    DatabricksTable(
+                        table_name=f"{schema_full_name}.{table.name}",
+                        on_behalf_of_user=self.on_behalf_of_user,
+                    )
+                    for table in tables
+                    if not any(table.name.endswith(suffix) for suffix in excluded_suffixes)
+                    and not any(table.name.startswith(prefix) for prefix in excluded_prefixes)
+                ]
+            )
+
+        return resources
 
 
 class LLMModel(BaseModel, IsDatabricksResource):
@@ -266,10 +320,12 @@ class LLMModel(BaseModel, IsDatabricksResource):
             "serving.serving-endpoints",
         ]
 
-    def as_resource(self) -> DatabricksResource:
-        return DatabricksServingEndpoint(
-            endpoint_name=self.name, on_behalf_of_user=self.on_behalf_of_user
-        )
+    def as_resources(self) -> Sequence[DatabricksResource]:
+        return [
+            DatabricksServingEndpoint(
+                endpoint_name=self.name, on_behalf_of_user=self.on_behalf_of_user
+            )
+        ]
 
     def as_chat_model(self) -> LanguageModelLike:
         # Retrieve langchain chat client from workspace client to enable OBO
@@ -345,10 +401,12 @@ class IndexModel(BaseModel, HasFullName, IsDatabricksResource):
             return f"{self.schema_model.catalog_name}.{self.schema_model.schema_name}.{self.name}"
         return self.name
 
-    def as_resource(self) -> DatabricksResource:
-        return DatabricksVectorSearchIndex(
-            index_name=self.full_name, on_behalf_of_user=self.on_behalf_of_user
-        )
+    def as_resources(self) -> Sequence[DatabricksResource]:
+        return [
+            DatabricksVectorSearchIndex(
+                index_name=self.full_name, on_behalf_of_user=self.on_behalf_of_user
+            )
+        ]
 
 
 class GenieRoomModel(BaseModel, IsDatabricksResource):
@@ -363,10 +421,12 @@ class GenieRoomModel(BaseModel, IsDatabricksResource):
             "dashboards.genie",
         ]
 
-    def as_resource(self) -> DatabricksResource:
-        return DatabricksGenieSpace(
-            genie_space_id=self.space_id, on_behalf_of_user=self.on_behalf_of_user
-        )
+    def as_resources(self) -> Sequence[DatabricksResource]:
+        return [
+            DatabricksGenieSpace(
+                genie_space_id=self.space_id, on_behalf_of_user=self.on_behalf_of_user
+            )
+        ]
 
 
 class VolumeModel(BaseModel, HasFullName):
@@ -394,7 +454,7 @@ class VolumePathModel(BaseModel, HasFullName):
     path: Optional[str] = None
 
     @model_validator(mode="after")
-    def validate_path_or_volume(self):
+    def validate_path_or_volume(self) -> "VolumePathModel":
         if not self.volume and not self.path:
             raise ValueError("Either 'volume' or 'path' must be provided")
         return self
@@ -502,8 +562,8 @@ class VectorStoreModel(BaseModel, IsDatabricksResource):
             "serving.serving-endpoints",
         ] + self.index.api_scopes
 
-    def as_resource(self) -> DatabricksResource:
-        return self.index.as_resource()
+    def as_resources(self) -> Sequence[DatabricksResource]:
+        return self.index.as_resources()
 
     def as_index(self, vsc: VectorSearchClient | None = None) -> VectorSearchIndex:
         from dao_ai.providers.base import ServiceProvider
@@ -524,18 +584,52 @@ class VectorStoreModel(BaseModel, IsDatabricksResource):
 class FunctionModel(BaseModel, HasFullName, IsDatabricksResource):
     model_config = ConfigDict()
     schema_model: Optional[SchemaModel] = Field(default=None, alias="schema")
-    name: str
+    name: Optional[str] = None
+
+    @model_validator(mode="after")
+    def validate_name_or_schema_required(self) -> "FunctionModel":
+        if not self.name and not self.schema_model:
+            raise ValueError(
+                "Either 'name' or 'schema_model' must be provided for FunctionModel"
+            )
+        return self
 
     @property
     def full_name(self) -> str:
         if self.schema_model:
-            return f"{self.schema_model.catalog_name}.{self.schema_model.schema_name}.{self.name}"
+            name: str = ""
+            if self.name:
+                name = f".{self.name}"
+            return f"{self.schema_model.catalog_name}.{self.schema_model.schema_name}{name}"
         return self.name
 
-    def as_resource(self) -> DatabricksResource:
-        return DatabricksFunction(
-            function_name=self.full_name, on_behalf_of_user=self.on_behalf_of_user
-        )
+    def as_resources(self) -> Sequence[DatabricksResource]:
+        resources: list[DatabricksResource] = []
+        if self.name:
+            resources.append(
+                DatabricksFunction(
+                    function_name=self.full_name,
+                    on_behalf_of_user=self.on_behalf_of_user,
+                )
+            )
+        else:
+            w: WorkspaceClient = self.workspace_client
+            schema_full_name: str = self.schema_model.full_name
+            functions: Iterator[FunctionInfo] = w.functions.list(
+                catalog_name=self.schema_model.catalog_name,
+                schema_name=self.schema_model.schema_name,
+            )
+            resources.extend(
+                [
+                    DatabricksFunction(
+                        function_name=f"{schema_full_name}.{function.name}",
+                        on_behalf_of_user=self.on_behalf_of_user,
+                    )
+                    for function in functions
+                ]
+            )
+
+        return resources
 
     @property
     def api_scopes(self) -> Sequence[str]:
@@ -556,10 +650,12 @@ class ConnectionModel(BaseModel, HasFullName, IsDatabricksResource):
             "catalog.connections",
         ]
 
-    def as_resource(self) -> DatabricksResource:
-        return DatabricksUCConnection(
-            connection_name=self.name, on_behalf_of_user=self.on_behalf_of_user
-        )
+    def as_resources(self) -> Sequence[DatabricksResource]:
+        return [
+            DatabricksUCConnection(
+                connection_name=self.name, on_behalf_of_user=self.on_behalf_of_user
+            )
+        ]
 
 
 class WarehouseModel(BaseModel, IsDatabricksResource):
@@ -575,10 +671,12 @@ class WarehouseModel(BaseModel, IsDatabricksResource):
             "sql.statement-execution",
         ]
 
-    def as_resource(self) -> DatabricksResource:
-        return DatabricksSQLWarehouse(
-            warehouse_id=self.warehouse_id, on_behalf_of_user=self.on_behalf_of_user
-        )
+    def as_resources(self) -> Sequence[DatabricksResource]:
+        return [
+            DatabricksSQLWarehouse(
+                warehouse_id=self.warehouse_id, on_behalf_of_user=self.on_behalf_of_user
+            )
+        ]
 
 
 class DatabaseModel(BaseModel):
@@ -1034,9 +1132,37 @@ class Message(BaseModel):
 
 class ChatPayload(BaseModel):
     model_config = ConfigDict(use_enum_values=True, extra="forbid")
-    messages: list[Message]
+    input: Optional[list[Message]] = None
+    messages: Optional[list[Message]] = None
     custom_inputs: dict
 
+    @model_validator(mode="after")
+    def validate_mutual_exclusion_and_alias(self) -> "ChatPayload":
+        """Handle dual field support with automatic aliasing."""
+        # If both fields are provided and they're the same, that's okay (redundant but valid)
+        if self.input is not None and self.messages is not None:
+            # Allow if they're identical (redundant specification)
+            if self.input == self.messages:
+                return self
+            # If they're different, prefer input and copy to messages
+            else:
+                self.messages = self.input
+                return self
+
+        # If neither field is provided, that's an error
+        if self.input is None and self.messages is None:
+            raise ValueError("Must specify either 'input' or 'messages' field.")
+
+        # Create alias: copy messages to input if input is None
+        if self.input is None and self.messages is not None:
+            self.input = self.messages
+
+        # Create alias: copy input to messages if messages is None
+        elif self.messages is None and self.input is not None:
+            self.messages = self.input
+
+        return self
+
 
 class ChatHistoryModel(BaseModel):
     model_config = ConfigDict(use_enum_values=True, extra="forbid")
@@ -1374,3 +1500,10 @@ class AppConfig(BaseModel):
         graph: CompiledStateGraph = self.as_graph()
         app: ChatModel = create_agent(graph)
         return app
+
+    def as_responses_agent(self) -> ResponsesAgent:
+        from dao_ai.models import create_responses_agent
+
+        graph: CompiledStateGraph = self.as_graph()
+        app: ResponsesAgent = create_responses_agent(graph)
+        return app

{dao_ai-0.0.17 → dao_ai-0.0.18}/src/dao_ai/messages.py

@@ -11,6 +11,9 @@ from langchain_core.messages import (
 )
 from langchain_core.messages.modifier import RemoveMessage
 from mlflow.types.llm import ChatMessage
+from mlflow.types.responses import (
+    ResponsesAgentRequest,
+)
 
 
 def remove_messages(
@@ -96,6 +99,10 @@ def has_mlflow_messages(messages: ChatMessage | Sequence[ChatMessage]) -> bool:
     return any(isinstance(m, ChatMessage) for m in messages)
 
 
+def has_mlflow_responses_messages(messages: ResponsesAgentRequest) -> bool:
+    return isinstance(messages, ResponsesAgentRequest)
+
+
 def has_image(messages: BaseMessage | Sequence[BaseMessage]) -> bool:
     """
     Check if a message contains an image.