Elegir el contexto adecuado en tareas de programación

Eres profesor y mantienes un proyecto de Python para gestionar expedientes de alumnos. Las calificaciones se guardan en un diccionario, con funciones de ayuda para obtener un estudiante y calcular su media. Las funciones existentes están guardadas en students.py, un archivo al que el LLM puede acceder y usar:

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

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

Quieres usar estos helpers para implementar una función add_grade() que añada una calificación al expediente de un estudiante y devuelva la media actualizada.