Skip to content

Commit 5cd1ecf

Browse files
committed
refactor: rename to dbx-agent-app, add eval bridge
- Rename import path: databricks_agents → dbx_agent_app - Rename PyPI package: databricks-agent-deploy → dbx-agent-app - Rename CLI command: databricks-agents → dbx-agent-app - Add bridge module (dbx_agent_app.bridge.eval) with app_predict_fn() for mlflow.genai.evaluate() interop with official databricks-agents - Update all src/, tests/, examples/, docs/, configs - 186 tests passing
1 parent a42025a commit 5cd1ecf

310 files changed

Lines changed: 2899 additions & 2527 deletions

File tree

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

databricks-agents/.github/workflows/test.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ jobs:
4242
4343
- name: Run tests with pytest
4444
run: |
45-
pytest tests/ -v --cov=databricks_agents --cov-report=xml --cov-report=term
45+
pytest tests/ -v --cov=dbx_agent_app --cov-report=xml --cov-report=term
4646
4747
- name: Upload coverage to Codecov
4848
uses: codecov/codecov-action@v4

databricks-agents/CLAUDE.md

Lines changed: 18 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,8 @@
1-
# databricks-agent-deploy
1+
# dbx-agent-app
22

33
## Project Overview
44

5-
Agent platform for Databricks Apps. Package name: `databricks-agent-deploy` (PyPI), import as `databricks_agents` (Python).
5+
Agent platform for Databricks Apps. Package name: `dbx-agent-app` (PyPI), import as `dbx_agent_app` (Python).
66

77
Build agents with any framework (official Databricks SDK, LangGraph, CrewAI, plain FastAPI). Deploy the platform to discover, test, trace, and govern all of them.
88

@@ -27,7 +27,7 @@ This is NOT an agent-building SDK. It's the **control plane** for agents in your
2727
The recommended way to build agents. One decorator gives you `/invocations`, agent card, health, and MCP.
2828

2929
```python
30-
from databricks_agents import app_agent, AgentRequest, AgentResponse
30+
from dbx_agent_app import app_agent, AgentRequest, AgentResponse
3131

3232
@app_agent(name="my_agent", description="Does stuff", capabilities=["search"])
3333
async def my_agent(request: AgentRequest) -> AgentResponse:
@@ -52,7 +52,7 @@ SDK types that replace `mlflow.types.responses` — no mlflow dependency needed:
5252
For agents using LangChain, replace `self.prep_msgs_for_llm()` with:
5353

5454
```python
55-
from databricks_agents.core.compat import to_langchain_messages
55+
from dbx_agent_app.core.compat import to_langchain_messages
5656
messages = to_langchain_messages(request) # -> List[BaseMessage]
5757
```
5858

@@ -62,7 +62,7 @@ For agents that want full control over their FastAPI app:
6262

6363
```python
6464
from fastapi import FastAPI
65-
from databricks_agents import add_agent_card, add_mcp_endpoints
65+
from dbx_agent_app import add_agent_card, add_mcp_endpoints
6666

6767
app = FastAPI()
6868

@@ -75,20 +75,20 @@ add_mcp_endpoints(app, tools=[...]) # optional
7575

7676
## Project Structure
7777

78-
- `src/databricks_agents/core/``@app_agent` decorator, wire types, helpers, compat
79-
- `src/databricks_agents/dashboard/`**the platform app** (FastAPI + React SPA)
80-
- `src/databricks_agents/deploy/` — multi-agent deploy orchestration (agents.yaml)
81-
- `src/databricks_agents/discovery/` — workspace agent scanning
82-
- `src/databricks_agents/registry/` — Unity Catalog registration
83-
- `src/databricks_agents/mcp/` — MCP JSON-RPC server
84-
- `src/databricks_agents/cli.py` — CLI entry point
78+
- `src/dbx_agent_app/core/``@app_agent` decorator, wire types, helpers, compat
79+
- `src/dbx_agent_app/dashboard/`**the platform app** (FastAPI + React SPA)
80+
- `src/dbx_agent_app/deploy/` — multi-agent deploy orchestration (agents.yaml)
81+
- `src/dbx_agent_app/discovery/` — workspace agent scanning
82+
- `src/dbx_agent_app/registry/` — Unity Catalog registration
83+
- `src/dbx_agent_app/mcp/` — MCP JSON-RPC server
84+
- `src/dbx_agent_app/cli.py` — CLI entry point
8585
- `examples/` — example agents using helpers + official framework
8686
- `tests/` — SDK tests
8787

8888
## Key Conventions
8989

9090
- SDK dependencies stay minimal (FastAPI, uvicorn, pydantic, httpx, databricks-sdk)
91-
- mlflow is optional (`pip install databricks-agent-deploy[mlflow]`)
91+
- mlflow is optional (`pip install dbx-agent-app[mlflow]`)
9292
- The dashboard is the primary product, not a side feature
9393
- `agents.yaml` defines agent systems (topology, dependencies, wiring)
9494
- Examples use `@app_agent` decorator (preferred) or plain FastAPI + `add_agent_card()`
@@ -97,9 +97,9 @@ add_mcp_endpoints(app, tools=[...]) # optional
9797
## CLI Commands
9898

9999
```
100-
databricks-agents deploy # Deploy agents from agents.yaml
101-
databricks-agents status # Show deployment status
102-
databricks-agents destroy # Tear down deployed agents
103-
databricks-agents dashboard # Launch platform locally (dev)
104-
databricks-agents platform # Deploy platform as a Databricks App
100+
dbx-agent-app deploy # Deploy agents from agents.yaml
101+
dbx-agent-app status # Show deployment status
102+
dbx-agent-app destroy # Tear down deployed agents
103+
dbx-agent-app dashboard # Launch platform locally (dev)
104+
dbx-agent-app platform # Deploy platform as a Databricks App
105105
```

databricks-agents/CONTRIBUTING.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
1-
# Contributing to databricks-agents
1+
# Contributing to dbx-agent-app
22

3-
Thank you for your interest in contributing to databricks-agents! This project is a Databricks Labs initiative to make it easier to build discoverable AI agents on Databricks Apps.
3+
Thank you for your interest in contributing to dbx-agent-app! This project is a Databricks Labs initiative to make it easier to build discoverable AI agents on Databricks Apps.
44

55
## Development Setup
66

databricks-agents/DEPLOYMENT_GUIDE.md

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -12,8 +12,8 @@ This framework is ready for deployment to the [Databricks Labs Sandbox](https://
1212
## Repository Structure
1313

1414
```
15-
databricks-agents/
16-
├── src/databricks_agents/ # Core framework
15+
dbx-agent-app/
16+
├── src/dbx_agent_app/ # Core framework
1717
│ ├── core/ # @app_agent decorator, agent request/response types
1818
│ ├── discovery/ # Agent discovery, A2A client
1919
│ ├── mcp/ # MCP server, UC Functions
@@ -74,25 +74,25 @@ databricks-agents/
7474
git clone <current-location>
7575

7676
# Add sandbox as remote
77-
cd databricks-agents
77+
cd dbx-agent-app
7878
git remote add sandbox git@github.com:databrickslabs/sandbox.git
7979

8080
# Create feature branch
81-
git checkout -b databricks-agents-framework
81+
git checkout -b dbx-agent-app-framework
8282

8383
# Push to sandbox
84-
git push sandbox databricks-agents-framework
84+
git push sandbox dbx-agent-app-framework
8585
```
8686

8787
### 2. Create PR to sandbox/main
8888

89-
**PR Title**: Add databricks-agents framework for building discoverable agents
89+
**PR Title**: Add dbx-agent-app framework for building discoverable agents
9090

9191
**PR Description**:
9292
```markdown
9393
## Summary
9494

95-
Adds `databricks-agents`, a lightweight Python framework for building discoverable AI agents on Databricks Apps. This framework makes it trivial to turn any Databricks App into a standards-compliant agent with auto-generated A2A protocol endpoints.
95+
Adds `dbx-agent-app`, a lightweight Python framework for building discoverable AI agents on Databricks Apps. This framework makes it trivial to turn any Databricks App into a standards-compliant agent with auto-generated A2A protocol endpoints.
9696

9797
## What It Does
9898

@@ -104,18 +104,18 @@ Adds `databricks-agents`, a lightweight Python framework for building discoverab
104104

105105
## Key Files
106106

107-
- `src/databricks_agents/core/decorator.py` - @app_agent decorator
108-
- `src/databricks_agents/core/types.py` - AgentRequest, AgentResponse types
109-
- `src/databricks_agents/discovery/` - Agent discovery and A2A client
110-
- `src/databricks_agents/registry/` - Unity Catalog integration
111-
- `src/databricks_agents/mcp/` - MCP server support
107+
- `src/dbx_agent_app/core/decorator.py` - @app_agent decorator
108+
- `src/dbx_agent_app/core/types.py` - AgentRequest, AgentResponse types
109+
- `src/dbx_agent_app/discovery/` - Agent discovery and A2A client
110+
- `src/dbx_agent_app/registry/` - Unity Catalog integration
111+
- `src/dbx_agent_app/mcp/` - MCP server support
112112
- `examples/` - Complete working examples
113113
- `tests/` - Test suite
114114

115115
## Example Usage
116116

117117
```python
118-
from databricks_agents import app_agent, AgentRequest, AgentResponse
118+
from dbx_agent_app import app_agent, AgentRequest, AgentResponse
119119

120120
@app_agent(
121121
name="customer_research",

databricks-agents/FRAMEWORK_OVERVIEW.md

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# databricks-agents Framework Overview
1+
# dbx-agent-app Framework Overview
22

33
## What We've Built
44

@@ -21,8 +21,8 @@ This framework treats Databricks Apps as first-class agents, allowing them to:
2121
## Framework Structure
2222

2323
```
24-
databricks-agents/
25-
├── src/databricks_agents/
24+
dbx-agent-app/
25+
├── src/dbx_agent_app/
2626
│ ├── core/
2727
│ │ ├── __init__.py
2828
│ │ ├── decorator.py # @app_agent decorator
@@ -62,7 +62,7 @@ Python decorator that transforms an async function into an agent with auto-gener
6262

6363
**Usage:**
6464
```python
65-
from databricks_agents import app_agent, AgentRequest, AgentResponse
65+
from dbx_agent_app import app_agent, AgentRequest, AgentResponse
6666

6767
@app_agent(
6868
name="my_agent",
@@ -84,7 +84,7 @@ Discovers agent-enabled apps in your workspace by:
8484

8585
**Usage:**
8686
```python
87-
from databricks_agents.discovery import AgentDiscovery
87+
from dbx_agent_app.discovery import AgentDiscovery
8888

8989
discovery = AgentDiscovery(profile="my-profile")
9090
result = await discovery.discover_agents()
@@ -103,7 +103,7 @@ Communicates with agents using A2A protocol:
103103

104104
**Usage:**
105105
```python
106-
from databricks_agents.discovery import A2AClient
106+
from dbx_agent_app.discovery import A2AClient
107107

108108
async with A2AClient() as client:
109109
card = await client.fetch_agent_card(agent_url)
@@ -138,7 +138,7 @@ Delegates to Databricks workspace OIDC provider for authentication
138138
### Creating an Agent with @app_agent
139139
```python
140140
# examples/customer_research_agent.py
141-
from databricks_agents import app_agent, AgentRequest, AgentResponse
141+
from dbx_agent_app import app_agent, AgentRequest, AgentResponse
142142

143143
@app_agent(
144144
name="customer_research",
@@ -155,7 +155,7 @@ app = customer_research.app
155155
```python
156156
# examples/plain_fastapi_agent.py
157157
from fastapi import FastAPI
158-
from databricks_agents import add_agent_card
158+
from dbx_agent_app import add_agent_card
159159

160160
app = FastAPI()
161161

@@ -174,7 +174,7 @@ add_agent_card(
174174
### Discovering Agents
175175
```python
176176
# examples/discover_agents.py
177-
from databricks_agents.discovery import AgentDiscovery
177+
from dbx_agent_app.discovery import AgentDiscovery
178178

179179
discovery = AgentDiscovery(profile="my-profile")
180180
result = await discovery.discover_agents()

databricks-agents/README.md

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,10 @@
1-
# databricks-agents
1+
# dbx-agent-app
22

33
A lightweight Python framework for building discoverable AI agents on Databricks Apps that automatically expose A2A (Agent-to-Agent) protocol endpoints.
44

55
## What It Does
66

7-
The `databricks-agents` framework makes it trivial to turn a Databricks App into a discoverable, standards-compliant agent:
7+
The `dbx-agent-app` framework makes it trivial to turn a Databricks App into a discoverable, standards-compliant agent:
88

99
- **Auto-generates A2A protocol endpoints** (`/.well-known/agent.json`, `/.well-known/openid-configuration`)
1010
- **Wraps FastAPI** to seamlessly integrate agent capabilities with your web app
@@ -31,21 +31,21 @@ The [A2A (Agent-to-Agent) protocol](https://a2a.so/) provides a standard way for
3131
## Installation
3232

3333
```bash
34-
pip install databricks-agents
34+
pip install dbx-agent-app
3535
```
3636

3737
Or with development dependencies:
3838

3939
```bash
40-
pip install databricks-agents[dev]
40+
pip install dbx-agent-app[dev]
4141
```
4242

4343
## Quick Start
4444

4545
### 1. Create an Agent
4646

4747
```python
48-
from databricks_agents import app_agent, AgentRequest, AgentResponse
48+
from dbx_agent_app import app_agent, AgentRequest, AgentResponse
4949

5050
# Create your agent with capabilities
5151
@app_agent(
@@ -94,7 +94,7 @@ databricks apps deploy customer-research --source-code-path ./
9494

9595
```python
9696
import asyncio
97-
from databricks_agents.discovery import AgentDiscovery
97+
from dbx_agent_app.discovery import AgentDiscovery
9898

9999
async def main():
100100
discovery = AgentDiscovery(profile="my-profile")
@@ -168,7 +168,7 @@ Standard health check endpoint:
168168
The `AgentDiscovery` class scans your workspace for agent-enabled apps:
169169

170170
```python
171-
from databricks_agents.discovery import AgentDiscovery
171+
from dbx_agent_app.discovery import AgentDiscovery
172172

173173
# Initialize with optional profile
174174
discovery = AgentDiscovery(profile="my-profile")
@@ -196,7 +196,7 @@ if result.errors:
196196
Communicate with other agents using the A2A protocol:
197197

198198
```python
199-
from databricks_agents.discovery import A2AClient
199+
from dbx_agent_app.discovery import A2AClient
200200

201201
async with A2AClient() as client:
202202
# Fetch an agent's card
@@ -222,7 +222,7 @@ For agents using plain FastAPI with the `add_agent_card()` helper, tools can be
222222
Example:
223223

224224
```python
225-
from databricks_agents import app_agent, AgentRequest, AgentResponse
225+
from dbx_agent_app import app_agent, AgentRequest, AgentResponse
226226

227227
@app_agent(
228228
name="customer_research",
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
11
# A2AClient
22

33
!!! info "Coming Soon"
4-
This page is under development. See the [source code](https://github.com/databricks-labs/databricks-agents) for current implementation details.
4+
This page is under development. See the [source code](https://github.com/databricks-labs/dbx-agent-app) for current implementation details.
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
11
# AgentDiscovery
22

33
!!! info "Coming Soon"
4-
This page is under development. See the [source code](https://github.com/databricks-labs/databricks-agents) for current implementation details.
4+
This page is under development. See the [source code](https://github.com/databricks-labs/dbx-agent-app) for current implementation details.
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
11
# UCAgentRegistry
22

33
!!! info "Coming Soon"
4-
This page is under development. See the [source code](https://github.com/databricks-labs/databricks-agents) for current implementation details.
4+
This page is under development. See the [source code](https://github.com/databricks-labs/dbx-agent-app) for current implementation details.
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
11
# Contributing
22

33
!!! info "Coming Soon"
4-
This page is under development. See the [source code](https://github.com/databricks-labs/databricks-agents) for current implementation details.
4+
This page is under development. See the [source code](https://github.com/databricks-labs/dbx-agent-app) for current implementation details.

0 commit comments

Comments
 (0)