[BUZZOK-28810] Add support for MCP server to genai agents (#1798)

* Add support for MCP server to genai agents

* Rework

* Cleanup

* Remove checks

* Set pythonpath

* Fix posix compatibility

* Fix remaining posix incompatibilties

Author: mjnitz02

public_dropin_environments/python311_genai_agents/Dockerfile

@@ -52,7 +52,8 @@ RUN uv venv ${VENV_PATH} && \
 WORKDIR ${WORKDIR}
 
 COPY ./agent/agent.py ./agent/cgroup_watchers.py ${AGENTDIR}/
-COPY ./jupyter_kernel_gateway_config.py ./start_server.sh ${WORKDIR}/
+COPY ./jupyter_kernel_gateway_config.py ${WORKDIR}/jupyter_kernel_gateway_config.py
+COPY ./start_server_codespaces.sh ${WORKDIR}/start_server.sh
 COPY ./ipython_config.py /etc/ipython/
 COPY ./extensions /etc/ipython/extensions
 

public_dropin_environments/python311_genai_agents/Dockerfile.local

@@ -92,7 +92,8 @@ RUN uv venv ${VENV_PATH} && \
 WORKDIR ${WORKDIR}
 
 COPY ./agent/agent.py ./agent/cgroup_watchers.py ${AGENTDIR}/
-COPY ./jupyter_kernel_gateway_config.py ./start_server.sh ${WORKDIR}/
+COPY ./jupyter_kernel_gateway_config.py ${WORKDIR}/jupyter_kernel_gateway_config.py
+COPY ./start_server_codespaces.sh ${WORKDIR}/start_server.sh
 COPY ./ipython_config.py /etc/ipython/
 COPY ./extensions /etc/ipython/extensions
 

…s/python311_genai_agents/start_server.sh …_genai_agents/start_server_codespaces.shpublic_dropin_environments/python311_genai_agents/start_server.sh renamed to public_dropin_environments/python311_genai_agents/start_server_codespaces.sh public_dropin_environments/python311_genai_agents/start_server.sh renamed to public_dropin_environments/python311_genai_agents/start_server_codespaces.sh

@@ -5,24 +5,68 @@
 # This is proprietary source code of DataRobot, Inc. and its affiliates.
 #
 # Released under the terms of DataRobot Tool and Utility Agreement.
-echo "Starting Custom Model environment with DRUM prediction server"
 
-if [ "${ENABLE_CUSTOM_MODEL_RUNTIME_ENV_DUMP}" = 1 ]; then
+# =============================================================================
+# Startup script for Custom Model or MCP Server environments
+# Determines which service to run based on directory contents
+# =============================================================================
+
+SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
+
+# Configure UV package manager
+export UV_PROJECT=${CODE_DIR}
+unset UV_COMPILE_BYTECODE  # Disable compilation (already done in build)
+unset UV_CACHE_DIR         # Disable caching for reproducibility
+
+# Activate the virtual environment
+. ${VENV_PATH}/bin/activate
+
+# Sync dependencies using UV
+# --active: Install into the active venv instead of creating a new one
+# --frozen: Skip dependency resolution, use exact versions from lock file
+# --extra: Install the 'agentic_playground' optional dependency group
+# Note: Compilation disabled since kernel venv is already compiled
+uv sync --frozen --active --no-progress --color never --extra agentic_playground || true
+
+# Optional: Dump environment variables for debugging
+if [ "${ENABLE_CUSTOM_MODEL_RUNTIME_ENV_DUMP}" = "1" ]; then
     echo "Environment variables:"
     env
 fi
 
-export UV_PROJECT=${CODE_DIR}
-unset UV_COMPILE_BYTECODE
-unset UV_CACHE_DIR
-source ${VENV_PATH}/bin/activate
-# `--active` to install into active kernel venv (instead of creating local from scratch)
-# `--frozen` to skip dependency resolution and just install exactly what's in lock file
-# Compilation DISABLED - kernel venv has already been compiled, and having compilation enabled
-# would re-compile all site-packages (takes quite some time)
-time uv sync --frozen --active --no-progress --color never --extra agentic_playground || true
-
-echo
-echo "Executing command: drum server $*"
-echo
-exec drum server "$@"
+# -----------------------------------------------------------------------------
+# Option 1: Custom Model with DRUM Server
+# Requires: custom.py file in the same directory
+# -----------------------------------------------------------------------------
+if [ -f "$SCRIPT_DIR/custom.py" ]; then
+    echo "Starting Custom Model environment with DRUM prediction server"
+
+    # Start DRUM server
+    echo
+    echo "Executing command: drum server $*"
+    echo
+    exec drum server "$@"
+
+# -----------------------------------------------------------------------------
+# Option 2: MCP Server
+# Requires: app/ directory in the same location
+# -----------------------------------------------------------------------------
+elif [ -d "$SCRIPT_DIR/app" ]; then
+    echo "Starting Custom Model environment with MCP server"
+
+    # Set Python path to script directory for module imports
+    export PYTHONPATH="$SCRIPT_DIR"
+
+    # Start the MCP server
+    exec python -m app.main
+
+# -----------------------------------------------------------------------------
+# Error: No valid entry point found
+# -----------------------------------------------------------------------------
+else
+    echo "Error: Neither custom.py nor app/ directory found in $SCRIPT_DIR"
+    echo "This script requires either:"
+    echo "  - custom.py file for DRUM-based Custom Models"
+    echo "  - app/ directory for MCP Server applications"
+    exit 1
+fi