To deal with changes in this [PR](https://github.com/llamastack/llama-stack/pull/3895), there is a need to change e.g. module imports