datarobot-genai 0.2.31__py3-none-any.whl → 0.2.32__py3-none-any.whl

datarobot_genai/drmcp/core/exceptions.py

@@ -19,7 +19,3 @@ class DynamicToolRegistrationError(Exception):
 
 class DynamicPromptRegistrationError(Exception):
     """Exception raised for errors in the dynamic prompt registration process."""
-
-
-class MCPError(Exception):
-    """Base class for MCP errors."""

datarobot_genai/drmcp/core/logging.py

@@ -20,7 +20,7 @@ from collections.abc import Callable
 from typing import Any
 from typing import TypeVar
 
-from .exceptions import MCPError
+from fastmcp.exceptions import ToolError
 
 # Secret patterns to redact from logs
 SECRET_PATTERNS = [
@@ -93,6 +93,6 @@ def log_execution(func: F) -> F:
             return result
         except Exception as e:
             error_msg = _log_error(logger, func.__name__, e, args=args, kwargs=kwargs)
-            raise MCPError(error_msg)
+            raise ToolError(error_msg)
 
     return wrapper  # type: ignore[return-value]

datarobot_genai/drmcp/test_utils/clients/base.py

@@ -27,7 +27,7 @@ from mcp.types import ListToolsResult
 from mcp.types import TextContent
 from openai.types.chat.chat_completion import ChatCompletion
 
-from ..utils import save_response_to_file
+from datarobot_genai.drmcp.test_utils.utils import save_response_to_file
 
 
 class ToolCall:

datarobot_genai/drmcp/tools/clients/gdrive.py

@@ -20,6 +20,7 @@ import logging
 import uuid
 from typing import Annotated
 from typing import Any
+from typing import Literal
 
 import httpx
 from datarobot.auth.datarobot.exceptions import OAuthServiceClientErr
@@ -821,6 +822,96 @@ class GoogleDriveClient:
             headers={"Content-Type": f"multipart/related; boundary={boundary}"},
         )
 
+    async def manage_access(
+        self,
+        *,
+        file_id: str,
+        action: Literal["add", "update", "remove"],
+        role: Literal["reader", "commenter", "writer", "fileOrganizer", "organizer", "owner"]
+        | None = None,
+        email_address: str | None = None,
+        permission_id: str | None = None,
+        transfer_ownership: bool = False,
+    ) -> str:
+        """Manage access permissions for a Google Drive file or folder.
+
+        Adds, updates, or removes sharing permissions on an existing Google Drive
+        file or folder using the Google Drive Permissions API.
+
+        This method supports granting access to users or groups, changing access
+        roles, and revoking permissions. Ownership transfer is supported for files
+        in "My Drive" when explicitly requested.
+
+        Args:
+            file_id: The ID of the Google Drive file or folder whose permissions
+                are being managed.
+            action: The permission operation to perform.
+            role: The access role to assign or update. Valid values include
+                Required for "add" and "update" actions.
+            email_address: The email address of the user or group to grant access to.
+                Required for the "add" action.
+            permission_id: The ID of the permission to update or remove.
+                Required for "update" and "remove" actions.
+            transfer_ownership: Whether to transfer ownership of the file.
+                Only applicable when action="update" and role="owner".
+
+        Returns
+        -------
+            Permission id.
+            For "add" its newly added permission.
+            For "update"/"remove" its previous permission.
+
+        Raises
+        ------
+            GoogleDriveError: If the permission operation fails (invalid arguments,
+                insufficient permissions, resource not found, ownership transfer
+                not allowed, rate limited, etc.).
+        """
+        if not file_id.strip():
+            raise GoogleDriveError("Argument validation error: 'file_id' cannot be empty.")
+
+        if action == "add" and not email_address:
+            raise GoogleDriveError("'email_address' is required for action 'add'.")
+
+        if action in ("update", "remove") and not permission_id:
+            raise GoogleDriveError("'permission_id' is required for action 'update' or 'remove'.")
+
+        if action != "remove" and not role:
+            raise GoogleDriveError("'role' is required for action 'add' or 'update'.")
+
+        if action == "add":
+            response = await self._client.post(
+                url=f"/{file_id}/permissions",
+                json={
+                    "type": "user",
+                    "role": role,
+                    "emailAddress": email_address,
+                },
+                params={"sendNotificationEmail": False, "supportsAllDrives": True},
+            )
+
+        elif action == "update":
+            response = await self._client.patch(
+                url=f"/{file_id}/permissions/{permission_id}",
+                json={"role": role},
+                params={"transferOwnership": transfer_ownership, "supportsAllDrives": True},
+            )
+
+        elif action == "remove":
+            response = await self._client.delete(url=f"/{file_id}/permissions/{permission_id}")
+
+        else:
+            raise GoogleDriveError(f"Invalid action '{action}'")
+
+        if response.status_code not in (200, 201, 204):
+            raise GoogleDriveError(f"Drive API error {response.status_code}: {response.text}")
+
+        if action == "add":
+            return response.json()["id"]
+
+        # Cannot be null here because of above validators
+        return permission_id  # type: ignore
+
     async def __aenter__(self) -> "GoogleDriveClient":
         """Async context manager entry."""
         return self

datarobot_genai/drmcp/tools/gdrive/tools.py

@@ -16,6 +16,7 @@
 
 import logging
 from typing import Annotated
+from typing import Literal
 
 from fastmcp.exceptions import ToolError
 from fastmcp.tools.tool import ToolResult
@@ -33,7 +34,9 @@ from datarobot_genai.drmcp.tools.clients.gdrive import get_gdrive_access_token
 logger = logging.getLogger(__name__)
 
 
-@dr_mcp_tool(tags={"google", "gdrive", "list", "search", "files", "find", "contents"})
+@dr_mcp_tool(
+    tags={"google", "gdrive", "list", "search", "files", "find", "contents"}, enabled=False
+)
 async def gdrive_find_contents(
     *,
     page_size: Annotated[
@@ -345,3 +348,99 @@ async def gdrive_update_metadata(
         content=f"Successfully updated file '{updated_file.name}': {changes_description}.",
         structured_content=updated_file.as_flat_dict(),
     )
+
+
+@dr_mcp_tool(tags={"google", "gdrive", "manage", "access", "acl"})
+async def gdrive_manage_access(
+    *,
+    file_id: Annotated[str, "The ID of the file or folder."],
+    action: Annotated[Literal["add", "update", "remove"], "The operation to perform."],
+    role: Annotated[
+        Literal["reader", "commenter", "writer", "fileOrganizer", "organizer", "owner"] | None,
+        "The access level.",
+    ] = None,
+    email_address: Annotated[
+        str | None, "The email of the user or group (required for 'add')."
+    ] = None,
+    permission_id: Annotated[
+        str | None, "The specific permission ID (required for 'update' or 'remove')."
+    ] = None,
+    transfer_ownership: Annotated[
+        bool, "Whether to transfer ownership (only for 'update' to 'owner' role)."
+    ] = False,
+) -> ToolResult:
+    """
+    Consolidated tool for sharing files and managing permissions.
+    Pushes all logic to the Google Drive API permissions resource (create, update, delete).
+
+    Usage:
+        - Add role: gdrive_manage_access(
+            file_id="SomeFileId",
+            action="add",
+            role="reader",
+            email_address="dummy@user.com"
+          )
+        - Update role: gdrive_manage_access(
+            file_id="SomeFileId",
+            action="update",
+            role="reader",
+            permission_id="SomePermissionId"
+          )
+        - Remove permission: gdrive_manage_access(
+            file_id="SomeFileId",
+            action="remove",
+            permission_id="SomePermissionId"
+          )
+    """
+    if not file_id or not file_id.strip():
+        raise ToolError("Argument validation error: 'file_id' cannot be empty.")
+
+    if action == "add" and not email_address:
+        raise ToolError("'email_address' is required for action 'add'.")
+
+    if action in ("update", "remove") and not permission_id:
+        raise ToolError("'permission_id' is required for action 'update' or 'remove'.")
+
+    if action != "remove" and not role:
+        raise ToolError("'role' is required for action 'add' or 'update'.")
+
+    access_token = await get_gdrive_access_token()
+    if isinstance(access_token, ToolError):
+        raise access_token
+
+    try:
+        async with GoogleDriveClient(access_token) as client:
+            permission_id = await client.manage_access(
+                file_id=file_id,
+                action=action,
+                role=role,
+                email_address=email_address,
+                permission_id=permission_id,
+                transfer_ownership=transfer_ownership,
+            )
+    except GoogleDriveError as e:
+        logger.error(f"Google Drive permission operation failed: {e}")
+        raise ToolError(str(e))
+    except Exception as e:
+        logger.error(f"Unexpected error changing permissions for Google Drive file {file_id}: {e}")
+        raise ToolError(
+            f"Unexpected error changing permissions for Google Drive file {file_id}: {str(e)}"
+        )
+
+    # Build response
+    structured_content = {"affectedFileId": file_id}
+    if action == "add":
+        content = (
+            f"Successfully added role '{role}' for '{email_address}' for gdrive file '{file_id}'. "
+            f"New permission id '{permission_id}'."
+        )
+        structured_content["newPermissionId"] = permission_id
+    elif action == "update":
+        content = (
+            f"Successfully updated role '{role}' (permission '{permission_id}') "
+            f"for gdrive file '{file_id}'."
+        )
+    else:  # action == "remove":
+        content = f"Successfully removed permission '{permission_id}' for gdrive file '{file_id}'."
+
+    return ToolResult(content=content, structured_content=structured_content)

datarobot_genai/drmcp/tools/predictive/data.py

@@ -35,9 +35,9 @@ async def upload_dataset_to_ai_catalog(
 ) -> ToolError | ToolResult:
     """Upload a dataset to the DataRobot AI Catalog / Data Registry."""
     if not file_path and not file_url:
-        return ToolError("Either file_path or file_url must be provided.")
+        raise ToolError("Either file_path or file_url must be provided.")
     if file_path and file_url:
-        return ToolError("Please provide either file_path or file_url, not both.")
+        raise ToolError("Please provide either file_path or file_url, not both.")
 
     # Get client
     client = get_sdk_client()
@@ -47,17 +47,17 @@ async def upload_dataset_to_ai_catalog(
         # Does file exist?
         if not os.path.exists(file_path):
             logger.error("File not found: %s", file_path)
-            return ToolError(f"File not found: {file_path}")
+            raise ToolError(f"File not found: {file_path}")
         catalog_item = client.Dataset.create_from_file(file_path)
     else:
         # Does URL exist?
         if file_url is None or not is_valid_url(file_url):
             logger.error("Invalid file URL: %s", file_url)
-            return ToolError(f"Invalid file URL: {file_url}")
+            raise ToolError(f"Invalid file URL: {file_url}")
         catalog_item = client.Dataset.create_from_url(file_url)
 
     if not catalog_item:
-        return ToolError("Failed to upload dataset.")
+        raise ToolError("Failed to upload dataset.")
 
     return ToolResult(
         content=f"Successfully uploaded dataset: {catalog_item.id}",

datarobot_genai/drmcp/tools/predictive/model.py

@@ -14,9 +14,12 @@
 
 import json
 import logging
+from typing import Annotated
 from typing import Any
 
 from datarobot.models.model import Model
+from fastmcp.exceptions import ToolError
+from fastmcp.tools.tool import ToolResult
 
 from datarobot_genai.drmcp.core.clients import get_sdk_client
 from datarobot_genai.drmcp.core.mcp_instance import dr_mcp_tool
@@ -50,33 +53,25 @@ class ModelEncoder(json.JSONEncoder):
         return super().default(obj)
 
 
-@dr_mcp_tool(tags={"model", "management", "info"})
-async def get_best_model(project_id: str, metric: str | None = None) -> str:
-    """
-    Get the best model for a DataRobot project, optionally by a specific metric.
+@dr_mcp_tool(tags={"predictive", "model", "read", "management", "info"})
+async def get_best_model(
+    *,
+    project_id: Annotated[str, "The DataRobot project ID"] | None = None,
+    metric: Annotated[str, "The metric to use for best model selection (e.g., 'AUC', 'LogLoss')"]
+    | None = None,
+) -> ToolError | ToolResult:
+    """Get the best model for a DataRobot project, optionally by a specific metric."""
+    if not project_id:
+        raise ToolError("Project ID must be provided")
 
-    Args:
-        project_id: The ID of the DataRobot project.
-        metric: (Optional) The metric to use for best model selection (e.g., 'AUC', 'LogLoss').
-
-    Returns
-    -------
-        A formatted string describing the best model.
-
-    Raises
-    ------
-        Exception: If project not found or no models exist in the project.
-    """
     client = get_sdk_client()
     project = client.Project.get(project_id)
     if not project:
-        logger.error(f"Project with ID {project_id} not found")
-        raise Exception(f"Project with ID {project_id} not found.")
+        raise ToolError(f"Project with ID {project_id} not found.")
 
     leaderboard = project.get_models()
     if not leaderboard:
-        logger.info(f"No models found for project {project_id}")
-        raise Exception("No models found for this project.")
+        raise ToolError("No models found for this project.")
 
     if metric:
         reverse_sort = metric.upper() in [
@@ -98,51 +93,91 @@ async def get_best_model(project_id: str, metric: str | None = None) -> str:
     best_model = leaderboard[0]
     logger.info(f"Found best model {best_model.id} for project {project_id}")
 
-    # Format the response as a human-readable string
     metric_info = ""
+    metric_value = None
+
     if metric and best_model.metrics and metric in best_model.metrics:
         metric_value = best_model.metrics[metric].get("validation")
         if metric_value is not None:
             metric_info = f" with {metric}: {metric_value:.2f}"
 
-    return f"Best model: {best_model.model_type}{metric_info}"
-
-
-@dr_mcp_tool(tags={"model", "prediction", "scoring"})
-async def score_dataset_with_model(project_id: str, model_id: str, dataset_url: str) -> str:
-    """
-    Score a dataset using a specific DataRobot model.
+    # Include full metrics in the response
+    best_model_dict = model_to_dict(best_model)
+    best_model_dict["metric"] = metric
+    best_model_dict["metric_value"] = metric_value
+
+    # Format metrics for human-readable content
+    metrics_text = ""
+    if best_model.metrics:
+        metrics_list = []
+        for metric_name, metric_data in best_model.metrics.items():
+            if isinstance(metric_data, dict) and "validation" in metric_data:
+                val = metric_data["validation"]
+                if val is not None:
+                    metrics_list.append(f"{metric_name}: {val:.4f}")
+        if metrics_list:
+            metrics_text = "\nPerformance metrics:\n" + "\n".join(f"  - {m}" for m in metrics_list)
+
+    return ToolResult(
+        content=f"Best model: {best_model.model_type}{metric_info}{metrics_text}",
+        structured_content={
+            "project_id": project_id,
+            "best_model": best_model_dict,
+        },
+    )
+
+
+@dr_mcp_tool(tags={"predictive", "model", "read", "scoring", "dataset"})
+async def score_dataset_with_model(
+    *,
+    project_id: Annotated[str, "The DataRobot project ID"] | None = None,
+    model_id: Annotated[str, "The DataRobot model ID"] | None = None,
+    dataset_url: Annotated[str, "The dataset URL"] | None = None,
+) -> ToolError | ToolResult:
+    """Score a dataset using a specific DataRobot model."""
+    if not project_id:
+        raise ToolError("Project ID must be provided")
+    if not model_id:
+        raise ToolError("Model ID must be provided")
+    if not dataset_url:
+        raise ToolError("Dataset URL must be provided")
 
-    Args:
-        project_id: The ID of the DataRobot project.
-        model_id: The ID of the DataRobot model to use for scoring.
-        dataset_url: The URL to the dataset to score (must be accessible to DataRobot).
-
-    Returns
-    -------
-        A string summary of the scoring job or a meaningful error message.
-    """
     client = get_sdk_client()
     project = client.Project.get(project_id)
     model = client.Model.get(project, model_id)
     job = model.score(dataset_url)
-    logger.info(f"Started scoring job {job.id} for model {model_id}")
-    return f"Scoring job started: {job.id}"
-
 
-@dr_mcp_tool(tags={"model", "management", "list"})
-async def list_models(project_id: str) -> str:
-    """
-    List all models in a project.
+    return ToolResult(
+        content=f"Scoring job started: {job.id}",
+        structured_content={
+            "scoring_job_id": job.id,
+            "project_id": project_id,
+            "model_id": model_id,
+            "dataset_url": dataset_url,
+        },
+    )
+
+
+@dr_mcp_tool(tags={"predictive", "model", "read", "management", "list"})
+async def list_models(
+    *,
+    project_id: Annotated[str, "The DataRobot project ID"] | None = None,
+) -> ToolError | ToolResult:
+    """List all models in a project."""
+    if not project_id:
+        raise ToolError("Project ID must be provided")
 
-    Args:
-        project_id: The ID of the DataRobot project.
-
-    Returns
-    -------
-        A string summary of the models in the project.
-    """
     client = get_sdk_client()
     project = client.Project.get(project_id)
     models = project.get_models()
-    return json.dumps(models, indent=2, cls=ModelEncoder)
+
+    return ToolResult(
+        content=(
+            f"Found {len(models)} models in project {project_id}, here are the details:\n"
+            f"{json.dumps(models, indent=2, cls=ModelEncoder)}"
+        ),
+        structured_content={
+            "project_id": project_id,
+            "models": [model_to_dict(model) for model in models],
+        },
+    )

datarobot_genai/drmcp/tools/predictive/project.py

@@ -54,9 +54,9 @@ async def get_project_dataset_by_name(
     The dataset ID and the dataset type (source or prediction) as a string, or an error message.
     """
     if not project_id:
-        return ToolError("Project ID is required.")
+        raise ToolError("Project ID is required.")
     if not dataset_name:
-        return ToolError("Dataset name is required.")
+        raise ToolError("Dataset name is required.")
 
     client = get_sdk_client()
     project = client.Project.get(project_id)

datarobot_genai/drmcp/tools/predictive/training.py

@@ -63,7 +63,7 @@ async def analyze_dataset(
 ) -> ToolError | ToolResult:
     """Analyze a dataset to understand its structure and potential use cases."""
     if not dataset_id:
-        return ToolError("Dataset ID must be provided")
+        raise ToolError("Dataset ID must be provided")
 
     client = get_sdk_client()
     dataset = client.Dataset.get(dataset_id)
@@ -116,7 +116,7 @@ async def suggest_use_cases(
 ) -> ToolError | ToolResult:
     """Analyze a dataset and suggest potential machine learning use cases."""
     if not dataset_id:
-        return ToolError("Dataset ID must be provided")
+        raise ToolError("Dataset ID must be provided")
 
     client = get_sdk_client()
     dataset = client.Dataset.get(dataset_id)
@@ -148,7 +148,7 @@ async def get_exploratory_insights(
 ) -> ToolError | ToolResult:
     """Generate exploratory data insights for a dataset."""
     if not dataset_id:
-        return ToolError("Dataset ID must be provided")
+        raise ToolError("Dataset ID must be provided")
 
     client = get_sdk_client()
     dataset = client.Dataset.get(dataset_id)
@@ -481,9 +481,9 @@ async def start_autopilot(
 
     if not project_id:
         if not dataset_url and not dataset_id:
-            return ToolError("Either dataset_url or dataset_id must be provided")
+            raise ToolError("Either dataset_url or dataset_id must be provided")
         if dataset_url and dataset_id:
-            return ToolError("Please provide either dataset_url or dataset_id, not both")
+            raise ToolError("Please provide either dataset_url or dataset_id, not both")
 
         if dataset_url:
             dataset = client.Dataset.create_from_url(dataset_url)
@@ -497,7 +497,7 @@ async def start_autopilot(
         project = client.Project.get(project_id)
 
     if not target:
-        return ToolError("Target variable must be specified")
+        raise ToolError("Target variable must be specified")
 
     try:
         # Start modeling
@@ -517,7 +517,7 @@ async def start_autopilot(
         )
 
     except Exception as e:
-        return ToolError(
+        raise ToolError(
             content=json.dumps(
                 {
                     "error": f"Failed to start Autopilot: {str(e)}",
@@ -546,9 +546,9 @@ async def get_model_roc_curve(
 ) -> ToolError | ToolResult:
     """Get detailed ROC curve for a specific model."""
     if not project_id:
-        return ToolError("Project ID must be provided")
+        raise ToolError("Project ID must be provided")
     if not model_id:
-        return ToolError("Model ID must be provided")
+        raise ToolError("Model ID must be provided")
 
     client = get_sdk_client()
     project = client.Project.get(project_id)
@@ -587,7 +587,7 @@ async def get_model_roc_curve(
             structured_content={"data": roc_data},
         )
     except Exception as e:
-        return ToolError(f"Failed to get ROC curve: {str(e)}")
+        raise ToolError(f"Failed to get ROC curve: {str(e)}")
 
 
 @dr_mcp_tool(tags={"predictive", "training", "read", "model", "evaluation"})
@@ -598,9 +598,9 @@ async def get_model_feature_impact(
 ) -> ToolError | ToolResult:
     """Get detailed feature impact for a specific model."""
     if not project_id:
-        return ToolError("Project ID must be provided")
+        raise ToolError("Project ID must be provided")
     if not model_id:
-        return ToolError("Model ID must be provided")
+        raise ToolError("Model ID must be provided")
 
     client = get_sdk_client()
     project = client.Project.get(project_id)
@@ -631,9 +631,9 @@ async def get_model_lift_chart(
 ) -> ToolError | ToolResult:
     """Get detailed lift chart for a specific model."""
     if not project_id:
-        return ToolError("Project ID must be provided")
+        raise ToolError("Project ID must be provided")
     if not model_id:
-        return ToolError("Model ID must be provided")
+        raise ToolError("Model ID must be provided")
 
     client = get_sdk_client()
     project = client.Project.get(project_id)

{datarobot_genai-0.2.31.dist-info → datarobot_genai-0.2.32.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: datarobot-genai
-Version: 0.2.31
+Version: 0.2.32
 Summary: Generic helpers for GenAI
 Project-URL: Homepage, https://github.com/datarobot-oss/datarobot-genai
 Author: DataRobot, Inc.

{datarobot_genai-0.2.31.dist-info → datarobot_genai-0.2.32.dist-info}/RECORD

@@ -33,8 +33,8 @@ datarobot_genai/drmcp/core/constants.py,sha256=lUwoW_PTrbaBGqRJifKqCn3EoFacoEgdO
 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/exceptions.py,sha256=9zoNh5ph6QihWIYuw37ljZ73_iUfy38YVYyFSnEwivc,839
+datarobot_genai/drmcp/core/logging.py,sha256=rnUkws0vIDy_uLevwNj-wgA9uijW1Go774JPCrG0Yfw,3423
 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
@@ -75,34 +75,34 @@ datarobot_genai/drmcp/test_utils/tool_base_ete.py,sha256=2RvVmwHYczl7F6qZHkKYiI7
 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/base.py,sha256=WoPdddYmmXGylEuKRtKHPfprcHMjbHqPB9PwzWmORV4,10637
 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/gdrive.py,sha256=RK4IISpYb99aK6WgDthesDoglaZxwGpG_PPAAe6xsVM,33064
 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/gdrive/tools.py,sha256=7bNrp7E3opKwsBDYfLIOsOGfPXW-Ae9KvcimEzetR0A,17631
 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/data.py,sha256=VbGs8ERP8vNFtTTryGhI61JItNVaJsx1gxpRX1ZFZcg,4626
 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/model.py,sha256=BVxOMHh3--liwBU4VB1OWRrqkhJ4y_Rq053f7y94TF8,6276
 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/drmcp/tools/predictive/project.py,sha256=Mzf7rQogBV6h1-MWQYTwtDHOsMWfjOyyJpSYmmvNNuc,3253
+datarobot_genai/drmcp/tools/predictive/training.py,sha256=WWzzGibYMSvI8kqHnvav6qNIVjoe1EG4RyiYa3XhFYA,23984
 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
@@ -117,9 +117,9 @@ datarobot_genai/nat/datarobot_llm_clients.py,sha256=-_q_KlKOVQecIYJd8YRiYnS4ZNaz
 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.32.dist-info/METADATA,sha256=I5_bDUsP1rsNCYVIXvZjYYb91UwhA7N51E_Y0EN-fSc,6301
+datarobot_genai-0.2.32.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+datarobot_genai-0.2.32.dist-info/entry_points.txt,sha256=jEW3WxDZ8XIK9-ISmTyt5DbmBb047rFlzQuhY09rGrM,284
+datarobot_genai-0.2.32.dist-info/licenses/AUTHORS,sha256=isJGUXdjq1U7XZ_B_9AH8Qf0u4eX0XyQifJZ_Sxm4sA,80
+datarobot_genai-0.2.32.dist-info/licenses/LICENSE,sha256=U2_VkLIktQoa60Nf6Tbt7E4RMlfhFSjWjcJJfVC-YCE,11341
+datarobot_genai-0.2.32.dist-info/RECORD,,