dao-ai 0.0.19__tar.gz → 0.0.20__tar.gz

{dao_ai-0.0.19 → dao_ai-0.0.20}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dao-ai
-Version: 0.0.19
+Version: 0.0.20
 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.19 → dao_ai-0.0.20}/config/quick_serve_restaurant/quick-serve-restaurant.yaml

@@ -225,11 +225,15 @@ retrievers:
 
 tools:
 
-  insert_coffee_order_uc_tool: &insert_coffee_order_uc_tool
+  insert_coffee_order_uc_tool: &insert_coffee_order_tool
     name: insert_coffee_order_tool_uc
     function:
       type: unity_catalog
       <<: *insert_coffee_order
+      partial_args:
+        host: *workspace_host_var
+        client_id: *client_id_var
+        client_secret: *client_secret_var
 
   match_item_by_description_and_price_tool: &match_item_by_description_and_price_tool
     name: match_item_by_description_and_price_uc
@@ -249,16 +253,6 @@ tools:
       type: unity_catalog
       <<: *lookup_items_by_descriptions
 
-  insert_coffee_order_tool: &insert_coffee_order_tool
-    name: insert_coffee_order
-    function:
-      type: factory
-      name: quick_serve_restaurant.tools.insert_coffee_order_tool
-      args:
-        tool: *insert_coffee_order_uc_tool
-        host: *workspace_host_var
-        client_id: *client_id_var
-        client_secret: *client_secret_var
 
   current_time_tool: &current_time_tool
     name: current_time

{dao_ai-0.0.19 → dao_ai-0.0.20}/pyproject.toml

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

@@ -664,6 +664,18 @@
     },
     "DatabaseModel": {
       "properties": {
+        "on_behalf_of_user": {
+          "anyOf": [
+            {
+              "type": "boolean"
+            },
+            {
+              "type": "null"
+            }
+          ],
+          "default": false,
+          "title": "On Behalf Of User"
+        },
         "name": {
           "title": "Name",
           "type": "string"
@@ -1214,7 +1226,34 @@
         "args": {
           "anyOf": [
             {
-              "additionalProperties": true,
+              "additionalProperties": {
+                "anyOf": [
+                  {
+                    "$ref": "#/$defs/CompositeVariableModel"
+                  },
+                  {
+                    "$ref": "#/$defs/EnvironmentVariableModel"
+                  },
+                  {
+                    "$ref": "#/$defs/SecretVariableModel"
+                  },
+                  {
+                    "$ref": "#/$defs/PrimitiveVariableModel"
+                  },
+                  {
+                    "type": "string"
+                  },
+                  {
+                    "type": "integer"
+                  },
+                  {
+                    "type": "number"
+                  },
+                  {
+                    "type": "boolean"
+                  }
+                ]
+              },
               "type": "object"
             },
             {
@@ -2517,6 +2556,45 @@
             }
           ],
           "default": null
+        },
+        "partial_args": {
+          "anyOf": [
+            {
+              "additionalProperties": {
+                "anyOf": [
+                  {
+                    "$ref": "#/$defs/CompositeVariableModel"
+                  },
+                  {
+                    "$ref": "#/$defs/EnvironmentVariableModel"
+                  },
+                  {
+                    "$ref": "#/$defs/SecretVariableModel"
+                  },
+                  {
+                    "$ref": "#/$defs/PrimitiveVariableModel"
+                  },
+                  {
+                    "type": "string"
+                  },
+                  {
+                    "type": "integer"
+                  },
+                  {
+                    "type": "number"
+                  },
+                  {
+                    "type": "boolean"
+                  }
+                ]
+              },
+              "type": "object"
+            },
+            {
+              "type": "null"
+            }
+          ],
+          "title": "Partial Args"
         }
       },
       "required": [

{dao_ai-0.0.19 → dao_ai-0.0.20}/src/dao_ai/config.py

@@ -867,7 +867,7 @@ class PythonFunctionModel(BaseFunctionModel, HasFullName):
 
 class FactoryFunctionModel(BaseFunctionModel, HasFullName):
     model_config = ConfigDict(use_enum_values=True, extra="forbid")
-    args: Optional[dict[str, Any]] = Field(default_factory=dict)
+    args: Optional[dict[str, AnyVariable]] = Field(default_factory=dict)
     type: Literal[FunctionType.FACTORY] = FunctionType.FACTORY
 
     @property
@@ -963,6 +963,7 @@ class McpFunctionModel(BaseFunctionModel, HasFullName):
 class UnityCatalogFunctionModel(BaseFunctionModel, HasFullName):
     model_config = ConfigDict(use_enum_values=True, extra="forbid")
     schema_model: Optional[SchemaModel] = Field(default=None, alias="schema")
+    partial_args: Optional[dict[str, AnyVariable]] = Field(default_factory=dict)
     type: Literal[FunctionType.UNITY_CATALOG] = FunctionType.UNITY_CATALOG
 
     @property

{dao_ai-0.0.19 → dao_ai-0.0.20}/src/dao_ai/guardrails.py

@@ -87,12 +87,12 @@ def judge_node(guardrails: GuardrailModel) -> RunnableLike:
         )
 
         if eval_result["score"]:
-            logger.debug("✅ Response approved by judge")
+            logger.debug("Response approved by judge")
             logger.debug(f"Judge's comment: {eval_result['comment']}")
             return
         else:
             # Otherwise, return the judge's critique as a new user message
-            logger.warning("⚠️ Judge requested improvements")
+            logger.warning("Judge requested improvements")
             comment: str = eval_result["comment"]
             logger.warning(f"Judge's critique: {comment}")
             content: str = "\n".join([human_message.content, comment])

{dao_ai-0.0.19 → dao_ai-0.0.20}/src/dao_ai/memory/postgres.py

@@ -20,137 +20,6 @@ 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()
@@ -251,7 +120,7 @@ class AsyncPostgresStoreManager(StoreManagerBase):
             )
 
             # Create store with the shared pool (using patched version)
-            self._store = PatchedAsyncPostgresStore(conn=self.pool)
+            self._store = AsyncPostgresStore(conn=self.pool)
 
             await self._store.setup()
 

{dao_ai-0.0.19 → dao_ai-0.0.20}/src/dao_ai/tools/human_in_the_loop.py

@@ -87,7 +87,11 @@ def as_human_in_the_loop(
     if isinstance(function, BaseFunctionModel):
         human_in_the_loop: HumanInTheLoopModel | None = function.human_in_the_loop
         if human_in_the_loop:
-            logger.debug(f"Adding human-in-the-loop to tool: {tool.name}")
+            # Get tool name safely - handle RunnableBinding objects
+            tool_name = getattr(tool, "name", None) or getattr(
+                getattr(tool, "bound", None), "name", "unknown_tool"
+            )
+            logger.debug(f"Adding human-in-the-loop to tool: {tool_name}")
             tool = add_human_in_the_loop(
                 tool=tool,
                 interrupt_config=human_in_the_loop.interupt_config,