datarobot-genai 0.2.31__py3-none-any.whl

datarobot_genai/nat/datarobot_mcp_client.py

@@ -0,0 +1,266 @@
+# Copyright 2025 DataRobot, Inc. and its affiliates.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from __future__ import annotations
+
+import logging
+from datetime import timedelta
+from typing import TYPE_CHECKING
+from typing import Literal
+
+from nat.cli.register_workflow import register_function_group
+from nat.data_models.component_ref import AuthenticationRef
+from nat.plugins.mcp.client_base import AuthAdapter
+from nat.plugins.mcp.client_base import MCPStreamableHTTPClient
+from nat.plugins.mcp.client_config import MCPServerConfig
+from nat.plugins.mcp.client_impl import MCPClientConfig
+from pydantic import Field
+from pydantic import HttpUrl
+
+if TYPE_CHECKING:
+    import httpx
+    from nat.authentication.interfaces import AuthProviderBase
+    from nat.builder.builder import Builder
+    from nat.plugins.mcp.client_impl import MCPFunctionGroup
+
+logger = logging.getLogger(__name__)
+
+
+def _default_transport() -> Literal["streamable-http", "sse", "stdio"]:
+    from datarobot_genai.core.mcp.common import MCPConfig  # noqa: PLC0415
+
+    server_config = MCPConfig().server_config
+    return server_config["transport"] if server_config else "stdio"
+
+
+def _default_url() -> HttpUrl | None:
+    from datarobot_genai.core.mcp.common import MCPConfig  # noqa: PLC0415
+
+    server_config = MCPConfig().server_config
+    return server_config["url"] if server_config else None
+
+
+def _default_auth_provider() -> str | AuthenticationRef | None:
+    from datarobot_genai.core.mcp.common import MCPConfig  # noqa: PLC0415
+
+    server_config = MCPConfig().server_config
+    return "datarobot_mcp_auth" if server_config else None
+
+
+def _default_command() -> str | None:
+    from datarobot_genai.core.mcp.common import MCPConfig  # noqa: PLC0415
+
+    server_config = MCPConfig().server_config
+    return None if server_config else "docker"
+
+
+class DataRobotMCPServerConfig(MCPServerConfig):
+    transport: Literal["streamable-http", "sse", "stdio"] = Field(
+        default_factory=_default_transport,
+        description="Transport type to connect to the MCP server (sse or streamable-http)",
+    )
+    url: HttpUrl | None = Field(
+        default_factory=_default_url,
+        description="URL of the MCP server (for sse or streamable-http transport)",
+    )
+    # Authentication configuration
+    auth_provider: str | AuthenticationRef | None = Field(
+        default_factory=_default_auth_provider,
+        description="Reference to authentication provider",
+    )
+    command: str | None = Field(
+        default_factory=_default_command,
+        description="Command to run for stdio transport (e.g. 'python' or 'docker')",
+    )
+
+
+class DataRobotMCPClientConfig(MCPClientConfig, name="datarobot_mcp_client"):  # type: ignore[call-arg]
+    server: DataRobotMCPServerConfig = Field(
+        default_factory=DataRobotMCPServerConfig,
+        description="DataRobot MCP Server configuration",
+    )
+
+
+class DataRobotAuthAdapter(AuthAdapter):
+    async def _get_auth_headers(
+        self, request: httpx.Request | None = None, response: httpx.Response | None = None
+    ) -> dict[str, str]:
+        """Get authentication headers from the NAT auth provider."""
+        try:
+            # Use the user_id passed to this AuthAdapter instance
+            auth_result = await self.auth_provider.authenticate(
+                user_id=self.user_id, response=response
+            )
+            as_kwargs = auth_result.as_requests_kwargs()
+            return as_kwargs["headers"]
+        except Exception as e:
+            logger.warning("Failed to get auth token: %s", e)
+            return {}
+
+
+class DataRobotMCPStreamableHTTPClient(MCPStreamableHTTPClient):
+    def __init__(
+        self,
+        url: str,
+        auth_provider: AuthProviderBase | None = None,
+        user_id: str | None = None,
+        tool_call_timeout: timedelta = timedelta(seconds=60),
+        auth_flow_timeout: timedelta = timedelta(seconds=300),
+        reconnect_enabled: bool = True,
+        reconnect_max_attempts: int = 2,
+        reconnect_initial_backoff: float = 0.5,
+        reconnect_max_backoff: float = 50.0,
+    ):
+        super().__init__(
+            url=url,
+            auth_provider=auth_provider,
+            user_id=user_id,
+            tool_call_timeout=tool_call_timeout,
+            auth_flow_timeout=auth_flow_timeout,
+            reconnect_enabled=reconnect_enabled,
+            reconnect_max_attempts=reconnect_max_attempts,
+            reconnect_initial_backoff=reconnect_initial_backoff,
+            reconnect_max_backoff=reconnect_max_backoff,
+        )
+        effective_user_id = user_id or (
+            auth_provider.config.default_user_id if auth_provider else None
+        )
+        self._httpx_auth = (
+            DataRobotAuthAdapter(auth_provider, effective_user_id) if auth_provider else None
+        )
+
+
+@register_function_group(config_type=DataRobotMCPClientConfig)
+async def datarobot_mcp_client_function_group(
+    config: DataRobotMCPClientConfig, _builder: Builder
+) -> MCPFunctionGroup:
+    """
+    Connect to an MCP server and expose tools as a function group.
+
+    Args:
+        config: The configuration for the MCP client
+        _builder: The builder
+    Returns:
+        The function group
+    """
+    from nat.plugins.mcp.client_base import MCPSSEClient  # noqa: PLC0415
+    from nat.plugins.mcp.client_base import MCPStdioClient  # noqa: PLC0415
+    from nat.plugins.mcp.client_impl import MCPFunctionGroup  # noqa: PLC0415
+    from nat.plugins.mcp.client_impl import mcp_apply_tool_alias_and_description  # noqa: PLC0415
+    from nat.plugins.mcp.client_impl import mcp_session_tool_function  # noqa: PLC0415
+
+    # Resolve auth provider if specified
+    auth_provider = None
+    if config.server.auth_provider:
+        auth_provider = await _builder.get_auth_provider(config.server.auth_provider)
+
+    # Build the appropriate client
+    if config.server.transport == "stdio":
+        if not config.server.command:
+            raise ValueError("command is required for stdio transport")
+        client = MCPStdioClient(
+            config.server.command,
+            config.server.args,
+            config.server.env,
+            tool_call_timeout=config.tool_call_timeout,
+            auth_flow_timeout=config.auth_flow_timeout,
+            reconnect_enabled=config.reconnect_enabled,
+            reconnect_max_attempts=config.reconnect_max_attempts,
+            reconnect_initial_backoff=config.reconnect_initial_backoff,
+            reconnect_max_backoff=config.reconnect_max_backoff,
+        )
+    elif config.server.transport == "sse":
+        client = MCPSSEClient(
+            str(config.server.url),
+            tool_call_timeout=config.tool_call_timeout,
+            auth_flow_timeout=config.auth_flow_timeout,
+            reconnect_enabled=config.reconnect_enabled,
+            reconnect_max_attempts=config.reconnect_max_attempts,
+            reconnect_initial_backoff=config.reconnect_initial_backoff,
+            reconnect_max_backoff=config.reconnect_max_backoff,
+        )
+    elif config.server.transport == "streamable-http":
+        # Use default_user_id for the base client
+        base_user_id = auth_provider.config.default_user_id if auth_provider else None
+        client = DataRobotMCPStreamableHTTPClient(
+            str(config.server.url),
+            auth_provider=auth_provider,
+            user_id=base_user_id,
+            tool_call_timeout=config.tool_call_timeout,
+            auth_flow_timeout=config.auth_flow_timeout,
+            reconnect_enabled=config.reconnect_enabled,
+            reconnect_max_attempts=config.reconnect_max_attempts,
+            reconnect_initial_backoff=config.reconnect_initial_backoff,
+            reconnect_max_backoff=config.reconnect_max_backoff,
+        )
+    else:
+        raise ValueError(f"Unsupported transport: {config.server.transport}")
+
+    logger.info("Configured to use MCP server at %s", client.server_name)
+
+    # Create the MCP function group
+    group = MCPFunctionGroup(config=config)
+
+    # Store shared components for session client creation
+    group._shared_auth_provider = auth_provider
+    group._client_config = config
+
+    async with client:
+        # Expose the live MCP client on the function group instance so other components
+        # (e.g., HTTP endpoints) can reuse the already-established session instead of creating a
+        # new client per request.
+        group.mcp_client = client
+        group.mcp_client_server_name = client.server_name
+        group.mcp_client_transport = client.transport
+
+        all_tools = await client.get_tools()
+        tool_overrides = mcp_apply_tool_alias_and_description(all_tools, config.tool_overrides)
+
+        # Add each tool as a function to the group
+        for tool_name, tool in all_tools.items():
+            # Get override if it exists
+            override = tool_overrides.get(tool_name)
+
+            # Use override values or defaults
+            function_name = override.alias if override and override.alias else tool_name
+            description = (
+                override.description if override and override.description else tool.description
+            )
+
+            # Create the tool function according to configuration
+            tool_fn = mcp_session_tool_function(tool, group)
+
+            # Normalize optional typing for linter/type-checker compatibility
+            single_fn = tool_fn.single_fn
+            if single_fn is None:
+                # Should not happen because FunctionInfo always sets a single_fn
+                logger.warning("Skipping tool %s because single_fn is None", function_name)
+                continue
+
+            input_schema = tool_fn.input_schema
+            # Convert NoneType sentinel to None for FunctionGroup.add_function signature
+            if input_schema is type(None):  # noqa: E721
+                input_schema = None
+
+            # Add to group
+            logger.info("Adding tool %s to group", function_name)
+            group.add_function(
+                name=function_name,
+                description=description,
+                fn=single_fn,
+                input_schema=input_schema,
+                converters=tool_fn.converters,
+            )
+
+        yield group

datarobot_genai/nat/helpers.py

@@ -0,0 +1,87 @@
+# Copyright 2025 DataRobot, Inc. and its affiliates.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from collections.abc import AsyncGenerator
+from contextlib import asynccontextmanager
+
+from nat.builder.workflow import Workflow
+from nat.builder.workflow_builder import WorkflowBuilder
+from nat.data_models.config import Config
+from nat.runtime.loader import PluginTypes
+from nat.runtime.loader import discover_and_register_plugins
+from nat.runtime.session import SessionManager
+from nat.utils.data_models.schema_validator import validate_schema
+from nat.utils.io.yaml_tools import yaml_load
+from nat.utils.type_utils import StrPath
+
+
+def load_config(config_file: StrPath, headers: dict[str, str] | None = None) -> Config:
+    """
+    Load a NAT configuration file with injected headers. It ensures that all plugins are
+    loaded and then validates the configuration file against the Config schema.
+
+    Parameters
+    ----------
+    config_file : StrPath
+        The path to the configuration file
+
+    Returns
+    -------
+    Config
+        The validated Config object
+    """
+    # Ensure all of the plugins are loaded
+    discover_and_register_plugins(PluginTypes.CONFIG_OBJECT)
+
+    config_yaml = yaml_load(config_file)
+
+    add_headers_to_datarobot_mcp_auth(config_yaml, headers)
+
+    # Validate configuration adheres to NAT schemas
+    validated_nat_config = validate_schema(config_yaml, Config)
+
+    return validated_nat_config
+
+
+def add_headers_to_datarobot_mcp_auth(config_yaml: dict, headers: dict[str, str] | None) -> None:
+    if headers:
+        if authentication := config_yaml.get("authentication"):
+            for auth_name in authentication:
+                auth_config = authentication[auth_name]
+                if auth_config.get("_type") == "datarobot_mcp_auth":
+                    auth_config["headers"] = headers
+
+
+@asynccontextmanager
+async def load_workflow(
+    config_file: StrPath, max_concurrency: int = -1, headers: dict[str, str] | None = None
+) -> AsyncGenerator[Workflow, None]:
+    """
+    Load the NAT configuration file and create a Runner object. This is the primary entry point for
+    running NAT workflows with injected headers.
+
+    Parameters
+    ----------
+    config_file : StrPath
+        The path to the configuration file
+    max_concurrency : int, optional
+        The maximum number of parallel workflow invocations to support. Specifying 0 or -1 will
+        allow an unlimited count, by default -1
+    """
+    # Load the config object
+    config = load_config(config_file, headers=headers)
+
+    # Must yield the workflow function otherwise it cleans up
+    async with WorkflowBuilder.from_config(config=config) as workflow:
+        yield SessionManager(await workflow.build(), max_concurrency=max_concurrency)

datarobot_genai-0.2.31.dist-info/METADATA

@@ -0,0 +1,145 @@
+Metadata-Version: 2.4
+Name: datarobot-genai
+Version: 0.2.31
+Summary: Generic helpers for GenAI
+Project-URL: Homepage, https://github.com/datarobot-oss/datarobot-genai
+Author: DataRobot, Inc.
+License: Apache-2.0
+License-File: AUTHORS
+License-File: LICENSE
+Requires-Python: <3.13,>=3.10
+Requires-Dist: ag-ui-protocol<0.2.0,>=0.1.9
+Requires-Dist: datarobot-drum<2.0.0,>=1.17.5
+Requires-Dist: datarobot-predict<2.0.0,>=1.13.2
+Requires-Dist: datarobot<4.0.0,>=3.10.0
+Requires-Dist: openai<2.0.0,>=1.76.2
+Requires-Dist: opentelemetry-instrumentation-aiohttp-client<1.0.0,>=0.43b0
+Requires-Dist: opentelemetry-instrumentation-httpx<1.0.0,>=0.43b0
+Requires-Dist: opentelemetry-instrumentation-openai<1.0.0,>=0.40.5
+Requires-Dist: opentelemetry-instrumentation-requests<1.0.0,>=0.43b0
+Requires-Dist: opentelemetry-instrumentation-threading<1.0.0,>=0.43b0
+Requires-Dist: pandas<3.0.0,>=2.2.3
+Requires-Dist: pyarrow==20.0.0
+Requires-Dist: pyjwt<3.0.0,>=2.10.1
+Requires-Dist: pypdf<7.0.0,>=6.1.3
+Requires-Dist: ragas<0.4.0,>=0.3.8
+Requires-Dist: requests<3.0.0,>=2.32.4
+Provides-Extra: crewai
+Requires-Dist: anthropic<1.0.0,~=0.71.0; extra == 'crewai'
+Requires-Dist: azure-ai-inference<2.0.0,>=1.0.0b9; extra == 'crewai'
+Requires-Dist: crewai-tools[mcp]<0.77.0,>=0.69.0; extra == 'crewai'
+Requires-Dist: crewai>=1.1.0; extra == 'crewai'
+Requires-Dist: opentelemetry-instrumentation-crewai<1.0.0,>=0.40.5; extra == 'crewai'
+Requires-Dist: pybase64<2.0.0,>=1.4.2; extra == 'crewai'
+Provides-Extra: drmcp
+Requires-Dist: aiohttp-retry<3.0.0,>=2.8.3; extra == 'drmcp'
+Requires-Dist: aiohttp<4.0.0,>=3.9.0; extra == 'drmcp'
+Requires-Dist: aiosignal<2.0.0,>=1.3.1; extra == 'drmcp'
+Requires-Dist: boto3<2.0.0,>=1.34.0; extra == 'drmcp'
+Requires-Dist: datarobot-asgi-middleware<1.0.0,>=0.2.0; extra == 'drmcp'
+Requires-Dist: fastmcp<3.0.0,>=2.13.0.2; extra == 'drmcp'
+Requires-Dist: httpx<1.0.0,>=0.28.1; extra == 'drmcp'
+Requires-Dist: opentelemetry-api<2.0.0,>=1.22.0; extra == 'drmcp'
+Requires-Dist: opentelemetry-exporter-otlp-proto-http<2.0.0,>=1.22.0; extra == 'drmcp'
+Requires-Dist: opentelemetry-exporter-otlp<2.0.0,>=1.22.0; extra == 'drmcp'
+Requires-Dist: opentelemetry-sdk<2.0.0,>=1.22.0; extra == 'drmcp'
+Requires-Dist: pydantic-settings<3.0.0,>=2.1.0; extra == 'drmcp'
+Requires-Dist: pydantic<3.0.0,>=2.6.1; extra == 'drmcp'
+Requires-Dist: python-dotenv<2.0.0,>=1.1.0; extra == 'drmcp'
+Provides-Extra: langgraph
+Requires-Dist: langchain-mcp-adapters<0.2.0,>=0.1.12; extra == 'langgraph'
+Requires-Dist: langgraph-prebuilt<0.7.0,>=0.2.3; extra == 'langgraph'
+Requires-Dist: langgraph<0.7.0,>=0.4.10; extra == 'langgraph'
+Requires-Dist: opentelemetry-instrumentation-langchain<1.0.0,>=0.40.5; extra == 'langgraph'
+Provides-Extra: llamaindex
+Requires-Dist: llama-index-core<0.14.0,>=0.13.6; extra == 'llamaindex'
+Requires-Dist: llama-index-llms-langchain<0.8.0,>=0.6.1; extra == 'llamaindex'
+Requires-Dist: llama-index-llms-litellm<0.7.0,>=0.4.1; extra == 'llamaindex'
+Requires-Dist: llama-index-llms-openai<0.6.0,>=0.3.38; extra == 'llamaindex'
+Requires-Dist: llama-index-tools-mcp<0.5.0,>=0.1.0; extra == 'llamaindex'
+Requires-Dist: llama-index<0.14.0,>=0.13.6; extra == 'llamaindex'
+Requires-Dist: opentelemetry-instrumentation-llamaindex<1.0.0,>=0.40.5; extra == 'llamaindex'
+Requires-Dist: pypdf<7.0.0,>=6.0.0; extra == 'llamaindex'
+Provides-Extra: nat
+Requires-Dist: anyio==4.11.0; extra == 'nat'
+Requires-Dist: crewai>=1.1.0; (python_version >= '3.11') and extra == 'nat'
+Requires-Dist: llama-index-llms-litellm<0.7.0,>=0.4.1; extra == 'nat'
+Requires-Dist: nvidia-nat-langchain==1.3.0; (python_version >= '3.11') and extra == 'nat'
+Requires-Dist: nvidia-nat-mcp==1.3.0; (python_version >= '3.11') and extra == 'nat'
+Requires-Dist: nvidia-nat-opentelemetry==1.3.0; (python_version >= '3.11') and extra == 'nat'
+Requires-Dist: nvidia-nat==1.3.0; (python_version >= '3.11') and extra == 'nat'
+Requires-Dist: opentelemetry-instrumentation-crewai<1.0.0,>=0.40.5; extra == 'nat'
+Requires-Dist: opentelemetry-instrumentation-langchain<1.0.0,>=0.40.5; extra == 'nat'
+Requires-Dist: opentelemetry-instrumentation-llamaindex<1.0.0,>=0.40.5; extra == 'nat'
+Provides-Extra: pydanticai
+Requires-Dist: logfire<5.0.0,>=4.6.0; extra == 'pydanticai'
+Requires-Dist: pydantic-ai<1.9.0,>=1.0.5; extra == 'pydanticai'
+Description-Content-Type: text/markdown
+
+<p align="center">
+  <a href="https://github.com/datarobot-oss/datarobot-genai">
+    <img src="docs/img/datarobot_logo.avif" width="600px" alt="DataRobot Logo"/>
+  </a>
+</p>
+<h3 align="center">DataRobot GenAI Library</h3>
+
+<p align="center">
+  <a href="https://www.datarobot.com/">Homepage</a>
+  ·
+  <a href="https://pypi.org/project/datarobot-genai/">PyPI</a>
+  ·
+  <a href="https://docs.datarobot.com/en/docs/get-started/troubleshooting/general-help.html">Support</a>
+</p>
+
+<p align="center">
+  <a href="/LICENSE">
+    <img src="https://img.shields.io/github/license/datarobot-oss/datarobot-genai" alt="License">
+  </a>
+  <a href="https://pypi.org/project/datarobot-genai/">
+    <img src="https://img.shields.io/pypi/v/datarobot-genai" alt="PyPI version">
+  </a>
+</p>
+
+
+## Features
+- Utilities for common GenAI workflows
+- Integrations: CrewAI, LangGraph, LlamaIndex, NAT, MCP
+
+## Installation
+- Requires Python 3.10–3.12.
+- Install:
+```bash
+pip install --upgrade pip
+pip install "datarobot-genai"
+```
+- Optional extras:
+```bash
+pip install "datarobot-genai[crewai]"
+pip install "datarobot-genai[langgraph]"
+pip install "datarobot-genai[llamaindex]"
+# Multiple extras
+pip install "datarobot-genai[crewai,langgraph,llamaindex]"
+```
+  Available extras include: `crewai`, `langgraph`, `llamaindex`, `nat`, `drmcp`, `pydanticai`.
+
+## Development
+Prerequisites: Python 3.10–3.12, uv, Task CLI, pre-commit.
+```bash
+uv sync --all-extras --dev
+pre-commit install
+task test
+```
+
+## Publishing
+- PRs (same-repo): dev builds are auto-published to TestPyPI (`.devN`).
+- Merge to `main`: tags `v{version}` and publishes to PyPI automatically.
+- Pushing a `v*` tag also triggers PyPI publish.
+- Optional: `task release:tag-and-push` creates and pushes `v{version}` locally.
+
+## Links
+- Home: https://github.com/datarobot-oss/datarobot-genai
+- PyPI: https://pypi.org/project/datarobot-genai/
+- TestPyPI: https://test.pypi.org/project/datarobot-genai/
+
+## License
+Apache-2.0

datarobot_genai-0.2.31.dist-info/RECORD

@@ -0,0 +1,125 @@
+datarobot_genai/__init__.py,sha256=QwWCADIZigzGvew8tGT9rAjgQjchSMjaZtFPbi3Bt2s,527
+datarobot_genai/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+datarobot_genai/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+datarobot_genai/core/custom_model.py,sha256=9rSJhKZ1Fo8BpVVLnRMUj3QV2AYUr5ADHEs0aW6s2tY,5813
+datarobot_genai/core/telemetry_agent.py,sha256=CxvoyResG3jXQ7ucU26NXCzWjWQyua-5qSYvVxpZJQg,5343
+datarobot_genai/core/agents/__init__.py,sha256=mTG_QVV5aoOWOgVA3KEq7KQLJllyxtG2ZQoq9wiUNYo,1542
+datarobot_genai/core/agents/base.py,sha256=63JRi0r7L94p6QHpor4JBFu_g5nxaQkqb_I4dsfrDc0,6722
+datarobot_genai/core/chat/__init__.py,sha256=kAxp4Dc-6HIM_cdBl-3IxwzJQr13UYYQ2Zc-hMwz2F8,638
+datarobot_genai/core/chat/auth.py,sha256=6qITKTHFtESsBc2NsA6cvJf78pPUrcA5XV3Vxlhb5us,5457
+datarobot_genai/core/chat/client.py,sha256=fk8MebXa8_R33VK0_DrXCS0Fgw3wFvPEvsuubC27c3s,6639
+datarobot_genai/core/chat/responses.py,sha256=vGxTA433f2AxGVlijV6O4EghyNPJCDmEqpAK2oWnsIs,10583
+datarobot_genai/core/cli/__init__.py,sha256=B93Yb6VavoZpatrh8ltCL6YglIfR5FHgytXbO9UuxBw,733
+datarobot_genai/core/cli/agent_environment.py,sha256=BJzQoiDvZF5gW4mFE71U0yeg-l72C--kxiE-fv6W194,1662
+datarobot_genai/core/cli/agent_kernel.py,sha256=3XX58DQ6XPpWB_tn5m3iGb3XTfhZf5X3W9tc6ADieU4,7790
+datarobot_genai/core/mcp/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+datarobot_genai/core/mcp/common.py,sha256=Y8SjuquUODKEfI7T9X-QuTMKdIlpCWFI1b3xs6tmHFA,7812
+datarobot_genai/core/utils/__init__.py,sha256=VxtRUz6iwb04eFQQy0zqTNXLAkYpPXcJxVoKV0nOdXk,59
+datarobot_genai/core/utils/auth.py,sha256=LpSoHdPD2siskYwG8q4f9cike4VQdgFWJpuJrpiszXU,8674
+datarobot_genai/core/utils/urls.py,sha256=tk0t13duDEPcmwz2OnS4vwEdatruiuX8lnxMMhSaJik,2289
+datarobot_genai/crewai/__init__.py,sha256=MtFnHA3EtmgiK_GjwUGPgQQ6G1MCEzz1SDBwQi9lE8M,706
+datarobot_genai/crewai/agent.py,sha256=vp8_2LExpeLls7Fpzo0R6ud5I6Ryfu3n3oVTN4Yyi6A,1417
+datarobot_genai/crewai/base.py,sha256=JLljEN7sj8zaH8OamYoevFBZzza5BjZ4f0CGHRp2jUU,6447
+datarobot_genai/crewai/events.py,sha256=K67bO1zwPrxmppz2wh8dFGNbVebyWGXAMD7oodFE2sQ,5462
+datarobot_genai/crewai/mcp.py,sha256=AJTrs-8KdiRSjRECfBT1lJOsszWMoFoN9NIa1p5_wsM,2115
+datarobot_genai/drmcp/__init__.py,sha256=WyKW7p77kS63EsmuW8cnZpqgcxCzIv1TmRPo2qo-Z8A,2988
+datarobot_genai/drmcp/server.py,sha256=KE4kjS5f9bfdYftG14HBHrfvxDfCD4pwCXePfvl1OvU,724
+datarobot_genai/drmcp/core/__init__.py,sha256=y4yapzp3KnFMzSR6HlNDS4uSuyNT7I1iPBvaCLsS0sU,577
+datarobot_genai/drmcp/core/auth.py,sha256=E-5wrGbBFEBlD5377g6Exddrc7HsazamwX8tWr2RLXY,5815
+datarobot_genai/drmcp/core/clients.py,sha256=y-yG8617LbmiZ_L7FWfMrk4WjIekyr76u_Q80aLqGpI,5524
+datarobot_genai/drmcp/core/config.py,sha256=SWLhVKoqI4vmA-04TFKpKm1_G2yEMEN1e_8cv8d_XRM,13774
+datarobot_genai/drmcp/core/config_utils.py,sha256=U-aieWw7MyP03cGDFIp97JH99ZUfr3vD9uuTzBzxn7w,6428
+datarobot_genai/drmcp/core/constants.py,sha256=lUwoW_PTrbaBGqRJifKqCn3EoFacoEgdO-CpoFVrUoU,739
+datarobot_genai/drmcp/core/credentials.py,sha256=PYEUDNMVw1BoMzZKLkPVTypNkVevEPtmk3scKnE-zYg,6706
+datarobot_genai/drmcp/core/dr_mcp_server.py,sha256=czcjbwhZAeW9EtG_Bys0GARPOuQulstkiU7FG48Q9bg,14118
+datarobot_genai/drmcp/core/dr_mcp_server_logo.py,sha256=hib-nfR1SNTW6CnpFsFCkL9H_OMwa4YYyinV7VNOuLk,4708
+datarobot_genai/drmcp/core/exceptions.py,sha256=eqsGI-lxybgvWL5w4BFhbm3XzH1eU5tetwjnhJxelpc,905
+datarobot_genai/drmcp/core/logging.py,sha256=Y_hig4eBWiXGaVV7B_3wBcaYVRNH4ydptbEQhrP9-mY,3414
+datarobot_genai/drmcp/core/mcp_instance.py,sha256=nt4gOlAQklMcqmohRIKovYcyhgLdb08NHMo28DBYmOk,18362
+datarobot_genai/drmcp/core/routes.py,sha256=dqE2M0UzAyyN9vQjlyTjYW4rpju3LT039po5weuO__I,17936
+datarobot_genai/drmcp/core/routes_utils.py,sha256=vSseXWlplMSnRgoJgtP_rHxWSAVYcx_tpTv4lyTpQoc,944
+datarobot_genai/drmcp/core/server_life_cycle.py,sha256=WKGJWGxalvqxupzJ2y67Kklc_9PgpZT0uyjlv_sr5wc,3419
+datarobot_genai/drmcp/core/telemetry.py,sha256=NEkSTC1w6uQgtukLHI-sWvR4EMgInysgATcvfQ5CplM,15378
+datarobot_genai/drmcp/core/tool_config.py,sha256=izUdM6dN3GRBzSBs-OagggM2dX5PGBnDbVv4N5bfWFI,3668
+datarobot_genai/drmcp/core/tool_filter.py,sha256=yKQlEtzyIeXGxZJkHbK36QI19vmgQkvqmfx5cTo2pp4,3156
+datarobot_genai/drmcp/core/utils.py,sha256=EvfpqKZ3tECMoxpIQ_tA_3rOgy6KJEYKC0lWZo_Daag,4517
+datarobot_genai/drmcp/core/dynamic_prompts/__init__.py,sha256=y4yapzp3KnFMzSR6HlNDS4uSuyNT7I1iPBvaCLsS0sU,577
+datarobot_genai/drmcp/core/dynamic_prompts/controllers.py,sha256=AGJlKqgHRO0Kd7Gl-Ulw9KYBgzjTTFXWBvOUF-SuKUI,5454
+datarobot_genai/drmcp/core/dynamic_prompts/dr_lib.py,sha256=4j33AKmq7kQX_EE2_RWAbP8-K5KPVEvpUginTWn_MHs,2701
+datarobot_genai/drmcp/core/dynamic_prompts/register.py,sha256=2c-vBaTfu3mq_8tSFfDAzG5hG06uS9CghIC1sJxHRNw,7173
+datarobot_genai/drmcp/core/dynamic_prompts/utils.py,sha256=BZ3792AgfvYlwL0_J0MzQfGecyEA5_OKUMynEZYzCds,1136
+datarobot_genai/drmcp/core/dynamic_tools/__init__.py,sha256=0kq9vMkF7EBsS6lkEdiLibmUrghTQqosHbZ5k-V9a5g,578
+datarobot_genai/drmcp/core/dynamic_tools/register.py,sha256=3M5-F0mhUYTZJWmFDmqzsj3QAd7ut7b0kPv-JZyaTzg,9204
+datarobot_genai/drmcp/core/dynamic_tools/schema.py,sha256=zw_fYleG2dZAYLVS3MWT2Dy_xXcL9JERG_mXgrusU1g,18457
+datarobot_genai/drmcp/core/dynamic_tools/deployment/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+datarobot_genai/drmcp/core/dynamic_tools/deployment/config.py,sha256=eVDsO3b16dY_jeEtIpZNpB085bDV5nhMUxIpuLxG9iU,8075
+datarobot_genai/drmcp/core/dynamic_tools/deployment/controllers.py,sha256=56M1z616kp32j40Ufy0SnHQmWg-UritCnExzYwNHA8s,2209
+datarobot_genai/drmcp/core/dynamic_tools/deployment/metadata.py,sha256=PNNnMX9zSsMxtHokeX_eNutUXBjP5u9G4ezzPMBLrfI,5702
+datarobot_genai/drmcp/core/dynamic_tools/deployment/register.py,sha256=e-or8WcGSe9EdiiPwr4fHlhwJfzV1GrWDWT7ES8jd7s,3300
+datarobot_genai/drmcp/core/dynamic_tools/deployment/adapters/__init__.py,sha256=0kq9vMkF7EBsS6lkEdiLibmUrghTQqosHbZ5k-V9a5g,578
+datarobot_genai/drmcp/core/dynamic_tools/deployment/adapters/base.py,sha256=wnua-dMcSRH3jTF8YsjwzA9UcsgNQHTH3r4y2xzVhkQ,2400
+datarobot_genai/drmcp/core/dynamic_tools/deployment/adapters/default.py,sha256=PLNLpy8_njT-9aLWt4QM0GsDDm8WqCI-Tl1ymNzcdXo,3232
+datarobot_genai/drmcp/core/dynamic_tools/deployment/adapters/drum.py,sha256=Ti7r8y-cszsyzagioWtfc-U3owdhRpQ6g5qrXrsJvQQ,8261
+datarobot_genai/drmcp/core/dynamic_tools/deployment/schemas/drum_agentic_fallback_schema.json,sha256=dG-6XyDmIJWirja7UmMhZGsYu851rn7As_dT5utmstQ,1147
+datarobot_genai/drmcp/core/dynamic_tools/deployment/schemas/drum_prediction_fallback_schema.json,sha256=D7SnrkpSlOtqhJ7OnbeelIV8DJ_EhCfX647VcTBEpuM,292
+datarobot_genai/drmcp/core/memory_management/__init__.py,sha256=y4yapzp3KnFMzSR6HlNDS4uSuyNT7I1iPBvaCLsS0sU,577
+datarobot_genai/drmcp/core/memory_management/manager.py,sha256=gmc_SQs12YQFMWl2UbfWR40QmLV9XuCnwPZgQwKWrbA,30552
+datarobot_genai/drmcp/core/memory_management/memory_tools.py,sha256=AxzpwOlldmhhDfKZcAxaGs7Xih2SCe0XbQuXX5nQczI,6397
+datarobot_genai/drmcp/test_utils/__init__.py,sha256=y4yapzp3KnFMzSR6HlNDS4uSuyNT7I1iPBvaCLsS0sU,577
+datarobot_genai/drmcp/test_utils/elicitation_test_tool.py,sha256=UVKwy39nl3XcVAh6IATcN-cWL2bfrprgRQ7fbK82jTI,3287
+datarobot_genai/drmcp/test_utils/integration_mcp_server.py,sha256=YSk19tbaka_0ziqi7LoXie4SJs-cvi9-H00Go0ZtQWE,3575
+datarobot_genai/drmcp/test_utils/mcp_utils_ete.py,sha256=46rH0fYYmUj7ygf968iRbdSp5u95v23BEw3Ts_c431Y,4788
+datarobot_genai/drmcp/test_utils/mcp_utils_integration.py,sha256=sHA_BWtpgIAFp9IXiNkUeBartBMjLAauqkV9bYtCr-g,3874
+datarobot_genai/drmcp/test_utils/test_interactive.py,sha256=KAScFT65GUkOxuiiBcjli8HHvV1NusVN01nOib3xVCc,7939
+datarobot_genai/drmcp/test_utils/tool_base_ete.py,sha256=2RvVmwHYczl7F6qZHkKYiI77IoL-PaMT3y59t0aQtTE,9328
+datarobot_genai/drmcp/test_utils/utils.py,sha256=JF2W9J4Q8pCqro7dj_bHObHNP7dfybDXesTLFOUsIVM,3039
+datarobot_genai/drmcp/test_utils/clients/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+datarobot_genai/drmcp/test_utils/clients/anthropic.py,sha256=qFhLvLZHMpZa2tZwI8pZQaaeG1lsM56VaONt6a9VU8c,2333
+datarobot_genai/drmcp/test_utils/clients/base.py,sha256=mPo8tmFlOgcHA3d3ynUsmH1XI49mj8jkUy3b0Y57slw,10606
+datarobot_genai/drmcp/test_utils/clients/dr_gateway.py,sha256=qlx0WxEOtTkxt9PiCxgWAp02k5jyUgXcKb9AwCGw6cw,2150
+datarobot_genai/drmcp/test_utils/clients/openai.py,sha256=tyIibvjtFp7u2BoHJqwIRlHn9UPtysKOgStoA9SZUYs,2566
+datarobot_genai/drmcp/tools/__init__.py,sha256=0kq9vMkF7EBsS6lkEdiLibmUrghTQqosHbZ5k-V9a5g,578
+datarobot_genai/drmcp/tools/clients/__init__.py,sha256=0kq9vMkF7EBsS6lkEdiLibmUrghTQqosHbZ5k-V9a5g,578
+datarobot_genai/drmcp/tools/clients/atlassian.py,sha256=__M_uz7FrcbKCYRzeMn24DCEYD6OmFx_LuywHCxgXsA,6472
+datarobot_genai/drmcp/tools/clients/confluence.py,sha256=h_G0By_kDnJeWDT_d-IREsaZ5-0xB5GoLXOqblYP5MA,20706
+datarobot_genai/drmcp/tools/clients/gdrive.py,sha256=Sonc8g52z8gDeaIGxh2GkoWYbtveCBGfDU0z3m4iSRU,29292
+datarobot_genai/drmcp/tools/clients/jira.py,sha256=Rm91JAyrNIqxu66-9rU1YqoRXVnWbEy-Ahvy6f6HlVg,9823
+datarobot_genai/drmcp/tools/clients/microsoft_graph.py,sha256=PASGThDPE8zkBZqach8lurJL1y47DWUPLwvf9N6uLGM,19234
+datarobot_genai/drmcp/tools/clients/s3.py,sha256=GmwzvurFdNfvxOooA8g5S4osRysHYU0S9ypg_177Glg,953
+datarobot_genai/drmcp/tools/confluence/__init__.py,sha256=0kq9vMkF7EBsS6lkEdiLibmUrghTQqosHbZ5k-V9a5g,578
+datarobot_genai/drmcp/tools/confluence/tools.py,sha256=_-ws65WLK8KZP_mKkf4yJ7ZunR8qdyoiMwHQX47MSMw,12362
+datarobot_genai/drmcp/tools/gdrive/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+datarobot_genai/drmcp/tools/gdrive/tools.py,sha256=jSKz0TuqAdHFitUZ-BW8NDl31Aq4VGpWa0yWSksLi00,13887
+datarobot_genai/drmcp/tools/jira/__init__.py,sha256=0kq9vMkF7EBsS6lkEdiLibmUrghTQqosHbZ5k-V9a5g,578
+datarobot_genai/drmcp/tools/jira/tools.py,sha256=dfkqTU2HH-7n44hX80ODFacKq0p0LOchFcZtIIKFNMM,9687
+datarobot_genai/drmcp/tools/microsoft_graph/__init__.py,sha256=CuOaMt1AJo7cHx_GuhO3s_aqxZas_wlDsoBorBsvbeU,577
+datarobot_genai/drmcp/tools/microsoft_graph/tools.py,sha256=zJ-UA1TMhPOYcExvgWv0YBjDsSIDPA-U1SEbBrVfAc8,7744
+datarobot_genai/drmcp/tools/predictive/__init__.py,sha256=WuOHlNNEpEmcF7gVnhckruJRKU2qtmJLE3E7zoCGLDo,1030
+datarobot_genai/drmcp/tools/predictive/data.py,sha256=sSFAmO6x0DSuolw8urhMaOj5PwfUH29oc2mEOZI3YU4,4631
+datarobot_genai/drmcp/tools/predictive/deployment.py,sha256=lm02Ayuo11L1hP41fgi3QpR1Eyty-Wc16rM0c8SgliM,3277
+datarobot_genai/drmcp/tools/predictive/deployment_info.py,sha256=BGEF_dmbxOBJR0n1Tt9TO2-iNTQSBTr-oQUyaxLZ0ZI,15297
+datarobot_genai/drmcp/tools/predictive/model.py,sha256=Yih5-KedJ-1yupPLXCJsCXOdyWWi9pRvgapXDlgXWJA,4891
+datarobot_genai/drmcp/tools/predictive/predict.py,sha256=Qoob2_t2crfWtyPzkXMRz2ITZumnczU6Dq4C7q9RBMI,9370
+datarobot_genai/drmcp/tools/predictive/predict_realtime.py,sha256=urq6rPyZFsAP-bPyclSNzrkvb6FTamdlFau8q0IWWJ0,13472
+datarobot_genai/drmcp/tools/predictive/project.py,sha256=xC52UdYvuFeNZC7Y5MfXcvzTL70WwAacQXESr6rqN6s,3255
+datarobot_genai/drmcp/tools/predictive/training.py,sha256=LzMxbBT8wxKYDrRlVElfmTUrzpmGvwrR-mTGf6YUnIA,23998
+datarobot_genai/langgraph/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+datarobot_genai/langgraph/agent.py,sha256=DRnywmS9KDywyChtuIZZwNKbJs8BpC259EG_kxYbiQ8,15828
+datarobot_genai/langgraph/mcp.py,sha256=iA2_j46mZAaNaL7ntXT-LW6C-NMJkzr3VfKDDfe7mh8,2851
+datarobot_genai/llama_index/__init__.py,sha256=JEMkLQLuP8n14kNE3bZ2j08NdajnkJMfYjDQYqj7C0c,407
+datarobot_genai/llama_index/agent.py,sha256=V6ZsD9GcBDJS-RJo1tJtIHhyW69_78gM6_fOHFV-Piw,1829
+datarobot_genai/llama_index/base.py,sha256=ovcQQtC-djD_hcLrWdn93jg23AmD6NBEj7xtw4a6K6c,14481
+datarobot_genai/llama_index/mcp.py,sha256=leXqF1C4zhuYEKFwNEfZHY4dsUuGZk3W7KArY-zxVL8,2645
+datarobot_genai/nat/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+datarobot_genai/nat/agent.py,sha256=DuGrgqt1FzvAE-cRH_P3LTFUlwuClvbVurdwA-RsbuY,11177
+datarobot_genai/nat/datarobot_auth_provider.py,sha256=Z4NSsrHxK8hUeiqtK_lryHsUuZC74ziNo_FHbsZgtiM,4230
+datarobot_genai/nat/datarobot_llm_clients.py,sha256=-_q_KlKOVQecIYJd8YRiYnS4ZNazQAiAdZBE1Zip_wQ,12684
+datarobot_genai/nat/datarobot_llm_providers.py,sha256=aDoQcTeGI-odqydPXEX9OGGNFbzAtpqzTvHHEkmJuEQ,4963
+datarobot_genai/nat/datarobot_mcp_client.py,sha256=jL8sXb8g4gvt0VYgB2tfMGsMjpB1GV2XIbN0iv_LxVU,10701
+datarobot_genai/nat/helpers.py,sha256=Q7E3ADZdtFfS8E6OQPyw2wgA6laQ58N3bhLj5CBWwJs,3265
+datarobot_genai-0.2.31.dist-info/METADATA,sha256=h36ncbvYaZpFSme4CEIH1WouCke0-S-dODtLZRqnNB8,6301
+datarobot_genai-0.2.31.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+datarobot_genai-0.2.31.dist-info/entry_points.txt,sha256=jEW3WxDZ8XIK9-ISmTyt5DbmBb047rFlzQuhY09rGrM,284
+datarobot_genai-0.2.31.dist-info/licenses/AUTHORS,sha256=isJGUXdjq1U7XZ_B_9AH8Qf0u4eX0XyQifJZ_Sxm4sA,80
+datarobot_genai-0.2.31.dist-info/licenses/LICENSE,sha256=U2_VkLIktQoa60Nf6Tbt7E4RMlfhFSjWjcJJfVC-YCE,11341
+datarobot_genai-0.2.31.dist-info/RECORD,,

datarobot_genai-0.2.31.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.28.0
+Root-Is-Purelib: true
+Tag: py3-none-any

datarobot_genai-0.2.31.dist-info/entry_points.txt

@@ -0,0 +1,5 @@
+[nat.plugins]
+datarobot_auth_provider = datarobot_genai.nat.datarobot_auth_provider
+datarobot_llm_clients = datarobot_genai.nat.datarobot_llm_clients
+datarobot_llm_providers = datarobot_genai.nat.datarobot_llm_providers
+datarobot_mcp_client = datarobot_genai.nat.datarobot_mcp_client

datarobot_genai-0.2.31.dist-info/licenses/AUTHORS

@@ -0,0 +1,2 @@
+Cavit Cakir cavit.cakir@datarobot.com
+James Clemens james.clemens@datarobot.com