🐛 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 |
|