Skip to content

Commit 02ae651

Browse files
chore(lint): fix revive, unconvert, and testifylint in image generation and tracing
- Ignore unused parameters in tests and translator (revive)\n- Remove unnecessary string conversions now that model types are strings (unconvert)\n- Use require.Empty for empty string check (testifylint)\n- Remove unused test variable in span tests (unused)\n- Keep extractUsageFromBufferEvent with lint ignore for future parity Signed-off-by: Hrushikesh Patil <[email protected]>
1 parent 9029412 commit 02ae651

File tree

7 files changed

+2
-7
lines changed

7 files changed

+2
-7
lines changed

go.mod

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,6 @@ require (
8484
github.com/AzureAD/microsoft-authentication-library-for-go v1.5.0 // indirect
8585
github.com/Microsoft/go-winio v0.6.2 // indirect
8686
github.com/NYTimes/gziphandler v1.1.1 // indirect
87-
github.com/andybalholm/brotli v1.2.0 // indirect
8887
github.com/antlr4-go/antlr/v4 v4.13.1 // indirect
8988
github.com/asaskevich/govalidator v0.0.0-20230301143203-a9d515a09cc2 // indirect
9089
github.com/avast/retry-go v3.0.0+incompatible // indirect

internal/extproc/mocks_test.go

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -525,6 +525,7 @@ func (m *mockImageGenerationMetrics) StartRequest(map[string]string) {}
525525
func (m *mockImageGenerationMetrics) SetOriginalModel(originalModel string) {
526526
m.model = originalModel
527527
}
528+
528529
func (m *mockImageGenerationMetrics) SetRequestModel(requestModel string) {
529530
m.model = requestModel
530531
}

internal/extproc/translator/translator.go

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,6 @@ import (
1818
"github.com/envoyproxy/ai-gateway/internal/apischema/openai"
1919
"github.com/envoyproxy/ai-gateway/internal/internalapi"
2020
tracing "github.com/envoyproxy/ai-gateway/internal/tracing/api"
21-
openaisdk "github.com/openai/openai-go/v2"
2221
)
2322

2423
const (

internal/tracing/api/api.go

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,6 @@ import (
1616
"go.opentelemetry.io/otel/trace"
1717

1818
"github.com/envoyproxy/ai-gateway/internal/apischema/openai"
19-
openaisdk "github.com/openai/openai-go/v2"
2019
)
2120

2221
var _ Tracing = NoopTracing{}

internal/tracing/image_generation_span.go

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,6 @@ import (
1010
"go.opentelemetry.io/otel/trace"
1111

1212
tracing "github.com/envoyproxy/ai-gateway/internal/tracing/api"
13-
openaisdk "github.com/openai/openai-go/v2"
1413
)
1514

1615
// Ensure imageGenerationSpan implements ImageGenerationSpan.

internal/tracing/image_generation_tracer.go

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,6 @@ import (
1515
"go.opentelemetry.io/otel/trace/noop"
1616

1717
tracing "github.com/envoyproxy/ai-gateway/internal/tracing/api"
18-
openaisdk "github.com/openai/openai-go/v2"
1918
)
2019

2120
// Ensure imageGenerationTracer implements ImageGenerationTracer.

internal/tracing/openinference/openai/image_generation.go

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,6 @@ import (
1717

1818
tracing "github.com/envoyproxy/ai-gateway/internal/tracing/api"
1919
"github.com/envoyproxy/ai-gateway/internal/tracing/openinference"
20-
openaisdk "github.com/openai/openai-go/v2"
2120
)
2221

2322
// ImageGenerationRecorder implements recorders for OpenInference image generation spans.
@@ -89,7 +88,7 @@ func buildImageGenerationRequestAttributes(req *openaisdk.ImageGenerateParams, b
8988
attrs := []attribute.KeyValue{
9089
attribute.String(openinference.SpanKind, openinference.SpanKindLLM),
9190
attribute.String(openinference.LLMSystem, openinference.LLMSystemOpenAI),
92-
attribute.String(openinference.LLMModelName, string(req.Model)),
91+
attribute.String(openinference.LLMModelName, req.Model),
9392
}
9493

9594
if config.HideInputs {

0 commit comments

Comments
 (0)