fn custom_reward(response: String) -> Float {
    let len = str_len(response)
    let score = 0.0
    if len > 10 {
        let score = 0.5
    }
    if len > 30 {
        let score = 1.0
    }
    return score
}

fn main() {
    println("Testing custom reward in training context")
    let r = custom_reward("hello this is a longer sentence for testing")
    println("Custom reward: " + r)
    println("It works!")
}
