datarobot-genai 0.1.63__py3-none-any.whl → 0.1.67__py3-none-any.whl

datarobot_genai/core/agents/base.py

@@ -52,6 +52,7 @@ class BaseAgent(Generic[TTool], abc.ABC):
         verbose: bool | str | None = True,
         timeout: int | None = 90,
         authorization_context: dict[str, Any] | None = None,
+        forwarded_headers: dict[str, str] | None = None,
         **_: Any,
     ) -> None:
         self.api_key = api_key or os.environ.get("DATAROBOT_API_TOKEN")
@@ -68,6 +69,7 @@ class BaseAgent(Generic[TTool], abc.ABC):
             self.verbose = bool(verbose)
         self._mcp_tools: list[TTool] = []
         self._authorization_context = authorization_context or {}
+        self._forwarded_headers: dict[str, str] = forwarded_headers or {}
 
     def set_mcp_tools(self, tools: list[TTool]) -> None:
         self._mcp_tools = tools
@@ -86,6 +88,11 @@ class BaseAgent(Generic[TTool], abc.ABC):
         """Return the authorization context for this agent."""
         return self._authorization_context
 
+    @property
+    def forwarded_headers(self) -> dict[str, str]:
+        """Return the forwarded headers for this agent."""
+        return self._forwarded_headers
+
     def litellm_api_base(self, deployment_id: str | None) -> str:
         return get_api_base(self.api_base, deployment_id)
 

datarobot_genai/core/custom_model.py

@@ -139,6 +139,11 @@ def chat_entrypoint(
     completion_create_params["authorization_context"] = resolve_authorization_context(
         completion_create_params, **kwargs
     )
+    # Keep only allowed headers from the forwarded_headers.
+    incoming_headers = kwargs.get("headers", {}) or {}
+    allowed_headers = {"x-datarobot-api-token", "x-datarobot-api-key"}
+    forwarded_headers = {k: v for k, v in incoming_headers.items() if k.lower() in allowed_headers}
+    completion_create_params["forwarded_headers"] = forwarded_headers
 
     # Instantiate user agent with all supplied completion params including auth context
     agent = agent_cls(**completion_create_params)

datarobot_genai/core/mcp/common.py

@@ -16,6 +16,7 @@ import json
 import re
 from typing import Any
 from typing import Literal
+from urllib.parse import urlparse
 
 from datarobot.core.config import DataRobotAppFrameworkBaseSettings
 from pydantic import field_validator
@@ -37,6 +38,7 @@ class MCPConfig(DataRobotAppFrameworkBaseSettings):
     datarobot_endpoint: str | None = None
     datarobot_api_token: str | None = None
     authorization_context: dict[str, Any] | None = None
+    forwarded_headers: dict[str, str] | None = None
 
     _auth_context_handler: AuthContextHeaderHandler | None = None
     _server_config: dict[str, Any] | None = None
@@ -121,18 +123,33 @@ class MCPConfig(DataRobotAppFrameworkBaseSettings):
         """
         if self.external_mcp_url:
             # External MCP URL - no authentication needed
+            headers: dict[str, str] = {}
+
+            # Forward headers for localhost connections
+            if self.forwarded_headers:
+                try:
+                    parsed_url = urlparse(self.external_mcp_url)
+                    hostname = parsed_url.hostname or ""
+                    # Check if hostname is localhost or 127.0.0.1
+                    if hostname in ("localhost", "127.0.0.1", "::1"):
+                        headers.update(self.forwarded_headers)
+                except Exception:
+                    # If URL parsing fails, fall back to simple string check
+                    if "localhost" in self.external_mcp_url or "127.0.0.1" in self.external_mcp_url:
+                        headers.update(self.forwarded_headers)
+
+            # Merge external headers if provided
             if self.external_mcp_headers:
-                headers = json.loads(self.external_mcp_headers)
-            else:
-                headers = {}
+                external_headers = json.loads(self.external_mcp_headers)
+                headers.update(external_headers)
 
-            config = {
+            return {
                 "url": self.external_mcp_url.rstrip("/"),
                 "transport": self.external_mcp_transport,
                 "headers": headers,
             }
-            return config
-        elif self.mcp_deployment_id:
+
+        if self.mcp_deployment_id:
             # DataRobot deployment ID - requires authentication
             if self.datarobot_endpoint is None:
                 raise ValueError(
@@ -142,15 +159,21 @@ class MCPConfig(DataRobotAppFrameworkBaseSettings):
                 raise ValueError(
                     "When using a DataRobot hosted MCP deployment, datarobot_api_token must be set."
                 )
+
             base_url = self.datarobot_endpoint.rstrip("/")
             if not base_url.endswith("/api/v2"):
-                base_url = base_url + "/api/v2"
+                base_url = f"{base_url}/api/v2"
+
             url = f"{base_url}/deployments/{self.mcp_deployment_id}/directAccess/mcp"
 
-            headers = {
-                **self._authorization_bearer_header(),
-                **self._authorization_context_header(),
-            }
+            # Start with forwarded headers if available
+            headers = {}
+            if self.forwarded_headers:
+                headers.update(self.forwarded_headers)
+
+            # Add authentication headers
+            headers.update(self._authorization_bearer_header())
+            headers.update(self._authorization_context_header())
 
             return {
                 "url": url,

datarobot_genai/crewai/base.py

@@ -93,6 +93,7 @@ class CrewAIAgent(BaseAgent[BaseTool], abc.ABC):
         # Use MCP context manager to handle connection lifecycle
         with mcp_tools_context(
             authorization_context=self._authorization_context,
+            forwarded_headers=self.forwarded_headers,
         ) as mcp_tools:
             # Set MCP tools for all agents if MCP is not configured this is effectively a no-op
             self.set_mcp_tools(mcp_tools)

datarobot_genai/crewai/mcp.py

@@ -30,9 +30,13 @@ from datarobot_genai.core.mcp.common import MCPConfig
 @contextmanager
 def mcp_tools_context(
     authorization_context: dict[str, Any] | None = None,
+    forwarded_headers: dict[str, str] | None = None,
 ) -> Generator[list[Any], None, None]:
     """Context manager for MCP tools that handles connection lifecycle."""
-    config = MCPConfig(authorization_context=authorization_context)
+    config = MCPConfig(
+        authorization_context=authorization_context,
+        forwarded_headers=forwarded_headers,
+    )
     # If no MCP server configured, return empty tools list
     if not config.server_config:
         print("No MCP server configured, using empty tools list", flush=True)

datarobot_genai/drmcp/core/dr_mcp_server.py

@@ -184,13 +184,17 @@ class DataRobotMCPServer:
             prompts = asyncio.run(self._mcp._list_prompts_mcp())
             resources = asyncio.run(self._mcp._list_resources_mcp())
 
-            self._logger.info(f"Registered tools: {len(tools)}")
+            tools_count = len(tools)
+            prompts_count = len(prompts)
+            resources_count = len(resources)
+
+            self._logger.info(f"Registered tools: {tools_count}")
             for tool in tools:
                 self._logger.info(f" > {tool.name}")
-            self._logger.info(f"Registered prompts: {len(prompts)}")
+            self._logger.info(f"Registered prompts: {prompts_count}")
             for prompt in prompts:
                 self._logger.info(f" > {prompt.name}")
-            self._logger.info(f"Registered resources: {len(resources)}")
+            self._logger.info(f"Registered resources: {resources_count}")
             for resource in resources:
                 self._logger.info(f" > {resource.name}")
 
@@ -209,6 +213,9 @@ class DataRobotMCPServer:
                         self._mcp,
                         self._mcp_transport,
                         port=self._config.mcp_server_port,
+                        tools_count=tools_count,
+                        prompts_count=prompts_count,
+                        resources_count=resources_count,
                     )
 
                 if self._mcp_transport == "stdio":

datarobot_genai/drmcp/core/dr_mcp_server_logo.py

@@ -54,6 +54,9 @@ def log_server_custom_banner(
     host: str | None = None,
     port: int | None = None,
     path: str | None = None,
+    tools_count: int | None = None,
+    prompts_count: int | None = None,
+    resources_count: int | None = None,
 ) -> None:
     """
     Create and log a formatted banner with server information and logo.
@@ -64,13 +67,20 @@ def log_server_custom_banner(
         host: Host address (for HTTP transports)
         port: Port number (for HTTP transports)
         path: Server path (for HTTP transports)
+        tools_count: Number of tools registered
+        prompts_count: Number of prompts registered
+        resources_count: Number of resources registered
     """
     # Create the logo text
     # Use Text with no_wrap and markup disabled to preserve ANSI escape codes
     logo_text = Text.from_ansi(DR_LOGO_ASCII, no_wrap=True)
 
     # Create the main title
-    title_text = Text(f"DataRobot MCP Server {datarobot_genai_version}", style="bold green")
+    title_text = Text(f"DataRobot MCP Server {datarobot_genai_version}", style="dim green")
+    stats_text = Text(
+        f"{tools_count} tools, {prompts_count} prompts, {resources_count} resources",
+        style="bold green",
+    )
 
     # Create the information table
     info_table = Table.grid(padding=(0, 1))
@@ -107,6 +117,7 @@ def log_server_custom_banner(
         Align.center(logo_text),
         "",
         Align.center(title_text),
+        Align.center(stats_text),
         "",
         "",
         Align.center(info_table),

datarobot_genai/drmcp/core/dynamic_prompts/dr_lib.py

@@ -11,7 +11,7 @@
 # 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 import defaultdict
 from dataclasses import dataclass
 
 import datarobot as dr
@@ -34,6 +34,7 @@ class DrVariable:
 @dataclass
 class DrPromptVersion:
     id: str
+    prompt_template_id: str
     version: int
     prompt_text: str
     variables: list[DrVariable]
@@ -45,6 +46,7 @@ class DrPromptVersion:
         ]
         return cls(
             id=d["id"],
+            prompt_template_id=d["promptTemplateId"],
             version=d["version"],
             prompt_text=d["promptText"],
             variables=variables,
@@ -58,7 +60,9 @@ class DrPrompt:
     description: str
 
     def get_latest_version(self) -> DrPromptVersion | None:
-        prompt_template_versions = get_datarobot_prompt_template_versions(self.id)
+        all_prompt_template_versions = get_datarobot_prompt_template_versions([self.id])
+        prompt_template_versions = all_prompt_template_versions.get(self.id)
+
         if not prompt_template_versions:
             return None
         latest_version = max(prompt_template_versions, key=lambda v: v.version)
@@ -77,15 +81,21 @@ def get_datarobot_prompt_templates() -> list[DrPrompt]:
     return [DrPrompt.from_dict(prompt_template) for prompt_template in prompt_templates_data]
 
 
-def get_datarobot_prompt_template_versions(prompt_template_id: str) -> list[DrPromptVersion]:
+def get_datarobot_prompt_template_versions(
+    prompt_template_ids: list[str],
+) -> dict[str, list[DrPromptVersion]]:
     prompt_template_versions_data = dr.utils.pagination.unpaginate(
-        initial_url=f"genai/promptTemplates/{prompt_template_id}/versions/",
-        initial_params={},
+        initial_url="genai/promptTemplates/versions/",
+        initial_params={
+            "promptTemplateIds": prompt_template_ids,
+        },
         client=get_api_client(),
     )
-    prompt_template_versions = []
+    prompt_template_versions = defaultdict(list)
     for prompt_template_version in prompt_template_versions_data:
-        prompt_template_versions.append(DrPromptVersion.from_dict(prompt_template_version))
+        prompt_template_versions[prompt_template_version["promptTemplateId"]].append(
+            DrPromptVersion.from_dict(prompt_template_version)
+        )
     return prompt_template_versions
 
 

datarobot_genai/drmcp/core/dynamic_prompts/register.py

@@ -27,6 +27,7 @@ from datarobot_genai.drmcp.core.mcp_instance import register_prompt
 from .dr_lib import DrPrompt
 from .dr_lib import DrPromptVersion
 from .dr_lib import DrVariable
+from .dr_lib import get_datarobot_prompt_template_versions
 from .dr_lib import get_datarobot_prompt_templates
 
 logger = logging.getLogger(__name__)
@@ -36,11 +37,21 @@ async def register_prompts_from_datarobot_prompt_management() -> None:
     """Register prompts from DataRobot Prompt Management."""
     prompts = get_datarobot_prompt_templates()
     logger.info(f"Found {len(prompts)} prompts in Prompts Management.")
+    all_prompts_versions = get_datarobot_prompt_template_versions(
+        prompt_template_ids=list({prompt.id for prompt in prompts})
+    )
 
     # Try to register each prompt, continue on failure
     for prompt in prompts:
+        prompt_versions = all_prompts_versions.get(prompt.id)
+        if not prompt_versions:
+            logger.warning(f"Prompt template id {prompt.id} has no versions.")
+            continue
+
+        latest_version = max(prompt_versions, key=lambda v: v.version)
+
         try:
-            await register_prompt_from_datarobot_prompt_management(prompt)
+            await register_prompt_from_datarobot_prompt_management(prompt, latest_version)
         except DynamicPromptRegistrationError:
             pass
 
@@ -114,15 +125,36 @@ async def register_prompt_from_datarobot_prompt_management(
         ) from exc
 
 
+def _escape_non_ascii(s: str) -> str:
+    out = []
+    for ch in s:
+        # If its space -> change to underscore
+        if ch.isspace():
+            out.append("_")
+        # ASCII letter, digit or underscore -> keep
+        elif ch.isascii() and (ch.isalnum() or ch == "_"):
+            out.append(ch)
+        # Everything else -> encode as 'xHEX'
+        else:
+            out.append(f"x{ord(ch):x}")
+    return "".join(out)
+
+
 def to_valid_mcp_prompt_name(s: str) -> str:
     """Convert an arbitrary string into a valid MCP prompt name."""
-    # Replace any sequence of invalid characters with '_'
-    s = re.sub(r"[^0-9a-zA-Z_]+", "_", s)
-
     # If its ONLY numbers return "prompt_[number]"
     if s.isdigit():
         return f"prompt_{s}"
 
+    # First, ASCII-transliterate using hex escape for non-ASCII
+    if not s.isascii():
+        # whole string non-ascii? -> escape and prefix with prompt_
+        encoded = _escape_non_ascii(s)
+        return f"prompt_{encoded}"
+
+    # Replace any sequence of invalid characters with '_'
+    s = re.sub(r"[^0-9a-zA-Z_]+", "_", s)
+
     # Remove leading characters that are not letters or underscores (can't start with a digit or _)
     s = re.sub(r"^[^a-zA-Z]+", "", s)
 

datarobot_genai/drmcp/core/mcp_server_tools.py

@@ -51,7 +51,7 @@ async def list_tools_by_tags(tags: list[str] | None = None, match_all: bool = Fa
     -------
         A formatted string listing tools that match the tag criteria.
     """
-    tools = await mcp._list_tools_mcp(tags=tags, match_all=match_all)
+    tools = await mcp.list_tools(tags=tags, match_all=match_all)
 
     if not tools:
         if tags:
@@ -95,7 +95,7 @@ async def get_tool_info_by_name(tool_name: str) -> str:
     -------
         A formatted string with detailed information about the tool.
     """
-    all_tools = await mcp._list_tools_mcp()
+    all_tools = await mcp.list_tools()
 
     for tool in all_tools:
         if tool.name == tool_name:

datarobot_genai/langgraph/agent.py

@@ -86,6 +86,7 @@ class LangGraphAgent(BaseAgent[BaseTool], abc.ABC):
             ]:
                 async with mcp_tools_context(
                     authorization_context=self._authorization_context,
+                    forwarded_headers=self.forwarded_headers,
                 ) as mcp_tools:
                     self.set_mcp_tools(mcp_tools)
                     result = await self._invoke(completion_create_params)
@@ -104,6 +105,7 @@ class LangGraphAgent(BaseAgent[BaseTool], abc.ABC):
             # For non-streaming, use async with directly
             async with mcp_tools_context(
                 authorization_context=self._authorization_context,
+                forwarded_headers=self.forwarded_headers,
             ) as mcp_tools:
                 self.set_mcp_tools(mcp_tools)
                 result = await self._invoke(completion_create_params)

datarobot_genai/langgraph/mcp.py

@@ -28,6 +28,7 @@ from datarobot_genai.core.mcp.common import MCPConfig
 @asynccontextmanager
 async def mcp_tools_context(
     authorization_context: dict[str, Any] | None = None,
+    forwarded_headers: dict[str, str] | None = None,
 ) -> AsyncGenerator[list[BaseTool], None]:
     """Yield a list of LangChain BaseTool instances loaded via MCP.
 
@@ -37,8 +38,13 @@ async def mcp_tools_context(
     ----------
     authorization_context : dict[str, Any] | None
         Authorization context to use for MCP connections
+    forwarded_headers : dict[str, str] | None
+        Forwarded headers, e.g. x-datarobot-api-key to use for MCP authentication
     """
-    mcp_config = MCPConfig(authorization_context=authorization_context)
+    mcp_config = MCPConfig(
+        authorization_context=authorization_context,
+        forwarded_headers=forwarded_headers,
+    )
     server_config = mcp_config.server_config
 
     if not server_config:

datarobot_genai/llama_index/base.py

@@ -84,6 +84,7 @@ class LlamaIndexAgent(BaseAgent[BaseTool], abc.ABC):
         # Load MCP tools (if configured) asynchronously before building workflow
         mcp_tools = await load_mcp_tools(
             authorization_context=self._authorization_context,
+            forwarded_headers=self.forwarded_headers,
         )
         self.set_mcp_tools(mcp_tools)
 

datarobot_genai/llama_index/mcp.py

@@ -30,18 +30,23 @@ from datarobot_genai.core.mcp.common import MCPConfig
 
 async def load_mcp_tools(
     authorization_context: dict[str, Any] | None = None,
+    forwarded_headers: dict[str, str] | None = None,
 ) -> list[Any]:
     """
     Asynchronously load MCP tools for LlamaIndex.
 
     Args:
         authorization_context: Optional authorization context for MCP connections
+        forwarded_headers: Optional forwarded headers, e.g. x-datarobot-api-key for MCP auth
 
     Returns
     -------
         List of MCP tools, or empty list if no MCP configuration is present.
     """
-    config = MCPConfig(authorization_context=authorization_context)
+    config = MCPConfig(
+        authorization_context=authorization_context,
+        forwarded_headers=forwarded_headers,
+    )
     server_params = config.server_config
 
     if not server_params:

{datarobot_genai-0.1.63.dist-info → datarobot_genai-0.1.67.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: datarobot-genai
-Version: 0.1.63
+Version: 0.1.67
 Summary: Generic helpers for GenAI
 Project-URL: Homepage, https://github.com/datarobot-oss/datarobot-genai
 Author: DataRobot, Inc.
@@ -32,7 +32,7 @@ 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==2.13.0.2; 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'

{datarobot_genai-0.1.63.dist-info → datarobot_genai-0.1.67.dist-info}/RECORD

@@ -1,10 +1,10 @@
 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=vnV62eyBpokWTA5BDU7GBgpB2nYop4oY8z4K1AOAet4,6799
+datarobot_genai/core/custom_model.py,sha256=8SMW3BAToI44331pF6fATQlS-FulDXop1OmmrVxuB1U,7155
 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=gG54Q7jckW8eFe-z89JdiX4mq6UYGXGsv-zBe3WhVuc,6390
+datarobot_genai/core/agents/base.py,sha256=kyMk9m1qGFT_XretYIEbP0F-vstPIWh_OZlECqWXvdw,6685
 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
@@ -13,15 +13,15 @@ datarobot_genai/core/cli/__init__.py,sha256=B93Yb6VavoZpatrh8ltCL6YglIfR5FHgytXb
 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=7ugPoj0YGJMoHGQVBqQw2LgYOhXh1YX21mfGIMMAS68,5762
+datarobot_genai/core/mcp/common.py,sha256=Y24Ij8e4BwOnxojnz2BJRIFcs_Y7EfbORLrN6txowpY,6833
 datarobot_genai/core/utils/__init__.py,sha256=VxtRUz6iwb04eFQQy0zqTNXLAkYpPXcJxVoKV0nOdXk,59
 datarobot_genai/core/utils/auth.py,sha256=Xo1PxVr6oMgtMHkmHdS02klDKK1cyDpjGvIMF4Tx0Lo,7874
 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=7zHq9pmfdjV-xtDUqjdRJ4bKVPArWik8RbI3ap54ab0,7452
+datarobot_genai/crewai/base.py,sha256=iDIbU5LrdEaBKd914WuZuVg5f9yMw7pZTsWZqMTO1TI,7506
 datarobot_genai/crewai/events.py,sha256=K67bO1zwPrxmppz2wh8dFGNbVebyWGXAMD7oodFE2sQ,5462
-datarobot_genai/crewai/mcp.py,sha256=1jSzLJ9GRA5wsz53uScJ-HSXwJJrdo6-a4z5cSSp09k,2002
+datarobot_genai/crewai/mcp.py,sha256=AJTrs-8KdiRSjRECfBT1lJOsszWMoFoN9NIa1p5_wsM,2115
 datarobot_genai/drmcp/__init__.py,sha256=JE83bfpGU7v77VzrDdlb0l8seM5OwUsUbaQErJ2eisc,2983
 datarobot_genai/drmcp/server.py,sha256=KE4kjS5f9bfdYftG14HBHrfvxDfCD4pwCXePfvl1OvU,724
 datarobot_genai/drmcp/core/__init__.py,sha256=y4yapzp3KnFMzSR6HlNDS4uSuyNT7I1iPBvaCLsS0sU,577
@@ -31,12 +31,12 @@ datarobot_genai/drmcp/core/config.py,sha256=D7bSi40Yc5J71_JxmpfppG83snbIJW9iz1J7
 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=KBF-8gJcWNaUAtoXJoVq_SKguNbSfGxPsNbYMEmzkBY,12555
-datarobot_genai/drmcp/core/dr_mcp_server_logo.py,sha256=vFsPHn8l_a6bNB_18se6ZjmItmA3l6PuCmRcJJ7Qb0g,4262
+datarobot_genai/drmcp/core/dr_mcp_server.py,sha256=7mu5UXHQmKNbIpNoQE0lPJaUI7AZa03avfHZRRtpjNI,12841
+datarobot_genai/drmcp/core/dr_mcp_server_logo.py,sha256=Npkn-cPPBv261kiiaNwfeS_y4a2WXlbiMS2dP80uDj8,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=wMsP39xqTmNBYqd49olEQb5UHTSsxj6BOIoIElorRB0,19235
-datarobot_genai/drmcp/core/mcp_server_tools.py,sha256=OyTDhvm38t8gzaMrSVn-SQOWtoaEugKtbZIhCmGkR6w,4288
+datarobot_genai/drmcp/core/mcp_server_tools.py,sha256=odNZKozfx0VV38SLZHw9lY0C0JM_JnRI06W3BBXnyE4,4278
 datarobot_genai/drmcp/core/routes.py,sha256=zW4Ukr9f_odF-_oXq_WlTaCH5P4mb7gBSFp67IiuAoo,17208
 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
@@ -45,8 +45,8 @@ datarobot_genai/drmcp/core/tool_filter.py,sha256=tLOcG50QBvS48cOVHM6OqoODYiiS6Ke
 datarobot_genai/drmcp/core/utils.py,sha256=dSjrayWVcnC5GxQcvOIOSHaoEymPIVtG_s2ZBMlmSOw,4336
 datarobot_genai/drmcp/core/dynamic_prompts/__init__.py,sha256=y4yapzp3KnFMzSR6HlNDS4uSuyNT7I1iPBvaCLsS0sU,577
 datarobot_genai/drmcp/core/dynamic_prompts/controllers.py,sha256=vCMYxwYjNDadRxSlRk6p8pHK6h_2K-PkbBTTW_lqBJ0,3318
-datarobot_genai/drmcp/core/dynamic_prompts/dr_lib.py,sha256=QEySFLKdTxsXk5VoGsHobbZ_olTB9nzPGo7M99TL5mc,3894
-datarobot_genai/drmcp/core/dynamic_prompts/register.py,sha256=Q2c9zuOuywS63AcGpD4774OhQKSGHX2tpLXqW_C4JKM,6058
+datarobot_genai/drmcp/core/dynamic_prompts/dr_lib.py,sha256=IEdD2Gqm4SfUdiXJB99RiWxkN6frGaxJ2SfATetMM3c,4243
+datarobot_genai/drmcp/core/dynamic_prompts/register.py,sha256=5AEh1m8GX-gPZHUdiE1VATt7IKJQk-eThcxh01sWn0I,7204
 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
@@ -83,19 +83,19 @@ datarobot_genai/drmcp/tools/predictive/predict_realtime.py,sha256=t7f28y_ealZoA6
 datarobot_genai/drmcp/tools/predictive/project.py,sha256=KaMDAvJY4s12j_4ybA7-KcCS1yMOj-KPIKNBgCSE2iM,2536
 datarobot_genai/drmcp/tools/predictive/training.py,sha256=kxeDVLqUh9ajDk8wK7CZRRydDK8UNuTVZCB3huUihF8,23660
 datarobot_genai/langgraph/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-datarobot_genai/langgraph/agent.py,sha256=us-DXSgmI2271tsbzOOtDbRHdStCsOro-YRQ30ePpkA,9739
-datarobot_genai/langgraph/mcp.py,sha256=_ggxlRFkydhJgJVcAdwDWZL0WWCejctNC5X9ycAWeUM,2610
+datarobot_genai/langgraph/agent.py,sha256=IcNawsi2Avsw3usTQ6Z8Ku0JdCsK3sYa2CQMhUe2Kk4,9859
+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=MwhSlRCwsP0Z9IxaTJ4tMZO2mCMX3pvqNFl3s9aj1P8,14427
-datarobot_genai/llama_index/mcp.py,sha256=3XULVxtLBV_ItM491zWRg2KMHGszy6OTyPzHW08zO2k,2439
+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=siBLDWAff2-JwZ8Q3iNpM_e4_IoSwG9IvY0hyEjNenw,10292
 datarobot_genai/nat/datarobot_llm_clients.py,sha256=IZq_kooUL8QyDTkpEreszLQk9vCzg6-FbTjIkXR9wc0,7203
 datarobot_genai/nat/datarobot_llm_providers.py,sha256=lOVaL_0Fl6-7GFYl3HmfqttqKpKt-2w8o92P3T7B6cU,3683
-datarobot_genai-0.1.63.dist-info/METADATA,sha256=wrBVB4hN9XykoMaTi9w-NPbeWTlwyis3FUO4jZ5--DM,5911
-datarobot_genai-0.1.63.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-datarobot_genai-0.1.63.dist-info/entry_points.txt,sha256=CZhmZcSyt_RBltgLN_b9xasJD6J5SaDc_z7K0wuOY9Y,150
-datarobot_genai-0.1.63.dist-info/licenses/AUTHORS,sha256=isJGUXdjq1U7XZ_B_9AH8Qf0u4eX0XyQifJZ_Sxm4sA,80
-datarobot_genai-0.1.63.dist-info/licenses/LICENSE,sha256=U2_VkLIktQoa60Nf6Tbt7E4RMlfhFSjWjcJJfVC-YCE,11341
-datarobot_genai-0.1.63.dist-info/RECORD,,
+datarobot_genai-0.1.67.dist-info/METADATA,sha256=hfSJaoB-QbjSH8wEMUAYxqzmdyudEz8IYdqG53tRdb4,5918
+datarobot_genai-0.1.67.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+datarobot_genai-0.1.67.dist-info/entry_points.txt,sha256=CZhmZcSyt_RBltgLN_b9xasJD6J5SaDc_z7K0wuOY9Y,150
+datarobot_genai-0.1.67.dist-info/licenses/AUTHORS,sha256=isJGUXdjq1U7XZ_B_9AH8Qf0u4eX0XyQifJZ_Sxm4sA,80
+datarobot_genai-0.1.67.dist-info/licenses/LICENSE,sha256=U2_VkLIktQoa60Nf6Tbt7E4RMlfhFSjWjcJJfVC-YCE,11341
+datarobot_genai-0.1.67.dist-info/RECORD,,

{datarobot_genai-0.1.63.dist-info → datarobot_genai-0.1.67.dist-info}/WHEEL

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