Escolhendo o contexto certo em tarefas de código

Você é um professor mantendo um projeto em Python para registros de alunos. As notas são armazenadas em um dicionário, com funções auxiliares para buscar um aluno e calcular sua média. As funções existentes estão salvas em students.py, um arquivo que o LLM pode acessar e usar:

def get_student(student_id):
  return {"name": "Alice", "grades": [88, 92, 95]}

def calculate_average(grades):
  return round(sum(grades) / len(grades), 2)

Você quer usar essas auxiliares para implementar uma função add_grade() que adiciona uma nota ao registro do aluno e retorna a média atualizada.