Need DockerFile for Labs + Spark + Jupyter-AI

Starting w/ Jupyter & Spark. Need a ‘perfect’ dockerFile for M2/Apple.

Tried to build my own with Jupyter/pyspark-notebook. Assign :8888, boots, get to localhost, switch to ‘JupyterLabs’, click install to install jupyter-ai. ERROR:

ERROR: “There seems to be a problem with the Chat backend, please look at the JupyterLab server logs or contact your administrator to correct this problem.”

Never asks me for the API KEY or setup. Just fails. Does anyone have a PERFECT DockerFile and ComposeUP YAML file that nails this?