☕ JavaMigration

🚀 Repository for automated Java code migration research, part of the MigrationBench project.

• 1. Overview
• 2. MigrationBench Datasets
• 3. Installation
  • 3.1 Prerequisites
  • 3.2 Install Package
• 4. Usage
  • 4.1 Agent Types
  • 4.2 Running Migration
  • 4.3 Command Line Options
• 5. Data
• 6. Resources
• 7. Citation

JavaMigrationAgent is a library for automated code migration from Java 8 to Java 17 (21) using LLM-based agents built on the Strands Agents framework.

It provides multiple agent strategies for migration:

1. Baseline: Direct LLM-based migration
2. PE (Prompt Engineering) (Baseline + PE): Baseline with enhanced prompts for dependency updates
3. RAG (Baseline + PE + RAG): Uses retrieval-augmented generation for dependency version lookup
4. Hybrid (Seed change, followed by baseline + PE): Pre-processes dependencies before LLM migration (More cost effective)

The agent relies on the MigrationBench package for evaluation.

Verify you have java 17 and maven 3.9.6 installed:

# java
$ java --version
openjdk 17.0.15 2025-04-15 LTS
OpenJDK Runtime Environment Corretto-17.0.15.6.1 (build 17.0.15+6-LTS)
OpenJDK 64-Bit Server VM Corretto-17.0.15.6.1 (build 17.0.15+6-LTS, mixed mode, sharing)

# maven
$ mvn --version
Apache Maven 3.9.6 (bc0240f3c744dd6b6ec2920b3cd08dcc295161ae)
Maven home: /usr/local/bin/apache-maven-3.9.6
Java version: 17.0.15, vendor: Amazon.com Inc., runtime: /usr/lib/jvm/java-17-amazon-corretto.x86_64

If you haven't done it yet, follow the instructions in MigrationBench to install Maven.

# cd .../JavaMigration/

pip install -r requirements.txt -e .

Or with uv:

# cd .../JavaMigration/

uv pip install -e .

See the full binary script at src/java_migration_agent/main.py.

Run batch migration on the MigrationBench dataset:

python -m java_migration_agent \
    --agent-type baseline \
    --exp-id exp-001 \
    --hf-dataset AmazonScience/migration-bench-java-selected \
    --model-id global.anthropic.claude-sonnet-4-5-20250929-v1:0 \
    --max-workers 8 \
    --output-dir ./migration_results

Agent trajectories and execution results are stored in the data/ folder.

1. 🤗 MigrationBench (Hugging Face)
2. 💻 MigrationBench (GitHub)
3. 📄 arXiv Paper

@inproceedings{liu2026migrationbench,
  title={Migrationbench: Repository-level code migration benchmark from java 8},
  author={Liu, Linbo and Liu, Xinle and Zhou, Qiang and Chen, Lin and Liu, Yihan and Nguyen, Hoan and Tehrani, Behrooz Omidvar and Shen, Xi and Huan, Jun and Tripp, Omer and others},
  booktitle={Proceedings of the 32nd ACM SIGKDD Conference on Knowledge Discovery and Data Mining V. 2},
  pages={9427--9438},
  year={2026}
}