🐛 Bug: Fix the bug of middleware statistics for unknown models
Browse files
main.py
CHANGED
@@ -127,7 +127,8 @@ class StatsMiddleware(BaseHTTPMiddleware):
|
|
127 |
self.request_times[endpoint] += process_time
|
128 |
self.ip_counts[endpoint][client_ip] += 1
|
129 |
self.request_arrivals[endpoint].append(arrival_time)
|
130 |
-
|
|
|
131 |
|
132 |
return response
|
133 |
|
|
|
127 |
self.request_times[endpoint] += process_time
|
128 |
self.ip_counts[endpoint][client_ip] += 1
|
129 |
self.request_arrivals[endpoint].append(arrival_time)
|
130 |
+
if model != "unknown":
|
131 |
+
self.model_counts[model] += 1
|
132 |
|
133 |
return response
|
134 |
|