第一次學 Kotlin Koog AI 就上手 Day 22：策略圖基礎：構建多階段 AI 工作流程

// 簡單的單次執行策略
val agent = AIAgent(
    executor = simpleOpenAIExecutor(ApiKeyManager.openAIApiKey!!),
    strategy = singleRunStrategy(), // 簡單策略
    systemPrompt = "你是一個客服助手"
)

val strategy = strategy<String, String>("strategy_name") {
    // 在這裡定義節點和邊
}

val processNode by node<String, String>("process_data") { input ->
    // 處理輸入資料
    val result = "處理後的資料：$input"
    result // 返回處理結果
}

// 基本連接
edge(nodeA forwardTo nodeB)

// 條件連接
edge(nodeA forwardTo nodeB onCondition { output ->
    output.length > 10
})

// 轉換輸出
edge(nodeA forwardTo nodeB transformed { output ->
    output.uppercase()
})

flowchart TD
    A[開始] --> B[驗證訂單節點]
    B --> C{驗證結果}
    C -->|驗證通過| D[計算價格節點]
    C -->|驗證失敗| E[錯誤處理節點]
    D --> F[生成確認節點]
    F --> G[結束]
    E --> G[結束]

    style A fill:#e1f5fe
    style B fill:#f3e5f5
    style C fill:#fff3e0
    style D fill:#f3e5f5
    style E fill:#ffebee
    style F fill:#f3e5f5
    style G fill:#e8f5e8

class OrderProcessingAgent {

    private val agent = AIAgent(
        executor = simpleOpenAIExecutor(ApiKeyManager.openAIApiKey!!),
        systemPrompt = "你是一個專業的訂單處理助手",
        llmModel = OpenAIModels.CostOptimized.GPT4_1Mini,
        // 使用自訂策略圖
        strategy = createOrderProcessingStrategy()
    )

    private fun createOrderProcessingStrategy() = strategy<String, String>("order_processing") {

        // 節點一：驗證訂單資料
        val validateOrderNode by node<String, String>("validate_order") { orderData ->
            println("🔍 正在驗證訂單資料...")

            // 模擬訂單驗證邏輯
            val lines = orderData.split("\n")
            val hasCustomerInfo = lines.any { it.contains("客戶：") }
            val hasProductInfo = lines.any { it.contains("商品：") }
            val hasAmount = lines.any { it.contains("金額：") }

            when {
                !hasCustomerInfo -> "錯誤：缺少客戶資訊"
                !hasProductInfo -> "錯誤：缺少商品資訊"
                !hasAmount -> "錯誤：缺少金額資訊"
                else -> {
                    println("✅ 訂單驗證通過")
                    "驗證通過：$orderData"
                }
            }
        }

        // 節點二：計算價格
        val calculatePriceNode by node<String, String>("calculate_price") { validatedOrder ->
            println("💰 正在計算訂單價格...")

            // 從訂單中提取金額
            val amountLine = validatedOrder.split("\n")
                .find { it.contains("金額：") }

            val amount = amountLine?.substringAfter("金額：")?.trim()?.toDoubleOrNull() ?: 0.0
            val tax = amount * 0.05 // 5% 稅金
            val total = amount + tax

            val result = "$validatedOrder\n稅金：$tax\n總計：$total"
            println("✅ 價格計算完成，總計：$total")
            result
        }

        // 節點三：生成訂單確認
        val generateConfirmationNode by node<String, String>("generate_confirmation") { orderWithPrice ->
            println("📝 正在生成訂單確認...")

            val confirmation = """
                📋 訂單確認書
                ================
                $orderWithPrice
                ================
                狀態：已確認
                確認時間：${java.time.LocalDateTime.now()}
            """.trimIndent()

            println("✅ 訂單確認書生成完成")
            confirmation
        }

        // 節點四：錯誤處理
        val handleErrorNode by node<String, String>("handle_error") { errorMessage ->
            println("❌ 處理訂單錯誤")

            """
                ⚠️ 訂單處理失敗
                錯誤原因：$errorMessage
                請檢查訂單資料並重新提交
            """.trimIndent()
        }

        // 定義執行流程
        edge(nodeStart forwardTo validateOrderNode)

        // 驗證成功 -> 計算價格
        edge(validateOrderNode forwardTo calculatePriceNode onCondition { result ->
            result.startsWith("驗證通過")
        })

        // 驗證失敗 -> 錯誤處理
        edge(validateOrderNode forwardTo handleErrorNode onCondition { result ->
            result.startsWith("錯誤")
        })

        // 計算價格 -> 生成確認
        edge(calculatePriceNode forwardTo generateConfirmationNode)

        // 所有路徑最終都到達結束點
        edge(generateConfirmationNode forwardTo nodeFinish)
        edge(handleErrorNode forwardTo nodeFinish)
    }

    suspend fun processOrder(orderData: String): String {
        return agent.run(orderData)
    }
}

suspend fun main() {
    val processor = OrderProcessingAgent()

    println("=== 訂單處理策略圖演示 ===\n")

    // 測試正常訂單
    val validOrder = """
        客戶：張小明
        商品：筆記型電腦
        金額：50000
    """.trimIndent()

    println("📝 處理正常訂單：")
    try {
        val result = processor.processOrder(validOrder)
        println("\n🎯 處理結果：")
        println(result)
    } catch (e: Exception) {
        println("❌ 處理失敗：${e.message}")
    }

    println("\n" + "=".repeat(50) + "\n")

    // 測試異常訂單
    val invalidOrder = """
        客戶：李小華
        商品：智慧型手機
        // 缺少金額資訊
    """.trimIndent()

    println("📝 處理異常訂單：")
    try {
        val result = processor.processOrder(invalidOrder)
        println("\n🎯 處理結果：")
        println(result)
    } catch (e: Exception) {
        println("❌ 處理失敗：${e.message}")
    }
}

=== 訂單處理策略圖演示 ===

📝 處理正常訂單：
🔍 正在驗證訂單資料...
✅ 訂單驗證通過
💰 正在計算訂單價格...
✅ 價格計算完成，總計：52500.0
📝 正在生成訂單確認...
✅ 訂單確認書生成完成

🎯 處理結果：
                📋 訂單確認書
                ================
                驗證通過：客戶：張小明
商品：筆記型電腦
金額：50000
稅金：2500.0
總計：52500.0
                ================
                狀態：已確認
                確認時間：2025-08-17T17:31:42.787342

==================================================

📝 處理異常訂單：
🔍 正在驗證訂單資料...
❌ 處理訂單錯誤

🎯 處理結果：
⚠️ 訂單處理失敗
錯誤原因：錯誤：缺少金額資訊
請檢查訂單資料並重新提交