export def "nu-complete tobi" [context: string, offset: int] {
    let tokens = ($context | split row " ")
    let query = if ($tokens | length) > 1 { $tokens | last } else { "" }
    try {
        with-env { TOBI_COMPLETE: "1" } {
            ^tobi $query | from json
        }
    } catch {
        []
    }
}

export def --wrapped tobi [
    query?: string@"nu-complete tobi"
    ...args
] {
    if ($query | is-empty) {
        ^tobi
    } else {
        ^tobi $query ...$args
    }
}