The LLM should provide code for a specific calculation based on the given dataframe, e.g. calculate the mean of a column