@atharvaghodekar, as mentioned in the doc, ai_query with Custom or External Models requires enabling the feature, which isn’t possible in Free Edition for now. However, querying Foundation Model APIs is enabled by default, so you can use ai_query with Databricks-hosted foundation models in Free Edition.