DatabricksSandbox integration

import time

from databricks.sdk import WorkspaceClient
from databricks.sdk.service.sandbox import (
    ComputeSpec,
    Sandbox,
    SandboxSpec,
    SandboxState,
)
from google.protobuf.duration_pb2 import Duration

from databricks_langchain import DatabricksSandbox

client = WorkspaceClient()

sandbox_id = "my-agent-sandbox"
name = f"sandboxes/{sandbox_id}"

client.sandbox.create_sandbox(
    Sandbox(
        spec=SandboxSpec(
            compute=ComputeSpec(inactivity_timeout=Duration(seconds=900))
        )
    ),
    sandbox_id=sandbox_id,
)

# Wait until the sandbox is RUNNING before running commands.
while client.sandbox.get_sandbox(name).status.state != SandboxState.SANDBOX_STATE_RUNNING:
    time.sleep(3)

backend = DatabricksSandbox(client, name)

result = backend.execute("echo hello")
print(result.output)

from deepagents import create_deep_agent
from langchain_anthropic import ChatAnthropic

# `backend` from the previous step, bound to a RUNNING Databricks sandbox.
agent = create_deep_agent(
    model=ChatAnthropic(model="claude-sonnet-4-6"),
    system_prompt="You are a coding assistant with sandbox access.",
    backend=backend,
)

result = agent.invoke(
    {"messages": [{"role": "user", "content": "Write and run a Python script"}]}
)

client.sandbox.delete_sandbox(name)