[BUZZOK-27768] Add NAT to python_311_genai_agents environment (#1704)

* remove EXT from Dockerfiles

* add nat

* add override

* update requirements

* Reconcile dependencies, updated IDs, tags

---------

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

Author: jpclemens0

public_dropin_environments/python311_genai_agents/Dockerfile

@@ -94,9 +94,8 @@ ENV ANNOY_COMPILER_ARGS="-D_CRT_SECURE_NO_WARNINGS,-DANNOYLIB_MULTITHREADED_BUIL
 
 ARG WORKDIR
 ARG VENV_PATH
-ARG EXT=""
 
-COPY ./pyproject.toml${EXT} ${WORKDIR}/pyproject.toml
+COPY ./pyproject.toml ${WORKDIR}/pyproject.toml
 
 WORKDIR ${WORKDIR}
 RUN . ${VENV_PATH}/bin/activate && \

public_dropin_environments/python311_genai_agents/Dockerfile.local

@@ -134,9 +134,8 @@ ENV ANNOY_COMPILER_ARGS="-D_CRT_SECURE_NO_WARNINGS,-DANNOYLIB_MULTITHREADED_BUIL
 
 ARG WORKDIR
 ARG VENV_PATH
-ARG EXT=""
 
-COPY ./pyproject.toml${EXT} ${WORKDIR}/pyproject.toml
+COPY ./pyproject.toml ${WORKDIR}/pyproject.toml
 
 WORKDIR ${WORKDIR}
 RUN . ${VENV_PATH}/bin/activate && \

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": "68e68b3735af84120828b074",
+  "environmentVersionId": "68e80d4278eb0c11d1a15b20",
   "environmentVersionDescription": "",
   "isPublic": true,
   "isDownloadable": true,
@@ -15,8 +15,8 @@
   "contextUrl": "https://github.com/datarobot/datarobot-user-models/tree/master/public_dropin_environments/python311_genai_agents",
   "imageRepository": "env-python-genai-agents",
   "tags": [
-    "v11.2.0-68e68b3735af84120828b074",
-    "68e68b3735af84120828b074",
+    "v11.2.0-68e80d4278eb0c11d1a15b20",
+    "68e80d4278eb0c11d1a15b20",
     "v11.2.0-latest"
   ]
 }

public_dropin_environments/python311_genai_agents/pyproject.toml

@@ -8,8 +8,10 @@ requires-python = ">=3.10, <3.13"
 dependencies = [
     "annotated-types>=0.7.0",
     "click>=8.1.8",
-    "datarobot-drum>=1.17.3",
+    "datarobot[auth]>=3.8.2",
+    "datarobot-drum>=1.17.5",
     "datarobot-mlops>=11.1.0",
+    "datarobot-moderations>=11.2.3",
     "datarobot-predict>=1.13.2",
     "dotenv>=0.9.9",
     "langchain-community>=0.3.27",
@@ -35,8 +37,6 @@ dependencies = [
     "requests>=2.32.4",
     "traceloop-sdk>=0.40.2",
     "uvicorn>=0.32.0",
-    "datarobot[auth]>=3.8.2",
-    "datarobot-moderations>=11.2.3",
 
     # Docker image requirements (These MUST be included to avoid docker conflicts)
     "ecs-logging>=2.2.0",
@@ -53,11 +53,19 @@ dependencies = [
     "ujson<5.8.0",
     "uwsgi>=2.0.30",
 
+    "nvidia-nat==1.3.0rc3; python_version >= '3.11'",
+    # "nvidia-nat-llama-index==1.3.0rc3; python_version >= '3.11'",
+    "nvidia-nat-opentelemetry==1.3.0rc3; python_version >= '3.11'",
+    "nvidia-nat-crewai==1.3.0rc3; python_version >= '3.11'",
+    "nvidia-nat-langchain==1.3.0rc3; python_version >= '3.11'",
+
     # Agent framework requirements
+
     "crewai>=0.177.0",
     "crewai-tools>=0.69.0",
     "opentelemetry-instrumentation-crewai>=0.40.5",
     "pybase64>=1.4.2",
+    "pylance<=0.38.1",
     "langgraph>=0.4.10",
     "langgraph-prebuilt>=0.2.3",
     "llama-index<0.14.0",
@@ -174,13 +182,10 @@ docstring-code-line-length = "dynamic"
 allow_duplicate_keys = true
 line_length = 120
 
-# Pydantic plugin must be disabled with llamaindex, see https://github.com/python/mypy/issues/15927
 [tool.mypy]
 strict = true
 python_version = "3.11"
-
 plugins = "pydantic.mypy"
-
 exclude = """(?x)(
     output/ # exclude files starting with "output"
     | deployment_.*/
@@ -212,3 +217,8 @@ warn_unused_ignores = true
 [[tool.mypy.overrides]]
 module = "datarobot.*"
 implicit_reexport = true
+
+[tool.uv]
+override-dependencies = [
+    "fastapi==0.118.2",
+]