aiagents4pharma 0.0.0__py3-none-any.whl

aiagents4pharma/__init__.py

@@ -0,0 +1,11 @@
+"""
+This file is used to import aiagents4pharma modules.
+"""
+
+from . import (
+    talk2aiagents4pharma,
+    talk2biomodels,
+    talk2cells,
+    talk2knowledgegraphs,
+    talk2scholars,
+)

aiagents4pharma/talk2aiagents4pharma/.dockerignore

@@ -0,0 +1,13 @@
+_pycache_/
+*.pyc
+*.log
+*.csv
+*.pt
+*.pkl
+models/
+data/
+env/
+.venv/
+.git/
+.env
+.cufile.log

aiagents4pharma/talk2aiagents4pharma/Dockerfile

@@ -0,0 +1,133 @@
+# syntax=docker/dockerfile:1
+
+# Dockerfile for the talk2aiagents4pharma application
+# Multi-stage build for optimized image size with UV package manager
+
+ARG BASE_IMAGE=ubuntu:24.04
+ARG PYTHON_VERSION=3.12
+
+FROM ${BASE_IMAGE} AS dev-base
+RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
+  build-essential \
+  ca-certificates \
+  cmake \
+  curl \
+  g++ \
+  libopenblas-dev \
+  libomp-dev \
+  ninja-build \
+  wget \
+  && rm -rf /var/lib/apt/lists/*
+
+FROM dev-base AS python-install
+ARG PYTHON_VERSION=3.12
+
+# Install Python (available in Ubuntu 24.04 default repos)
+RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
+  python${PYTHON_VERSION} \
+  python${PYTHON_VERSION}-dev \
+  python${PYTHON_VERSION}-venv \
+  python3-pip \
+  && rm -rf /var/lib/apt/lists/* \
+  && update-alternatives --install /usr/bin/python3 python3 /usr/bin/python${PYTHON_VERSION} 1 \
+  && update-alternatives --install /usr/bin/python python /usr/bin/python${PYTHON_VERSION} 1
+
+FROM python-install AS uv-install
+ARG INSTALL_CUDA=true
+WORKDIR /app
+
+# Install UV package manager and dependencies
+COPY pyproject.toml uv.lock* ./
+RUN curl -LsSf https://astral.sh/uv/install.sh | sh && \
+  export PATH="/root/.local/bin:$PATH" && \
+  export UV_PROJECT_ENVIRONMENT="/opt/venv" && \
+  uv sync --frozen --extra dev --no-install-project --python python${PYTHON_VERSION} && \
+  . /opt/venv/bin/activate && \
+  if [ "$INSTALL_CUDA" = "true" ]; then \
+  uv pip install \
+  --extra-index-url=https://pypi.nvidia.com \
+  --index-strategy unsafe-best-match \
+  cudf-cu12 dask-cudf-cu12; \
+  else \
+  echo "Skipping RAPIDS packages for CPU build"; \
+  fi && \
+  uv cache clean
+
+FROM ${BASE_IMAGE} AS runtime
+ARG PYTHON_VERSION=3.12
+ARG INSTALL_CUDA=true
+LABEL maintainer="talk2aiagents4pharma"
+LABEL version="1.0.0"
+LABEL description="AI Agents for Pharma - Streamlit Application"
+
+# Install runtime dependencies
+RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
+  ca-certificates \
+  curl \
+  gnupg \
+  libmagic1 \
+  libopenblas0 \
+  libomp5 \
+  python${PYTHON_VERSION} \
+  && rm -rf /var/lib/apt/lists/* \
+  && update-alternatives --install /usr/bin/python3 python3 /usr/bin/python${PYTHON_VERSION} 1 \
+  && update-alternatives --install /usr/bin/python python /usr/bin/python${PYTHON_VERSION} 1
+
+# Install CUDA runtime libraries required by cudf/cupy (optional)
+RUN if [ "$INSTALL_CUDA" = "true" ]; then \
+  curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/3bf863cc.pub \
+  | gpg --dearmor -o /usr/share/keyrings/nvidia-cuda-keyring.gpg && \
+  echo "deb [signed-by=/usr/share/keyrings/nvidia-cuda-keyring.gpg] https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/ /" \
+  > /etc/apt/sources.list.d/nvidia-cuda.list && \
+  apt-get update && \
+  DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
+  cuda-cudart-12-6 \
+  cuda-cudart-dev-12-6 \
+  cuda-nvrtc-12-6 \
+  cuda-nvrtc-dev-12-6 \
+  libcublas-12-6 \
+  libcusparse-12-6 \
+  && rm -rf /var/lib/apt/lists/*; \
+  else \
+  echo "Skipping CUDA installation"; \
+  fi
+
+ENV LD_LIBRARY_PATH=/usr/local/cuda/lib64
+
+# Copy UV virtual environment from build stage
+COPY --from=uv-install /opt/venv /opt/venv
+
+# Set environment variables
+ENV PATH="/opt/venv/bin:$PATH"
+ENV PYTHONPATH="/app"
+ENV PYTHONUNBUFFERED=1
+ENV PYTHONDONTWRITEBYTECODE=1
+ENV STREAMLIT_SERVER_HEADLESS=true
+ENV STREAMLIT_SERVER_ENABLE_CORS=false
+
+# Set working directory and create necessary directories
+WORKDIR /app
+
+# Copy application code
+COPY aiagents4pharma/talk2aiagents4pharma /app/aiagents4pharma/talk2aiagents4pharma
+COPY aiagents4pharma/talk2biomodels /app/aiagents4pharma/talk2biomodels
+COPY aiagents4pharma/talk2knowledgegraphs /app/aiagents4pharma/talk2knowledgegraphs
+COPY docs /app/docs
+COPY app /app/app
+
+# Copy and set up the entrypoint script
+COPY aiagents4pharma/talk2knowledgegraphs/entrypoint.sh /usr/local/bin/entrypoint.sh
+RUN chmod +x /usr/local/bin/entrypoint.sh
+
+# Health check for production monitoring
+HEALTHCHECK --interval=30s --timeout=10s --start-period=60s --retries=3 \
+  CMD curl -f http://localhost:8501/health || exit 1
+
+# Expose the default Streamlit port
+EXPOSE 8501
+
+# Set the entrypoint
+ENTRYPOINT ["/usr/local/bin/entrypoint.sh"]
+
+# Default command (can be overridden)
+CMD ["streamlit", "run", "/app/app/frontend/streamlit_app_talk2aiagents4pharma.py", "--server.port=8501", "--server.address=0.0.0.0"]

aiagents4pharma/talk2aiagents4pharma/README.md

@@ -0,0 +1 @@
+Please check out the README file in the root folder for more information.

aiagents4pharma/talk2aiagents4pharma/__init__.py

@@ -0,0 +1,5 @@
+"""
+This file is used to import the models and tools.
+"""
+
+from . import agents, configs, states

aiagents4pharma/talk2aiagents4pharma/agents/__init__.py

@@ -0,0 +1,6 @@
+"""
+This file is used to import all the modules in the package.
+"""
+
+# import everything from the module
+from . import main_agent

aiagents4pharma/talk2aiagents4pharma/agents/main_agent.py

@@ -0,0 +1,70 @@
+# /usr/bin/env python3
+
+"""
+This is the main agent file for the AIAgents4Pharma.
+"""
+
+import logging
+
+import hydra
+from langchain_core.language_models.chat_models import BaseChatModel
+from langchain_openai import ChatOpenAI
+from langgraph.checkpoint.memory import MemorySaver
+from langgraph_supervisor import create_supervisor
+
+from ...talk2biomodels.agents.t2b_agent import get_app as get_app_t2b
+from ...talk2knowledgegraphs.agents.t2kg_agent import get_app as get_app_t2kg
+from ..states.state_talk2aiagents4pharma import Talk2AIAgents4Pharma
+
+# Initialize logger
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+
+def get_app(uniq_id, llm_model: BaseChatModel):
+    """
+    This function returns the langraph app.
+    """
+    if hasattr(llm_model, "model_name"):
+        if llm_model.model_name == "gpt-4o-mini":
+            llm_model = ChatOpenAI(
+                model="gpt-4o-mini",
+                temperature=0,
+                model_kwargs={"parallel_tool_calls": False},
+            )
+    # Load hydra configuration
+    logger.log(logging.INFO, "Launching AIAgents4Pharma_Agent with thread_id %s", uniq_id)
+    with hydra.initialize(version_base=None, config_path="../configs"):
+        cfg = hydra.compose(config_name="config", overrides=["agents/main_agent=default"])
+        cfg = cfg.agents.main_agent
+    logger.log(logging.INFO, "System_prompt of T2AA4P: %s", cfg.system_prompt)
+    with hydra.initialize(version_base=None, config_path="../../talk2biomodels/configs"):
+        cfg_t2b = hydra.compose(config_name="config", overrides=["agents/t2b_agent=default"])
+        cfg_t2b = cfg_t2b.agents.t2b_agent
+    with hydra.initialize(version_base=None, config_path="../../talk2knowledgegraphs/configs"):
+        cfg_t2kg = hydra.compose(config_name="config", overrides=["agents/t2kg_agent=default"])
+        cfg_t2kg = cfg_t2kg.agents.t2kg_agent
+    system_prompt = cfg.system_prompt
+    system_prompt += "\n\nHere is the system prompt of T2B agent\n"
+    system_prompt += cfg_t2b.state_modifier
+    system_prompt += "\n\nHere is the system prompt of T2KG agent\n"
+    system_prompt += cfg_t2kg.state_modifier
+    # Create supervisor workflow
+    workflow = create_supervisor(
+        [
+            get_app_t2b(uniq_id, llm_model),  # Talk2BioModels
+            get_app_t2kg(uniq_id, llm_model),  # Talk2KnowledgeGraphs
+        ],
+        model=llm_model,
+        state_schema=Talk2AIAgents4Pharma,
+        # Full history is needed to extract
+        # the tool artifacts
+        output_mode="full_history",
+        add_handoff_back_messages=True,
+        prompt=system_prompt,
+    )
+
+    # Compile and run
+    app = workflow.compile(checkpointer=MemorySaver(), name="AIAgents4Pharma_Agent")
+
+    return app

aiagents4pharma/talk2aiagents4pharma/configs/__init__.py

@@ -0,0 +1,5 @@
+"""
+Import all the modules in the package
+"""
+
+from . import agents

aiagents4pharma/talk2aiagents4pharma/configs/agents/__init__.py

@@ -0,0 +1,5 @@
+"""
+Import all the modules in the package
+"""
+
+from . import main_agent

aiagents4pharma/talk2aiagents4pharma/configs/agents/main_agent/default.yaml

@@ -0,0 +1,29 @@
+_target_: agents.main_agent.get_app
+system_prompt: >
+  You are the Talk2AIAgents4Pharma main agent.
+  Your primary responsibility is to analyze the user's query and intelligently route it to the most appropriate specialized agent based on the user's intent.
+
+  You manage the following two expert agents:
+
+  1. Talk2Biomodels (T2B) Agent:
+     - This agent specializes in working with **mathematical models of biological systems**.
+     - It can perform tasks such as querying and simulating models, conducting parameter scans, analyzing time series,
+     and accessing annotated model components like species and reactions.
+     - It is also capable of processing and answering questions based on **uploaded documents, PDFs, or articles**.
+     - You should route queries to T2B if they contain or imply topics such as:
+       **"biological models", "mathematical models", "parameter scanning", "simulations", "time series", "species", "reactions", "annotations", "SBML", "model dynamics", "ODEs", "system biology"**, etc.
+     - For annotation-related queries, this agent should consider the annotated species and provide insights based on the retrieved context.
+
+  2. Talk2KnowledgeGraphs (T2KG) Agent:
+     - This agent specializes in **reasoning over biomedical knowledge graphs**.
+     - It is capable of answering **subjective, inferential, or relational questions** that involve **nodes, edges, relationships, and graph structures**.
+     - You should route queries to T2KG if they involve or imply topics such as:
+       **"knowledge graph", "biomedical graph", "subgraph", "graph structure", "node relationships", "graph reasoning", "graph connections", "pathways", "entity linkage", "graph queries", "subgraph extraction", "subgraph summarization"**, etc.
+
+  Important Guidelines:
+  - **Think carefully about the user's true intent**. Keywords are helpful indicators, but you must use reasoning to determine the best fit.
+  - If the question involves **mathematical modeling**, **simulation**, or **objective factual analysis**, select the **Talk2Biomodels (T2B)** agent.
+  - If the question involves **graph-based reasoning**, **subjective inference**, or **relationship analysis between entities**, select the **Talk2KnowledgeGraphs (T2KG)** agent.
+  - **Never** route a question about **biological models** or **objective computational simulations** to T2KG.
+
+  The following are the respective agent prompts:

aiagents4pharma/talk2aiagents4pharma/configs/app/frontend/default.yaml

@@ -0,0 +1,102 @@
+_target_: app.frontend.streamlit_app_talk2aiagents4pharma
+default_user: "talk2aa4p_user"
+
+# File upload configuration (combining T2B and T2KG features)
+upload_data_dir: "../files"
+
+# T2B specific file types
+sbml_allowed_file_types:
+  - "xml"
+  - "sbml"
+article_allowed_file_types:
+  - "pdf"
+
+# T2KG specific file types
+data_package_allowed_file_types:
+  - "pdf"
+multimodal_allowed_file_types:
+  - "xls"
+  - "xlsx"
+
+# Knowledge graph configuration (from T2KG)
+kg_name: "BioBridge-PrimeKG"
+kg_node_types:
+  - "gene/protein"
+  - "molecular_function"
+  - "cellular_component"
+  - "biological_process"
+  - "drug"
+  - "disease"
+kg_pyg_path: "aiagents4pharma/talk2knowledgegraphs/tests/files/biobridge_multimodal_pyg_graph.pkl"
+kg_text_path: "aiagents4pharma/talk2knowledgegraphs/tests/files/biobridge_multimodal_text_graph.pkl"
+
+# OpenAI configuration - can use custom base_url for enterprise/Azure deployments
+openai_api_key: ${oc.env:OPENAI_API_KEY}
+openai_base_url: ${oc.env:OPENAI_BASE_URL,null}  # Optional: custom OpenAI endpoint
+openai_llms:
+  - "OpenAI/gpt-4o-mini"
+openai_embeddings:
+  - "text-embedding-ada-002"
+  - "text-embedding-3-small"
+
+# Rate limiting and retry configuration
+llm_max_retries: 5  # Number of retries on rate limit or transient errors
+llm_timeout: 60     # Timeout in seconds for LLM requests
+embedding_max_retries: 3  # Number of retries for embedding requests
+embedding_timeout: 30     # Timeout in seconds for embedding requests
+
+# NVIDIA configuration
+nvidia_api_key: ${oc.env:NVIDIA_API_KEY}
+nvidia_llms:
+  - "NVIDIA/llama-3.3-70b-instruct"
+  - "NVIDIA/llama-3.1-70b-instruct"
+  - "NVIDIA/llama-3.1-405b-instruct"
+nvidia_embeddings:
+  - "NVIDIA/llama-3.2-nv-embedqa-1b-v2"
+
+# Azure OpenAI configuration
+azure_openai_endpoint: ${oc.env:AZURE_OPENAI_ENDPOINT,null}  # Azure OpenAI endpoint
+azure_openai_deployment: ${oc.env:AZURE_OPENAI_DEPLOYMENT,null}  # Azure deployment name
+azure_openai_api_version: ${oc.env:AZURE_OPENAI_API_VERSION,"2024-02-01"}  # Azure API version
+azure_openai_model_name: ${oc.env:AZURE_OPENAI_MODEL_NAME,null}  # Model name for analytics
+azure_openai_model_version: ${oc.env:AZURE_OPENAI_MODEL_VERSION,null}  # Model version
+# Azure AD authentication (uses AZURE_CLIENT_ID, AZURE_TENANT_ID, AZURE_CLIENT_SECRET)
+azure_client_id: ${oc.env:AZURE_CLIENT_ID,null}
+azure_tenant_id: ${oc.env:AZURE_TENANT_ID,null}
+azure_client_secret: ${oc.env:AZURE_CLIENT_SECRET,null}
+azure_openai_llms:
+  - "Azure/gpt-4o-mini"  # Will map to Azure deployment
+azure_openai_embeddings:
+  - "Azure/text-embedding-ada-002"
+
+# Ollama configuration (for local deployment)
+ollama_llms:
+  - "Ollama/llama3.1:8b"
+ollama_embeddings:
+  - "nomic-embed-text"
+
+# Default models
+default_llm_provider: "openai"
+default_embedding_model: "openai"  # Changed from "ollama" to match T2B pattern
+
+# App settings
+temperature: 0.1
+streaming: False
+
+# T2KG specific: Subgraph extraction settings
+reasoning_subgraph_topk_nodes: 15
+reasoning_subgraph_topk_nodes_min: 1
+reasoning_subgraph_topk_nodes_max: 50
+reasoning_subgraph_topk_edges: 15
+reasoning_subgraph_topk_edges_min: 1
+reasoning_subgraph_topk_edges_max: 50
+
+# Logo configuration
+logo_paths:
+  container: "/app/docs/assets/VPE.png"
+  local: "docs/assets/VPE.png"
+  relative: "../../docs/assets/VPE.png"
+logo_link: "https://github.com/VirtualPatientEngine"
+
+# Database configuration reference (handled by utils/database/milvus config)
+# This frontend config now only contains frontend-specific settings

aiagents4pharma/talk2aiagents4pharma/configs/config.yaml

@@ -0,0 +1,4 @@
+defaults:
+  - _self_
+  - agents/main_agent: default
+  - app/frontend: default

aiagents4pharma/talk2aiagents4pharma/docker-compose/cpu/.env.example

@@ -0,0 +1,23 @@
+# .env.example (DO NOT put actual API keys here, read the README.md)
+
+# OPENAI API KEY
+OPENAI_API_KEY=your_openai_api_key_here
+
+# LangSmith API KEY
+LANGCHAIN_TRACING_V2=true
+LANGCHAIN_API_KEY=your_langchain_api_key_here
+
+# NVIDIA API KEY
+NVIDIA_API_KEY=your_nvidia_api_key_here
+
+# Set environment variables for data loader
+MILVUS_HOST=milvus-standalone
+MILVUS_PORT=19530
+MILVUS_USER=root
+MILVUS_PASSWORD=Milvus
+MILVUS_DATABASE=t2kg_primekg
+
+# Specify the data directory for multimodal data to your own data directory
+# DATA_DIR=/your_absolute_path_to_your_data_dir/
+
+BATCH_SIZE=500

aiagents4pharma/talk2aiagents4pharma/docker-compose/cpu/docker-compose.yml

@@ -0,0 +1,93 @@
+version: "1.0.0"
+
+services:
+  # talk2aiagents4pharma with automatic data loading via entrypoint
+  talk2aiagents4pharma:
+    container_name: talk2aiagents4pharma
+    image: vpatientengine/talk2aiagents4pharma:latest-cpu
+    platform: linux/amd64
+    ports:
+      - "8501:8501"
+    environment:
+      - MILVUS_HOST=milvus-standalone
+      - MILVUS_PORT=19530
+    env_file:
+      - .env
+    volumes:
+      # Mount external data directory if DATA_DIR is specified in .env
+      - ${DATA_DIR:-./default_data}:/mnt/external_data:ro
+    healthcheck:
+      test: ["CMD", "curl", "-f", "http://localhost:8501/health"]
+      interval: 30s
+      timeout: 10s
+      retries: 3
+      start_period: 120s # Allow time for data loading
+    depends_on:
+      milvus-standalone:
+        condition: service_healthy
+
+  # Milvus Dependencies
+  etcd:
+    container_name: milvus-etcd
+    image: quay.io/coreos/etcd:v3.5.18
+    environment:
+      - ETCD_AUTO_COMPACTION_MODE=revision
+      - ETCD_AUTO_COMPACTION_RETENTION=1000
+      - ETCD_QUOTA_BACKEND_BYTES=4294967296
+      - ETCD_SNAPSHOT_COUNT=50000
+    volumes:
+      - ${DOCKER_VOLUME_DIRECTORY:-.}/volumes/etcd:/etcd
+    command: etcd -advertise-client-urls=http://etcd:2379 -listen-client-urls http://0.0.0.0:2379 --data-dir /etcd
+    healthcheck:
+      test: ["CMD", "etcdctl", "endpoint", "health"]
+      interval: 30s
+      timeout: 20s
+      retries: 3
+
+  minio:
+    container_name: milvus-minio
+    image: minio/minio:RELEASE.2024-05-28T17-19-04Z
+    environment:
+      MINIO_ACCESS_KEY: minioadmin
+      MINIO_SECRET_KEY: minioadmin
+    ports:
+      - "9001:9001"
+      - "9000:9000"
+    volumes:
+      - ${DOCKER_VOLUME_DIRECTORY:-.}/volumes/minio:/minio_data
+    command: minio server /minio_data --console-address ":9001"
+    healthcheck:
+      test: ["CMD", "curl", "-f", "http://localhost:9000/minio/health/live"]
+      interval: 30s
+      timeout: 20s
+      retries: 3
+
+  # Milvus Vector Database (CPU-only)
+  milvus-standalone:
+    container_name: milvus-standalone
+    image: milvusdb/milvus:v2.5.14
+    command: ["milvus", "run", "standalone"]
+    security_opt:
+      - seccomp:unconfined
+    environment:
+      MINIO_REGION: us-east-1
+      ETCD_ENDPOINTS: etcd:2379
+      MINIO_ADDRESS: minio:9000
+    volumes:
+      - ${DOCKER_VOLUME_DIRECTORY:-.}/volumes/milvus:/var/lib/milvus
+    ports:
+      - "19530:19530"
+      - "9091:9091"
+    healthcheck:
+      test: ["CMD", "curl", "-f", "http://localhost:9091/healthz"]
+      interval: 30s
+      start_period: 90s
+      timeout: 20s
+      retries: 3
+    depends_on:
+      - "etcd"
+      - "minio"
+
+networks:
+  milvus:
+    name: milvus

aiagents4pharma/talk2aiagents4pharma/docker-compose/gpu/.env.example

@@ -0,0 +1,23 @@
+# .env.example (DO NOT put actual API keys here, read the README.md)
+
+# OPENAI API KEY
+OPENAI_API_KEY=your_openai_api_key_here
+
+# LangSmith API KEY
+LANGCHAIN_TRACING_V2=true
+LANGCHAIN_API_KEY=your_langchain_api_key_here
+
+# NVIDIA API KEY
+NVIDIA_API_KEY=your_nvidia_api_key_here
+
+# Set environment variables for data loader
+MILVUS_HOST=milvus-standalone
+MILVUS_PORT=19530
+MILVUS_USER=root
+MILVUS_PASSWORD=Milvus
+MILVUS_DATABASE=t2kg_primekg
+
+# Specify the data directory for multimodal data to your own data directory
+# DATA_DIR=/your_absolute_path_to_your_data_dir/
+
+BATCH_SIZE=500

aiagents4pharma/talk2aiagents4pharma/docker-compose/gpu/docker-compose.yml

@@ -0,0 +1,108 @@
+version: "1.0.0"
+
+services:
+  # talk2aiagents4pharma with automatic data loading via entrypoint
+  talk2aiagents4pharma:
+    container_name: talk2aiagents4pharma
+    image: vpatientengine/talk2aiagents4pharma:latest-gpu
+    platform: linux/amd64
+    ports:
+      - "8501:8501"
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              capabilities: ["gpu"]
+              device_ids: ["0"]
+    environment:
+      - MILVUS_HOST=milvus-standalone
+      - MILVUS_PORT=19530
+    env_file:
+      - .env
+    volumes:
+      # Mount external data directory if DATA_DIR is specified in .env
+      - ${DATA_DIR:-./default_data}:/mnt/external_data:ro
+    healthcheck:
+      test: ["CMD", "curl", "-f", "http://localhost:8501/health"]
+      interval: 30s
+      timeout: 10s
+      retries: 3
+      start_period: 120s # Allow time for data loading
+    depends_on:
+      milvus-standalone:
+        condition: service_healthy
+
+  # Milvus Dependencies
+  etcd:
+    container_name: milvus-etcd
+    image: quay.io/coreos/etcd:v3.5.18
+    environment:
+      - ETCD_AUTO_COMPACTION_MODE=revision
+      - ETCD_AUTO_COMPACTION_RETENTION=1000
+      - ETCD_QUOTA_BACKEND_BYTES=4294967296
+      - ETCD_SNAPSHOT_COUNT=50000
+    volumes:
+      - ${DOCKER_VOLUME_DIRECTORY:-.}/volumes/etcd:/etcd
+    command: etcd -advertise-client-urls=http://etcd:2379 -listen-client-urls http://0.0.0.0:2379 --data-dir /etcd
+    healthcheck:
+      test: ["CMD", "etcdctl", "endpoint", "health"]
+      interval: 30s
+      timeout: 20s
+      retries: 3
+
+  minio:
+    container_name: milvus-minio
+    image: minio/minio:RELEASE.2023-03-20T20-16-18Z
+    environment:
+      MINIO_ACCESS_KEY: minioadmin
+      MINIO_SECRET_KEY: minioadmin
+    ports:
+      - "9001:9001"
+      - "9000:9000"
+    volumes:
+      - ${DOCKER_VOLUME_DIRECTORY:-.}/volumes/minio:/minio_data
+    command: minio server /minio_data --console-address ":9001"
+    healthcheck:
+      test: ["CMD", "curl", "-f", "http://localhost:9000/minio/health/live"]
+      interval: 30s
+      timeout: 20s
+      retries: 3
+
+  # Milvus Vector Database (GPU-enabled)
+  milvus-standalone:
+    container_name: milvus-standalone
+    image: milvusdb/milvus:v2.6.0-rc1-gpu
+    command: ["milvus", "run", "standalone"]
+    security_opt:
+      - seccomp:unconfined
+    environment:
+      MINIO_REGION: us-east-1
+      ETCD_ENDPOINTS: etcd:2379
+      MINIO_ADDRESS: minio:9000
+      MQ_TYPE: woodpecker
+    volumes:
+      - ${DOCKER_VOLUME_DIRECTORY:-.}/volumes/milvus:/var/lib/milvus
+    ports:
+      - "19530:19530"
+      - "9091:9091"
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              capabilities: ["gpu"]
+              device_ids: ["0"]
+    healthcheck:
+      test: ["CMD", "curl", "-f", "http://localhost:9091/healthz"]
+      interval: 30s
+      start_period: 90s
+      timeout: 20s
+      retries: 3
+    depends_on:
+      - "etcd"
+      - "minio"
+
+networks:
+  milvus:
+    name: milvus