datarobot-genai 0.2.44__tar.gz → 0.3.0__tar.gz

{datarobot_genai-0.2.44 → datarobot_genai-0.3.0}/PKG-INFO

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

{datarobot_genai-0.2.44 → datarobot_genai-0.3.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "datarobot-genai"
-version = "0.2.44"
+version = "0.3.0"
 description = "Generic helpers for GenAI"
 readme = "README.md"
 requires-python = ">=3.10, <3.13"
@@ -85,6 +85,7 @@ drmcp = [
   "boto3>=1.34.0,<2.0.0",
   "httpx>=0.28.1,<1.0.0",
   "tavily-python>=0.7.20,<1.0.0",
+  "perplexityai>=0.27,<1.0",
   "pydantic>=2.6.1,<3.0.0",
   "pydantic-settings>=2.1.0,<3.0.0",
   "opentelemetry-api>=1.22.0,<2.0.0",

{datarobot_genai-0.2.44 → datarobot_genai-0.3.0}/src/datarobot_genai/core/agents/__init__.py

@@ -17,7 +17,7 @@ This package provides:
 - BaseAgent: common initialization for agent env/config fields
 - Common helpers: make_system_prompt, extract_user_prompt_content
 - Framework utilities (optional extras):
-  - crewai: build_llm, create_pipeline_interactions_from_messages
+  - crewai: create_pipeline_interactions_from_messages
   - langgraph: create_pipeline_interactions_from_events
   - llamaindex: DataRobotLiteLLM, create_pipeline_interactions_from_events
 """

{datarobot_genai-0.2.44 → datarobot_genai-0.3.0}/src/datarobot_genai/crewai/__init__.py

@@ -2,22 +2,19 @@
 
 Public API:
 - mcp_tools_context: Context manager returning available MCP tools for CrewAI.
-- build_llm: Construct a CrewAI LLM configured for DataRobot endpoints.
 - create_pipeline_interactions_from_messages: Convert messages to MultiTurnSample.
 """
 
 from datarobot_genai.core.mcp.common import MCPConfig
 
-from .agent import build_llm
+from .agent import CrewAIAgent
 from .agent import create_pipeline_interactions_from_messages
-from .base import CrewAIAgent
 from .events import CrewAIEventListener
 from .mcp import mcp_tools_context
 
 __all__ = [
     "mcp_tools_context",
     "CrewAIAgent",
-    "build_llm",
     "create_pipeline_interactions_from_messages",
     "CrewAIEventListener",
     "MCPConfig",

datarobot_genai-0.2.44/src/datarobot_genai/crewai/base.py → datarobot_genai-0.3.0/src/datarobot_genai/crewai/agent.py

@@ -41,11 +41,24 @@ from datarobot_genai.core.agents.base import default_usage_metrics
 from datarobot_genai.core.agents.base import extract_user_prompt_content
 from datarobot_genai.core.agents.base import is_streaming
 
-from .agent import create_pipeline_interactions_from_messages
 from .mcp import mcp_tools_context
 
 if TYPE_CHECKING:
     from ragas import MultiTurnSample
+    from ragas.messages import AIMessage
+    from ragas.messages import HumanMessage
+    from ragas.messages import ToolMessage
+
+
+def create_pipeline_interactions_from_messages(
+    messages: list[HumanMessage | AIMessage | ToolMessage] | None,
+) -> MultiTurnSample | None:
+    if not messages:
+        return None
+    # Lazy import to reduce memory overhead when ragas is not used
+    from ragas import MultiTurnSample
+
+    return MultiTurnSample(user_input=messages)
 
 
 class CrewAIAgent(BaseAgent[BaseTool], abc.ABC):

{datarobot_genai-0.2.44 → datarobot_genai-0.3.0}/src/datarobot_genai/drmcp/core/config.py

@@ -76,7 +76,16 @@ class MCPToolConfig(BaseSettings):
             RUNTIME_PARAM_ENV_VAR_NAME_PREFIX + "ENABLE_MICROSOFT_GRAPH_TOOLS",
             "ENABLE_MICROSOFT_GRAPH_TOOLS",
         ),
-        description="Enable/disable Sharepoint tools",
+        description="Enable/disable Microsoft Graph (Sharepoint/OneDrive) tools",
+    )
+
+    enable_perplexity_tools: bool = Field(
+        default=False,
+        validation_alias=AliasChoices(
+            RUNTIME_PARAM_ENV_VAR_NAME_PREFIX + "ENABLE_PERPLEXITY_TOOLS",
+            "ENABLE_PERPLEXITY_TOOLS",
+        ),
+        description="Enable/disable Perplexity tools",
     )
 
     enable_tavily_tools: bool = Field(
@@ -140,6 +149,7 @@ class MCPToolConfig(BaseSettings):
         "enable_confluence_tools",
         "enable_gdrive_tools",
         "enable_microsoft_graph_tools",
+        "enable_perplexity_tools",
         "enable_tavily_tools",
         "is_atlassian_oauth_provider_configured",
         "is_google_oauth_provider_configured",

{datarobot_genai-0.2.44 → datarobot_genai-0.3.0}/src/datarobot_genai/drmcp/core/tool_config.py

@@ -31,6 +31,7 @@ class ToolType(str, Enum):
     CONFLUENCE = "confluence"
     GDRIVE = "gdrive"
     MICROSOFT_GRAPH = "microsoft_graph"
+    PERPLEXITY = "perplexity"
     TAVILY = "tavily"
 
 
@@ -81,6 +82,13 @@ TOOL_CONFIGS: dict[ToolType, ToolConfig] = {
         package_prefix="datarobot_genai.drmcp.tools.microsoft_graph",
         config_field_name="enable_microsoft_graph_tools",
     ),
+    ToolType.PERPLEXITY: ToolConfig(
+        name="perplexity",
+        oauth_check=None,  # OAuth for Perplexity is not supported
+        directory="perplexity",
+        package_prefix="datarobot_genai.drmcp.tools.perplexity",
+        config_field_name="enable_perplexity_tools",
+    ),
     ToolType.TAVILY: ToolConfig(
         name="tavily",
         oauth_check=None,

datarobot_genai-0.3.0/src/datarobot_genai/drmcp/tools/clients/perplexity.py

@@ -0,0 +1,173 @@
+# Copyright 2025 DataRobot, Inc.
+#
+# 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.
+
+import logging
+from typing import Any
+from typing import Literal
+
+from fastmcp.exceptions import ToolError
+from fastmcp.server.dependencies import get_http_headers
+from perplexity import AsyncPerplexity
+from perplexity.types import search_create_response
+from pydantic import BaseModel
+from pydantic import ConfigDict
+
+logger = logging.getLogger(__name__)
+
+MAX_QUERIES: int = 5
+MAX_RESULTS: int = 20
+MAX_TOKENS_PER_PAGE: int = 8192
+MAX_SEARCH_DOMAIN_FILTER: int = 20
+
+MAX_RESULTS_DEFAULT: int = 10
+MAX_TOKENS_PER_PAGE_DEFAULT: int = 2048
+
+
+async def get_perplexity_access_token() -> str | ToolError:
+    """
+    Get Perplexity API key from HTTP headers.
+
+    At the moment of creating this fn. Perplexity does not support OAuth.
+    It allows only API-KEY authorized flow.
+
+    Returns
+    -------
+        Access token string on success, ToolError on failure
+
+    Example:
+        ```python
+        token = await get_perplexity_access_token()
+        if isinstance(token, ToolError):
+            # Handle error
+            return token
+        # Use token
+        ```
+    """
+    try:
+        headers = get_http_headers()
+
+        if api_key := headers.get("x-perplexity-api-key"):
+            return api_key
+
+        logger.warning("Perplexity API key not found in headers.")
+        return ToolError(
+            "Perplexity API key not found in headers. "
+            "Please provide it via 'x-perplexity-api-key' header."
+        )
+    except Exception as e:
+        logger.error(f"Unexpected error obtaining Perplexity API key: {e}.", exc_info=e)
+        return ToolError("An unexpected error occured while obtaining Perplexity API key.")
+
+
+class PerplexityError(Exception):
+    """Exception for Perplexity API errors."""
+
+    def __init__(self, message: str) -> None:
+        super().__init__(message)
+
+
+class PerplexitySearchResult(BaseModel):
+    snippet: str
+    title: str
+    url: str
+    date: str | None = None
+    last_updated: str | None = None
+
+    model_config = ConfigDict(populate_by_name=True)
+
+    @classmethod
+    def from_perplexity_sdk(cls, result: search_create_response.Result) -> "PerplexitySearchResult":
+        """Create a PerplexitySearchResult from perplexity sdk response data."""
+        return cls(**result.model_dump())
+
+    def as_flat_dict(self) -> dict[str, Any]:
+        """Return a flat dictionary representation of the search result."""
+        return self.model_dump(by_alias=True)
+
+
+class PerplexityClient:
+    """Client for interacting with Perplexity API.
+    Its simple wrapper around perplexity python sdk.
+    """
+
+    def __init__(self, access_token: str) -> None:
+        self._client = AsyncPerplexity(api_key=access_token)
+
+    async def search(
+        self,
+        query: str | list[str],
+        search_domain_filter: list[str] | None = None,
+        recency: Literal["hour", "day", "week", "month", "year"] | None = None,
+        max_results: int = MAX_RESULTS_DEFAULT,
+        max_tokens_per_page: int = MAX_TOKENS_PER_PAGE_DEFAULT,
+    ) -> list[PerplexitySearchResult]:
+        """
+        Search using Perplexity.
+
+        Args:
+            query: Query to filter results.
+            search_domain_filter: Up to 20 domains/URLs to allowlist or denylist.
+            recency: Filter results by time period.
+            max_results: Number of ranked results to return.
+            max_tokens_per_page: Context extraction cap per page.
+
+        Returns
+        -------
+            List of Perplexity search results.
+        """
+        if not query:
+            raise PerplexityError("Error: query cannot be empty.")
+        if query and isinstance(query, str) and not query.strip():
+            raise PerplexityError("Error: query cannot be empty.")
+        if query and isinstance(query, list) and len(query) > MAX_QUERIES:
+            raise PerplexityError(f"Error: query list cannot be bigger than {MAX_QUERIES}.")
+        if query and isinstance(query, list) and not all(q.strip() for q in query):
+            raise PerplexityError("Error: query cannot contain empty str.")
+        if search_domain_filter and len(search_domain_filter) > MAX_SEARCH_DOMAIN_FILTER:
+            raise PerplexityError("Error: maximum number of search domain filters is 20.")
+        if max_results <= 0:
+            raise PerplexityError("Error: max_results must be greater than 0.")
+        if max_results > MAX_RESULTS:
+            raise PerplexityError("Error: max_results must be smaller than or equal to 20.")
+        if max_tokens_per_page <= 0:
+            raise PerplexityError("Error: max_tokens_per_page must be greater than 0.")
+        if max_tokens_per_page > MAX_TOKENS_PER_PAGE:
+            raise PerplexityError(
+                "Error: max_tokens_per_page must be smaller than or equal to 8192."
+            )
+
+        max_results = min(max_results, MAX_RESULTS)
+        max_tokens_per_page = min(max_tokens_per_page, MAX_TOKENS_PER_PAGE)
+
+        search_result = await self._client.search.create(
+            query=query,
+            search_domain_filter=search_domain_filter,
+            search_recency_filter=recency,
+            max_results=max_results,
+            max_tokens_per_page=max_tokens_per_page,
+        )
+
+        return [
+            PerplexitySearchResult.from_perplexity_sdk(result) for result in search_result.results
+        ]
+
+    async def __aenter__(self) -> "PerplexityClient":
+        """Async context manager entry."""
+        return self
+
+    async def __aexit__(
+        self, exc_type: type[BaseException] | None, exc_val: BaseException | None, exc_tb: Any
+    ) -> None:
+        """Async context manager exit."""
+        await self._client.close()

datarobot_genai-0.3.0/src/datarobot_genai/drmcp/tools/perplexity/tools.py

@@ -0,0 +1,121 @@
+# Copyright 2026 DataRobot, Inc.
+#
+# 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.
+
+"""Perplexity MCP tools."""
+
+import logging
+from typing import Annotated
+from typing import Literal
+
+from fastmcp.exceptions import ToolError
+from fastmcp.tools.tool import ToolResult
+
+from datarobot_genai.drmcp.core.mcp_instance import dr_mcp_tool
+from datarobot_genai.drmcp.tools.clients.perplexity import MAX_QUERIES
+from datarobot_genai.drmcp.tools.clients.perplexity import MAX_RESULTS
+from datarobot_genai.drmcp.tools.clients.perplexity import MAX_RESULTS_DEFAULT
+from datarobot_genai.drmcp.tools.clients.perplexity import MAX_SEARCH_DOMAIN_FILTER
+from datarobot_genai.drmcp.tools.clients.perplexity import MAX_TOKENS_PER_PAGE
+from datarobot_genai.drmcp.tools.clients.perplexity import MAX_TOKENS_PER_PAGE_DEFAULT
+from datarobot_genai.drmcp.tools.clients.perplexity import PerplexityClient
+from datarobot_genai.drmcp.tools.clients.perplexity import get_perplexity_access_token
+
+logger = logging.getLogger(__name__)
+
+
+@dr_mcp_tool(tags={"perplexity", "web", "search", "websearch"})
+async def perplexity_search(
+    *,
+    query: Annotated[
+        str,
+        list[str],
+        f"The search query string OR "
+        f"a list of up to {MAX_QUERIES} sub-queries for multi-query research.",
+    ],
+    search_domain_filter: Annotated[
+        list[str] | None,
+        f"Up to {MAX_SEARCH_DOMAIN_FILTER} domains/URLs "
+        f"to allowlist or denylist (prefix with '-').",
+    ] = None,
+    recency: Annotated[
+        Literal["day", "week", "month", "year"] | None, "Filter results by time period."
+    ] = None,
+    max_results: Annotated[
+        int, f"Number of ranked results to return (1-{MAX_RESULTS})."
+    ] = MAX_RESULTS_DEFAULT,
+    max_tokens_per_page: Annotated[
+        int,
+        f"Content extraction cap per page (1-{MAX_TOKENS_PER_PAGE}) "
+        f"(default {MAX_TOKENS_PER_PAGE_DEFAULT}).",
+    ] = MAX_TOKENS_PER_PAGE_DEFAULT,
+) -> ToolResult:
+    """Perplexity web search tool combining multi-query research and content extraction control."""
+    if not query:
+        raise ToolError("Argument validation error: query cannot be empty.")
+    if query and isinstance(query, str) and not query.strip():
+        raise ToolError("Argument validation error: query cannot be empty.")
+    if query and isinstance(query, list) and len(query) > MAX_QUERIES:
+        raise ToolError(
+            f"Argument validation error: query list cannot be bigger than {MAX_QUERIES}."
+        )
+    if query and isinstance(query, list) and not all(q.strip() for q in query):
+        raise ToolError("Argument validation error: query cannot contain empty str.")
+    if search_domain_filter and len(search_domain_filter) > MAX_SEARCH_DOMAIN_FILTER:
+        raise ToolError(
+            f"Argument validation error: "
+            f"maximum number of search domain filters is {MAX_SEARCH_DOMAIN_FILTER}."
+        )
+    if max_results <= 0:
+        raise ToolError("Argument validation error: max_results must be greater than 0.")
+    if max_results > MAX_RESULTS:
+        raise ToolError(
+            f"Argument validation error: "
+            f"max_results must be smaller than or equal to {MAX_RESULTS}."
+        )
+    if max_tokens_per_page <= 0:
+        raise ToolError("Argument validation error: max_tokens_per_page must be greater than 0.")
+    if max_tokens_per_page > MAX_TOKENS_PER_PAGE:
+        raise ToolError(
+            f"Argument validation error: "
+            f"max_tokens_per_page must be smaller than or equal to {MAX_TOKENS_PER_PAGE}."
+        )
+
+    access_token = await get_perplexity_access_token()
+    if isinstance(access_token, ToolError):
+        raise access_token
+
+    async with PerplexityClient(access_token=access_token) as perplexity_client:
+        results = await perplexity_client.search(
+            query=query,
+            search_domain_filter=search_domain_filter,
+            recency=recency,
+            max_results=max_results,
+            max_tokens_per_page=max_tokens_per_page,
+        )
+
+    query_txt = f"query '{query}'" if isinstance(query, str) else f"queries '{', '.join(query)}'"
+    n = len(results)
+
+    return ToolResult(
+        content=f"Successfully executed search for {query_txt}. Found {n} result(s).",
+        structured_content={
+            "results": results,
+            "count": n,
+            "metadata": {
+                "queriesExecuted": len(query) if isinstance(query, list) else 1,
+                "filtersApplied": {"domains": search_domain_filter, "recency": recency},
+                "extractionLimit": max_tokens_per_page,
+            },
+        },
+    )

{datarobot_genai-0.2.44 → datarobot_genai-0.3.0}/src/datarobot_genai/llama_index/__init__.py

@@ -3,8 +3,8 @@
 from datarobot_genai.core.mcp.common import MCPConfig
 
 from .agent import DataRobotLiteLLM
+from .agent import LlamaIndexAgent
 from .agent import create_pipeline_interactions_from_events
-from .base import LlamaIndexAgent
 from .mcp import load_mcp_tools
 
 __all__ = [

datarobot_genai-0.2.44/src/datarobot_genai/llama_index/base.py → datarobot_genai-0.3.0/src/datarobot_genai/llama_index/agent.py

@@ -11,15 +11,6 @@
 # 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.
-
-"""
-Base class for LlamaIndex-based agents.
-
-Provides a standard ``invoke`` that runs an AgentWorkflow, collects events,
-and converts them into pipeline interactions. Subclasses provide the workflow
-and response extraction logic.
-"""
-
 from __future__ import annotations
 
 import abc
@@ -27,8 +18,12 @@ import inspect
 from collections.abc import AsyncGenerator
 from typing import TYPE_CHECKING
 from typing import Any
+from typing import cast
 
+from llama_index.core.base.llms.types import LLMMetadata
 from llama_index.core.tools import BaseTool
+from llama_index.core.workflow import Event
+from llama_index.llms.litellm import LiteLLM
 from openai.types.chat import CompletionCreateParams
 
 from datarobot_genai.core.agents.base import BaseAgent
@@ -38,13 +33,45 @@ from datarobot_genai.core.agents.base import default_usage_metrics
 from datarobot_genai.core.agents.base import extract_user_prompt_content
 from datarobot_genai.core.agents.base import is_streaming
 
-from .agent import create_pipeline_interactions_from_events
 from .mcp import load_mcp_tools
 
 if TYPE_CHECKING:
     from ragas import MultiTurnSample
 
 
+class DataRobotLiteLLM(LiteLLM):
+    """LiteLLM wrapper providing chat/function capability metadata for LlamaIndex."""
+
+    @property
+    def metadata(self) -> LLMMetadata:
+        """Return LLM metadata."""
+        return LLMMetadata(
+            context_window=128000,
+            num_output=self.max_tokens or -1,
+            is_chat_model=True,
+            is_function_calling_model=True,
+            model_name=self.model,
+        )
+
+
+def create_pipeline_interactions_from_events(
+    events: list[Event] | None,
+) -> MultiTurnSample | None:
+    if not events:
+        return None
+    # Lazy import to reduce memory overhead when ragas is not used
+    from ragas import MultiTurnSample
+    from ragas.integrations.llama_index import convert_to_ragas_messages
+    from ragas.messages import AIMessage
+    from ragas.messages import HumanMessage
+    from ragas.messages import ToolMessage
+
+    # convert_to_ragas_messages expects a list[Event]
+    ragas_trace = convert_to_ragas_messages(list(events))
+    ragas_messages = cast(list[HumanMessage | AIMessage | ToolMessage], ragas_trace)
+    return MultiTurnSample(user_input=ragas_messages)
+
+
 class LlamaIndexAgent(BaseAgent[BaseTool], abc.ABC):
     """Abstract base agent for LlamaIndex workflows."""
 

datarobot_genai-0.2.44/src/datarobot_genai/crewai/agent.py

@@ -1,50 +0,0 @@
-# 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
-
-from typing import TYPE_CHECKING
-
-from crewai import LLM
-
-from datarobot_genai.core.utils.urls import get_api_base
-
-if TYPE_CHECKING:
-    from ragas import MultiTurnSample
-    from ragas.messages import AIMessage
-    from ragas.messages import HumanMessage
-    from ragas.messages import ToolMessage
-
-
-def build_llm(
-    *,
-    api_base: str,
-    api_key: str | None,
-    model: str,
-    deployment_id: str | None,
-    timeout: int,
-) -> LLM:
-    """Create a CrewAI LLM configured for DataRobot LLM Gateway or deployment."""
-    base = get_api_base(api_base, deployment_id)
-    return LLM(model=model, api_base=base, api_key=api_key, timeout=timeout)
-
-
-def create_pipeline_interactions_from_messages(
-    messages: list[HumanMessage | AIMessage | ToolMessage] | None,
-) -> MultiTurnSample | None:
-    if not messages:
-        return None
-    # Lazy import to reduce memory overhead when ragas is not used
-    from ragas import MultiTurnSample
-
-    return MultiTurnSample(user_input=messages)

datarobot_genai-0.2.44/src/datarobot_genai/llama_index/agent.py

@@ -1,57 +0,0 @@
-# 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
-
-from typing import TYPE_CHECKING
-from typing import cast
-
-from llama_index.core.base.llms.types import LLMMetadata
-from llama_index.core.workflow import Event
-from llama_index.llms.litellm import LiteLLM
-
-if TYPE_CHECKING:
-    from ragas import MultiTurnSample
-
-
-class DataRobotLiteLLM(LiteLLM):
-    """LiteLLM wrapper providing chat/function capability metadata for LlamaIndex."""
-
-    @property
-    def metadata(self) -> LLMMetadata:
-        """Return LLM metadata."""
-        return LLMMetadata(
-            context_window=128000,
-            num_output=self.max_tokens or -1,
-            is_chat_model=True,
-            is_function_calling_model=True,
-            model_name=self.model,
-        )
-
-
-def create_pipeline_interactions_from_events(
-    events: list[Event] | None,
-) -> MultiTurnSample | None:
-    if not events:
-        return None
-    # Lazy import to reduce memory overhead when ragas is not used
-    from ragas import MultiTurnSample
-    from ragas.integrations.llama_index import convert_to_ragas_messages
-    from ragas.messages import AIMessage
-    from ragas.messages import HumanMessage
-    from ragas.messages import ToolMessage
-
-    # convert_to_ragas_messages expects a list[Event]
-    ragas_trace = convert_to_ragas_messages(list(events))
-    ragas_messages = cast(list[HumanMessage | AIMessage | ToolMessage], ragas_trace)
-    return MultiTurnSample(user_input=ragas_messages)