Skip to content

Commit 845c435

Browse files
authored
Add AI tools integration (#2025)
## Changes Integrate the `databricks aitools` family of commands into the extension. - on install (or first launch of the extension with this feature) it prompts to install AI tools - add `AI tools` row to the configuration pane ## Tests Added tests pass
1 parent cdd5b8c commit 845c435

24 files changed

Lines changed: 4335 additions & 78 deletions

packages/databricks-vscode/package.json

Lines changed: 85 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -98,6 +98,48 @@
9898
"enablement": "databricks.context.activated && databricks.context.loggedIn && !databricks.context.remoteMode",
9999
"icon": "$(debug-disconnect)"
100100
},
101+
{
102+
"command": "databricks.aitools.install",
103+
"title": "Install AI tools",
104+
"category": "Databricks",
105+
"icon": "$(cloud-download)"
106+
},
107+
{
108+
"command": "databricks.aitools.checkForUpdates",
109+
"title": "Check for AI tools updates",
110+
"category": "Databricks",
111+
"icon": "$(refresh)"
112+
},
113+
{
114+
"command": "databricks.aitools.reload",
115+
"title": "Reload AI tools",
116+
"category": "Databricks",
117+
"icon": "$(refresh)"
118+
},
119+
{
120+
"command": "databricks.aitools.update",
121+
"title": "Update AI tools",
122+
"category": "Databricks",
123+
"icon": "$(arrow-circle-up)"
124+
},
125+
{
126+
"command": "databricks.aitools.uninstall",
127+
"title": "Uninstall AI tools",
128+
"category": "Databricks",
129+
"icon": "$(trash)"
130+
},
131+
{
132+
"command": "databricks.aitools.addCursorPlugin",
133+
"title": "Add Databricks plugin to Cursor",
134+
"category": "Databricks",
135+
"icon": "$(plug)"
136+
},
137+
{
138+
"command": "databricks.aitools.installAgent",
139+
"title": "Install AI tools for this agent",
140+
"category": "Databricks",
141+
"icon": "$(cloud-download)"
142+
},
101143
{
102144
"command": "databricks.cluster.filterByAll",
103145
"title": "All",
@@ -1009,6 +1051,21 @@
10091051
"when": "view == configurationView && viewItem =~ /databricks.configuration.cluster.*\\.terminated.*/ && databricks.context.bundle.deploymentState == idle",
10101052
"group": "navigation@0"
10111053
},
1054+
{
1055+
"command": "databricks.aitools.addCursorPlugin",
1056+
"when": "view == configurationView && viewItem =~ /^databricks.configuration.aitools.(installed|upToDate|updateAvailable|checking|updating|error)$/ && databricks.context.aitools.showCursorPlugin",
1057+
"group": "inline@0"
1058+
},
1059+
{
1060+
"command": "databricks.aitools.uninstall",
1061+
"when": "view == configurationView && viewItem =~ /^databricks.configuration.aitools.(installed|upToDate|updateAvailable|checking|updating|error)$/",
1062+
"group": "navigation@9"
1063+
},
1064+
{
1065+
"command": "databricks.aitools.installAgent",
1066+
"when": "view == configurationView && viewItem == databricks.configuration.aitools.agent.notInstalled",
1067+
"group": "inline@0"
1068+
},
10121069
{
10131070
"command": "databricks.bundle.deployAndRunJob",
10141071
"when": "view == dabsResourceExplorerView && viewItem =~ /^databricks.bundle.resource=jobs.runnable.*$/ && databricks.context.bundle.deploymentState == idle",
@@ -1161,6 +1218,34 @@
11611218
}
11621219
],
11631220
"commandPalette": [
1221+
{
1222+
"command": "databricks.aitools.addCursorPlugin",
1223+
"when": "databricks.context.aitools.showCursorPlugin && !databricks.context.remoteMode"
1224+
},
1225+
{
1226+
"command": "databricks.aitools.install",
1227+
"when": "!databricks.context.aitools.installed && !databricks.context.remoteMode"
1228+
},
1229+
{
1230+
"command": "databricks.aitools.uninstall",
1231+
"when": "databricks.context.aitools.installed && !databricks.context.remoteMode"
1232+
},
1233+
{
1234+
"command": "databricks.aitools.update",
1235+
"when": "databricks.context.aitools.installed && !databricks.context.remoteMode"
1236+
},
1237+
{
1238+
"command": "databricks.aitools.checkForUpdates",
1239+
"when": "databricks.context.aitools.installed && !databricks.context.remoteMode"
1240+
},
1241+
{
1242+
"command": "databricks.aitools.reload",
1243+
"when": "false"
1244+
},
1245+
{
1246+
"command": "databricks.aitools.installAgent",
1247+
"when": "false"
1248+
},
11641249
{
11651250
"command": "databricks.environment.setupPythonEnv",
11661251
"when": "false"

0 commit comments

Comments
 (0)