Uh oh!
There was an error while loading. Please reload this page.
Conversation
Adds libs/agent package to detect agents (Claude Code, Gemini CLI, Cursor) via environment variables. Used in user agent tracking and available for telemetry and conditional behavior. Detection only succeeds when exactly one agent environment variable is set, preventing ambiguous detection in environments with multiple agents. Implementation: - libs/agent: Detection logic with Detect(), Product(), and Mock() - cmd/root: Integrated agent detection into command execution - Test config: Clear agent env vars in acceptance tests to prevent interference with test outputs Environment variable sources: - CLAUDECODE=1: anthropics/claude-code#531 - GEMINI_CLI=1: https://github.com/google-gemini/gemini-cli/blob/main/docs/cli/commands.md - CURSOR_AGENT=1: https://forum.cursor.com/t/cursor-cli-is-not-setting-cursor-agent-1-environment-variable-while-executing-bash-commands/132427 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
pietern
requested review from
andrewnester, anton-107, denik and shreyas-goenka
as code ownersJanuary 15, 2026 10:30
pieterntemporarily deployed
to
test-trigger-is
January 15, 2026 10:31 — with 
GitHub Actions
Inactive
Collaborator
Commit: c299cef
21 interesting tests: 10 KNOWN, 8 RECOVERED, 2 flaky, 1 SKIP
Top 48 slowest tests (at least 2 minutes):
|
lennartkats-db
approved these changes
Jan 15, 2026
| claudeCodeEnvVar = "CLAUDECODE" | ||
| geminiCliEnvVar = "GEMINI_CLI" | ||
| cursorAgentEnvVar = "CURSOR_AGENT" | ||
| ) |
Contributor
There was a problem hiding this comment.
As discussed offline, Codex seems to set CODEX_SANDBOX.
ContributorAuthor
There was a problem hiding this comment.
I'll submit a separate PR for this.
fjakobs
approved these changes
Jan 15, 2026
Uh oh!
There was an error while loading. Please reload this page.
pietern added a commit
that referenced
this pull request
Jan 19, 2026
This packages was added in #4287 and centralizes agent detection.
eng-dev-ecosystem-bot
commented
Jan 19, 2026
Collaborator
Commit: 48d2d47
72 interesting tests: 26 FAIL, 23 KNOWN, 22 BUG, 1 SKIP
Top 50 slowest tests (at least 2 minutes):
|
pietern added a commit
that referenced
this pull request
Jan 19, 2026
## Why This packages was added in #4287 and centralizes agent detection. ## Tests Manually confirmed install with `CLAUDECODE=1` and `CURSOR_AGENT=1`.
vikrantpuppala added a commit
to vikrantpuppala/databricks-jdbc
that referenced
this pull request
Feb 24, 2026
Detect when the JDBC driver is invoked by an AI coding agent (e.g. Claude Code, Cursor, Gemini CLI) by checking well-known environment variables, and append `agent/<product>` to the User-Agent string. This enables Databricks to understand how much driver usage originates from AI coding agents. Detection only succeeds when exactly one agent is detected to avoid ambiguous attribution. Mirrors the approach in databricks/cli#4287. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
Merged
3 tasks
vikrantpuppala added a commit
to vikrantpuppala/databricks-sql-python
that referenced
this pull request
Feb 24, 2026
Detect when the Python SQL connector is invoked by an AI coding agent (e.g. Claude Code, Cursor, Gemini CLI) by checking well-known environment variables, and append `agent/<product>` to the User-Agent string. This enables Databricks to understand how much driver usage originates from AI coding agents. Detection only succeeds when exactly one agent is detected to avoid ambiguous attribution. Mirrors the approach in databricks/cli#4287. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
3 tasks
vikrantpuppala added a commit
to vikrantpuppala/databricks-sql-go
that referenced
this pull request
Feb 24, 2026
Detect when the Go SQL driver is invoked by an AI coding agent (e.g. Claude Code, Cursor, Gemini CLI) by checking well-known environment variables, and append `agent/<product>` to the User-Agent string. This enables Databricks to understand how much driver usage originates from AI coding agents. Detection only succeeds when exactly one agent is detected to avoid ambiguous attribution. Mirrors the approach in databricks/cli#4287. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
Merged
4 tasks
vikrantpuppala added a commit
to vikrantpuppala/databricks-sql-nodejs
that referenced
this pull request
Feb 24, 2026
Detect when the Node.js SQL driver is invoked by an AI coding agent (e.g. Claude Code, Cursor, Gemini CLI) by checking well-known environment variables, and append `agent/<product>` to the User-Agent string. This enables Databricks to understand how much driver usage originates from AI coding agents. Detection only succeeds when exactly one agent is detected to avoid ambiguous attribution. Mirrors the approach in databricks/cli#4287. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
4 tasks
vikrantpuppala added a commit
to vikrantpuppala/databricks-jdbc
that referenced
this pull request
Feb 24, 2026
Detect when the JDBC driver is invoked by an AI coding agent (e.g. Claude Code, Cursor, Gemini CLI) by checking well-known environment variables, and append `agent/<product>` to the User-Agent string. This enables Databricks to understand how much driver usage originates from AI coding agents. Detection only succeeds when exactly one agent is detected to avoid ambiguous attribution. Mirrors the approach in databricks/cli#4287. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com> Signed-off-by: Vikrant Puppala <vikrant.puppala@databricks.com>
vikrantpuppala added a commit
to vikrantpuppala/databricks-sql-python
that referenced
this pull request
Feb 24, 2026
Detect when the Python SQL connector is invoked by an AI coding agent (e.g. Claude Code, Cursor, Gemini CLI) by checking well-known environment variables, and append `agent/<product>` to the User-Agent string. This enables Databricks to understand how much driver usage originates from AI coding agents. Detection only succeeds when exactly one agent is detected to avoid ambiguous attribution. Mirrors the approach in databricks/cli#4287. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com> Signed-off-by: Vikrant Puppala <vikrant.puppala@databricks.com>
vikrantpuppala added a commit
to vikrantpuppala/databricks-sql-go
that referenced
this pull request
Feb 24, 2026
Detect when the Go SQL driver is invoked by an AI coding agent (e.g. Claude Code, Cursor, Gemini CLI) by checking well-known environment variables, and append `agent/<product>` to the User-Agent string. This enables Databricks to understand how much driver usage originates from AI coding agents. Detection only succeeds when exactly one agent is detected to avoid ambiguous attribution. Mirrors the approach in databricks/cli#4287. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com> Signed-off-by: Vikrant Puppala <vikrant.puppala@databricks.com>
vikrantpuppala added a commit
to vikrantpuppala/databricks-sql-nodejs
that referenced
this pull request
Feb 24, 2026
Detect when the Node.js SQL driver is invoked by an AI coding agent (e.g. Claude Code, Cursor, Gemini CLI) by checking well-known environment variables, and append `agent/<product>` to the User-Agent string. This enables Databricks to understand how much driver usage originates from AI coding agents. Detection only succeeds when exactly one agent is detected to avoid ambiguous attribution. Mirrors the approach in databricks/cli#4287. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com> Signed-off-by: Vikrant Puppala <vikrant.puppala@databricks.com>
vikrantpuppala added a commit
to vikrantpuppala/databricks-jdbc
that referenced
this pull request
Apr 13, 2026
Detect when the JDBC driver is invoked by an AI coding agent (e.g. Claude Code, Cursor, Gemini CLI) by checking well-known environment variables, and append `agent/<product>` to the User-Agent string. This enables Databricks to understand how much driver usage originates from AI coding agents. Detection only succeeds when exactly one agent is detected to avoid ambiguous attribution. Mirrors the approach in databricks/cli#4287. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com> Signed-off-by: Vikrant Puppala <vikrant.puppala@databricks.com>
vikrantpuppala added a commit
to databricks/databricks-sql-python
that referenced
this pull request
Apr 13, 2026
Detect when the Python SQL connector is invoked by an AI coding agent (e.g. Claude Code, Cursor, Gemini CLI) by checking well-known environment variables, and append `agent/<product>` to the User-Agent string. This enables Databricks to understand how much driver usage originates from AI coding agents. Detection only succeeds when exactly one agent is detected to avoid ambiguous attribution. Mirrors the approach in databricks/cli#4287. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com> Signed-off-by: Vikrant Puppala <vikrant.puppala@databricks.com>
vikrantpuppala added a commit
to vikrantpuppala/databricks-sql-go
that referenced
this pull request
Apr 13, 2026
Detect when the Go SQL driver is invoked by an AI coding agent (e.g. Claude Code, Cursor, Gemini CLI) by checking well-known environment variables, and append `agent/<product>` to the User-Agent string. This enables Databricks to understand how much driver usage originates from AI coding agents. Detection only succeeds when exactly one agent is detected to avoid ambiguous attribution. Mirrors the approach in databricks/cli#4287. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com> Signed-off-by: Vikrant Puppala <vikrant.puppala@databricks.com>
vikrantpuppala added a commit
to databricks/databricks-sql-nodejs
that referenced
this pull request
Apr 13, 2026
Detect when the Node.js SQL driver is invoked by an AI coding agent (e.g. Claude Code, Cursor, Gemini CLI) by checking well-known environment variables, and append `agent/<product>` to the User-Agent string. This enables Databricks to understand how much driver usage originates from AI coding agents. Detection only succeeds when exactly one agent is detected to avoid ambiguous attribution. Mirrors the approach in databricks/cli#4287. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com> Signed-off-by: Vikrant Puppala <vikrant.puppala@databricks.com>
vikrantpuppala added a commit
to databricks/databricks-sql-go
that referenced
this pull request
Apr 13, 2026
## Summary - Adds `internal/agent` package that detects 7 AI coding agents (Claude Code, Cursor, Gemini CLI, Cline, Codex, OpenCode, Antigravity) by checking well-known environment variables they set in spawned shell processes - Integrates detection into `InitThriftClient` to append `agent/<product>` to the User-Agent header - Uses exactly-one detection rule: if zero or multiple agent env vars are set, no agent is attributed (avoids ambiguity) ## Approach Mirrors the implementation in [databricks/cli#4287](databricks/cli#4287) and aligns with the latest agent list in [`libs/agent/agent.go`](https://github.com/databricks/cli/blob/main/libs/agent/agent.go#L35). | Agent | Product String | Environment Variable | |-------|---------------|---------------------| | Google Antigravity | `antigravity` | `ANTIGRAVITY_AGENT` | | Claude Code | `claude-code` | `CLAUDECODE` | | Cline | `cline` | `CLINE_ACTIVE` | | OpenAI Codex | `codex` | `CODEX_CI` | | Cursor | `cursor` | `CURSOR_AGENT` | | Gemini CLI | `gemini-cli` | `GEMINI_CLI` | | OpenCode | `opencode` | `OPENCODE` | Adding a new agent requires only a new constant and a new entry in `knownAgents`. ## Changes - **New**: `internal/agent/agent.go` — environment-variable-based agent detection with injectable env lookup for testability - **New**: `internal/agent/agent_test.go` — 11 test cases covering all agents, no agent, multiple agents, empty values, and real `os.Getenv` - **Modified**: `internal/client/client.go` — calls `agent.Detect()` when building User-Agent in `InitThriftClient` ## Test plan - [x] `internal/agent` — 11 tests pass - [x] `internal/client` — all existing tests continue to pass - [x] Manual: verified User-Agent contains `agent/claude-code` when run from Claude Code via `GODEBUG=http2debug=2` ``` http2: Transport encoding header "user-agent" = "godatabrickssqlconnector/1.10.0 agent/claude-code" ``` - [x] Executed `SELECT 1` successfully against dogfood warehouse with the new header 🤖 Generated with [Claude Code](https://claude.com/claude-code) Signed-off-by: Vikrant Puppala <vikrant.puppala@databricks.com> Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>
vikrantpuppala added a commit
to databricks/databricks-jdbc
that referenced
this pull request
Apr 13, 2026
) ## Summary - Adds `AgentDetector` utility class that detects 7 AI coding agents (Claude Code, Cursor, Gemini CLI, Cline, Codex, OpenCode, Antigravity) by checking well-known environment variables they set in spawned shell processes - Integrates detection into `UserAgentManager` to append `agent/<product>` to the User-Agent header on both SDK-based and connector-service UA paths - Uses exactly-one detection rule: if zero or multiple agent env vars are set, no agent is attributed (avoids ambiguity) ## Approach Mirrors the implementation in [databricks/cli#4287](databricks/cli#4287) and aligns with the latest agent list in [`libs/agent/agent.go`](https://github.com/databricks/cli/blob/main/libs/agent/agent.go#L35). | Agent | Product String | Environment Variable | |-------|---------------|---------------------| | Google Antigravity | `antigravity` | `ANTIGRAVITY_AGENT` | | Claude Code | `claude-code` | `CLAUDECODE` | | Cline | `cline` | `CLINE_ACTIVE` | | OpenAI Codex | `codex` | `CODEX_CI` | | Cursor | `cursor` | `CURSOR_AGENT` | | Gemini CLI | `gemini-cli` | `GEMINI_CLI` | | OpenCode | `opencode` | `OPENCODE` | Adding a new agent requires only a new constant and a new entry in the `KNOWN_AGENTS` array. ## Changes - **New**: `AgentDetector.java` — environment-variable-based agent detection with injectable env lookup for testability - **Modified**: `UserAgentManager.java` — calls `AgentDetector.detect()` in both `setUserAgent()` and `buildUserAgentForConnectorService()` - **New**: `AgentDetectorTest.java` — 12 test cases covering single agent, no agent, multiple agents, empty values, null values, and full coverage of all known agents ## Test plan - [x] `AgentDetectorTest` — 12 tests pass (single agent detection for all 7 agents, no agent, multiple agents, empty/null values) - [x] `UserAgentManagerTest` — all 7 existing tests continue to pass - [x] Manual: verify User-Agent header contains `agent/claude-code` when run from Claude Code 🤖 Generated with [Claude Code](https://claude.com/claude-code) Signed-off-by: Vikrant Puppala <vikrant.puppala@databricks.com> Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>
vikrantpuppala added a commit
to databricks/databricks-sql-nodejs
that referenced
this pull request
Apr 13, 2026
Add AI coding agent detection to User-Agent header Detect when the Node.js SQL driver is invoked by an AI coding agent (e.g. Claude Code, Cursor, Gemini CLI) by checking well-known environment variables, and append `agent/<product>` to the User-Agent string. This enables Databricks to understand how much driver usage originates from AI coding agents. Detection only succeeds when exactly one agent is detected to avoid ambiguous attribution. Mirrors the approach in databricks/cli#4287. Signed-off-by: Vikrant Puppala <vikrant.puppala@databricks.com> Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>
vikrantpuppala added a commit
to databricks/databricks-sql-python
that referenced
this pull request
Apr 13, 2026
Detect when the Python SQL connector is invoked by an AI coding agent (e.g. Claude Code, Cursor, Gemini CLI) by checking well-known environment variables, and append `agent/<product>` to the User-Agent string. This enables Databricks to understand how much driver usage originates from AI coding agents. Detection only succeeds when exactly one agent is detected to avoid ambiguous attribution. Mirrors the approach in databricks/cli#4287. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com> Signed-off-by: Vikrant Puppala <vikrant.puppala@databricks.com>
vikrantpuppala added a commit
to databricks/databricks-sql-python
that referenced
this pull request
Apr 13, 2026
Add AI coding agent detection to User-Agent header Detect when the Python SQL connector is invoked by an AI coding agent (e.g. Claude Code, Cursor, Gemini CLI) by checking well-known environment variables, and append `agent/<product>` to the User-Agent string. This enables Databricks to understand how much driver usage originates from AI coding agents. Detection only succeeds when exactly one agent is detected to avoid ambiguous attribution. Mirrors the approach in databricks/cli#4287. Signed-off-by: Vikrant Puppala <vikrant.puppala@databricks.com> Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>
jiabin-hu pushed a commit
to jiabin-hu/databricks-sql-nodejs
that referenced
this pull request
Apr 28, 2026
…tabricks#333) Add AI coding agent detection to User-Agent header Detect when the Node.js SQL driver is invoked by an AI coding agent (e.g. Claude Code, Cursor, Gemini CLI) by checking well-known environment variables, and append `agent/<product>` to the User-Agent string. This enables Databricks to understand how much driver usage originates from AI coding agents. Detection only succeeds when exactly one agent is detected to avoid ambiguous attribution. Mirrors the approach in databricks/cli#4287. Signed-off-by: Vikrant Puppala <vikrant.puppala@databricks.com> Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com> Signed-off-by: Jiabin Hu <jiabin.hu@databricks.com>
denik pushed a commit
that referenced
this pull request
May 20, 2026
## Changes Adds libs/agent package to detect AI agents (Claude Code, Gemini CLI, Cursor) via environment variables. Used in user agent tracking and available for telemetry and (future, optional) conditional behavior. Detection only succeeds when exactly one agent environment variable is set, preventing ambiguous detection in environments with multiple agents. Implementation: - libs/agent: Detection logic with Detect(), Product(), and Mock() - cmd/root: Integrated agent detection into command execution Environment variable sources: - CLAUDECODE=1: anthropics/claude-code#531 - GEMINI_CLI=1: https://github.com/google-gemini/gemini-cli/blob/main/docs/cli/commands.md - CURSOR_AGENT=1: https://forum.cursor.com/t/cursor-cli-is-not-setting-cursor-agent-1-environment-variable-while-executing-bash-commands/132427 Co-authored-by: Claude Sonnet 4.5 <noreply@anthropic.com>
denik pushed a commit
that referenced
this pull request
May 20, 2026
## Why This packages was added in #4287 and centralizes agent detection. ## Tests Manually confirmed install with `CLAUDECODE=1` and `CURSOR_AGENT=1`.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for freeto join this conversation on GitHub.
Already have an account?
Sign in to comment
4 participants
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Changes
Adds libs/agent package to detect AI agents (Claude Code, Gemini CLI, Cursor) via environment variables. Used in user agent tracking and available for telemetry and (future, optional) conditional behavior.
Detection only succeeds when exactly one agent environment variable is set, preventing ambiguous detection in environments with multiple agents.
Implementation:
Environment variable sources: