Fix Key Handling

The model receives the prompt (and optional system message). The run uses scorer contains_all with the JSON configuration below. Pass/fail and partial credit are determined entirely by that scorer against the model output; no human grading.

Fix the bug and return corrected Python code only.

def user_city(user):
    return user["address"]["city"].strip().lower()

{
  "expected_contains": [
    "address = user.get(\"address\")",
    "if not address",
    "address.get(\"city\")"
  ]
}