diff --git a/internal/api/modules/amp/fallback_handlers.go b/internal/api/modules/amp/fallback_handlers.go index 46c3d3d9..d43ea3c1 100644 --- a/internal/api/modules/amp/fallback_handlers.go +++ b/internal/api/modules/amp/fallback_handlers.go @@ -180,6 +180,7 @@ func (fh *FallbackHandler) WrapHandler(handler gin.HandlerFunc) gin.HandlerFunc if usedMapping { // Log: Model was mapped to another model + log.Debugf("amp model mapping: request %s -> %s", normalizedModel, resolvedModel) logAmpRouting(RouteTypeModelMapping, modelName, resolvedModel, providerName, requestPath) rewriter := NewResponseRewriter(c.Writer, normalizedModel) c.Writer = rewriter @@ -188,7 +189,7 @@ func (fh *FallbackHandler) WrapHandler(handler gin.HandlerFunc) gin.HandlerFunc c.Request.Body = io.NopCloser(bytes.NewReader(bodyBytes)) handler(c) rewriter.Flush() - log.Debugf("amp response rewriter: rewrote model %s -> %s in response", resolvedModel, normalizedModel) + log.Debugf("amp model mapping: response %s -> %s", resolvedModel, normalizedModel) } else if len(providers) > 0 { // Log: Using local provider (free) logAmpRouting(RouteTypeLocalProvider, modelName, resolvedModel, providerName, requestPath) diff --git a/internal/api/modules/amp/model_mapping.go b/internal/api/modules/amp/model_mapping.go index c07f41c4..87384a80 100644 --- a/internal/api/modules/amp/model_mapping.go +++ b/internal/api/modules/amp/model_mapping.go @@ -66,7 +66,6 @@ func (m *DefaultModelMapper) MapModel(requestedModel string) string { } // Note: Detailed routing log is handled by logAmpRouting in fallback_handlers.go - log.Debugf("amp model mapping: resolved %s -> %s", requestedModel, targetModel) return targetModel } diff --git a/internal/api/modules/amp/response_rewriter.go b/internal/api/modules/amp/response_rewriter.go index 5be36078..e906f143 100644 --- a/internal/api/modules/amp/response_rewriter.go +++ b/internal/api/modules/amp/response_rewriter.go @@ -1,9 +1,7 @@ package amp import ( - "bufio" "bytes" - "net" "net/http" "strings" @@ -98,11 +96,3 @@ func (rw *ResponseRewriter) rewriteStreamChunk(chunk []byte) []byte { return bytes.Join(lines, []byte("\n")) } - -// Hijack implements http.Hijacker for WebSocket support -func (rw *ResponseRewriter) Hijack() (net.Conn, *bufio.ReadWriter, error) { - if hijacker, ok := rw.ResponseWriter.(http.Hijacker); ok { - return hijacker.Hijack() - } - return nil, nil, http.ErrNotSupported -}