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