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

Author: Linbo-Liu

examples/strands_migration_agent/README.md

@@ -141,10 +141,10 @@ curl -X POST http://localhost:8080/invocations \
   -H "Content-Type: application/json" \
   -d '{
     "prompt": "Please help migrate this repo: {repo_path}. There are {num_tests} test cases in it.",
-    "repo_uri": "s3://{BUCKET}/tars/test/15093015999__EJServer/15093015999__EJServer.tar.gz",
-    "metadata_uri": "s3://{BUCKET}/tars/test/15093015999__EJServer/metadata.json",
+    "repo_uri": "s3://my-migration-bench-data/tars/test/15093015999__EJServer/15093015999__EJServer.tar.gz",
+    "metadata_uri": "s3://my-migration-bench-data/tars/test/15093015999__EJServer/metadata.json",
     "require_maximal_migration": false,
-    "agent_type": "baseline",
+    "prompt_type": "dummy",
     "_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

@@ -229,11 +229,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 +260,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)