AI completion in WASM notebooks?

[alchem0x2A]

I'm looking for a way to have usable AI completion in wasm notebooks. I know that marimo.app wasm notebooks come with the AI completion so I think it should also be able to deploy that in static web deployment. It seem however whenever running wasm notebooks at local test environment, I always get this type of error (currently only using gemini)

the html-wasm notebooks are generated by marimo export html-wasm --mode edit some-notebook.py.

Basically my question is whether one can achieve the same wasm setup as marimo.app, as clearly the AI feature is possible there.

If possible are the marimo team maintaining a repo of the config for marimo.app website?

[alchem0x2A]

would be nice if someone has similar requests!