Skip to content

DrugAgent (Liu et al.)

DrugAgent is a research multi-agent framework that combines LLM planning with domain-guided code generation to build and evaluate machine-learning workflows for drug discovery.

Catalog updated ·

Overview

DrugAgent (Liu et al.) addresses the programming work needed to turn drug-discovery modelling ideas into machine-learning experiments. The framework pairs an LLM Planner, responsible for proposing and refining solution strategies, with an LLM Instructor that implements those strategies using domain-specific documentation. Its role is computational workflow development, from data preparation through model evaluation, rather than experimental drug discovery or a standalone molecular prediction model. The architecture is described in the paper’s methodology.

A task comprises a natural-language description of objectives and constraints, starter files such as datasets or code templates, and an evaluator. The Planner proposes alternatives and sends selected ideas to the Instructor. The Instructor can read and edit scripts, execute code, consult guidance on biological data acquisition and featurization, and return performance or failure reports. These reports inform subsequent planning; the framework ultimately submits the best-performing idea within its iteration limit. Outputs include implemented ML solutions and evaluation reports, with submission files used in the reported experiments.

The paper studies binary-classification workflows using PAMPA for ADMET, HIV for high-throughput screening, and DAVIS for drug-target interaction prediction. These are limited research case studies, not evidence of broad production readiness. The authors identify basic documentation, restricted benchmark coverage, and hallucination risks, and explicitly call for safety checks and human oversight before deployment. They link an anonymous code repository, but the supplied repository excerpt exposes no files, setup instructions, or software licence.

Key Features

  • Idea-space planning: generates alternative ML strategies, explores selected ideas, and revises the candidate set using implementation success or failure reports.
  • Domain-guided implementation: uses an LLM Instructor to translate modelling ideas into code while consulting specialized drug-discovery documentation.
  • Environment interaction: supports reading and editing scripts and running code as part of the ML experimentation workflow.
  • Biological representation guidance: documentation covers molecule and protein encoding, including fingerprints and graph-based representations.
  • Model preparation guidance: references domain-specific pretrained models, including ChemBERTa for molecules and ESM for protein sequences.
  • Evaluator-driven reporting: returns performance reports for implemented ideas or failure reports when critical functionality is missing.

Use Cases

  • Intended evaluation: reproduce the PAMPA case study to examine how the Planner and Instructor construct an absorption-related classification workflow.
  • Intended evaluation: study HIV assay-outcome classification to assess molecular preprocessing, representation choices, and generated submission validity.
  • Intended evaluation: reproduce DAVIS drug-target interaction classification and inspect how molecular structures and protein sequences are handled.
  • Intended evaluation: compare full DrugAgent with Planner or Instructor ablations to investigate the contribution of idea exploration and domain guidance.

How to Use

  1. Read the paper and methodology to establish the research scope and the division between Planner and Instructor.
  2. Inspect the author-linked code location for available files, dependency instructions, and licence information. The supplied excerpt does not establish an installation procedure or runnable package.
  3. For a reproduction study, choose PAMPA, HIV, or DAVIS and consult the experimental setup. Prepare a task description, starter files, and an evaluator matching the selected classification objective.
  4. Use the documented planning-and-implementation workflow as the reproduction specification. Record proposed ideas, generated code, execution failures, and evaluation reports rather than accepting agent summaries as verified results.
  5. Assess submission validity and predictive evaluation separately. Review the limitations and ethics discussion, require human inspection of generated implementations, and keep this evaluation separate from operational drug-discovery decisions.

Related resources

Therapeutics Data Commons provides therapeutic machine learning datasets, Python loaders, data splits, evaluation metrics and benchmarks for prediction and molecule-generation research.

Open sourcePython

Molecular Property Prediction · Drug Discovery

TorchDrug

Open Source

TorchDrug is a PyTorch-based toolkit for graph and molecular machine learning, with SMILES input, graph operations, property-prediction building blocks and multi-CPU/GPU execution.

Open sourcePython

Molecular Property Prediction · Drug Discovery

AIDDISON Explorer is a hosted drug-discovery platform that generates and ranks molecular candidates against target profiles, design constraints, predicted properties and synthetic feasibility.

Molecular Generation · Drug Discovery

AutoDock Vina

Open Source

AutoDock Vina is an open-source molecular docking and virtual screening program with Vina and AutoDock4.2 scoring, multi-ligand docking, macrocycle support and Python 3 bindings.

Open sourcePython

Drug Discovery

Boltz

Model

Boltz is a biomolecular interaction model family. Boltz-2 provides workflows for complex structure and binding affinity prediction.

Open sourcePython

Drug Discovery

ChemBERTa provides BERT-like models for chemical SMILES, with RoBERTa masked-language modelling checkpoints and notebooks for pre-training, fine-tuning and molecular property prediction research.

Open sourcePython

Molecular Property Prediction

Related guides

Agents

AI Agents for Chemistry: From Chatbots to Autonomous Research

A practical guide to eight chemistry and biomedical research agents: how tools, memory, planning and multi-agent roles work, which projects fit different tasks, and how to evaluate bounded autonomy with scientific oversight.

Workflows

AI for Drug Discovery: Agents, MCP Servers and Platforms

Choose tools by workflow stage: target evidence, protein structures, known binding sites, molecular generation, property analysis and screening. Compare research agents, MCP integrations, hosted APIs and platforms, with a proposed end-to-end example and practical evaluation gates.

Workflows

Protein & Structural Biology AI Workflow

Build an evidence-led protein workflow from UniProt identity and sequences to RCSB structure metadata, observed ligand contacts, and proposed downstream candidate evaluation—with explicit checks for species, isoforms, chains, missing data, and experimental validation.