refactor(amp): consolidate and simplify model mapping debug logs

This commit is contained in:
hkfires
2025-12-06 14:54:38 +08:00
parent 5d12d4ce33
commit 22579155c5
3 changed files with 2 additions and 12 deletions

View File

@@ -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)

View File

@@ -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
}

View File

@@ -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
}