[BUZZOK-26356] Synchronize Genai Agents image with af-component-agent (#1514)

* Synchronize Genai Agents image with af-component-agent

* Update

* Lint

* Bump requirements.txt

* Bump version

* Reconcile dependencies, updated IDs, tags

* Unpin chainguard

* Update run_agent.py

---------

Co-authored-by: svc-harness-git2 <[email protected]>

Author: mjnitz02

public_dropin_environments/python311_genai_agents/Dockerfile

@@ -14,7 +14,7 @@ ARG UNAME=notebooks
 ARG UID=10101
 ARG GID=10101
 
-FROM datarobotdev/mirror_chainguard_datarobot.com_python-fips:3.11-dev-drbuild-1328 AS base
+FROM datarobotdev/mirror_chainguard_datarobot.com_python-fips:3.11-dev AS base
 
 ARG UNAME
 ARG UID
@@ -36,7 +36,9 @@ ENV PYTHONUNBUFFERED=1 \
     PYTHONDONTWRITEBYTECODE=1 \
     VENV_PATH=${VENV_PATH} \
     PIP_NO_CACHE_DIR=1 \
-    NOTEBOOKS_KERNEL="python"
+    NOTEBOOKS_KERNEL="python" \
+    DEEPEVAL_HOME=/tmp/.deepeval \
+    DEEPEVAL_TELEMETRY_OPT_OUT="YES"
 
 ENV PATH="$VENV_PATH/bin:$PATH" \
   PYTHONPATH="/home/notebooks/.ipython/extensions:/home/notebooks/storage"
@@ -98,7 +100,7 @@ RUN source ${VENV_PATH}/bin/activate && \
     uv pip install --no-cache-dir -r ${WORKDIR}/requirements-agent.txt && rm ${WORKDIR}/requirements-agent.txt && \
     # Generative AI Agents
     uv pip uninstall litellm && \
-    uv pip install --no-cache-dir git+https://github.com/BerriAI/litellm@v1.72.1.dev1
+    uv pip install --no-cache-dir litellm>=1.72.1
 
 # Copy agent runtime into work directory
 COPY ./run_agent.py ${WORKDIR}/

public_dropin_environments/python311_genai_agents/Dockerfile.local

@@ -76,7 +76,9 @@ ENV PYTHONUNBUFFERED=1 \
     PYTHONDONTWRITEBYTECODE=1 \
     VENV_PATH=${VENV_PATH} \
     PIP_NO_CACHE_DIR=1 \
-    NOTEBOOKS_KERNEL="python"
+    NOTEBOOKS_KERNEL="python" \
+    DEEPEVAL_HOME=/tmp/.deepeval \
+    DEEPEVAL_TELEMETRY_OPT_OUT="YES"
 
 ENV PATH="$VENV_PATH/bin:$PATH" \
   PYTHONPATH="/home/notebooks/.ipython/extensions:/home/notebooks/storage"
@@ -138,7 +140,7 @@ RUN source ${VENV_PATH}/bin/activate && \
     uv pip install --no-cache-dir -r ${WORKDIR}/requirements-agent.txt && rm ${WORKDIR}/requirements-agent.txt && \
     # Generative AI Agents
     uv pip uninstall litellm && \
-    uv pip install --no-cache-dir git+https://github.com/BerriAI/litellm@v1.72.1.dev1
+    uv pip install --no-cache-dir litellm>=1.72.1
 
 # Copy agent runtime into work directory
 COPY ./run_agent.py ${WORKDIR}/

public_dropin_environments/python311_genai_agents/env_info.json

@@ -4,7 +4,7 @@
   "description": "This template environment can be used to create GenAI-powered agents using CrewAI, LangGraph, or Llama-Index. Similar to other drop-in environments, you can either include a .pth artifact or any other code needed to deserialize your model, and optionally a custom.py file. You can also use this environment in codespaces.",
   "programmingLanguage": "python",
   "label": "",
-  "environmentVersionId": "6848b58f2081a8179c7a7704",
+  "environmentVersionId": "6849eae09604e94bd6c25d17",
   "environmentVersionDescription": "",
   "isPublic": true,
   "useCases": [
@@ -14,8 +14,8 @@
   "contextUrl": "https://github.com/datarobot/datarobot-user-models/tree/master/public_dropin_environments/python311_genai_agents",
   "imageRepository": "env-python-genai-agents",
   "tags": [
-    "v11.1.0-6848b58f2081a8179c7a7704",
-    "6848b58f2081a8179c7a7704",
+    "v11.1.0-6849eae09604e94bd6c25d17",
+    "6849eae09604e94bd6c25d17",
     "v11.1.0-latest"
   ]
 }

public_dropin_environments/python311_genai_agents/extensions/dataframe_formatter.py

@@ -20,8 +20,6 @@
 from enum import Enum
 from typing import Any
 from typing import Callable
-from typing import Dict
-from typing import List
 from typing import Optional
 from typing import Union
 from typing import cast
@@ -75,15 +73,15 @@ class DataframesProcessSteps(str, Enum):
     DEFAULT = "get_columns"
 
 
-Columns = List[Dict[str, Any]]
+Columns = list[dict[str, Any]]
 
 DEFAULT_INDEX_KEY = "index"
 
 
 def _register_exception(
     e: Exception,
     step: str,
-) -> Dict[str, Any]:
+) -> dict[str, Any]:
     exc_info = sys.exc_info()
     traceback_msg = traceback.format_exception(*exc_info)
 
@@ -174,7 +172,7 @@ def _transform_to_json(data: DataFrame) -> Any:
     return json.loads(data.to_json(orient="table", index=True, default_handler=str))["data"]
 
 
-def _prepare_df_for_chart_cell(val: DataFrame, columns: List[str]) -> Union[DataFrame, List[str]]:
+def _prepare_df_for_chart_cell(val: DataFrame, columns: list[str]) -> Union[DataFrame, list[str]]:
     if len(columns) == 0:
         data = []
     elif len(columns) == 1:
@@ -190,9 +188,9 @@ def _prepare_df_for_chart_cell(val: DataFrame, columns: List[str]) -> Union[Data
 # This formatter can operate with data that we have received as a DataFrame
 def formatter(  # noqa: C901,PLR0912
     val: "DataFrame",
-    formatter: Optional[Callable[..., List[str]]] = None,
+    formatter: Optional[Callable[..., list[str]]] = None,
     **formatter_kwargs: Any,
-) -> Dict[str, Any]:
+) -> dict[str, Any]:
     error = []
     dataframe_limit = 5000
     dataframe_id = id(val)

public_dropin_environments/python311_genai_agents/requirements.in

@@ -3,12 +3,12 @@ setuptools
 ecs-logging
 jupyter-client
 jupyter_kernel_gateway
-jupyter_core
+jupyter_core>=5.8.1
 ipykernel<6.29.0
 pandas
 numpy
 mistune
-datarobot-drum>=1.16.14
+datarobot-drum>=1.16.16
 datarobot
 numpy
 uwsgi
@@ -25,7 +25,8 @@ llama-index-llms-openai
 llama-index-llms-langchain
 llama-index-llms-litellm
 openai
-ragas
+ragas @ git+https://github.com/explodinggradients/ragas@5d59549ad5ef511f621502c563bc55ac5aeb9188#subdirectory=ragas
+datarobot-moderations>=11.1.14
 
 # Generative AI Telemetry
 traceloop-sdk