Skip to content

feat(useragent): include ai-agent in structured format #34

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
platinummonkey merged 1 commit into from
Feb 11, 2026
Merged

feat(useragent): include ai-agent in structured format #34

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 6 additions & 6 deletions pkg/useragent/useragent.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -22,26 +22,26 @@ import (
//
// Format with agent:
//
// pup/v0.1.0 (go go1.25.0; os darwin; arch arm64) claude-code
// pup/v0.1.0 (go go1.25.0; os darwin; arch arm64; ai-agent claude-code)
//
// AI agents are detected via environment variables:
// - CLAUDECODE=1 or CLAUDE_CODE=1 → appends "claude-code"
// - CURSOR_AGENT=true or CURSOR_AGENT=1 → appends "cursor"
// - CLAUDECODE=1 or CLAUDE_CODE=1 → adds "ai-agent claude-code"
// - CURSOR_AGENT=true or CURSOR_AGENT=1 → adds "ai-agent cursor"
//
// If multiple agents are detected, CLAUDECODE takes precedence.
func Get() string {
base := fmt.Sprintf(
"pup/%s (go %s; os %s; arch %s)",
"pup/%s (go %s; os %s; arch %s",
version.Version,
runtime.Version(),
runtime.GOOS,
runtime.GOARCH,
)

if agent := detectAgent(); agent != "" {
return base + " " + agent
return base + fmt.Sprintf("; ai-agent %s)", agent)
}
return base
return base + ")"
}

// detectAgent detects AI coding assistant from environment variables.
Expand Down
41 changes: 21 additions & 20 deletions pkg/useragent/useragent_test.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -28,9 +28,9 @@ func TestGet_NoAgent(t *testing.T) {
t.Errorf("Get() = %q, want %q", result, expected)
}

// Verify no agent suffix
if strings.Contains(result, "claude-code") || strings.Contains(result, "cursor") {
t.Errorf("Get() should not contain agent suffix, got %q", result)
// Verify no agent in output
if strings.Contains(result, "ai-agent") {
t.Errorf("Get() should not contain ai-agent, got %q", result)
}
}

Expand Down Expand Up @@ -58,13 +58,14 @@ func TestGet_WithClaudeCode(t *testing.T) {

result := Get()

// Verify suffix is present
if !strings.HasSuffix(result, " "+tt.wantSuf) {
t.Errorf("Get() = %q, want suffix %q", result, tt.wantSuf)
// Verify agent is present in structured format
expectedAgent := "; ai-agent " + tt.wantSuf + ")"
if !strings.HasSuffix(result, expectedAgent) {
t.Errorf("Get() = %q, want suffix %q", result, expectedAgent)
}

// Verify base format is still correct
expectedBase := "pup/" + version.Version + " (go " + runtime.Version() + "; os " + runtime.GOOS + "; arch " + runtime.GOARCH + ")"
expectedBase := "pup/" + version.Version + " (go " + runtime.Version() + "; os " + runtime.GOOS + "; arch " + runtime.GOARCH
if !strings.HasPrefix(result, expectedBase) {
t.Errorf("Get() = %q, want prefix %q", result, expectedBase)
}
Expand Down Expand Up @@ -95,13 +96,13 @@ func TestGet_WithCursor(t *testing.T) {

result := Get()

// Verify suffix is present
if !strings.HasSuffix(result, " cursor") {
t.Errorf("Get() = %q, want suffix 'cursor'", result)
// Verify agent is present in structured format
if !strings.HasSuffix(result, "; ai-agent cursor)") {
t.Errorf("Get() = %q, want suffix '; ai-agent cursor)'", result)
}

// Verify base format is still correct
expectedBase := "pup/" + version.Version + " (go " + runtime.Version() + "; os " + runtime.GOOS + "; arch " + runtime.GOARCH + ")"
expectedBase := "pup/" + version.Version + " (go " + runtime.Version() + "; os " + runtime.GOOS + "; arch " + runtime.GOARCH
if !strings.HasPrefix(result, expectedBase) {
t.Errorf("Get() = %q, want prefix %q", result, expectedBase)
}
Expand All @@ -124,9 +125,9 @@ func TestGet_WithMultipleAgents(t *testing.T) {

result := Get()

// Should have claude-code, not cursor
if !strings.HasSuffix(result, " claude-code") {
t.Errorf("Get() = %q, want suffix 'claude-code' (CLAUDECODE should take precedence)", result)
// Should have ai-agent claude-code, not cursor
if !strings.HasSuffix(result, "; ai-agent claude-code)") {
t.Errorf("Get() = %q, want suffix '; ai-agent claude-code)' (CLAUDECODE should take precedence)", result)
}
if strings.Contains(result, "cursor") {
t.Errorf("Get() = %q, should not contain 'cursor' when CLAUDECODE is set", result)
Expand Down Expand Up @@ -281,15 +282,15 @@ func TestGet_Format(t *testing.T) {

resultWithAgent := Get()

// Verify agent is appended with single space
expectedSuffix := " claude-code"
// Verify agent is in structured format
expectedSuffix := "; ai-agent claude-code)"
if !strings.HasSuffix(resultWithAgent, expectedSuffix) {
t.Errorf("Get() with agent should end with %q, got %q", expectedSuffix, resultWithAgent)
}

// Verify base part is unchanged
basePart := strings.TrimSuffix(resultWithAgent, expectedSuffix)
if basePart != result {
t.Errorf("Get() base part should be unchanged, got %q, want %q", basePart, result)
// Verify base part is present
expectedBase := "pup/" + version.Version + " (go " + runtime.Version() + "; os " + runtime.GOOS + "; arch " + runtime.GOARCH
if !strings.HasPrefix(resultWithAgent, expectedBase) {
t.Errorf("Get() with agent should start with %q, got %q", expectedBase, resultWithAgent)
}
}