databricks_ai_bridge_quickstart_unity_catalog_query.py

1from databricks.ai_bridge import DatabricksAIBridge
2
3# Initialize the AI Bridge
4# If running inside a Databricks Notebook, it will automatically use the current context.
5# Otherwise, ensure DATABRICKS_HOST and DATABRICKS_TOKEN environment variables are set.
6bridge = DatabricksAIBridge()
7
8# Example: Using the bridge to execute a query or interact with data
9# Note: Specific methods may vary based on the version, but the primary entry point 
10# is the DatabricksAIBridge class.
11result = bridge.display_data("SELECT * FROM samples.nyctaxi.trips LIMIT 5")
12
13print(result)