rename agent_type to prompt_type; add error handling for unknown prompt_type.

Author: Linbo-Liu

examples/strands_migration_agent/README.md

@@ -1,7 +1,7 @@
 # Strands Migration Agent
 
 This agent tackles the problem of code migration from Java 8 to Java 17 as introduced in [MigrationBench](https://github.com/amazon-science/MigrationBench).
-It is a re-implementation of the [JavaMigrationAgent](https://github.com/amazon-science/JavaMigration/tree/main/java_migration_agent) with open source LLMs.
+It builds upon the official [JavaMigrationAgent](https://github.com/amazon-science/JavaMigration/tree/main/java_migration_agent) with open source LLMs.
 This example is under active development alongside the `agentcore-rl-toolkit` library.
 
 ## Basic Setup
@@ -144,7 +144,7 @@ curl -X POST http://localhost:8080/invocations \
     "repo_uri": "s3://{BUCKET}/tars/test/15093015999__EJServer/15093015999__EJServer.tar.gz",
     "metadata_uri": "s3://{BUCKET}/tars/test/15093015999__EJServer/metadata.json",
     "require_maximal_migration": false,
-    "agent_type": "baseline",
+    "prompt_type": "baseline",
     "_rollout": {
         "exp_id": "dev",
         "s3_bucket": "agentcore-rl",

examples/strands_migration_agent/eval_utils.py

@@ -40,7 +40,7 @@ def get_s3_folder_uris(s3_uri: str) -> list[str]:
     return folder_uris
 
 
-def prepare_payload(folder_uri: str, require_maximal_migration: bool = False, agent_type: str = "baseline") -> dict:
+def prepare_payload(folder_uri: str, require_maximal_migration: bool = False, prompt_type: str = "baseline") -> dict:
     """
     Prepare a single payload for a repository folder.
 
@@ -61,7 +61,7 @@ def prepare_payload(folder_uri: str, require_maximal_migration: bool = False, ag
         "repo_uri": repo_uri,
         "metadata_uri": metadata_uri,
         "require_maximal_migration": require_maximal_migration,
-        "agent_type": agent_type,
+        "prompt_type": prompt_type,
     }
 
 

examples/strands_migration_agent/evaluate.py

@@ -97,11 +97,11 @@ def main():
         help="Whether a repository is evaluated under maximal migration",
     )
     parser.add_argument(
-        "--agent_type",
+        "--prompt_type",
         type=str,
         default="baseline",
         choices=["baseline", "rag", "hybrid"],
-        help="Specify Java migration agent type",
+        help="Specify Java migration prompt type",
     )
 
     args = parser.parse_args()
@@ -127,7 +127,7 @@ def main():
     logger.info(f"Found {len(s3_folder_uris)} repositories to evaluate")
 
     # Prepare payloads
-    payloads = [prepare_payload(uri, args.require_maximal_migration, args.agent_type) for uri in s3_folder_uris]
+    payloads = [prepare_payload(uri, args.require_maximal_migration, args.prompt_type) for uri in s3_folder_uris]
 
     # Setup results directory and file
     results_dir = Path(__file__).parent / "results"

examples/strands_migration_agent/evaluate_async.py

@@ -7,12 +7,7 @@
 import time
 from pathlib import Path
 
-from eval_utils import (
-    append_result_to_file,
-    get_s3_folder_uris,
-    load_config,
-    prepare_payload,
-)
+from eval_utils import append_result_to_file, get_s3_folder_uris, load_config, prepare_payload
 
 from agentcore_rl_toolkit import RolloutClient
 
@@ -229,11 +224,11 @@ async def main():
         help="Whether a repository is evaluated under maximal migration",
     )
     parser.add_argument(
-        "--agent_type",
+        "--prompt_type",
         type=str,
         default="baseline",
         choices=["baseline", "rag", "hybrid"],
-        help="Specify Java migration agent type",
+        help="Specify Java migration prompt type",
     )
 
     args = parser.parse_args()
@@ -260,7 +255,7 @@ async def main():
     logger.info(f"Found {len(s3_folder_uris)} repositories to evaluate")
 
     # Prepare payloads
-    payloads = [prepare_payload(uri, args.require_maximal_migration, args.agent_type) for uri in s3_folder_uris]
+    payloads = [prepare_payload(uri, args.require_maximal_migration, args.prompt_type) for uri in s3_folder_uris]
 
     # Setup results directory and file
     results_dir = Path(__file__).parent / "results"

examples/strands_migration_agent/rl_app.py

@@ -44,7 +44,7 @@
 def invoke_agent(payload: dict):
     base_url = payload["_rollout"]["base_url"]
     model_id = payload["_rollout"]["model_id"]
-    agent_type = payload.get("agent_type", "baseline")
+    prompt_type = payload.get("prompt_type", "baseline")
     params = payload["_rollout"].get("sampling_params", {})
     tools = [shell, editor]
 
@@ -54,7 +54,7 @@ def invoke_agent(payload: dict):
         prompt += (
             "\nYou should update all dependencies in the `pom.xml` file to their latest versions that support Java 17."
         )
-    if agent_type == "rag":
+    if prompt_type == "rag":
         prompt += (
             "\nYou have access to a dependency version lookup tool. When updating dependencies "
             "in pom.xml:\n"
@@ -65,14 +65,16 @@ def invoke_agent(payload: dict):
             "3. Update all dependencies to their Java 17 compatible versions"
         )
         tools.append(search_dependency_version)
-    elif agent_type == "hybrid":
+    elif prompt_type == "hybrid":
         prompt += (
             "\nDependencies in the `pom.xml` file have been updated to their "
             "latest versions that support Java 17, but these changes might introduce "
             "compatibility issues in the codebase. Please fix any such issues in your "
             "migration. Do not downgrade the dependency versions back to their JDK 8 "
             "compatible versions."
         )
+    elif prompt_type != "baseline":
+        logger.warning(f"Unavailable prompt_type: {prompt_type}. Set to default prompt_type baseline.")
 
     model = OpenAIModel(client_args={"api_key": "EMPTY", "base_url": base_url}, model_id=model_id, params=params)
 
@@ -90,7 +92,7 @@ def invoke_agent(payload: dict):
     logger.info(f"Loaded repo into: {repo_path} (took {load_duration:.2f}s)")
 
     start_time = time.time()
-    setup_repo_environment(repo_path, agent_type)
+    setup_repo_environment(repo_path, prompt_type)
     setup_duration = time.time() - start_time
     logger.info(f"Finished repo setup for: {repo_path} (took {setup_duration:.2f}s)")
 

examples/strands_migration_agent/utils.py

@@ -30,7 +30,7 @@ def load_metadata_from_s3(s3_uri: str) -> dict:
     return json.loads(content)
 
 
-def setup_repo_environment(repo_path: str, agent_type: str = "baseline"):
+def setup_repo_environment(repo_path: str, prompt_type: str = "baseline"):
     """
     1. Pre-warm Maven caches (best-effort)
     2. Make sure git works.
@@ -58,7 +58,7 @@ def setup_repo_environment(repo_path: str, agent_type: str = "baseline"):
         capture_output=True,
     )
     logger.info("git working properly!")
-    if agent_type == "hybrid":
+    if prompt_type == "hybrid":
         logger.info("static update on jdk and dependency versions")
         update_jdk_related(repo_path)
         update_dependency_version(repo_path)