dao-ai 0.0.4__tar.gz → 0.0.6__tar.gz

{dao_ai-0.0.4 → dao_ai-0.0.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dao-ai
-Version: 0.0.4
+Version: 0.0.6
 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,7 +24,7 @@ Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Classifier: Topic :: System :: Distributed Computing
 Requires-Python: >=3.11
-Requires-Dist: databricks-agents>=0.20.0
+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-vectorsearch>=0.56

{dao_ai-0.0.4 → dao_ai-0.0.6}/config/dais2025/model_config_dais.yaml

@@ -161,6 +161,7 @@ resources:
       schema: *dais_schema
       name: dais_rct
 
+
   functions:
     find_inventory_by_sku: &find_inventory_by_sku
       schema: *dais_schema
@@ -413,14 +414,20 @@ tools:
       args:
         llm_model: *tool_calling_llm
 
+  # dais_find_store_by_number_tool: &dais_find_store_by_number_tool
+  #   name: dais_find_store_by_number
+  #   function:
+  #     type: factory
+  #     name: dais2025.tools.store.create_find_store_by_number_tool
+  #     args:
+  #       schema: *dais_schema
+  #       warehouse: *dais_warehouse
+
   dais_find_store_by_number_tool: &dais_find_store_by_number_tool
     name: dais_find_store_by_number
     function:
-      type: factory
-      name: dais2025.tools.store.create_find_store_by_number_tool
-      args:
-        schema: *dais_schema
-        warehouse: *dais_warehouse
+      type: unity_catalog
+      <<: *find_store_by_number
 
   dais_store_number_extraction_tool: &dais_store_number_extraction_tool
     name: dais_store_number_extraction
@@ -1343,6 +1350,10 @@ evaluation:
     schema: *dais_schema
     name: evaluation
   num_evals: 25
+  custom_inputs:
+    configurable:
+      thread_id: "1"
+      user_id: my_user_id
 
 datasets:
   # Core product and inventory tables

{dao_ai-0.0.4 → dao_ai-0.0.6}/config/hardware_store/supervisor.yaml

@@ -919,7 +919,7 @@ app:
     - *recommendation                               # Product recommendation agent
     - *general                                      # General purpose agent
   orchestration:                                    # Agent orchestration configuration
-    memory: *memory                                 # Reference to memory configuration
+    #memory: *memory                                 # Reference to memory configuration
     supervisor:                                     # Supervisor orchestration pattern
       model: *tool_calling_llm                      # LLM for routing decisions
   message_hooks: 

{dao_ai-0.0.4 → dao_ai-0.0.6}/config/hardware_store/supervisor_postgres.yaml

@@ -956,11 +956,11 @@ app:
           thread_id: "1"
           user_id: my_user_id
           store_num: 87887
-  chat_history: 
+  #chat_history: 
     #max_tokens: 256
-    max_messages_before_summary: 5
-    max_summary_tokens: 128
-    model: *fast_llm                          # Model for summarization tasks
+  #  max_messages_before_summary: 5
+  #  max_summary_tokens: 128
+  #  model: *fast_llm                          # Model for summarization tasks
     # swarm:
     #   model: *tool_calling_llm
     #   default_agent: *general

{dao_ai-0.0.4 → dao_ai-0.0.6}/databricks.yaml

@@ -86,25 +86,11 @@ resources:
           environment_key: default
           
 targets:
-  azure:
-    default: True
-    workspace:
-      host: https://adb-984752964297111.11.azuredatabricks.net
-      root_path: /Users/${workspace.current_user.userName}/.bundle/${var.app_name}/${bundle.target}
+  dev:
+    mode: development
+    default: true
 
-  aws:
-    workspace:
-      host: https://e2-demo-field-eng.cloud.databricks.com
-      root_path: /Users/${workspace.current_user.userName}/.bundle/${var.app_name}/${bundle.target}
-  gcp:
-    workspace:
-      host: https://416411475796958.8.gcp.databricks.com/
-      root_path: /Users/${workspace.current_user.userName}/.bundle/${var.app_name}/${bundle.target}
 
-  retail:
-    workspace:
-      host: https://adb-6447127900489650.10.azuredatabricks.net/
-      root_path: /Users/${workspace.current_user.userName}/.bundle/${var.app_name}/${bundle.target}
 
 
 

{dao_ai-0.0.4 → dao_ai-0.0.6}/notebooks/02_provision_vector_search.py

@@ -1,5 +1,7 @@
 # Databricks notebook source
 # MAGIC %pip install --quiet -r ../requirements.txt
+# MAGIC %pip uninstall -y databricks-connect pyspark pyspark-connect
+# MAGIC %pip install databricks-connect
 # MAGIC %restart_python
 
 # COMMAND ----------

{dao_ai-0.0.4 → dao_ai-0.0.6}/notebooks/03_generate_evaluation_data.py

@@ -1,5 +1,7 @@
 # Databricks notebook source
 # MAGIC %pip install --quiet -r ../requirements.txt
+# MAGIC %pip uninstall -y databricks-connect pyspark pyspark-connect
+# MAGIC %pip install databricks-connect
 # MAGIC %restart_python
 
 # COMMAND ----------

{dao_ai-0.0.4 → dao_ai-0.0.6}/notebooks/04_unity_catalog_tools.py

@@ -1,5 +1,7 @@
 # Databricks notebook source
 # MAGIC %pip install --quiet -r ../requirements.txt
+# MAGIC %pip uninstall -y databricks-connect pyspark pyspark-connect
+# MAGIC %pip install databricks-connect
 # MAGIC %restart_python
 
 # COMMAND ----------

{dao_ai-0.0.4 → dao_ai-0.0.6}/notebooks/05_agent_as_code_driver.py

@@ -1,5 +1,7 @@
 # Databricks notebook source
 # MAGIC %pip install --quiet -r ../requirements.txt
+# MAGIC %pip uninstall -y databricks-connect pyspark pyspark-connect
+# MAGIC %pip install databricks-connect
 # MAGIC %restart_python
 
 # COMMAND ----------

{dao_ai-0.0.4 → dao_ai-0.0.6}/notebooks/06_run_evaluation.py

@@ -1,6 +1,8 @@
 # Databricks notebook source
 # DBTITLE 1,Install Required Python Packages and Restart Kernel
 # MAGIC %pip install --quiet -r ../requirements.txt
+# MAGIC %pip uninstall -y databricks-connect pyspark pyspark-connect
+# MAGIC %pip install databricks-connect
 # MAGIC %restart_python
 
 # COMMAND ----------

{dao_ai-0.0.4 → dao_ai-0.0.6}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "dao-ai"
-version = "0.0.4"
+version = "0.0.6"
 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" }
@@ -42,7 +42,7 @@ classifiers = [
     "Topic :: System :: Distributed Computing",
 ]
 dependencies = [
-    "databricks-agents>=0.20.0",
+    "databricks-agents>=1.2.0",
     "databricks-langchain>=0.4.2",
     "databricks-sdk[openai]>=0.55.0",
     "databricks-vectorsearch>=0.56",

{dao_ai-0.0.4 → dao_ai-0.0.6}/requirements.txt

@@ -7,27 +7,22 @@ annotated-types==0.7.0
 anthropic==0.54.0
 anyio==4.9.0
 attrs==25.3.0
-azure-core==1.34.0
-azure-storage-blob==12.25.1
-azure-storage-file-datalake==12.20.0
 blinker==1.9.0
-boto3==1.38.41
-botocore==1.38.41
+boto3==1.40.1
+botocore==1.40.1
 cachetools==5.5.2
 certifi==2025.6.15
-cffi==1.17.1
 cfgv==3.4.0
 charset-normalizer==3.4.2
 click==8.2.1
 cloudpickle==3.1.1
 contourpy==1.3.2
-cryptography==45.0.4
 cycler==0.12.1
-databricks-agents==1.1.0
+databricks-agents==1.2.0
 databricks-ai-bridge==0.5.1
 databricks-connect==16.1.6
 databricks-langchain==0.5.1
-databricks-sdk==0.55.0
+databricks-sdk==0.61.0
 databricks-vectorsearch==0.56
 dataclasses-json==0.6.7
 deprecation==2.1.0
@@ -44,12 +39,7 @@ fonttools==4.58.4
 frozenlist==1.7.0
 gitdb==4.0.12
 gitpython==3.1.44
-google-api-core==2.25.1
 google-auth==2.40.3
-google-cloud-core==2.4.3
-google-cloud-storage==3.1.1
-google-crc32c==1.7.1
-google-resumable-media==2.7.2
 googleapis-common-protos==1.70.0
 grandalf==0.8
 graphene==3.4.3
@@ -67,7 +57,6 @@ identify==2.6.12
 idna==3.10
 importlib-metadata==8.7.0
 iniconfig==2.1.0
-isodate==0.7.2
 itsdangerous==2.2.0
 jinja2==3.1.6
 jiter==0.10.0
@@ -126,7 +115,6 @@ pluggy==1.6.0
 pre-commit==4.2.0
 primp==0.15.0
 propcache==0.3.2
-proto-plus==1.26.1
 protobuf==5.29.5
 psycopg==3.2.9
 psycopg-binary==3.2.9
@@ -135,7 +123,6 @@ py4j==0.10.9.7
 pyarrow==20.0.0
 pyasn1==0.6.1
 pyasn1-modules==0.4.2
-pycparser==2.22
 pydantic==2.11.7
 pydantic-core==2.33.2
 pydantic-settings==2.10.0
@@ -153,7 +140,7 @@ requests-toolbelt==1.0.0
 rich==14.0.0
 rsa==4.9.1
 ruff==0.12.0
-s3transfer==0.13.0
+s3transfer==0.13.1
 scikit-learn==1.7.0
 scipy==1.13.1
 setuptools==80.9.0
@@ -181,6 +168,7 @@ urllib3==2.5.0
 uvicorn==0.34.3
 virtualenv==20.31.2
 werkzeug==3.1.3
+whenever==0.7.3
 xxhash==3.5.0
 yarl==1.20.1
 zipp==3.23.0

{dao_ai-0.0.4 → dao_ai-0.0.6}/src/dao_ai/tools/core.py

@@ -200,8 +200,6 @@ def create_mcp_tools(
         else:
             logger.debug("Using existing authentication token")
 
-        logger.debug(f"Connection headers: {headers}")
-
         response = {
             "url": function.url,
             "transport": function.transport,
@@ -230,7 +228,9 @@ def create_mcp_tools(
         logger.debug(f"Retrieved {len(mcp_tools)} MCP tools")
     except Exception as e:
         logger.error(f"Failed to get tools from MCP server: {e}")
-        return []
+        raise RuntimeError(
+            f"Failed to list MCP tools for function '{function.name}' with transport '{function.transport}' and URL '{function.url}': {e}"
+        )
 
     # Create wrapper tools with fresh session per invocation
     def _create_tool_wrapper(mcp_tool: Tool) -> RunnableLike:

{dao_ai-0.0.4 → dao_ai-0.0.6}/src/quick_serve_restaurant/tools.py

@@ -4,7 +4,6 @@ from langchain_core.tools import tool as create_tool
 from loguru import logger
 
 from dao_ai.config import CompositeVariableModel, ToolModel, UnityCatalogFunctionModel
-from dao_ai.tools.core import BaseTool
 
 
 def insert_coffee_order_tool(
@@ -52,21 +51,32 @@ def insert_coffee_order_tool(
         if isinstance(unity_catalog_function, dict):
             unity_catalog_function = UnityCatalogFunctionModel(**unity_catalog_function)
 
-        unity_catalog_tool: BaseTool = next(
-            iter(unity_catalog_function.as_tools() or []), None
+        from databricks_langchain import DatabricksFunctionClient
+
+        client = DatabricksFunctionClient()
+        logger.debug(
+            f"Calling Unity Catalog function directly: {unity_catalog_function.full_name}"
         )
-        logger.debug(f"Invoking Unity Catalog tool: {unity_catalog_tool.name}")
-        result: str = unity_catalog_tool.invoke(
-            {
+
+        result = client.execute_function(
+            function_name=unity_catalog_function.full_name,
+            parameters={
                 "host": host.as_value(),
                 "client_id": client_id.as_value(),
                 "client_secret": client_secret.as_value(),
                 "coffee_name": coffee_name,
                 "size": size,
                 "session_id": session_id,
-            }
+            },
         )
-        logger.debug(f"Order result: {result}")
-        return result
+
+        # Handle the result from execute_function
+        if hasattr(result, "error") and result.error:
+            logger.error(f"Unity Catalog function error: {result.error}")
+            raise RuntimeError(f"Function execution failed: {result.error}")
+
+        result_value = result.value if hasattr(result, "value") else str(result)
+        logger.debug(f"Order result: {result_value}")
+        return result_value
 
     return insert_coffee_order