app 対応

Author: kazuhitogo

README.md

@@ -165,13 +165,13 @@ cd ./agent/
 `run.sh` および `main.py` にはいくつかのオプションがあります。
 
 ```bash
-# --system-prompt <ファイル名>: カスタムシステムプロンプトファイルを指定
-# -f, --file <ファイル名>: 処理対象のオブジェクトリストファイルを指定（デフォルト: object_list.ini）
+# --mode custom: agent/prompts/custom/custon.txt に記載のプロンプトで動作します。
+# (run.sh のみで動作) -f, --file <ファイル名>: 処理対象のオブジェクトリストファイルを指定（デフォルト: object_list.ini）
 # --avoid-throttling: Bedrockのトークン制限エラー時に自動リトライ及び token 使用量に応じた sleep を有効化
 
 # カスタムシステムプロンプト
-uv run main.py --system-prompt custom_prompt.txt
-./run.sh --system-prompt custom_prompt.txt
+uv run main.py --mode custom
+./run.sh --mode custom
 
 # 一括変換のオブジェクト一覧指定
 ./run.sh -f custom_object_list.ini
@@ -200,7 +200,7 @@ uv run main.py --avoid-throttling
 cd ./agent # リポジトリルートディレクトリがカレントディレクトリの前提です
 
 # アプリケーションの変換
-# 例) uv run main.py --prompt "<ソースの配置場所> <アプリ名> <テスト名>" --system-prompt "system_prompt_app.txt"
+# 例) uv run main.py --prompt "<ソースの配置場所> <アプリ名> <テスト名>" --mode "app"
 uv run main.py --prompt "../application/employee-mgmt/application employee-mgmt test01" --mode "app"
 
 ```

agent/main.py

@@ -118,7 +118,7 @@ def main():
         "--mode",
         type=str,
         default="db_object",
-        help="db_object or app",
+        help="db_object or app or custom",
     )
     parser.add_argument(
         "--avoid-throttling",

agent/prompts/custom/custom.txt

@@ -1,2 +1,2 @@
-あなたはユーザーの望みを叶える親切な AI エージェントです。
-持っているツールを駆使してユーザーの要望に応えてください。
+与えられた Oracle の DB Object をデータベースから抽出して PostgreSQL に変換して実行してください。
+変換後の SQL は ./result/{オブジェクトの名前}.sql に保存してください。

agent/run.sh

@@ -4,13 +4,13 @@
 trap 'echo "\nInterrupted. Exiting..."; exit 1' INT
 
 # Parse options
-SYSTEM_PROMPT_OPT=""
+MODE_OPT=""
 CONFIG_FILE="object_list.ini"
 AVOID_THROTTLING=""
 while [[ $# -gt 0 ]]; do
     case $1 in
-        --system-prompt)
-            SYSTEM_PROMPT_OPT="--system-prompt $2"
+        --mode)
+            MODE_OPT="--mode $2"
             shift 2
             ;;
         -f|--file)
@@ -38,7 +38,7 @@ while IFS= read -r line; do
     fi
 
     echo "Processing: $line"
-    if uv run main.py --prompt "$line" $SYSTEM_PROMPT_OPT $AVOID_THROTTLING; then
+    if uv run main.py --prompt "$line" $MODE_OPT $AVOID_THROTTLING; then
         # Comment out the processed line (macOS compatible)
         if [[ "$OSTYPE" == "darwin"* ]]; then
             # macOS (BSD sed)