Skip to content

Commit ac2a4b4

Browse files
committed
pass Databricks Host toi deployments
1 parent 5c430db commit ac2a4b4

File tree

2 files changed

+7
-2
lines changed

2 files changed

+7
-2
lines changed

experimental/apps-mcp/lib/providers/databricks/deployment.go

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@ package databricks
33
import (
44
"context"
55
"fmt"
6+
"os"
67
"os/exec"
78
"time"
89

@@ -53,8 +54,9 @@ func GetUserInfo(ctx context.Context, cfg *mcp.Config) (*iam.User, error) {
5354
return user, nil
5455
}
5556

56-
func SyncWorkspace(appInfo *apps.App, sourceDir string) error {
57+
func SyncWorkspace(ctx context.Context, appInfo *apps.App, sourceDir string) error {
5758
targetPath := GetSourcePath(appInfo)
59+
host := middlewares.MustGetDatabricksClient(ctx).Config.Host
5860

5961
cmd := exec.Command(
6062
"databricks",
@@ -65,6 +67,9 @@ func SyncWorkspace(appInfo *apps.App, sourceDir string) error {
6567
targetPath,
6668
)
6769
cmd.Dir = sourceDir
70+
env := os.Environ()
71+
env = append(env, "DATABRICKS_HOST="+host)
72+
cmd.Env = env
6873

6974
output, err := cmd.CombinedOutput()
7075
if err != nil {

experimental/apps-mcp/lib/providers/deployment/provider.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -179,7 +179,7 @@ func (p *Provider) deployDatabricksApp(ctx context.Context, args *DeployDatabric
179179
syncStart := time.Now()
180180
log.Infof(ctx, "Syncing workspace: source=%s, target=%s", workPath, databricks.GetSourcePath(appInfo))
181181

182-
if err := databricks.SyncWorkspace(appInfo, workPath); err != nil {
182+
if err := databricks.SyncWorkspace(ctx, appInfo, workPath); err != nil {
183183
return &DeployResult{
184184
Success: false,
185185
Message: fmt.Sprintf("Failed to sync workspace: %v", err),

0 commit comments

Comments
 (0)