diff --git a/llmproxymetrics.go b/llmproxymetrics.go index 3150bd1..52b1643 100644 --- a/llmproxymetrics.go +++ b/llmproxymetrics.go @@ -42,10 +42,12 @@ func createProxy(target *url.URL) func(http.ResponseWriter, *http.Request) { var jsonData map[string]interface{} json.NewDecoder(r.Body).Decode(&jsonData) - if jsonData["stream"].(bool) == false { - notStreamed.Inc() - } else { - streamed.Inc() + if jsonData["stream"] != nil { + if !jsonData["stream"].(bool) { + notStreamed.Inc() + } else { + streamed.Inc() + } } modifyResponse := func(response *http.Response) error {