Docker container for AgentIQ

docker/Dockerfile

@@ -0,0 +1,59 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# 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.
+
+ARG BASE_IMAGE_URL=nvcr.io/nvidia/base/ubuntu
+ARG BASE_IMAGE_TAG=22.04_20240212
+ARG PYTHON_VERSION=3.12
+ARG AIQ_VERSION=1.0.0
+ARG UV_VERSION=0.5.31
+
+FROM --platform=$TARGETPLATFORM ghcr.io/astral-sh/uv:${UV_VERSION} AS uv_base
+
+FROM --platform=$TARGETPLATFORM ${BASE_IMAGE_URL}:${BASE_IMAGE_TAG} AS base
+ARG AIQ_VERSION
+ARG PYTHON_VERSION
+ARG UV_VERSION
+
+COPY --from=uv_base /uv /uvx /bin/
+
+ENV PYTHONDONTWRITEBYTECODE=1
+
+# Install certificates
+RUN export DEBIAN_FRONTEND=noninteractive && \
+    export TZ=Etc/UTC && \
+    apt-get update && \
+    apt upgrade -y && \
+    apt-get install --no-install-recommends  -y ca-certificates && \
+    apt clean && \
+    update-ca-certificates
+
+# Set SSL environment variables
+ENV REQUESTS_CA_BUNDLE=/etc/ssl/certs/ca-certificates.crt
+ENV SSL_CERT_FILE=/etc/ssl/certs/ca-certificates.crt
+
+# Set working directory
+WORKDIR /workspace
+
+# Install the AgentIQ package and the example package
+RUN --mount=type=cache,id=uv_cache,target=/root/.cache/uv,sharing=locked \
+    uv venv --python ${PYTHON_VERSION} /workspace/.venv && \
+    . /workspace/.venv/bin/activate && \
+    uv pip install "agentiq[crewai, langchain, llama-index, mem0ai, semantic-kernel, zep-cloud] == ${AIQ_VERSION}"
+
+# Enivronment variables for the venv
+ENV PATH="/workspace/.venv/bin:$PATH"
+
+# Define the entry point to start the aiq CLI tool
+ENTRYPOINT ["aiq"]

docker/build_container.sh

@@ -0,0 +1,55 @@
+#!/bin/bash
+# SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# 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.
+
+SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )"
+
+# Get the path to REPO_ROOT without altering the docker context (in case we are in a submodule)
+pushd ${SCRIPT_DIR} &> /dev/null
+export REPO_ROOT=${REPO_ROOT:-"$(git rev-parse --show-toplevel)"}
+popd &> /dev/null
+
+HOST_ARCH=$(dpkg --print-architecture)
+DOCKER_TARGET_ARCH=${DOCKER_TARGET_ARCH:-${HOST_ARCH}}
+
+if [ ${DOCKER_TARGET_ARCH} != ${HOST_ARCH} ]; then
+    echo -n "Performing cross-build for ${DOCKER_TARGET_ARCH} on ${HOST_ARCH}, please ensure qemu is installed, "
+    echo "details in ${REPO_ROOT}/docs/source/advanced/running-ci-locally.md"
+fi
+
+git describe --tags --abbrev=0 2>/dev/null || echo "no-tag"
+
+DOCKER_IMAGE_NAME=${DOCKER_IMAGE_NAME:-"agentiq"}
+DOCKER_IMAGE_TAG=${DOCKER_IMAGE_TAG:-"$(git describe --tags --abbrev=0 2> /dev/null)"}
+
+DOCKER_EXTRA_ARGS=${DOCKER_EXTRA_ARGS:-""}
+
+# Build the docker arguments
+DOCKER_ARGS="-t ${DOCKER_IMAGE_NAME}:${DOCKER_IMAGE_TAG}"
+DOCKER_ARGS="${DOCKER_ARGS} --platform=linux/${DOCKER_TARGET_ARCH}"
+DOCKER_ARGS="${DOCKER_ARGS} --network=host"
+
+# Last add any extra args (duplicates override earlier ones)
+DOCKER_ARGS="${DOCKER_ARGS} ${DOCKER_EXTRA_ARGS}"
+
+# Export buildkit variable
+export DOCKER_BUILDKIT=1
+
+echo "Building ${DOCKER_IMAGE_NAME}:${DOCKER_TAG} with args..."
+echo ""
+echo "   COMMAND: docker build ${DOCKER_ARGS} -f ${SCRIPT_DIR}/Dockerfile ."
+echo "   Note: add '--progress plain' to DOCKER_EXTRA_ARGS to show all container build output"
+
+docker build ${DOCKER_ARGS} -f ${SCRIPT_DIR}/Dockerfile .