Skip to content

Commit 2bc6469

Browse files
committed
chore: lint fixes part 1 (automatic fixes)
1 parent 6f1a39d commit 2bc6469

Some content is hidden

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

66 files changed

+412
-337
lines changed

api.go

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,14 +3,15 @@ package aibridge
33
import (
44
"context"
55

6+
"github.com/prometheus/client_golang/prometheus"
7+
"go.opentelemetry.io/otel/trace"
8+
69
"cdr.dev/slog/v3"
710
"github.com/coder/aibridge/config"
811
aibcontext "github.com/coder/aibridge/context"
912
"github.com/coder/aibridge/metrics"
1013
"github.com/coder/aibridge/provider"
1114
"github.com/coder/aibridge/recorder"
12-
"github.com/prometheus/client_golang/prometheus"
13-
"go.opentelemetry.io/otel/trace"
1415
)
1516

1617
// Const + Type + function aliases for backwards compatibility.

bridge.go

Lines changed: 13 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,12 @@ import (
1111
"sync/atomic"
1212
"time"
1313

14+
"github.com/hashicorp/go-multierror"
15+
"github.com/sony/gobreaker/v2"
16+
"go.opentelemetry.io/otel/codes"
17+
"go.opentelemetry.io/otel/trace"
18+
"golang.org/x/xerrors"
19+
1420
"cdr.dev/slog/v3"
1521
"github.com/coder/aibridge/circuitbreaker"
1622
aibcontext "github.com/coder/aibridge/context"
@@ -19,10 +25,6 @@ import (
1925
"github.com/coder/aibridge/provider"
2026
"github.com/coder/aibridge/recorder"
2127
"github.com/coder/aibridge/tracing"
22-
"github.com/hashicorp/go-multierror"
23-
"github.com/sony/gobreaker/v2"
24-
"go.opentelemetry.io/otel/codes"
25-
"go.opentelemetry.io/otel/trace"
2628
)
2729

2830
const (
@@ -67,10 +69,10 @@ func validateProviders(providers []provider.Provider) error {
6769
for _, prov := range providers {
6870
name := prov.Name()
6971
if !validProviderName.MatchString(name) {
70-
return fmt.Errorf("invalid provider name %q: must contain only lowercase alphanumeric characters and hyphens", name)
72+
return xerrors.Errorf("invalid provider name %q: must contain only lowercase alphanumeric characters and hyphens", name)
7173
}
7274
if names[name] {
73-
return fmt.Errorf("duplicate provider name: %q", name)
75+
return xerrors.Errorf("duplicate provider name: %q", name)
7476
}
7577
names[name] = true
7678
}
@@ -125,7 +127,7 @@ func NewRequestBridge(ctx context.Context, providers []provider.Provider, rec re
125127
slog.F("prefix", prov.RoutePrefix()),
126128
slog.F("path", path),
127129
)
128-
return nil, fmt.Errorf("failed to configure provider '%v': failed to join bridged path: %w", providerName, err)
130+
return nil, xerrors.Errorf("failed to configure provider '%v': failed to join bridged path: %w", providerName, err)
129131
}
130132
mux.Handle(route, handler)
131133
}
@@ -144,7 +146,7 @@ func NewRequestBridge(ctx context.Context, providers []provider.Provider, rec re
144146
slog.F("prefix", prov.RoutePrefix()),
145147
slog.F("path", path),
146148
)
147-
return nil, fmt.Errorf("failed to configure provider '%v': failed to join passed through path: %w", providerName, err)
149+
return nil, xerrors.Errorf("failed to configure provider '%v': failed to join passed through path: %w", providerName, err)
148150
}
149151
mux.Handle(route, http.StripPrefix(prov.RoutePrefix(), ftr))
150152
}
@@ -325,7 +327,7 @@ func (b *RequestBridge) Shutdown(ctx context.Context) error {
325327
select {
326328
case <-ctx.Done():
327329
// Cancel all inflight requests, if any are still running.
328-
b.logger.Debug(ctx, "shutdown context canceled; cancelling inflight requests", slog.Error(ctx.Err()))
330+
b.logger.Debug(ctx, "shutdown context canceled; canceling inflight requests", slog.Error(ctx.Err()))
329331
b.inflightCancel()
330332
<-done
331333
err = ctx.Err()
@@ -347,8 +349,8 @@ func (b *RequestBridge) InflightRequests() int32 {
347349
return b.inflightReqs.Load()
348350
}
349351

350-
// mergeContexts merges two contexts together, so that if either is cancelled
351-
// the returned context is cancelled. The context values will only be used from
352+
// mergeContexts merges two contexts together, so that if either is canceled
353+
// the returned context is canceled. The context values will only be used from
352354
// the first context.
353355
func mergeContexts(base, other context.Context) context.Context {
354356
ctx, cancel := context.WithCancel(base)

bridge_test.go

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,12 +5,13 @@ import (
55
"net/http/httptest"
66
"testing"
77

8+
"github.com/stretchr/testify/assert"
9+
"github.com/stretchr/testify/require"
10+
811
"cdr.dev/slog/v3/sloggers/slogtest"
912
"github.com/coder/aibridge/config"
1013
"github.com/coder/aibridge/internal/testutil"
1114
"github.com/coder/aibridge/provider"
12-
"github.com/stretchr/testify/assert"
13-
"github.com/stretchr/testify/require"
1415
)
1516

1617
func TestValidateProvider_Names(t *testing.T) {

buildinfo/buildinfo_test.go

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,9 @@ package buildinfo_test
33
import (
44
"testing"
55

6-
"github.com/coder/aibridge/buildinfo"
76
"github.com/stretchr/testify/assert"
7+
8+
"github.com/coder/aibridge/buildinfo"
89
)
910

1011
func TestBuildInfo(t *testing.T) {

circuitbreaker/circuitbreaker.go

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,14 +7,16 @@ import (
77
"sync"
88
"time"
99

10+
"github.com/sony/gobreaker/v2"
11+
"golang.org/x/xerrors"
12+
1013
"github.com/coder/aibridge/config"
1114
"github.com/coder/aibridge/metrics"
12-
"github.com/sony/gobreaker/v2"
1315
)
1416

1517
// ErrCircuitOpen is returned by Execute when the circuit breaker is open
1618
// and the request was rejected without calling the handler.
17-
var ErrCircuitOpen = errors.New("circuit breaker is open")
19+
var ErrCircuitOpen = xerrors.New("circuit breaker is open")
1820

1921
// DefaultIsFailure returns true for standard HTTP status codes that typically
2022
// indicate upstream overload.
@@ -153,7 +155,7 @@ func (p *ProviderCircuitBreakers) Execute(endpoint, model string, w http.Respons
153155
_, err := cb.Execute(func() (struct{}, error) {
154156
handlerErr = handler(sw)
155157
if p.isFailure(sw.statusCode) {
156-
return struct{}{}, fmt.Errorf("upstream error: %d", sw.statusCode)
158+
return struct{}{}, xerrors.Errorf("upstream error: %d", sw.statusCode)
157159
}
158160
return struct{}{}, nil
159161
})

circuitbreaker/circuitbreaker_test.go

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,9 +8,10 @@ import (
88
"testing"
99
"time"
1010

11-
"github.com/coder/aibridge/config"
1211
"github.com/sony/gobreaker/v2"
1312
"github.com/stretchr/testify/assert"
13+
14+
"github.com/coder/aibridge/config"
1415
)
1516

1617
func TestExecute_PerModelIsolation(t *testing.T) {

intercept/actor_headers.go

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,8 +5,9 @@ import (
55
"strings"
66

77
ant_option "github.com/anthropics/anthropic-sdk-go/option"
8-
"github.com/coder/aibridge/context"
98
oai_option "github.com/openai/openai-go/v3/option"
9+
10+
"github.com/coder/aibridge/context"
1011
)
1112

1213
const (

intercept/actor_headers_test.go

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,10 +3,11 @@ package intercept
33
import (
44
"testing"
55

6-
"github.com/coder/aibridge/context"
7-
"github.com/coder/aibridge/recorder"
86
"github.com/google/uuid"
97
"github.com/stretchr/testify/require"
8+
9+
"github.com/coder/aibridge/context"
10+
"github.com/coder/aibridge/recorder"
1011
)
1112

1213
func TestNilActor(t *testing.T) {

intercept/apidump/apidump.go

Lines changed: 14 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -12,11 +12,14 @@ import (
1212
"slices"
1313
"strings"
1414

15+
"golang.org/x/xerrors"
16+
1517
"cdr.dev/slog/v3"
1618

17-
"github.com/coder/quartz"
1819
"github.com/google/uuid"
1920
"github.com/tidwall/pretty"
21+
22+
"github.com/coder/quartz"
2023
)
2124

2225
const (
@@ -71,7 +74,7 @@ type dumper struct {
7174
func (d *dumper) dumpRequest(req *http.Request) error {
7275
dumpPath := d.dumpPath + SuffixRequest
7376
if err := os.MkdirAll(filepath.Dir(dumpPath), 0o755); err != nil {
74-
return fmt.Errorf("create dump dir: %w", err)
77+
return xerrors.Errorf("create dump dir: %w", err)
7578
}
7679

7780
// Read and restore body
@@ -80,7 +83,7 @@ func (d *dumper) dumpRequest(req *http.Request) error {
8083
var err error
8184
bodyBytes, err = io.ReadAll(req.Body)
8285
if err != nil {
83-
return fmt.Errorf("read request body: %w", err)
86+
return xerrors.Errorf("read request body: %w", err)
8487
}
8588
req.Body = io.NopCloser(bytes.NewReader(bodyBytes))
8689
}
@@ -91,18 +94,18 @@ func (d *dumper) dumpRequest(req *http.Request) error {
9194
var buf bytes.Buffer
9295
_, err := fmt.Fprintf(&buf, "%s %s %s\r\n", req.Method, req.URL.RequestURI(), req.Proto)
9396
if err != nil {
94-
return fmt.Errorf("write request uri: %w", err)
97+
return xerrors.Errorf("write request uri: %w", err)
9598
}
9699
err = d.writeRedactedHeaders(&buf, req.Header, sensitiveRequestHeaders, map[string]string{
97100
"Content-Length": fmt.Sprintf("%d", len(prettyBody)),
98101
})
99102
if err != nil {
100-
return fmt.Errorf("write request headers: %w", err)
103+
return xerrors.Errorf("write request headers: %w", err)
101104
}
102105

103106
_, err = fmt.Fprintf(&buf, "\r\n")
104107
if err != nil {
105-
return fmt.Errorf("write request header terminator: %w", err)
108+
return xerrors.Errorf("write request header terminator: %w", err)
106109
}
107110
buf.Write(prettyBody)
108111
buf.WriteByte('\n')
@@ -117,15 +120,15 @@ func (d *dumper) dumpResponse(resp *http.Response) error {
117120
var headerBuf bytes.Buffer
118121
_, err := fmt.Fprintf(&headerBuf, "%s %s\r\n", resp.Proto, resp.Status)
119122
if err != nil {
120-
return fmt.Errorf("write response status: %w", err)
123+
return xerrors.Errorf("write response status: %w", err)
121124
}
122125
err = d.writeRedactedHeaders(&headerBuf, resp.Header, sensitiveResponseHeaders, nil)
123126
if err != nil {
124-
return fmt.Errorf("write response headers: %w", err)
127+
return xerrors.Errorf("write response headers: %w", err)
125128
}
126129
_, err = fmt.Fprintf(&headerBuf, "\r\n")
127130
if err != nil {
128-
return fmt.Errorf("write response header terminator: %w", err)
131+
return xerrors.Errorf("write response header terminator: %w", err)
129132
}
130133

131134
// Wrap the response body to capture it as it streams
@@ -175,7 +178,7 @@ func (d *dumper) writeRedactedHeaders(w io.Writer, headers http.Header, sensitiv
175178
if override, ok := overrides[key]; ok {
176179
_, err := fmt.Fprintf(w, "%s: %s\r\n", key, override)
177180
if err != nil {
178-
return fmt.Errorf("write response header override: %w", err)
181+
return xerrors.Errorf("write response header override: %w", err)
179182
}
180183
}
181184
continue
@@ -190,7 +193,7 @@ func (d *dumper) writeRedactedHeaders(w io.Writer, headers http.Header, sensitiv
190193
}
191194
_, err := fmt.Fprintf(w, "%s: %s\r\n", key, value)
192195
if err != nil {
193-
return fmt.Errorf("write response headers: %w", err)
196+
return xerrors.Errorf("write response headers: %w", err)
194197
}
195198
}
196199
}

intercept/apidump/apidump_test.go

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,11 +10,12 @@ import (
1010
"strings"
1111
"testing"
1212

13+
"github.com/google/uuid"
14+
"github.com/stretchr/testify/require"
15+
1316
"cdr.dev/slog/v3"
1417
"cdr.dev/slog/v3/sloggers/slogtest"
1518
"github.com/coder/quartz"
16-
"github.com/google/uuid"
17-
"github.com/stretchr/testify/require"
1819
)
1920

2021
// findDumpFile finds a dump file matching the pattern in the given directory.

0 commit comments

Comments
 (0)