diff --git a/llmproxymetrics.go b/llmproxymetrics.go index a85e765..60eeef4 100644 --- a/llmproxymetrics.go +++ b/llmproxymetrics.go @@ -70,7 +70,9 @@ func createProxy(target *url.URL) func(http.ResponseWriter, *http.Request) { proxy := httputil.NewSingleHostReverseProxy(target) proxy.Director = director - proxy.ModifyResponse = modifyResponse + if r.URL.Path == "/api/generate" { + proxy.ModifyResponse = modifyResponse + } proxy.ServeHTTP(w, r) }