def fits(input_tokens, max_output, context_limit): return input_tokens + max_output <= context_limit print(fits(3800, 200, 4096)) print(fits(4000, 200, 4096))