dao-ai 0.0.17__py3-none-any.whl → 0.0.19__py3-none-any.whl

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/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 (
@@ -37,6 +39,7 @@ from mlflow.models import ModelConfig
 from mlflow.models.resources import (
     DatabricksFunction,
     DatabricksGenieSpace,
+    DatabricksLakebase,
     DatabricksResource,
     DatabricksServingEndpoint,
     DatabricksSQLWarehouse,
@@ -44,8 +47,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 +78,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 +244,70 @@ 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 +323,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,17 +404,19 @@ 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):
     model_config = ConfigDict(use_enum_values=True, extra="forbid")
     name: str
     description: Optional[str] = None
-    space_id: str
+    space_id: AnyVariable
 
     @property
     def api_scopes(self) -> Sequence[str]:
@@ -363,10 +424,17 @@ 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
+            )
+        ]
+
+    @model_validator(mode="after")
+    def update_space_id(self):
+        self.space_id = value_of(self.space_id)
+        return self
 
 
 class VolumeModel(BaseModel, HasFullName):
@@ -394,7 +462,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 +570,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 +592,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]:
@@ -554,19 +656,22 @@ class ConnectionModel(BaseModel, HasFullName, IsDatabricksResource):
     def api_scopes(self) -> Sequence[str]:
         return [
             "catalog.connections",
+            "serving.serving-endpoints",
         ]
 
-    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):
     model_config = ConfigDict()
     name: str
     description: Optional[str] = None
-    warehouse_id: str
+    warehouse_id: AnyVariable
 
     @property
     def api_scopes(self) -> Sequence[str]:
@@ -575,13 +680,20 @@ 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
+            )
+        ]
+
+    @model_validator(mode="after")
+    def update_warehouse_id(self):
+        self.warehouse_id = value_of(self.warehouse_id)
+        return self
 
 
-class DatabaseModel(BaseModel):
+class DatabaseModel(BaseModel, IsDatabricksResource):
     model_config = ConfigDict(frozen=True)
     name: str
     description: Optional[str] = None
@@ -597,6 +709,18 @@ class DatabaseModel(BaseModel):
     client_secret: Optional[AnyVariable] = None
     workspace_host: Optional[AnyVariable] = None
 
+    @property
+    def api_scopes(self) -> Sequence[str]:
+        return []
+
+    def as_resources(self) -> Sequence[DatabricksResource]:
+        return [
+            DatabricksLakebase(
+                database_instance_name=self.name,
+                on_behalf_of_user=self.on_behalf_of_user,
+            )
+        ]
+
     @model_validator(mode="after")
     def validate_auth_methods(self):
         oauth_fields: Sequence[Any] = [
@@ -1034,9 +1158,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")
@@ -1064,13 +1216,13 @@ class AppModel(BaseModel):
     endpoint_name: Optional[str] = None
     tags: Optional[dict[str, Any]] = Field(default_factory=dict)
     scale_to_zero: Optional[bool] = True
-    environment_vars: Optional[dict[str, Any]] = Field(default_factory=dict)
+    environment_vars: Optional[dict[str, AnyVariable]] = Field(default_factory=dict)
     budget_policy_id: Optional[str] = None
     workload_size: Optional[WorkloadSize] = "Small"
     permissions: Optional[list[AppPermissionModel]] = Field(default_factory=list)
     agents: list[AgentModel] = Field(default_factory=list)
 
-    orchestration: OrchestrationModel
+    orchestration: Optional[OrchestrationModel] = None
     alias: Optional[str] = None
     initialization_hooks: Optional[FunctionHook | list[FunctionHook]] = Field(
         default_factory=list
@@ -1089,7 +1241,38 @@ class AppModel(BaseModel):
     @model_validator(mode="after")
     def validate_agents_not_empty(self):
         if not self.agents:
-            raise ValueError("agents must contain at least one item")
+            raise ValueError("At least one agent must be specified")
+        return self
+
+    @model_validator(mode="after")
+    def update_environment_vars(self):
+        for key, value in self.environment_vars.items():
+            if isinstance(value, SecretVariableModel):
+                updated_value = str(value)
+            else:
+                updated_value = value_of(value)
+
+            self.environment_vars[key] = updated_value
+        return self
+
+    @model_validator(mode="after")
+    def set_default_orchestration(self):
+        if self.orchestration is None:
+            if len(self.agents) > 1:
+                default_agent: AgentModel = self.agents[0]
+                self.orchestration = OrchestrationModel(
+                    swarm=SupervisorModel(model=default_agent.model)
+                )
+            elif len(self.agents) == 1:
+                default_agent: AgentModel = self.agents[0]
+                self.orchestration = OrchestrationModel(
+                    supervisor=SwarmModel(
+                        model=default_agent.model, default_agent=default_agent
+                    )
+                )
+            else:
+                raise ValueError("At least one agent must be specified")
+
         return self
 
     @model_validator(mode="after")
@@ -1374,3 +1557,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/graph.py

@@ -136,8 +136,8 @@ def _create_supervisor_graph(config: AppConfig) -> CompiledStateGraph:
 
     workflow: StateGraph = StateGraph(
         SharedState,
-        input=IncomingState,
-        output=OutgoingState,
+        input_schema=IncomingState,
+        output_schema=OutgoingState,
         context_schema=Context,
     )
 
@@ -200,8 +200,8 @@ def _create_swarm_graph(config: AppConfig) -> CompiledStateGraph:
 
     workflow: StateGraph = StateGraph(
         SharedState,
-        input=IncomingState,
-        output=OutgoingState,
+        input_schema=IncomingState,
+        output_schema=OutgoingState,
         context_schema=Context,
     )
 

dao_ai/memory/core.py

@@ -99,13 +99,13 @@ class CheckpointManager:
                         checkpointer_manager
                     )
             case StorageType.POSTGRES:
-                from dao_ai.memory.postgres import PostgresCheckpointerManager
+                from dao_ai.memory.postgres import AsyncPostgresCheckpointerManager
 
                 checkpointer_manager = cls.checkpoint_managers.get(
                     checkpointer_model.database.name
                 )
                 if checkpointer_manager is None:
-                    checkpointer_manager = PostgresCheckpointerManager(
+                    checkpointer_manager = AsyncPostgresCheckpointerManager(
                         checkpointer_model
                     )
                     cls.checkpoint_managers[checkpointer_model.database.name] = (

dao_ai/memory/postgres.py

@@ -20,6 +20,137 @@ from dao_ai.memory.base import (
 )
 
 
+class PatchedAsyncPostgresStore(AsyncPostgresStore):
+    """
+    Patched version of AsyncPostgresStore that properly handles event loop initialization
+    and task lifecycle management.
+
+    The issues occur because:
+    1. AsyncBatchedBaseStore.__init__ calls asyncio.get_running_loop() and fails if no event loop is running
+    2. The background _task can complete/fail, causing assertions in asearch/other methods to fail
+    3. Destructor tries to access _task even when it doesn't exist
+
+    This patch ensures proper initialization and handles task lifecycle robustly.
+    """
+
+    def __init__(self, *args, **kwargs):
+        # Ensure we have a running event loop before calling super().__init__()
+        loop = None
+        try:
+            loop = asyncio.get_running_loop()
+        except RuntimeError:
+            # No running loop - create one temporarily for initialization
+            loop = asyncio.new_event_loop()
+            asyncio.set_event_loop(loop)
+
+        try:
+            super().__init__(*args, **kwargs)
+        except Exception as e:
+            # If parent initialization fails, ensure _task is at least defined
+            if not hasattr(self, "_task"):
+                self._task = None
+            logger.warning(f"AsyncPostgresStore initialization failed: {e}")
+            raise
+
+    def _ensure_task_running(self):
+        """
+        Ensure the background task is running. Recreate it if necessary.
+        """
+        if not hasattr(self, "_task") or self._task is None:
+            logger.error("AsyncPostgresStore task not initialized")
+            raise RuntimeError("Store task not properly initialized")
+
+        if self._task.done():
+            logger.warning(
+                "AsyncPostgresStore background task completed, attempting to restart"
+            )
+            # Try to get the task exception for debugging
+            try:
+                exception = self._task.exception()
+                if exception:
+                    logger.error(f"Background task failed with: {exception}")
+                else:
+                    logger.info("Background task completed normally")
+            except Exception as e:
+                logger.warning(f"Could not determine task completion reason: {e}")
+
+            # Try to restart the task
+            try:
+                import weakref
+
+                from langgraph.store.base.batch import _run
+
+                self._task = self._loop.create_task(
+                    _run(self._aqueue, weakref.ref(self))
+                )
+                logger.info("Successfully restarted AsyncPostgresStore background task")
+            except Exception as e:
+                logger.error(f"Failed to restart background task: {e}")
+                raise RuntimeError(
+                    f"Store background task failed and could not be restarted: {e}"
+                )
+
+    async def asearch(
+        self,
+        namespace_prefix,
+        /,
+        *,
+        query=None,
+        filter=None,
+        limit=10,
+        offset=0,
+        refresh_ttl=None,
+    ):
+        """
+        Override asearch to handle task lifecycle issues gracefully.
+        """
+        self._ensure_task_running()
+
+        # Call parent implementation if task is healthy
+        return await super().asearch(
+            namespace_prefix,
+            query=query,
+            filter=filter,
+            limit=limit,
+            offset=offset,
+            refresh_ttl=refresh_ttl,
+        )
+
+    async def aget(self, namespace, key, /, *, refresh_ttl=None):
+        """Override aget with task lifecycle management."""
+        self._ensure_task_running()
+        return await super().aget(namespace, key, refresh_ttl=refresh_ttl)
+
+    async def aput(self, namespace, key, value, /, *, refresh_ttl=None):
+        """Override aput with task lifecycle management."""
+        self._ensure_task_running()
+        return await super().aput(namespace, key, value, refresh_ttl=refresh_ttl)
+
+    async def adelete(self, namespace, key):
+        """Override adelete with task lifecycle management."""
+        self._ensure_task_running()
+        return await super().adelete(namespace, key)
+
+    async def alist_namespaces(self, *, prefix=None):
+        """Override alist_namespaces with task lifecycle management."""
+        self._ensure_task_running()
+        return await super().alist_namespaces(prefix=prefix)
+
+    def __del__(self):
+        """
+        Override destructor to handle missing _task attribute gracefully.
+        """
+        try:
+            # Only try to cancel if _task exists and is not None
+            if hasattr(self, "_task") and self._task is not None:
+                if not self._task.done():
+                    self._task.cancel()
+        except Exception as e:
+            # Log but don't raise - destructors should not raise exceptions
+            logger.debug(f"AsyncPostgresStore destructor cleanup: {e}")
+            pass
+
+
 class AsyncPostgresPoolManager:
     _pools: dict[str, AsyncConnectionPool] = {}
     _lock: asyncio.Lock = asyncio.Lock()
@@ -119,8 +250,9 @@ class AsyncPostgresStoreManager(StoreManagerBase):
                 self.store_model.database
             )
 
-            # Create store with the shared pool
-            self._store = AsyncPostgresStore(conn=self.pool)
+            # Create store with the shared pool (using patched version)
+            self._store = PatchedAsyncPostgresStore(conn=self.pool)
+
             await self._store.setup()
 
             self._setup_complete = True

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.

dao_ai/models.py

@@ -1,13 +1,14 @@
 import uuid
 from os import PathLike
 from pathlib import Path
-from typing import Any, Generator, Optional, Sequence
+from typing import Any, Generator, Optional, Sequence, Union
 
 from langchain_core.messages import AIMessage, AIMessageChunk, BaseMessage
 from langgraph.graph.state import CompiledStateGraph
 from loguru import logger
 from mlflow import MlflowClient
-from mlflow.pyfunc import ChatAgent, ChatModel
+from mlflow.pyfunc import ChatAgent, ChatModel, ResponsesAgent
+from mlflow.types.agent import ChatContext
 from mlflow.types.llm import (
     ChatChoice,
     ChatChoiceDelta,
@@ -17,8 +18,21 @@ from mlflow.types.llm import (
     ChatMessage,
     ChatParams,
 )
+from mlflow.types.responses import (
+    ResponsesAgentRequest,
+    ResponsesAgentResponse,
+    ResponsesAgentStreamEvent,
+)
+from mlflow.types.responses_helpers import (
+    Message,
+    ResponseInputTextParam,
+)
 
-from dao_ai.messages import has_langchain_messages, has_mlflow_messages
+from dao_ai.messages import (
+    has_langchain_messages,
+    has_mlflow_messages,
+    has_mlflow_responses_messages,
+)
 from dao_ai.state import Context
 
 
@@ -185,6 +199,266 @@ class LanggraphChatModel(ChatModel):
         return [m.to_dict() for m in messages]
 
 
+class LanggraphResponsesAgent(ResponsesAgent):
+    """
+    ResponsesAgent that delegates requests to a LangGraph CompiledStateGraph.
+
+    This is the modern replacement for LanggraphChatModel, providing better
+    support for streaming, tool calling, and async execution.
+    """
+
+    def __init__(self, graph: CompiledStateGraph) -> None:
+        self.graph = graph
+
+    def predict(self, request: ResponsesAgentRequest) -> ResponsesAgentResponse:
+        """
+        Process a ResponsesAgentRequest and return a ResponsesAgentResponse.
+        """
+        logger.debug(f"ResponsesAgent request: {request}")
+
+        # Convert ResponsesAgent input to LangChain messages
+        messages = self._convert_request_to_langchain_messages(request)
+
+        # Prepare context
+        context: Context = self._convert_request_to_context(request)
+        custom_inputs: dict[str, Any] = {"configurable": context.model_dump()}
+
+        # Use async ainvoke internally for parallel execution
+        import asyncio
+
+        async def _async_invoke():
+            try:
+                return await self.graph.ainvoke(
+                    {"messages": messages}, context=context, config=custom_inputs
+                )
+            except Exception as e:
+                logger.error(f"Error in graph.ainvoke: {e}")
+                raise
+
+        try:
+            loop = asyncio.get_event_loop()
+        except RuntimeError:
+            # Handle case where no event loop exists (common in some deployment scenarios)
+            loop = asyncio.new_event_loop()
+            asyncio.set_event_loop(loop)
+
+        try:
+            response: dict[str, Sequence[BaseMessage]] = loop.run_until_complete(
+                _async_invoke()
+            )
+        except Exception as e:
+            logger.error(f"Error in async execution: {e}")
+            raise
+
+        # Convert response to ResponsesAgent format
+        last_message: BaseMessage = response["messages"][-1]
+
+        output_item = self.create_text_output_item(
+            text=last_message.content, id=f"msg_{uuid.uuid4().hex[:8]}"
+        )
+
+        custom_outputs = custom_inputs
+        return ResponsesAgentResponse(
+            output=[output_item], custom_outputs=custom_outputs
+        )
+
+    def predict_stream(
+        self, request: ResponsesAgentRequest
+    ) -> Generator[ResponsesAgentStreamEvent, None, None]:
+        """
+        Process a ResponsesAgentRequest and yield ResponsesAgentStreamEvent objects.
+        """
+        logger.debug(f"ResponsesAgent stream request: {request}")
+
+        # Convert ResponsesAgent input to LangChain messages
+        messages: list[BaseMessage] = self._convert_request_to_langchain_messages(
+            request
+        )
+
+        # Prepare context
+        context: Context = self._convert_request_to_context(request)
+        custom_inputs: dict[str, Any] = {"configurable": context.model_dump()}
+
+        # Use async astream internally for parallel execution
+        import asyncio
+
+        async def _async_stream():
+            item_id = f"msg_{uuid.uuid4().hex[:8]}"
+            accumulated_content = ""
+
+            try:
+                async for nodes, stream_mode, messages_batch in self.graph.astream(
+                    {"messages": messages},
+                    context=context,
+                    config=custom_inputs,
+                    stream_mode=["messages", "custom"],
+                    subgraphs=True,
+                ):
+                    nodes: tuple[str, ...]
+                    stream_mode: str
+                    messages_batch: Sequence[BaseMessage]
+
+                    for message in messages_batch:
+                        if (
+                            isinstance(
+                                message,
+                                (
+                                    AIMessageChunk,
+                                    AIMessage,
+                                ),
+                            )
+                            and message.content
+                            and "summarization" not in nodes
+                        ):
+                            content = message.content
+                            accumulated_content += content
+
+                            # Yield streaming delta
+                            yield ResponsesAgentStreamEvent(
+                                **self.create_text_delta(delta=content, item_id=item_id)
+                            )
+
+                custom_outputs = custom_inputs
+                # Yield final output item
+                yield ResponsesAgentStreamEvent(
+                    type="response.output_item.done",
+                    item=self.create_text_output_item(
+                        text=accumulated_content, id=item_id
+                    ),
+                    custom_outputs=custom_outputs,
+                )
+            except Exception as e:
+                logger.error(f"Error in graph.astream: {e}")
+                raise
+
+        # Convert async generator to sync generator
+        try:
+            loop = asyncio.get_event_loop()
+        except RuntimeError:
+            # Handle case where no event loop exists (common in some deployment scenarios)
+            loop = asyncio.new_event_loop()
+            asyncio.set_event_loop(loop)
+
+        async_gen = _async_stream()
+
+        try:
+            while True:
+                try:
+                    item = loop.run_until_complete(async_gen.__anext__())
+                    yield item
+                except StopAsyncIteration:
+                    break
+                except Exception as e:
+                    logger.error(f"Error in streaming: {e}")
+                    raise
+        finally:
+            try:
+                loop.run_until_complete(async_gen.aclose())
+            except Exception as e:
+                logger.warning(f"Error closing async generator: {e}")
+
+    def _extract_text_from_content(
+        self,
+        content: Union[str, list[Union[ResponseInputTextParam, str, dict[str, Any]]]],
+    ) -> str:
+        """Extract text content from various MLflow content formats.
+
+        MLflow ResponsesAgent supports multiple content formats:
+        - str: Simple text content
+        - list[ResponseInputTextParam]: Structured text objects with .text attribute
+        - list[dict]: Dictionaries with "text" key
+        - Mixed lists of the above types
+
+        This method normalizes all formats to a single concatenated string.
+
+        Args:
+            content: The content to extract text from
+
+        Returns:
+            Concatenated text string from all content items
+        """
+        if isinstance(content, str):
+            return content
+        elif isinstance(content, list):
+            text_parts = []
+            for content_item in content:
+                if isinstance(content_item, ResponseInputTextParam):
+                    text_parts.append(content_item.text)
+                elif isinstance(content_item, str):
+                    text_parts.append(content_item)
+                elif isinstance(content_item, dict) and "text" in content_item:
+                    text_parts.append(content_item["text"])
+            return "".join(text_parts)
+        else:
+            # Fallback for unknown types - try to extract text attribute
+            return getattr(content, "text", str(content))
+
+    def _convert_request_to_langchain_messages(
+        self, request: ResponsesAgentRequest
+    ) -> list[dict[str, Any]]:
+        """Convert ResponsesAgent input to LangChain message format."""
+        messages = []
+
+        for input_item in request.input:
+            if isinstance(input_item, Message):
+                # Handle MLflow Message objects
+                content = self._extract_text_from_content(input_item.content)
+                messages.append({"role": input_item.role, "content": content})
+            elif isinstance(input_item, dict):
+                # Handle dict format
+                if "role" in input_item and "content" in input_item:
+                    content = self._extract_text_from_content(input_item["content"])
+                    messages.append({"role": input_item["role"], "content": content})
+            else:
+                # Fallback for other object types with role/content attributes
+                role = getattr(input_item, "role", "user")
+                content = self._extract_text_from_content(
+                    getattr(input_item, "content", "")
+                )
+                messages.append({"role": role, "content": content})
+
+        return messages
+
+    def _convert_request_to_context(self, request: ResponsesAgentRequest) -> Context:
+        """Convert ResponsesAgent context to internal Context."""
+
+        logger.debug(f"request.context: {request.context}")
+        logger.debug(f"request.custom_inputs: {request.custom_inputs}")
+
+        configurable: dict[str, Any] = {}
+
+        # Process context values first (lower priority)
+        # Use strong typing with forward-declared type hints instead of hasattr checks
+        chat_context: Optional[ChatContext] = request.context
+        if chat_context is not None:
+            conversation_id: Optional[str] = chat_context.conversation_id
+            user_id: Optional[str] = chat_context.user_id
+
+            if conversation_id is not None:
+                configurable["conversation_id"] = conversation_id
+                configurable["thread_id"] = conversation_id
+
+            if user_id is not None:
+                configurable["user_id"] = user_id
+
+        # Process custom_inputs after context so they can override context values (higher priority)
+        if request.custom_inputs:
+            if "configurable" in request.custom_inputs:
+                configurable.update(request.custom_inputs.pop("configurable"))
+
+            configurable.update(request.custom_inputs)
+
+        if "user_id" in configurable:
+            configurable["user_id"] = configurable["user_id"].replace(".", "_")
+
+        if "thread_id" not in configurable:
+            configurable["thread_id"] = str(uuid.uuid4())
+
+        logger.debug(f"Creating context from: {configurable}")
+
+        return Context(**configurable)
+
+
 def create_agent(graph: CompiledStateGraph) -> ChatAgent:
     """
     Create an MLflow-compatible ChatAgent from a LangGraph state machine.
@@ -201,6 +475,22 @@ def create_agent(graph: CompiledStateGraph) -> ChatAgent:
     return LanggraphChatModel(graph)
 
 
+def create_responses_agent(graph: CompiledStateGraph) -> ResponsesAgent:
+    """
+    Create an MLflow-compatible ResponsesAgent from a LangGraph state machine.
+
+    Factory function that wraps a compiled LangGraph in the LanggraphResponsesAgent
+    class to make it deployable through MLflow.
+
+    Args:
+        graph: A compiled LangGraph state machine
+
+    Returns:
+        An MLflow-compatible ResponsesAgent instance
+    """
+    return LanggraphResponsesAgent(graph)
+
+
 def _process_langchain_messages(
     app: LanggraphChatModel | CompiledStateGraph,
     messages: Sequence[BaseMessage],
@@ -266,7 +556,14 @@ def _process_langchain_messages_stream(
                     yield message
 
     # Convert async generator to sync generator
-    loop = asyncio.get_event_loop()
+
+    try:
+        loop = asyncio.get_event_loop()
+    except RuntimeError:
+        # Handle case where no event loop exists (common in some deployment scenarios)
+        loop = asyncio.new_event_loop()
+        asyncio.set_event_loop(loop)
+
     async_gen = _async_stream()
 
     try:
@@ -288,6 +585,14 @@ def _process_mlflow_messages(
     return app.predict(None, messages, custom_inputs)
 
 
+def _process_mlflow_response_messages(
+    app: ResponsesAgent,
+    messages: ResponsesAgentRequest,
+) -> ResponsesAgentResponse:
+    """Process MLflow ResponsesAgent request in batch mode."""
+    return app.predict(messages)
+
+
 def _process_mlflow_messages_stream(
     app: ChatModel,
     messages: Sequence[ChatMessage],
@@ -298,32 +603,68 @@ def _process_mlflow_messages_stream(
         yield event
 
 
+def _process_mlflow_response_messages_stream(
+    app: ResponsesAgent,
+    messages: ResponsesAgentRequest,
+) -> Generator[ResponsesAgentStreamEvent, None, None]:
+    """Process MLflow ResponsesAgent request in streaming mode."""
+    for event in app.predict_stream(messages):
+        event: ResponsesAgentStreamEvent
+        yield event
+
+
 def _process_config_messages(
-    app: ChatModel,
+    app: LanggraphChatModel | LanggraphResponsesAgent,
     messages: dict[str, Any],
     custom_inputs: Optional[dict[str, Any]] = None,
-) -> ChatCompletionResponse:
-    messages: Sequence[ChatMessage] = [ChatMessage(**m) for m in messages]
-    params: ChatParams = ChatParams(**{"custom_inputs": custom_inputs})
-
-    return _process_mlflow_messages(app, messages, params)
+) -> ChatCompletionResponse | ResponsesAgentResponse:
+    if isinstance(app, LanggraphChatModel):
+        messages: Sequence[ChatMessage] = [ChatMessage(**m) for m in messages]
+        params: ChatParams = ChatParams(**{"custom_inputs": custom_inputs})
+        return _process_mlflow_messages(app, messages, params)
+
+    elif isinstance(app, LanggraphResponsesAgent):
+        input_messages: list[Message] = [Message(**m) for m in messages]
+        request = ResponsesAgentRequest(
+            input=input_messages, custom_inputs=custom_inputs
+        )
+        return _process_mlflow_response_messages(app, request)
 
 
 def _process_config_messages_stream(
-    app: ChatModel, messages: dict[str, Any], custom_inputs: dict[str, Any]
-) -> Generator[ChatCompletionChunk, None, None]:
-    messages: Sequence[ChatMessage] = [ChatMessage(**m) for m in messages]
-    params: ChatParams = ChatParams(**{"custom_inputs": custom_inputs})
+    app: LanggraphChatModel | LanggraphResponsesAgent,
+    messages: dict[str, Any],
+    custom_inputs: dict[str, Any],
+) -> Generator[ChatCompletionChunk | ResponsesAgentStreamEvent, None, None]:
+    if isinstance(app, LanggraphChatModel):
+        messages: Sequence[ChatMessage] = [ChatMessage(**m) for m in messages]
+        params: ChatParams = ChatParams(**{"custom_inputs": custom_inputs})
 
-    for event in _process_mlflow_messages_stream(app, messages, custom_inputs=params):
-        yield event
+        for event in _process_mlflow_messages_stream(
+            app, messages, custom_inputs=params
+        ):
+            yield event
+
+    elif isinstance(app, LanggraphResponsesAgent):
+        input_messages: list[Message] = [Message(**m) for m in messages]
+        request = ResponsesAgentRequest(
+            input=input_messages, custom_inputs=custom_inputs
+        )
+
+        for event in _process_mlflow_response_messages_stream(app, request):
+            yield event
 
 
 def process_messages_stream(
-    app: LanggraphChatModel,
-    messages: Sequence[BaseMessage] | Sequence[ChatMessage] | dict[str, Any],
+    app: LanggraphChatModel | LanggraphResponsesAgent,
+    messages: Sequence[BaseMessage]
+    | Sequence[ChatMessage]
+    | ResponsesAgentRequest
+    | dict[str, Any],
     custom_inputs: Optional[dict[str, Any]] = None,
-) -> Generator[ChatCompletionChunk | AIMessageChunk, None, None]:
+) -> Generator[
+    ChatCompletionChunk | ResponsesAgentStreamEvent | AIMessageChunk, None, None
+]:
     """
     Process messages through a ChatAgent in streaming mode.
 
@@ -338,7 +679,10 @@ def process_messages_stream(
         Individual message chunks from the streaming response
     """
 
-    if has_mlflow_messages(messages):
+    if has_mlflow_responses_messages(messages):
+        for event in _process_mlflow_response_messages_stream(app, messages):
+            yield event
+    elif has_mlflow_messages(messages):
         for event in _process_mlflow_messages_stream(app, messages, custom_inputs):
             yield event
     elif has_langchain_messages(messages):
@@ -350,10 +694,13 @@ def process_messages_stream(
 
 
 def process_messages(
-    app: LanggraphChatModel,
-    messages: Sequence[BaseMessage] | Sequence[ChatMessage] | dict[str, Any],
+    app: LanggraphChatModel | LanggraphResponsesAgent,
+    messages: Sequence[BaseMessage]
+    | Sequence[ChatMessage]
+    | ResponsesAgentRequest
+    | dict[str, Any],
     custom_inputs: Optional[dict[str, Any]] = None,
-) -> ChatCompletionResponse | dict[str, Any] | Any:
+) -> ChatCompletionResponse | ResponsesAgentResponse | dict[str, Any] | Any:
     """
     Process messages through a ChatAgent in batch mode.
 
@@ -368,7 +715,9 @@ def process_messages(
         Complete response from the agent
     """
 
-    if has_mlflow_messages(messages):
+    if has_mlflow_responses_messages(messages):
+        return _process_mlflow_response_messages(app, messages)
+    elif has_mlflow_messages(messages):
         return _process_mlflow_messages(app, messages, custom_inputs)
     elif has_langchain_messages(messages):
         return _process_langchain_messages(app, messages, custom_inputs)

dao_ai/providers/databricks.py

@@ -42,6 +42,7 @@ import dao_ai
 from dao_ai.config import (
     AppConfig,
     ConnectionModel,
+    DatabaseModel,
     DatasetModel,
     FunctionModel,
     GenieRoomModel,
@@ -224,6 +225,7 @@ class DatabricksProvider(ServiceProvider):
         connections: Sequence[ConnectionModel] = list(
             config.resources.connections.values()
         )
+        databases: Sequence[DatabaseModel] = list(config.resources.databases.values())
 
         resources: Sequence[IsDatabricksResource] = (
             llms
@@ -233,14 +235,19 @@ class DatabricksProvider(ServiceProvider):
             + functions
             + tables
             + connections
+            + databases
         )
 
-        # all_resources: Sequence[DatabricksResource] = [
-        #     r.as_resource() for r in resources
-        # ]
+        # Flatten all resources from all models into a single list
+        all_resources: list[DatabricksResource] = []
+        for r in resources:
+            all_resources.extend(r.as_resources())
 
         system_resources: Sequence[DatabricksResource] = [
-            r.as_resource() for r in resources if not r.on_behalf_of_user
+            resource
+            for r in resources
+            for resource in r.as_resources()
+            if not r.on_behalf_of_user
         ]
         logger.debug(f"system_resources: {[r.name for r in system_resources]}")
 

{dao_ai-0.0.17.dist-info → dao_ai-0.0.19.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dao-ai
-Version: 0.0.17
+Version: 0.0.19
 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
@@ -24,22 +24,22 @@ Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Classifier: Topic :: System :: Distributed Computing
 Requires-Python: >=3.12
-Requires-Dist: databricks-agents>=1.2.0
-Requires-Dist: databricks-langchain>=0.4.2
-Requires-Dist: databricks-sdk[openai]>=0.55.0
+Requires-Dist: databricks-agents>=1.6.0
+Requires-Dist: databricks-langchain>=0.8.0
+Requires-Dist: databricks-sdk[openai]>=0.66.0
 Requires-Dist: duckduckgo-search>=8.0.2
 Requires-Dist: grandalf>=0.8
-Requires-Dist: langchain-mcp-adapters>=0.1.9
+Requires-Dist: langchain-mcp-adapters>=0.1.10
 Requires-Dist: langchain-tavily>=0.2.11
 Requires-Dist: langchain>=0.3.27
 Requires-Dist: langgraph-checkpoint-postgres>=2.0.23
 Requires-Dist: langgraph-supervisor>=0.0.29
 Requires-Dist: langgraph-swarm>=0.0.14
-Requires-Dist: langgraph>=0.6.5
+Requires-Dist: langgraph>=0.6.7
 Requires-Dist: langmem>=0.0.29
 Requires-Dist: loguru>=0.7.3
-Requires-Dist: mcp>=1.9.1
-Requires-Dist: mlflow>=3.3.2
+Requires-Dist: mcp>=1.14.1
+Requires-Dist: mlflow>=3.4.0
 Requires-Dist: nest-asyncio>=1.6.0
 Requires-Dist: openevals>=0.0.19
 Requires-Dist: openpyxl>=3.1.5

{dao_ai-0.0.17.dist-info → dao_ai-0.0.19.dist-info}/RECORD

@@ -1,13 +1,13 @@
 dao_ai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-dao_ai/agent_as_code.py,sha256=rMWMC0nabtABHPD5H-Yy9ej7QNebLnXfvlZPiwrthoE,457
+dao_ai/agent_as_code.py,sha256=kPSeDz2-1jRaed1TMs4LA3VECoyqe9_Ed2beRLB9gXQ,472
 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=VyHqkW1UMQJ0fzyme1fV_3fi_6wDmKRQeCrx881fDQ4,45173
-dao_ai/graph.py,sha256=kXaGLGFVekDWqm-AHzti6LmrXnyi99VQ-AdCGuNb_xM,7831
+dao_ai/config.py,sha256=N_Vc-rJHvBzbia4TyAExGhCvZKXlk49bskrI_sbxwjg,51869
+dao_ai/graph.py,sha256=gmD9mxODfXuvn9xWeBfewm1FiuVAWMLEdnZz7DNmSH0,7859
 dao_ai/guardrails.py,sha256=-Qh0f_2Db9t4Nbrrx9FM7tnpqShjMoyxepZ0HByItfU,4027
-dao_ai/messages.py,sha256=tRZQTeb5YFKu8cm1xeaCkKhidq-0tdzncNEzVePvits,6806
-dao_ai/models.py,sha256=Zf5Rqus5xcdpxSvuLlDau4JM1j1fF9v_MnQ7HW4BXU4,13862
+dao_ai/messages.py,sha256=xl_3-WcFqZKCFCiov8sZOPljTdM3gX3fCHhxq-xFg2U,7005
+dao_ai/models.py,sha256=Xb23U-lhDG8KyNRIijcJ4InluadlaGNy4rrYx7Cjgfg,26939
 dao_ai/nodes.py,sha256=SSuFNTXOdFaKg_aX-yUkQO7fM9wvNGu14lPXKDapU1U,8461
 dao_ai/prompts.py,sha256=vpmIbWs_szXUgNNDs5Gh2LcxKZti5pHDKSfoClUcgX0,1289
 dao_ai/state.py,sha256=GwbMbd1TWZx1T5iQrEOX6_rpxOitlmyeJ8dMr2o_pag,1031
@@ -18,11 +18,11 @@ dao_ai/hooks/__init__.py,sha256=LlHGIuiZt6vGW8K5AQo1XJEkBP5vDVtMhq0IdjcLrD4,417
 dao_ai/hooks/core.py,sha256=ZShHctUSoauhBgdf1cecy9-D7J6-sGn-pKjuRMumW5U,6663
 dao_ai/memory/__init__.py,sha256=1kHx_p9abKYFQ6EYD05nuc1GS5HXVEpufmjBGw_7Uho,260
 dao_ai/memory/base.py,sha256=99nfr2UZJ4jmfTL_KrqUlRSCoRxzkZyWyx5WqeUoMdQ,338
-dao_ai/memory/core.py,sha256=K45iCEFbqJCVxMi4m3vmBJi4c6TQ-UtKGzyugDTkPP0,4141
-dao_ai/memory/postgres.py,sha256=YILzA7xtqawPAOLFaGG_i17zW7cQxXTzTD8yd-ipe8k,12480
+dao_ai/memory/core.py,sha256=g7chjBgVgx3iKjR2hghl0QL1j3802uIM_e7mgszur9M,4151
+dao_ai/memory/postgres.py,sha256=ncvEKFYX-ZjUDYVmuWBMcZnykcp2eK4TP-ojzqkwDsk,17433
 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=XRPOqwF5SeA9rPAOWMg2gSMC7lw31BI5VI_4K0KIOqo,27931
+dao_ai/providers/databricks.py,sha256=fZ8mGotfA3W3t5yUej2xGmGHSybjBFYr895mOctT418,28203
 dao_ai/tools/__init__.py,sha256=ye6MHaJY7tUnJ8336YJiLxuZr55zDPNdOw6gm7j5jlc,1103
 dao_ai/tools/agent.py,sha256=WbQnyziiT12TLMrA7xK0VuOU029tdmUBXbUl-R1VZ0Q,1886
 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=EDYQs51zIPaAP0ma1D81wJT77GQ-v-cjb2XrFVWfWdg,2621
-dao_ai-0.0.17.dist-info/METADATA,sha256=G50pidDVsQt5j4T2NhygeeYJyMOdL7fSI_zyMaWIGUo,41378
-dao_ai-0.0.17.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-dao_ai-0.0.17.dist-info/entry_points.txt,sha256=Xa-UFyc6gWGwMqMJOt06ZOog2vAfygV_DSwg1AiP46g,43
-dao_ai-0.0.17.dist-info/licenses/LICENSE,sha256=YZt3W32LtPYruuvHE9lGk2bw6ZPMMJD8yLrjgHybyz4,1069
-dao_ai-0.0.17.dist-info/RECORD,,
+dao_ai-0.0.19.dist-info/METADATA,sha256=hus4RZHOCTgDR6Rs8zS9l0OusplrFzryWCLsXZpTxgw,41380
+dao_ai-0.0.19.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+dao_ai-0.0.19.dist-info/entry_points.txt,sha256=Xa-UFyc6gWGwMqMJOt06ZOog2vAfygV_DSwg1AiP46g,43
+dao_ai-0.0.19.dist-info/licenses/LICENSE,sha256=YZt3W32LtPYruuvHE9lGk2bw6ZPMMJD8yLrjgHybyz4,1069
+dao_ai-0.0.19.dist-info/RECORD,,