devrel-examples/demos/bee-pollinator/databricks.yml at main · databricks-solutions/devrel-examples · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
# Databricks Asset Bundle configuration for the Bee Pollinator demo.
#
# Deploys a single job that loads data, uploads PDFs, and creates
# the Genie Space, Knowledge Assistant, and Supervisor Agent end-to-end.
# Fully programmatic — no UI steps required.
#
# Usage:
#   databricks bundle validate
#   databricks bundle deploy
#   databricks bundle run setup_demo

bundle:
  name: bee-pollinator-demo

variables:
  catalog:
    description: Unity Catalog catalog name
    default: main
  schema:
    description: Schema name for bee health data
    default: bee_pollinator
  warehouse_id:
    description: SQL Warehouse ID (for Genie Space)

resources:
  jobs:
    setup_demo:
      name: "[${bundle.target}] bee-pollinator-setup"
      tasks:
        - task_key: load_data
          notebook_task:
            notebook_path: ./scripts/load_data.py
            source: WORKSPACE
            base_parameters:
              catalog: ${var.catalog}
              schema: ${var.schema}
          environment_key: default
        - task_key: create_agents
          depends_on:
            - task_key: load_data
          notebook_task:
            notebook_path: ./scripts/create_agents.py
            source: WORKSPACE
            base_parameters:
              catalog: ${var.catalog}
              schema: ${var.schema}
              warehouse_id: ${var.warehouse_id}
          environment_key: default
      environments:
        - environment_key: default
          spec:
            client: "1"
            dependencies:
              # >=0.106.0 exposes w.supervisor_agents (programmatic
              # Supervisor Agent creation, no UI step required).
              - databricks-sdk>=0.106.0

targets:
  dev:
    mode: development
    default: true