Python model warehouse cfg with Databricks

The problem I’m having

I have Python models that i want to run with with an All Purpose Compute cluster for staging env and with a Job Compute cluster for production env.
I have tried creating a new Connection for the staging env with an All Purpose Compute cluster and Private Link for the Network but i do not know how to reference this connection from a single Python model(there is no configuration key at all for “connection”).

The context of why I’m trying to do this

I would like to get advice on what is the best practices for achieving that purpose. How can configure Job Compute clusters with the Private Link