As per the doc , one can execute an ai_query in python. But I am sot sure why I can'tWorks in SQLselect 1 as colA, ai_query('databricks-llama-4-maverick', 'what is 2+2') as testBut not in sparkdf = spark.sql("select 1 as colA")
df = df.selectExpr(""...