dao-ai 0.0.24__py3-none-any.whl → 0.0.26__py3-none-any.whl

dao_ai/tools/mcp.py

@@ -5,7 +5,6 @@ from databricks_mcp import DatabricksOAuthClientProvider
 from langchain_core.runnables.base import RunnableLike
 from langchain_core.tools import tool as create_tool
 from langchain_mcp_adapters.client import MultiServerMCPClient
-from langchain_mcp_adapters.tools import load_mcp_tools
 from loguru import logger
 from mcp import ClientSession
 from mcp.client.streamable_http import streamablehttp_client
@@ -49,8 +48,8 @@ def create_mcp_tools(
             mcp_url = f"{workspace_host}/api/2.0/mcp/external/{connection_name}"
             logger.debug(f"Constructed MCP URL from connection: {mcp_url}")
 
-        async def _get_tools_with_connection():
-            """Get tools using DatabricksOAuthClientProvider."""
+        async def _list_tools_with_connection():
+            """List available tools using DatabricksOAuthClientProvider."""
             workspace_client = function.connection.workspace_client
 
             async with streamablehttp_client(
@@ -59,20 +58,16 @@ def create_mcp_tools(
                 async with ClientSession(read_stream, write_stream) as session:
                     # Initialize and list tools
                     await session.initialize()
-                    tools = await load_mcp_tools(session)
-                    return tools
+                    return await session.list_tools()
 
         try:
-            langchain_tools = asyncio.run(_get_tools_with_connection())
-            logger.debug(
-                f"Retrieved {len(langchain_tools)} MCP tools via UC Connection"
+            mcp_tools: list[Tool] | ListToolsResult = asyncio.run(
+                _list_tools_with_connection()
             )
+            if isinstance(mcp_tools, ListToolsResult):
+                mcp_tools = mcp_tools.tools
 
-            # Wrap tools with human-in-the-loop if needed
-            wrapped_tools = [
-                as_human_in_the_loop(tool, function) for tool in langchain_tools
-            ]
-            return wrapped_tools
+            logger.debug(f"Retrieved {len(mcp_tools)} MCP tools via UC Connection")
 
         except Exception as e:
             logger.error(f"Failed to get tools from MCP server via UC Connection: {e}")
@@ -80,6 +75,39 @@ def create_mcp_tools(
                 f"Failed to list MCP tools for function '{function.name}' via UC Connection '{function.connection.name}': {e}"
             )
 
+        # Create wrapper tools with fresh session per invocation
+        def _create_tool_wrapper_with_connection(mcp_tool: Tool) -> RunnableLike:
+            @create_tool(
+                mcp_tool.name,
+                description=mcp_tool.description or f"MCP tool: {mcp_tool.name}",
+                args_schema=mcp_tool.inputSchema,
+            )
+            async def tool_wrapper(**kwargs):
+                """Execute MCP tool with fresh UC Connection session."""
+                logger.debug(
+                    f"Invoking MCP tool {mcp_tool.name} with fresh UC Connection session"
+                )
+                workspace_client = function.connection.workspace_client
+
+                try:
+                    async with streamablehttp_client(
+                        mcp_url, auth=DatabricksOAuthClientProvider(workspace_client)
+                    ) as (read_stream, write_stream, _):
+                        async with ClientSession(read_stream, write_stream) as session:
+                            await session.initialize()
+                            result = await session.call_tool(mcp_tool.name, kwargs)
+                            logger.debug(
+                                f"MCP tool {mcp_tool.name} completed successfully"
+                            )
+                            return result
+                except Exception as e:
+                    logger.error(f"MCP tool {mcp_tool.name} failed: {e}")
+                    raise
+
+            return as_human_in_the_loop(tool_wrapper, function)
+
+        return [_create_tool_wrapper_with_connection(tool) for tool in mcp_tools]
+
     else:
         # Use direct MCP connection with MultiServerMCPClient
         logger.debug("Using direct MCP connection with MultiServerMCPClient")

dao_ai/utils.py

@@ -3,7 +3,8 @@ import importlib.metadata
 import os
 import re
 import site
-from importlib.metadata import version
+from importlib.metadata import PackageNotFoundError, version
+from pathlib import Path
 from typing import Any, Callable, Sequence
 
 from loguru import logger
@@ -37,6 +38,59 @@ def normalize_name(name: str) -> str:
     return normalized.strip("_")
 
 
+def dao_ai_version() -> str:
+    """
+    Get the dao-ai package version, with fallback for source installations.
+
+    Tries to get the version from installed package metadata first. If the package
+    is not installed (e.g., running from source), falls back to reading from
+    pyproject.toml. Returns "dev" if neither method works.
+
+    Returns:
+        str: The version string, or "dev" if version cannot be determined
+    """
+    try:
+        # Try to get version from installed package metadata
+        return version("dao-ai")
+    except PackageNotFoundError:
+        # Package not installed, try reading from pyproject.toml
+        logger.debug(
+            "dao-ai package not installed, attempting to read version from pyproject.toml"
+        )
+        try:
+            import tomllib  # Python 3.11+
+        except ImportError:
+            try:
+                import tomli as tomllib  # Fallback for Python < 3.11
+            except ImportError:
+                logger.warning(
+                    "Cannot determine dao-ai version: package not installed and tomllib/tomli not available"
+                )
+                return "dev"
+
+        try:
+            # Find pyproject.toml relative to this file
+            project_root = Path(__file__).parents[2]
+            pyproject_path = project_root / "pyproject.toml"
+
+            if not pyproject_path.exists():
+                logger.warning(
+                    f"Cannot determine dao-ai version: pyproject.toml not found at {pyproject_path}"
+                )
+                return "dev"
+
+            with open(pyproject_path, "rb") as f:
+                pyproject_data = tomllib.load(f)
+                pkg_version = pyproject_data.get("project", {}).get("version", "dev")
+                logger.debug(
+                    f"Read version {pkg_version} from pyproject.toml at {pyproject_path}"
+                )
+                return pkg_version
+        except Exception as e:
+            logger.warning(f"Cannot determine dao-ai version from pyproject.toml: {e}")
+            return "dev"
+
+
 def get_installed_packages() -> dict[str, str]:
     """Get all installed packages with versions"""
 
@@ -65,6 +119,7 @@ def get_installed_packages() -> dict[str, str]:
         f"psycopg[binary,pool]=={version('psycopg')}",
         f"pydantic=={version('pydantic')}",
         f"pyyaml=={version('pyyaml')}",
+        f"tomli=={version('tomli')}",
         f"unitycatalog-ai[databricks]=={version('unitycatalog-ai')}",
         f"unitycatalog-langchain[databricks]=={version('unitycatalog-langchain')}",
     ]

{dao_ai-0.0.24.dist-info → dao_ai-0.0.26.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dao-ai
-Version: 0.0.24
+Version: 0.0.26
 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
@@ -29,6 +29,7 @@ Requires-Dist: databricks-langchain>=0.8.1
 Requires-Dist: databricks-mcp>=0.3.0
 Requires-Dist: databricks-sdk[openai]>=0.67.0
 Requires-Dist: duckduckgo-search>=8.0.2
+Requires-Dist: gepa>=0.0.17
 Requires-Dist: grandalf>=0.8
 Requires-Dist: langchain-mcp-adapters>=0.1.10
 Requires-Dist: langchain-tavily>=0.2.11
@@ -40,7 +41,7 @@ Requires-Dist: langgraph>=0.6.10
 Requires-Dist: langmem>=0.0.29
 Requires-Dist: loguru>=0.7.3
 Requires-Dist: mcp>=1.17.0
-Requires-Dist: mlflow>=3.4.0
+Requires-Dist: mlflow>=3.5.0
 Requires-Dist: nest-asyncio>=1.6.0
 Requires-Dist: openevals>=0.0.19
 Requires-Dist: openpyxl>=3.1.5
@@ -51,6 +52,7 @@ Requires-Dist: pyyaml>=6.0.2
 Requires-Dist: rich>=14.0.0
 Requires-Dist: scipy<=1.15
 Requires-Dist: sqlparse>=0.5.3
+Requires-Dist: tomli>=2.3.0
 Requires-Dist: unitycatalog-ai[databricks]>=0.3.0
 Provides-Extra: databricks
 Requires-Dist: databricks-connect>=15.0.0; extra == 'databricks'

{dao_ai-0.0.24.dist-info → dao_ai-0.0.26.dist-info}/RECORD

@@ -2,17 +2,17 @@ dao_ai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 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=j9SAdf7UHSoS2pLos-oypJNDPS48A2rRq55OEW1wsMI,56755
-dao_ai/graph.py,sha256=APYc2y3cig4P52X4sOHSFSZNK8j5EtEPJLFwWeJ3KQQ,7956
+dao_ai/cli.py,sha256=gq-nsapWxDA1M6Jua3vajBvIwf0Oa6YLcB58lEtMKUo,22503
+dao_ai/config.py,sha256=MdX7zGOXf9Fg03lo1vA7G9yoiIzYmJ6kqTJH4aiQ3VQ,65068
+dao_ai/graph.py,sha256=RrcYfdyj9lYXUZyx4Ax4VIlCUbOWviwr1t2qGDM69LQ,8318
 dao_ai/guardrails.py,sha256=4TKArDONRy8RwHzOT1plZ1rhy3x9GF_aeGpPCRl6wYA,4016
 dao_ai/messages.py,sha256=xl_3-WcFqZKCFCiov8sZOPljTdM3gX3fCHhxq-xFg2U,7005
 dao_ai/models.py,sha256=8r8GIG3EGxtVyWsRNI56lVaBjiNrPkzh4HdwMZRq8iw,31689
-dao_ai/nodes.py,sha256=SSuFNTXOdFaKg_aX-yUkQO7fM9wvNGu14lPXKDapU1U,8461
+dao_ai/nodes.py,sha256=iQ_5vL6mt1UcRnhwgz-l1D8Ww4CMQrSMVnP_Lu7fFjU,8781
 dao_ai/prompts.py,sha256=7Hcstmv514P0s9s-TVoIlbkDV2XXOphGCW6gcPeyUYE,1628
 dao_ai/state.py,sha256=_lF9krAYYjvFDMUwZzVKOn0ZnXKcOrbjWKdre0C5B54,1137
 dao_ai/types.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-dao_ai/utils.py,sha256=yXgqHrYdO5qDxgxUs2G5XJeLFgwg8D0BIJvbFkqSbhs,4519
+dao_ai/utils.py,sha256=xfBo9-6k9ss1c7QlC26QDCwS2sY3mgUYgWzTTV6443M,6662
 dao_ai/vector_search.py,sha256=jlaFS_iizJ55wblgzZmswMM3UOL-qOp2BGJc0JqXYSg,2839
 dao_ai/hooks/__init__.py,sha256=LlHGIuiZt6vGW8K5AQo1XJEkBP5vDVtMhq0IdjcLrD4,417
 dao_ai/hooks/core.py,sha256=ZShHctUSoauhBgdf1cecy9-D7J6-sGn-pKjuRMumW5U,6663
@@ -22,20 +22,20 @@ dao_ai/memory/core.py,sha256=DnEjQO3S7hXr3CDDd7C2eE7fQUmcCS_8q9BXEgjPH3U,4271
 dao_ai/memory/postgres.py,sha256=vvI3osjx1EoU5GBA6SCUstTBKillcmLl12hVgDMjfJY,15346
 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=1BPYQxi4-Z4I1ygZYlKV8ycdxZTtWNXplySToayHCEI,43096
+dao_ai/providers/databricks.py,sha256=jfQEEgr9mEIttde1182wPpYm3kHq4sRwXoeMNfkpEmA,66512
 dao_ai/tools/__init__.py,sha256=G5-5Yi6zpQOH53b5IzLdtsC6g0Ep6leI5GxgxOmgw7Q,1203
 dao_ai/tools/agent.py,sha256=WbQnyziiT12TLMrA7xK0VuOU029tdmUBXbUl-R1VZ0Q,1886
 dao_ai/tools/core.py,sha256=Kei33S8vrmvPOAyrFNekaWmV2jqZ-IPS1QDSvU7RZF0,1984
 dao_ai/tools/genie.py,sha256=8HSOCzSg6PlBzBYXMmNfUnl-LO03p3Ki3fxLPm_dhPg,15051
 dao_ai/tools/human_in_the_loop.py,sha256=yk35MO9eNETnYFH-sqlgR-G24TrEgXpJlnZUustsLkI,3681
-dao_ai/tools/mcp.py,sha256=CYv59yn-LIY11atUgNtN2W6vR7C6Qyo7-rvPcVJnXVk,7461
+dao_ai/tools/mcp.py,sha256=gTY2xTInC5sy9ckZubwp1QIyZ6_qaUXGeJM_NlmPnGE,8892
 dao_ai/tools/python.py,sha256=XcQiTMshZyLUTVR5peB3vqsoUoAAy8gol9_pcrhddfI,1831
 dao_ai/tools/slack.py,sha256=SCvyVcD9Pv_XXPXePE_fSU1Pd8VLTEkKDLvoGTZWy2Y,4775
 dao_ai/tools/time.py,sha256=Y-23qdnNHzwjvnfkWvYsE7PoWS1hfeKy44tA7sCnNac,8759
 dao_ai/tools/unity_catalog.py,sha256=uX_h52BuBAr4c9UeqSMI7DNz3BPRLeai5tBVW4sJqRI,13113
 dao_ai/tools/vector_search.py,sha256=EDYQs51zIPaAP0ma1D81wJT77GQ-v-cjb2XrFVWfWdg,2621
-dao_ai-0.0.24.dist-info/METADATA,sha256=69qYBqdZg8tZF1ni90RBrSG9OAmE4jXdf5lec6U0TL8,42639
-dao_ai-0.0.24.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-dao_ai-0.0.24.dist-info/entry_points.txt,sha256=Xa-UFyc6gWGwMqMJOt06ZOog2vAfygV_DSwg1AiP46g,43
-dao_ai-0.0.24.dist-info/licenses/LICENSE,sha256=YZt3W32LtPYruuvHE9lGk2bw6ZPMMJD8yLrjgHybyz4,1069
-dao_ai-0.0.24.dist-info/RECORD,,
+dao_ai-0.0.26.dist-info/METADATA,sha256=V5Bvxs1b3FJR-rIjfJHqrNyv9r46rZMHeY9ze8xkMX0,42695
+dao_ai-0.0.26.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+dao_ai-0.0.26.dist-info/entry_points.txt,sha256=Xa-UFyc6gWGwMqMJOt06ZOog2vAfygV_DSwg1AiP46g,43
+dao_ai-0.0.26.dist-info/licenses/LICENSE,sha256=YZt3W32LtPYruuvHE9lGk2bw6ZPMMJD8yLrjgHybyz4,1069
+dao_ai-0.0.26.dist-info/RECORD,,